Noyau/N_FONCTION.py

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