Branch data Line data Source code
1 : : #include "NCHelperFV.hpp"
2 : : #include "moab/FileOptions.hpp"
3 : :
4 : : #include <cmath>
5 : : #include <sstream>
6 : :
7 : : namespace moab
8 : : {
9 : :
10 : 0 : bool NCHelperFV::can_read_file( ReadNC* readNC, int fileId )
11 : : {
12 : 0 : std::vector< std::string >& dimNames = readNC->dimNames;
13 : :
14 : : // If dimension names "lon" AND "lat" AND "slon" AND "slat" exist then it should be the FV grid
15 [ # # ][ # # ]: 0 : if( ( std::find( dimNames.begin(), dimNames.end(), std::string( "lon" ) ) != dimNames.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # # #
# # # # #
# ]
16 [ # # ][ # # ]: 0 : ( std::find( dimNames.begin(), dimNames.end(), std::string( "lat" ) ) != dimNames.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # # #
# # # # ]
17 [ # # ][ # # ]: 0 : ( std::find( dimNames.begin(), dimNames.end(), std::string( "slon" ) ) != dimNames.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
18 [ # # ][ # # ]: 0 : ( std::find( dimNames.begin(), dimNames.end(), std::string( "slat" ) ) != dimNames.end() ) )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
19 : : {
20 : : // Make sure it is CAM grid
21 [ # # ][ # # ]: 0 : std::map< std::string, ReadNC::AttData >::iterator attIt = readNC->globalAtts.find( "source" );
22 [ # # ][ # # ]: 0 : if( attIt == readNC->globalAtts.end() ) return false;
23 [ # # ]: 0 : unsigned int sz = attIt->second.attLen;
24 [ # # ]: 0 : std::string att_data;
25 [ # # ]: 0 : att_data.resize( sz + 1 );
26 [ # # ]: 0 : att_data[sz] = '\000';
27 : : int success =
28 [ # # ][ # # ]: 0 : NCFUNC( get_att_text )( fileId, attIt->second.attVarId, attIt->second.attName.c_str(), &att_data[0] );
[ # # ][ # # ]
29 [ # # ]: 0 : if( success ) return false;
30 [ # # ][ # # ]: 0 : if( att_data.find( "CAM" ) == std::string::npos ) return false;
31 : :
32 : 0 : return true;
33 : : }
34 : :
35 : 0 : return false;
36 : : }
37 : :
38 : 0 : ErrorCode NCHelperFV::init_mesh_vals()
39 : : {
40 : 0 : Interface*& mbImpl = _readNC->mbImpl;
41 : 0 : std::vector< std::string >& dimNames = _readNC->dimNames;
42 : 0 : std::vector< int >& dimLens = _readNC->dimLens;
43 : 0 : std::map< std::string, ReadNC::VarData >& varInfo = _readNC->varInfo;
44 : 0 : DebugOutput& dbgOut = _readNC->dbgOut;
45 : 0 : bool& isParallel = _readNC->isParallel;
46 : 0 : int& partMethod = _readNC->partMethod;
47 : 0 : ScdParData& parData = _readNC->parData;
48 : :
49 : : // Look for names of i/j dimensions
50 : : // First i
51 : 0 : std::vector< std::string >::iterator vit;
52 : : unsigned int idx;
53 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "slon" ) ) != dimNames.end() )
[ # # ]
54 [ # # ]: 0 : idx = vit - dimNames.begin();
55 : : else
56 : : {
57 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'slon' variable" );
[ # # ][ # # ]
[ # # ]
58 : : }
59 : 0 : iDim = idx;
60 : 0 : gDims[0] = 0;
61 [ # # ]: 0 : gDims[3] = dimLens[idx] - 1;
62 : :
63 : : // Then j
64 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "slat" ) ) != dimNames.end() )
[ # # ]
65 [ # # ]: 0 : idx = vit - dimNames.begin();
66 : : else
67 : : {
68 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'slat' variable" );
[ # # ][ # # ]
[ # # ]
69 : : }
70 : 0 : jDim = idx;
71 : 0 : gDims[1] = 0;
72 [ # # ]: 0 : gDims[4] = dimLens[idx] - 1 + 2; // Add 2 for the pole points
73 : :
74 : : // Look for names of center i/j dimensions
75 : : // First i
76 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "lon" ) ) != dimNames.end() )
[ # # ]
77 [ # # ]: 0 : idx = vit - dimNames.begin();
78 : : else
79 : : {
80 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'lon' variable" );
[ # # ][ # # ]
[ # # ]
81 : : }
82 : 0 : iCDim = idx;
83 : 0 : gCDims[0] = 0;
84 [ # # ]: 0 : gCDims[3] = dimLens[idx] - 1;
85 : :
86 : : // Check i periodicity and set globallyPeriodic[0]
87 [ # # ]: 0 : std::vector< double > til_vals( 2 );
88 [ # # ][ # # ]: 0 : ErrorCode rval = read_coordinate( "lon", gCDims[3] - 1, gCDims[3], til_vals );MB_CHK_SET_ERR( rval, "Trouble reading 'lon' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
89 [ # # ][ # # ]: 0 : if( std::fabs( 2 * til_vals[1] - til_vals[0] - 360 ) < 0.001 ) globallyPeriodic[0] = 1;
[ # # ]
90 [ # # ]: 0 : if( globallyPeriodic[0] )
91 [ # # ]: 0 : assert( "Number of vertices and edges should be same" && gDims[3] == gCDims[3] );
92 : : else
93 [ # # ]: 0 : assert( "Number of vertices should equal to number of edges plus one" && gDims[3] == gCDims[3] + 1 );
94 : :
95 : : // Then j
96 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "lat" ) ) != dimNames.end() )
[ # # ]
97 [ # # ]: 0 : idx = vit - dimNames.begin();
98 : : else
99 : : {
100 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'lat' dimension" );
[ # # ][ # # ]
[ # # ]
101 : : }
102 : 0 : jCDim = idx;
103 : 0 : gCDims[1] = 0;
104 [ # # ]: 0 : gCDims[4] = dimLens[idx] - 1;
105 : :
106 : : // For FV models, will always be non-periodic in j
107 [ # # ]: 0 : assert( gDims[4] == gCDims[4] + 1 );
108 : :
109 : : // Try a truly 2D mesh
110 : 0 : gDims[2] = -1;
111 : 0 : gDims[5] = -1;
112 : :
113 : : // Look for time dimension
114 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "time" ) ) != dimNames.end() )
[ # # ]
115 [ # # ]: 0 : idx = vit - dimNames.begin();
116 [ # # ][ # # ]: 0 : else if( ( vit = std::find( dimNames.begin(), dimNames.end(), "t" ) ) != dimNames.end() )
[ # # ]
117 [ # # ]: 0 : idx = vit - dimNames.begin();
118 : : else
119 : : {
120 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'time' or 't' dimension" );
[ # # ][ # # ]
[ # # ]
121 : : }
122 : 0 : tDim = idx;
123 [ # # ]: 0 : nTimeSteps = dimLens[idx];
124 : :
125 : : // Get number of levels
126 [ # # ][ # # ]: 0 : if( ( vit = std::find( dimNames.begin(), dimNames.end(), "lev" ) ) != dimNames.end() )
[ # # ]
127 [ # # ]: 0 : idx = vit - dimNames.begin();
128 [ # # ][ # # ]: 0 : else if( ( vit = std::find( dimNames.begin(), dimNames.end(), "ilev" ) ) != dimNames.end() )
[ # # ]
129 [ # # ]: 0 : idx = vit - dimNames.begin();
130 : : else
131 : : {
132 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'lev' or 'ilev' dimension" );
[ # # ][ # # ]
[ # # ]
133 : : }
134 : 0 : levDim = idx;
135 [ # # ]: 0 : nLevels = dimLens[idx];
136 : :
137 : : // Parse options to get subset
138 : 0 : int rank = 0, procs = 1;
139 : : #ifdef MOAB_HAVE_MPI
140 [ # # ]: 0 : if( isParallel )
141 : : {
142 : 0 : ParallelComm*& myPcomm = _readNC->myPcomm;
143 [ # # ][ # # ]: 0 : rank = myPcomm->proc_config().proc_rank();
144 [ # # ][ # # ]: 0 : procs = myPcomm->proc_config().proc_size();
145 : : }
146 : : #endif
147 [ # # ]: 0 : if( procs > 1 )
148 : : {
149 [ # # ]: 0 : for( int i = 0; i < 6; i++ )
150 : 0 : parData.gDims[i] = gDims[i];
151 [ # # ]: 0 : for( int i = 0; i < 3; i++ )
152 : 0 : parData.gPeriodic[i] = globallyPeriodic[i];
153 : 0 : parData.partMethod = partMethod;
154 : : int pdims[3];
155 : :
156 [ # # ][ # # ]: 0 : rval = ScdInterface::compute_partition( procs, rank, parData, lDims, locallyPeriodic, pdims );MB_CHK_ERR( rval );
[ # # ][ # # ]
157 [ # # ]: 0 : for( int i = 0; i < 3; i++ )
158 : 0 : parData.pDims[i] = pdims[i];
159 : :
160 : 0 : dbgOut.tprintf( 1, "Partition: %dx%d (out of %dx%d)\n", lDims[3] - lDims[0] + 1, lDims[4] - lDims[1] + 1,
161 [ # # ]: 0 : gDims[3] - gDims[0] + 1, gDims[4] - gDims[1] + 1 );
162 [ # # ]: 0 : if( 0 == rank )
163 : : dbgOut.tprintf( 1, "Contiguous chunks of size %d bytes.\n",
164 [ # # ]: 0 : 8 * ( lDims[3] - lDims[0] + 1 ) * ( lDims[4] - lDims[1] + 1 ) );
165 : : }
166 : : else
167 : : {
168 [ # # ]: 0 : for( int i = 0; i < 6; i++ )
169 : 0 : lDims[i] = gDims[i];
170 : 0 : locallyPeriodic[0] = globallyPeriodic[0];
171 : : }
172 : :
173 [ # # ]: 0 : _opts.get_int_option( "IMIN", lDims[0] );
174 [ # # ]: 0 : _opts.get_int_option( "IMAX", lDims[3] );
175 [ # # ]: 0 : _opts.get_int_option( "JMIN", lDims[1] );
176 [ # # ]: 0 : _opts.get_int_option( "JMAX", lDims[4] );
177 : :
178 : : // Now get actual coordinate values for vertices and cell centers
179 : 0 : lCDims[0] = lDims[0];
180 [ # # ]: 0 : if( locallyPeriodic[0] )
181 : : // If locally periodic, doesn't matter what global periodicity is, # vertex coords = # elem
182 : : // coords
183 : 0 : lCDims[3] = lDims[3];
184 : : else
185 : 0 : lCDims[3] = lDims[3] - 1;
186 : :
187 : : // For FV models, will always be non-periodic in j
188 : 0 : lCDims[1] = lDims[1];
189 : 0 : lCDims[4] = lDims[4] - 1;
190 : :
191 : : // Resize vectors to store values later
192 [ # # ][ # # ]: 0 : if( -1 != lDims[0] ) ilVals.resize( lDims[3] - lDims[0] + 1 );
193 [ # # ][ # # ]: 0 : if( -1 != lCDims[0] ) ilCVals.resize( lCDims[3] - lCDims[0] + 1 );
194 [ # # ][ # # ]: 0 : if( -1 != lDims[1] ) jlVals.resize( lDims[4] - lDims[1] + 1 );
195 [ # # ][ # # ]: 0 : if( -1 != lCDims[1] ) jlCVals.resize( lCDims[4] - lCDims[1] + 1 );
196 [ # # ][ # # ]: 0 : if( nTimeSteps > 0 ) tVals.resize( nTimeSteps );
197 : :
198 : : // Now read coord values
199 [ # # ]: 0 : std::map< std::string, ReadNC::VarData >::iterator vmit;
200 [ # # ]: 0 : if( -1 != lCDims[0] )
201 : : {
202 [ # # ][ # # ]: 0 : if( ( vmit = varInfo.find( "lon" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
203 : : {
204 [ # # ][ # # ]: 0 : rval = read_coordinate( "lon", lCDims[0], lCDims[3], ilCVals );MB_CHK_SET_ERR( rval, "Trouble reading 'lon' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
205 : : }
206 : : else
207 : : {
208 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'lon' variable" );
[ # # ][ # # ]
[ # # ]
209 : : }
210 : : }
211 : :
212 [ # # ]: 0 : if( -1 != lCDims[1] )
213 : : {
214 [ # # ][ # # ]: 0 : if( ( vmit = varInfo.find( "lat" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
215 : : {
216 [ # # ][ # # ]: 0 : rval = read_coordinate( "lat", lCDims[1], lCDims[4], jlCVals );MB_CHK_SET_ERR( rval, "Trouble reading 'lat' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
217 : : }
218 : : else
219 : : {
220 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'lat' variable" );
[ # # ][ # # ]
[ # # ]
221 : : }
222 : : }
223 : :
224 [ # # ]: 0 : if( -1 != lDims[0] )
225 : : {
226 [ # # ][ # # ]: 0 : if( ( vmit = varInfo.find( "slon" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
227 : : {
228 : : // Last column
229 [ # # ][ # # ]: 0 : if( !locallyPeriodic[0] && globallyPeriodic[0] && lDims[3] > gDims[3] )
[ # # ]
230 : : {
231 [ # # ]: 0 : assert( lDims[3] == gDims[3] + 1 );
232 [ # # ]: 0 : std::vector< double > dummyVar( lDims[3] - lDims[0] );
233 [ # # ]: 0 : rval = read_coordinate( "slon", lDims[0], lDims[3] - 1, dummyVar );
234 [ # # ][ # # ]: 0 : double dif = dummyVar[1] - dummyVar[0];
235 : : std::size_t i;
236 [ # # ]: 0 : for( i = 0; i != dummyVar.size(); i++ )
237 [ # # ][ # # ]: 0 : ilVals[i] = dummyVar[i];
238 [ # # ][ # # ]: 0 : ilVals[i] = ilVals[i - 1] + dif;
239 : : }
240 : : else
241 : : {
242 [ # # ][ # # ]: 0 : rval = read_coordinate( "slon", lDims[0], lDims[3], ilVals );MB_CHK_SET_ERR( rval, "Trouble reading 'slon' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
243 : : }
244 : : }
245 : : else
246 : : {
247 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'slon' variable" );
[ # # ][ # # ]
[ # # ]
248 : : }
249 : : }
250 : :
251 [ # # ]: 0 : if( -1 != lDims[1] )
252 : : {
253 [ # # ][ # # ]: 0 : if( ( vmit = varInfo.find( "slat" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
254 : : {
255 [ # # ][ # # ]: 0 : if( !isParallel || ( ( gDims[4] - gDims[1] ) == ( lDims[4] - lDims[1] ) ) )
256 : : {
257 [ # # ]: 0 : std::vector< double > dummyVar( lDims[4] - lDims[1] - 1 );
258 [ # # ][ # # ]: 0 : rval = read_coordinate( "slat", lDims[1], lDims[4] - 2, dummyVar );MB_CHK_SET_ERR( rval, "Trouble reading 'slat' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
259 : : // Copy the correct piece
260 [ # # ]: 0 : jlVals[0] = -90.0;
261 : 0 : std::size_t i = 0;
262 [ # # ]: 0 : for( i = 1; i != dummyVar.size() + 1; i++ )
263 [ # # ][ # # ]: 0 : jlVals[i] = dummyVar[i - 1];
264 [ # # ][ # # ]: 0 : jlVals[i] = 90.0; // Using value of i after loop exits.
265 : : }
266 : : else
267 : : {
268 : : // If this is the first row
269 : : // Need to read one less then available and read it into a dummy var
270 [ # # ]: 0 : if( lDims[1] == gDims[1] )
271 : : {
272 [ # # ]: 0 : std::vector< double > dummyVar( lDims[4] - lDims[1] );
273 [ # # ][ # # ]: 0 : rval = read_coordinate( "slat", lDims[1], lDims[4] - 1, dummyVar );MB_CHK_SET_ERR( rval, "Trouble reading 'slat' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
274 : : // Copy the correct piece
275 [ # # ]: 0 : jlVals[0] = -90.0;
276 [ # # ][ # # ]: 0 : for( int i = 1; i < lDims[4] + 1; i++ )
277 [ # # ][ # # ]: 0 : jlVals[i] = dummyVar[i - 1];
278 : : }
279 : : // Or if it's the last row
280 [ # # ]: 0 : else if( lDims[4] == gDims[4] )
281 : : {
282 [ # # ]: 0 : std::vector< double > dummyVar( lDims[4] - lDims[1] );
283 [ # # ][ # # ]: 0 : rval = read_coordinate( "slat", lDims[1] - 1, lDims[4] - 2, dummyVar );MB_CHK_SET_ERR( rval, "Trouble reading 'slat' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
284 : : // Copy the correct piece
285 : 0 : std::size_t i = 0;
286 [ # # ]: 0 : for( i = 0; i != dummyVar.size(); i++ )
287 [ # # ][ # # ]: 0 : jlVals[i] = dummyVar[i];
288 [ # # ][ # # ]: 0 : jlVals[i] = 90.0; // Using value of i after loop exits.
289 : : }
290 : : // It's in the middle
291 : : else
292 : : {
293 [ # # ][ # # ]: 0 : rval = read_coordinate( "slat", lDims[1] - 1, lDims[4] - 1, jlVals );MB_CHK_SET_ERR( rval, "Trouble reading 'slat' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
294 : : }
295 : : }
296 : : }
297 : : else
298 : : {
299 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Couldn't find 'slat' variable" );
[ # # ][ # # ]
[ # # ]
300 : : }
301 : : }
302 : :
303 : : // Store time coordinate values in tVals
304 [ # # ]: 0 : if( nTimeSteps > 0 )
305 : : {
306 [ # # ][ # # ]: 0 : if( ( vmit = varInfo.find( "time" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
307 : : {
308 [ # # ][ # # ]: 0 : rval = read_coordinate( "time", 0, nTimeSteps - 1, tVals );MB_CHK_SET_ERR( rval, "Trouble reading 'time' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
309 : : }
310 [ # # ][ # # ]: 0 : else if( ( vmit = varInfo.find( "t" ) ) != varInfo.end() && ( *vmit ).second.varDims.size() == 1 )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # #
# ]
311 : : {
312 [ # # ][ # # ]: 0 : rval = read_coordinate( "t", 0, nTimeSteps - 1, tVals );MB_CHK_SET_ERR( rval, "Trouble reading 't' variable" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
313 : : }
314 : : else
315 : : {
316 : : // If expected time variable is not available, set dummy time coordinate values to tVals
317 [ # # ]: 0 : for( int t = 0; t < nTimeSteps; t++ )
318 [ # # ]: 0 : tVals.push_back( (double)t );
319 : : }
320 : : }
321 : :
322 [ # # ]: 0 : dbgOut.tprintf( 1, "I=%d-%d, J=%d-%d\n", lDims[0], lDims[3], lDims[1], lDims[4] );
323 : 0 : dbgOut.tprintf( 1, "%d elements, %d vertices\n", ( lDims[3] - lDims[0] ) * ( lDims[4] - lDims[1] ),
324 [ # # ]: 0 : ( lDims[3] - lDims[0] + 1 ) * ( lDims[4] - lDims[1] + 1 ) );
325 : :
326 : : // For each variable, determine the entity location type and number of levels
327 [ # # ]: 0 : std::map< std::string, ReadNC::VarData >::iterator mit;
328 [ # # ][ # # ]: 0 : for( mit = varInfo.begin(); mit != varInfo.end(); ++mit )
[ # # ]
329 : : {
330 [ # # ]: 0 : ReadNC::VarData& vd = ( *mit ).second;
331 : :
332 : : // Default entLoc is ENTLOCSET
333 [ # # ][ # # ]: 0 : if( std::find( vd.varDims.begin(), vd.varDims.end(), tDim ) != vd.varDims.end() )
[ # # ]
334 : : {
335 [ # # ][ # # ]: 0 : if( ( std::find( vd.varDims.begin(), vd.varDims.end(), iCDim ) != vd.varDims.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # # #
# # ]
336 [ # # ][ # # ]: 0 : ( std::find( vd.varDims.begin(), vd.varDims.end(), jCDim ) != vd.varDims.end() ) )
[ # # ][ # # ]
[ # # # # ]
337 : 0 : vd.entLoc = ReadNC::ENTLOCFACE;
338 [ # # ][ # # ]: 0 : else if( ( std::find( vd.varDims.begin(), vd.varDims.end(), jDim ) != vd.varDims.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # # #
# # ]
339 [ # # ][ # # ]: 0 : ( std::find( vd.varDims.begin(), vd.varDims.end(), iCDim ) != vd.varDims.end() ) )
[ # # ][ # # ]
[ # # # # ]
340 : 0 : vd.entLoc = ReadNC::ENTLOCNSEDGE;
341 [ # # ][ # # ]: 0 : else if( ( std::find( vd.varDims.begin(), vd.varDims.end(), jCDim ) != vd.varDims.end() ) &&
[ # # ][ # # ]
[ # # ][ # # ]
[ # # # #
# # ]
342 [ # # ][ # # ]: 0 : ( std::find( vd.varDims.begin(), vd.varDims.end(), iDim ) != vd.varDims.end() ) )
[ # # ][ # # ]
[ # # # # ]
343 : 0 : vd.entLoc = ReadNC::ENTLOCEWEDGE;
344 : : }
345 : :
346 : : // Default numLev is 0
347 [ # # ][ # # ]: 0 : if( std::find( vd.varDims.begin(), vd.varDims.end(), levDim ) != vd.varDims.end() ) vd.numLev = nLevels;
[ # # ]
348 : : }
349 : :
350 [ # # ]: 0 : std::vector< std::string > ijdimNames( 4 );
351 [ # # ][ # # ]: 0 : ijdimNames[0] = "__slon";
352 [ # # ][ # # ]: 0 : ijdimNames[1] = "__slat";
353 [ # # ][ # # ]: 0 : ijdimNames[2] = "__lon";
354 [ # # ][ # # ]: 0 : ijdimNames[3] = "__lat";
355 : :
356 [ # # ]: 0 : std::string tag_name;
357 : : Tag tagh;
358 : :
359 : : // __<dim_name>_LOC_MINMAX (for slon, slat, lon and lat)
360 [ # # ]: 0 : for( unsigned int i = 0; i != ijdimNames.size(); i++ )
361 : : {
362 [ # # ]: 0 : std::vector< int > val( 2, 0 );
363 [ # # ][ # # ]: 0 : if( ijdimNames[i] == "__slon" )
[ # # ]
364 : : {
365 [ # # ]: 0 : val[0] = lDims[0];
366 [ # # ]: 0 : val[1] = lDims[3];
367 : : }
368 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__slat" )
[ # # ]
369 : : {
370 [ # # ]: 0 : val[0] = lDims[1];
371 [ # # ]: 0 : val[1] = lDims[4];
372 : : }
373 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lon" )
[ # # ]
374 : : {
375 [ # # ]: 0 : val[0] = lCDims[0];
376 [ # # ]: 0 : val[1] = lCDims[3];
377 : : }
378 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lat" )
[ # # ]
379 : : {
380 [ # # ]: 0 : val[0] = lCDims[1];
381 [ # # ]: 0 : val[1] = lCDims[4];
382 : : }
383 [ # # ][ # # ]: 0 : std::stringstream ss_tag_name;
[ # # ]
384 [ # # ][ # # ]: 0 : ss_tag_name << ijdimNames[i] << "_LOC_MINMAX";
[ # # ]
385 [ # # ][ # # ]: 0 : tag_name = ss_tag_name.str();
386 [ # # ][ # # ]: 0 : rval = mbImpl->tag_get_handle( tag_name.c_str(), 2, MB_TYPE_INTEGER, tagh, MB_TAG_SPARSE | MB_TAG_CREAT );MB_CHK_SET_ERR( rval, "Trouble creating conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
387 [ # # ][ # # ]: 0 : rval = mbImpl->tag_set_data( tagh, &_fileSet, 1, &val[0] );MB_CHK_SET_ERR( rval, "Trouble setting data to conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
388 [ # # ][ # # ]: 0 : if( MB_SUCCESS == rval ) dbgOut.tprintf( 2, "Conventional tag %s is created.\n", tag_name.c_str() );
[ # # ]
389 : 0 : }
390 : :
391 : : // __<dim_name>_LOC_VALS (for slon, slat, lon and lat)
392 : : // Assume all have the same data type as lon (expected type is float or double)
393 [ # # ][ # # ]: 0 : switch( varInfo["lon"].varDataType )
[ # # ]
394 : : {
395 : : case NC_FLOAT:
396 : : case NC_DOUBLE:
397 : 0 : break;
398 : : default:
399 [ # # ][ # # ]: 0 : MB_SET_ERR( MB_FAILURE, "Unexpected variable data type for 'lon'" );
[ # # ][ # # ]
[ # # ]
400 : : }
401 : :
402 [ # # ]: 0 : for( unsigned int i = 0; i != ijdimNames.size(); i++ )
403 : : {
404 : 0 : void* val = NULL;
405 : 0 : int val_len = 0;
406 [ # # ][ # # ]: 0 : if( ijdimNames[i] == "__slon" )
[ # # ]
407 : : {
408 [ # # ]: 0 : val = &ilVals[0];
409 : 0 : val_len = ilVals.size();
410 : : }
411 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__slat" )
[ # # ]
412 : : {
413 [ # # ]: 0 : val = &jlVals[0];
414 : 0 : val_len = jlVals.size();
415 : : }
416 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lon" )
[ # # ]
417 : : {
418 [ # # ]: 0 : val = &ilCVals[0];
419 : 0 : val_len = ilCVals.size();
420 : : }
421 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lat" )
[ # # ]
422 : : {
423 [ # # ]: 0 : val = &jlCVals[0];
424 : 0 : val_len = jlCVals.size();
425 : : }
426 : :
427 [ # # ][ # # ]: 0 : std::stringstream ss_tag_name;
428 [ # # ][ # # ]: 0 : ss_tag_name << ijdimNames[i] << "_LOC_VALS";
[ # # ]
429 [ # # ][ # # ]: 0 : tag_name = ss_tag_name.str();
430 : : rval = mbImpl->tag_get_handle( tag_name.c_str(), 0, MB_TYPE_DOUBLE, tagh,
431 [ # # ][ # # ]: 0 : MB_TAG_CREAT | MB_TAG_SPARSE | MB_TAG_VARLEN );MB_CHK_SET_ERR( rval, "Trouble creating conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
432 [ # # ][ # # ]: 0 : rval = mbImpl->tag_set_by_ptr( tagh, &_fileSet, 1, &val, &val_len );MB_CHK_SET_ERR( rval, "Trouble setting data to conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
433 [ # # ][ # # ]: 0 : if( MB_SUCCESS == rval ) dbgOut.tprintf( 2, "Conventional tag %s is created.\n", tag_name.c_str() );
[ # # ]
434 : 0 : }
435 : :
436 : : // __<dim_name>_GLOBAL_MINMAX (for slon, slat, lon and lat)
437 [ # # ]: 0 : for( unsigned int i = 0; i != ijdimNames.size(); i++ )
438 : : {
439 [ # # ]: 0 : std::vector< int > val( 2, 0 );
440 [ # # ][ # # ]: 0 : if( ijdimNames[i] == "__slon" )
[ # # ]
441 : : {
442 [ # # ]: 0 : val[0] = gDims[0];
443 [ # # ]: 0 : val[1] = gDims[3];
444 : : }
445 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__slat" )
[ # # ]
446 : : {
447 [ # # ]: 0 : val[0] = gDims[1];
448 [ # # ]: 0 : val[1] = gDims[4];
449 : : }
450 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lon" )
[ # # ]
451 : : {
452 [ # # ]: 0 : val[0] = gCDims[0];
453 [ # # ]: 0 : val[1] = gCDims[3];
454 : : }
455 [ # # ][ # # ]: 0 : else if( ijdimNames[i] == "__lat" )
[ # # ]
456 : : {
457 [ # # ]: 0 : val[0] = gCDims[1];
458 [ # # ]: 0 : val[1] = gCDims[4];
459 : : }
460 [ # # ][ # # ]: 0 : std::stringstream ss_tag_name;
[ # # ]
461 [ # # ][ # # ]: 0 : ss_tag_name << ijdimNames[i] << "_GLOBAL_MINMAX";
[ # # ]
462 [ # # ][ # # ]: 0 : tag_name = ss_tag_name.str();
463 [ # # ][ # # ]: 0 : rval = mbImpl->tag_get_handle( tag_name.c_str(), 2, MB_TYPE_INTEGER, tagh, MB_TAG_SPARSE | MB_TAG_CREAT );MB_CHK_SET_ERR( rval, "Trouble creating conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
464 [ # # ][ # # ]: 0 : rval = mbImpl->tag_set_data( tagh, &_fileSet, 1, &val[0] );MB_CHK_SET_ERR( rval, "Trouble setting data to conventional tag " << tag_name );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
465 [ # # ][ # # ]: 0 : if( MB_SUCCESS == rval ) dbgOut.tprintf( 2, "Conventional tag %s is created.\n", tag_name.c_str() );
[ # # ]
466 : 0 : }
467 : :
468 : : // Hack: create dummy variables, if needed, for dimensions with no corresponding coordinate
469 : : // variables
470 [ # # ][ # # ]: 0 : rval = create_dummy_variables();MB_CHK_SET_ERR( rval, "Failed to create dummy variables" );
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
471 : :
472 : 0 : return MB_SUCCESS;
473 : : }
474 : :
475 [ + - ][ + - ]: 228 : } // namespace moab
|
