1 # -*- coding: utf-8 -*-
3 # Copyright (C) 2008-2015 EDF R&D
5 # This file is part of SALOME ADAO module
7 # This library is free software; you can redistribute it and/or
8 # modify it under the terms of the GNU Lesser General Public
9 # License as published by the Free Software Foundation; either
10 # version 2.1 of the License.
12 # This library is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 # Lesser General Public License for more details.
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this library; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 # Author: André Ribes, andre.ribes@edf.fr, EDF R&D
25 from generator.generator_python import PythonGenerator
31 Retourne les informations necessaires pour le chargeur de plugins
33 Ces informations sont retournees dans un dictionnaire
38 # La factory pour creer une instance du plugin
39 'factory' : AdaoGenerator,
42 class AdaoGenerator(PythonGenerator):
44 def __init__(self,cr=None):
45 PythonGenerator.__init__(self, cr)
49 self.text_da_status = False
50 self.logger = logging.getLogger('ADAO EFICAS GENERATOR')
51 self.logger.setLevel(logging.INFO)
52 ch = logging.StreamHandler()
53 ch.setLevel(logging.INFO)
54 formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")
55 ch.setFormatter(formatter)
56 self.logger.addHandler(ch)
58 def gener(self,obj,format='brut',config=None):
59 self.logger.debug("method gener called")
60 self.text_comm = PythonGenerator.gener(self, obj, format, config)
61 for key, value in self.dictMCVal.iteritems():
62 self.logger.debug("dictMCVAl %s %s" % (key,value))
65 self.text_da_status = False
67 self.text_da_status = True
69 self.logger.info("Case is not correct, python command file for YACS schema generation cannot be created")
70 self.logger.debug(self.text_da)
75 def writeDefault(self, fn):
76 if self.text_da_status:
77 self.logger.debug("write adao python command file")
78 filename = fn[:fn.rfind(".")] + '.py'
79 f = open( str(filename), 'wb')
80 f.write( self.text_da )
83 def generMCSIMP(self,obj) :
85 Convertit un objet MCSIMP en texte python
88 for i in obj.get_genealogie() :
90 self.dictMCVal[clef]=obj.valeur
92 s=PythonGenerator.generMCSIMP(self,obj)
95 def generate_da(self):
97 if "__CHECKING_STUDY__StudyName" in self.dictMCVal.keys():
98 self.type_of_study = "CHECKING_STUDY"
100 self.type_of_study = "ASSIMILATION_STUDY"
102 self.text_da += "#-*-coding:iso-8859-1-*- \n"
103 self.text_da += "study_config = {} \n"
105 # Extraction de Study_type
106 self.text_da += "study_config['StudyType'] = '" + self.type_of_study + "'\n"
107 # Extraction de StudyName
108 self.text_da += "study_config['Name'] = '" + self.dictMCVal["__"+self.type_of_study+"__StudyName"] + "'\n"
109 # Extraction de Debug
110 if "__"+self.type_of_study+"__Debug" in self.dictMCVal.keys():
111 self.text_da += "study_config['Debug'] = '" + str(self.dictMCVal["__"+self.type_of_study+"__Debug"]) + "'\n"
113 self.text_da += "study_config['Debug'] = '0'\n"
115 # Extraction de Algorithm et de ses parametres
116 if "__"+self.type_of_study+"__AlgorithmParameters__Algorithm" in self.dictMCVal.keys():
117 self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Algorithm"] + "'\n"
118 self.add_AlgorithmParameters()
119 elif "__"+self.type_of_study+"__Algorithm" in self.dictMCVal.keys():
120 self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__Algorithm"] + "'\n"
122 if "__"+self.type_of_study+"__Background__INPUT_TYPE" in self.dictMCVal.keys():
123 self.add_data("Background")
124 if "__"+self.type_of_study+"__BackgroundError__INPUT_TYPE" in self.dictMCVal.keys():
125 self.add_data("BackgroundError")
126 if "__"+self.type_of_study+"__Observation__INPUT_TYPE" in self.dictMCVal.keys():
127 self.add_data("Observation")
128 if "__"+self.type_of_study+"__ObservationError__INPUT_TYPE" in self.dictMCVal.keys():
129 self.add_data("ObservationError")
130 if "__"+self.type_of_study+"__CheckingPoint__INPUT_TYPE" in self.dictMCVal.keys():
131 self.add_data("CheckingPoint")
132 if "__"+self.type_of_study+"__ObservationOperator__INPUT_TYPE" in self.dictMCVal.keys():
133 self.add_data("ObservationOperator")
134 if "__"+self.type_of_study+"__EvolutionModel__INPUT_TYPE" in self.dictMCVal.keys():
135 self.add_data("EvolutionModel")
136 if "__"+self.type_of_study+"__EvolutionError__INPUT_TYPE" in self.dictMCVal.keys():
137 self.add_data("EvolutionError")
138 if "__"+self.type_of_study+"__ControlInput__INPUT_TYPE" in self.dictMCVal.keys():
139 self.add_data("ControlInput")
142 # Parametres optionnels
144 # Extraction du StudyRepertory
145 if "__"+self.type_of_study+"__StudyRepertory" in self.dictMCVal.keys():
146 self.text_da += "study_config['Repertory'] = '" + self.dictMCVal["__"+self.type_of_study+"__StudyRepertory"] + "'\n"
147 # Extraction de UserPostAnalysis
148 if "__"+self.type_of_study+"__UserPostAnalysis__FROM" in self.dictMCVal.keys():
149 self.add_UserPostAnalysis()
150 if "__"+self.type_of_study+"__UserDataInit__INIT_FILE" in self.dictMCVal.keys():
152 if "__"+self.type_of_study+"__Observers__SELECTION" in self.dictMCVal.keys():
155 def add_data(self, data_name):
157 # Extraction des données
158 search_text = "__"+self.type_of_study+"__" + data_name + "__"
159 data_type = self.dictMCVal[search_text + "INPUT_TYPE"]
160 search_type = search_text + data_type + "__data__"
161 from_type = self.dictMCVal[search_type + "FROM"]
163 if from_type == "String":
164 data = self.dictMCVal[search_type + "STRING_DATA__STRING"]
165 elif from_type == "Script":
166 data = self.dictMCVal[search_type + "SCRIPT_DATA__SCRIPT_FILE"]
167 elif from_type == "ScriptWithSwitch":
168 data = self.dictMCVal[search_type + "SCRIPTWITHSWITCH_DATA__SCRIPTWITHSWITCH_FILE"]
169 elif from_type == "ScriptWithFunctions":
170 data = self.dictMCVal[search_type + "SCRIPTWITHFUNCTIONS_DATA__SCRIPTWITHFUNCTIONS_FILE"]
171 elif from_type == "ScriptWithOneFunction":
172 data = self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__SCRIPTWITHONEFUNCTION_FILE"]
173 elif from_type == "FunctionDict":
174 data = self.dictMCVal[search_type + "FUNCTIONDICT_DATA__FUNCTIONDICT_FILE"]
176 raise Exception('From Type unknown', from_type)
178 if from_type == "String" or from_type == "Script":
179 self.text_da += data_name + "_config = {}\n"
180 self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
181 self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
182 self.text_da += data_name + "_config['Data'] = '" + data + "'\n"
183 if search_text+"Stored" in self.dictMCVal.keys():
184 self.text_da += data_name + "_config['Stored'] = '" + str(self.dictMCVal[search_text+"Stored"]) + "'\n"
185 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
187 if from_type == "ScriptWithSwitch":
188 self.text_da += data_name + "_ScriptWithSwitch = {}\n"
189 self.text_da += data_name + "_ScriptWithSwitch['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
190 self.text_da += data_name + "_ScriptWithSwitch['Script'] = {}\n"
191 self.text_da += data_name + "_ScriptWithSwitch['Script']['Direct'] = '" + data + "'\n"
192 self.text_da += data_name + "_ScriptWithSwitch['Script']['Tangent'] = '" + data + "'\n"
193 self.text_da += data_name + "_ScriptWithSwitch['Script']['Adjoint'] = '" + data + "'\n"
194 self.text_da += data_name + "_config = {}\n"
195 self.text_da += data_name + "_config['Type'] = 'Function'\n"
196 self.text_da += data_name + "_config['From'] = 'ScriptWithSwitch'\n"
197 self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithSwitch\n"
198 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
200 if from_type == "ScriptWithFunctions":
201 self.text_da += data_name + "_ScriptWithFunctions = {}\n"
202 self.text_da += data_name + "_ScriptWithFunctions['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
203 self.text_da += data_name + "_ScriptWithFunctions['Script'] = {}\n"
204 self.text_da += data_name + "_ScriptWithFunctions['Script']['Direct'] = '" + data + "'\n"
205 self.text_da += data_name + "_ScriptWithFunctions['Script']['Tangent'] = '" + data + "'\n"
206 self.text_da += data_name + "_ScriptWithFunctions['Script']['Adjoint'] = '" + data + "'\n"
207 self.text_da += data_name + "_config = {}\n"
208 self.text_da += data_name + "_config['Type'] = 'Function'\n"
209 self.text_da += data_name + "_config['From'] = 'ScriptWithFunctions'\n"
210 self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithFunctions\n"
211 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
213 if from_type == "ScriptWithOneFunction":
214 self.text_da += data_name + "_ScriptWithOneFunction = {}\n"
215 self.text_da += data_name + "_ScriptWithOneFunction['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
216 self.text_da += data_name + "_ScriptWithOneFunction['Script'] = {}\n"
217 self.text_da += data_name + "_ScriptWithOneFunction['Script']['Direct'] = '" + data + "'\n"
218 self.text_da += data_name + "_ScriptWithOneFunction['Script']['Tangent'] = '" + data + "'\n"
219 self.text_da += data_name + "_ScriptWithOneFunction['Script']['Adjoint'] = '" + data + "'\n"
220 self.text_da += data_name + "_ScriptWithOneFunction['DifferentialIncrement'] = " + str(float(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__DifferentialIncrement"])) + "\n"
221 self.text_da += data_name + "_ScriptWithOneFunction['CenteredFiniteDifference'] = " + str(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__CenteredFiniteDifference"]) + "\n"
222 if search_type + "SCRIPTWITHONEFUNCTION_DATA__EnableMultiProcessing" in self.dictMCVal.keys():
223 self.text_da += data_name + "_ScriptWithOneFunction['EnableMultiProcessing'] = " + str(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__EnableMultiProcessing"]) + "\n"
224 self.text_da += data_name + "_config = {}\n"
225 self.text_da += data_name + "_config['Type'] = 'Function'\n"
226 self.text_da += data_name + "_config['From'] = 'ScriptWithOneFunction'\n"
227 self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithOneFunction\n"
228 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
230 if from_type == "FunctionDict":
231 self.text_da += data_name + "_FunctionDict = {}\n"
232 self.text_da += data_name + "_FunctionDict['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
233 self.text_da += data_name + "_FunctionDict['Script'] = {}\n"
234 self.text_da += data_name + "_FunctionDict['Script']['Direct'] = '" + data + "'\n"
235 self.text_da += data_name + "_FunctionDict['Script']['Tangent'] = '" + data + "'\n"
236 self.text_da += data_name + "_FunctionDict['Script']['Adjoint'] = '" + data + "'\n"
237 self.text_da += data_name + "_config = {}\n"
238 self.text_da += data_name + "_config['Type'] = 'Function'\n"
239 self.text_da += data_name + "_config['From'] = 'FunctionDict'\n"
240 self.text_da += data_name + "_config['Data'] = " + data_name + "_FunctionDict\n"
241 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
245 init_file_data = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__INIT_FILE"]
246 init_target_list = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__TARGET_LIST"]
248 self.text_da += "Init_config = {}\n"
249 self.text_da += "Init_config['Type'] = 'Dict'\n"
250 self.text_da += "Init_config['From'] = 'Script'\n"
251 self.text_da += "Init_config['Data'] = '" + init_file_data + "'\n"
252 self.text_da += "Init_config['Target'] = ["
253 if type(init_target_list) is type("str"):
254 self.text_da += "'" + init_target_list + "',"
256 for target in init_target_list:
257 self.text_da += "'" + target + "',"
258 self.text_da += "]\n"
259 self.text_da += "study_config['UserDataInit'] = Init_config\n"
261 def add_UserPostAnalysis(self):
263 from_type = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__FROM"]
265 if from_type == "String":
266 data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__STRING_DATA__STRING"]
267 self.text_da += "Analysis_config = {}\n"
268 self.text_da += "Analysis_config['From'] = 'String'\n"
269 self.text_da += "Analysis_config['Data'] = \"\"\"" + data + "\"\"\"\n"
270 self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n"
271 elif from_type == "Script":
272 data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__SCRIPT_DATA__SCRIPT_FILE"]
273 self.text_da += "Analysis_config = {}\n"
274 self.text_da += "Analysis_config['From'] = 'Script'\n"
275 self.text_da += "Analysis_config['Data'] = '" + data + "'\n"
276 self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n"
277 elif from_type == "Template":
278 tmpl = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__TEMPLATE_DATA__Template"]
279 data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__TEMPLATE_DATA__%s__ValueTemplate"%tmpl]
280 self.text_da += "Analysis_config = {}\n"
281 self.text_da += "Analysis_config['From'] = 'String'\n"
282 self.text_da += "Analysis_config['Data'] = \"\"\"" + data + "\"\"\"\n"
283 self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n"
285 raise Exception('From Type unknown', from_type)
287 def add_AlgorithmParameters(self):
289 if not self.dictMCVal.has_key("__"+self.type_of_study+"__AlgorithmParameters__Parameters"): return
291 data_name = "AlgorithmParameters"
293 para_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Parameters"]
294 if para_type == "Defaults":
295 from_type = para_type
296 elif para_type == "Dict":
297 from_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__FROM"]
299 if from_type == "Script":
300 data = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__SCRIPT_DATA__SCRIPT_FILE"]
301 self.text_da += data_name + "_config = {} \n"
302 self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
303 self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
304 self.text_da += data_name + "_config['Data'] = '" + data + "'\n"
305 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
306 elif from_type == "String":
307 data = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__STRING_DATA__STRING"]
308 self.text_da += data_name + "_config = {} \n"
309 self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
310 self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
311 self.text_da += data_name + "_config['Data'] = '" + data + "'\n"
312 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
313 elif from_type == "Defaults":
314 base = "__"+self.type_of_study+"__AlgorithmParameters__Parameters"
315 keys = [k for k in self.dictMCVal.keys() if base in k]
317 keys = [k.replace(base,'') for k in keys]
320 data += '"%s":"%s",'%(k.split('__')[-1],self.dictMCVal[base+k])
322 self.text_da += data_name + "_config = {} \n"
323 self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
324 self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
325 self.text_da += data_name + "_config['Data'] = '" + data + "'\n"
326 self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
328 def add_variables(self):
331 if "__"+self.type_of_study+"__InputVariables__NAMES" in self.dictMCVal.keys():
334 if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"], type("")):
335 names.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"])
337 names = self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"]
338 if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"], type(1)):
339 sizes.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"])
341 sizes = self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"]
343 self.text_da += "inputvariables_config = {}\n"
344 self.text_da += "inputvariables_config['Order'] = %s\n" % list(names)
345 for name, size in zip(names, sizes):
346 self.text_da += "inputvariables_config['%s'] = %s\n" % (name,size)
347 self.text_da += "study_config['InputVariables'] = inputvariables_config\n"
349 self.text_da += "inputvariables_config = {}\n"
350 self.text_da += "inputvariables_config['Order'] =['adao_default']\n"
351 self.text_da += "inputvariables_config['adao_default'] = -1\n"
352 self.text_da += "study_config['InputVariables'] = inputvariables_config\n"
355 if "__"+self.type_of_study+"__OutputVariables__NAMES" in self.dictMCVal.keys():
358 if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"], type("")):
359 names.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"])
361 names = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"]
362 if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"], type(1)):
363 sizes.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"])
365 sizes = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"]
367 self.text_da += "outputvariables_config = {}\n"
368 self.text_da += "outputvariables_config['Order'] = %s\n" % list(names)
369 for name, size in zip(names, sizes):
370 self.text_da += "outputvariables_config['%s'] = %s\n" % (name,size)
371 self.text_da += "study_config['OutputVariables'] = outputvariables_config\n"
373 self.text_da += "outputvariables_config = {}\n"
374 self.text_da += "outputvariables_config['Order'] = ['adao_default']\n"
375 self.text_da += "outputvariables_config['adao_default'] = -1\n"
376 self.text_da += "study_config['OutputVariables'] = outputvariables_config\n"
378 def add_observers(self):
380 observer = self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"]
381 if isinstance(observer, type("")):
382 self.add_observer_in_dict(observer, observers)
384 for observer in self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"]:
385 self.add_observer_in_dict(observer, observers)
387 # Write observers in the python command file
389 self.text_da += "observers = {}\n"
390 for observer in observers.keys():
392 self.text_da += "observers[\"" + observer + "\"] = {}\n"
393 self.text_da += "observers[\"" + observer + "\"][\"number\"] = " + str(number) + "\n"
394 self.text_da += "observers[\"" + observer + "\"][\"nodetype\"] = \"" + observers[observer]["nodetype"] + "\"\n"
395 if observers[observer]["nodetype"] == "String":
396 self.text_da += "observers[\"" + observer + "\"][\"String\"] = \"\"\"" + observers[observer]["script"] + "\"\"\"\n"
397 elif observers[observer]["nodetype"] == "Template":
398 self.text_da += "observers[\"" + observer + "\"][\"String\"] = \"\"\"" + observers[observer]["script"] + "\"\"\"\n"
399 self.text_da += "observers[\"" + observer + "\"][\"Template\"] = \"\"\"" + observers[observer]["template"] + "\"\"\"\n"
401 self.text_da += "observers[\"" + observer + "\"][\"Script\"] = \"" + observers[observer]["file"] + "\"\n"
402 if "scheduler" in observers[observer].keys():
403 self.text_da += "observers[\"" + observer + "\"][\"scheduler\"] = \"\"\"" + observers[observer]["scheduler"] + "\"\"\"\n"
404 if "info" in observers[observer].keys():
405 self.text_da += "observers[\"" + observer + "\"][\"info\"] = \"\"\"" + observers[observer]["info"] + "\"\"\"\n"
406 self.text_da += "study_config['Observers'] = observers\n"
408 def add_observer_in_dict(self, observer, observers):
410 Add observer in the observers dict.
412 observers[observer] = {}
413 observers[observer]["name"] = observer
414 observer_eficas_name = "__"+self.type_of_study+"__Observers__" + observer + "__" + observer + "_data__"
416 node_type_key_name = observer_eficas_name + "NodeType"
417 observers[observer]["nodetype"] = self.dictMCVal[node_type_key_name]
419 # NodeType script/file
420 if observers[observer]["nodetype"] == "String":
421 observers[observer]["script"] = self.dictMCVal[observer_eficas_name + "PythonScript__Value"]
422 elif observers[observer]["nodetype"] == "Template":
423 observers[observer]["nodetype"] = "String"
424 observer_template_key = observer_eficas_name + "ObserverTemplate__"
425 observers[observer]["template"] = self.dictMCVal[observer_template_key + "Template"]
426 observers[observer]["script"] = self.dictMCVal[observer_template_key + observers[observer]["template"] + "__ValueTemplate"]
428 observers[observer]["file"] = self.dictMCVal[observer_eficas_name + "UserFile__Value"]
431 scheduler_key_name = observer_eficas_name + "Scheduler"
432 if scheduler_key_name in self.dictMCVal.keys():
433 observers[observer]["scheduler"] = self.dictMCVal[scheduler_key_name]
436 info_key_name = observer_eficas_name + "Info"
437 if info_key_name in self.dictMCVal.keys():
438 observers[observer]["info"] = self.dictMCVal[info_key_name]