gen_data Namespace Reference

Functions
def	to_vtk (item, xDim, yDim, zDim, nframes, filename)
def	wfc_density (xDim, yDim, zDim, data_dir, pltval, i)
def	wfc_phase (xDim, yDim, zDim, data_dir, pltval, i)
def	proj_phase_2d (xDim, yDim, zDim, data_dir, pltval, i)
def	var (xDim, yDim, zDim, data_dir, pltval)
def	var_r2 (xDim, yDim, zDim, data_dir)
def	proj_var2d (xdim, yDim, zDim, data_dir, pltval)
def	proj_var2d (xdim, yDim, zDim, data_dir, pltval, file_string)
def	proj_var1d (xdim, yDim, zDim, data_dir, pltval, file_string)
def	proj_2d (xDim, yDim, zDim, data_dir, pltval, i)
def	proj_k2d (xDim, yDim, zDim, data_dir, pltval, i)
def	to_bvox (item, xDim, yDim, zDim, nframes, filename)
def	wfc_com (xDim, yDim, zDim, data_dir, pltval, i)

Variables
int	xDim = 256

Function Documentation

proj_2d()

def gen_data.proj_2d	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 176 of file gen_data.py.

References print().

def proj_2d(xDim, yDim, zDim, data_dir, pltval, i):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir

    filename = data_dir + "/Pwfc_%s" %i
    print(i)
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    if (pltval == "wfc_ev"):
        data_real = data_dir + "/wfc_ev_%s" % i
        data_im = data_dir + "/wfc_evi_%s" % i
    elif (pltval == "wfc_ramp"):
        data_real = data_dir + "/wfc_0_ramp_%s" % i
        data_im = data_dir + "/wfc_0_rampi_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    print(len(lines_real))
    wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = abs(wfc_real + 1j * wfc_im)
    wfc = wfc * wfc
    file = open(filename,'w')
    for k in range(0,xDim):
        for j in range(0,yDim):
            file.write(str(wfc[k][j][zDim/2]) + '\n')
    file.close()

Here is the call graph for this function:

proj_k2d()

def gen_data.proj_k2d	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 203 of file gen_data.py.

References print().

def proj_k2d(xDim, yDim, zDim, data_dir, pltval, i):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir

    filename = data_dir + "/wfc_1"
    print(i)
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    print(len(lines_real))
    wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = np.fft.fftshift(np.fft.fftn(wfc_real + 1j * wfc_im))
    wfc = abs(wfc) * abs(wfc)
    file = open(filename,'w')
    for k in range(0,xDim):
        for j in range(0,yDim):
            file.write(str(wfc[j][k][zDim/2]) + '\n')
    file.close()

# function to output the bvox bin data of a matrix

Here is the call graph for this function:

proj_phase_2d()

def gen_data.proj_phase_2d	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 85 of file gen_data.py.

References print().

def proj_phase_2d(xDim, yDim, zDim, data_dir, pltval, i):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir
    filename = data_dir + "/wfc_ph_%s" %i
    print(i)
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    if (pltval == "wfc_ev"):
        data_real = data_dir + "/wfc_ev_%s" % i
        data_im = data_dir + "/wfc_evi_%s" % i
    elif (pltval == "wfc_ramp"):
        data_real = data_dir + "/wfc_0_ramp_%s" % i
        data_im = data_dir + "/wfc_0_rampi_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    print(len(lines_real))
    wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = (wfc_real + 1j * wfc_im)
    wfc = np.angle(wfc)
    file = open(filename,'w')
    for k in range(0,xDim):
        for j in range(0,yDim):
            file.write(str(wfc[j][k][zDim/2]) + '\n')
    file.close()

Here is the call graph for this function:

proj_var1d()

def gen_data.proj_var1d	(		xdim,
			yDim,
			zDim,
			data_dir,
			pltval,
			file_string
	)

Definition at line 161 of file gen_data.py.

def proj_var1d(xdim, yDim, zDim, data_dir, pltval, file_string):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir

    filename = data_dir + "/" + file_string
    file = open(filename,"w")
    data = data_dir + "/" + pltval
    lines = np.loadtxt(data)
    var_data = np.reshape(lines, (xDim, yDim, zDim))
    for i in range(0,xDim):
        file.write(str(var_data[zDim/2][yDim/2][i])+'\n')
    file.close

proj_var2d() [1/2]

def gen_data.proj_var2d	(		xdim,
			yDim,
			zDim,
			data_dir,
			pltval
	)

Definition at line 144 of file gen_data.py.

Referenced by proj_var2d().

def proj_var2d(xdim, yDim, zDim, data_dir, pltval):
    proj_var2d(xdim, yDim, zDim, data_dir, pltval, "var")

Here is the caller graph for this function:

proj_var2d() [2/2]

def gen_data.proj_var2d	(		xdim,
			yDim,
			zDim,
			data_dir,
			pltval,
			file_string
	)

Definition at line 147 of file gen_data.py.

References proj_var2d().

def proj_var2d(xdim, yDim, zDim, data_dir, pltval, file_string):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir

    filename = data_dir + "/" + file_string
    file = open(filename,"w")
    data = data_dir + "/" + pltval
    lines = np.loadtxt(data)
    var_data = np.reshape(lines, (xDim, yDim, zDim))
    for k in range(0,xDim):
        for j in range(0,yDim):
            file.write(str(var_data[k][j][xDim/2])+'\n')
    file.close

Here is the call graph for this function:

to_bvox()

def gen_data.to_bvox	(		item,
			xDim,
			yDim,
			zDim,
			nframes,
			filename
	)

Definition at line 225 of file gen_data.py.

def to_bvox(item, xDim, yDim, zDim, nframes, filename):
    header = np.array([xDim, yDim, zDim, nframes])
    binfile = open(filename, "wb")
    header.astype('<i4').tofile(binfile)
    item.astype('<f4').tofile(binfile)
    binfile.close()

# find Center of Mass of toroidal condensate

to_vtk()

def gen_data.to_vtk	(		item,
			xDim,
			yDim,
			zDim,
			nframes,
			filename
	)

Definition at line 14 of file gen_data.py.

def to_vtk(item, xDim, yDim, zDim, nframes, filename):
    outfile = open(filename, "w")
    outfile.write("# vtk DataFile Version 3.0\n")
    outfile.write("vtkfile\n")
    outfile.write("ASCII\n")
    outfile.write("DATASET STRUCTURED_POINTS\n")
    outfile.write("DIMENSIONS "+str(xDim)+" "+str(yDim)+" "+str(zDim)+"\n")
    outfile.write("ORIGIN 0 0 0\n")
    outfile.write("SPACING 1 1 1\n")
    outfile.write("POINT_DATA " + str(xDim*yDim*zDim) + "\n")
    outfile.write("SCALARS scalars float 1\n")
    outfile.write("LOOKUP_TABLE default\n")
    for i in range(xDim):
        for j in range(yDim):
            for k in range(zDim):
                outfile.write(str(item[i][j][k]) + " ")
            outfile.write('\n')
        outfile.write('\n') 

# Function to plot wfc with pltvar as a variable to modify the type of plot

var()

def gen_data.var	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval
	)

Definition at line 112 of file gen_data.py.

def var(xDim, yDim, zDim, data_dir, pltval):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir
    data = data_dir + "/" + pltval
    lines = np.loadtxt(data)
    maximum = max(lines)
    minimum = min(lines)
    for i in range(0,len(lines)):
        lines[i] = (lines[i] - minimum) / (maximum - minimum)
    val = np.reshape(lines, (xDim,yDim,zDim));
    return val

var_r2()

def gen_data.var_r2	(		xDim,
			yDim,
			zDim,
			data_dir
	)

Definition at line 124 of file gen_data.py.

def var_r2(xDim, yDim, zDim, data_dir):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir
    data_ax = data_dir + "/Ax_0"
    data_ay = data_dir + "/Ay_0"

    lines_ax = np.loadtxt(data_ax)
    lines_ay = np.loadtxt(data_ay)

    lines_out = [0 for i in range(len(lines_ax))]
    for i in range(0,len(lines_ax)):
        lines_out[i] = np.sqrt(lines_ay[i]**2 + lines_ax[i]**2)

    maximum = max(lines_out)
    minimum = min(lines_out)
    for i in range(0,len(lines_out)):
        lines_out[i] = (lines_out[i] - minimum) / (maximum - minimum)
    val = np.reshape(lines_out, (xDim,yDim,zDim));
    return val

wfc_com()

def gen_data.wfc_com	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 233 of file gen_data.py.

References print().

def wfc_com(xDim, yDim, zDim, data_dir, pltval, i):
    if data_dir[0] != "/":
        data_dir = "../" + data_dir

    filename = data_dir + "/wfc_%s" %i
    print(i)
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    if (pltval == "wfc_ev"):
        data_real = data_dir + "/wfc_ev_%s" % i
        data_im = data_dir + "/wfc_evi_%s" % i
    elif (pltval == "wfc_ramp"):
        data_real = data_dir + "/wfc_0_ramp_%s" % i
        data_im = data_dir + "/wfc_0_rampi_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    print(len(lines_real))
    wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = abs(wfc_real + 1j * wfc_im)
    wfc = wfc * wfc

    # Here we are finding the CoM
    comx = 0
    comy = 0
    sum = 0
    for i in range(xDim/2,xDim):
        for j in range(0,zDim):
            comx += wfc[j][i][zDim/2]*i
            comy += wfc[j][i][zDim/2]*j
            sum += wfc[j][i][zDim/2]

    comx /= sum
    comy /= sum

    return comx, comy

wfc_density()

def gen_data.wfc_density	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 34 of file gen_data.py.

References print().

def wfc_density(xDim, yDim, zDim, data_dir, pltval, i):
    print(i)
    if data_dir[0] != "/":
        data_dir = "../" + data_dir
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    if (pltval == "wfc_ev"):
        data_real = data_dir + "/wfc_ev_%s" % i
        data_im = data_dir + "/wfc_evi_%s" % i
    elif (pltval == "wfc_ramp"):
        data_real = data_dir + "/wfc_0_ramp_%s" % i
        data_im = data_dir + "/wfc_0_rampi_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    print(len(lines_real))
    wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = abs(wfc_real + 1j * wfc_im)
    wfc = wfc * wfc
    wfc = np.reshape(wfc,(xDim*yDim*zDim))
    maximum = max(wfc)
    wfc /= maximum
    wfc = np.reshape(wfc,(xDim,yDim,zDim))
    return wfc

Here is the call graph for this function:

wfc_phase()

def gen_data.wfc_phase	(		xDim,
			yDim,
			zDim,
			data_dir,
			pltval,
			i
	)

Definition at line 59 of file gen_data.py.

References print().

def wfc_phase(xDim, yDim, zDim, data_dir, pltval, i):
    print(i)
    if data_dir[0] != "/":
        data_dir = "../" + data_dir
    data_real = data_dir + "/wfc_0_const_%s" % i
    data_im = data_dir + "/wfc_0_consti_%s" % i
    if (pltval == "wfc_ev"):
        data_real = data_dir + "/wfc_ev_%s" % i
        data_im = data_dir + "/wfc_evi_%s" % i
    elif (pltval == "wfc_ramp"):
        data_real = data_dir + "/wfc_0_ramp_%s" % i
        data_im = data_dir + "/wfc_0_rampi_%s" % i
    lines_real = np.loadtxt(data_real)
    lines_im = np.loadtxt(data_im)
    wfc_real = np.reshape(lines_real, (xDim,yDim, zDim));
    wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
    wfc = (wfc_real + 1j * wfc_im)
    wfc = np.angle(wfc)
    wfc = np.reshape(wfc,(xDim*yDim*zDim))
    maximum = max(wfc)
    minimum = min(wfc)
    for i in range(0,len(wfc)):
        wfc[i] = (wfc[i] - minimum) / (maximum - minimum)
    wfc = np.reshape(wfc,(xDim,yDim,zDim))
    return wfc

Here is the call graph for this function:

Variable Documentation

xDim

int gen_data.xDim = 256

Definition at line 11 of file gen_data.py.