generator/generator_mapVP.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2007-2021   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 """
  21    Ce module contient le plugin generateur de fichier au format
  22    CARMEL3D pour EFICAS.
  23
  24 """
  25 from __future__ import print_function
  26 from __future__ import absolute_import
  27 try :
  28     from builtins import str
  29 except : pass
  30
  31 import traceback
  32 import types,re,os
  33 import Accas
  34
  35 from .generator_python import PythonGenerator
  36
  37 listeCalParName = ('Time' , 'Temperature', 'DoseRate', 'Thickness')        #'calculation_parameter_names'
  38
  39
  40 def entryPoint():
  41     """
  42        Retourne les informations necessaires pour le chargeur de plugins
  43
  44        Ces informations sont retournees dans un dictionnaire
  45     """
  46     return {
  47          # Le nom du plugin
  48            'name' : 'MAPVp',
  49          # La factory pour creer une instance du plugin
  50            'factory' : MapGenerator,
  51            }
  52
  53
  54 class MapGenerator(PythonGenerator):
  55     """
  56        Ce generateur parcourt un objet de type JDC et produit
  57        un texte au format eficas et
  58        un texte au format py
  59
  60     """
  61
  62     def gener(self,obj,format='brut',config=None,appliEficas=None):
  63         self.appliEficas=appliEficas
  64         self.cata=self.appliEficas.readercata.cata
  65         self.initDico()
  66         self.text=PythonGenerator.gener(self,obj,format)
  67         if obj.isValid() :self.genereTexte(obj)
  68         return self.text
  69
  70     def initDico(self) :
  71         self.texteInput = ""
  72         self.dictParam={}
  73         self.dictValeur={}
  74         self.listeEquations=[]
  75         self.typeEtude = ""
  76
  77
  78     def genereTexte(self,obj) :
  79         print ('genereTexte', self.typeEtude)
  80         if self.typeEtude == "Calculation" : self.genereCalculation()
  81
  82
  83     def generPROC_ETAPE(self,obj):
  84         s=PythonGenerator.generPROC_ETAPE(self,obj)
  85         if obj.nom == "Calculation_for_Mechanistic" : print ('hjkhjkh');self.typeEtude="Calculation"
  86         return s
  87
  88     def genereCalculation(self) :
  89         '''
  90         Prepare le contenu du fichier de parametres python
  91         '''
  92         self.texteInput = ""
  93         self.texteInput += self.genereCsv()
  94         self.texteInput += self.genereCalculationParams()
  95         self.texteInput += self.txtNomCst
  96         self.texteInput += self.txtVal
  97         self.texteInput += self.txtValAct
  98         self.texteInput += self.txtNomCstNA
  99         self.texteInput += self.txtValNA
 100         self.texteInput += self.txtInitName
 101         self.texteInput += self.txtInitVal
 102         self.texteInput += self.genereEquations()
 103         print (self.texteInput)
 104
 105     def writeDefault(self, fn):
 106         # normalement c_solver_polymer_kinetics_myStudy.input ou myStudy est le nom de l etude
 107         fileInput = fn[:fn.rfind(".")] + '.input'
 108         f = open( str(fileInput), 'wb')
 109         f.write( self.texteInput )
 110         f.close()
 111
 112
 113     def genereCalculationParams(self) :
 114         txtNom  = "calculation_parameter_names = [ "
 115         txtVal = "calculation_parameters = [ "
 116         for param in ('Time' , 'Temperature', 'DoseRate', 'Thickness')  :
 117             if param in self.dictValeur.keys() :
 118                 txtNom  += "'"+param +  "', "
 119                 txtVal += str(self.dictValeur[param]) + ", "
 120         # on enleve les dernieres , et on ferme
 121         txtNom = txtNom[0:-2]
 122         txtNom += "]\n"
 123         txtVal = txtVal[0:-2]
 124         txtVal += "]\n"
 125         txt = txtNom + txtVal
 126         return txt
 127
 128     def genereEquations(self) :
 129         txt="equation =["
 130         index=0
 131         TechnicalUse = self.dictValeur['TechnicalUse']
 132         ModelName    = self.dictValeur['ModelName']
 133         for param in  self.listInitialParameters:
 134             print ('*******************************************')
 135             print (' je  traite ', param , 'index : ', index)
 136             trouveParam=False
 137
 138             if index != 0 : txtParam = 'Dy[j*5 + '+str(index)+ '] = '
 139             else :          txtParam = 'Dy[j*5] = '
 140
 141             for equation in  self.listeEquations :
 142                 if param in (self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation].keys()):
 143                     print ('____________ trouve : ', param , 'in ', equation, ' ',  self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation][param])
 144                     trouveParam = True
 145                     if self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation][param][0] == '-' :
 146                         txtParam += ' ' + self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation][param]
 147                     else :
 148                         if index != 0 :
 149                             txtParam += ' + ' + self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation][param]
 150                         else :
 151                             txtParam +=  self.cata.dicoEquations[TechnicalUse][ModelName]['equa_diff'][equation][param]
 152                     print ('         txtParam   intermediaire ', txtParam)
 153
 154             if trouveParam :
 155                 txtParam = txtParam + ", "
 156                 txt += txtParam
 157                 index = index+1
 158             print (txtParam)
 159             print ('fin param', param, 'trouve ', trouveParam, '___________________________')
 160             print ('*************************************************')
 161             print (' ')
 162         print ('_________________fin for')
 163         txt=txt[0:-3]
 164         txt+="]\n"
 165         return txt
 166
 167     def genereCsv(self) :
 168         txt =  'study_name = ' +self.dictValeur['SimulationName'] +  "\n"
 169         txt += 'csv_output_file_name = ' + self.dictValeur['OutPutFolder'] + '/c_solver_stiff_ode_1d_' + self.dictValeur['SimulationName']+ '.csv\n'
 170         return txt
 171
 172
 173     def generMCList(self,obj):
 174         s=PythonGenerator.generMCList(self,obj)
 175         if obj.nom == 'ConstantesArrhenius' :
 176             self.txtNomCst  = "Arrhenius_Name = [ "
 177             self.txtVal     = "Arrhenius_A = [ "
 178             self.txtValAct  = "Arrhenius_Ea = [ "
 179             for objFils in obj.data:
 180                 for mc in objFils.mcListe :
 181                     self.txtNomCst +=  "'" + mc.nom + "', "
 182                     self.txtVal    +=  str(mc.valeur[0]) + ", "
 183                     self.txtValAct +=  str(mc.valeur[1]) + ", "
 184             self.txtNomCst = self.txtNomCst[0:-2]
 185             self.txtVal    = self.txtVal[0:-2]
 186             self.txtValAct = self.txtValAct [0:-2]
 187             self.txtNomCst += ']\n'
 188             self.txtVal    += ']\n'
 189             self.txtValAct += ']\n'
 190
 191         if obj.nom == 'ConstantesNonArrhenius' :
 192             self.txtNomCstNA  = "non_Arrhenius_coefs_names = [ "
 193             self.txtValNA     = "non_Arrhenius_coefs = [ "
 194             for objFils in obj.data:
 195                 for mc in objFils.mcListe :
 196                     self.txtNomCstNA +=  "'" + mc.nom + "', "
 197                     self.txtValNA    +=  str(mc.valeur) + ", "
 198             self.txtNomCstNA  = self.txtNomCstNA[0:-2]
 199             self.txtValNA     = self.txtValNA[0:-2]
 200             self.txtNomCstNA += ']\n'
 201             self.txtValNA    += ']\n'
 202
 203         if obj.nom == 'InitialParameters' :
 204             self.listInitialParameters =[]
 205             self.txtInitName  = "initial_Value_names = [ "
 206             self.txtInitVal   = "initial_Values = [ "
 207             for objFils in obj.data:
 208                 for mc in objFils.mcListe :
 209                     self.txtInitName +=  "'" + mc.nom + "', "
 210                     self.txtInitVal  +=  str(mc.valeur) + ", "
 211                     self.listInitialParameters.append(mc.nom)
 212             self.txtInitName  = self.txtInitName[0:-2]
 213             self.txtInitVal   = self.txtInitVal[0:-2]
 214             self.txtInitName += ']\n'
 215             self.txtInitVal  += ']\n'
 216
 217         if obj.nom in( 'initiation','propagation','termination','stabilization') :
 218             for o in obj :
 219                 for mc  in o.mcListe :
 220                     nom=mc.nom.replace (' ','').replace ('+','_').replace ('-','_').replace ('>','_').replace('(','').replace(')','').replace('*','').replace('[','').replace(']','')
 221                     nom=obj.nom+'_'+nom
 222                     self.listeEquations.append(nom )
 223         return s
 224
 225     def generMCSIMP(self,obj) :
 226         """
 227         Convertit un objet MCSIMP en texte python
 228         Remplit le dictionnaire des MCSIMP
 229         """
 230         s=PythonGenerator.generMCSIMP(self,obj)
 231         if obj.nom=='Consigne' : return s
 232
 233         if obj.getGenealogie()[0][-6:-1]=="_PARA":
 234             self.dictParam[obj.nom]=obj.valeur
 235         else :
 236             self.dictValeur[obj.nom]=obj.valeur
 237         return s