def __init__(self):
BurstBaseController.__init__(self)
self.range_parameters = SimulatorRangeParameters()
self.burst_service = BurstService()
self.simulator_service = SimulatorService()
self.cached_simulator_algorithm = self.algorithm_service.get_algorithm_by_module_and_class(
IntrospectionRegistry.SIMULATOR_MODULE, IntrospectionRegistry.SIMULATOR_CLASS)
self.context = SimulatorContext()
self.monitors_handler = MonitorsWizardHandler()
def _mark_selected(self, project):
"""
Set the project passed as parameter as the selected project.
"""
previous_project = common.get_current_project()
# Update project stored in selection, with latest Project entity from DB.
members = self.user_service.get_users_for_project("", project.id)[1]
project.members = members
if previous_project is None or previous_project.id != project.id:
# Clean Burst selection from session in case of a different project.
SimulatorContext().clean_project_data_from_session()
# Store in DB new project selection
user = common.get_from_session(common.KEY_USER)
if user is not None:
self.user_service.save_project_to_user(user.id, project.id)
# Display info message about project change
self.logger.debug("Selected project is now " + project.name)
common.set_info_message("Your current working project is: " +
str(project.name))
linked_dt = self.project_service.get_linked_datatypes_storage_path(
project)
storage_interface = StorageInterface()
storage_interface.set_project_active(project, linked_dt)
if previous_project is not None:
storage_interface.set_project_inactive(previous_project)
# Add the project entity to session every time, as it might be changed (e.g. after edit)
common.add2session(common.KEY_PROJECT, project)
def _remove_project(self, project_id):
"""Private method for removing project."""
try:
self.project_service.remove_project(project_id)
except ServicesBaseException as exc:
self.logger.error("Could not delete project!")
self.logger.exception(exc)
common.set_error_message(exc.message)
prj = common.get_current_project()
if prj is not None and prj.id == int(project_id):
SimulatorContext().clean_project_data_from_session()
def logout(self):
"""
Logging out user and clean session
"""
user = common.remove_from_session(common.KEY_USER)
if user is not None:
self.logger.debug("User " + user.username + " is just logging out!")
current_project = common.get_current_project()
if current_project is not None and encryption_handler.encryption_enabled():
encryption_handler.set_project_inactive(current_project)
SimulatorContext().clean_project_data_from_session()
common.set_info_message("Thank you for using The Virtual Brain!")
common.expire_session()
raise cherrypy.HTTPRedirect("/user")
class SimulatorController(BurstBaseController):
KEY_IS_LOAD_AFTER_REDIRECT = "is_load_after_redirect"
COPY_NAME_FORMAT = "copy_of_{}"
BRANCH_NAME_FORMAT = "{}_branch{}"
KEY_KEEP_SAME_SIM_WIZARD = "keep_same_wizard"
def __init__(self):
BurstBaseController.__init__(self)
self.range_parameters = SimulatorRangeParameters()
self.burst_service = BurstService()
self.simulator_service = SimulatorService()
self.cached_simulator_algorithm = self.algorithm_service.get_algorithm_by_module_and_class(
IntrospectionRegistry.SIMULATOR_MODULE,
IntrospectionRegistry.SIMULATOR_CLASS)
self.context = SimulatorContext()
self.monitors_handler = MonitorsWizardHandler()
@expose_json
def cancel_or_remove_burst(self, burst_id):
"""
Cancel or Remove the burst entity given by burst_id (and all linked entities: op, DTs)
:returns True: if the op was successfully.
"""
burst_config = BurstService.load_burst_configuration(int(burst_id))
op_id, is_group = burst_config.operation_info_for_burst_removal
return self.cancel_or_remove_operation(op_id, is_group,
burst_config.is_finished)
def cancel_or_remove_operation(self,
operation_id,
is_group,
remove_after_stop=False):
"""
Stop the operation given by operation_id. If is_group is true stop all the
operations from that group.
"""
# Load before we remove, to have its data in memory here
burst_config = BurstService.get_burst_for_operation_id(
operation_id, is_group)
if burst_config is not None:
self.burst_service.mark_burst_finished(
burst_config,
BurstConfiguration.BURST_CANCELED,
store_h5_file=False)
while GROUP_BURST_PENDING.get(burst_config.id, False):
pass
GROUP_BURST_PENDING.pop(burst_config.id, False)
result = OperationService.stop_operation(operation_id, is_group,
remove_after_stop)
if remove_after_stop:
current_burst = self.context.burst_config
if (current_burst is not None and burst_config is not None
and current_burst.id == burst_config.id and
((current_burst.fk_simulation == operation_id and not is_group)
or (current_burst.fk_operation_group == operation_id
and is_group))):
self.reset_simulator_configuration()
if burst_config is not None:
burst_config = BurstService.load_burst_configuration(
burst_config.id)
if burst_config:
BurstService.remove_burst_configuration(burst_config.id)
return result
@expose_page
@settings
@context_selected
def index(self):
"""Get on burst main page"""
template_specification = dict(
mainContent="burst/main_burst",
title="Simulation Cockpit",
includedResources='project/included_resources')
if not self.context.last_loaded_fragment_url:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_CONNECTIVITY_URL)
self.context.set_burst_config()
_, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if self.context.burst_config.start_time is not None:
is_simulation_load = True
self.context.add_simulator_load_to_session(True)
template_specification['burstConfig'] = self.context.burst_config
template_specification[
'burst_list'] = self.burst_service.get_available_bursts(
self.context.project.id)
form = self.prepare_first_fragment()
rendering_rules = SimulatorFragmentRenderingRules(
form,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL,
None,
is_simulation_copy,
is_simulation_load,
last_form_url=self.context.last_loaded_fragment_url,
last_request_type=cherrypy.request.method,
is_first_fragment=True,
is_branch=is_branch)
template_specification.update(**rendering_rules.to_dict())
cherrypy.response.headers[
'Cache-Control'] = 'no-cache, no-store, must-revalidate'
cherrypy.response.headers['Pragma'] = 'no-cache'
cherrypy.response.headers['Expires'] = '0'
return self.fill_default_attributes(template_specification)
def prepare_first_fragment(self):
self.context.set_simulator()
simulator, _, _, is_branch = self.context.get_common_params()
branch_conditions = self.simulator_service.compute_conn_branch_conditions(
is_branch, simulator)
form = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorAdapterForm(),
project_id=self.context.project.id,
extra_conditions=branch_conditions)
self.simulator_service.validate_first_fragment(form,
self.context.project.id,
ConnectivityIndex)
form.fill_from_trait(self.context.simulator)
return form
@expose_json
def set_fragment_url(self, **data):
try:
self.context.add_last_loaded_form_url_to_session(data['url'])
except KeyError:
self.logger.error(
"Cannot set last loaded url to session because the required data was not found."
)
@expose_fragment('simulator_fragment')
def set_connectivity(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, _ = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_COUPLING_PARAMS_URL)
form = SimulatorAdapterForm()
form.fill_from_post(data)
self.simulator_service.reset_at_connectivity_change(
is_simulation_copy, form, session_stored_simulator)
form.fill_trait(session_stored_simulator)
next_form = self.algorithm_service.prepare_adapter_form(
form_instance=get_form_for_coupling(
type(session_stored_simulator.coupling))())
self.range_parameters.coupling_parameters = next_form.get_range_parameters(
)
next_form.fill_from_trait(session_stored_simulator.coupling)
rendering_rules = SimulatorFragmentRenderingRules(
next_form, SimulatorWizzardURLs.SET_COUPLING_PARAMS_URL,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL, is_simulation_copy,
is_simulation_load, self.context.last_loaded_fragment_url,
cherrypy.request.method)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_coupling_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_SURFACE_URL)
form = get_form_for_coupling(
type(session_stored_simulator.coupling))()
form.fill_from_post(data)
form.fill_trait(session_stored_simulator.coupling)
surface_fragment = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorSurfaceFragment(),
project_id=self.context.project.id)
surface_fragment.fill_from_trait(session_stored_simulator.surface)
rendering_rules = SimulatorFragmentRenderingRules(
surface_fragment,
SimulatorWizzardURLs.SET_SURFACE_URL,
SimulatorWizzardURLs.SET_COUPLING_PARAMS_URL,
is_simulation_copy,
is_simulation_load,
self.context.last_loaded_fragment_url,
cherrypy.request.method,
is_branch=is_branch)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_surface(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
rendering_rules = SimulatorFragmentRenderingRules(
previous_form_action_url=SimulatorWizzardURLs.SET_SURFACE_URL,
is_simulation_copy=is_simulation_copy,
is_simulation_readonly_load=is_simulation_load,
last_form_url=self.context.last_loaded_fragment_url,
last_request_type=cherrypy.request.method,
is_branch=is_branch)
if cherrypy.request.method == POST_REQUEST:
form = SimulatorSurfaceFragment()
form.fill_from_post(data)
self.simulator_service.reset_at_surface_change(
is_simulation_copy, form, session_stored_simulator)
form.fill_trait(session_stored_simulator)
if session_stored_simulator.surface:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_CORTEX_URL)
else:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_STIMULUS_URL)
return SimulatorSurfaceFragment.prepare_next_fragment_after_surface(
session_stored_simulator, rendering_rules, self.context.project.id,
SimulatorWizzardURLs.SET_CORTEX_URL,
SimulatorWizzardURLs.SET_STIMULUS_URL)
@expose_fragment('simulator_fragment')
def set_cortex(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, _ = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_STIMULUS_URL)
rm_fragment = SimulatorRMFragment()
rm_fragment.fill_from_post(data)
rm_fragment.fill_trait(session_stored_simulator.surface)
rendering_rules = SimulatorFragmentRenderingRules(
None, None, SimulatorWizzardURLs.SET_CORTEX_URL,
is_simulation_copy, is_simulation_load,
self.context.last_loaded_fragment_url, cherrypy.request.method)
return SimulatorStimulusFragment.prepare_stimulus_fragment(
session_stored_simulator, rendering_rules, True,
SimulatorWizzardURLs.SET_STIMULUS_URL, self.context.project.id)
@expose_fragment('simulator_fragment')
def set_stimulus(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_MODEL_URL)
stimuli_fragment = SimulatorStimulusFragment(
session_stored_simulator.is_surface_simulation)
stimuli_fragment.fill_from_post(data)
stimuli_fragment.fill_trait(session_stored_simulator)
model_fragment = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorModelFragment())
model_fragment.fill_from_trait(session_stored_simulator)
rendering_rules = SimulatorFragmentRenderingRules(
model_fragment,
SimulatorWizzardURLs.SET_MODEL_URL,
SimulatorWizzardURLs.SET_STIMULUS_URL,
is_simulation_copy,
is_simulation_load,
self.context.last_loaded_fragment_url,
cherrypy.request.method,
is_model_fragment=True,
is_surface_simulation=session_stored_simulator.
is_surface_simulation,
is_branch=is_branch)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_model(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
set_next_wizard = True
if SimulatorController.KEY_KEEP_SAME_SIM_WIZARD in data:
set_next_wizard = False
if set_next_wizard:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_MODEL_PARAMS_URL)
form = SimulatorModelFragment()
form.fill_from_post(data)
form.fill_trait(session_stored_simulator)
form = self.algorithm_service.prepare_adapter_form(
form_instance=get_form_for_model(
type(session_stored_simulator.model))(is_branch))
self.range_parameters.model_parameters = form.get_range_parameters()
form.fill_from_trait(session_stored_simulator.model)
rendering_rules = SimulatorFragmentRenderingRules(
form, SimulatorWizzardURLs.SET_MODEL_PARAMS_URL,
SimulatorWizzardURLs.SET_MODEL_URL, is_simulation_copy,
is_simulation_load, self.context.last_loaded_fragment_url,
cherrypy.request.method)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_model_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_INTEGRATOR_URL)
form = get_form_for_model(type(
session_stored_simulator.model))(is_branch)
if is_branch:
data['variables_of_interest'] = list(
session_stored_simulator.model.variables_of_interest)
form.fill_from_post(data)
form.fill_trait(session_stored_simulator.model)
integrator_fragment = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorIntegratorFragment())
integrator_fragment.integrator.display_subform = False
integrator_fragment.fill_from_trait(session_stored_simulator)
rendering_rules = SimulatorFragmentRenderingRules(
integrator_fragment,
SimulatorWizzardURLs.SET_INTEGRATOR_URL,
SimulatorWizzardURLs.SET_MODEL_PARAMS_URL,
is_simulation_copy,
is_simulation_load,
self.context.last_loaded_fragment_url,
cherrypy.request.method,
is_branch=is_branch)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_integrator(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_INTEGRATOR_PARAMS_URL)
fragment = SimulatorIntegratorFragment()
fragment.fill_from_post(data)
fragment.fill_trait(session_stored_simulator)
form = self.algorithm_service.prepare_adapter_form(
form_instance=get_form_for_integrator(
type(session_stored_simulator.integrator))(is_branch))
if hasattr(form, 'noise'):
form.noise.display_subform = False
form.fill_from_trait(session_stored_simulator.integrator)
rendering_rules = SimulatorFragmentRenderingRules(
form, SimulatorWizzardURLs.SET_INTEGRATOR_PARAMS_URL,
SimulatorWizzardURLs.SET_INTEGRATOR_URL, is_simulation_copy,
is_simulation_load, self.context.last_loaded_fragment_url,
cherrypy.request.method)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_integrator_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
form = get_form_for_integrator(
type(session_stored_simulator.integrator))(is_branch)
if is_branch:
data['dt'] = str(session_stored_simulator.integrator.dt)
form.fill_from_post(data)
form.fill_trait(session_stored_simulator.integrator)
if isinstance(session_stored_simulator.integrator,
IntegratorStochasticViewModel):
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_NOISE_PARAMS_URL)
else:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_MONITORS_URL)
rendering_rules = SimulatorFragmentRenderingRules(
None,
None,
SimulatorWizzardURLs.SET_INTEGRATOR_PARAMS_URL,
is_simulation_copy,
is_simulation_load,
self.context.last_loaded_fragment_url,
cherrypy.request.method,
is_noise_fragment=False)
if not isinstance(session_stored_simulator.integrator,
IntegratorStochasticViewModel):
return self.monitors_handler.prepare_monitor_fragment(
session_stored_simulator, rendering_rules,
SimulatorWizzardURLs.SET_MONITORS_URL)
integrator_noise_fragment = get_form_for_noise(
type(session_stored_simulator.integrator.noise))()
if hasattr(integrator_noise_fragment, 'equation'):
integrator_noise_fragment.equation.display_subform = False
self.range_parameters.integrator_noise_parameters = integrator_noise_fragment.get_range_parameters(
)
integrator_noise_fragment.fill_from_trait(
session_stored_simulator.integrator.noise)
rendering_rules.form = integrator_noise_fragment
rendering_rules.form_action_url = SimulatorWizzardURLs.SET_NOISE_PARAMS_URL
rendering_rules.is_noise_fragment = True
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_noise_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
form = get_form_for_noise(
type(session_stored_simulator.integrator.noise))()
form.fill_from_post(data)
form.fill_trait(session_stored_simulator.integrator.noise)
if isinstance(session_stored_simulator.integrator.noise,
AdditiveNoiseViewModel):
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_MONITORS_URL)
else:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_NOISE_EQUATION_PARAMS_URL)
rendering_rules = SimulatorFragmentRenderingRules(
None, None, SimulatorWizzardURLs.SET_NOISE_PARAMS_URL,
is_simulation_copy, is_simulation_load,
self.context.last_loaded_fragment_url, cherrypy.request.method)
return self.monitors_handler.prepare_next_fragment_after_noise(
session_stored_simulator, is_branch, rendering_rules,
SimulatorWizzardURLs.SET_MONITORS_URL,
SimulatorWizzardURLs.SET_NOISE_EQUATION_PARAMS_URL)
@expose_fragment('simulator_fragment')
def set_noise_equation_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, _ = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_MONITORS_URL)
form = get_form_for_equation(
type(session_stored_simulator.integrator.noise.b))()
form.fill_from_post(data)
form.fill_trait(session_stored_simulator.integrator.noise.b)
rendering_rules = SimulatorFragmentRenderingRules(
None, None, SimulatorWizzardURLs.SET_NOISE_EQUATION_PARAMS_URL,
is_simulation_copy, is_simulation_load,
self.context.last_loaded_fragment_url, cherrypy.request.method)
return self.monitors_handler.prepare_monitor_fragment(
session_stored_simulator, rendering_rules,
SimulatorWizzardURLs.SET_MONITORS_URL)
@staticmethod
def build_monitor_url(fragment_url, monitor):
url_regex = '{}/{}'
url = url_regex.format(fragment_url, monitor)
return url
def get_first_monitor_fragment_url(self, simulator, monitors_url):
first_monitor = simulator.first_monitor
if first_monitor is not None:
monitor_vm_name = type(first_monitor).__name__
return self.build_monitor_url(monitors_url, monitor_vm_name)
return SimulatorWizzardURLs.SETUP_PSE_URL
@expose_fragment('simulator_fragment')
def set_monitors(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, is_branch = self.context.get_common_params(
)
if cherrypy.request.method == POST_REQUEST:
fragment = SimulatorMonitorFragment(
is_surface_simulation=session_stored_simulator.
is_surface_simulation)
fragment.fill_from_post(data)
self.monitors_handler.set_monitors_list_on_simulator(
session_stored_simulator, fragment.monitors.value)
last_loaded_fragment_url = self.get_first_monitor_fragment_url(
session_stored_simulator,
SimulatorWizzardURLs.SET_MONITOR_PARAMS_URL)
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
last_loaded_fragment_url)
rendering_rules = SimulatorFragmentRenderingRules(
is_simulation_copy=is_simulation_copy,
is_simulation_readonly_load=is_simulation_load,
last_request_type=cherrypy.request.method,
form_action_url=last_loaded_fragment_url,
previous_form_action_url=SimulatorWizzardURLs.SET_MONITORS_URL)
return self.monitors_handler.get_fragment_after_monitors(
session_stored_simulator, self.context.burst_config,
self.context.project.id, is_branch, rendering_rules,
SimulatorWizzardURLs.SETUP_PSE_URL)
def get_url_after_monitors(self, current_monitor, monitor_name,
next_monitor):
if isinstance(current_monitor, BoldViewModel):
return self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_EQUATION_URL, monitor_name)
if next_monitor is not None:
return self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_PARAMS_URL,
type(next_monitor).__name__)
return SimulatorWizzardURLs.SETUP_PSE_URL
@staticmethod
def get_url_for_final_fragment(burst_config):
if burst_config.is_pse_burst():
return SimulatorWizzardURLs.LAUNCH_PSE_URL
return SimulatorWizzardURLs.SETUP_PSE_URL
def get_urls_for_next_monitor_fragment(self, next_monitor,
current_monitor):
form_action_url = self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_PARAMS_URL,
type(next_monitor).__name__)
if_bold_url = self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_EQUATION_URL,
type(current_monitor).__name__)
return form_action_url, if_bold_url
@expose_fragment('simulator_fragment')
def set_monitor_params(self, current_monitor_name, **data):
session_stored_simulator, is_simulation_copy, is_simulator_load, is_branch = self.context.get_common_params(
)
current_monitor, next_monitor = self.monitors_handler.get_current_and_next_monitor_form(
current_monitor_name, session_stored_simulator)
if cherrypy.request.method == POST_REQUEST:
form = get_form_for_monitor(type(current_monitor))(
session_stored_simulator, is_branch)
if is_branch:
data['period'] = str(current_monitor.period)
data['variables_of_interest'] = [
session_stored_simulator.model.variables_of_interest[i]
for i in current_monitor.variables_of_interest
]
form.fill_from_post(data)
form.fill_trait(current_monitor)
last_loaded_form_url = self.get_url_after_monitors(
current_monitor, current_monitor_name, next_monitor)
self.context.add_last_loaded_form_url_to_session(
last_loaded_form_url)
previous_form_action_url = self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_PARAMS_URL, current_monitor_name)
rendering_rules = SimulatorFragmentRenderingRules(
is_simulation_copy=is_simulation_copy,
is_simulation_readonly_load=is_simulator_load,
last_request_type=cherrypy.request.method,
last_form_url=self.context.last_loaded_fragment_url,
previous_form_action_url=previous_form_action_url)
form_action_url, if_bold_url = self.get_urls_for_next_monitor_fragment(
next_monitor, current_monitor)
self.monitors_handler.update_monitor(current_monitor)
return self.monitors_handler.handle_next_fragment_for_monitors(
self.context, rendering_rules, current_monitor, next_monitor,
False, form_action_url, if_bold_url)
def get_url_after_monitor_equation(self, next_monitor):
if next_monitor is None:
return SimulatorWizzardURLs.SETUP_PSE_URL
last_loaded_fragment_url = self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_PARAMS_URL,
type(next_monitor).__name__)
return last_loaded_fragment_url
@expose_fragment('simulator_fragment')
def set_monitor_equation(self, current_monitor_name, **data):
session_stored_simulator, is_simulation_copy, is_simulator_load, is_branch = self.context.get_common_params(
)
current_monitor, next_monitor = self.monitors_handler.get_current_and_next_monitor_form(
current_monitor_name, session_stored_simulator)
if cherrypy.request.method == POST_REQUEST:
form = get_form_for_equation(type(current_monitor.hrf_kernel))()
form.fill_from_post(data)
form.fill_trait(current_monitor.hrf_kernel)
last_loaded_fragment_url = self.get_url_after_monitor_equation(
next_monitor)
self.context.add_last_loaded_form_url_to_session(
last_loaded_fragment_url)
previous_form_action_url = self.build_monitor_url(
SimulatorWizzardURLs.SET_MONITOR_EQUATION_URL,
current_monitor_name)
rendering_rules = SimulatorFragmentRenderingRules(
None, None, previous_form_action_url, is_simulation_copy,
is_simulator_load, self.context.last_loaded_fragment_url,
cherrypy.request)
form_action_url, if_bold_url = self.get_urls_for_next_monitor_fragment(
next_monitor, current_monitor)
return self.monitors_handler.handle_next_fragment_for_monitors(
self.context, rendering_rules, current_monitor, next_monitor, True,
form_action_url, if_bold_url)
@expose_fragment('simulator_fragment')
def setup_pse(self, **data):
session_stored_simulator, is_simulation_copy, is_simulator_load, _ = self.context.get_common_params(
)
burst_config = self.context.burst_config
all_range_parameters = self.range_parameters.get_all_range_parameters()
next_form = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorPSEConfigurationFragment(
self.range_parameters.get_all_range_parameters()))
if cherrypy.request.method == POST_REQUEST:
session_stored_simulator.simulation_length = float(
data['simulation_length'])
burst_config.name = data['input_simulation_name_id']
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_PSE_PARAMS_URL)
param1, param2 = self.burst_service.handle_range_params_at_loading(
burst_config, all_range_parameters)
if param1:
param_dict = {'pse_param1': param1.name}
if param2 is not None:
param_dict['pse_param2'] = param2.name
next_form.fill_from_post(param_dict)
rendering_rules = SimulatorFragmentRenderingRules(
next_form, SimulatorWizzardURLs.SET_PSE_PARAMS_URL,
SimulatorWizzardURLs.SETUP_PSE_URL, is_simulation_copy,
is_simulator_load, self.context.last_loaded_fragment_url,
cherrypy.request.method)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def set_pse_params(self, **data):
session_stored_simulator, is_simulation_copy, is_simulation_load, _ = self.context.get_common_params(
)
burst_config = self.context.burst_config
form = SimulatorPSEConfigurationFragment(
self.range_parameters.get_all_range_parameters())
if cherrypy.request.method == POST_REQUEST:
self.context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.LAUNCH_PSE_URL)
form.fill_from_post(data)
param1 = form.pse_param1.value
burst_config.range1 = param1.to_json()
param2 = None
if form.pse_param2.value:
param2 = form.pse_param2.value
burst_config.range2 = param2.to_json()
else:
all_range_parameters = self.range_parameters.get_all_range_parameters(
)
param1, param2 = self.burst_service.handle_range_params_at_loading(
burst_config, all_range_parameters)
next_form = self.algorithm_service.prepare_adapter_form(
form_instance=SimulatorPSERangeFragment(param1, param2))
rendering_rules = SimulatorFragmentRenderingRules(
next_form,
SimulatorWizzardURLs.LAUNCH_PSE_URL,
SimulatorWizzardURLs.SET_PSE_PARAMS_URL,
is_simulation_copy,
is_simulation_load,
last_form_url=self.context.last_loaded_fragment_url,
is_launch_pse_fragment=True)
return rendering_rules.to_dict()
@expose_json
def launch_pse(self, **data):
session_stored_simulator = self.context.simulator
all_range_parameters = self.range_parameters.get_all_range_parameters()
range_param1, range_param2 = SimulatorPSERangeFragment.fill_from_post(
all_range_parameters, **data)
burst_config = self.context.burst_config
burst_config.start_time = datetime.now()
burst_config.range1 = range_param1.to_json()
if range_param2:
burst_config.range2 = range_param2.to_json()
burst_config = self.burst_service.prepare_burst_for_pse(burst_config)
session_stored_simulator.operation_group_gid = uuid.UUID(
burst_config.operation_group.gid)
session_stored_simulator.ranges = json.dumps(burst_config.ranges)
try:
thread = threading.Thread(
target=self.simulator_service.async_launch_and_prepare_pse,
kwargs={
'burst_config': burst_config,
'user': self.context.logged_user,
'project': self.context.project,
'simulator_algo': self.cached_simulator_algorithm,
'range_param1': range_param1,
'range_param2': range_param2,
'session_stored_simulator': session_stored_simulator
})
thread.start()
return {'id': burst_config.id}
except BurstServiceException as e:
self.logger.exception("Could not launch burst!")
return {'error': e.message}
@expose_json
def launch_simulation(self, launch_mode, **data):
current_form = SimulatorFinalFragment()
burst_config = self.context.burst_config
burst_config.range1 = None
burst_config.range2 = None
try:
current_form.fill_from_post(data)
except Exception as exc:
self.logger.exception(exc)
return {'error': str(exc)}
burst_name = current_form.simulation_name.value
session_stored_simulator = self.context.simulator
session_stored_simulator.simulation_length = current_form.simulation_length.value
if burst_name != 'none_undefined':
burst_config.name = burst_name
if launch_mode == self.burst_service.LAUNCH_BRANCH:
simulation_state_index = self.simulator_service.get_simulation_state_index(
burst_config, SimulationHistoryIndex)
session_stored_simulator.history_gid = simulation_state_index[
0].gid
burst_config.start_time = datetime.now()
session_burst_config = self.burst_service.store_burst(burst_config)
try:
thread = threading.Thread(target=self.simulator_service.
async_launch_and_prepare_simulation,
kwargs={
'burst_config': session_burst_config,
'user': self.context.logged_user,
'project': self.context.project,
'simulator_algo':
self.cached_simulator_algorithm,
'simulator': session_stored_simulator
})
thread.start()
return {'id': session_burst_config.id}
except BurstServiceException as e:
self.logger.exception('Could not launch burst!')
return {'error': e.message}
@expose_fragment('burst/burst_history')
def load_burst_history(self, initBurst=None):
"""
Load the available burst that are stored in the database at this time.
This is one alternative to 'chrome-back problem'.
"""
bursts = self.burst_service.get_available_bursts(
self.context.project.id)
self.burst_service.populate_burst_disk_usage(bursts)
fromInit = False
if initBurst is not None:
fromInit = True
return {
'burst_list': bursts,
'fromInit': fromInit,
'selectedBurst': self.context.burst_config.id,
'first_fragment_url':
SimulatorFragmentRenderingRules.FIRST_FORM_URL
}
@cherrypy.expose
def get_last_fragment_url(self, burst_config_id):
burst_config = self.burst_service.load_burst_configuration(
burst_config_id)
return self.get_url_for_final_fragment(burst_config)
@expose_fragment('simulator_fragment')
def load_burst_read_only(self, burst_config_id):
try:
burst_config = self.burst_service.load_burst_configuration(
burst_config_id)
storage_path = StorageInterface().get_project_folder(
self.context.project.name, str(burst_config.fk_simulation))
simulator = h5.load_view_model(burst_config.simulator_gid,
storage_path)
last_loaded_form_url = self.get_url_for_final_fragment(
burst_config)
self.context.init_session_at_burst_loading(burst_config, simulator,
last_loaded_form_url)
form = self.prepare_first_fragment()
self.monitors_handler.build_list_of_monitors_from_view_models(
self.context.simulator)
rendering_rules = SimulatorFragmentRenderingRules(
form,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL,
is_simulation_readonly_load=True,
is_first_fragment=True)
return rendering_rules.to_dict()
except Exception:
# Most probably Burst was removed. Delete it from session, so that client
# has a good chance to get a good response on refresh
self.logger.exception("Error loading burst")
self.context.remove_burst_config_from_session()
raise
def _prepare_first_fragment_for_burst_copy(self, burst_config_id,
burst_name_format):
simulator, burst_config_copy = self.burst_service.prepare_data_for_burst_copy(
burst_config_id, burst_name_format, self.context.project)
self.monitors_handler.build_list_of_monitors_from_view_models(
simulator)
last_loaded_form_url = self.get_url_for_final_fragment(
burst_config_copy)
self.context.init_session_at_copy_preparation(burst_config_copy,
simulator,
last_loaded_form_url)
return self.prepare_first_fragment()
@expose_fragment('simulator_fragment')
def copy_simulator_configuration(self, burst_config_id):
self.context.add_branch_and_copy_to_session(False, True)
form = self._prepare_first_fragment_for_burst_copy(
burst_config_id, self.COPY_NAME_FORMAT)
rendering_rules = SimulatorFragmentRenderingRules(
form,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL,
is_simulation_copy=True,
is_simulation_readonly_load=True,
is_first_fragment=True)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def branch_simulator_configuration(self, burst_config_id):
self.context.add_branch_and_copy_to_session(True, False)
form = self._prepare_first_fragment_for_burst_copy(
burst_config_id, self.BRANCH_NAME_FORMAT)
rendering_rules = SimulatorFragmentRenderingRules(
form,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL,
is_simulation_copy=True,
is_simulation_readonly_load=True,
is_first_fragment=True)
return rendering_rules.to_dict()
@expose_fragment('simulator_fragment')
def reset_simulator_configuration(self):
burst_config = BurstConfiguration(self.context.project.id)
self.context.init_session_at_sim_reset(
burst_config, SimulatorWizzardURLs.SET_CONNECTIVITY_URL)
self.monitors_handler.clear_next_monitors_dict()
form = self.prepare_first_fragment()
rendering_rules = SimulatorFragmentRenderingRules(
form,
SimulatorWizzardURLs.SET_CONNECTIVITY_URL,
is_first_fragment=True)
return rendering_rules.to_dict()
@expose_json
def rename_burst(self, burst_id, burst_name):
"""
Rename the burst given by burst_id, setting it's new name to
burst_name.
"""
validation_result = SimulatorFinalFragment.is_burst_name_ok(burst_name)
if validation_result is True:
self.burst_service.rename_burst(burst_id, burst_name)
return {'success': "Simulation successfully renamed!"}
else:
self.logger.exception(validation_result)
return {'error': validation_result}
@expose_json
def get_history_status(self, **data):
"""
For each burst id received, get the status and return it.
"""
return self.burst_service.update_history_status(
json.loads(data['burst_ids']))
@cherrypy.expose
@handle_error(redirect=False)
@check_user
def export(self, burst_id):
export_manager = ExportManager()
export_zip = export_manager.export_simulator_configuration(burst_id)
result_name = "tvb_simulation_" + str(burst_id) + ".zip"
return serve_file(export_zip, "application/x-download", "attachment",
result_name)
@expose_fragment("overlay")
def get_upload_overlay(self):
template_specification = self.fill_overlay_attributes(
None, "Upload", "Simulation ZIP", "burst/upload_burst_overlay",
"dialog-upload")
template_specification[
'first_fragment_url'] = SimulatorWizzardURLs.SET_CONNECTIVITY_URL
return self.fill_default_attributes(template_specification)
@cherrypy.expose
@handle_error(redirect=True)
@check_user
@settings
def load_simulator_configuration_from_zip(self, **data):
"""Upload Simulator from previously exported ZIP file"""
self.logger.debug("Uploading ..." + str(data))
last_loaded_form_url = SimulatorWizzardURLs.SETUP_PSE_URL
try:
upload_param = "uploadedfile"
if upload_param in data and data[upload_param]:
simulator, burst_config, sim_folder = self.burst_service.load_simulation_from_zip(
data[upload_param], self.context.project)
dts_folder = os.path.join(
sim_folder, StorageInterface.EXPORTED_SIMULATION_DTS_DIR)
ImportService().import_project_operations(
self.context.project, dts_folder, False, None)
self.monitors_handler.build_list_of_monitors_from_view_models(
simulator)
if burst_config.is_pse_burst():
last_loaded_form_url = SimulatorWizzardURLs.LAUNCH_PSE_URL
self.context.init_session_at_sim_config_from_zip(
burst_config, simulator, last_loaded_form_url)
except IOError as ioexcep:
self.logger.exception(ioexcep)
self.context.set_warning_message(
"This ZIP does not contain a complete simulator configuration")
except ServicesBaseException as excep:
self.logger.warning(excep.message)
self.context.set_warning_message(excep.message)
raise cherrypy.HTTPRedirect('/burst/')
def __init__(self):
super(NoiseConfigurationController, self).__init__()
self.simulator_context = SimulatorContext()
class NoiseConfigurationController(BurstBaseController):
"""
Controller class for placing noise parameters in nodes.
"""
def __init__(self):
super(NoiseConfigurationController, self).__init__()
self.simulator_context = SimulatorContext()
@expose_page
def index(self):
des = SerializationManager(self.simulator_context.simulator)
conn_idx = load.load_entity_by_gid(des.conf.connectivity)
model = des.conf.model
integrator = des.conf.integrator
state_vars = model.state_variables
noise_values = self.init_noise_config_values(model, integrator,
conn_idx)
initial_noise = self.group_noise_array_by_state_var(
noise_values, state_vars, conn_idx.number_of_regions)
current_project = common.get_current_project()
file_handler = FilesHelper()
conn_path = file_handler.get_project_folder(
current_project, str(conn_idx.fk_from_operation))
params = ConnectivityViewer.get_connectivity_parameters(
conn_idx, conn_path)
params.update({
'title': 'Noise configuration',
'mainContent': 'burst/noise',
'isSingleMode': True,
'submit_parameters_url': '/burst/noise/submit',
'stateVars': state_vars,
'stateVarsJson': json.dumps(state_vars),
'noiseInputValues': initial_noise[0],
'initialNoiseValues': json.dumps(initial_noise)
})
return self.fill_default_attributes(params, 'regionmodel')
@cherrypy.expose
@handle_error(redirect=True)
@check_user
def submit(self, node_values):
"""
Submit noise dispersions
:param node_values: A map from state variable names to noise dispersion arrays. Ex {'V': [1,2...74]}
"""
des = SerializationManager(self.simulator_context.simulator)
des.write_noise_parameters(json.loads(node_values))
self.simulator_context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_NOISE_PARAMS_URL)
raise cherrypy.HTTPRedirect("/burst/")
@staticmethod
def group_noise_array_by_state_var(noise_values, state_vars,
number_of_regions):
initial_noise = []
for i in range(number_of_regions):
node_noise = {}
for sv_idx, sv in enumerate(state_vars):
node_noise[sv] = noise_values[sv_idx][i]
initial_noise.append(node_noise)
return initial_noise
@staticmethod
def init_noise_config_values(model, integrator, connectivity):
"""
Initialize a state var x number of nodes array with noise values.
"""
state_variables = model.state_variables
nr_nodes = connectivity.number_of_regions
nr_state_vars = len(state_variables)
try:
nsig = integrator.noise.nsig
noise_values = nsig.tolist()
except AttributeError:
# Just fallback to default
return [[1 for _ in range(nr_nodes)] for _ in state_variables]
if nsig.shape == (1, ):
# Only one number for noise
return [noise_values * nr_nodes for _ in state_variables]
elif nsig.shape == (nr_state_vars,
1) or nsig.shape == (nr_state_vars, ):
# Only one number per state variable
return [[noise_values[idx]] * nr_nodes
for idx in range(nr_state_vars)]
elif nsig.shape == (nr_state_vars, nr_nodes):
return noise_values
else:
raise ValueError("Got unexpected noise shape %s." % (nsig.shape, ))
def __init__(self):
super(SurfaceModelParametersController, self).__init__()
self.simulator_context = SimulatorContext()
class SurfaceModelParametersController(SpatioTemporalController):
"""
Control for defining parameters of a model in a visual manner.
Here we focus on model-parameters spread over a brain surface.
"""
MODEL_PARAM_FIELD = 'set_model_parameter'
EQUATION_FIELD = 'set_equation'
EQUATION_PARAMS_FIELD = 'set_equation_param'
base_url = '/spatial/modelparameters/surface'
def __init__(self):
super(SurfaceModelParametersController, self).__init__()
self.simulator_context = SimulatorContext()
def get_data_from_burst_configuration(self):
"""
Returns the model and surface instances from the burst configuration.
"""
des = SerializationManager(self.simulator_context.simulator)
### Read from session current burst-configuration
if des.conf is None:
return None, None
# if des.has_model_pse_ranges():
# common.set_error_message("When configuring model parameters you are not allowed to specify range values.")
# raise cherrypy.HTTPRedirect("/burst/")
try:
model = des.conf.model
except Exception:
self.logger.exception(
"Some of the provided parameters have an invalid value.")
common.set_error_message(
"Some of the provided parameters have an invalid value.")
raise cherrypy.HTTPRedirect("/burst/")
cortex = des.conf.surface
return model, cortex
def _prepare_model_params_dict(self, model):
model_form = get_model_to_form_dict().get(type(model))
model_params = model_form.get_params_configurable_in_phase_plane()
if len(model_params) == 0:
self.logger.warning(
"The list with configurable parameters for the current model is empty!"
)
model_params_dict = {}
for param in model_params:
model_params_dict.update({param: param})
return model_params_dict
def _fill_form_from_context(self, config_form, context):
if context.current_model_param in context.applied_equations:
current_equation = context.get_equation_for_parameter(
context.current_model_param)
context.current_equation = current_equation
config_form.equation.data = type(current_equation)
config_form.equation.subform_field.form = get_form_for_equation(
type(current_equation))()
config_form.equation.subform_field.form.fill_from_trait(
current_equation)
else:
context.current_equation = SurfaceModelParametersForm.default_equation(
)
config_form.equation.data = type(context.current_equation)
config_form.equation.subform_field.form.fill_from_trait(
context.current_equation)
def _prepare_reload(self, context):
template_specification = {
'baseUrl': self.base_url,
'equationsPrefixes': self.plotted_equation_prefixes
}
template_specification.update(
{'applied_equations': context.get_configure_info()})
config_form = SurfaceModelParametersForm(self.model_params_dict)
config_form.model_param.data = context.current_model_param
self._fill_form_from_context(config_form, context)
template_specification.update(
{'adapter_form': self.render_adapter_form(config_form)})
parameters_equation_plot_form = EquationPlotForm()
template_specification.update({
'parametersEquationPlotForm':
self.render_adapter_form(parameters_equation_plot_form)
})
return template_specification
@expose_page
def edit_model_parameters(self):
"""
Main method, to initialize Model-Parameter visual-set.
"""
model, cortex = self.get_data_from_burst_configuration()
surface_gid = cortex.surface_gid
surface_index = load.load_entity_by_gid(surface_gid)
self.model_params_dict = self._prepare_model_params_dict(model)
context_model_parameters = SurfaceContextModelParameters(
surface_index, model,
SurfaceModelParametersForm.default_equation(),
list(self.model_params_dict.values())[0])
common.add2session(KEY_CONTEXT_MPS, context_model_parameters)
template_specification = dict(
title="Spatio temporal - Model parameters")
template_specification.update(
self.display_surface(surface_gid.hex, cortex.region_mapping_data))
dummy_form_for_initialization = SurfaceModelParametersForm(
self.model_params_dict)
self.plotted_equation_prefixes = {
self.MODEL_PARAM_FIELD:
dummy_form_for_initialization.model_param.name,
self.EQUATION_FIELD:
dummy_form_for_initialization.equation.name,
self.EQUATION_PARAMS_FIELD:
dummy_form_for_initialization.equation.subform_field.name[1:]
}
template_specification.update(
self._prepare_reload(context_model_parameters))
template_specification.update(
submit_parameters_url=
'/spatial/modelparameters/surface/submit_model_parameters',
mainContent='spatial/model_param_surface_main',
submitSurfaceParametersBtn=True)
return self.fill_default_attributes(template_specification)
@expose_fragment('spatial/model_param_surface_left')
def set_model_parameter(self, model_parameter):
context = common.get_from_session(KEY_CONTEXT_MPS)
context.current_model_param = model_parameter
template_specification = self._prepare_reload(context)
return self.fill_default_attributes(template_specification)
@cherrypy.expose
@using_template("form_fields/form_field")
@handle_error(redirect=False)
@check_user
def refresh_subform(self, equation, mapping_key):
eq_class = get_ui_name_to_equation_dict().get(equation)
context = common.get_from_session(KEY_CONTEXT_MPS)
context.current_equation = eq_class()
eq_params_form = SubformHelper.get_subform_for_field_value(
equation, mapping_key)
return {
'adapter_form': eq_params_form,
'equationsPrefixes': self.plotted_equation_prefixes
}
@cherrypy.expose
def set_equation_param(self, **param):
context = common.get_from_session(KEY_CONTEXT_MPS)
eq_params_form_class = get_form_for_equation(
type(context.current_equation))
eq_params_form = eq_params_form_class()
eq_params_form.fill_from_trait(context.current_equation)
eq_params_form.fill_from_post(param)
eq_params_form.fill_trait(context.current_equation)
@expose_fragment('spatial/model_param_surface_left')
def apply_equation(self, **kwargs):
"""
Applies an equations for computing a model parameter.
"""
context_model_parameters = common.get_from_session(KEY_CONTEXT_MPS)
context_model_parameters.apply_equation(
context_model_parameters.current_model_param,
context_model_parameters.current_equation)
template_specification = self._prepare_reload(context_model_parameters)
return self.fill_default_attributes(template_specification)
@expose_fragment('spatial/model_param_surface_focal_points')
def apply_focal_point(self, model_param, triangle_index):
"""
Adds the given focal point to the list of focal points specified for
the equation used for computing the values for the specified model param.
"""
template_specification = {}
context_model_parameters = common.get_from_session(KEY_CONTEXT_MPS)
if context_model_parameters.get_equation_for_parameter(
model_param) is not None:
context_model_parameters.apply_focal_point(model_param,
triangle_index)
else:
template_specification[
'error_msg'] = "You have no equation applied for this parameter."
template_specification[
'focal_points'] = context_model_parameters.get_focal_points_for_parameter(
model_param)
template_specification['focal_points_json'] = json.dumps(
context_model_parameters.get_focal_points_for_parameter(
model_param))
return template_specification
@expose_fragment('spatial/model_param_surface_focal_points')
def remove_focal_point(self, model_param, vertex_index):
"""
Removes the given focal point from the list of focal points specified for
the equation used for computing the values for the specified model param.
"""
context_model_parameters = common.get_from_session(KEY_CONTEXT_MPS)
context_model_parameters.remove_focal_point(model_param, vertex_index)
return {
'focal_points':
context_model_parameters.get_focal_points_for_parameter(
model_param),
'focal_points_json':
json.dumps(
context_model_parameters.get_focal_points_for_parameter(
model_param))
}
@expose_fragment('spatial/model_param_surface_focal_points')
def get_focal_points(self, model_param):
"""
Returns the html which displays the list of focal points selected for the
equation used for computing the values for the given model parameter.
"""
context_model_parameters = common.get_from_session(KEY_CONTEXT_MPS)
return {
'focal_points':
context_model_parameters.get_focal_points_for_parameter(
model_param),
'focal_points_json':
json.dumps(
context_model_parameters.get_focal_points_for_parameter(
model_param))
}
@cherrypy.expose
@handle_error(redirect=True)
@check_user
def submit_model_parameters(self, submit_action="cancel_action"):
"""
Collects the model parameters values from all the models used for the surface vertices.
@:param submit_action: a post parameter. It distinguishes if this request is a cancel or a submit
"""
if submit_action == "submit_action":
context_model_parameters = common.get_from_session(KEY_CONTEXT_MPS)
simulator = self.simulator_context.simulator
for param_name in self.model_params_dict.values():
param_data = context_model_parameters.get_data_for_model_param(
param_name)
if param_data is None:
continue
setattr(simulator.model, param_name, param_data)
### Update in session the last loaded URL for burst-page.
self.simulator_context.add_last_loaded_form_url_to_session(
SimulatorWizzardURLs.SET_INTEGRATOR_URL)
### Clean from session drawing context
common.remove_from_session(KEY_CONTEXT_MPS)
raise cherrypy.HTTPRedirect("/burst/")
def fill_default_attributes(self, template_dictionary):
"""
Overwrite base controller to add required parameters for adapter templates.
"""
template_dictionary[common.KEY_SECTION] = 'burst'
template_dictionary[common.KEY_SUB_SECTION] = 'surfacemodel'
template_dictionary[
common.KEY_INCLUDE_RESOURCES] = 'spatial/included_resources'
BaseController.fill_default_attributes(self, template_dictionary)
return template_dictionary
@expose_fragment('spatial/equation_displayer')
def get_equation_chart(self, **form_data):
"""
Returns the html which contains the plot with the equation selected by the user for a certain model param.
"""
try:
plot_form = EquationPlotForm()
if form_data:
plot_form.fill_from_post(form_data)
min_x, max_x, ui_message = self.get_x_axis_range(
plot_form.min_x.value, plot_form.max_x.value)
context_mps = common.get_from_session(KEY_CONTEXT_MPS)
equation = context_mps.current_equation
series_data, display_ui_message = equation.get_series_data(
min_range=min_x, max_range=max_x)
all_series = self.get_series_json(series_data, "Spatial")
ui_message = ''
if display_ui_message:
ui_message = self.get_ui_message(["spatial"])
return {
'allSeries': all_series,
'prefix': 'spatial',
'message': ui_message
}
except NameError as ex:
self.logger.exception(ex)
return {
'allSeries': None,
'errorMsg': "Incorrect parameters for equation passed."
}
except SyntaxError as ex:
self.logger.exception(ex)
return {
'allSeries': None,
'errorMsg': "Some of the parameters hold invalid characters."
}
except Exception as ex:
self.logger.exception(ex)
return {'allSeries': None, 'errorMsg': ex}
class RegionsModelParametersController(BurstBaseController):
"""
Controller class for placing model parameters in nodes.
"""
def __init__(self):
super(RegionsModelParametersController, self).__init__()
self.simulator_context = SimulatorContext()
@staticmethod
def _dynamics_json(dynamics):
""" Construct a json representation of a list of dynamics as used by the js client """
ret = {}
for d in dynamics:
ret[d.id] = {
'id': d.id,
'name': d.name,
'model_class': d.model_class
}
return json.dumps(ret)
def no_dynamics_page(self):
params = ({
'title': 'Model parameters',
'mainContent': 'burst/model_param_region_empty',
})
return self.fill_default_attributes(params)
@expose_page
def index(self):
current_user_id = self.simulator_context.logged_user.id
# In case the number of dynamics gets big we should add a filter in the ui.
dynamics = dao.get_dynamics_for_user(current_user_id)
if not dynamics:
return self.no_dynamics_page()
sim_config = self.simulator_context.simulator
connectivity = sim_config.connectivity
if connectivity is None:
msg = 'You have to select a connectivity before setting up the region Model. '
common.set_error_message(msg)
raise ValueError(msg)
current_project = common.get_current_project()
file_handler = FilesHelper()
conn_idx = load.load_entity_by_gid(connectivity)
conn_path = file_handler.get_project_folder(current_project, str(conn_idx.fk_from_operation))
params = ConnectivityViewer.get_connectivity_parameters(conn_idx, conn_path)
burst_config = self.simulator_context.burst_config
params.update({
'title': 'Model parameters',
'mainContent': 'burst/model_param_region',
'isSingleMode': True,
'submit_parameters_url': '/burst/modelparameters/regions/submit_model_parameters',
'dynamics': dynamics,
'dynamics_json': self._dynamics_json(dynamics),
'initial_dynamic_ids': burst_config.dynamic_ids
})
return self.fill_default_attributes(params, 'regionmodel')
@cherrypy.expose
@handle_error(redirect=True)
@check_user
def submit_model_parameters(self, node_values):
"""
Collects the model parameters values from all the models used for the connectivity nodes.
Assumes that the array indices are consistent with the node order.
"""
dynamic_ids = json.loads(node_values)
dynamics = [dao.get_dynamic(did) for did in dynamic_ids]
for dynamic in dynamics[1:]:
if dynamic.model_class != dynamics[0].model_class:
raise Exception("All dynamics must have the same model type")
model_name = dynamics[0].model_class
burst_config = self.simulator_context.burst_config
simulator_config = self.simulator_context.simulator
# update model parameters in burst config
des = SerializationManager(simulator_config)
model_parameters = [dict(json.loads(d.model_parameters)) for d in dynamics]
des.write_model_parameters(model_name, model_parameters)
# update dynamic ids in burst config
burst_config.dynamic_ids = json.dumps(dynamic_ids)
# Update in session the simulator configuration and the current form URL in wizzard for burst-page.
self.simulator_context.set_burst_config(burst_config)
self.simulator_context.add_last_loaded_form_url_to_session(SimulatorWizzardURLs.SET_INTEGRATOR_URL)
raise cherrypy.HTTPRedirect("/burst/")
