diff --git a/setup.py b/setup.py index 0a4fb1963e69d6605503af02e2c6fde4aa929d83..925a23c6d9566208c65b657788ffd49f1adb446e 100644 --- a/setup.py +++ b/setup.py @@ -6,4 +6,5 @@ setup(name='Yngve Tools', author='Yngve Inntjore Levinsen', author_email='yngve.levinsen@esss.se', packages=['ess'], + scripts=['tracewin'], ) diff --git a/tracewin b/tracewin new file mode 100644 index 0000000000000000000000000000000000000000..5dd53831017bc02286fe85bd4f95bedba1068fb6 --- /dev/null +++ b/tracewin @@ -0,0 +1,45 @@ +#!/usr/bin/env python + +import argparse +import os +parser = argparse.ArgumentParser(description="Creates a TraceWin batch command") + +parser.add_argument('-p', '--project', dest='project', + help="Project file", required=True) +parser.add_argument('-l', '--lattice', dest='lattice', + help="Lattice/structure file (.dat) to use") +parser.add_argument('-f', '--freq', dest='frequency', + help="Bunch Frequency of beam 1 [MHz]") +parser.add_argument('-c', '--current', dest='current', + help="Beam Current of beam 1 [mA]") +parser.add_argument('-n', '--npart', dest='npart', + help="Number of particles in beam 1") +parser.add_argument('-e', '--energy', dest='energy', + help="Input beam energy for beam 1 [MeV]") +parser.add_argument('-o', '--output', dest='outpath', + help="Path to calculation directory") +parser.add_argument('-r', '--seed', dest='seed', + help="Initial random seed", type=int) + +args = parser.parse_args() + +# A list to be filled depending on arguments given... +cmd = ["TraceWin64_noX11", "'"+args.project+"'"] + +if args.lattice: + cmd.append("dat_file="+args.lattice) +if args.frequency: + cmd.append("freq1="+args.lattice) +if args.current: + cmd.append("current1="+args.current) +if args.npart: + cmd.append("npart1="+args.npart) +if args.energy: + cmd.append("energy1="+args.energy) +if args.outpath: + cmd.append("path_cal='"+os.path.abspath(args.outpath)+"'") +if args.seed: + cmd.append("random_seed="+str(args.seed)) + +# For now we just print the command: +print ' '.join(cmd)