add_noise_to_field Namespace Reference

Functions
None	plot_my_hist (datum)
None	generate_file (input_fileName, output, xSpace=73, ySpace=128, zSpace=1214, step=2.5, args=None)

Variables
argparse	parser = argparse.ArgumentParser(description="Process field.")
	default
	type
	str
	action
	False
	float
	int
argparse	args = parser.parse_args()
f	first_line = f.readline().strip().split()
	xSpace
	ySpace
	zSpace
	step

Function Documentation

generate_file()

None add_noise_to_field.generate_file	(		input_fileName,
			output,
			xSpace = 73,
			ySpace = 128,
			zSpace = 1214,
			step = 2.5,
			args = None
	)

Definition at line 23 of file add_noise_to_field.py.

def generate_file(input_fileName, output, xSpace=73, ySpace=128, zSpace=1214, step=2.5, args=None) -> None:
    # (min, max, max/stepSize + 1)  in case of Z: (0, nSteps*2.5 - 2.5, nSteps)
    field = pd.read_csv(input_fileName, skiprows=1, sep=r"\s+", names=["x", "y", "z", "bx", "by", "bz"])
 
    field_mask = field.copy()
    field_mask[["bx", "by", "bz"]] = field_mask[["bx", "by", "bz"]] != 0
 
    field_new = field.copy()
 
    if args.sidesOnly:
        temp_by = np.array(field_new["by"]).reshape([xSpace, ySpace, zSpace])
        temp_by = gaussian_filter(temp_by, sigma=args.sigma)
        field_new["by"] = temp_by.reshape(-1)
        field_new["by"] = field_new["by"] * field_mask["by"]
        rezult = field_new
    else:
        field_new[["bx", "by", "bz"]] = 0
        index_range = np.random.choice(field_new.index, size=args.nCores)
        field_new.loc[index_range, "by"] = random.uniform(-args.peak, args.peak)
        temp_by = np.array(field_new["by"]).reshape([xSpace, ySpace, zSpace])
        temp_by = gaussian_filter(temp_by, sigma=args.sigma)
        field_new["by"] = temp_by.reshape(-1)
        field_new["by"] = field_new["by"] / (field_new["by"].abs().max())  # *field_mask['by']
        field_new["by"] = field_new["by"] * field_mask["by"]
        rezult = field.copy()
        rezult["by"] = rezult["by"] + rezult["by"] * field_new["by"] * args.fraction
 
    # plot_my_hist(field_mask)
    # plot_my_hist(field)
    # plot_my_hist(field_new)
    # plot_my_hist(rezult)
    rezult.to_csv(output, sep="\t", header=None, index=None)
 
 

plot_my_hist()

None add_noise_to_field.plot_my_hist

(

datum

)

Definition at line 14 of file add_noise_to_field.py.

def plot_my_hist(datum) -> None:
    plotData = datum[datum["y"] == 0]
    H, _xedges, _yedges = np.histogram2d(plotData["x"], plotData["z"], bins=[50, 500], weights=plotData["by"])
    plt.figure(figsize=[20, 10])
    plt.imshow(H, interpolation="nearest", origin="low")
    # plt.colorbar()
    plt.show()
 
 

Variable Documentation

action

add_noise_to_field.action

Definition at line 59 of file add_noise_to_field.py.

args

add_noise_to_field.args = parser.parse_args()

Definition at line 68 of file add_noise_to_field.py.

default

add_noise_to_field.default

Definition at line 59 of file add_noise_to_field.py.

False

add_noise_to_field.False

Definition at line 63 of file add_noise_to_field.py.

first_line

f add_noise_to_field.first_line = f.readline().strip().split()

Definition at line 71 of file add_noise_to_field.py.

float

add_noise_to_field.float

Definition at line 64 of file add_noise_to_field.py.

int

add_noise_to_field.int

Definition at line 65 of file add_noise_to_field.py.

parser

argparse add_noise_to_field.parser = argparse.ArgumentParser(description="Process field.")

Definition at line 58 of file add_noise_to_field.py.

step

add_noise_to_field.step

Definition at line 79 of file add_noise_to_field.py.

str

add_noise_to_field.str

Definition at line 59 of file add_noise_to_field.py.

type

add_noise_to_field.type

Definition at line 59 of file add_noise_to_field.py.

xSpace

add_noise_to_field.xSpace

Definition at line 76 of file add_noise_to_field.py.

ySpace

add_noise_to_field.ySpace

Definition at line 77 of file add_noise_to_field.py.

zSpace

add_noise_to_field.zSpace

Definition at line 78 of file add_noise_to_field.py.