Noyau/N_FONCTION.py

   1 #@ MODIF N_FONCTION Noyau  DATE 17/05/2005   AUTEUR DURAND C.DURAND
   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 from N_ASSD import ASSD
  24 import string
  25
  26 class FONCTION(ASSD):pass
  27
  28 class formule(ASSD):
  29    def __init__(self,**args):
  30       ASSD.__init__(self,**args)
  31       self.nompar    =None
  32       self.expression=None
  33
  34    def __call__(self,*val):
  35       if hasattr(self.parent,'contexte_fichier_init'):
  36                         context=self.parent.contexte_fichier_init
  37       else            : context={}
  38       i=0
  39       for param in self.nompar :
  40          context[param]=val[i]
  41          i=i+1
  42       try :
  43        res=eval(self.expression,self.jdc.const_context, context)
  44       except :
  45        print 75*'!'
  46        print '! '+string.ljust('Erreur evaluation formule '+self.nom,72)+'!'
  47        print 75*'!'
  48        raise
  49       return res
  50
  51    def setFormule(self,nom_para,texte):
  52       """
  53          Cette methode sert a initialiser les attributs
  54          nompar, expression et code qui sont utilisés
  55          dans l'évaluation de la formule
  56       """
  57       self.nompar     = nom_para
  58       self.expression = texte
  59       try :
  60         self.code=compile(texte,texte,'eval')
  61       except SyntaxError :
  62         print 75*'!'
  63         print '! '+string.ljust('Erreur evaluation formule '+self.nom,72)+'!'
  64         print 75*'!'
  65         raise
  66
  67    def __setstate__(self,state):
  68       """
  69          Cette methode sert a restaurer l'attribut code
  70          lors d'un unpickle
  71       """
  72       self.__dict__.update(state)                  # update attributes
  73       self.setFormule(self.nompar,self.expression) # restore code attribute
  74
  75    def __getstate__(self):
  76       """
  77          Pour les formules, il faut enlever l'attribut code
  78          qui n'est pas picklable
  79       """
  80       d=ASSD.__getstate__(self)
  81       del d['code']
  82       return d
  83