Noyau/N_FONCTION.py

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