1 |
jgs |
144 |
# $Id$ |
2 |
|
|
|
3 |
|
|
|
4 |
|
|
from esys.modelframe import Model |
5 |
|
|
from esys.escript import inf,sup,Lsup |
6 |
|
|
|
7 |
|
|
|
8 |
|
|
class Probe(Model): |
9 |
|
|
"""@brief tests values against a reference which may depend on time and spatial coordinates |
10 |
|
|
it prints out the relative error in each time step and the maximum relative error over |
11 |
|
|
all time steps at the end |
12 |
|
|
|
13 |
|
|
WARNING: this class use python's eval function!!!!! |
14 |
|
|
|
15 |
|
|
@param value (in) - values to be tested |
16 |
|
|
@param reference (in) - expressions defining reference values to test against. If None only value is reported. |
17 |
|
|
@param line_tag (in) - tag to be used when printing error |
18 |
|
|
@param max_error (out) - maximum error |
19 |
|
|
@param t_max (out) - time of maximum error |
20 |
|
|
|
21 |
|
|
""" |
22 |
|
|
|
23 |
|
|
def __init__(self,debug=False): |
24 |
|
|
"""set up parameters""" |
25 |
|
|
Model.__init__(self,debug=debug) |
26 |
|
|
self.declareParameter(reference=None, \ |
27 |
|
|
value=0., \ |
28 |
|
|
line_tag="PROBE") |
29 |
|
|
|
30 |
|
|
def doInitialization(self,t): |
31 |
|
|
"""initializes values""" |
32 |
|
|
self.__tn=t |
33 |
|
|
self.t_max=None |
34 |
|
|
self.max_err=0. |
35 |
|
|
|
36 |
|
|
def doStep(self,dt): |
37 |
|
|
t=self.__tn+dt |
38 |
|
|
print "%s : time %d"%t |
39 |
|
|
v=self.value |
40 |
|
|
x=v.getFunctionSpace().getX() |
41 |
|
|
if v.getRank()==0: |
42 |
|
|
if self.reference==None: |
43 |
|
|
print "%s : (min,max)= (%e,%e)"%(self.line_tag,inf(v),max(v)) |
44 |
|
|
else: |
45 |
|
|
ref=eval(self.reference) |
46 |
|
|
err=Lsup(v-ref)/Lsup(ref) |
47 |
|
|
print "%s : (min,max,rel error)= (%e,%e,%e)"%(self.line_tag,inf(v),max(v),err) |
48 |
|
|
else: |
49 |
|
|
err=0 |
50 |
|
|
for i in range(v.getRank()): |
51 |
|
|
vi=v[i] |
52 |
|
|
if self.reference==None: |
53 |
|
|
print "%s : component %d (min,max)= (%e,%e)"%(self.line_tag,i,inf(vi),max(vi)) |
54 |
|
|
else: |
55 |
|
|
refi=eval(self.reference[i]) |
56 |
|
|
erri=Lsup(vi-refi)/Lsup(refi) |
57 |
|
|
print "%s : component %d (min,max,rel error)= (%e,%e,%e)"%(self.line_tag,i,inf(vi),max(vi),erri) |
58 |
|
|
err=max(err,erri) |
59 |
|
|
if not self.reference==None: print "%s : maximum error %e",err |
60 |
|
|
self.__tn=t |
61 |
|
|
|
62 |
|
|
if not self.reference==None: |
63 |
|
|
if err>self.max_err: |
64 |
|
|
self.t_max=self.__tn |
65 |
|
|
self.max_error=err/max_ref |
66 |
|
|
|
67 |
|
|
def finalize(self): |
68 |
|
|
"""print out the maximum error""" |
69 |
|
|
if not self.t_max==None: print "%s : component %d (min,max)= (%e,%e)"%(self.line_tag,self.t_max,self.max_err) |