source: XIOS/dev/dev_olga/src/node/grid.hpp @ 1568

Last change on this file since 1568 was 1568, checked in by oabramkina, 3 years ago

Grid mask is now applied in the source filter of clients: unmasked values are replaced by NaN. It is not reconstructed any more by servers.

This needs to be tested more rigorously before commiting to trunk.

  • Property copyright set to
    Software name : XIOS (Xml I/O Server)
    http://forge.ipsl.jussieu.fr/ioserver
    Creation date : January 2009
    Licence : CeCCIL version2
    see license file in root directory : Licence_CeCILL_V2-en.txt
    or http://www.cecill.info/licences/Licence_CeCILL_V2-en.html
    Holder : CEA/LSCE (Laboratoire des Sciences du CLimat et de l'Environnement)
    CNRS/IPSL (Institut Pierre Simon Laplace)
    Project Manager : Yann Meurdesoif
    yann.meurdesoif@cea.fr
  • Property svn:executable set to *
File size: 24.0 KB
Line 
1#ifndef __XIOS_CGrid__
2#define __XIOS_CGrid__
3
4/// XIOS headers ///
5#include "xios_spl.hpp"
6#include "group_factory.hpp"
7
8#include "declare_group.hpp"
9#include "domain.hpp"
10#include "axis.hpp"
11#include "scalar.hpp"
12#include "array_new.hpp"
13#include "attribute_array.hpp"
14#include "distribution_server.hpp"
15#include "client_server_mapping.hpp"
16#include "utils.hpp"
17#include "transformation_enum.hpp"
18
19namespace xios {
20
21   /// ////////////////////// Declarations ////////////////////// ///
22
23   class CGridGroup;
24   class CGridAttributes;
25   class CDomainGroup;
26   class CAxisGroup;
27   class CScalarGroup;
28   class CGrid;
29   class CDistributionClient;
30   class CDistributionServer;
31   class CServerDistributionDescription;
32   class CClientServerMapping;
33   class CGridTransformation;
34
35   ///--------------------------------------------------------------
36
37   // Declare/Define CGridAttribute
38   BEGIN_DECLARE_ATTRIBUTE_MAP(CGrid)
39#  include "grid_attribute.conf"
40   END_DECLARE_ATTRIBUTE_MAP(CGrid)
41
42   ///--------------------------------------------------------------
43
44   class CGrid
45      : public CObjectTemplate<CGrid>
46      , public CGridAttributes
47   {
48         /// typedef ///
49         typedef CObjectTemplate<CGrid>   SuperClass;
50         typedef CGridAttributes SuperClassAttribute;
51
52      public:
53
54         typedef CGridAttributes RelAttributes;
55         typedef CGridGroup      RelGroup;
56
57         enum EEventId
58         {
59           EVENT_ID_INDEX, EVENT_ID_ADD_DOMAIN, EVENT_ID_ADD_AXIS, EVENT_ID_ADD_SCALAR
60         };
61
62         /// Constructeurs ///
63         CGrid(void);
64         explicit CGrid(const StdString& id);
65         CGrid(const CGrid& grid);       // Not implemented yet.
66         CGrid(const CGrid* const grid); // Not implemented yet.
67
68         /// Traitements ///
69//         void solveReference(void);
70
71         void checkEligibilityForCompressedOutput();
72
73         void solveDomainAxisRef(bool areAttributesChecked);
74
75         void checkMaskIndex(bool doCalculateIndex);
76
77 //        virtual void toBinary  (StdOStream& os) const;
78//         virtual void fromBinary(StdIStream& is);
79
80         void addRelFileCompressed(const StdString& filename);
81
82         /// Tests ///
83         bool isCompressible(void) const;
84         bool isWrittenCompressed(const StdString& filename) const;
85
86      public:
87
88         /// Accesseurs ///
89         StdSize getDimension(void);
90
91         StdSize  getDataSize(void) const;
92
93         /// Entrees-sorties de champs
94         template <int n>
95         void inputField(const CArray<double,n>& field, CArray<double,1>& stored) const;
96         template <int n>
97         void maskField(const CArray<double,n>& field, CArray<double,1>& stored) const;
98         template <int n>
99         void outputField(const CArray<double,1>& stored, CArray<double,n>& field) const; 
100         template <int n>
101         void uncompressField(const CArray<double,n>& data, CArray<double,1>& outData) const; 
102
103         virtual void parse(xml::CXMLNode& node);
104
105         /// Destructeur ///
106         virtual ~CGrid(void);
107
108      public:
109
110         /// Accesseurs statiques ///
111         static StdString GetName(void);
112         static StdString GetDefName(void);
113
114         static ENodeType GetType(void);
115
116         /// Instanciateurs Statiques ///
117         static CGrid* createGrid(CDomain* domain);
118         static CGrid* createGrid(CDomain* domain, CAxis* axis);
119         static CGrid* createGrid(const std::vector<CDomain*>& domains, const std::vector<CAxis*>& axis,
120                                  const CArray<int,1>& axisDomainOrder = CArray<int,1>());
121         static CGrid* createGrid(StdString id, const std::vector<CDomain*>& domains, const std::vector<CAxis*>& axis,
122                                  const std::vector<CScalar*>& scalars, const CArray<int,1>& axisDomainOrder = CArray<int,1>());
123         static CGrid* createGrid(const std::vector<CDomain*>& domains, const std::vector<CAxis*>& axis,
124                                  const std::vector<CScalar*>& scalars, const CArray<int,1>& axisDomainOrder);
125         static StdString generateId(const std::vector<CDomain*>& domains, const std::vector<CAxis*>& axis,
126                                     const std::vector<CScalar*>& scalars, const CArray<int,1>& axisDomainOrder = CArray<int,1>());
127         static StdString generateId(const CGrid* gridSrc, const CGrid* gridDest);
128         static CGrid* cloneGrid(const StdString& idNewGrid, CGrid* gridSrc);
129
130      public:           
131         void computeIndexServer(void);
132         void computeIndex(void);
133         void computeIndexScalarGrid();
134         void computeWrittenIndex();
135
136         void solveDomainRef(bool checkAtt);
137         void solveAxisRef(bool checkAtt);
138         void solveScalarRef(bool checkAtt);
139         void solveDomainAxisRefInheritance(bool apply = true);
140         void solveTransformations();
141         void solveDomainAxisBaseRef();
142
143         CDomain* addDomain(const std::string& id=StdString());
144         CAxis* addAxis(const std::string& id=StdString());
145         CScalar* addScalar(const std::string& id=StdString());
146         void sendAddDomain(const std::string& id="");
147         void sendAddAxis(const std::string& id="");
148         void sendAddScalar(const std::string& id="");
149         void sendAllDomains();
150         void sendAllAxis();
151         void sendAllScalars();
152
153         static void recvAddDomain(CEventServer& event);
154         void recvAddDomain(CBufferIn& buffer);
155         static void recvAddAxis(CEventServer& event);
156         void recvAddAxis(CBufferIn& buffer);
157         static void recvAddScalar(CEventServer& event);
158         void recvAddScalar(CBufferIn& buffer);
159
160         static bool dispatchEvent(CEventServer& event);
161         static void recvIndex(CEventServer& event);
162         void recvIndex(vector<int> ranks, vector<CBufferIn*> buffers);
163         void sendIndex(void);
164         void sendIndexScalarGrid();
165
166         void setContextClient(CContextClient* contextClient);
167
168         void computeDomConServer();
169         std::map<int, int> getDomConServerSide();
170         std::map<int, StdSize> getAttributesBufferSize(CContextClient* client, bool bufferForWriting = false);
171         std::map<int, StdSize> getDataBufferSize(CContextClient* client, const std::string& id = "", bool bufferForWriting = false);
172         std::vector<StdString> getDomainList();
173         std::vector<StdString> getAxisList();
174         std::vector<StdString> getScalarList();
175         std::vector<CDomain*> getDomains();
176         std::vector<CAxis*> getAxis();
177         std::vector<CScalar*> getScalars();
178         CDomain* getDomain(int domainIndex);
179         CAxis* getAxis(int axisIndex);
180         CScalar* getScalar(int scalarIndex);
181         std::vector<int> getAxisOrder();
182         std::vector<int> getGlobalDimension();
183         bool isScalarGrid() const;         
184
185         bool doGridHaveDataToWrite();
186         bool doGridHaveDataDistributed(CContextClient* client = 0);
187         size_t getWrittenDataSize() const;
188         int getNumberWrittenIndexes() const;
189         int getTotalNumberWrittenIndexes() const;
190         int getOffsetWrittenIndexes() const;
191
192         CDistributionServer* getDistributionServer();
193         CDistributionClient* getDistributionClient();
194         CGridTransformation* getTransformations();
195
196         void transformGrid(CGrid* transformGridSrc);
197         void completeGrid(CGrid* transformGridSrc = 0);
198         void doAutoDistribution(CGrid* transformGridSrc);
199         bool isTransformed();
200         void setTransformed();
201         bool isGenerated();
202         void setGenerated();
203         void addTransGridSource(CGrid* gridSrc);
204         std::map<CGrid*, std::pair<bool,StdString> >& getTransGridSource();
205         bool hasTransform();
206         size_t getGlobalWrittenSize(void) ;
207         void getLocalMask(CArray<bool,1>& localMask) ;
208         template<int N>
209         void getLocalMask(const CArray<bool,N>& gridMask, CArray<bool,1>& localMask) ;
210      public:
211         CArray<int, 1> storeIndex_client;
212         CArray<bool, 1> storeMask_client;
213
214/** Map containing indexes that will be sent in sendIndex(). */
215         std::map<CContextClient*, map<int, CArray<int, 1> > > storeIndex_toSrv;
216
217/** Map storing the number of senders. Key = size of receiver's intracomm */
218         std::map<int, std::map<int,int> > nbSenders;
219
220         std::map<CContextClient*, std::map<int,int> > nbReadSenders;
221
222         map<int, CArray<int, 1> > storeIndex_fromSrv; // Support, for now, reading with level-1 server
223
224         map<int, CArray<size_t, 1> > outIndexFromClient;  // Deprecated
225
226         map<int, CArray<size_t, 1> > compressedOutIndexFromClient;
227
228/** Map storing received indexes. Key = sender rank, value = index array. */
229         map<int, CArray<size_t, 1> > outGlobalIndexFromClient;
230
231// Manh Ha's comment: " A client receives global index from other clients (via recvIndex)
232// then does mapping these index into local index of STORE_CLIENTINDEX
233// In this way, store_clientIndex can be used as an input of a source filter
234// Maybe we need a flag to determine whether a client wants to write. TODO "
235
236/** Map storing received data. Key = sender rank, value = data array.
237 *  The map is created in CGrid::computeClientIndex and filled upon receiving data in CField::recvUpdateData() */
238         map<int, CArray<size_t, 1> > outLocalIndexStoreOnClient; 
239
240/** Indexes calculated based on server-like distribution.
241 *  They are used for writing/reading data and only calculated for server level that does the writing/reading.
242 *  Along with localIndexToWriteOnClient, these indexes are used to correctly place incoming data. */
243         CArray<size_t,1> localIndexToWriteOnServer;
244
245/** Indexes calculated based on client-like distribution.
246 *  They are used for writing/reading data and only calculated for server level that does the writing/reading.
247 *  Along with localIndexToWriteOnServer, these indexes are used to correctly place incoming data. */
248         CArray<size_t,1> localIndexToWriteOnClient;
249
250         CArray<size_t,1> indexFromClients;
251
252         void checkMask(void);
253         void createMask(void);
254         void modifyMask(const CArray<int,1>& indexToModify, bool valueToModify = false);
255         void modifyMaskSize(const std::vector<int>& newDimensionSize, bool newValue = false);
256
257         void computeGridGlobalDimension(const std::vector<CDomain*>& domains,
258                                         const std::vector<CAxis*>& axis,
259                                         const std::vector<CScalar*>& scalars,
260                                         const CArray<int,1>& axisDomainOrder);
261
262      private:
263       template<int N>
264       void checkGridMask(CArray<bool,N>& gridMask,
265                          const std::vector<CArray<bool,1>* >& domainMasks,
266                          const std::vector<CArray<bool,1>* >& axisMasks,
267                          const CArray<int,1>& axisDomainOrder,
268                          bool createMask = false);
269        template<int N>
270        void modifyGridMask(CArray<bool,N>& gridMask, const CArray<int,1>& indexToModify, bool valueToModify);
271
272        template<int N>
273        void modifyGridMaskSize(CArray<bool,N>& gridMask, const std::vector<int>& eachDimSize, bool newValue);
274
275        void storeField_arr(const double* const data, CArray<double, 1>& stored) const;
276        void restoreField_arr(const CArray<double, 1>& stored, double* const data) const;
277        void uncompressField_arr(const double* const data, CArray<double, 1>& outData) const;
278        void maskField_arr(const double* const data, CArray<double, 1>& stored) const;
279
280        void setVirtualDomainGroup(CDomainGroup* newVDomainGroup);
281        void setVirtualAxisGroup(CAxisGroup* newVAxisGroup);
282        void setVirtualScalarGroup(CScalarGroup* newVScalarGroup);
283
284        void setDomainList(const std::vector<CDomain*> domains = std::vector<CDomain*>());
285        void setAxisList(const std::vector<CAxis*> axis = std::vector<CAxis*>());
286        void setScalarList(const std::vector<CScalar*> scalars = std::vector<CScalar*>());
287
288        CDomainGroup* getVirtualDomainGroup() const;
289        CAxisGroup* getVirtualAxisGroup() const;
290        CScalarGroup* getVirtualScalarGroup() const;
291
292        void checkAttributesAfterTransformation();
293        void setTransformationAlgorithms();
294        void computeIndexByElement(const std::vector<std::unordered_map<size_t,std::vector<int> > >& indexServerOnElement,
295                                   const CContextClient* client,
296                                   CClientServerMapping::GlobalIndexMap& globalIndexOnServer);
297        int computeGridGlobalDimension(std::vector<int>& globalDim,
298                                       const std::vector<CDomain*> domains,
299                                       const std::vector<CAxis*> axis,
300                                       const std::vector<CScalar*> scalars,
301                                       const CArray<int,1>& axisDomainOrder);
302        int getDistributedDimension();
303
304        void computeClientIndex();
305        void computeConnectedClients();
306        void computeClientIndexScalarGrid(); 
307        void computeConnectedClientsScalarGrid(); 
308
309      private:
310
311/** Clients that have to send a grid. There can be multiple clients in case of secondary server, otherwise only one client. */
312        std::list<CContextClient*> clients;
313        std::set<CContextClient*> clientsSet;
314
315        bool isChecked;
316        bool isDomainAxisChecked;
317        bool isIndexSent;
318
319        CDomainGroup* vDomainGroup_;
320        CAxisGroup* vAxisGroup_;
321        CScalarGroup* vScalarGroup_;
322        std::vector<std::string> axisList_, domList_, scalarList_;
323        bool isAxisListSet, isDomListSet, isScalarListSet;
324
325/** Client-like distribution calculated based on the knowledge of the entire grid */
326        CDistributionClient* clientDistribution_;
327
328/** Server-like distribution calculated upon receiving indexes */
329        CDistributionServer* serverDistribution_;
330
331        CClientServerMapping* clientServerMap_;
332        size_t writtenDataSize_;
333        int numberWrittenIndexes_, totalNumberWrittenIndexes_, offsetWrittenIndexes_;
334
335/** Map storing local ranks of connected receivers. Key = size of receiver's intracomm.
336  * It is calculated in computeConnectedClients(). */
337        std::map<int, std::vector<int> > connectedServerRank_;
338
339/** Map storing the size of data to be send. Key = size of receiver's intracomm
340  * It is calculated in computeConnectedClients(). */
341        std::map<int, std::map<int,size_t> > connectedDataSize_;
342
343/** Ranks of connected receivers in case of reading. It is calculated in recvIndex(). */
344        std::vector<int> connectedServerRankRead_;
345
346/** Size of data to be send in case of reading. It is calculated in recvIndex(). */
347        std::map<int,size_t> connectedDataSizeRead_;
348
349        bool isDataDistributed_;       
350         //! True if and only if the data defined on the grid can be outputted in a compressed way
351        bool isCompressible_;
352        std::set<std::string> relFilesCompressed;
353
354        bool isTransformed_, isGenerated_;
355        bool computedWrittenIndex_;
356        std::vector<int> axisPositionInGrid_;
357        CGridTransformation* transformations_;
358        bool hasDomainAxisBaseRef_;       
359        std::map<CGrid*, std::pair<bool,StdString> > gridSrc_;
360        bool hasTransform_;
361
362/** Map storing global indexes of server-like (band-wise) distribution for sending to receivers.
363  * Key = size of receiver's intracomm.
364  */
365//        std::map<CContextClient*, CClientServerMapping::GlobalIndexMap> globalIndexOnServer_;
366        std::map<int, CClientServerMapping::GlobalIndexMap> globalIndexOnServer_;
367
368
369/** List order of axis and domain in a grid, if there is a domain, it will take value 1 (true), axis 0 (false) */
370        std::vector<int> order_;
371
372   }; // class CGrid
373
374   ///--------------------------------------------------------------
375
376   template <int n>
377   void CGrid::inputField(const CArray<double,n>& field, CArray<double,1>& stored) const
378   {
379//#ifdef __XIOS_DEBUG
380      if (this->getDataSize() != field.numElements())
381         ERROR("void CGrid::inputField(const  CArray<double,n>& field, CArray<double,1>& stored) const",
382                << "[ Awaiting data of size = " << this->getDataSize() << ", "
383                << "Received data size = "      << field.numElements() << " ] "
384                << "The data array does not have the right size! "
385                << "Grid = " << this->getId())
386//#endif
387      this->storeField_arr(field.dataFirst(), stored);
388   }
389
390   template <int n>
391   void CGrid::maskField(const CArray<double,n>& field, CArray<double,1>& stored) const
392   {
393//#ifdef __XIOS_DEBUG
394      if (this->getDataSize() != field.numElements())
395         ERROR("void CGrid::inputField(const  CArray<double,n>& field, CArray<double,1>& stored) const",
396                << "[ Awaiting data of size = " << this->getDataSize() << ", "
397                << "Received data size = "      << field.numElements() << " ] "
398                << "The data array does not have the right size! "
399                << "Grid = " << this->getId())
400//#endif
401      this->maskField_arr(field.dataFirst(), stored);
402   }
403
404   template <int n>
405   void CGrid::outputField(const CArray<double,1>& stored, CArray<double,n>& field) const
406   {
407//#ifdef __XIOS_DEBUG
408      if (this->getDataSize() != field.numElements())
409         ERROR("void CGrid::outputField(const CArray<double,1>& stored, CArray<double,n>& field) const",
410                << "[ Size of the data = " << this->getDataSize() << ", "
411                << "Output data size = "   << field.numElements() << " ] "
412                << "The ouput array does not have the right size! "
413                << "Grid = " << this->getId())
414//#endif
415      this->restoreField_arr(stored, field.dataFirst());
416   }
417
418   /*!
419     This function removes the effect of mask on received data on the server.
420     This function only serve for the checking purpose. TODO: Something must be done to seperate mask and data_index from each other in received data
421     \data data received data with masking effect on the server
422     \outData data without masking effect
423   */
424   template <int N>
425   void CGrid::uncompressField(const CArray<double,N>& data, CArray<double,1>& outData) const
426   {     
427     uncompressField_arr(data.dataFirst(), outData);
428   }
429
430   template<int N>
431   void CGrid::checkGridMask(CArray<bool,N>& gridMask,
432                             const std::vector<CArray<bool,1>* >& domainMasks,
433                             const std::vector<CArray<bool,1>* >& axisMasks,
434                             const CArray<int,1>& axisDomainOrder,
435                             bool createMask)
436   {
437     int idx = 0;
438     int numElement = axisDomainOrder.numElements();
439     int dim = domainMasks.size() * 2 + axisMasks.size();
440     std::vector<CDomain*> domainP = this->getDomains();
441     std::vector<CAxis*> axisP = this->getAxis();
442
443     std::vector<int> idxLoop(dim,0), indexMap(numElement), eachDimSize(dim);
444     std::vector<int> currentIndex(dim);
445     int idxDomain = 0, idxAxis = 0;
446    for (int i = 0; i < numElement; ++i)
447    {
448      indexMap[i] = idx;
449      if (2 == axisDomainOrder(i)) {
450          eachDimSize[indexMap[i]]   = domainP[idxDomain]->ni;
451          eachDimSize[indexMap[i]+1] = domainP[idxDomain]->nj;
452          idx += 2; ++idxDomain;
453      }
454      else if (1 == axisDomainOrder(i)) {
455//        eachDimSize[indexMap[i]] = axisMasks[idxAxis]->numElements();
456        eachDimSize[indexMap[i]] = axisP[idxAxis]->n;
457        ++idx; ++idxAxis;
458      }
459      else {};
460    }
461
462    if (!gridMask.isEmpty() && !createMask)
463    {
464      for (int i = 0; i < dim; ++i)
465      {
466        if (gridMask.extent(i) != eachDimSize[i])
467          ERROR("CGrid::checkMask(void)",
468                << "The mask has one dimension whose size is different from the one of the local grid." << std::endl
469                << "Local size of dimension " << i << " is " << eachDimSize[i] << "." << std::endl
470                << "Mask size for dimension " << i << " is " << gridMask.extent(i) << "." << std::endl
471                << "Grid = " << this->getId())
472      }
473    }
474    else {
475        CArrayBoolTraits<CArray<bool,N> >::resizeArray(gridMask,eachDimSize);
476        gridMask = true;
477    }
478
479    int ssize = gridMask.numElements();
480    idx = 0;
481    while (idx < ssize)
482    {
483      for (int i = 0; i < dim-1; ++i)
484      {
485        if (idxLoop[i] == eachDimSize[i])
486        {
487          idxLoop[i] = 0;
488          ++idxLoop[i+1];
489        }
490      }
491
492      // Find out outer index
493      idxDomain = idxAxis = 0;
494      bool maskValue = true;
495      for (int i = 0; i < numElement; ++i)
496      {
497        if (2 == axisDomainOrder(i))
498        {
499          int idxTmp = idxLoop[indexMap[i]] + idxLoop[indexMap[i]+1] * eachDimSize[indexMap[i]];
500          if (idxTmp < (*domainMasks[idxDomain]).numElements())
501            maskValue = maskValue && (*domainMasks[idxDomain])(idxTmp);
502          else
503            maskValue = false;
504          ++idxDomain;
505        }
506        else if (1 == axisDomainOrder(i))
507        {
508          int idxTmp = idxLoop[indexMap[i]];
509          if (idxTmp < (*axisMasks[idxAxis]).numElements())
510            maskValue = maskValue && (*axisMasks[idxAxis])(idxTmp);
511          else
512            maskValue = false;
513
514          ++idxAxis;
515        }
516      }
517
518      int maskIndex = idxLoop[0];
519      int mulDim = 1;
520      for (int k = 1; k < dim; ++k)
521      {
522        mulDim *= eachDimSize[k-1];
523        maskIndex += idxLoop[k]*mulDim;
524      }
525      *(gridMask.dataFirst()+maskIndex) &= maskValue;
526
527      ++idxLoop[0];
528      ++idx;
529    }
530   }
531
532   template<int N>
533   void CGrid::modifyGridMaskSize(CArray<bool,N>& gridMask,
534                                  const std::vector<int>& eachDimSize,
535                                  bool newValue)
536   {
537      if (N != eachDimSize.size())
538      {
539        // ERROR("CGrid::modifyGridMaskSize(CArray<bool,N>& gridMask,
540        //                                  const std::vector<int>& eachDimSize,
541        //                                  bool newValue)",
542        //       << "Dimension size of the mask is different from input dimension size." << std::endl
543        //       << "Mask dimension is " << N << "." << std::endl
544        //       << "Input dimension is " << eachDimSize.size() << "." << std::endl
545        //       << "Grid = " << this->getId())
546      }
547      CArrayBoolTraits<CArray<bool,N> >::resizeArray(gridMask,eachDimSize);
548      gridMask = newValue;
549   }
550                                 
551
552   /*!
553     Modify the current mask of grid, the local index to be modified will take value false
554     \param [in/out] gridMask current mask of grid
555     \param [in] indexToModify local index to modify
556   */
557   template<int N>
558   void CGrid::modifyGridMask(CArray<bool,N>& gridMask, const CArray<int,1>& indexToModify, bool valueToModify)
559   {     
560     int num = indexToModify.numElements();
561     for (int idx = 0; idx < num; ++idx)
562     {
563       *(gridMask.dataFirst()+indexToModify(idx)) = valueToModify;
564     }
565   }
566   ///--------------------------------------------------------------
567
568
569/*!
570  A grid can have multiple dimension, so can its mask in the form of multi-dimension array.
571It's not a good idea to store all multi-dimension arrays corresponding to each mask.
572One of the ways is to convert this array into 1-dimension one and every process is taken place on it.
573  \param [in] multi-dimension array grid mask
574*/
575template<int N>
576void CGrid::getLocalMask(const CArray<bool,N>& gridMask, CArray<bool,1>& localMask)
577{
578  if (gridMask.isEmpty()) return ;
579  int dim = gridMask.dimensions();
580  std::vector<int> dimensionSizes(dim);
581  for (int i = 0; i < dim; ++i) dimensionSizes[i] = gridMask.extent(i);
582
583  std::vector<int> idxLoop(dim,0);
584  int ssize = gridMask.numElements(), idx = 0;
585  localMask.resize(ssize);
586  while (idx < ssize)
587  {
588    for (int i = 0; i < dim-1; ++i)
589    {
590      if (idxLoop[i] == dimensionSizes[i])
591      {
592        idxLoop[i] = 0;
593        ++idxLoop[i+1];
594      }
595    }
596
597    int maskIndex = idxLoop[0];
598    int mulDim = 1;
599    for (int k = 1; k < dim; ++k)
600    {
601      mulDim *= dimensionSizes[k-1];
602      maskIndex += idxLoop[k]*mulDim;
603    }
604    localMask(maskIndex) = *(gridMask.dataFirst()+maskIndex);
605
606    ++idxLoop[0];
607    ++idx;
608  }
609}
610
611   // Declare/Define CGridGroup and CGridDefinition
612   DECLARE_GROUP(CGrid);
613
614   ///--------------------------------------------------------------
615
616} // namespace xios
617
618#endif // __XIOS_CGrid__
Note: See TracBrowser for help on using the repository browser.