Aster/Cata/cataSTA9/Macro/calc_table_ops.py

   1 #@ MODIF calc_table_ops Macro  DATE 19/02/2008   AUTEUR COURTOIS M.COURTOIS
   2 # -*- coding: iso-8859-1 -*-
   3 #            CONFIGURATION MANAGEMENT OF EDF VERSION
   4 # ======================================================================
   5 # COPYRIGHT (C) 1991 - 2005  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 # RESPONSABLE MCOURTOI M.COURTOIS
  22 from types import *
  23 EnumTypes = (ListType, TupleType)
  24
  25 def calc_table_ops(self, TABLE, ACTION, INFO, **args):
  26    """
  27    Macro CALC_TABLE permettant de faire des opérations sur une table
  28    """
  29    import aster
  30
  31    macro = 'CALC_TABLE'
  32    from Accas                 import _F
  33    from Cata.cata             import table_sdaster, table_fonction, table_jeveux
  34    from Utilitai.Utmess       import  UTMESS
  35    from Utilitai              import transpose
  36    from Utilitai.Table        import Table, merge
  37
  38    ier = 0
  39    # La macro compte pour 1 dans la numerotation des commandes
  40    self.set_icmd(1)
  41
  42    # Le concept sortant (de type table_sdaster ou dérivé) est tabout
  43    self.DeclareOut('tabout', self.sd)
  44    if self.sd.__class__ == table_fonction:
  45       typ_tabout = 'TABLE_FONCTION'
  46    else:
  47       typ_tabout = 'TABLE'
  48
  49    # On importe les definitions des commandes a utiliser dans la macro
  50    # Le nom de la variable doit etre obligatoirement le nom de la commande
  51    CREA_TABLE    = self.get_cmd('CREA_TABLE')
  52    DETRUIRE      = self.get_cmd('DETRUIRE')
  53
  54    # 0. faut-il utiliser une table dérivée
  55    form_sens='\n... SENSIBILITE AU PARAMETRE %s (SD COMP %s)'
  56    if args['SENSIBILITE']:
  57       ncomp = self.jdc.memo_sensi.get_nocomp(TABLE.nom, args['SENSIBILITE'].nom)
  58       sdtab = table_jeveux(ncomp)
  59       tab = sdtab.EXTR_TABLE()
  60    else:
  61       tab = TABLE.EXTR_TABLE()
  62
  63    #----------------------------------------------
  64    # Boucle sur les actions à effectuer
  65    for fOP in ACTION:
  66       occ = fOP.cree_dict_valeurs(fOP.mc_liste)
  67       for mc, val in occ.items():
  68          if val == None:
  69             del occ[mc]
  70
  71       #----------------------------------------------
  72       # 1. Traitement du FILTRE
  73       # format pour l'impression des filtres
  74       form_filtre = '\nFILTRE -> NOM_PARA: %-16s CRIT_COMP: %-4s VALE: %s'
  75       if occ['OPERATION'] == 'FILTRE':
  76          # peu importe le type, c'est la meme méthode d'appel
  77          opts = [occ[k] for k in ('VALE','VALE_I','VALE_C','VALE_K') if occ.has_key(k)]
  78          kargs = {}
  79          for k in ('CRITERE','PRECISION'):
  80             if occ.has_key(k):
  81                kargs[k] = occ[k]
  82
  83          col = getattr(tab, occ['NOM_PARA'])
  84          tab = getattr(col, occ['CRIT_COMP'])(*opts,**kargs)
  85
  86          # trace l'operation dans le titre
  87          #if FORMAT in ('TABLEAU','ASTER'):
  88          tab.titr += form_filtre % (occ['NOM_PARA'], occ['CRIT_COMP'], \
  89             ' '.join([str(v) for v in opts]))
  90
  91       #----------------------------------------------
  92       # 2. Traitement de EXTR
  93       if occ['OPERATION'] == 'EXTR':
  94          lpar = occ['NOM_PARA']
  95          if not type(lpar) in EnumTypes:
  96             lpar = [lpar]
  97          for p in lpar:
  98             if not p in tab.para:
  99               UTMESS('F','TABLE0_2',valk=[p,TABLE.nom])
 100          tab = tab[occ['NOM_PARA']]
 101
 102       #----------------------------------------------
 103       # 3. Traitement de RENOMME
 104       if occ['OPERATION'] == 'RENOMME':
 105          try:
 106             tab.Renomme(*occ['NOM_PARA'])
 107          except KeyError, msg:
 108             UTMESS('F','TABLE0_3',valk=msg)
 109
 110       #----------------------------------------------
 111       # 4. Traitement du TRI
 112       if occ['OPERATION'] == 'TRI':
 113          tab.sort(CLES=occ['NOM_PARA'], ORDRE=occ['ORDRE'])
 114
 115       #----------------------------------------------
 116       # 5. Traitement de COMB
 117       if occ['OPERATION'] == 'COMB':
 118          tab2 = occ['TABLE'].EXTR_TABLE()
 119          opts = [tab, tab2]
 120          if occ.get('NOM_PARA') != None:
 121             lpar = occ['NOM_PARA']
 122             if not type(lpar) in EnumTypes:
 123                lpar = [lpar]
 124             for p in lpar:
 125                if not p in tab.para:
 126                   UTMESS('F','TABLE0_4',valk=[p, TABLE.nom])
 127                if not p in tab2.para:
 128                   UTMESS('F','TABLE0_5',valk=[p,occ['TABLE'].nom] )
 129             opts.append(lpar)
 130          tab = merge(*opts)
 131
 132       #----------------------------------------------
 133       # 6. Traitement de OPER
 134       if occ['OPERATION'] == 'OPER':
 135          # ajout de la colonne dans la table
 136          tab.fromfunction(occ['NOM_PARA'], occ['FORMULE'])
 137          if INFO == 2:
 138             vectval = getattr(tab, occ['NOM_PARA']).values()
 139             aster.affiche('MESSAGE', 'Ajout de la colonne %s : %s' % (occ['NOM_PARA']+repr(vectval))+'\n')
 140
 141       #----------------------------------------------
 142       # 6. Traitement de AJOUT
 143       if occ['OPERATION'] == 'AJOUT':
 144          if len(occ['NOM_PARA']) != len(occ['VALE']):
 145             UTMESS('F', 'TABLE0_14')
 146          dnew = dict(zip(occ['NOM_PARA'], occ['VALE']))
 147          # ajout de la ligne avec vérification des types
 148          tab.append(dnew)
 149
 150    #----------------------------------------------
 151    # 99. Création de la table_sdaster résultat
 152    # cas réentrant : il faut détruire l'ancienne table_sdaster
 153    if self.sd.nom == TABLE.nom:
 154       DETRUIRE(CONCEPT=_F(NOM=TABLE.nom,), INFO=1)
 155
 156    dprod = tab.dict_CREA_TABLE()
 157    if INFO == 2:
 158       echo_mess = []
 159       echo_mess.append( '@-'*30+'\n' )
 160       echo_mess.append( tab )
 161       from pprint import pformat
 162       echo_mess.append( pformat(dprod) )
 163       echo_mess.append( '@-'*30+'\n' )
 164       texte_final = ' '.join(echo_mess)
 165       aster.affiche('MESSAGE', texte_final)
 166
 167    # surcharge par le titre fourni
 168    tit = args['TITRE']
 169    if tit != None:
 170       if not type(tit) in EnumTypes:
 171          tit = [tit]
 172       dprod['TITRE'] = tuple(['%-80s' % lig for lig in tit])
 173    # type de la table de sortie à passer à CREA_TABLE
 174    tabout = CREA_TABLE(TYPE_TABLE=typ_tabout,
 175                        **dprod)
 176
 177    return ier
 178