Noyau/N_FONCTION.py

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