Aster/Cata/cataSTA73/Macro/macro_elas_mult_ops.py

   1 #@ MODIF macro_elas_mult_ops Macro  DATE 14/09/2004   AUTEUR MCOURTOI M.COURTOIS
   2 # -*- coding: iso-8859-1 -*-
   3 #            CONFIGURATION MANAGEMENT OF EDF VERSION
   4 # ======================================================================
   5 # COPYRIGHT (C) 1991 - 2002  EDF R&D                  WWW.CODE-ASTER.ORG
   6 # THIS PROGRAM IS FREE SOFTWARE; YOU CAN REDISTRIBUTE IT AND/OR MODIFY
   7 # IT UNDER THE TERMS OF THE GNU GENERAL PUBLIC LICENSE AS PUBLISHED BY
   8 # THE FREE SOFTWARE FOUNDATION; EITHER VERSION 2 OF THE LICENSE, OR
   9 # (AT YOUR OPTION) ANY LATER VERSION.
  10 #
  11 # THIS PROGRAM IS DISTRIBUTED IN THE HOPE THAT IT WILL BE USEFUL, BUT
  12 # WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF
  13 # MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. SEE THE GNU
  14 # GENERAL PUBLIC LICENSE FOR MORE DETAILS.
  15 #
  16 # YOU SHOULD HAVE RECEIVED A COPY OF THE GNU GENERAL PUBLIC LICENSE
  17 # ALONG WITH THIS PROGRAM; IF NOT, WRITE TO EDF R&D CODE_ASTER,
  18 #    1 AVENUE DU GENERAL DE GAULLE, 92141 CLAMART CEDEX, FRANCE.
  19 # ======================================================================
  20
  21
  22
  23 def macro_elas_mult_ops(self,MODELE,CHAM_MATER,CARA_ELEM,NUME_DDL,
  24                         CHAR_MECA_GLOBAL,CHAR_CINE_GLOBAL,LIAISON_DISCRET,
  25                         CAS_CHARGE,SOLVEUR,**args):
  26   """
  27      Ecriture de la macro MACRO_ELAS_MULT
  28   """
  29   ier=0
  30   import types
  31   from Accas import _F
  32
  33   # On met le mot cle NUME_DDL dans une variable locale pour le proteger
  34   numeddl=NUME_DDL
  35   # On importe les definitions des commandes a utiliser dans la macro
  36   CALC_MATR_ELEM  =self.get_cmd('CALC_MATR_ELEM')
  37   NUME_DDL        =self.get_cmd('NUME_DDL')
  38   ASSE_MATRICE    =self.get_cmd('ASSE_MATRICE')
  39   FACT_LDLT       =self.get_cmd('FACT_LDLT')
  40   CALC_VECT_ELEM  =self.get_cmd('CALC_VECT_ELEM')
  41   ASSE_VECTEUR    =self.get_cmd('ASSE_VECTEUR')
  42   RESO_LDLT       =self.get_cmd('RESO_LDLT')
  43   CREA_RESU       =self.get_cmd('CREA_RESU')
  44   CALC_ELEM       =self.get_cmd('CALC_ELEM')
  45   CALC_NO         =self.get_cmd('CALC_NO')
  46   # La macro compte pour 1 dans la numerotation des commandes
  47   #self.icmd=1
  48   self.set_icmd(1)
  49
  50   # Le concept sortant (de type mult_elas ou fourier_elas) est nommé
  51   # 'nomres' dans le contexte de la macro
  52
  53   self.DeclareOut('nomres',self.sd)
  54
  55   ielas = 0
  56   ifour = 0
  57   for m in CAS_CHARGE:
  58      if m['NOM_CAS']:
  59         ielas=1                 # mot clé NOM_CAS      présent sous CAS_CHARGE
  60         tyresu = 'MULT_ELAS'
  61      else:
  62         ifour=1                 # mot clé MODE_FOURIER présent sous CAS_CHARGE
  63         tyresu = 'FOURIER_ELAS'
  64   if ielas==1 and ifour==1:
  65      ier=ier+1
  66      self.cr.fatal("""<F> <MACRO_ELAS_MULT> On ne peut avoir a la fois NOM_CAS et MODE_FOURIER""")
  67      return ier
  68
  69   if (numeddl in self.sdprods) or (numeddl==None):
  70     # Si le concept numeddl est dans self.sdprods ou n est pas nommé
  71     # il doit etre  produit par la macro
  72     # il faudra donc appeler la commande NUME_DDL
  73     lnume = 1
  74   else:
  75     lnume = 0
  76
  77   if ielas==1 :
  78      motscles={}
  79      if   CHAR_MECA_GLOBAL: motscles['CHARGE']    =CHAR_MECA_GLOBAL
  80      elif CHAR_CINE_GLOBAL: motscles['CHARGE']    =CHAR_CINE_GLOBAL
  81      if   CHAM_MATER      : motscles['CHAM_MATER']=CHAM_MATER
  82      if   CARA_ELEM       : motscles['CARA_ELEM'] =CARA_ELEM
  83      __nomrig=CALC_MATR_ELEM(OPTION='RIGI_MECA',MODELE=MODELE,**motscles)
  84
  85      if lnume:
  86        # On peut passer des mots cles egaux a None. Ils sont ignores
  87        motscles={}
  88        if SOLVEUR:
  89           motscles['METHODE'] =SOLVEUR['METHODE']
  90           motscles['RENUM']   =SOLVEUR['RENUM']
  91        else:
  92           motscles['METHODE'] ='MULT_FRONT'
  93           motscles['RENUM']   ='METIS'
  94        if numeddl!=None:
  95           self.DeclareOut('num',numeddl)
  96           num=NUME_DDL(MATR_RIGI=__nomrig,**motscles)
  97        else:
  98           _num=NUME_DDL(MATR_RIGI=__nomrig,**motscles)
  99           num=_num
 100      else:
 101        num=numeddl
 102
 103      __nomras=ASSE_MATRICE(MATR_ELEM=__nomrig,NUME_DDL=num)
 104
 105      __nomraf=FACT_LDLT(MATR_ASSE=__nomras,NPREC=SOLVEUR['NPREC'],STOP_SINGULIER=SOLVEUR['STOP_SINGULIER'])
 106
 107 #####################################################################
 108 # boucle sur les items de CAS_CHARGE
 109
 110   nomchn=[]
 111   iocc=0
 112   for m in CAS_CHARGE:
 113      iocc=iocc+1
 114
 115      if ifour:
 116         motscles={}
 117         if   CHAR_MECA_GLOBAL: motscles['CHARGE']       =CHAR_MECA_GLOBAL
 118         elif CHAR_CINE_GLOBAL: motscles['CHARGE']       =CHAR_CINE_GLOBAL
 119         if   CHAM_MATER      : motscles['CHAM_MATER']   =CHAM_MATER
 120         if   CARA_ELEM       : motscles['CARA_ELEM']    =CARA_ELEM
 121         motscles['MODE_FOURIER'] =m['MODE_FOURIER']
 122         __nomrig=CALC_MATR_ELEM(OPTION='RIGI_MECA',MODELE=MODELE,**motscles)
 123
 124         if lnume:
 125            _num=NUME_DDL(MATR_RIGI=__nomrig,METHODE=SOLVEUR['METHODE'],RENUM=SOLVEUR['RENUM'])
 126            num=_num
 127            lnume=0
 128
 129         __nomras=ASSE_MATRICE(MATR_ELEM=__nomrig,NUME_DDL=num)
 130
 131         __nomraf=FACT_LDLT(MATR_ASSE=__nomras,NPREC=SOLVEUR['NPREC'],STOP_SINGULIER=SOLVEUR['STOP_SINGULIER'])
 132
 133
 134      if m['VECT_ASSE']==None :
 135         motscles={}
 136         if   CHAM_MATER      : motscles['CHAM_MATER']   =CHAM_MATER
 137         if   CARA_ELEM       : motscles['CARA_ELEM']    =CARA_ELEM
 138         if   ifour           : motscles['MODE_FOURIER'] =m['MODE_FOURIER']
 139         if   m['CHAR_MECA']  : motscles['CHARGE']       =m['CHAR_MECA']
 140         elif m['CHAR_CINE']  : motscles['CHARGE']       =m['CHAR_CINE']
 141         __nomvel=CALC_VECT_ELEM(OPTION='CHAR_MECA',**motscles)
 142         __nomasv=ASSE_VECTEUR(VECT_ELEM=__nomvel,NUME_DDL=num)
 143      else :
 144         __nomasv=m['VECT_ASSE']
 145
 146
 147      __nomchn=RESO_LDLT(MATR_FACT=__nomraf,CHAM_NO=__nomasv,TITRE=m['SOUS_TITRE'])
 148      nomchn.append(__nomchn)
 149
 150 # fin de la boucle sur les items de CAS_CHARGE
 151 #####################################################################
 152
 153   motscles={}
 154   iocc=0
 155   motscle2={}
 156   if CHAM_MATER : motscle2['CHAM_MATER']=CHAM_MATER
 157   if CARA_ELEM  : motscle2['CARA_ELEM']=CARA_ELEM
 158   if CHAM_MATER or CARA_ELEM :
 159     if ielas :
 160        motscles['AFFE']=[]
 161        for m in CAS_CHARGE:
 162           motscles['AFFE'].append(_F(MODELE=MODELE,
 163                                      CHAM_GD=nomchn[iocc],
 164                                      NOM_CAS=m['NOM_CAS'],
 165                                      **motscle2) )
 166           iocc=iocc+1
 167     else :
 168        motscles['AFFE']=[]
 169        for m in CAS_CHARGE:
 170           motscles['AFFE'].append(_F(MODELE=MODELE,
 171                                      CHAM_GD=nomchn[iocc],
 172                                      NUME_MODE=m['MODE_FOURIER'],
 173                                      TYPE_MODE=m['TYPE_MODE'],
 174                                      **motscle2) )
 175           iocc=iocc+1
 176   else:
 177     if ielas :
 178        motscles['AFFE']=[]
 179        for m in CAS_CHARGE:
 180           motscles['AFFE'].append(_F(MODELE=MODELE,
 181                                      CHAM_GD=nomchn[iocc],
 182                                      NOM_CAS=m['NOM_CAS'],) )
 183           iocc=iocc+1
 184     else :
 185        motscles['AFFE']=[]
 186        for m in CAS_CHARGE:
 187           motscles['AFFE'].append(_F(MODELE=MODELE,
 188                                      CHAM_GD=nomchn[iocc],
 189                                      NUME_MODE=m['MODE_FOURIER'],
 190                                      TYPE_MODE=m['TYPE_MODE'],) )
 191           iocc=iocc+1
 192
 193
 194   nomres=CREA_RESU(OPERATION='AFFE',TYPE_RESU=tyresu,NOM_CHAM='DEPL',**motscles)
 195
 196 #####################################################################
 197 # boucle sur les items de CAS_CHARGE pour CALC_ELEM ete CALC_NO
 198
 199   iocc=0
 200   for m in CAS_CHARGE:
 201      iocc=iocc+1
 202
 203      if m['OPTION']:
 204         nbel=0
 205         nbno=0
 206         liste_el=[]
 207         liste_no=[]
 208         if type(m['OPTION'])==types.StringType:
 209            if m['OPTION'] in ('FORC_NODA','REAC_NODA',
 210                               'EPSI_NOEU_DEPL','SIGM_NOEU_DEPL','EFGE_NOEU_DEPL',
 211                               'EQUI_NOEU_SIGM','EQUI_NOEU_EPSI','FLUX_NOEU_TEMP',):
 212               nbno=nbno+1
 213               liste_no.append(m['OPTION'])
 214            else:
 215               nbel=nbel+1
 216               liste_el.append(m['OPTION'])
 217         else:
 218            for opt in m['OPTION']:
 219               if opt in ('FORC_NODA','REAC_NODA',
 220                          'EPSI_NOEU_DEPL','SIGM_NOEU_DEPL','EFGE_NOEU_DEPL',
 221                          'EQUI_NOEU_SIGM','EQUI_NOEU_EPSI','FLUX_NOEU_TEMP',):
 222                  nbno=nbno+1
 223                  liste_no.append(opt)
 224               else:
 225                  nbel=nbel+1
 226                  liste_el.append(opt)
 227
 228         lreac=0
 229         if nbel:
 230            motscles={}
 231            if   CHAM_MATER : motscles['CHAM_MATER'] =CHAM_MATER
 232            if   CARA_ELEM  : motscles['CARA_ELEM']  =CARA_ELEM
 233            if ielas:
 234               motscles['NOM_CAS']=m['NOM_CAS']
 235            else:
 236               motscles['NUME_MODE']=m['MODE_FOURIER']
 237            motscles['EXCIT']=[]
 238            if   m['CHAR_MECA'] :
 239               for chargt in m['CHAR_MECA'] : motscles['EXCIT'].append(_F(CHARGE=chargt))
 240            elif m['CHAR_CINE'] :
 241               for chargt in m['CHAR_CINE'] : motscles['EXCIT'].append(_F(CHARGE=chargt))
 242            if   CHAR_MECA_GLOBAL:            motscles['EXCIT'].append(_F(CHARGE=CHAR_MECA_GLOBAL))
 243            elif CHAR_CINE_GLOBAL:            motscles['EXCIT'].append(_F(CHARGE=CHAR_CINE_GLOBAL))
 244            CALC_ELEM(reuse=nomres,
 245                      RESULTAT=nomres,
 246                      MODELE=MODELE,
 247                      NIVE_COUCHE=m['NIVE_COUCHE'],
 248                      NUME_COUCHE=m['NUME_COUCHE'],
 249                      OPTION=tuple(liste_el),
 250                      **motscles)
 251         if nbno:
 252            motscles={}
 253            if   CHAM_MATER : motscles['CHAM_MATER'] =CHAM_MATER
 254            if   CARA_ELEM  : motscles['CARA_ELEM']  =CARA_ELEM
 255            if ielas:
 256               motscles['NOM_CAS']=m['NOM_CAS']
 257            else:
 258               motscles['NUME_MODE']=m['MODE_FOURIER']
 259            motscles['EXCIT']=[]
 260            if   m['CHAR_MECA'] :
 261               for chargt in m['CHAR_MECA'] : motscles['EXCIT'].append(_F(CHARGE=chargt))
 262            elif m['CHAR_CINE'] :
 263               for chargt in m['CHAR_CINE'] : motscles['EXCIT'].append(_F(CHARGE=chargt))
 264            if   CHAR_MECA_GLOBAL:            motscles['EXCIT'].append(_F(CHARGE=CHAR_MECA_GLOBAL))
 265            elif CHAR_CINE_GLOBAL:            motscles['EXCIT'].append(_F(CHARGE=CHAR_CINE_GLOBAL))
 266            CALC_NO(reuse=nomres,
 267                    RESULTAT=nomres,
 268                    MODELE=MODELE,
 269                    OPTION=tuple(liste_no),
 270                    **motscles)
 271
 272 # fin de la boucle sur les items de CAS_CHARGE
 273 #####################################################################
 274   return ier
 275