def setUp(self):
solver = 'SLSQP'
num_seg = 10
seg_ncn = 2
rel_lengths = 'lgl'
# Instantiate a problem and set it's root to an empty Trajectory
prob = Problem()
prob.add_traj(Trajectory("traj0"))
if solver == 'SNOPT' and pyOptSparseDriver is not None:
driver = pyOptSparseDriver()
driver.options['optimizer'] = solver
driver.opt_settings['Major iterations limit'] = 1000
driver.opt_settings['iSumm'] = 6
driver.opt_settings['Major step limit'] = 0.5
driver.opt_settings["Major feasibility tolerance"] = 1.0E-5
driver.opt_settings["Major optimality tolerance"] = 1.0E-5
driver.opt_settings["Minor feasibility tolerance"] = 1.0E-4
driver.opt_settings['Verify level'] = 3
else:
driver = ScipyOptimizer()
driver.options['tol'] = 1.0E-6
driver.options['disp'] = True
driver.options['maxiter'] = 500
prob.driver = driver
dynamic_controls = [{
'name': 'theta',
'units': 'rad'
}, {
'name': 'psi',
'units': 'rad'
}]
static_controls = [{
'name': 'mass',
'units': 'kg'
}, {
'name': 'g',
'units': 'm/s/s'
}, {
'name': 'Cd',
'units': 'unitless'
}, {
'name': 'D_magnetic',
'units': 'N'
}, {
'name': 'R',
'units': 'J/(kg*K)'
}, {
'name': 'S',
'units': 'm**2'
}, {
'name': 'T_ambient',
'units': 'K'
}, {
'name': 'p_tube',
'units': 'Pa'
}]
phase0 = CollocationPhase(name='phase0',
rhs_class=MagnePlaneRHS,
num_seg=num_seg,
seg_ncn=seg_ncn,
rel_lengths=rel_lengths,
dynamic_controls=dynamic_controls,
static_controls=static_controls)
prob.trajectories["traj0"].add_phase(phase0)
phase0.set_state_options('x',
lower=0,
upper=100,
ic_val=0,
ic_fix=True,
fc_val=10,
fc_fix=True,
scaler=10.0,
defect_scaler=0.1)
phase0.set_state_options('y',
lower=0,
upper=0,
ic_val=0,
ic_fix=True,
fc_val=0,
fc_fix=True,
scaler=10.0,
defect_scaler=0.1)
phase0.set_state_options('z',
lower=-10,
upper=10,
ic_val=-10,
ic_fix=True,
fc_val=-5,
fc_fix=True,
scaler=10.0,
defect_scaler=0.1)
phase0.set_state_options('v',
lower=0,
upper=np.inf,
ic_val=0.0,
ic_fix=True,
fc_val=10.0,
fc_fix=False,
scaler=10.0,
defect_scaler=0.1)
phase0.set_dynamic_control_options('theta',
ic_val=-1.57,
fc_val=0.2,
opt=True,
lower=-1.58,
upper=1.58,
scaler=0.01)
phase0.set_dynamic_control_options('psi',
ic_val=0.0,
fc_val=0.0,
opt=True,
lower=0.0,
upper=0.0,
scaler=0.01)
phase0.set_static_control_options(name='g', val=9.80665, opt=False)
phase0.set_static_control_options(name='mass', val=1200, opt=False)
phase0.set_static_control_options(name='Cd', val=0.0, opt=False)
phase0.set_static_control_options(name='S', val=0.0, opt=False)
phase0.set_static_control_options(name='p_tube', val=0.0, opt=False)
phase0.set_static_control_options(name='T_ambient',
val=298.0,
opt=False)
phase0.set_static_control_options(name='R', val=287.0, opt=False)
# Set D_magnetic equal to the thrust value for now, no net forces
# on the pod except gravity.
phase0.set_static_control_options(name='D_magnetic',
val=30000.0,
opt=False)
phase0.set_time_options(t0_val=0,
t0_lower=0,
t0_upper=0,
tp_val=1.81,
tp_lower=0.5,
tp_upper=10.0)
prob.trajectories["traj0"].add_objective(name="t",
phase="phase0",
place="end",
scaler=1.0)
# Do top-level FD
prob.root.deriv_options['type'] = 'fd'
self.prob = prob
def magneplane_brachistochrone(solver='SLSQP', num_seg=3, seg_ncn=3):
prob = Problem()
traj = prob.add_traj(Trajectory("traj0"))
if solver == 'SNOPT':
if pyOptSparseDriver is None:
raise ValueError(
'Requested SNOPT but pyoptsparse is not available')
driver = pyOptSparseDriver()
driver.options['optimizer'] = solver
driver.opt_settings['Major iterations limit'] = 1000
driver.opt_settings['iSumm'] = 6
driver.opt_settings['Major step limit'] = 0.5
driver.opt_settings["Major feasibility tolerance"] = 1.0E-6
driver.opt_settings["Major optimality tolerance"] = 1.0E-6
driver.opt_settings["Minor feasibility tolerance"] = 1.0E-4
driver.opt_settings['Verify level'] = 3
else:
driver = ScipyOptimizer()
driver.options['tol'] = 1.0E-6
driver.options['disp'] = True
driver.options['maxiter'] = 500
prob.driver = driver
dynamic_controls = [{
'name': 'g',
'units': 'm/s**2'
}, {
'name': 'T',
'units': 'N'
}, {
'name': 'D',
'units': 'N'
}, {
'name': 'mass',
'units': 'kg'
}, {
'name': 'psi',
'units': 'rad'
}, {
'name': 'theta',
'units': 'rad'
}, {
'name': 'phi',
'units': 'rad'
}]
phase0 = CollocationPhase(name='phase0',
rhs_class=MagneplaneRHS,
num_seg=num_seg,
seg_ncn=seg_ncn,
rel_lengths="equal",
dynamic_controls=dynamic_controls,
static_controls=None)
traj.add_phase(phase0)
phase0.set_state_options('x',
lower=0,
upper=10,
ic_val=0,
ic_fix=True,
fc_val=10,
fc_fix=True,
defect_scaler=0.1)
phase0.set_state_options('y',
lower=0,
upper=0,
ic_val=0,
ic_fix=True,
fc_val=0,
fc_fix=True,
defect_scaler=0.1)
phase0.set_state_options('z',
lower=-10,
upper=0,
ic_val=-10,
ic_fix=True,
fc_val=-5,
fc_fix=True,
defect_scaler=0.1)
phase0.set_state_options('v',
lower=0,
upper=np.inf,
ic_val=0.0,
ic_fix=True,
fc_val=10.0,
fc_fix=False,
defect_scaler=0.1)
phase0.set_dynamic_control_options(name='psi',
val=phase0.node_space(0.0, 0.0),
opt=False)
phase0.set_dynamic_control_options(name='theta',
val=phase0.node_space(-.46, -.46),
opt=True,
lower=-1.57,
upper=1.57,
scaler=1.0)
phase0.set_dynamic_control_options(name='phi',
val=phase0.node_space(0.0, 0.0),
opt=False)
phase0.set_dynamic_control_options(name='g',
val=phase0.node_space(9.80665, 9.80665),
opt=False)
phase0.set_dynamic_control_options(name='T',
val=phase0.node_space(0.0, 0.0),
opt=False)
phase0.set_dynamic_control_options(name='D',
val=phase0.node_space(0.0, 0.0),
opt=False)
phase0.set_dynamic_control_options(name='mass',
val=phase0.node_space(1000.0, 1000.0),
opt=False)
phase0.set_time_options(t0_val=0,
t0_lower=0,
t0_upper=0,
tp_val=2.0,
tp_lower=0.5,
tp_upper=10.0)
traj.add_objective(name="t", phase="phase0", place="end", scaler=1.0)
return prob
