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.
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.
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 # ======================================================================
21 # RESPONSABLE MCOURTOI M.COURTOIS
23 EnumTypes = (ListType, TupleType)
25 def calc_table_ops(self, TABLE, ACTION, INFO, **args):
27 Macro CALC_TABLE permettant de faire des opérations sur une table
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
39 # La macro compte pour 1 dans la numerotation des commandes
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'
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')
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()
61 tab = TABLE.EXTR_TABLE()
63 #----------------------------------------------
64 # Boucle sur les actions à effectuer
66 occ = fOP.cree_dict_valeurs(fOP.mc_liste)
67 for mc, val in occ.items():
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)]
79 for k in ('CRITERE','PRECISION'):
83 col = getattr(tab, occ['NOM_PARA'])
84 tab = getattr(col, occ['CRIT_COMP'])(*opts,**kargs)
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]))
91 #----------------------------------------------
92 # 2. Traitement de EXTR
93 if occ['OPERATION'] == 'EXTR':
94 lpar = occ['NOM_PARA']
95 if not type(lpar) in EnumTypes:
99 UTMESS('F','TABLE0_2',valk=[p,TABLE.nom])
100 tab = tab[occ['NOM_PARA']]
102 #----------------------------------------------
103 # 3. Traitement de RENOMME
104 if occ['OPERATION'] == 'RENOMME':
106 tab.Renomme(*occ['NOM_PARA'])
107 except KeyError, msg:
108 UTMESS('F','TABLE0_3',valk=msg)
110 #----------------------------------------------
111 # 4. Traitement du TRI
112 if occ['OPERATION'] == 'TRI':
113 tab.sort(CLES=occ['NOM_PARA'], ORDRE=occ['ORDRE'])
115 #----------------------------------------------
116 # 5. Traitement de COMB
117 if occ['OPERATION'] == 'COMB':
118 tab2 = occ['TABLE'].EXTR_TABLE()
120 if occ.get('NOM_PARA') != None:
121 lpar = occ['NOM_PARA']
122 if not type(lpar) in EnumTypes:
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] )
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'])
138 vectval = getattr(tab, occ['NOM_PARA']).values()
139 aster.affiche('MESSAGE', 'Ajout de la colonne %s : %s' % (occ['NOM_PARA']+repr(vectval))+'\n')
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
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)
156 dprod = tab.dict_CREA_TABLE()
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)
167 # surcharge par le titre fourni
170 if not type(tit) in EnumTypes:
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,