def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','RUN','commit_parameters')
handler.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('INITIALIZED', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('RUN', 'startup_viz')
handler.add_method('RUN', 'new_resource')
handler.add_method('RUN', 'new_pilot')
handler.add_method('RUN', 'submit_script_job')
handler.add_method('RUN', 'submit_function_job')
handler.add_method('RUN', 'get_resource_state')
handler.add_method('RUN', 'get_pilot_state')
handler.add_method('RUN', 'get_pilot_status')
handler.add_method('RUN', 'get_script_job_state')
handler.add_method('RUN', 'get_script_job_status')
handler.add_method('RUN', 'get_function_job_state')
handler.add_method('RUN', 'get_function_job_status')
handler.add_method('RUN', 'get_worker_state')
handler.add_method('RUN', 'get_worker_status')
handler.add_method('RUN', 'use_for_distributed_workers')
handler.add_method('RUN', 'use_for_all_workers')
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
handler.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
handler.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('EDIT', 'before_get_parameter')
handler.add_method('UPDATE','before_get_parameter')
handler.add_transition('EDIT', 'UPDATE', 'generate_particles', False)
handler.add_transition('UPDATE', 'RUN', 'update_particle_set')
handler.add_transition('RUN', 'EDIT', 'clear_particle_set')
handler.add_method('RUN', 'invoke_state_change_updated')
handler.add_method('EDIT', 'get_number_of_particles_updated')
handler.add_method('UPDATE', 'get_number_of_particles_updated')
handler.add_method('RUN', 'get_number_of_particles_updated')
handler.add_method('RUN', 'get_mass')
handler.add_method('RUN', 'get_position')
handler.add_method('RUN', 'get_velocity')
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
handler.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
handler.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('EDIT', 'before_get_parameter')
handler.add_method('UPDATE','before_get_parameter')
handler.add_method('EDIT', 'new_particle')
handler.add_method('EDIT', 'delete_particle')
handler.add_method('UPDATE', 'new_particle')
handler.add_method('UPDATE', 'delete_particle')
handler.add_transition('EDIT', 'RUN', 'commit_particles')
handler.add_transition('RUN', 'UPDATE', 'new_particle', False)
handler.add_transition('RUN', 'UPDATE', 'delete_particle', False)
handler.add_transition('UPDATE', 'RUN', 'recommit_particles')
handler.add_method('RUN', 'calculate_densities')
handler.add_method('RUN', 'do_hop')
handler.add_method('RUN', 'get_mass')
handler.add_method('RUN', 'get_position')
handler.add_method('RUN', 'get_density')
handler.add_method('RUN', 'get_group_id')
def define_state(self, object):
CommonCode.define_state(self, object)
for method_name in [
'initialize_from_dyn',
'initialize_from_elements',
'transform_to_time',
'advance_to_radius',
'return_to_radius',
'advance_to_periastron',
'advance_to_apastron',
'return_to_periastron',
'return_to_apastron',
'get_total_mass',
'get_time',
'get_period',
'get_elements',
'get_integrals',
'get_separation_vector',
'get_separation',
'set_periastron',
'get_periastron',
'get_apastron',
'get_velocity_vector',
'get_angles',
'set_longitudinal_unit_vector',
'set_normal_unit_vector',
'get_longitudinal_unit_vector',
'get_transverse_unit_vector',
'set_transverse_unit_vector',
'get_normal_unit_vector',
'print_all',
'set_random',
'get_random',
'make_binary_scattering']:
object.add_method('!UNINITIALIZED!END', method_name)
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method("generate_particles", (), (handler.ERROR_CODE, ))
handler.add_method("get_number_of_particles_updated", (), (
handler.NO_UNIT,
handler.NO_UNIT,
handler.ERROR_CODE,
))
handler.add_method("get_mass", (handler.INDEX, ),
(nbody_system.mass, handler.ERROR_CODE))
handler.add_method("get_position", (handler.INDEX, ),
(nbody_system.length, nbody_system.length,
nbody_system.length, handler.ERROR_CODE))
handler.add_method("get_velocity", (handler.INDEX, ),
(nbody_system.speed, nbody_system.speed,
nbody_system.speed, handler.ERROR_CODE))
handler.add_method("get_output_path", (), (
handler.NO_UNIT,
handler.ERROR_CODE,
))
handler.add_method("set_output_path", (handler.NO_UNIT, ),
(handler.ERROR_CODE, ))
for par in parameters:
if hasattr(parameters[par]["default"], "unit"):
unit = parameters[par]["default"].unit
else:
unit = handler.NO_UNIT
handler.add_method("get_" + par, (), (
unit,
handler.ERROR_CODE,
))
handler.add_method("set_" + par, (unit, ), (handler.ERROR_CODE, ))
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE, ))
object.add_method("get_number_of_particles_updated", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("get_state", (object.INDEX, ),
[nbody_system.length] * 3 +
[nbody_system.speed] * 3 + [object.ERROR_CODE])
object.add_method("get_target_number_of_particles", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set_target_number_of_particles", (object.NO_UNIT, ),
(object.ERROR_CODE, ))
object.add_method("get_fractal_dimension", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set_fractal_dimension", (object.NO_UNIT, ),
(object.ERROR_CODE, ))
object.add_method("get_random_seed", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set_random_seed", (object.NO_UNIT, ),
(object.ERROR_CODE, ))
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_transition('EDIT', 'UPDATE', 'generate_particles', False)
object.add_transition('UPDATE', 'RUN', 'update_particle_set')
object.add_transition('RUN', 'EDIT', 'clear_particle_set')
object.add_method('RUN', 'invoke_state_change_updated')
object.add_method('EDIT', 'get_number_of_particles_updated')
object.add_method('UPDATE', 'get_number_of_particles_updated')
object.add_method('RUN', 'get_number_of_particles_updated')
object.add_method('RUN', 'get_mass')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_velocity')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('END', 'INITIALIZED', 'initialize_code', False)
object.add_transition('INITIALIZED', 'EDIT', 'commit_parameters')
object.add_transition('RUN', 'CHANGE_PARAMETERS_RUN', 'before_set_parameter', False)
object.add_transition('EDIT', 'CHANGE_PARAMETERS_EDIT', 'before_set_parameter', False)
object.add_transition('UPDATE', 'CHANGE_PARAMETERS_UPDATE', 'before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN', 'RUN', 'recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT', 'EDIT', 'recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE', 'UPDATE', 'recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_method('EDIT', 'new_particle')
object.add_method('EDIT', 'delete_particle')
object.add_method('UPDATE', 'new_particle')
object.add_method('UPDATE', 'delete_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
GravityFieldCode.define_state(self, object)
object.add_method('RUN', 'get_potential_energy')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_method('EDIT', 'new_particle')
object.add_method('EDIT', 'delete_particle')
object.add_method('UPDATE', 'new_particle')
object.add_method('UPDATE', 'delete_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_method('RUN', 'calculate_densities')
object.add_method('RUN', 'do_hop')
object.add_method('RUN', 'get_mass')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_density')
object.add_method('RUN', 'get_group_id')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_method('EVOLVED','before_get_parameter')
object.add_method('RUNCOMMIT','before_get_parameter')
object.add_method('EDIT', 'new_particle')
object.add_method('EDIT', 'delete_particle')
object.add_transition('EDIT', 'RUNCOMMIT', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_transition('RUN','RUNCOMMIT', 'recommit_particles')
object.add_transition('RUNCOMMIT','RUN', 'auto_go_to_run')
object.add_transition('RUNCOMMIT', 'EVOLVED', 'evolve_model', False)
# object.add_method('EVOLVED', 'evolve_model')
object.define_state('RUNCOMMIT')
object.add_transition('EVOLVED','RUN', 'synchronize_model')
object.add_method('RUN', 'synchronize_model')
object.add_method('RUN', 'get_state')
object.add_method('RUN', 'get_density')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_flux')
object.add_method('RUN', 'get_ionisation')
object.add_method('RUN', 'get_internal_energy')
object.add_method('RUN', 'set_dinternal_energy_dt')
object.add_method('RUN', 'get_dinternal_energy_dt')
object.add_method('UPDATE', 'set_dinternal_energy_dt')
object.add_method('UPDATE', 'get_dinternal_energy_dt')
object.add_method('INITIALIZED', 'set_hilbert_order')
object.add_method('INITIALIZED', 'set_box_size')
object.add_method('INITIALIZED', 'set_timestep')
object.add_method('INITIALIZED', 'set_source_Teff')
object.add_method('INITIALIZED', 'set_number_frequency_bins')
object.add_method('INITIALIZED', 'set_thermal_evolution')
object.add_method('INITIALIZED', 'set_blackbody_spectrum')
object.add_method('INITIALIZED', 'set_metal_cooling')
object.add_method('INITIALIZED', 'set_recombination_radiation')
object.add_method('INITIALIZED', 'set_collisional_ionization')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','RUN','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('RUN', 'new_resource')
object.add_method('RUN', 'new_pilot')
object.add_method('RUN', 'submit_script_job')
object.add_method('RUN', 'submit_function_job')
object.add_method('RUN', 'get_resource_state')
object.add_method('RUN', 'get_pilot_state')
object.add_method('RUN', 'get_pilot_status')
object.add_method('RUN', 'get_script_job_state')
object.add_method('RUN', 'get_script_job_status')
object.add_method('RUN', 'get_function_job_state')
object.add_method('RUN', 'get_function_job_status')
object.add_method('RUN', 'get_worker_state')
object.add_method('RUN', 'get_worker_status')
object.add_method('RUN', 'use_for_distributed_workers')
object.add_method('RUN', 'use_for_all_workers')
object.add_method('RUN', 'get_worker_port')
def define_state(self, handler):
CommonCode.define_state(self, handler)
for method_name in [
'initialize_from_dyn',
'initialize_from_elements',
'transform_to_time',
'advance_to_radius',
'return_to_radius',
'advance_to_periastron',
'advance_to_apastron',
'return_to_periastron',
'return_to_apastron',
'get_total_mass',
'get_time',
'get_period',
'get_elements',
'get_integrals',
'get_separation_vector',
'get_separation',
'set_periastron',
'get_periastron',
'get_apastron',
'get_velocity_vector',
'get_angles',
'set_longitudinal_unit_vector',
'set_normal_unit_vector',
'get_longitudinal_unit_vector',
'get_transverse_unit_vector',
'set_transverse_unit_vector',
'get_normal_unit_vector',
'print_all',
'set_random',
'get_random',
'make_binary_scattering']:
handler.add_method('!UNINITIALIZED!END', method_name)
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE,))
object.add_method("get_number_of_particles_updated", (), (object.NO_UNIT,object.NO_UNIT, object.ERROR_CODE,))
object.add_method("get_mass", (object.INDEX,),
(nbody_system.mass, object.ERROR_CODE)
)
object.add_method("get_position", (object.INDEX,),
(nbody_system.length, nbody_system.length, nbody_system.length, object.ERROR_CODE)
)
object.add_method("get_velocity", (object.INDEX,),
(nbody_system.speed, nbody_system.speed, nbody_system.speed, object.ERROR_CODE)
)
object.add_method("get_output_path", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_output_path", (object.NO_UNIT,), (object.ERROR_CODE,))
for par in parameters:
if hasattr(parameters[par]["default"],"unit"):
unit=parameters[par]["default"].unit
else:
unit=object.NO_UNIT
object.add_method("get_"+par, (), (unit, object.ERROR_CODE,))
object.add_method("set_"+par, (unit, ), (object.ERROR_CODE,))
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_method('EVOLVED','before_get_parameter')
object.add_method('EDIT', 'new_gas_particle')
object.add_method('EDIT', 'remove_gas_particle')
object.add_method('EDIT', 'new_src_particle')
object.add_method('EDIT', 'remove_src_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_gas_particle', False)
object.add_transition('RUN', 'UPDATE', 'remove_gas_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_src_particle', False)
object.add_transition('RUN', 'UPDATE', 'remove_src_particle', False)
object.add_transition('RUN', 'UPDATE', 'set_pos_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_rho_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_hsml_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_u_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_dudt_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_state_gas', False)
object.add_transition('RUN', 'UPDATE', 'set_state_src', False)
object.add_transition('RUN', 'UPDATE', 'set_luminosity_src', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_method('RUN', 'evolve_model')
object.add_method('RUN', 'get_state_gas')
object.add_method('RUN', 'get_state_src')
object.add_method('INITIALIZED', 'set_momentum_kicks')
object.add_method('INITIALIZED', 'set_isothermal')
object.add_method('INITIALIZED', 'set_boxsize')
object.add_method('INITIALIZED', 'set_globalHefraction')
object.add_method('INITIALIZED', 'set_raynumber')
object.add_method('INITIALIZED', 'set_iontempsolver')
object.add_method('INITIALIZED', 'set_boundary')
object.add_method('INITIALIZED', 'set_H_caseA')
object.add_method('INITIALIZED', 'set_He_caseA')
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
handler.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
handler.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('EDIT', 'before_get_parameter')
handler.add_method('UPDATE','before_get_parameter')
handler.add_method('EVOLVED','before_get_parameter')
handler.add_method('EDIT', 'new_gas_particle')
handler.add_method('EDIT', 'remove_gas_particle')
handler.add_method('EDIT', 'new_src_particle')
handler.add_method('EDIT', 'remove_src_particle')
handler.add_transition('EDIT', 'RUN', 'commit_particles')
handler.add_transition('RUN', 'UPDATE', 'new_gas_particle', False)
handler.add_transition('RUN', 'UPDATE', 'remove_gas_particle', False)
handler.add_transition('RUN', 'UPDATE', 'new_src_particle', False)
handler.add_transition('RUN', 'UPDATE', 'remove_src_particle', False)
handler.add_transition('RUN', 'UPDATE', 'set_pos_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_rho_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_hsml_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_u_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_dudt_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_state_gas', False)
handler.add_transition('RUN', 'UPDATE', 'set_state_src', False)
handler.add_transition('RUN', 'UPDATE', 'set_luminosity_src', False)
handler.add_transition('UPDATE', 'RUN', 'recommit_particles')
handler.add_method('RUN', 'evolve_model')
handler.add_method('RUN', 'get_state_gas')
handler.add_method('RUN', 'get_state_src')
handler.add_method('INITIALIZED', 'set_momentum_kicks')
handler.add_method('INITIALIZED', 'set_isothermal')
handler.add_method('INITIALIZED', 'set_boxsize')
handler.add_method('INITIALIZED', 'set_globalHefraction')
handler.add_method('INITIALIZED', 'set_raynumber')
handler.add_method('INITIALIZED', 'set_iontempsolver')
handler.add_method('INITIALIZED', 'set_boundary')
handler.add_method('INITIALIZED', 'set_H_caseA')
handler.add_method('INITIALIZED', 'set_He_caseA')
def __init__(self, mode=MODE_2D, coordinates="cartesian", **options):
self.mode = mode
self.coordinates = coordinates
CommonCode.__init__(self, AdcircInterface(**options), **options)
self._nodes = None
self._elements = None
self._elev_boundary = None
self._flow_boundary = None
self._parameters = None
self.stopping_conditions = StoppingConditions(self)
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method(
"new_particle",
(units.MSun,),
(object.INDEX, object.ERROR_CODE,)
)
object.add_method(
"delete_particle",
(object.INDEX,),
(object.ERROR_CODE,)
)
object.add_method(
"read_usm",
(object.NO_UNIT,),
(object.INDEX, object.ERROR_CODE,)
)
object.add_method(
"add_shell",
(object.INDEX, units.MSun, units.MSun, units.RSun, units.g / units.cm**3, units.barye,
units.K, units.LSun, units.amu, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT,
object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT),
(object.ERROR_CODE,)
)
object.add_method(
"get_stellar_model_element",
(object.INDEX, object.INDEX,),
(units.MSun, units.MSun, units.RSun, units.g / units.cm**3, units.barye,
object.NO_UNIT, units.K, units.LSun, units.amu,
object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT,
object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"get_number_of_zones",
(object.INDEX, ),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"get_number_of_particles",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"merge_two_stars",
(object.INDEX, object.INDEX,),
(object.INDEX, object.ERROR_CODE,)
)
object.add_method("get_target_n_shells_mixing", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_target_n_shells_mixing", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_target_n_shells", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_target_n_shells", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_number_of_particles", (), (object.NO_UNIT, object.ERROR_CODE,))
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method(
"new_particle",
(units.MSun,),
(handler.INDEX, handler.ERROR_CODE,)
)
handler.add_method(
"delete_particle",
(handler.INDEX,),
(handler.ERROR_CODE,)
)
handler.add_method(
"read_usm",
(handler.NO_UNIT,),
(handler.INDEX, handler.ERROR_CODE,)
)
handler.add_method(
"add_shell",
(handler.INDEX, units.MSun, units.MSun, units.RSun, units.g / units.cm**3, units.barye,
units.K, units.LSun, units.amu, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT,
handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT),
(handler.ERROR_CODE,)
)
handler.add_method(
"get_stellar_model_element",
(handler.INDEX, handler.INDEX,),
(units.MSun, units.MSun, units.RSun, units.g / units.cm**3, units.barye,
handler.NO_UNIT, units.K, units.LSun, units.amu,
handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT,
handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.NO_UNIT, handler.ERROR_CODE,)
)
handler.add_method(
"get_number_of_zones",
(handler.INDEX, ),
(handler.NO_UNIT, handler.ERROR_CODE,)
)
handler.add_method(
"get_number_of_particles",
(),
(handler.NO_UNIT, handler.ERROR_CODE,)
)
handler.add_method(
"merge_two_stars",
(handler.INDEX, handler.INDEX,),
(handler.INDEX, handler.ERROR_CODE,)
)
handler.add_method("get_target_n_shells_mixing", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_target_n_shells_mixing", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_target_n_shells", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_target_n_shells", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_number_of_particles", (), (handler.NO_UNIT, handler.ERROR_CODE,))
def __init__(self,
mode=POPInterface.MODE_NORMAL,
namelist_file="pop_in",
**options):
self._nml_file = namelist_file
self.nprocs = options.setdefault('number_of_workers', 8)
CodeWithNamelistParameters.__init__(self,
options.get("parameters", dict()))
CommonCode.__init__(self, POPInterface(mode=mode, **options),
**options)
self.parameters.namelist_file = self._nml_file
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('END', 'INITIALIZED', 'initialize_code', False)
object.add_transition('INITIALIZED', 'EDIT', 'commit_parameters')
object.add_transition('RUN', 'CHANGE_PARAMETERS_RUN',
'before_set_parameter', False)
object.add_transition('EDIT', 'CHANGE_PARAMETERS_EDIT',
'before_set_parameter', False)
object.add_transition('UPDATE', 'CHANGE_PARAMETERS_UPDATE',
'before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN', 'RUN',
'recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT', 'EDIT',
'recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE', 'UPDATE',
'recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE', 'before_get_parameter')
object.add_method('EDIT', 'new_star_particle')
object.add_method('EDIT', 'new_gas_particle')
object.add_method('EDIT', 'new_sphere_particle')
object.add_method('EDIT', 'new_marker_particle')
object.add_method('EDIT', 'delete_particle')
object.add_method('UPDATE', 'new_star_particle')
object.add_method('UPDATE', 'new_gas_particle')
object.add_method('UPDATE', 'new_sphere_particle')
object.add_method('UPDATE', 'new_marker_particle')
object.add_method('UPDATE', 'delete_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_star_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_gas_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_sphere_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_marker_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_method('RUN', 'store_view')
object.add_method('RUN', 'set_position')
object.add_method('RUN', 'set_color')
object.add_method('RUN', 'set_opacity')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_color')
object.add_method('RUN', 'get_opacity')
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("new_particle", [nbody_system.mass] + [nbody_system.length]*3,
(object.INDEX, object.ERROR_CODE))
object.add_method(
"get_eps2",
(),
(nbody_system.length * nbody_system.length, object.ERROR_CODE,)
)
object.add_method(
"set_eps2",
(nbody_system.length * nbody_system.length, ),
(object.ERROR_CODE,)
)
def define_state(self, handler):
CommonCode.define_state(self, handler)
#handler.add_transition('END', 'INITIALIZED', 'initialize_code', False)
handler.add_method('INITIALIZED', 'set_parallel_decomposition')
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
handler.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
handler.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('EDIT', 'before_get_parameter')
handler.add_transition('EDIT', 'RUN', 'initialize_grid')
handler.add_method('RUN', 'evolve_model')
handler.add_method('RUN', 'get_hydro_state_at_point')
for state in ['EDIT', 'RUN']:
for methodname in [
'get_grid_state',
'set_grid_state',
'get_grid_density',
'set_grid_density',
'set_grid_energy_density',
'get_grid_energy_density',
'get_grid_momentum_density',
'set_grid_momentum_density',
'get_position_of_index',
'get_index_of_position',
'set_grid_scalar',
'get_grid_scalar',
'get_mesh_size',
'set_gravity_field',
'get_gravity_field',
'get_boundary_state',
'set_boundary_state',
'get_boundary_position_if_index',
'get_boundary_index_range_inclusive'
]:
handler.add_method(state, methodname)
self.stopping_conditions.define_state(handler)
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('END', 'INITIALIZED', 'initialize_code', False)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('UPDATE','CHANGE_PARAMETERS_UPDATE','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_UPDATE','UPDATE','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_UPDATE','before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_method('UPDATE','before_get_parameter')
object.add_method('EDIT', 'new_star_particle')
object.add_method('EDIT', 'new_gas_particle')
object.add_method('EDIT', 'new_sphere_particle')
object.add_method('EDIT', 'new_marker_particle')
object.add_method('EDIT', 'delete_particle')
object.add_method('UPDATE', 'new_star_particle')
object.add_method('UPDATE', 'new_gas_particle')
object.add_method('UPDATE', 'new_sphere_particle')
object.add_method('UPDATE', 'new_marker_particle')
object.add_method('UPDATE', 'delete_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_star_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_gas_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_sphere_particle', False)
object.add_transition('RUN', 'UPDATE', 'new_marker_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_method('RUN', 'store_view')
object.add_method('RUN', 'set_position')
object.add_method('RUN', 'set_color')
object.add_method('RUN', 'set_opacity')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_color')
object.add_method('RUN', 'get_opacity')
def define_state(self, object):
CommonCode.define_state(self, object)
#object.add_transition('END', 'INITIALIZED', 'initialize_code', False)
object.add_method('INITIALIZED', 'set_parallel_decomposition')
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_transition('EDIT', 'RUN', 'initialize_grid')
object.add_method('RUN', 'evolve_model')
object.add_method('RUN', 'get_hydro_state_at_point')
for state in ['EDIT', 'RUN']:
for methodname in [
'get_grid_state',
'set_grid_state',
'get_grid_density',
'set_grid_density',
'set_grid_energy_density',
'get_grid_energy_density',
'get_grid_momentum_density',
'set_grid_momentum_density',
'get_position_of_index',
'get_index_of_position',
'set_grid_scalar',
'get_grid_scalar',
'get_mesh_size',
'set_gravity_field',
'get_gravity_field',
'get_boundary_state',
'set_boundary_state',
'get_boundary_position_if_index',
'get_boundary_index_range_inclusive'
]:
object.add_method(state, methodname)
self.stopping_conditions.define_state(object)
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED', 'RUN', 'commit_parameters')
handler.add_transition('RUN', 'CHANGE_PARAMETERS_RUN', 'before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_RUN', 'RUN', 'recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
GravityFieldCode.define_state(self, handler)
handler.add_method('RUN', 'get_local_density')
handler.add_method('RUN', 'get_velcirc')
handler.add_method('RUN', 'get_epifreq')
handler.add_method('RUN', 'get_spiral_density')
handler.add_method('RUN', 'get_tidal_tensor')
handler.add_method('RUN', 'get_eigen_values')
handler.add_method('RUN', 'get_tidal_radius')
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('RUN','PARAMETER_CHANGE_A','invoke_state_change2')
handler.add_transition('EDIT','PARAMETER_CHANGE_B','invoke_state_change2')
handler.add_transition('PARAMETER_CHANGE_A','RUN','recommit_parameters')
handler.add_transition('PARAMETER_CHANGE_B','EDIT','recommit_parameters')
handler.add_method('EDIT', 'new_particle')
handler.add_method('EDIT', 'delete_particle')
handler.add_transition('EDIT', 'RUN', 'commit_particles')
handler.add_transition('RUN', 'UPDATE', 'new_particle', False)
handler.add_transition('RUN', 'UPDATE', 'delete_particle', False)
handler.add_transition('UPDATE', 'RUN', 'recommit_particles')
handler.add_method('RUN', 'evolve_model')
handler.add_method('RUN', 'get_state')
handler.add_method('RUN', 'get_abundance')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','PARAMETER_CHANGE_A','invoke_state_change2')
object.add_transition('EDIT','PARAMETER_CHANGE_B','invoke_state_change2')
object.add_transition('PARAMETER_CHANGE_A','RUN','recommit_parameters')
object.add_transition('PARAMETER_CHANGE_B','EDIT','recommit_parameters')
object.add_method('EDIT', 'new_particle')
object.add_method('EDIT', 'delete_particle')
object.add_transition('EDIT', 'RUN', 'commit_particles')
object.add_transition('RUN', 'UPDATE', 'new_particle', False)
object.add_transition('RUN', 'UPDATE', 'delete_particle', False)
object.add_transition('UPDATE', 'RUN', 'recommit_particles')
object.add_method('RUN', 'evolve_model')
object.add_method('RUN', 'get_state')
object.add_method('RUN', 'get_abundance')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED', 'RUN', 'commit_parameters')
object.add_transition('RUN', 'CHANGE_PARAMETERS_RUN', 'before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN', 'RUN', 'recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
GravityFieldCode.define_state(self, object)
object.add_method('RUN', 'get_local_density')
object.add_method('RUN', 'get_velcirc')
object.add_method('RUN', 'get_epifreq')
object.add_method('RUN', 'get_spiral_density')
object.add_method('RUN', 'get_tidal_tensor')
object.add_method('RUN', 'get_eigen_values')
object.add_method('RUN', 'get_tidal_radius')
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method("generate_particles", (), (handler.ERROR_CODE,))
handler.add_method("get_number_of_particles_updated", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("get_mass", (handler.INDEX,),
(nbody_system.mass, handler.ERROR_CODE)
)
handler.add_method("get_position", (handler.INDEX,),
(nbody_system.length, nbody_system.length, nbody_system.length, handler.ERROR_CODE)
)
handler.add_method("get_velocity", (handler.INDEX,),
(nbody_system.speed, nbody_system.speed, nbody_system.speed, handler.ERROR_CODE)
)
handler.add_method("get_model_alpha", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_model_alpha", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_model_beta", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_model_beta", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_model_gamma", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_model_gamma", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_total_mass", (), (nbody_system.mass, handler.ERROR_CODE,))
handler.add_method("set_total_mass", (nbody_system.mass, ), (handler.ERROR_CODE,))
handler.add_method("get_scale_radius", (), (nbody_system.length, handler.ERROR_CODE,))
handler.add_method("set_scale_radius", (nbody_system.length, ), (handler.ERROR_CODE,))
handler.add_method("get_cutoff_radius", (), (nbody_system.length, handler.ERROR_CODE,))
handler.add_method("set_cutoff_radius", (nbody_system.length, ), (handler.ERROR_CODE,))
handler.add_method("get_target_number_of_particles", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_target_number_of_particles", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_black_hole_mass", (), (nbody_system.mass, handler.ERROR_CODE,))
handler.add_method("set_black_hole_mass", (nbody_system.mass, ), (handler.ERROR_CODE,))
handler.add_method("get_random_seed", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_random_seed", (handler.NO_UNIT, ), (handler.ERROR_CODE,))
handler.add_method("get_output_path", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_output_path", (handler.NO_UNIT,), (handler.ERROR_CODE,))
handler.add_method("get_output_basename", (), (handler.NO_UNIT, handler.ERROR_CODE,))
handler.add_method("set_output_basename", (handler.NO_UNIT,), (handler.ERROR_CODE,))
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE,))
object.add_method("get_number_of_particles_updated", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("get_mass", (object.INDEX,),
(nbody_system.mass, object.ERROR_CODE)
)
object.add_method("get_position", (object.INDEX,),
(nbody_system.length, nbody_system.length, nbody_system.length, object.ERROR_CODE)
)
object.add_method("get_velocity", (object.INDEX,),
(nbody_system.speed, nbody_system.speed, nbody_system.speed, object.ERROR_CODE)
)
object.add_method("get_model_alpha", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_model_alpha", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_model_beta", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_model_beta", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_model_gamma", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_model_gamma", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_total_mass", (), (nbody_system.mass, object.ERROR_CODE,))
object.add_method("set_total_mass", (nbody_system.mass, ), (object.ERROR_CODE,))
object.add_method("get_scale_radius", (), (nbody_system.length, object.ERROR_CODE,))
object.add_method("set_scale_radius", (nbody_system.length, ), (object.ERROR_CODE,))
object.add_method("get_cutoff_radius", (), (nbody_system.length, object.ERROR_CODE,))
object.add_method("set_cutoff_radius", (nbody_system.length, ), (object.ERROR_CODE,))
object.add_method("get_target_number_of_particles", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_target_number_of_particles", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_black_hole_mass", (), (nbody_system.mass, object.ERROR_CODE,))
object.add_method("set_black_hole_mass", (nbody_system.mass, ), (object.ERROR_CODE,))
object.add_method("get_random_seed", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_random_seed", (object.NO_UNIT, ), (object.ERROR_CODE,))
object.add_method("get_output_path", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_output_path", (object.NO_UNIT,), (object.ERROR_CODE,))
object.add_method("get_output_basename", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_output_basename", (object.NO_UNIT,), (object.ERROR_CODE,))
def define_state(self, handler):
CommonCode.define_state(self, handler)
handler.add_transition('INITIALIZED','EDIT','commit_parameters')
handler.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
handler.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
handler.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
handler.add_method('RUN', 'before_get_parameter')
handler.add_method('EDIT', 'before_get_parameter')
handler.add_transition('EDIT', 'RUN', 'generate_particles', False)
handler.add_transition('RUN', 'EDIT', 'clear_particle_set')
handler.add_method('EDIT', 'get_number_of_particles_updated')
handler.add_method('RUN', 'get_number_of_particles_updated')
handler.add_method('RUN', 'get_state')
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method('evolve_model', (units.s, ), (handler.ERROR_CODE, ))
handler.add_method("new_particle", (
units.cm**-3,
units.K,
units.s**-1,
), (
handler.INDEX,
handler.ERROR_CODE,
))
handler.add_method("set_state", (
handler.NO_UNIT,
units.cm**-3,
units.K,
units.s**-1,
), (handler.ERROR_CODE, ))
handler.add_method("get_state", (handler.INDEX, ), (
units.cm**-3,
units.K,
units.s**-1,
handler.ERROR_CODE,
))
handler.add_method("get_abundance", (
handler.INDEX,
handler.INDEX,
), (
handler.NO_UNIT,
handler.ERROR_CODE,
))
handler.add_method("set_abundance", (
handler.INDEX,
handler.INDEX,
handler.NO_UNIT,
), (handler.ERROR_CODE, ))
handler.add_method("delete_particle", (handler.INDEX, ),
(handler.ERROR_CODE, ))
handler.add_method("get_firstlast_abundance", (), (
handler.NO_UNIT,
handler.NO_UNIT,
handler.ERROR_CODE,
))
handler.add_method("get_time", (), (
units.s,
handler.ERROR_CODE,
))
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition("INITIALIZED", "EDIT", "commit_parameters")
object.add_transition("EDIT", "CHANGE_PARAMETERS_EDIT", "before_set_parameter", False)
object.add_transition("CHANGE_PARAMETERS_EDIT", "EDIT", "recommit_parameters")
object.add_method("CHANGE_PARAMETERS_EDIT", "before_set_parameter")
object.add_method("CHANGE_PARAMETERS_EDIT", "before_get_parameter")
object.add_method("RUN", "before_get_parameter")
object.add_method("EDIT", "before_get_parameter")
object.add_transition("EDIT", "RUN", "generate_particles", False)
object.add_transition("RUN", "EDIT", "clear_particle_set")
object.add_method("EDIT", "get_number_of_particles_updated")
object.add_method("RUN", "get_number_of_particles_updated")
object.add_method("RUN", "get_state")
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method('evolve_model', (units.s, ), (object.ERROR_CODE, ))
object.add_method("new_particle", (
units.cm**-3,
units.K,
units.s**-1,
), (
object.INDEX,
object.ERROR_CODE,
))
object.add_method("set_state", (
object.NO_UNIT,
units.cm**-3,
units.K,
units.s**-1,
), (object.ERROR_CODE, ))
object.add_method("get_state", (object.INDEX, ), (
units.cm**-3,
units.K,
units.s**-1,
object.ERROR_CODE,
))
object.add_method("get_abundance", (
object.INDEX,
object.INDEX,
), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set_abundance", (
object.INDEX,
object.INDEX,
object.NO_UNIT,
), (object.ERROR_CODE, ))
object.add_method("delete_particle", (object.INDEX, ),
(object.ERROR_CODE, ))
object.add_method("get_firstlast_abundance", (), (
object.NO_UNIT,
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("get_time", (), (
units.s,
object.ERROR_CODE,
))
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE,))
object.add_method("get_number_of_particles_updated", (), (object.NO_UNIT, object.ERROR_CODE))
object.add_method(
"get_state", (object.INDEX,), [nbody_system.length] * 3 + [nbody_system.speed] * 3 + [object.ERROR_CODE]
)
object.add_method("get_target_number_of_particles", (), (object.NO_UNIT, object.ERROR_CODE))
object.add_method("set_target_number_of_particles", (object.NO_UNIT,), (object.ERROR_CODE,))
object.add_method("get_fractal_dimension", (), (object.NO_UNIT, object.ERROR_CODE))
object.add_method("set_fractal_dimension", (object.NO_UNIT,), (object.ERROR_CODE,))
object.add_method("get_random_seed", (), (object.NO_UNIT, object.ERROR_CODE))
object.add_method("set_random_seed", (object.NO_UNIT,), (object.ERROR_CODE,))
def __init__(self, **options):
#print('__init__')
self.stopping_conditions = StoppingConditions(self)
self.itot = 0
self.ktot = 0
self.latitudes = None
self.longitudes = None
self.exp_name = 'TEST'
CommonCode.__init__(self, OpenIFSInterface(**options), **options)
#if workdir is given, assume inputfile, backupfile are relative to workdir
#TODO should store the changed file names - they are used in cleanup_code
inputfile = OpenIFS.inputfile
backupfile = OpenIFS.backupfile
if 'workdir' in options:
# print('OpenIFS.__init__() : setting workdir.')
self.set_workdir(options['workdir'])
inputfile = os.path.join(options['workdir'], inputfile)
backupfile = os.path.join(options['workdir'], backupfile)
if not os.path.exists(inputfile):
print('inputfile:' + inputfile)
print('cwd:' + os.getcwd())
raise Exception(
"File fort.4 not found. Creating an openIFS model from scratch is not supported yet."
)
else:
os.rename(inputfile, backupfile)
self.params = f90nml.read(backupfile)
self.patch = {
"NAMPAR0": {
"NPROC": options.get("number_of_workers", 1)
}
}
restart_steps = options.get("restart_steps", None)
if restart_steps:
self.patch["NAMRES"] = {"NFRRES": restart_steps}
f90nml.patch(backupfile, self.patch, inputfile)
print('Done patching openIFS input file', inputfile)
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method("new_particle",
[nbody_system.mass] + [nbody_system.length] * 3,
(handler.INDEX, handler.ERROR_CODE))
handler.add_method("get_eps2", (), (
nbody_system.length * nbody_system.length,
handler.ERROR_CODE,
))
handler.add_method("set_eps2",
(nbody_system.length * nbody_system.length, ),
(handler.ERROR_CODE, ))
handler.add_method("set_mass", (
handler.NO_UNIT,
nbody_system.mass,
), (handler.ERROR_CODE))
handler.add_method("get_mass", (handler.NO_UNIT, ),
(nbody_system.mass, handler.ERROR_CODE))
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE,))
object.add_method("get_number_of_particles_updated", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("get_mass", (object.INDEX,),
(nbody_system.mass, object.ERROR_CODE)
)
object.add_method("get_position", (object.INDEX,),
(nbody_system.length, nbody_system.length, nbody_system.length, object.ERROR_CODE)
)
object.add_method("get_velocity", (object.INDEX,),
(nbody_system.speed, nbody_system.speed, nbody_system.speed, object.ERROR_CODE)
)
object.add_method("get_output_path", (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set_output_path", (object.NO_UNIT,), (object.ERROR_CODE,))
for par in ["_number_of_grid_intervals", "_order_of_multipole_expansion",
"_number_of_radial_steps_correction_fns", "_number_of_iterations",
"_halo_number_of_particles", "_halo_random_seed",
"_bulge_number_of_particles", "_bulge_random_seed",
"_disk_number_of_particles", "_disk_random_seed",
"_inner_cusp_slope", "_outer_slope", "_Sersic_index_n", "_central_radial_vel_dispersion",
"_halo_streaming_fraction", "_bulge_streaming_fraction"]:
object.add_method("get"+par, (), (object.NO_UNIT, object.ERROR_CODE,))
object.add_method("set"+par, (object.NO_UNIT, ), (object.ERROR_CODE,))
for par in ["_halo_outer_radius", "_scale_radius", "_truncation_delta_r",
"_disk_scale_length", "_disk_outer_radius", "_disk_scale_height_sech2",
"_disk_truncation_dr", "_bulge_scale_radius", "_radial_grid_delta_r",
"_scale_length_of_sigR2"]:
object.add_method("get"+par, (), (nbody_system.length, object.ERROR_CODE,))
object.add_method("set"+par, (nbody_system.length, ), (object.ERROR_CODE,))
for par in ["_scale_velocity", "_bulge_velocity"]:
object.add_method("get"+par, (), (nbody_system.speed, object.ERROR_CODE,))
object.add_method("set"+par, (nbody_system.speed, ), (object.ERROR_CODE,))
object.add_method("get_disk_mass", (), (nbody_system.mass, object.ERROR_CODE,))
object.add_method("set_disk_mass", (nbody_system.mass, ), (object.ERROR_CODE,))
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('INITIALIZED', 'EDIT', 'commit_parameters')
object.add_transition('RUN', 'PARAMETER_CHANGE_A',
'invoke_state_change2')
object.add_transition('EDIT', 'PARAMETER_CHANGE_B',
'invoke_state_change2')
object.add_transition('PARAMETER_CHANGE_A', 'RUN',
'recommit_parameters')
object.add_transition('PARAMETER_CHANGE_B', 'EDIT',
'recommit_parameters')
object.add_transition('EDIT', 'UPDATE', 'generate_particles', False)
object.add_transition('UPDATE', 'RUN', 'update_particle_set')
object.add_transition('RUN', 'EDIT', 'clear_particle_set')
object.add_method('RUN', 'invoke_state_change_updated')
object.add_method('EDIT', 'get_number_of_particles_updated')
object.add_method('UPDATE', 'get_number_of_particles_updated')
object.add_method('RUN', 'get_number_of_particles_updated')
object.add_method('RUN', 'get_mass')
object.add_method('RUN', 'get_position')
object.add_method('RUN', 'get_velocity')
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition("INITIALIZED", "EDIT", "commit_parameters")
object.add_transition("RUN", "CHANGE_PARAMETERS_RUN", "before_set_parameter", False)
object.add_transition("EDIT", "CHANGE_PARAMETERS_EDIT", "before_set_parameter", False)
object.add_transition("UPDATE", "CHANGE_PARAMETERS_UPDATE", "before_set_parameter", False)
object.add_transition("CHANGE_PARAMETERS_RUN", "RUN", "recommit_parameters")
object.add_transition("CHANGE_PARAMETERS_EDIT", "EDIT", "recommit_parameters")
object.add_transition("CHANGE_PARAMETERS_UPDATE", "UPDATE", "recommit_parameters")
object.add_method("CHANGE_PARAMETERS_RUN", "before_set_parameter")
object.add_method("CHANGE_PARAMETERS_EDIT", "before_set_parameter")
object.add_method("CHANGE_PARAMETERS_UPDATE", "before_set_parameter")
object.add_method("CHANGE_PARAMETERS_RUN", "model_present")
object.add_method("CHANGE_PARAMETERS_EDIT", "model_present")
object.add_method("CHANGE_PARAMETERS_UPDATE", "model_present")
object.add_method("INITIALIZED", "model_present")
object.add_method("CHANGE_PARAMETERS_RUN", "before_get_parameter")
object.add_method("CHANGE_PARAMETERS_EDIT", "before_get_parameter")
object.add_method("CHANGE_PARAMETERS_UPDATE", "before_get_parameter")
object.add_method("RUN", "before_get_parameter")
object.add_method("EDIT", "before_get_parameter")
object.add_method("UPDATE", "before_get_parameter")
object.add_transition("EDIT", "UPDATE", "generate_particles", False)
object.add_transition("UPDATE", "RUN", "update_particle_set")
object.add_transition("RUN", "EDIT", "clear_particle_set")
object.add_method("RUN", "invoke_state_change_updated")
object.add_method("EDIT", "get_number_of_particles_updated")
object.add_method("UPDATE", "get_number_of_particles_updated")
object.add_method("RUN", "get_number_of_particles_updated")
object.add_method("RUN", "get_number_of_particles")
object.add_method("RUN", "get_mass")
object.add_method("RUN", "get_position")
object.add_method("RUN", "get_velocity")
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method('evolve_model', (units.Myr,), ( object.ERROR_CODE, ))
object.add_method(
"new_particle",
(
units.parsec,
units.parsec,
units.parsec,
units.amu / units.cm**3,
1.0e48 / units.s,
object.NO_UNIT,
units.cm**2 / units.s**2,
object.NO_UNIT
),
(
object.INDEX,
object.ERROR_CODE,
)
)
object.add_method(
"delete_particle",
(
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_state",
(
object.NO_UNIT,
),
(
units.parsec,
units.parsec,
units.parsec,
units.amu / units.cm**3,
1.0e48 / units.s,
object.NO_UNIT,
units.cm**2 / units.s**2,
object.NO_UNIT,
object.ERROR_CODE
)
)
object.add_method(
"set_state",
(
object.NO_UNIT,
units.parsec,
units.parsec,
units.parsec,
units.amu / units.cm**3,
1.0e48 / units.s,
object.NO_UNIT,
units.cm**2 / units.s**2,
object.NO_UNIT
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_internal_energy",
(
object.NO_UNIT,
),
(
units.cm**2 / units.s**2,
object.ERROR_CODE
)
)
object.add_method(
"set_internal_energy",
(
object.NO_UNIT,
units.cm**2 / units.s**2,
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_dinternal_energy_dt",
(
object.NO_UNIT,
),
(
units.cm**2 / units.s**3,
object.ERROR_CODE
)
)
object.add_method(
"set_dinternal_energy_dt",
(
object.NO_UNIT,
units.cm**2 / units.s**3,
),
(
object.ERROR_CODE
)
)
object.add_method(
"set_position",
(
object.NO_UNIT,
units.parsec,
units.parsec,
units.parsec,
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_position",
(
object.NO_UNIT,
),
(
units.parsec,
units.parsec,
units.parsec,
object.ERROR_CODE
)
)
object.add_method(
"set_density",
(
object.NO_UNIT,
units.amu / units.cm**3,
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_density",
(
object.NO_UNIT,
),
(
units.amu / units.cm**3,
object.ERROR_CODE
)
)
object.add_method(
"set_flux",
(
object.NO_UNIT,
1.0e48 / units.s,
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_flux",
(
object.NO_UNIT,
),
(
1.0e48 / units.s,
object.ERROR_CODE
)
)
object.add_method(
"get_mean_intensity",
(
object.NO_UNIT,
),
(
1.0e48 / units.s,
object.ERROR_CODE
)
)
object.add_method(
"get_diffuse_intensity",
(
object.NO_UNIT,
),
(
1.0e48 / units.s,
object.ERROR_CODE
)
)
object.add_method(
"set_ionisation",
(
object.NO_UNIT,
object.NO_UNIT
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_ionisation",
(
object.NO_UNIT,
),
(
object.NO_UNIT,
object.ERROR_CODE
)
)
object.add_method(
"set_metallicity",
(
object.NO_UNIT,
object.NO_UNIT
),
(
object.ERROR_CODE
)
)
object.add_method(
"get_metallicity",
(
object.NO_UNIT,
),
(
object.NO_UNIT,
object.ERROR_CODE
)
)
object.add_method(
'commit_particles',
(),
(object.ERROR_CODE)
)
object.add_method(
'recommit_particles',
(),
(object.ERROR_CODE)
)
object.add_method(
"get_timestep",
(),
(units.Myr, object.ERROR_CODE,)
)
object.add_method(
"set_timestep",
(units.Myr, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_source_Teff",
(),
(units.K, object.ERROR_CODE,)
)
object.add_method(
"set_source_Teff",
(units.K, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_hilbert_order",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_hilbert_order",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"set_blackbody_spectrum",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_blackbody_spectrum",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"get_thermal_evolution",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_thermal_evolution",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_collisional_ionization",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_collisional_ionization",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_number_frequency_bins",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_number_frequency_bins",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_metal_cooling",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_metal_cooling",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_recombination_radiation",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_recombination_radiation",
(object.NO_UNIT, ),
(object.ERROR_CODE,)
)
object.add_method(
"get_box_size",
(),
(units.parsec, object.ERROR_CODE,)
)
object.add_method(
"set_box_size",
(units.parsec, ),
(object.ERROR_CODE,)
)
object.add_method(
'get_time',
(),
(units.s,object.ERROR_CODE,)
)
object.add_method(
'set_time',
(units.s,),
(object.ERROR_CODE,)
)
object.add_method(
"get_simplex_data_directory",
(),
(object.NO_UNIT, object.ERROR_CODE,)
)
object.add_method(
"set_simplex_data_directory",
(object.NO_UNIT,),
(object.ERROR_CODE,)
)
def __init__(self, unit_converter = None, **options):
self.unit_converter = unit_converter
legacy_interface = BarAndSpiralsInterface(**options)
self.interface_doc = legacy_interface.__doc__
CommonCode.__init__(self, legacy_interface, **options)
def __init__(self, unit_converter = None, **options):
self.unit_converter = unit_converter
legacy_interface = FastKickInterface(**options)
self.legacy_doc = legacy_interface.__doc__
CommonCode.__init__(self, legacy_interface, **options)
def define_methods(self, object):
CommonCode.define_methods(self, object)
def __init__(self, unit_converter=None, **options):
self.unit_converter = unit_converter
CommonCode.__init__(self, KeplerInterface(**options), **options)
def __init__(self, unit_converter=None, **options):
self.unit_converter = unit_converter
self.stopping_conditions = StoppingConditions(self)
CommonCode.__init__(self, CapreoleInterface(**options), **options)
def define_state(self, object):
CommonCode.define_state(self, object)
object.add_transition('END', 'INITIALIZED', 'initialize_code', False)
object.add_transition('INITIALIZED','EDIT','commit_parameters')
object.add_transition('RUN','CHANGE_PARAMETERS_RUN','before_set_parameter', False)
object.add_transition('EDIT','CHANGE_PARAMETERS_EDIT','before_set_parameter', False)
object.add_transition('CHANGE_PARAMETERS_RUN','RUN','recommit_parameters')
object.add_transition('CHANGE_PARAMETERS_EDIT','EDIT','recommit_parameters')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_set_parameter')
object.add_method('CHANGE_PARAMETERS_RUN', 'before_get_parameter')
object.add_method('CHANGE_PARAMETERS_EDIT', 'before_get_parameter')
object.add_method('RUN', 'before_get_parameter')
object.add_method('EDIT', 'before_get_parameter')
object.add_transition('EDIT', 'RUN', 'initialize_grid')
object.add_method('RUN', 'evolve_model')
object.add_method('RUN', 'get_hydro_state_at_point')
for state in ['EDIT', 'RUN']:
for methodname in [
'get_grid_state',
'set_grid_state',
'get_potential_at_point',
'get_potential',
'get_gravity_at_point',
# 'set_potential',
'get_grid_density',
'set_grid_density',
'set_grid_energy_density',
'get_grid_energy_density',
'get_grid_momentum_density',
'set_grid_momentum_density',
'get_grid_velocity',
'set_grid_velocity',
'get_position_of_index',
'get_index_of_position',
'get_max_refinement',
# 'set_grid_scalar',
# 'get_grid_scalar',
'get_number_of_grids',
'get_index_range_inclusive',
'get_cell_volume',
'get_number_of_procs',
'get_all_local_num_grids',
'get_data_all_local_blks',
'get_1blk_cell_coords',
'get_leaf_indices',
'kick_grid',
'kick_block',
# 'get_boundary_state',
# 'set_boundary_state',
# 'get_boundary_position_if_index',
# 'get_boundary_index_range_inclusive',
'get_timestep',
'set_timestep',
'get_end_time',
'set_end_time',
'get_time',
'get_max_num_steps',
'set_max_num_steps',
'get_restart',
'get_number_of_particles',
'get_accel_gas_on_particles',
'get_particle_position',
'set_particle_position',
'get_particle_velocity',
'set_particle_velocity',
'make_sink',
'get_particle_mass',
'set_particle_mass',
'set_particle_gpot',
'get_particle_gpot',
'get_particle_tags',
'get_particle_proc',
'get_particle_block',
'get_new_tags',
'get_number_of_new_tags',
'clear_new_tags',
'particles_gather',
#'make_particle_tree',
'write_chpt'
]:
object.add_method(state, methodname)
def __init__(self, unit_converter=None, **options):
self.unit_converter = unit_converter
legacy_interface = FastKickInterface(**options)
self.legacy_doc = legacy_interface.__doc__
CommonCode.__init__(self, legacy_interface, **options)
def define_methods(self, handler):
CommonCode.define_methods(self, handler)
handler.add_method('evolve_model', (generic_unit_system.time,), ( handler.ERROR_CODE, ))
handler.add_method(
"new_gas_particle",
(
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
handler.NO_UNIT,
generic_unit_system.speed**2,
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
),
(
handler.INDEX,
handler.ERROR_CODE,
)
)
handler.add_method(
"set_state_gas",
(
handler.NO_UNIT,
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
handler.NO_UNIT,
generic_unit_system.speed**2
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_vel_gas",
(
handler.NO_UNIT,
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_pos_gas",
(
handler.NO_UNIT,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_rho_gas",
(
handler.NO_UNIT,
generic_unit_system.density,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_hsml_gas",
(
handler.NO_UNIT,
generic_unit_system.length,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_u_gas",
(
handler.NO_UNIT,
generic_unit_system.speed**2
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_dudt_gas",
(
handler.NO_UNIT,
generic_unit_system.length**2/generic_unit_system.time**3
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_dudt_gas",
(
handler.INDEX,
),
(
generic_unit_system.length**2/generic_unit_system.time**3,
handler.ERROR_CODE,
)
)
handler.add_method(
"get_state_gas",
(
handler.INDEX,
),
(
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
handler.NO_UNIT,
generic_unit_system.speed**2,
handler.ERROR_CODE,
)
)
handler.add_method(
"get_vel_gas",
(
handler.INDEX,
),
(
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
handler.ERROR_CODE,
)
)
handler.add_method(
"new_src_particle",
(
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
handler.NO_UNIT,
),
(
handler.INDEX,
handler.ERROR_CODE,
)
)
handler.add_method(
"set_state_src",
(
handler.NO_UNIT,
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
handler.NO_UNIT,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_luminosity_src",
(
handler.NO_UNIT,
1.e50 * units.s**-1,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_state_src",
(
handler.INDEX,
),
(
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
handler.NO_UNIT,
handler.ERROR_CODE,
)
)
handler.add_method(
"remove_src_particle",
(
handler.INDEX,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"remove_gas_particle",
(
handler.INDEX,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_boxsize",
(
generic_unit_system.length,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_defaultspectype",
(
handler.NO_UNIT,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"set_globalHefraction",
(
handler.NO_UNIT,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_boxsize",
(
),
(
generic_unit_system.length,
handler.ERROR_CODE,
)
)
handler.add_method(
"get_globalHefraction",
(
),
(
handler.NO_UNIT,
handler.ERROR_CODE,
)
)
handler.add_method(
"get_defaultspectype",
(
),
(
handler.NO_UNIT,
handler.ERROR_CODE,
)
)
handler.add_method(
"set_raynumber",
(
generic_unit_system.time**-1,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_raynumber",
(
),
(
generic_unit_system.time**-1,
handler.ERROR_CODE,
)
)
handler.add_method(
"set_iontempsolver",
(
handler.NO_UNIT,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_iontempsolver",
(
),
(
handler.NO_UNIT,
handler.ERROR_CODE,
)
)
handler.add_method(
"set_boundary",
(
handler.NO_UNIT,
),
(
handler.ERROR_CODE,
)
)
handler.add_method(
"get_boundary",
(
),
(
handler.NO_UNIT,
handler.ERROR_CODE,
)
)
handler.add_method(
"get_time",
(
),
(
generic_unit_system.time,
handler.ERROR_CODE,
)
)
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method(
'evolve_model',
(
units.s,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"new_particle",
(
units.cm**-3,
units.K,
units.s**-1,
),
(
object.INDEX,
object.ERROR_CODE,
)
)
object.add_method(
"set_state",
(
object.NO_UNIT,
units.cm**-3,
units.K,
units.s**-1,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_state",
(
object.INDEX,
),
(
units.cm**-3,
units.K,
units.s**-1,
object.ERROR_CODE,
)
)
object.add_method(
"get_abundance",
(
object.INDEX,
object.INDEX,
),
(
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"set_abundance",
(
object.INDEX,
object.INDEX,
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"delete_particle",
(
object.INDEX,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_firstlast_abundance",
(
),
(
object.NO_UNIT,
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"get_time",
(
),
(
units.s,
object.ERROR_CODE,
)
)
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method("generate_particles", (), (object.ERROR_CODE, ))
object.add_method("get_number_of_particles_updated", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("get_mass", (object.INDEX, ),
(nbody_system.mass, object.ERROR_CODE))
object.add_method("get_position", (object.INDEX, ),
(nbody_system.length, nbody_system.length,
nbody_system.length, object.ERROR_CODE))
object.add_method("get_velocity", (object.INDEX, ),
(nbody_system.speed, nbody_system.speed,
nbody_system.speed, object.ERROR_CODE))
object.add_method("get_output_path", (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set_output_path", (object.NO_UNIT, ),
(object.ERROR_CODE, ))
for par in [
"_number_of_grid_intervals", "_order_of_multipole_expansion",
"_number_of_radial_steps_correction_fns",
"_number_of_iterations", "_halo_number_of_particles",
"_halo_random_seed", "_bulge_number_of_particles",
"_bulge_random_seed", "_disk_number_of_particles",
"_disk_random_seed", "_inner_cusp_slope", "_outer_slope",
"_Sersic_index_n", "_central_radial_vel_dispersion",
"_halo_streaming_fraction", "_bulge_streaming_fraction"
]:
object.add_method("get" + par, (), (
object.NO_UNIT,
object.ERROR_CODE,
))
object.add_method("set" + par, (object.NO_UNIT, ),
(object.ERROR_CODE, ))
for par in [
"_halo_outer_radius", "_scale_radius", "_truncation_delta_r",
"_disk_scale_length", "_disk_outer_radius",
"_disk_scale_height_sech2", "_disk_truncation_dr",
"_bulge_scale_radius", "_radial_grid_delta_r",
"_scale_length_of_sigR2"
]:
object.add_method("get" + par, (), (
nbody_system.length,
object.ERROR_CODE,
))
object.add_method("set" + par, (nbody_system.length, ),
(object.ERROR_CODE, ))
for par in ["_scale_velocity", "_bulge_velocity"]:
object.add_method("get" + par, (), (
nbody_system.speed,
object.ERROR_CODE,
))
object.add_method("set" + par, (nbody_system.speed, ),
(object.ERROR_CODE, ))
object.add_method("get_disk_mass", (), (
nbody_system.mass,
object.ERROR_CODE,
))
object.add_method("set_disk_mass", (nbody_system.mass, ),
(object.ERROR_CODE, ))
def define_methods(self, object):
CommonCode.define_methods(self, object)
object.add_method('evolve_model', (generic_unit_system.time,), ( object.ERROR_CODE, ))
object.add_method(
"new_gas_particle",
(
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
object.NO_UNIT,
generic_unit_system.speed**2,
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
),
(
object.INDEX,
object.ERROR_CODE,
)
)
object.add_method(
"set_state_gas",
(
object.NO_UNIT,
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
object.NO_UNIT,
generic_unit_system.speed**2
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_vel_gas",
(
object.NO_UNIT,
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_pos_gas",
(
object.NO_UNIT,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_rho_gas",
(
object.NO_UNIT,
generic_unit_system.density,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_hsml_gas",
(
object.NO_UNIT,
generic_unit_system.length,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_u_gas",
(
object.NO_UNIT,
generic_unit_system.speed**2
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_dudt_gas",
(
object.NO_UNIT,
generic_unit_system.length**2/generic_unit_system.time**3
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_dudt_gas",
(
object.INDEX,
),
(
generic_unit_system.length**2/generic_unit_system.time**3,
object.ERROR_CODE,
)
)
object.add_method(
"get_state_gas",
(
object.INDEX,
),
(
generic_unit_system.mass,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.density,
object.NO_UNIT,
generic_unit_system.speed**2,
object.ERROR_CODE,
)
)
object.add_method(
"get_vel_gas",
(
object.INDEX,
),
(
generic_unit_system.speed,
generic_unit_system.speed,
generic_unit_system.speed,
object.ERROR_CODE,
)
)
object.add_method(
"new_src_particle",
(
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
object.NO_UNIT,
),
(
object.INDEX,
object.ERROR_CODE,
)
)
object.add_method(
"set_state_src",
(
object.NO_UNIT,
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_state_src",
(
object.INDEX,
),
(
1.e50 * units.s**-1,
generic_unit_system.length,
generic_unit_system.length,
generic_unit_system.length,
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"remove_src_particle",
(
object.INDEX,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"remove_gas_particle",
(
object.INDEX,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_boxsize",
(
generic_unit_system.length,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_defaultspectype",
(
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"set_globalHefraction",
(
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_boxsize",
(
),
(
generic_unit_system.length,
object.ERROR_CODE,
)
)
object.add_method(
"get_globalHefraction",
(
),
(
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"get_defaultspectype",
(
),
(
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"set_raynumber",
(
generic_unit_system.time**-1,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_raynumber",
(
),
(
generic_unit_system.time**-1,
object.ERROR_CODE,
)
)
object.add_method(
"set_iontempsolver",
(
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_iontempsolver",
(
),
(
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"set_boundary",
(
object.NO_UNIT,
),
(
object.ERROR_CODE,
)
)
object.add_method(
"get_boundary",
(
),
(
object.NO_UNIT,
object.ERROR_CODE,
)
)
object.add_method(
"get_time",
(
),
(
generic_unit_system.time,
object.ERROR_CODE,
)
)
def __init__(self, unit_converter = None, **options):
self.unit_converter = unit_converter
CommonCode.__init__(self,
KeplerInterface(**options),
**options)
def __init__(self, **options):
CommonCode.__init__(self, DistributedAmuseInterface(**options), **options)
def __init__(self, **options):
CommonCode.__init__(self, DistributedAmuseInterface(**options), **options)
def __init__(self, unit_converter = None, **options):
self.unit_converter = unit_converter
legacy_interface = BarAndSpiralsInterface(**options)
self.interface_doc = legacy_interface.__doc__
CommonCode.__init__(self, legacy_interface, **options)
def __init__(self, unit_converter = None, **options):
self.unit_converter = unit_converter
self.stopping_conditions = StoppingConditions(self)
CommonCode.__init__(self, CapreoleInterface(**options), **options)
