-# -*- coding: utf-8 -*-
-#@ MODIF N_FONCTION Noyau DATE 11/03/2003 AUTEUR DURAND C.DURAND
+#@ MODIF N_FONCTION Noyau DATE 19/09/2005 AUTEUR DURAND C.DURAND
+# -*- coding: iso-8859-1 -*-
# CONFIGURATION MANAGEMENT OF EDF VERSION
# ======================================================================
# COPYRIGHT (C) 1991 - 2002 EDF R&D WWW.CODE-ASTER.ORG
#
#
# ======================================================================
+
+# attention ! cet import permet d'avoir, dans les formules, le comportement
+# de la division réelle pour les entiers, et non la division entière
+# 1/2=0.5 (et non 0)
+from __future__ import division
+
from N_ASSD import ASSD
+import string
class FONCTION(ASSD):pass
-class formule(FONCTION) : pass
+class formule(ASSD):
+ def __init__(self,**args):
+ ASSD.__init__(self,**args)
+ self.nompar =None
+ self.expression=None
+
+ def __call__(self,*val):
+ if hasattr(self.parent,'contexte_fichier_init'):
+ context=self.parent.contexte_fichier_init
+ else : context={}
+ i=0
+ for param in self.nompar :
+ context[param]=val[i]
+ i=i+1
+ try :
+ res=eval(self.expression,self.jdc.const_context, context)
+ except :
+ print 75*'!'
+ print '! '+string.ljust('Erreur evaluation formule '+self.nom,72)+'!'
+ print 75*'!'
+ raise
+ return res
+
+ def setFormule(self,nom_para,texte):
+ """
+ Cette methode sert a initialiser les attributs
+ nompar, expression et code qui sont utilisés
+ dans l'évaluation de la formule
+ """
+ self.nompar = nom_para
+ self.expression = texte
+ try :
+ self.code=compile(texte,texte,'eval')
+ except SyntaxError :
+ print 75*'!'
+ print '! '+string.ljust('Erreur evaluation formule '+self.nom,72)+'!'
+ print 75*'!'
+ raise
+
+ def __setstate__(self,state):
+ """
+ Cette methode sert a restaurer l'attribut code
+ lors d'un unpickle
+ """
+ self.__dict__.update(state) # update attributes
+ self.setFormule(self.nompar,self.expression) # restore code attribute
+
+ def __getstate__(self):
+ """
+ Pour les formules, il faut enlever l'attribut code
+ qui n'est pas picklable
+ """
+ d=ASSD.__getstate__(self)
+ del d['code']
+ return d
+