source: issm/trunk/src/m/consistency/checkfield.py@ 14067

Last change on this file since 14067 was 14067, checked in by Mathieu Morlighem, 12 years ago

merged trunk-jpl and trunk for revision 14066

File size: 6.1 KB
Line 
1import numpy
2import os
3from pairoptions import *
4from MatlabFuncs import *
5
6def checkfield(md,fieldname,*args):
7 """
8 CHECKFIELD - check field consistency
9
10 Used to check model consistency.
11 Available options:
12 - NaN: 1 if check that there is no NaN
13 - size: [lines cols], NaN for non checked dimensions
14 - >: greater than provided value
15 - >=: greater or equal to provided value
16 - <: smallerthan provided value
17 - <=: smaller or equal to provided value
18 - < vec: smallerthan provided values on each vertex
19 - forcing: 1 if check forcing consistency (size and time)
20 - values: cell of strings or vector of acceptable values
21 - numel: list of acceptable number of elements
22 - cell: 1 if check that is cell
23 - empty: 1 if check that non empty
24 - message: overloaded error message
25
26 Usage:
27 md = checkfield(md,fieldname,options);
28
29 Example:
30 md = checkfield(md,'mesh.elementonbed','size',[md.mesh.numberofelements 1],'values',[0 1]);
31 md = checkfield(md,'diagnostic.icefront','size',[NaN 4],'NaN',1);
32 md = checkfield(md,'diagnostic.icefront(:,end)','values',[0 1 2]);
33 """
34
35 #get options
36 options=pairoptions(*args)
37
38 #get field from model
39# field=getattr(md,fieldname)
40 exec("field=md.%s" % fieldname)
41 if isinstance(field,(bool,int,long,float)):
42 field=numpy.array([field])
43
44 #check empty
45 if options.exist('empty'):
46 if not field:
47 md = md.checkmessage(options.getfieldvalue('message',\
48 "field '%s' is empty" % fieldname))
49
50 #Check size
51 if options.exist('size'):
52 fieldsize=options.getfieldvalue('size')
53 if len(fieldsize) == 1:
54 if numpy.isnan(fieldsize[0]):
55 pass
56 elif not numpy.size(field,0)==fieldsize[0]:
57 md = md.checkmessage(options.getfieldvalue('message',\
58 "field '%s' size should be %d" % (fieldname,fieldsize[0])))
59 elif len(fieldsize) == 2:
60 if numpy.isnan(fieldsize[0]):
61 if not numpy.size(field,1)==fieldsize[1]:
62 md = md.checkmessage(options.getfieldvalue('message',\
63 "field '%s' should have %d columns" % (fieldname,fieldsize[1])))
64 elif numpy.isnan(fieldsize[1]):
65 if not numpy.size(field,0)==fieldsize[0]:
66 md = md.checkmessage(options.getfieldvalue('message',\
67 "field '%s' should have %d lines" % (fieldname,fieldsize[0])))
68 else:
69 if (not numpy.size(field,0)==fieldsize[0]) or (not numpy.size(field,1)==fieldsize[1]):
70 md = md.checkmessage(options.getfieldvalue('message',\
71 "field '%s' size should be %d x %d" % (fieldname,fieldsize[0],fieldsize[1])))
72
73 #Check numel
74 if options.exist('numel'):
75 fieldnumel=options.getfieldvalue('numel')
76 if numpy.size(field) not in fieldnumel:
77 if len(fieldnumel)==1:
78 md = md.checkmessage(options.getfieldvalue('message',\
79 "field '%s' size should be %d" % (fieldname,fieldnumel)))
80 elif len(fieldnumel)==2:
81 md = md.checkmessage(options.getfieldvalue('message',\
82 "field '%s' size should be %d or %d" % (fieldname,fieldnumel[0],fieldnumel[1])))
83 else:
84 md = md.checkmessage(options.getfieldvalue('message',\
85 "field '%s' size should be %s" % (fieldname,fieldnumel)))
86
87 #check NaN
88 if options.getfieldvalue('NaN',0):
89 if numpy.any(numpy.isnan(field)):
90 md = md.checkmessage(options.getfieldvalue('message',\
91 "NaN values found in field '%s'" % fieldname))
92
93 #check cell
94 if options.getfieldvalue('cell',0):
95 if not isinstance(field,(tuple,list,dict)):
96 md = md.checkmessage(options.getfieldvalue('message',\
97 "field '%s' should be a cell" % fieldname))
98
99 #check values
100 if options.exist('values'):
101 fieldvalues=options.getfieldvalue('values')
102 if False in ismember(field,fieldvalues):
103 if len(fieldvalues)==1:
104 md = md.checkmessage(options.getfieldvalue('message',\
105 "field '%s' value should be '%s'" % (fieldname,fieldvalues[0])))
106 elif len(fieldvalues)==2:
107 md = md.checkmessage(options.getfieldvalue('message',\
108 "field '%s' values should be '%s' or '%s'" % (fieldname,fieldvalues[0],fieldvalues[1])))
109 else:
110 md = md.checkmessage(options.getfieldvalue('message',\
111 "field '%s' should have values in %s" % (fieldname,fieldvalues)))
112
113 #check greater
114 if options.exist('>='):
115 lowerbound=options.getfieldvalue('>=')
116 if numpy.any(field<lowerbound):
117 md = md.checkmessage(options.getfieldvalue('message',\
118 "field '%s' should have values above %d" % (fieldname,lowerbound)))
119 if options.exist('>'):
120 lowerbound=options.getfieldvalue('>')
121 if numpy.any(field<=lowerbound):
122 md = md.checkmessage(options.getfieldvalue('message',\
123 "field '%s' should have values above %d" % (fieldname,lowerbound)))
124
125 #check smaller
126 if options.exist('<='):
127 upperbound=options.getfieldvalue('<=')
128 if numpy.any(field>upperbound):
129 md = md.checkmessage(options.getfieldvalue('message',\
130 "field '%s' should have values below %d" % (fieldname,upperbound)))
131 if options.exist('<'):
132 upperbound=options.getfieldvalue('<')
133 if numpy.any(field>=upperbound):
134 md = md.checkmessage(options.getfieldvalue('message',\
135 "field '%s' should have values below %d" % (fieldname,upperbound)))
136
137 #check file
138 if options.getfieldvalue('file',0):
139 if not os.path.exists(field):
140 md = md.checkmessage("file provided in '%s': '%s' does not exist" % (fieldname,field))
141
142 #Check forcings (size and times)
143 if options.getfieldvalue('forcing',0):
144 if numpy.size(field,0)==md.mesh.numberofvertices:
145 if numpy.ndim(field)>1 and not numpy.size(field,1)==1:
146 md = md.checkmessage(options.getfieldvalue('message',\
147 "field '%s' should have only one column as there are md.mesh.numberofvertices lines" % fieldname))
148 elif numpy.size(field,0)==md.mesh.numberofvertices+1:
149 if not all(field[-1,:]==numpy.sort(field[-1,:])):
150 md = md.checkmessage(options.getfieldvalue('message',\
151 "field '%s' columns should be sorted chronologically" % fieldname))
152 if any(field[-1,0:-1]==field[-1,1:]):
153 md = md.checkmessage(options.getfieldvalue('message',\
154 "field '%s' columns must not contain duplicate timesteps" % fieldname))
155 else:
156 md = md.checkmessage(options.getfieldvalue('message',\
157 "field '%s' should have md.mesh.numberofvertices or md.mesh.numberofvertices+1 lines" % fieldname))
158
159 return md
160
Note: See TracBrowser for help on using the repository browser.