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, unit_converter = None, **options): self.unit_converter = unit_converter self.stopping_conditions = StoppingConditions(self) CommonCode.__init__(self, CapreoleInterface(**options), **options)