convert/convert_TELEMAC.py

   1 # Copyright (C) 2007-2013   EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19 import re, string
  20 from Extensions.i18n import tr
  21
  22
  23 from convert_python import PythonParser
  24
  25 pattern_comment_slash   = re.compile(r"^\s*/")
  26 pattern_eta   = re.compile(r".*&ETA.*")
  27 pattern_fin   = re.compile(r".*&FIN.*")
  28 pattern_oui   = re.compile(r"^\s*(oui|OUI|YES|yes|TRUE|VRAI)\s*$")
  29 pattern_non   = re.compile(r"^\s*(non|NON|NO|no|FALSE|FAUX)\s*$")
  30 pattern_blanc = re.compile(r"^\s*$")
  31 pattern_listeVide = re.compile(r"^\s*'\s*'\s*$")
  32 pattern_commence_par_quote=re.compile(r'^\s*[\'"].*')
  33 pattern_finit_par_virgule_ou_affect=re.compile(r'^.*(,|;|=|:)\s*$')
  34
  35 pattern_ligne=re.compile(r'^\s*(?P<ident>[^=:]*)\s*[:=]\s*(?P<reste>.*)$')
  36
  37 pattern_variables=re.compile (r"^\s*(?P<ident>VARIABLES FOR GRAPHIC PRINTOUTS|VARIABLES POUR LES SORTIES GRAPHIQUES)\s*[:=]\s*(?P<valeur>[A-Za-z]+(\d*|\*)(,[A-Za-z]+(\d*|\*))*)\s*(?P<reste>.*)$")
  38
  39 # Attention aux listes de flottants
  40 pattern_liste=re.compile(r'^\s*(?P<valeur>[+-.\w]+(\s*;\s*[+-.\w]+)+)\s*(?P<reste>.*)$')
  41 pattern_liste_texte=re.compile(r"^\s*(?P<valeur>('.*(';\s*))+('.*'\s*)?)(?P<reste>.*)$")
  42 pattern_flottant=re.compile(r'^\s*(?P<valeur>[+-]?((\d+(\.\d*)?)|(\.\d+))([dDeE][+-]?\d+)?)\s*(?P<reste>.*)$')
  43 pattern_texteQuote  = re.compile (r"^\s*(?P<valeur>'[^']+(''[^']+)*')\s*(?P<reste>.*)$")
  44 pattern_texteSimple = re.compile (r"(?P<valeur>(^|\s)\s*[\w\.-]+)\s*(?P<reste>.*)$")
  45 pattern_texteVide   = re.compile (r"^\s*(?P<valeur>'')\s*(?P<reste>.*)$")
  46
  47 pattern_ContientDouble=re.compile (r"^.*''.*$")
  48
  49 # le pattern texte reconnait
  50 #nom1 nom 2 : ou = chaine entre '
  51 # avec eventuellement  des quotes au milieu par exemple
  52 # TITRE = 'TELEMAC 2D : GOUTTE D''EAU DANS UN BASSIN$'
  53 # m.group("texte") va rendre 'TELEMAC 2D : GOUTTE D''EAU DANS UN BASSIN$'
  54
  55
  56 #Si le code n est pas Telemac
  57 try :
  58 #if 1 :
  59    from aideAuxConvertisseurs import ListeSupprimeCasToEficas
  60    from enumDicoTelemac       import TelemacdicoEn
  61 except :
  62    pass
  63
  64 from Extensions import localisation
  65
  66 from determine import monEnvQT5
  67
  68
  69
  70 def entryPoint():
  71    """
  72    Return a dictionary containing the description needed to load the plugin
  73    """
  74    return {
  75           'name' : 'TELEMAC3',
  76           'factory' : TELEMACParser
  77           }
  78
  79 class TELEMACParser(PythonParser):
  80    """
  81    This converter works like PythonParser, except that it also initializes all
  82    model variables to None in order to avoid Python syntax errors when loading
  83    a file with a different or inexistent definition of variables.
  84    """
  85
  86    def convert(self, outformat, appli=None):
  87
  88
  89       from Accas import A_BLOC, A_FACT, A_SIMP
  90       self.dicoCasToCata=appli.readercata.dicoCasToCata
  91       self.dicoInverse=appli.readercata.dicoInverse
  92       self.dicoMC=appli.readercata.dicoMC
  93       self.Ordre_Des_Commandes=appli.readercata.Ordre_Des_Commandes
  94
  95       if appli.langue=='fr' :
  96           from enumDicoTelemac       import DicoEnumCasFrToEnumCasEn
  97           for k in DicoEnumCasFrToEnumCasEn.keys() :
  98               TelemacdicoEn[k]=DicoEnumCasFrToEnumCasEn[k]
  99
 100       text=""
 101       self.dictSimp={}
 102
 103       l_lignes_texte_all = string.split(self.text,'\n')
 104       l_lignes_texte = []
 105       for l  in l_lignes_texte_all :
 106         if not(pattern_comment_slash.match(l)): l_lignes_texte.append(l)
 107
 108       l_lignes=[]
 109       i=0
 110       while (i < len(l_lignes_texte)) :
 111           ligne=l_lignes_texte[i]
 112           i=i+1
 113           if not(pattern_finit_par_virgule_ou_affect.match(ligne)):
 114              l_lignes.append(ligne)
 115              continue
 116           nouvelle_ligne=ligne
 117           while (i < len(l_lignes_texte)):
 118              ligne_traitee=l_lignes_texte[i]
 119              i=i+1
 120              nouvelle_ligne += ligne_traitee
 121              if not(pattern_finit_par_virgule_ou_affect.match(ligne_traitee)):
 122                 l_lignes.append(nouvelle_ligne)
 123                 break
 124
 125
 126       for ligne in l_lignes :
 127           if pattern_comment_slash.match(ligne) : continue
 128           if pattern_eta.match(ligne) : continue
 129           if pattern_fin.match(ligne) : continue
 130           if pattern_blanc.match(ligne) : continue
 131
 132
 133           finLigne=ligne
 134           while finLigne != "" :
 135               if pattern_comment_slash.match(finLigne) : finLigne=""; continue
 136               valeur=""
 137               if pattern_variables.match(finLigne) :
 138                  m=pattern_variables.match(finLigne)
 139                  simpCas=self.traiteIdent(m.group('ident'))
 140                  valeur=m.group('valeur')
 141                  finLigne=m.group('reste')
 142                  self.dictSimp[simpCas]=valeur
 143                  continue
 144
 145
 146               m=pattern_ligne.match(finLigne)
 147               if m == None :
 148                  #print( "________________________________________________")
 149                  print ('pb avec ****', finLigne , '**** dans ', ligne)
 150                  #print( "________________________________________________")
 151                  break
 152
 153               simpCas=self.traiteIdent(m.group('ident'))
 154               if not simpCas :
 155                  finLigne=m.group('reste')
 156                  continue
 157
 158               finLigne=m.group('reste')
 159               # attention, l ordre des if est important
 160               if pattern_liste.match(finLigne) :
 161                  m=pattern_liste.match(finLigne)
 162               elif pattern_liste_texte.match(finLigne) :
 163                  m=pattern_liste_texte.match(finLigne)
 164               elif pattern_texteQuote.match(finLigne) :
 165                  m=pattern_texteQuote.match(finLigne)
 166               elif pattern_flottant.match(finLigne) :
 167                  m=pattern_flottant.match(finLigne)
 168               elif pattern_texteVide.match(finLigne):
 169                  m=pattern_texteVide.match(finLigne)
 170               elif pattern_texteSimple.match(finLigne):
 171                  m=pattern_texteSimple.match(finLigne)
 172               else :
 173                  #print ("________________________________________________")
 174                  print ('pb avec ****', finLigne , '**** dans ', ligne)
 175                  print ("non match")
 176                  #print ("________________________________________________")
 177                  break
 178
 179
 180               valeur=m.group('valeur')
 181               if pattern_blanc.match(valeur) : valeur=None
 182
 183               if pattern_flottant.match(finLigne) :
 184                  valeur=re.sub("d","e",valeur)
 185                  valeur=re.sub("D","E",valeur)
 186
 187               if pattern_liste.match(finLigne) or pattern_liste_texte.match(finLigne):
 188                  valeur=valeur.split(";")
 189
 190
 191               finLigne=m.group('reste')
 192               self.dictSimp[simpCas]=valeur
 193
 194       if 'TITLE' not in self.dictSimp.keys() :
 195           import os
 196           #self.dictSimp['TITLE']=os.path.basename(self.filename)
 197
 198       dicoParMC={}
 199       for simp in self.dictSimp.keys():
 200           if simp in TELEMACParser.__dict__.keys() : apply(TELEMACParser.__dict__[simp],(self,))
 201
 202       for simp in self.dictSimp.keys():
 203           if simp in ListeSupprimeCasToEficas: continue
 204           if simp not in self.dicoInverse.keys() :
 205              #print ( "************")
 206              print  ("pb avec dans dicoInverse", simp,'------')
 207              #print  ("************")
 208              continue
 209           listeGenea=self.dicoInverse[simp]
 210           listeGeneaReverse=[]
 211           for (u,v) in listeGenea :
 212               if isinstance(v,A_BLOC.BLOC): continue
 213               listeGeneaReverse.append(u)
 214           listeGeneaReverse.reverse()
 215           dicoTravail=dicoParMC
 216           i=0
 217           while i < len(listeGeneaReverse[0:-1]) :
 218             mot=listeGeneaReverse[i]
 219             i=i+1
 220             if mot not in dicoTravail.keys(): dicoTravail[mot]={}
 221             dicoTravail=dicoTravail[mot]
 222           dicoTravail[simp]=self.dictSimp[simp]
 223
 224       self.textePy=""
 225       listeMC=self.tri(dicoParMC.keys())
 226       for k in listeMC :
 227           self.textePy += str(k )+ "("
 228           self.traiteMC(dicoParMC[k])
 229           self.textePy += ");\n"
 230
 231
 232       appli.listeTelemac=self.dictSimp.keys()
 233       return self.textePy
 234
 235
 236    #----------------------------------------
 237    def traiteIdent(self,ident):
 238    # enleve les espaces de part et autre
 239    # traduit du langage Telemac vers le langage Catalogue
 240    #----------------------------------------
 241           while ident[-1] == " " or ident[-1] == '\t' : ident=ident[0:-1]
 242           while ident[0]  == " " or ident[0]  == '\t' : ident=ident[1:]
 243           try : identCata=self.dicoCasToCata[ident]
 244           except :
 245             print ( "---> ", "pb mot clef  pour", ident)
 246             identCata=None
 247           return identCata
 248
 249
 250    def traiteMC(self,dico) :
 251        from Accas import A_BLOC, A_FACT, A_SIMP
 252        for k in dico.keys() :
 253            valeur= dico[k]
 254            if k not in self.dicoMC.keys() : kA=self.dicoFrancaisAnglais[k]
 255            else : kA=k
 256            obj=self.dicoMC[kA]
 257            if isinstance(obj,A_FACT.FACT):   self.convertFACT(obj,kA,valeur)
 258            elif isinstance(obj,A_BLOC.BLOC): self.convertBLOC(obj,kA,valeur)
 259            elif isinstance(obj,A_SIMP.SIMP): self.convertSIMP(obj,kA,valeur)
 260            else : print ("%%%%%%%%%%%\n", "pb conversion type pour", k, obj, "\n%%%%%%%%%%%")
 261
 262
 263    def convertFACT(self,obj,nom,valeur):
 264        if nom in TELEMACParser.__dict__.keys() :
 265           apply(TELEMACParser.__dict__[nom],(self,))
 266           return
 267        self.textePy +=  nom + "=_F( "
 268        self.traiteMC(valeur)
 269        self.textePy += '),\n'
 270
 271
 272    def convertBLOC(self,obj,nom,valeur):
 273        print ("ANOMALIE _________ BLOC ")
 274        print (nom)
 275
 276    def convertSIMP(self,obj,nom,valeur):
 277        #print 'in convertSIMP', nom,valeur
 278        if nom in ("PRESCRIBED_FLOWRATES", "PRESCRIBED_VELOCITIES", "PRESCRIBED_ELEVATIONS" ): return
 279        if obj.max==1 :
 280           if hasattr(obj.type[0],'ntuple') :
 281              lval=[]
 282              for v in valeur :
 283                try :    v=eval(v,{})
 284                except : pass
 285                lval.append(v)
 286              self.textePy += nom + "=" + str(lval) +","
 287              return
 288           if 'TXM' in obj.type :
 289
 290               if pattern_ContientDouble.match(str(valeur)):
 291                  valeur=re.sub("''","\'\'",str(valeur))
 292                  self.textePy += nom + "=" + str(valeur) +","
 293                  return
 294               valeur=str(valeur)
 295
 296               # ceinture et bretelle si les re sont correctes -)
 297               while valeur[-1] == " " or valeur[-1] == '\t' : valeur=valeur[0:-1]
 298               while valeur[0]  == " " or valeur[0]  == '\t' : valeur=valeur[1:]
 299
 300
 301
 302           # Pour les enum
 303           try    : valeur=eval(valeur,{})
 304           except : pass
 305
 306           if nom in TelemacdicoEn.keys():
 307              try    :
 308                valeur=TelemacdicoEn[nom][valeur]
 309                self.textePy += nom + "= '" + str(valeur) +"',"
 310                return
 311              except : pass
 312
 313
 314           if obj.into != [] and obj.into != None and not('R' in obj.type) and not('I' in obj.type):
 315              for possible in obj.into :
 316                 try :
 317                   if possible.upper() == valeur.upper():
 318                      valeur=possible
 319                      break
 320                   v=valeur[0].upper()+valeur[1:].lower()
 321                   v2=tr(v)
 322                   if possible.upper() == v2.upper():
 323                      valeur=possible
 324                      break
 325                 except:
 326                    if valeur != None :
 327                       print ("pb avec le type de ", obj.nom, obj.type, 'et la valeur ', valeur)
 328
 329           if 'Fichier' in obj.type or 'TXM' in obj.type or 'Repertoire' in obj.type :
 330               valeur=str(valeur)
 331               if valeur == "" or valeur == " " :
 332                  self.textePy += nom + "= '" + str(valeur) +"' ,"
 333                  return
 334               while valeur[-1] == " " : valeur=valeur[0:-1]
 335               while valeur[0]  == " " : valeur=valeur[1:]
 336               self.textePy += nom + "= '" + str(valeur) +"' ,"
 337               return
 338
 339           if bool in obj.type :
 340             if   valeur == True  :  self.textePy += nom + "= True,"
 341             elif valeur == False :  self.textePy += nom + "= False,"
 342             elif pattern_oui.match(valeur) : self.textePy += nom + "= True,"
 343             elif pattern_non.match(valeur) : self.textePy += nom + "= False,"
 344             else :  self.textePy += nom + "= None,"
 345             return
 346           self.textePy += nom + "=" + str(valeur) +","
 347
 348        else :
 349           if valeur == () or valeur ==[] or pattern_listeVide.match(str(valeur)) :
 350              self.textePy += nom + "= None,"
 351              return
 352
 353           # les 4 lignes suivantes sont probablement inutiles
 354           while valeur[-1] == " " or  valeur[-1]=="'" : valeur=valeur[0:-1]
 355           while valeur[0]  == " " or  valeur[-0]=="'" : valeur=valeur[1:]
 356           oldValeur=valeur
 357           if isinstance(valeur, basestring) :
 358              if   ";" in valeur : valeur=valeur.split(';')
 359              else  : valeur=valeur.split(',')
 360
 361           if len(valeur)< 2 and pattern_flottant.match(oldValeur):
 362           # Attention : on attend une liste mais on a une seule valeur!
 363              try :    oldValeur=eval(oldValeur,{})
 364              except : pass
 365              if nom in TelemacdicoEn.keys() :
 366                 v=TelemacdicoEn[nom][oldValeur]
 367                 self.textePy += nom + "= ('" + str(v) +"',),"
 368              else :
 369                 self.textePy += nom + "= (" + str(oldValeur) +",),"
 370              return
 371
 372
 373           if valeur == None : return
 374           newVal=[]
 375           for v in valeur :
 376             try :    v=eval(v,{})
 377             except : pass
 378             if nom in TelemacdicoEn.keys():
 379                try    : v=TelemacdicoEn[nom][v]
 380                except : pass
 381             newVal.append(v)
 382           self.textePy += nom + "=" + str(newVal) +","
 383
 384
 385
 386    def tri(self, listeIn):
 387       if len(listeIn) == 1 : return listeIn
 388       if self.Ordre_Des_Commandes == None : return listeIn
 389       listeOut=[listeIn[0],]
 390       for k in listeIn[1:]:
 391           #k=str(self.dicoFrancaisAnglais[kF])
 392           ordreK=self.Ordre_Des_Commandes.index(k)
 393           i=0
 394           while i < len(listeOut):
 395              #ordreI=self.Ordre_Des_Commandes.index(self.dicoFrancaisAnglais[listeOut[i]])
 396              ordreI=self.Ordre_Des_Commandes.index(listeOut[i])
 397              if ordreK < ordreI : break
 398              i=i+1
 399           #listeOut.insert(i,kF)
 400           listeOut.insert(i,k)
 401       return listeOut
 402
 403    def LIQUID_BOUNDARIES(self):
 404        texte_Boundaries="LIQUID_BOUNDARIES=( "
 405        if 'PRESCRIBED_ELEVATIONS' in self.dictSimp.keys():
 406               valeursPE=self.dictSimp["PRESCRIBED_ELEVATIONS"]
 407               if not type(valeursPE)==list : valeursPE = (valeursPE,)
 408               longueur=len(self.dictSimp["PRESCRIBED_ELEVATIONS"])
 409        else : valeursPE=None
 410        if 'PRESCRIBED_FLOWRATES' in self.dictSimp.keys():
 411               valeursPF=self.dictSimp["PRESCRIBED_FLOWRATES"]
 412               if not type(valeursPF)==list : valeursPF = (valeursPF,)
 413               longueur=len(self.dictSimp["PRESCRIBED_FLOWRATES"])
 414        else : valeursPF=None
 415        if 'PRESCRIBED_VELOCITIES' in self.dictSimp.keys():
 416               valeursPV=self.dictSimp["PRESCRIBED_VELOCITIES"]
 417               if not type(valeursPV)==list : valeursPV = (valeursPV,)
 418               longueur=len(self.dictSimp["PRESCRIBED_VELOCITIES"])
 419        else : valeursPV=None
 420
 421        if valeursPE == None and valeursPF == None and valeursPV == None :
 422              texte_Boundaries +="),\n"
 423              return
 424
 425        if valeursPE == None or valeursPF == None or valeursPV == None :
 426           listNulle=[]
 427           for i in range(longueur) : listNulle.append('0')
 428
 429
 430        if valeursPE == None : valeursPE = listNulle
 431        if valeursPF == None : valeursPF = listNulle
 432        if valeursPV == None : valeursPV = listNulle
 433
 434
 435        for e in range(len(valeursPE)):
 436           if valeursPE[e] != "" or valeursPE[e] != "\n" :
 437             if eval(valeursPE[e],{}) != 0 :
 438                texte_Boundaries += "_F(BOUNDARY_TYPE = 'Prescribed Elevations',\n"
 439                texte_Boundaries += "PRESCRIBED_ELEVATIONS = " + str(valeursPE[e]) + "),\n"
 440                continue
 441
 442           if valeursPF[e] != "" or valeursPF[e] != "\n" :
 443             if eval(valeursPF[e],{}) != 0 :
 444                texte_Boundaries += "_F(BOUNDARY_TYPE = 'Prescribed Flowrates',\n"
 445                texte_Boundaries += "PRESCRIBED_FLOWRATES = " + str(valeursPF[e]) + "),\n"
 446                continue
 447
 448           if valeursPV[e] != "" or valeursPV[e] != "\n" :
 449              if eval(valeursPV[e],{})!=0 :
 450                 texte_Boundaries += "_F( BOUNDARY_TYPE= 'Prescribed Velocity',\n"
 451                 texte_Boundaries += "PRESCRIBED_VELOCITIES = " + str(valeursPV[e]) + "),\n"
 452                 continue
 453           print ("pb texte_Boundaries avec la valeur numero ", e)
 454
 455        texte_Boundaries +="),\n"
 456        self.textePy += texte_Boundaries
 457