X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FdaEficas%2Fgenerator_adao.py;h=0bd1bfff841ed4f467139f5807401715d49de02d;hb=b4ec54a708ac4f11e9c710edd09219849ebf3242;hp=b23cf6871702b1c561d9c5100d758972843019c5;hpb=6a033450ee6ef0b1b9104db784707eeb551d7ca7;p=modules%2Fadao.git diff --git a/src/daEficas/generator_adao.py b/src/daEficas/generator_adao.py index b23cf68..0bd1bff 100644 --- a/src/daEficas/generator_adao.py +++ b/src/daEficas/generator_adao.py @@ -1,7 +1,30 @@ # -*- coding: utf-8 -*- -print "import generator_adao" +# +# Copyright (C) 2008-2017 EDF R&D +# +# This file is part of SALOME ADAO module +# +# This library is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2.1 of the License. +# +# This library is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +# +# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +# +# Author: André Ribes, andre.ribes@edf.fr, EDF R&D from generator.generator_python import PythonGenerator +import traceback +import logging def entryPoint(): """ @@ -24,26 +47,34 @@ class AdaoGenerator(PythonGenerator): self.text_comm = "" self.text_da = "" self.text_da_status = False + self.logger = logging.getLogger('ADAO EFICAS GENERATOR') + self.logger.setLevel(logging.INFO) + ch = logging.StreamHandler() + ch.setLevel(logging.INFO) + formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s") + ch.setFormatter(formatter) + self.logger.addHandler(ch) - def gener(self,obj,format='brut',config=None): - print "AdaoGenerator gener" + def gener(self,obj,format='brut',config=None,appli=None): + self.logger.debug("method gener called") self.text_comm = PythonGenerator.gener(self, obj, format, config) + for key, value in self.dictMCVal.iteritems(): + self.logger.debug("dictMCVAl %s %s" % (key,value)) - print "Dictionnaire" - print self.dictMCVal - try : self.text_da_status = False self.generate_da() self.text_da_status = True except: - print "Case seems not be correct" - pass + self.logger.debug("EFICAS case is not valid, python command file for YACS schema generation cannot be created") + self.logger.debug(self.text_da) + self.dictMCVal = {} + # traceback.print_exc() return self.text_comm def writeDefault(self, fn): if self.text_da_status: - print "write adao python command file" + self.logger.debug("write adao python command file") filename = fn[:fn.rfind(".")] + '.py' f = open( str(filename), 'wb') f.write( self.text_da ) @@ -61,74 +92,358 @@ class AdaoGenerator(PythonGenerator): s=PythonGenerator.generMCSIMP(self,obj) return s - def generMCFACT(self,obj): - - if obj.parent.nom == "ALGORITHM": - self.dictMCVal["ALGORITHM_NAME"] = obj.nom - - s=PythonGenerator.generMCFACT(self,obj) - return s - def generate_da(self): - + + if "__CHECKING_STUDY__StudyName" in self.dictMCVal.keys(): + self.type_of_study = "CHECKING_STUDY" + else: + self.type_of_study = "ASSIMILATION_STUDY" + self.text_da += "#-*-coding:iso-8859-1-*- \n" self.text_da += "study_config = {} \n" - self.text_da += "study_config[\"Name\"] = \"" + self.dictMCVal["__ASSIM_STUDY__STUDY_NAME"] + "\"\n" - self.text_da += "study_config[\"Algorithm\"] = \"" + self.dictMCVal["ALGORITHM_NAME"] + "\"\n" + # Extraction de Study_type + self.text_da += "study_config['StudyType'] = '" + self.type_of_study + "'\n" + # Extraction de StudyName + self.text_da += "study_config['Name'] = '" + self.dictMCVal["__"+self.type_of_study+"__StudyName"] + "'\n" + # Extraction de Debug + if "__"+self.type_of_study+"__Debug" in self.dictMCVal.keys(): + self.text_da += "study_config['Debug'] = '" + str(self.dictMCVal["__"+self.type_of_study+"__Debug"]) + "'\n" + else: + self.text_da += "study_config['Debug'] = '0'\n" + + # Extraction de Algorithm et de ses parametres + if "__"+self.type_of_study+"__AlgorithmParameters__Algorithm" in self.dictMCVal.keys(): + self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Algorithm"] + "'\n" + self.add_AlgorithmParameters() + elif "__"+self.type_of_study+"__Algorithm" in self.dictMCVal.keys(): + self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__Algorithm"] + "'\n" - self.add_data("Background") - self.add_data("BackgroundError") - self.add_data("Observation") - self.add_data("ObservationError") - self.add_data("ObservationOperator") - self.add_analysis() + if "__"+self.type_of_study+"__Background__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("Background") + if "__"+self.type_of_study+"__BackgroundError__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("BackgroundError") + if "__"+self.type_of_study+"__Observation__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("Observation") + if "__"+self.type_of_study+"__ObservationError__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("ObservationError") + if "__"+self.type_of_study+"__CheckingPoint__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("CheckingPoint") + if "__"+self.type_of_study+"__ObservationOperator__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("ObservationOperator") + if "__"+self.type_of_study+"__EvolutionModel__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("EvolutionModel") + if "__"+self.type_of_study+"__EvolutionError__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("EvolutionError") + if "__"+self.type_of_study+"__ControlInput__INPUT_TYPE" in self.dictMCVal.keys(): + self.add_data("ControlInput") + + self.add_variables() + # Parametres optionnels + + # Extraction du StudyRepertory + if "__"+self.type_of_study+"__StudyRepertory" in self.dictMCVal.keys(): + self.text_da += "study_config['Repertory'] = '" + self.dictMCVal["__"+self.type_of_study+"__StudyRepertory"] + "'\n" + # Extraction de UserPostAnalysis + if "__"+self.type_of_study+"__UserPostAnalysis__FROM" in self.dictMCVal.keys(): + self.add_UserPostAnalysis() + if "__"+self.type_of_study+"__UserDataInit__INIT_FILE" in self.dictMCVal.keys(): + self.add_init() + if "__"+self.type_of_study+"__Observers__SELECTION" in self.dictMCVal.keys(): + self.add_observers() def add_data(self, data_name): - search_text = "__ASSIM_STUDY__ALGORITHM__" + self.dictMCVal["ALGORITHM_NAME"] + "__" - back_search_text = search_text + data_name + "__" - try : - back_from = self.dictMCVal[back_search_text + "VECTOR__FROM"] - back_data = self.dictMCVal[back_search_text + "VECTOR__DATA"] - self.text_da += data_name + "_config = {} \n" - self.text_da += data_name + "_config[\"Type\"] = \"Vector\" \n" - self.text_da += data_name + "_config[\"From\"] = \"" + back_from + "\" \n" - self.text_da += data_name + "_config[\"Data\"] = \"" + back_data + "\" \n" - self.text_da += "study_config[\"" + data_name + "\"] = " + data_name + "_config \n" - except: - pass - try : - back_from = self.dictMCVal[back_search_text + "MATRIX__FROM"] - back_data = self.dictMCVal[back_search_text + "MATRIX__DATA"] + # Extraction des données + search_text = "__"+self.type_of_study+"__" + data_name + "__" + data_type = self.dictMCVal[search_text + "INPUT_TYPE"] + search_type = search_text + data_type + "__data__" + from_type = self.dictMCVal[search_type + "FROM"] + data = "" + if from_type == "String": + data = self.dictMCVal[search_type + "STRING_DATA__STRING"] + elif from_type == "Script": + data = self.dictMCVal[search_type + "SCRIPT_DATA__SCRIPT_FILE"] + elif from_type == "ScriptWithSwitch": + data = self.dictMCVal[search_type + "SCRIPTWITHSWITCH_DATA__SCRIPTWITHSWITCH_FILE"] + elif from_type == "ScriptWithFunctions": + data = self.dictMCVal[search_type + "SCRIPTWITHFUNCTIONS_DATA__SCRIPTWITHFUNCTIONS_FILE"] + elif from_type == "ScriptWithOneFunction": + data = self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__SCRIPTWITHONEFUNCTION_FILE"] + elif from_type == "FunctionDict": + data = self.dictMCVal[search_type + "FUNCTIONDICT_DATA__FUNCTIONDICT_FILE"] + else: + raise Exception('From Type unknown', from_type) + + if from_type == "String" or from_type == "Script": + self.text_da += data_name + "_config = {}\n" + self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n" + self.text_da += data_name + "_config['From'] = '" + from_type + "'\n" + self.text_da += data_name + "_config['Data'] = '" + data + "'\n" + if search_text+"Stored" in self.dictMCVal.keys(): + self.text_da += data_name + "_config['Stored'] = '" + str(self.dictMCVal[search_text+"Stored"]) + "'\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + + if from_type == "ScriptWithSwitch": + self.text_da += data_name + "_ScriptWithSwitch = {}\n" + self.text_da += data_name + "_ScriptWithSwitch['Function'] = ['Direct', 'Tangent', 'Adjoint']\n" + self.text_da += data_name + "_ScriptWithSwitch['Script'] = {}\n" + self.text_da += data_name + "_ScriptWithSwitch['Script']['Direct'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithSwitch['Script']['Tangent'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithSwitch['Script']['Adjoint'] = '" + data + "'\n" + self.text_da += data_name + "_config = {}\n" + self.text_da += data_name + "_config['Type'] = 'Function'\n" + self.text_da += data_name + "_config['From'] = 'ScriptWithSwitch'\n" + self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithSwitch\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + + if from_type == "ScriptWithFunctions": + self.text_da += data_name + "_ScriptWithFunctions = {}\n" + self.text_da += data_name + "_ScriptWithFunctions['Function'] = ['Direct', 'Tangent', 'Adjoint']\n" + self.text_da += data_name + "_ScriptWithFunctions['Script'] = {}\n" + self.text_da += data_name + "_ScriptWithFunctions['Script']['Direct'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithFunctions['Script']['Tangent'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithFunctions['Script']['Adjoint'] = '" + data + "'\n" + self.text_da += data_name + "_config = {}\n" + self.text_da += data_name + "_config['Type'] = 'Function'\n" + self.text_da += data_name + "_config['From'] = 'ScriptWithFunctions'\n" + self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithFunctions\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + + if from_type == "ScriptWithOneFunction": + self.text_da += data_name + "_ScriptWithOneFunction = {}\n" + self.text_da += data_name + "_ScriptWithOneFunction['Function'] = ['Direct', 'Tangent', 'Adjoint']\n" + self.text_da += data_name + "_ScriptWithOneFunction['Script'] = {}\n" + self.text_da += data_name + "_ScriptWithOneFunction['Script']['Direct'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithOneFunction['Script']['Tangent'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithOneFunction['Script']['Adjoint'] = '" + data + "'\n" + self.text_da += data_name + "_ScriptWithOneFunction['DifferentialIncrement'] = " + str(float(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__DifferentialIncrement"])) + "\n" + self.text_da += data_name + "_ScriptWithOneFunction['CenteredFiniteDifference'] = " + str(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__CenteredFiniteDifference"]) + "\n" + if search_type + "SCRIPTWITHONEFUNCTION_DATA__EnableMultiProcessing" in self.dictMCVal.keys(): + self.text_da += data_name + "_ScriptWithOneFunction['EnableMultiProcessing'] = " + str(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__EnableMultiProcessing"]) + "\n" + if search_type + "SCRIPTWITHONEFUNCTION_DATA__NumberOfProcesses" in self.dictMCVal.keys(): + self.text_da += data_name + "_ScriptWithOneFunction['NumberOfProcesses'] = " + str(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__NumberOfProcesses"]) + "\n" + self.text_da += data_name + "_config = {}\n" + self.text_da += data_name + "_config['Type'] = 'Function'\n" + self.text_da += data_name + "_config['From'] = 'ScriptWithOneFunction'\n" + self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithOneFunction\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + + if from_type == "FunctionDict": + self.text_da += data_name + "_FunctionDict = {}\n" + self.text_da += data_name + "_FunctionDict['Function'] = ['Direct', 'Tangent', 'Adjoint']\n" + self.text_da += data_name + "_FunctionDict['Script'] = {}\n" + self.text_da += data_name + "_FunctionDict['Script']['Direct'] = '" + data + "'\n" + self.text_da += data_name + "_FunctionDict['Script']['Tangent'] = '" + data + "'\n" + self.text_da += data_name + "_FunctionDict['Script']['Adjoint'] = '" + data + "'\n" + self.text_da += data_name + "_config = {}\n" + self.text_da += data_name + "_config['Type'] = 'Function'\n" + self.text_da += data_name + "_config['From'] = 'FunctionDict'\n" + self.text_da += data_name + "_config['Data'] = " + data_name + "_FunctionDict\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + + def add_init(self): + + init_file_data = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__INIT_FILE"] + init_target_list = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__TARGET_LIST"] + + self.text_da += "Init_config = {}\n" + self.text_da += "Init_config['Type'] = 'Dict'\n" + self.text_da += "Init_config['From'] = 'Script'\n" + self.text_da += "Init_config['Data'] = '" + init_file_data + "'\n" + self.text_da += "Init_config['Target'] = [" + if type(init_target_list) is type("str"): + self.text_da += "'" + init_target_list + "'," + else: + for target in init_target_list: + self.text_da += "'" + target + "'," + self.text_da += "]\n" + self.text_da += "study_config['UserDataInit'] = Init_config\n" + + def add_UserPostAnalysis(self): + from_type = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__FROM"] + data = "" + if from_type == "String": + data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__STRING_DATA__STRING"] + self.text_da += "Analysis_config = {}\n" + self.text_da += "Analysis_config['From'] = 'String'\n" + self.text_da += "Analysis_config['Data'] = \"\"\"" + data + "\"\"\"\n" + self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n" + elif from_type == "Script": + data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__SCRIPT_DATA__SCRIPT_FILE"] + self.text_da += "Analysis_config = {}\n" + self.text_da += "Analysis_config['From'] = 'Script'\n" + self.text_da += "Analysis_config['Data'] = '" + data + "'\n" + self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n" + elif from_type == "Template": + tmpl = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__TEMPLATE_DATA__Template"] + data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__TEMPLATE_DATA__%s__ValueTemplate"%tmpl] + self.text_da += "Analysis_config = {}\n" + self.text_da += "Analysis_config['From'] = 'String'\n" + self.text_da += "Analysis_config['Data'] = \"\"\"" + data + "\"\"\"\n" + self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n" + else: + raise Exception('From Type unknown', from_type) + + def add_AlgorithmParameters(self): + + if not self.dictMCVal.has_key("__"+self.type_of_study+"__AlgorithmParameters__Parameters"): return + + data_name = "AlgorithmParameters" + data_type = "Dict" + para_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Parameters"] + if para_type == "Defaults": + from_type = para_type + elif para_type == "Dict": + from_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__FROM"] + + if from_type == "Script": + data = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__SCRIPT_DATA__SCRIPT_FILE"] self.text_da += data_name + "_config = {} \n" - self.text_da += data_name + "_config[\"Type\"] = \"Matrix\" \n" - self.text_da += data_name + "_config[\"From\"] = \"" + back_from + "\" \n" - self.text_da += data_name + "_config[\"Data\"] = \"" + back_data + "\" \n" - self.text_da += "study_config[\"" + data_name + "\"] = " + data_name + "_config \n" - except: - pass + self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n" + self.text_da += data_name + "_config['From'] = '" + from_type + "'\n" + self.text_da += data_name + "_config['Data'] = '" + data + "'\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + elif from_type == "String": + data = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__STRING_DATA__STRING"] + self.text_da += data_name + "_config = {} \n" + self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n" + self.text_da += data_name + "_config['From'] = '" + from_type + "'\n" + self.text_da += data_name + "_config['Data'] = '" + data + "'\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" + elif from_type == "Defaults": + base = "__"+self.type_of_study+"__AlgorithmParameters__Parameters" + keys = [k for k in self.dictMCVal.keys() if base in k] + keys.remove(base) + keys = [k.replace(base,'') for k in keys] + data = '{' + for k in keys: + key = k.split('__')[-1] + val = self.dictMCVal[base+k] + # print key," = ",val," ",type(val) + if isinstance(val, str) and key == "SetSeed": + data += '"%s":%s,'%(key,int(val)) + elif isinstance(val, str) and not (val.count('[')>=2 or val.count('(')>=2): + data += '"%s":"%s",'%(key,val) + else: + data += '"%s":%s,'%(key,val) + data = data.replace("'",'"') + data += '}' + self.text_da += data_name + "_config = {} \n" + self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n" + self.text_da += data_name + "_config['From'] = 'String'\n" + self.text_da += data_name + "_config['Data'] = '" + data + "'\n" + self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n" - def add_analysis(self): - search_text = "__ASSIM_STUDY__ALGORITHM__" + self.dictMCVal["ALGORITHM_NAME"] + "__Analysis__" - try : - ana_from = self.dictMCVal[search_text + "FROM"] - print ana_from - if ana_from == "String": - ana_data = self.dictMCVal[search_text + "STRING_DATA__STRING"] - self.text_da += "Analysis_config = {} \n" - self.text_da += "Analysis_config[\"From\"] = \"String\" \n" - self.text_da += "Analysis_config[\"Data\"] = \"\"\"" + ana_data + "\"\"\" \n" - self.text_da += "study_config[\"Analysis\"] = Analysis_config \n" - pass - if ana_from == "File": - ana_data = self.dictMCVal[search_text + "FILE_DATA__FILE"] - self.text_da += "Analysis_config = {} \n" - self.text_da += "Analysis_config[\"From\"] = \"File\" \n" - self.text_da += "Analysis_config[\"Data\"] = \"" + ana_data + "\" \n" - self.text_da += "study_config[\"Analysis\"] = Analysis_config \n" - pass - except: - pass + def add_variables(self): + + # Input variables + if "__"+self.type_of_study+"__InputVariables__NAMES" in self.dictMCVal.keys(): + names = [] + sizes = [] + if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"], type("")): + names.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"]) + else: + names = self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"] + if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"], type(1)): + sizes.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"]) + else: + sizes = self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"] + + self.text_da += "inputvariables_config = {}\n" + self.text_da += "inputvariables_config['Order'] = %s\n" % list(names) + for name, size in zip(names, sizes): + self.text_da += "inputvariables_config['%s'] = %s\n" % (name,size) + self.text_da += "study_config['InputVariables'] = inputvariables_config\n" + else: + self.text_da += "inputvariables_config = {}\n" + self.text_da += "inputvariables_config['Order'] =['adao_default']\n" + self.text_da += "inputvariables_config['adao_default'] = -1\n" + self.text_da += "study_config['InputVariables'] = inputvariables_config\n" + + # Output variables + if "__"+self.type_of_study+"__OutputVariables__NAMES" in self.dictMCVal.keys(): + names = [] + sizes = [] + if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"], type("")): + names.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"]) + else: + names = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"] + if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"], type(1)): + sizes.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"]) + else: + sizes = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"] + + self.text_da += "outputvariables_config = {}\n" + self.text_da += "outputvariables_config['Order'] = %s\n" % list(names) + for name, size in zip(names, sizes): + self.text_da += "outputvariables_config['%s'] = %s\n" % (name,size) + self.text_da += "study_config['OutputVariables'] = outputvariables_config\n" + else: + self.text_da += "outputvariables_config = {}\n" + self.text_da += "outputvariables_config['Order'] = ['adao_default']\n" + self.text_da += "outputvariables_config['adao_default'] = -1\n" + self.text_da += "study_config['OutputVariables'] = outputvariables_config\n" + + def add_observers(self): + observers = {} + observer = self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"] + if isinstance(observer, type("")): + self.add_observer_in_dict(observer, observers) + else: + for observer in self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"]: + self.add_observer_in_dict(observer, observers) + + # Write observers in the python command file + number = 2 + self.text_da += "observers = {}\n" + for observer in observers.keys(): + number += 1 + self.text_da += "observers[\"" + observer + "\"] = {}\n" + self.text_da += "observers[\"" + observer + "\"][\"number\"] = " + str(number) + "\n" + self.text_da += "observers[\"" + observer + "\"][\"nodetype\"] = \"" + observers[observer]["nodetype"] + "\"\n" + if observers[observer]["nodetype"] == "String": + self.text_da += "observers[\"" + observer + "\"][\"String\"] = \"\"\"" + observers[observer]["script"] + "\"\"\"\n" + elif observers[observer]["nodetype"] == "Template": + self.text_da += "observers[\"" + observer + "\"][\"String\"] = \"\"\"" + observers[observer]["script"] + "\"\"\"\n" + self.text_da += "observers[\"" + observer + "\"][\"Template\"] = \"\"\"" + observers[observer]["template"] + "\"\"\"\n" + else: + self.text_da += "observers[\"" + observer + "\"][\"Script\"] = \"" + observers[observer]["file"] + "\"\n" + if "scheduler" in observers[observer].keys(): + self.text_da += "observers[\"" + observer + "\"][\"scheduler\"] = \"\"\"" + observers[observer]["scheduler"] + "\"\"\"\n" + if "info" in observers[observer].keys(): + self.text_da += "observers[\"" + observer + "\"][\"info\"] = \"\"\"" + observers[observer]["info"] + "\"\"\"\n" + self.text_da += "study_config['Observers'] = observers\n" + + def add_observer_in_dict(self, observer, observers): + """ + Add observer in the observers dict. + """ + observers[observer] = {} + observers[observer]["name"] = observer + observer_eficas_name = "__"+self.type_of_study+"__Observers__" + observer + "__" + observer + "_data__" + # NodeType + node_type_key_name = observer_eficas_name + "NodeType" + observers[observer]["nodetype"] = self.dictMCVal[node_type_key_name] + + # NodeType script/file + if observers[observer]["nodetype"] == "String": + observers[observer]["script"] = self.dictMCVal[observer_eficas_name + "PythonScript__Value"] + elif observers[observer]["nodetype"] == "Template": + observers[observer]["nodetype"] = "String" + observer_template_key = observer_eficas_name + "ObserverTemplate__" + observers[observer]["template"] = self.dictMCVal[observer_template_key + "Template"] + observers[observer]["script"] = self.dictMCVal[observer_template_key + observers[observer]["template"] + "__ValueTemplate"] + else: + observers[observer]["file"] = self.dictMCVal[observer_eficas_name + "UserFile__Value"] + + # Scheduler + scheduler_key_name = observer_eficas_name + "Scheduler" + if scheduler_key_name in self.dictMCVal.keys(): + observers[observer]["scheduler"] = self.dictMCVal[scheduler_key_name] + # Info + info_key_name = observer_eficas_name + "Info" + if info_key_name in self.dictMCVal.keys(): + observers[observer]["info"] = self.dictMCVal[info_key_name]