class TestGlobalOptimizer(unittest.TestCase):
def setUp(self):
linac = Linac()
linac.add_beamline('astra',
name='gun',
swd=test_path,
fin='injector.in',
template=osp.join(test_path, 'injector.in.000'),
pout='injector.0150.001')
self.opt = LinacOptimization(linac)
self.opt.add_obj('f', expr='gun.out.Sx', scale=1.e6)
self.opt.add_icon('g1',
func=lambda a: a['gun'].max.emitx * 1e6,
ub=0.041)
self.opt.add_econ('g2', func=lambda a: a['gun'].out.gamma, eq=10.0)
self.opt.add_var('laser_spot', value=0.1, lb=0.04, ub=0.3)
self.opt.add_var('main_sole_b', value=0.1, lb=0.0, ub=0.3)
self.opt.add_var('gun_gradient', value=130, lb=90.0, ub=130.0)
def tearDown(self):
for file in glob.glob(osp.join(test_path, "injector.*.001")):
os.remove(file)
os.remove(osp.join(test_path, "injector.in"))
def test_optimization(self):
optimizer = ALPSO()
opt_f, opt_x = self.opt.solve(optimizer)
self.assertAlmostEqual(opt_f, 65.62, delta=0.10)
self.assertAlmostEqual(opt_x[0], 0.040000, delta=0.00010)
self.assertAlmostEqual(opt_x[1], 0.2521, delta=0.0010)
self.assertAlmostEqual(opt_x[2], 100.4, delta=1.0)
linac = Linac()
linac.add_beamline('astra',
name='gun',
swd='../astra_files',
fin='injector.in',
template='injector.in.000',
pout='injector.0450.001')
opt = LinacOptimization(linac)
opt.add_obj('St', expr='gun.out.St', scale=1e15)
# equality constraint (the No. of particles at the end of the 'gun' beamline)
opt.add_econ('n_pars', expr='gun.out.n', eq=2000)
# inequality constraint (the beta [x] function at the end of the 'gun' beamline) with the upper boundary.
opt.add_icon('emitx', expr='gun.out.emitx', scale=1e6, ub=0.3)
# inequality constraint (the Lorentz factor at the end of the 'gun' beamline) with the lower boundary.
opt.add_icon('gamma', func=lambda a: a['gun'].out.gamma, lb=20.0)
# inequality constraint (the maximum beam size [x] throughout the 'gun' beamline) with upper boundary.
opt.add_icon('max_Sx', func=lambda a: a['gun'].max.Sx * 1e3, ub=3.0)
opt.add_var('laser_spot', value=0.1, lb=0.04, ub=0.5)
opt.add_var('main_sole_b', value=0.2, lb=0.00, ub=0.4)
opt.add_var('gun_phase', value=0.0, lb=-10, ub=10)
opt.add_var('tws_phase', value=0.0, lb=-90, ub=0)
optimizer = ALPSO()
optimizer.swarm_size = 40 # configure the optimizer
opt.solve(optimizer)