src/daEficas/generator_adao.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2010-2013 EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 # Author: André Ribes, andre.ribes@edf.fr, EDF R&D
  21
  22 from generator.generator_python import PythonGenerator
  23 import traceback
  24 import logging
  25
  26 def entryPoint():
  27    """
  28       Retourne les informations necessaires pour le chargeur de plugins
  29
  30       Ces informations sont retournees dans un dictionnaire
  31    """
  32    return {
  33         # Le nom du plugin
  34         'name' : 'adao',
  35         # La factory pour creer une instance du plugin
  36           'factory' : AdaoGenerator,
  37           }
  38
  39 class AdaoGenerator(PythonGenerator):
  40
  41   def __init__(self,cr=None):
  42     PythonGenerator.__init__(self, cr)
  43     self.dictMCVal={}
  44     self.text_comm = ""
  45     self.text_da = ""
  46     self.text_da_status = False
  47     self.logger = logging.getLogger('ADAO EFICAS GENERATOR')
  48     self.logger.setLevel(logging.INFO)
  49     ch = logging.StreamHandler()
  50     ch.setLevel(logging.INFO)
  51     formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")
  52     ch.setFormatter(formatter)
  53     self.logger.addHandler(ch)
  54
  55   def gener(self,obj,format='brut',config=None):
  56     self.logger.debug("method gener called")
  57     self.text_comm = PythonGenerator.gener(self, obj, format, config)
  58     for key, value in self.dictMCVal.iteritems():
  59       self.logger.debug("dictMCVAl %s %s" % (key,value))
  60
  61     try :
  62       self.text_da_status = False
  63       self.generate_da()
  64       self.text_da_status = True
  65     except:
  66       self.logger.info("Case is not correct, python command file for YACS schema generation cannot be created")
  67       self.logger.debug(self.text_da)
  68       self.dictMCVal = {}
  69       traceback.print_exc()
  70     return self.text_comm
  71
  72   def writeDefault(self, fn):
  73     if self.text_da_status:
  74       print "write adao python command file"
  75       filename = fn[:fn.rfind(".")] + '.py'
  76       f = open( str(filename), 'wb')
  77       f.write( self.text_da )
  78       f.close()
  79
  80   def generMCSIMP(self,obj) :
  81     """
  82     Convertit un objet MCSIMP en texte python
  83     """
  84     clef=""
  85     for i in obj.get_genealogie() :
  86       clef=clef+"__"+i
  87     self.dictMCVal[clef]=obj.valeur
  88
  89     s=PythonGenerator.generMCSIMP(self,obj)
  90     return s
  91
  92   def generate_da(self):
  93
  94     if "__CHECKING_STUDY__Study_name" in self.dictMCVal.keys():
  95       self.type_of_study = "CHECKING_STUDY"
  96     else:
  97       self.type_of_study = "ASSIMILATION_STUDY"
  98
  99     self.text_da += "#-*-coding:iso-8859-1-*- \n"
 100     self.text_da += "study_config = {} \n"
 101
 102     # Extraction de Study_type
 103     self.text_da += "study_config['StudyType'] = '" + self.type_of_study + "'\n"
 104     # Extraction de Study_name
 105     self.text_da += "study_config['Name'] = '" + self.dictMCVal["__"+self.type_of_study+"__Study_name"] + "'\n"
 106     # Extraction de Debug
 107     self.text_da += "study_config['Debug'] = '" + str(self.dictMCVal["__"+self.type_of_study+"__Debug"]) + "'\n"
 108     # Extraction de Algorithm
 109     self.text_da += "study_config['Algorithm'] = '" + self.dictMCVal["__"+self.type_of_study+"__Algorithm"] + "'\n"
 110
 111     if "__"+self.type_of_study+"__Background__INPUT_TYPE" in self.dictMCVal.keys():
 112       self.add_data("Background")
 113     if "__"+self.type_of_study+"__BackgroundError__INPUT_TYPE" in self.dictMCVal.keys():
 114       self.add_data("BackgroundError")
 115     if "__"+self.type_of_study+"__Observation__INPUT_TYPE" in self.dictMCVal.keys():
 116       self.add_data("Observation")
 117     if "__"+self.type_of_study+"__ObservationError__INPUT_TYPE" in self.dictMCVal.keys():
 118       self.add_data("ObservationError")
 119     if "__"+self.type_of_study+"__CheckingPoint__INPUT_TYPE" in self.dictMCVal.keys():
 120       self.add_data("CheckingPoint")
 121     if "__"+self.type_of_study+"__ObservationOperator__INPUT_TYPE" in self.dictMCVal.keys():
 122       self.add_data("ObservationOperator")
 123     if "__"+self.type_of_study+"__EvolutionModel__INPUT_TYPE" in self.dictMCVal.keys():
 124       self.add_data("EvolutionModel")
 125     if "__"+self.type_of_study+"__EvolutionError__INPUT_TYPE" in self.dictMCVal.keys():
 126       self.add_data("EvolutionError")
 127     if "__"+self.type_of_study+"__ControlInput__INPUT_TYPE" in self.dictMCVal.keys():
 128       self.add_data("ControlInput")
 129
 130     self.add_variables()
 131     # Parametres optionnels
 132
 133     # Extraction du Study_repertory
 134     if "__"+self.type_of_study+"__Study_repertory" in self.dictMCVal.keys():
 135       self.text_da += "study_config['Repertory'] = '" + self.dictMCVal["__"+self.type_of_study+"__Study_repertory"] + "'\n"
 136     # Extraction de AlgorithmParameters
 137     if "__"+self.type_of_study+"__AlgorithmParameters__INPUT_TYPE" in self.dictMCVal.keys():
 138       self.add_algorithm_parameters()
 139     # Extraction de UserPostAnalysis
 140     if "__"+self.type_of_study+"__UserPostAnalysis__FROM" in self.dictMCVal.keys():
 141       self.add_UserPostAnalysis()
 142     if "__"+self.type_of_study+"__UserDataInit__INIT_FILE" in self.dictMCVal.keys():
 143       self.add_init()
 144     if "__"+self.type_of_study+"__Observers__SELECTION" in self.dictMCVal.keys():
 145       self.add_observers()
 146
 147   def add_data(self, data_name):
 148
 149     # Extraction des données
 150     search_text = "__"+self.type_of_study+"__" + data_name + "__"
 151     data_type = self.dictMCVal[search_text + "INPUT_TYPE"]
 152     search_type = search_text + data_type + "__data__"
 153     from_type = self.dictMCVal[search_type + "FROM"]
 154     data = ""
 155     if from_type == "String":
 156       data = self.dictMCVal[search_type + "STRING_DATA__STRING"]
 157     elif from_type == "Script":
 158       data = self.dictMCVal[search_type + "SCRIPT_DATA__SCRIPT_FILE"]
 159     elif from_type == "ScriptWithSwitch":
 160       data = self.dictMCVal[search_type + "SCRIPTWITHSWITCH_DATA__SCRIPTWITHSWITCH_FILE"]
 161     elif from_type == "ScriptWithFunctions":
 162       data = self.dictMCVal[search_type + "SCRIPTWITHFUNCTIONS_DATA__SCRIPTWITHFUNCTIONS_FILE"]
 163     elif from_type == "ScriptWithOneFunction":
 164       data = self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__SCRIPTWITHONEFUNCTION_FILE"]
 165     elif from_type == "FunctionDict":
 166       data = self.dictMCVal[search_type + "FUNCTIONDICT_DATA__FUNCTIONDICT_FILE"]
 167     else:
 168       raise Exception('From Type unknown', from_type)
 169
 170     if from_type == "String" or from_type == "Script":
 171       self.text_da += data_name + "_config = {}\n"
 172       self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
 173       self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
 174       self.text_da += data_name + "_config['Data'] = '" + data      + "'\n"
 175       if search_text+"Stored" in self.dictMCVal.keys():
 176         self.text_da += data_name + "_config['Stored'] = '" +  str(self.dictMCVal[search_text+"Stored"])  + "'\n"
 177       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 178
 179     if from_type == "ScriptWithSwitch":
 180       self.text_da += data_name + "_ScriptWithSwitch = {}\n"
 181       self.text_da += data_name + "_ScriptWithSwitch['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
 182       self.text_da += data_name + "_ScriptWithSwitch['Script'] = {}\n"
 183       self.text_da += data_name + "_ScriptWithSwitch['Script']['Direct'] = '"  + data + "'\n"
 184       self.text_da += data_name + "_ScriptWithSwitch['Script']['Tangent'] = '" + data + "'\n"
 185       self.text_da += data_name + "_ScriptWithSwitch['Script']['Adjoint'] = '" + data + "'\n"
 186       self.text_da += data_name + "_config = {}\n"
 187       self.text_da += data_name + "_config['Type'] = 'Function'\n"
 188       self.text_da += data_name + "_config['From'] = 'ScriptWithSwitch'\n"
 189       self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithSwitch\n"
 190       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 191
 192     if from_type == "ScriptWithFunctions":
 193       self.text_da += data_name + "_ScriptWithFunctions = {}\n"
 194       self.text_da += data_name + "_ScriptWithFunctions['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
 195       self.text_da += data_name + "_ScriptWithFunctions['Script'] = {}\n"
 196       self.text_da += data_name + "_ScriptWithFunctions['Script']['Direct'] = '"  + data + "'\n"
 197       self.text_da += data_name + "_ScriptWithFunctions['Script']['Tangent'] = '" + data + "'\n"
 198       self.text_da += data_name + "_ScriptWithFunctions['Script']['Adjoint'] = '" + data + "'\n"
 199       self.text_da += data_name + "_config = {}\n"
 200       self.text_da += data_name + "_config['Type'] = 'Function'\n"
 201       self.text_da += data_name + "_config['From'] = 'ScriptWithFunctions'\n"
 202       self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithFunctions\n"
 203       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 204
 205     if from_type == "ScriptWithOneFunction":
 206       self.text_da += data_name + "_ScriptWithOneFunction = {}\n"
 207       self.text_da += data_name + "_ScriptWithOneFunction['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
 208       self.text_da += data_name + "_ScriptWithOneFunction['Script'] = {}\n"
 209       self.text_da += data_name + "_ScriptWithOneFunction['Script']['Direct'] = '"  + data + "'\n"
 210       self.text_da += data_name + "_ScriptWithOneFunction['Script']['Tangent'] = '" + data + "'\n"
 211       self.text_da += data_name + "_ScriptWithOneFunction['Script']['Adjoint'] = '" + data + "'\n"
 212       self.text_da += data_name + "_ScriptWithOneFunction['DifferentialIncrement'] = " + str(float(self.dictMCVal[search_type + "SCRIPTWITHONEFUNCTION_DATA__DifferentialIncrement"])) + "\n"
 213       self.text_da += data_name + "_config = {}\n"
 214       self.text_da += data_name + "_config['Type'] = 'Function'\n"
 215       self.text_da += data_name + "_config['From'] = 'ScriptWithOneFunction'\n"
 216       self.text_da += data_name + "_config['Data'] = " + data_name + "_ScriptWithOneFunction\n"
 217       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 218
 219     if from_type == "FunctionDict":
 220       self.text_da += data_name + "_FunctionDict = {}\n"
 221       self.text_da += data_name + "_FunctionDict['Function'] = ['Direct', 'Tangent', 'Adjoint']\n"
 222       self.text_da += data_name + "_FunctionDict['Script'] = {}\n"
 223       self.text_da += data_name + "_FunctionDict['Script']['Direct'] = '"  + data + "'\n"
 224       self.text_da += data_name + "_FunctionDict['Script']['Tangent'] = '" + data + "'\n"
 225       self.text_da += data_name + "_FunctionDict['Script']['Adjoint'] = '" + data + "'\n"
 226       self.text_da += data_name + "_config = {}\n"
 227       self.text_da += data_name + "_config['Type'] = 'Function'\n"
 228       self.text_da += data_name + "_config['From'] = 'FunctionDict'\n"
 229       self.text_da += data_name + "_config['Data'] = " + data_name + "_FunctionDict\n"
 230       self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 231
 232   def add_algorithm_parameters(self):
 233
 234     data_name = "AlgorithmParameters"
 235     data_type = "Dict"
 236     from_type = "Script"
 237     data = self.dictMCVal["__"+self.type_of_study+"__AlgorithmParameters__Dict__data__SCRIPT_DATA__SCRIPT_FILE"]
 238
 239     self.text_da += data_name + "_config = {} \n"
 240     self.text_da += data_name + "_config['Type'] = '" + data_type + "'\n"
 241     self.text_da += data_name + "_config['From'] = '" + from_type + "'\n"
 242     self.text_da += data_name + "_config['Data'] = '" + data + "'\n"
 243     self.text_da += "study_config['" + data_name + "'] = " + data_name + "_config\n"
 244
 245   def add_init(self):
 246
 247       init_file_data = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__INIT_FILE"]
 248       init_target_list = self.dictMCVal["__"+self.type_of_study+"__UserDataInit__TARGET_LIST"]
 249
 250       self.text_da += "Init_config = {}\n"
 251       self.text_da += "Init_config['Type'] = 'Dict'\n"
 252       self.text_da += "Init_config['From'] = 'Script'\n"
 253       self.text_da += "Init_config['Data'] = '" + init_file_data + "'\n"
 254       self.text_da += "Init_config['Target'] = ["
 255       if type(init_target_list) is type("str"):
 256         self.text_da +=  "'" + init_target_list + "',"
 257       else:
 258         for target in init_target_list:
 259           self.text_da += "'" + target + "',"
 260       self.text_da += "]\n"
 261       self.text_da += "study_config['UserDataInit'] = Init_config\n"
 262
 263   def add_UserPostAnalysis(self):
 264
 265     from_type = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__FROM"]
 266     data = ""
 267     if from_type == "String":
 268       data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__STRING_DATA__STRING"]
 269       self.text_da += "Analysis_config = {}\n"
 270       self.text_da += "Analysis_config['From'] = 'String'\n"
 271       self.text_da += "Analysis_config['Data'] = \"\"\"" + data + "\"\"\"\n"
 272       self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n"
 273     elif from_type == "Script":
 274       data = self.dictMCVal["__"+self.type_of_study+"__UserPostAnalysis__SCRIPT_DATA__SCRIPT_FILE"]
 275       self.text_da += "Analysis_config = {}\n"
 276       self.text_da += "Analysis_config['From'] = 'Script'\n"
 277       self.text_da += "Analysis_config['Data'] = '" + data + "'\n"
 278       self.text_da += "study_config['UserPostAnalysis'] = Analysis_config\n"
 279     else:
 280       raise Exception('From Type unknown', from_type)
 281
 282   def add_variables(self):
 283
 284     # Input variables
 285     if "__"+self.type_of_study+"__InputVariables__NAMES" in self.dictMCVal.keys():
 286       names = []
 287       sizes = []
 288       if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"], type("")):
 289         names.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"])
 290       else:
 291         names = self.dictMCVal["__"+self.type_of_study+"__InputVariables__NAMES"]
 292       if isinstance(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"], type(1)):
 293         sizes.append(self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"])
 294       else:
 295         sizes = self.dictMCVal["__"+self.type_of_study+"__InputVariables__SIZES"]
 296
 297       self.text_da += "inputvariables_config = {}\n"
 298       self.text_da += "inputvariables_config['Order'] = %s\n" % list(names)
 299       for name, size in zip(names, sizes):
 300         self.text_da += "inputvariables_config['%s'] = %s\n" % (name,size)
 301       self.text_da += "study_config['InputVariables'] = inputvariables_config\n"
 302     else:
 303       self.text_da += "inputvariables_config = {}\n"
 304       self.text_da += "inputvariables_config['Order'] =['adao_default']\n"
 305       self.text_da += "inputvariables_config['adao_default'] = -1\n"
 306       self.text_da += "study_config['InputVariables'] = inputvariables_config\n"
 307
 308     # Output variables
 309     if "__"+self.type_of_study+"__OutputVariables__NAMES" in self.dictMCVal.keys():
 310       names = []
 311       sizes = []
 312       if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"], type("")):
 313         names.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"])
 314       else:
 315         names = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__NAMES"]
 316       if isinstance(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"], type(1)):
 317         sizes.append(self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"])
 318       else:
 319         sizes = self.dictMCVal["__"+self.type_of_study+"__OutputVariables__SIZES"]
 320
 321       self.text_da += "outputvariables_config = {}\n"
 322       self.text_da += "outputvariables_config['Order'] = %s\n" % list(names)
 323       for name, size in zip(names, sizes):
 324         self.text_da += "outputvariables_config['%s'] = %s\n" % (name,size)
 325       self.text_da += "study_config['OutputVariables'] = outputvariables_config\n"
 326     else:
 327       self.text_da += "outputvariables_config = {}\n"
 328       self.text_da += "outputvariables_config['Order'] = ['adao_default']\n"
 329       self.text_da += "outputvariables_config['adao_default'] = -1\n"
 330       self.text_da += "study_config['OutputVariables'] = outputvariables_config\n"
 331
 332   def add_observers(self):
 333     observers = {}
 334     observer = self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"]
 335     if isinstance(observer, type("")):
 336       self.add_observer_in_dict(observer, observers)
 337     else:
 338       for observer in self.dictMCVal["__"+self.type_of_study+"__Observers__SELECTION"]:
 339         self.add_observer_in_dict(observer, observers)
 340
 341     # Write observers in the python command file
 342     number = 2
 343     self.text_da += "observers = {}\n"
 344     for observer in observers.keys():
 345       number += 1
 346       self.text_da += "observers[\"" + observer + "\"] = {}\n"
 347       self.text_da += "observers[\"" + observer + "\"][\"number\"] = " + str(number) + "\n"
 348       self.text_da += "observers[\"" + observer + "\"][\"nodetype\"] = \"" + observers[observer]["nodetype"] + "\"\n"
 349       if observers[observer]["nodetype"] == "String":
 350         self.text_da += "observers[\"" + observer + "\"][\"String\"] = \"\"\"" + observers[observer]["script"] + "\"\"\"\n"
 351       else:
 352         self.text_da += "observers[\"" + observer + "\"][\"Script\"] = \"" + observers[observer]["file"] + "\"\n"
 353       if "scheduler" in observers[observer].keys():
 354         self.text_da += "observers[\"" + observer + "\"][\"scheduler\"] = \"\"\"" + observers[observer]["scheduler"] + "\"\"\"\n"
 355       if "info" in observers[observer].keys():
 356         self.text_da += "observers[\"" + observer + "\"][\"info\"] = \"\"\"" + observers[observer]["info"] + "\"\"\"\n"
 357     self.text_da += "study_config['Observers'] = observers\n"
 358
 359   def add_observer_in_dict(self, observer, observers):
 360     """
 361       Add observer in the observers dict.
 362     """
 363     observers[observer] = {}
 364     observers[observer]["name"] = observer
 365     observer_eficas_name = "__"+self.type_of_study+"__Observers__" + observer + "__" + observer + "_data__"
 366     # NodeType
 367     node_type_key_name = observer_eficas_name + "NodeType"
 368     observers[observer]["nodetype"] = self.dictMCVal[node_type_key_name]
 369
 370     # NodeType script/file
 371     if observers[observer]["nodetype"] == "String":
 372       observers[observer]["script"] = self.dictMCVal[observer_eficas_name + "PythonScript__Value"]
 373     else:
 374       observers[observer]["file"] = self.dictMCVal[observer_eficas_name + "UserFile__Value"]
 375
 376     # Scheduler
 377     scheduler_key_name = observer_eficas_name + "Scheduler"
 378     if scheduler_key_name in self.dictMCVal.keys():
 379       observers[observer]["scheduler"] = self.dictMCVal[scheduler_key_name]
 380
 381     # Info
 382     info_key_name = observer_eficas_name + "Info"
 383     if info_key_name in self.dictMCVal.keys():
 384       observers[observer]["info"] = self.dictMCVal[info_key_name]