GPUE  v1.0
GPU Gross-Pitaevskii Equation numerical solver for Bose-Einstein condensates
extract_gauge Namespace Reference

Functions

def extract_field (filename, varname)
 
def write_2d (var, outfile, fudge_factor)
 
def write_3d (var, outfile, fudge_factor)
 
def write_constant (outfile, res)
 

Variables

float hbar = 1.05457148e-34
 
def var = extract_field("Avec_broad_256.mat","avec")
 

Function Documentation

◆ extract_field()

def extract_gauge.extract_field (   filename,
  varname 
)

Definition at line 13 of file extract_gauge.py.

13 def extract_field(filename, varname):
14  mat = scipy.io.loadmat(filename)
15  return mat[varname]
16 
17 # function to output file in 2d format
def extract_field(filename, varname)

◆ write_2d()

def extract_gauge.write_2d (   var,
  outfile,
  fudge_factor 
)

Definition at line 18 of file extract_gauge.py.

18 def write_2d(var, outfile, fudge_factor):
19  file = open(outfile,'w')
20  for i in range(0,len(var)):
21  for j in range(0,len(var)):
22  file.write(str(var[i][j] * fudge_factor) + '\n')
23  file.close()
24 
25 # function to output 2d gauge field in 3d for GPU simulation
def write_2d(var, outfile, fudge_factor)

◆ write_3d()

def extract_gauge.write_3d (   var,
  outfile,
  fudge_factor 
)

Definition at line 26 of file extract_gauge.py.

26 def write_3d(var, outfile, fudge_factor):
27  file = open(outfile,'w')
28  for i in range(0,len(var)):
29  for j in range(0,len(var)):
30  for k in range(0,len(var)):
31  file.write(str(var[k][j] * fudge_factor) + '\n')
32  file.close()
33 
def write_3d(var, outfile, fudge_factor)

◆ write_constant()

def extract_gauge.write_constant (   outfile,
  res 
)

Definition at line 34 of file extract_gauge.py.

34 def write_constant(outfile, res):
35  file = open(outfile,'w')
36  for i in range(0,res*res*res):
37  file.write(str(0) + '\n')
38  file.close()
39 
def write_constant(outfile, res)

Variable Documentation

◆ hbar

float extract_gauge.hbar = 1.05457148e-34

Definition at line 10 of file extract_gauge.py.

◆ var

def extract_gauge.var = extract_field("Avec_broad_256.mat","avec")

Definition at line 40 of file extract_gauge.py.