Salome HOME
Documentation and code minor corrections
[modules/adao.git] / src / daEficas / generator_adao.py
index d1bcd293a4581c46c67e9bce9072aa6d4ecda4b4..79c786f4318c66008154190148a2cad5de7d5aab 100644 (file)
@@ -1,5 +1,8 @@
 # -*- coding: utf-8 -*-
-# Copyright (C) 2010-2013 EDF R&D
+#
+# Copyright (C) 2008-2018 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
@@ -52,10 +55,10 @@ class AdaoGenerator(PythonGenerator):
     ch.setFormatter(formatter)
     self.logger.addHandler(ch)
 
-  def gener(self,obj,format='brut',config=None):
+  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():
+    for key, value in self.dictMCVal.items():
       self.logger.debug("dictMCVAl %s %s" % (key,value))
 
     try :
@@ -63,17 +66,18 @@ class AdaoGenerator(PythonGenerator):
       self.generate_da()
       self.text_da_status = True
     except:
-      self.logger.info("Case is not correct, python command file for YACS schema generation cannot be created")
+      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()
+      traceback.print_exc()
     return self.text_comm
 
   def writeDefault(self, fn):
     if self.text_da_status:
       self.logger.debug("write adao python command file")
       filename = fn[:fn.rfind(".")] + '.py'
-      f = open( str(filename), 'wb')
+      #~ f = open( str(filename), 'wb')
+      f = open( str(filename), 'w')
       f.write( self.text_da )
       f.close()
 
@@ -82,7 +86,7 @@ class AdaoGenerator(PythonGenerator):
     Convertit un objet MCSIMP en texte python
     """
     clef=""
-    for i in obj.get_genealogie() :
+    for i in obj.getGenealogie() :
       clef=clef+"__"+i
     self.dictMCVal[clef]=obj.valeur
 
@@ -90,23 +94,31 @@ class AdaoGenerator(PythonGenerator):
     return s
 
   def generate_da(self):
-  
-    if "__CHECKING_STUDY__Study_name" in self.dictMCVal.keys():
+
+    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 += "#-*- coding: utf-8 -*-\n"
+    self.text_da += "study_config = {}\n"
 
     # Extraction de Study_type
     self.text_da += "study_config['StudyType'] = '" + self.type_of_study + "'\n"
-    # Extraction de Study_name
-    self.text_da += "study_config['Name'] = '" + self.dictMCVal["__"+self.type_of_study+"__Study_name"] + "'\n"
+    # Extraction de StudyName
+    self.text_da += "study_config['Name'] = '" + self.dictMCVal["__"+self.type_of_study+"__StudyName"] + "'\n"
     # Extraction de Debug
-    self.text_da += "study_config['Debug'] = '" + str(self.dictMCVal["__"+self.type_of_study+"__Debug"]) + "'\n"
-    # Extraction de Algorithm
-    self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__Algorithm"] + "'\n"
+    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"
 
     if "__"+self.type_of_study+"__Background__INPUT_TYPE" in self.dictMCVal.keys():
       self.add_data("Background")
@@ -130,12 +142,9 @@ class AdaoGenerator(PythonGenerator):
     self.add_variables()
     # Parametres optionnels
 
-    # Extraction du Study_repertory
-    if "__"+self.type_of_study+"__Study_repertory" in self.dictMCVal.keys():
-      self.text_da += "study_config['Repertory'] = '" + self.dictMCVal["__"+self.type_of_study+"__Study_repertory"] + "'\n"
-    # Extraction de AlgorithmParameters
-    if "__"+self.type_of_study+"__AlgorithmParameters__INPUT_TYPE" in self.dictMCVal.keys():
-      self.add_algorithm_parameters()
+    # 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()
@@ -146,7 +155,7 @@ class AdaoGenerator(PythonGenerator):
 
   def add_data(self, data_name):
 
-    # Extraction des données
+    # Extraction des donnees
     search_text = "__"+self.type_of_study+"__" + data_name + "__"
     data_type = self.dictMCVal[search_text + "INPUT_TYPE"]
     search_type = search_text + data_type + "__data__"
@@ -211,6 +220,10 @@ class AdaoGenerator(PythonGenerator):
       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"
@@ -230,19 +243,6 @@ class AdaoGenerator(PythonGenerator):
       self.text_da += data_name + "_config['Data'] = " + data_name + "_FunctionDict\n"
       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 
-  def add_algorithm_parameters(self):
-
-    data_name = "AlgorithmParameters"
-    data_type = "Dict"
-    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'] = '" + 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"
-
   def add_init(self):
 
       init_file_data = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__INIT_FILE"]
@@ -253,7 +253,7 @@ class AdaoGenerator(PythonGenerator):
       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"):
+      if isinstance(init_target_list, "str"):
         self.text_da +=  "'" + init_target_list + "',"
       else:
         for target in init_target_list:
@@ -277,9 +277,72 @@ class AdaoGenerator(PythonGenerator):
       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):
+
+    data_name = "AlgorithmParameters"
+    data_type = "Dict"
+    #
+    if "__"+self.type_of_study+"__AlgorithmParameters__Parameters" in self.dictMCVal:
+        para_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Parameters"]
+    else:
+        para_type = "Defaults"
+    #
+    if para_type == "Defaults":
+        from_type = para_type
+    elif para_type == "Dict":
+        from_type = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__FROM"]
+    else:
+        return
+
+    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'] = '" + 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]
+      if base in keys: 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 += '}'
+      if 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_variables(self):
 
     # Input variables