def main():
parser = ArgumentParser()
parser.add_argument('-g', '--grid-params', type=FileType('r'), required=True)
parser.add_argument('--part-quant', default=1)
parser.add_argument('--isec', default=4)
parser.add_argument('--idir', default=0)
parser.add_argument('--nqua', default=1)
parser.add_argument('--ist1', default=10)
parser.add_argument('--ist2', default=10)
parser.add_argument('--jst1', default=50)
parser.add_argument('--jst2', default=50)
parser.add_argument('--kst1', default=5)
parser.add_argument('--kst2', default=5)
args = parser.parse_args()
read_params(args.grid_params)
tm.mod_param.partquant = args.part_quant
tm.mod_param.partquant = args.part_quant
tm.mod_seed.isec = args.isec
tm.mod_seed.idir = args.idir
tm.mod_seed.nqua = args.nqua
tm.mod_seed.seedtype = 1
tm.mod_seed.ist1 = args.ist1
tm.mod_seed.ist2 = args.ist2
tm.mod_seed.jst1 = args.jst1
tm.mod_seed.jst2 = args.jst2
tm.mod_seed.kst1 = args.kst1
tm.mod_seed.kst2 = args.kst2
tm.init_params2()
tm.coordinat()
tm.mod_grid.kmt[:] = tm.mod_param.km
if tm.mod_seed.nqua == 1: # number of trajectories (per time resolution)
# num=NTRACMAX
tm.mod_seed.num = tm.mod_param.partquant
elif tm.mod_seed.nqua == 2:
tm.mod_param.voltr = tm.mod_param.partquant
elif tm.mod_seed.nqua == 3:
tm.mod_param.voltr = tm.mod_param.partquant
tm.mod_traj.trj[:] = 0
tm.mod_traj.nrj[:] = 0
tm.init_seed()
tm.mod_seed.seed(0, 0)
print tm.mod_seed.seed_ijk
print tm.mod_traj.trj
def main():
tm.init_params()
tm.coordinat()
writesetup()
print 'tm.time.intmin', tm.mod_time.intmin
print 'tm.time.intmax', tm.mod_time.intmax
if tm.mod_seed.nff == 1: #Forward
print "Forward"
tm.mod_time.intstart = tm.mod_time.intmin
tm.mod_time.intend = tm.mod_time.intmax
else: #Backward
print "Backward"
tm.mod_time.intstart = tm.mod_time.intmin
tm.mod_time.intend = tm.mod_time.intmax
print "tm.time.intstart", tm.mod_time.intstart
print "tm.time.intend", tm.mod_time.intend
tm.init_seed()
filename = abspath(join(curdir, 'results-old', 'data'))
print "filename", filename
print "tm.mod_seed.nqua", tm.mod_seed.nqua
if tm.mod_seed.nqua == 1: # number of trajectories (per time resolution)
# num=NTRACMAX
print "tm.params.part_quant", tm.mod_param.partquant, tm.mod_seed.num
tm.mod_seed.num = tm.mod_param.partquant
elif tm.mod_seed.nqua == 2:
tm.mod_param.voltr = tm.mod_param.partquant
elif tm.mod_seed.nqua == 3:
tm.mod_param.voltr = tm.mod_param.partquant
tm.fortran_file(56, filename + '_run.asc') # trajectory path
tm.fortran_file(57, filename + '_out.asc') # exit position
tm.fortran_file(58, filename + '_in.asc') # entrance position
tm.fortran_file(59, filename + '_err.asc') # Error position
tm.loop.readfields = readfields
print_state()
# tm.loop()
print "Done!"