framework = Framework('basefiles_FEBE_'+str(scaling), overwrite=False)
framework.loadSettings('Lattices/claraX400_v12_80MVm.def')
if not os.name == 'nt':
framework.defineASTRACommand(['mpiexec','-np',str(3*scaling),'/opt/ASTRA/astra_MPICH2.sh'])
framework.defineGeneratorCommand(['/opt/ASTRA/generator.sh'])
framework.defineCSRTrackCommand(['/opt/OpenMPI-1.4.3/bin/mpiexec','-n',str(3*scaling),'/opt/CSRTrack/csrtrack_openmpi.sh'])
framework.generator.number_of_particles = 2**(3*scaling)
framework.track(run=True)
# for i in [3,4,5,6]:
# create_base_files(i)
# exit()
framework = Framework('basefiles_6', overwrite=False)
framework.loadSettings('Lattices/claraX400_v12_80MVm.def')
parameters = framework.getElementType('quadrupole','k1')
print 'parameters = ', parameters
best = parameters
class fitnessFunc():
def __init__(self, args, tempdir, scaling=4, overwrite=True, verbose=False, summary=False, clean=False):
self.cons = constraintsClass()
self.beam = rbf.beam()
self.twiss = rtf.twiss()
self.tmpdir = tempdir
self.scaling = scaling
self.verbose = verbose
self.summary = summary
self.parameters = list(args)
self.dirname = os.path.basename(self.tmpdir)
framework.defineASTRACommand(['mpiexec','-np',str(3*scaling),'/opt/ASTRA/astra_MPICH2.sh'])
framework.defineGeneratorCommand(['/opt/ASTRA/generator.sh'])
framework.defineCSRTrackCommand(['/opt/OpenMPI-1.4.3/bin/mpiexec','-n',str(3*scaling),'/opt/CSRTrack/csrtrack_openmpi.sh'])
framework.defineElegantCommand(['elegant'])
framework['S07'].file_block['input']['prefix'] = '../../basefiles_'+str(scaling)+'/'
framework.track(track=True, startfile='S07')
# for i in [6]:
# create_base_files(i)
# exit()
framework = Framework('basefiles_6', overwrite=False)
framework.loadSettings('Lattices/clara400_v12_FEBE.def')
# posstart = 60
# parameters = [a['k1'] for a in framework.getElementType('quadrupole') if a['position_start'][2] > posstart]
variables = [a for a in framework.getElementType('quadrupole') if 'S07F' in a.objectName]
preparameters = parameters = [a['k1'] for a in variables]
print('parameters = ', parameters)
# exit()
best = parameters
class fitnessFunc():
def __init__(self, args, tempdir, scaling=4, overwrite=True, verbose=False, summary=False, clean=False):
global preparameters, variables
self.cons = constraintsClass()
self.beam = rbf.beam()
self.twiss = rtf.twiss()
self.tmpdir = tempdir
self.scaling = scaling
self.verbose = verbose