source: XIOS/trunk/src/node/domain.cpp @ 610

Last change on this file since 610 was 610, checked in by rlacroix, 6 years ago

Domain initialization: Reduce communications and simplify the code.

  • 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
File size: 33.8 KB
Line 
1#include "domain.hpp"
2
3#include "attribute_template.hpp"
4#include "object_template.hpp"
5#include "group_template.hpp"
6
7#include "xios_spl.hpp"
8#include "event_client.hpp"
9#include "event_server.hpp"
10#include "buffer_in.hpp"
11#include "message.hpp"
12#include "type.hpp"
13#include "context.hpp"
14#include "context_client.hpp"
15#include "array_new.hpp"
16#include "server_distribution_description.hpp"
17#include "client_server_mapping_distributed.hpp"
18
19namespace xios {
20
21   /// ////////////////////// Définitions ////////////////////// ///
22
23   CDomain::CDomain(void)
24      : CObjectTemplate<CDomain>(), CDomainAttributes()
25      , isChecked(false), relFiles(), isClientChecked(false), nbConnectedClients_(), indSrv_(), connectedServerRank_()
26      , isDistributed_(false)
27   { /* Ne rien faire de plus */ }
28
29   CDomain::CDomain(const StdString & id)
30      : CObjectTemplate<CDomain>(id), CDomainAttributes()
31      , isChecked(false), relFiles(), isClientChecked(false), nbConnectedClients_(), indSrv_(), connectedServerRank_()
32      , isDistributed_(false)
33         { /* Ne rien faire de plus */ }
34
35   CDomain::~CDomain(void)
36   {
37   }
38
39   ///---------------------------------------------------------------
40
41   const std::set<StdString> & CDomain::getRelFiles(void) const
42   {
43      return (this->relFiles);
44   }
45
46   //----------------------------------------------------------------
47
48   bool CDomain::hasZoom(void) const
49   {
50      return ((this->zoom_ni.getValue() != this->ni_glo.getValue()) &&
51              (this->zoom_nj.getValue() != this->nj_glo.getValue()));
52   }
53
54   //----------------------------------------------------------------
55
56   bool CDomain::isEmpty(void) const
57   {
58      return ((this->zoom_ni_srv == 0) ||
59              (this->zoom_nj_srv == 0));
60   }
61
62   //----------------------------------------------------------------
63
64   bool CDomain::IsWritten(const StdString & filename) const
65   {
66      return (this->relFiles.find(filename) != this->relFiles.end());
67   }
68
69   //----------------------------------------------------------------
70
71   bool CDomain::isDistributed(void) const
72   {
73      return isDistributed_;
74   }
75
76   //----------------------------------------------------------------
77
78   void CDomain::addRelFile(const StdString & filename)
79   {
80      this->relFiles.insert(filename);
81   }
82
83
84   StdString CDomain::GetName(void)   { return (StdString("domain")); }
85   StdString CDomain::GetDefName(void){ return (CDomain::GetName()); }
86   ENodeType CDomain::GetType(void)   { return (eDomain); }
87
88   //----------------------------------------------------------------
89
90   void CDomain::checkDomain(void)
91   {
92      if (!type.isEmpty() && type==type_attr::unstructured)
93      {
94         if (ni_glo.isEmpty() || ni_glo <= 0 )
95         {
96            ERROR("CDomain::checkAttributes(void)",
97               << "[ Id = " << this->getId() << " ] "
98               << "The global domain is badly defined,"
99               << " check the \'ni_glo\'  value !")
100         }
101         nj_glo=ni_glo ;
102         ni_glo=1 ;
103         if (!ni.isEmpty()) nj=ni ;
104         if (!ibegin.isEmpty()) jbegin=ibegin ;
105         if (!iend.isEmpty()) jend=iend ;
106         if (!i_index.isEmpty())
107         {
108          j_index.resize(1,nj) ;
109          for(int i=0;i<ni;i++) j_index(0,i)=i_index(i,0) ;
110          i_index.resize(1,nj) ;
111          for(int j=0;j<nj;j++) i_index(0,j)=0 ;
112         }
113
114         if (!mask.isEmpty())
115         {
116          CArray<int,2> mask_tmp(nj,1) ;
117          mask_tmp = mask ;
118          mask.resize(1,nj) ;
119          for(int j=0;j<nj;j++) mask(0,j)=mask_tmp(j,0) ;
120         }
121         ni=1 ;
122         ibegin=0 ;
123         iend=0 ;
124
125      }
126      else if ((ni_glo.isEmpty() || ni_glo.getValue() <= 0 ) ||
127          (nj_glo.isEmpty() || nj_glo.getValue() <= 0 ))
128      {
129         ERROR("CDomain::checkAttributes(void)",
130               << "[ Id = " << this->getId() << " ] "
131               << "The global domain is badly defined,"
132               << " check the \'ni_glo\' et \'nj_glo\' values !")
133      }
134
135      isDistributed_ = !ibegin.isEmpty() || !iend.isEmpty() || !ni.isEmpty() || !jbegin.isEmpty() || !jend.isEmpty() || !nj.isEmpty();
136
137      checkLocalIDomain();
138      checkLocalJDomain();
139
140      ibegin_client = ibegin; iend_client = iend; ni_client = ni;
141      jbegin_client = jbegin; jend_client = jend; nj_client = nj;
142
143      if (i_index.isEmpty())
144      {
145        i_index.resize(ni,nj);
146        for (int j = 0; j < nj; j++)
147          for (int i = 0; i < ni; i++) i_index(i,j) = i;
148      }
149
150      if (j_index.isEmpty())
151      {
152        j_index.resize(ni,nj);
153        for (int j = 0; j < nj; j++)
154          for (int i = 0; i < ni; i++) j_index(i,j) = j;
155      }
156   }
157
158   //----------------------------------------------------------------
159
160   void CDomain::checkLocalIDomain(void)
161   {
162      if (!ni.isEmpty() && !ibegin.isEmpty() && iend.isEmpty())
163        iend.setValue(ibegin.getValue() + ni.getValue() - 1);
164      else if (!ni.isEmpty() && !iend.isEmpty() && ibegin.isEmpty())
165        ibegin.setValue(iend.getValue() - ni.getValue()  + 1);
166      else if (!ibegin.isEmpty() && !iend.isEmpty() && ni.isEmpty())
167        ni.setValue(iend.getValue() - ibegin.getValue() + 1);
168      else if (!ibegin.isEmpty() && !iend.isEmpty() && !ni.isEmpty())
169      {
170        if (iend.getValue() != ibegin.getValue() + ni.getValue() - 1)
171          ERROR("CDomain::checkLocalIDomain(void)",
172                << "[ Id = " << this->getId() << " ] "
173                << "The local domain is wrongly defined,"
174                << " iend is different from (ibegin + ni - 1)");
175      }
176      else if (ibegin.isEmpty() && iend.isEmpty() && ni.isEmpty())
177      {
178        ibegin = 0;
179        iend = ni_glo - 1;
180        ni = ni_glo;
181      }
182      else
183      {
184        ERROR("CDomain::checkLocalIDomain(void)",
185              << "[ Id = " << this->getId() << " ] "
186              << "The local domain is wrongly defined,"
187              << " defining just one attribute among 'ibegin', 'iend' or 'ni' is invalid");
188      }
189
190      if (ni.getValue() < 0 || ibegin.getValue() > iend.getValue() ||
191          ibegin.getValue() < 0 || iend.getValue() > (ni_glo.getValue() - 1))
192      {
193        ERROR("CDomain::checkLocalIDomain(void)",
194              << "[ Id = " << this->getId() << " ] "
195              << "The local domain is wrongly defined,"
196              << " check the attributes 'ni_glo', 'ni', 'ibegin' and 'iend'");
197      }
198   }
199
200   //----------------------------------------------------------------
201
202   void CDomain::checkLocalJDomain(void)
203   {
204      if (!nj.isEmpty() && !jbegin.isEmpty() && jend.isEmpty())
205        jend.setValue(jbegin.getValue() + nj.getValue() - 1);
206      else if (!nj.isEmpty() && !jend.isEmpty() && jbegin.isEmpty())
207        jbegin.setValue(jend.getValue() - nj.getValue() + 1);
208      else if (!jbegin.isEmpty() && !jend.isEmpty() && nj.isEmpty())
209        nj.setValue(jend.getValue() - jbegin.getValue() + 1);
210      else if (!jbegin.isEmpty() && !jend.isEmpty() && !nj.isEmpty())
211      {
212        if (jend.getValue() != jbegin.getValue() + nj.getValue() - 1)
213          ERROR("CDomain::checkLocalJDomain(void)",
214                << "[ Id = " << this->getId() << " ] "
215                << "The local domain is wrongly defined,"
216                << " jend is different from (jbegin + nj - 1)");
217      }
218      else if (jbegin.isEmpty() && jend.isEmpty() && nj.isEmpty())
219      {
220        jbegin = 0;
221        jend = nj_glo - 1;
222        nj = nj_glo;
223      }
224      else
225      {
226        ERROR("CDomain::checkLocalJDomain(void)",
227              << "[ Id = " << this->getId() << " ] "
228              << "The local domain is wrongly defined,"
229              << " defining just one attribute among 'jbegin', 'jend' or 'nj' is invalid");
230      }
231
232      if (nj.getValue() < 0 || jbegin.getValue() > jend.getValue() ||
233          jbegin.getValue() < 0 || jend.getValue() > (nj_glo.getValue() - 1))
234      {
235        ERROR("CDomain::checkLocalJDomain(void)",
236              << "[ Id = " << this->getId() << " ] "
237              << "The local domain is wrongly defined,"
238              << " check the attributes 'nj_glo', 'nj', 'jbegin' and 'jend'");
239      }
240   }
241
242   //----------------------------------------------------------------
243
244   void CDomain::checkMask(void)
245   {
246      using namespace std;
247
248      int ibegin_mask = 0,
249          jbegin_mask = 0,
250          iend_mask = iend.getValue() - ibegin.getValue(),
251          jend_mask = jend.getValue() - jbegin.getValue();
252
253      if (!zoom_ibegin.isEmpty())
254      {
255         int zoom_iend = zoom_ibegin.getValue() + zoom_ni.getValue() - 1;
256         int zoom_jend = zoom_jbegin.getValue() + zoom_nj.getValue() - 1;
257
258         ibegin_mask = max (ibegin.getValue(), zoom_ibegin.getValue());
259         jbegin_mask = max (jbegin.getValue(), zoom_jbegin.getValue());
260         iend_mask   = min (iend.getValue(), zoom_iend);
261         jend_mask   = min (jend.getValue(), zoom_jend);
262
263         ibegin_mask -= ibegin.getValue();
264         jbegin_mask -= jbegin.getValue();
265         iend_mask   -= ibegin.getValue();
266         jend_mask   -= jbegin.getValue();
267      }
268
269
270      if (!mask.isEmpty())
271      {
272         if ((mask.extent(0) != ni) ||
273             (mask.extent(1) != nj))
274            ERROR("CDomain::checkAttributes(void)",
275                  <<"the mask has not the same size than the local domain"<<endl
276                   <<"Local size is "<<ni<<"x"<<nj<<endl
277                  <<"Mask size is "<<mask.extent(0)<<"x"<<mask.extent(1));
278         for (int i = 0; i < ni; i++)
279         {
280            for (int j = 0; j < nj; j++)
281            {
282               if (i < ibegin_mask && i > iend_mask &&
283                   j < jbegin_mask && j > jend_mask )
284                     mask(i,j) = false;
285            }
286         }
287      }
288      else // (!mask.hasValue())
289      { // Si aucun masque n'est défini,
290        // on en crée un nouveau qui valide l'intégralité du domaine.
291         mask.resize(ni,nj) ;
292         for (int i = 0; i < ni.getValue(); i++)
293         {
294            for (int j = 0; j < nj.getValue(); j++)
295            {
296               if (i >= ibegin_mask && i <= iend_mask &&
297                   j >= jbegin_mask && j <= jend_mask )
298                     mask(i,j) = true;
299               else  mask(i,j) = false;
300            }
301         }
302      }
303   }
304
305
306   //----------------------------------------------------------------
307
308   void CDomain::checkDomainData(void)
309   {
310      if (!data_dim.isEmpty() &&
311         !(data_dim.getValue() == 1 || data_dim.getValue() == 2))
312      {
313         ERROR("CDomain::checkAttributes(void)",
314               << "Data dimension incompatible (must be 1 or 2) !") ;
315      }
316      else if (data_dim.isEmpty())
317      {
318         ERROR("CDomain::checkAttributes(void)",
319               << "Data dimension undefined !") ;
320      }
321
322      if (data_ibegin.isEmpty())
323         data_ibegin.setValue(0) ;
324      if (data_jbegin.isEmpty() && (data_dim.getValue() == 2))
325         data_jbegin.setValue(0) ;
326
327      if (!data_ni.isEmpty() && (data_ni.getValue() <= 0))
328      {
329         ERROR("CDomain::checkAttributes(void)",
330               << "Data dimension is negative (data_ni).") ;
331      }
332      else if (data_ni.isEmpty())
333      {
334         data_ni.setValue((data_dim.getValue() == 1)
335                           ? (ni.getValue() * nj.getValue())
336                           : ni.getValue());
337      }
338
339      if (data_dim.getValue() == 2)
340      {
341         if (!data_nj.isEmpty() && (data_nj.getValue() <= 0) )
342         {
343            ERROR("CDomain::checkAttributes(void)",
344                  << "Data dimension is negative (data_nj).") ;
345         }
346         else if (data_nj.isEmpty())
347            data_nj.setValue(nj.getValue()) ;
348      }
349
350   }
351
352   //----------------------------------------------------------------
353
354   void CDomain::checkCompression(void)
355   {
356      if (!data_i_index.isEmpty())
357      {
358         int ssize = data_i_index.numElements();
359         if (!data_n_index.isEmpty() &&
360            (data_n_index.getValue() != ssize))
361         {
362            ERROR("CDomain::checkAttributes(void)",
363                  <<"Dimension data_i_index incompatible with data_n_index.") ;
364         }
365         else if (data_n_index.isEmpty())
366            data_n_index.setValue(ssize) ;
367
368         if (data_dim.getValue() == 2)
369         {
370            if (!data_j_index.isEmpty() &&
371               (data_j_index.numElements() != data_i_index.numElements()))
372            {
373               ERROR("CDomain::checkAttributes(void)",
374                     <<"Dimension data_j_index incompatible with data_i_index.") ;
375            }
376            else if (data_j_index.isEmpty())
377            {
378               ERROR("CDomain::checkAttributes(void)",
379                     <<"data_j_index must be defined !") ;
380            }
381         }
382      }
383      else
384      {
385         if (!data_n_index.isEmpty() ||
386            ((data_dim.getValue() == 2) && (!data_j_index.isEmpty())))
387            ERROR("CDomain::checkAttributes(void)", << "data_i_index undefined") ;
388      }
389
390      if (data_n_index.isEmpty())
391      { // -> bloc re-vérifié OK
392         if (data_dim.getValue() == 1)
393         {
394            const int dni = data_ni.getValue();
395            data_i_index.resize(dni) ;
396            data_n_index.setValue(dni);
397            for (int i = 0; i < dni; i++) data_i_index(i) = i+1 ;
398         }
399         else   // (data_dim == 2)
400         {
401            const int dni = data_ni.getValue() * data_nj.getValue();
402            data_i_index.resize(dni) ;
403            data_j_index.resize(dni) ;
404
405            data_n_index.setValue(dni);
406
407            for(int count = 0, j = 0; j  < data_nj.getValue(); j++)
408            {
409               for(int i = 0; i < data_ni.getValue(); i++, count++)
410               {
411                  data_i_index(count) = i+1 ;
412                  data_j_index(count) = j+1 ;
413               }
414            }
415         }
416      }
417   }
418
419   //----------------------------------------------------------------
420
421   void CDomain::completeLonLatClient(void)
422   {
423      int i,j,k ;
424      CArray<double,1> lonvalue_temp(ni*nj) ;
425      CArray<double,1> latvalue_temp(ni*nj) ;
426      CArray<double,2> bounds_lon_temp(nvertex,ni*nj) ;
427      CArray<double,2> bounds_lat_temp(nvertex,ni*nj) ;
428
429      if (type.isEmpty())
430      {
431        if ( lonvalue.numElements() == ni*nj && latvalue.numElements() == ni*nj )
432        {
433          type.setValue(type_attr::curvilinear) ;
434          isCurvilinear=true ;
435        }
436        else if ( lonvalue.numElements() == ni && latvalue.numElements() == nj )
437        {
438          type.setValue(type_attr::regular) ;
439          isCurvilinear=false ;
440        }
441        else ERROR("void CDomain::completeLonLatClient(void)",<<"the grid is nor curvilinear, nor cartesian, because the size of longitude and latitude array is not coherent with the domain size"<<endl
442                                                              <<"lonvalue size = " << lonvalue.numElements() << "different of ni or ni*nj"<<endl
443                                                              <<"latvalue size = " << latvalue.numElements() << "different of nj or ni*nj" ) ;
444      }
445      if (type==type_attr::curvilinear || type==type_attr::unstructured)
446      {
447        lonvalue_temp=lonvalue ;
448        latvalue_temp=latvalue ;
449        if (hasBounds) bounds_lon_temp=bounds_lon ;
450        if (hasBounds) bounds_lat_temp=bounds_lat ;
451      }
452      else
453      {
454        for(j=0;j<nj;j++)
455          for(i=0;i<ni;i++)
456          {
457            k=j*ni+i ;
458            lonvalue_temp(k)=lonvalue(i) ;
459            latvalue_temp(k)=latvalue(j) ;
460            if (hasBounds)
461            {
462              for(int n=0;n<nvertex;n++)
463              {
464                bounds_lon_temp(n,k)=bounds_lon(n,i) ;
465                bounds_lat_temp(n,k)=bounds_lat(n,j) ;
466              }
467            }
468          }
469      }
470
471      StdSize dm = zoom_ni_client * zoom_nj_client;
472
473      // Make sure that this attribute is non-empty for every client.
474      if (0 != dm)
475      {
476        lonvalue.resize(dm);
477        latvalue.resize(dm);
478      }
479
480
481      for (int i = 0; i < zoom_ni_client; i++)
482      {
483        for (int j = 0; j < zoom_nj_client; j++)
484        {
485          lonvalue(i + j * zoom_ni_client) = lonvalue_temp( (i + zoom_ibegin_client-ibegin) + (j + zoom_jbegin_client -jbegin)*ni );
486          latvalue(i + j * zoom_ni_client) = latvalue_temp( (i + zoom_ibegin_client - ibegin)+(j + zoom_jbegin_client - jbegin)*ni );
487          if (hasBounds)
488          {
489            for(int n=0;n<nvertex;n++)
490            {
491              bounds_lon(n,i + j * zoom_ni_client) = bounds_lon_temp( n, (i + zoom_ibegin_client - ibegin) + (j + zoom_jbegin_client -jbegin)*ni );
492              bounds_lat(n,i + j * zoom_ni_client) = bounds_lat_temp( n, (i + zoom_ibegin_client - ibegin)+(j + zoom_jbegin_client -jbegin)*ni );
493            }
494          }
495        }
496      }
497    }
498
499
500   //----------------------------------------------------------------
501
502   void CDomain::checkZoom(void)
503   {
504      // Résolution et vérification des données globales de zoom.
505      if (!this->zoom_ni.isEmpty() || !this->zoom_nj.isEmpty() ||
506          !this->zoom_ibegin.isEmpty() || !this->zoom_jbegin.isEmpty())
507      {
508         if (this->zoom_ni.isEmpty()     || this->zoom_nj.isEmpty() ||
509             this->zoom_ibegin.isEmpty() || this->zoom_jbegin.isEmpty())
510         {
511            ERROR("CDomain::checkZoom(void)",
512                  <<"if one of zoom attributes is defined then all zoom attributes must be defined") ;
513         }
514         else
515         {
516            int zoom_iend = zoom_ibegin + zoom_ni - 1;
517            int zoom_jend = zoom_jbegin + zoom_nj - 1;
518
519            if (zoom_ibegin < 0  || zoom_jbegin < 0 || zoom_iend > (ni_glo-1) || zoom_jend > (nj_glo-1))
520               ERROR("CDomain::checkZoom(void)",
521                     << "Zoom is wrongly defined,"
522                     << " Check the values : zoom_ni, zoom_nj, zoom_ibegin, zoom_jbegin") ;
523         }
524      }
525      else
526      {
527         zoom_ni = ni_glo;
528         zoom_nj = nj_glo;
529         zoom_ibegin = 0;
530         zoom_jbegin = 0;
531      }
532
533      // compute client zoom indices
534
535      int zoom_iend=zoom_ibegin+zoom_ni-1 ;
536      zoom_ibegin_client = ibegin_client > zoom_ibegin ? ibegin_client : zoom_ibegin ;
537      zoom_iend_client = iend_client < zoom_iend ? iend_client : zoom_iend ;
538      zoom_ni_client=zoom_iend_client-zoom_ibegin_client+1 ;
539      if (zoom_ni_client<0) zoom_ni_client=0 ;
540
541
542      int zoom_jend=zoom_jbegin+zoom_nj-1 ;
543      zoom_jbegin_client = jbegin_client > zoom_jbegin ? jbegin_client : zoom_jbegin ;
544      zoom_jend_client = jend_client < zoom_jend ? jend_client : zoom_jend ;
545      zoom_nj_client=zoom_jend_client-zoom_jbegin_client+1 ;
546      if (zoom_nj_client<0) zoom_nj_client=0 ;
547
548   }
549
550   void CDomain::checkBounds(void)
551   {
552     if (!nvertex.isEmpty() && !bounds_lon.isEmpty() && !bounds_lat.isEmpty())
553     {
554       hasBounds=true ;
555
556     }
557     else
558     {
559       hasBounds=false;
560       nvertex=0 ;
561     }
562   }
563
564   //----------------------------------------------------------------
565   // Divide function checkAttributes into 2 seperate ones
566   // This function only checks all attributes of current domain
567   void CDomain::checkAttributesOnClient()
568   {
569     if (this->isClientChecked) return;
570     CContext* context=CContext::getCurrent();
571
572      this->checkDomain();
573      this->checkZoom();
574      this->checkBounds();
575
576      if (context->hasClient)
577      { // CÃŽté client uniquement
578         this->checkMask();
579         this->checkDomainData();
580         this->checkCompression();
581         this->completeLonLatClient();
582         this->computeConnectedServer() ;
583      }
584      else
585      { // CÃŽté serveur uniquement
586//         if (!this->isEmpty())
587// ne sert plus //   this->completeLonLatServer();
588      }
589
590      this->isClientChecked = true;
591   }
592
593   // Send all checked attributes to server
594   void CDomain::sendCheckedAttributes()
595   {
596     if (!this->isClientChecked) checkAttributesOnClient();
597     CContext* context=CContext::getCurrent() ;
598
599     if (this->isChecked) return;
600     if (context->hasClient)
601     {
602       sendServerAttribut() ;
603       sendLonLat() ;
604     }
605
606     this->isChecked = true;
607   }
608
609   void CDomain::checkAttributes(void)
610   {
611      if (this->isChecked) return;
612      CContext* context=CContext::getCurrent() ;
613
614      this->checkDomain();
615      this->checkZoom();
616      this->checkBounds();
617
618      if (context->hasClient)
619      { // CÃŽté client uniquement
620         this->checkMask();
621         this->checkDomainData();
622         this->checkCompression();
623         this->completeLonLatClient();
624      }
625      else
626      { // CÃŽté serveur uniquement
627//         if (!this->isEmpty())
628// ne sert plus //   this->completeLonLatServer();
629      }
630
631      if (context->hasClient)
632      {
633        computeConnectedServer() ;
634        sendServerAttribut() ;
635        sendLonLat() ;
636      }
637
638      this->isChecked = true;
639   }
640
641  void CDomain::sendServerAttribut(void)
642  {
643    std::vector<int> nGlobDomain(2);
644    nGlobDomain[0] = ni_glo.getValue();
645    nGlobDomain[1] = nj_glo.getValue();
646    CServerDistributionDescription serverDescription(nGlobDomain);
647
648    CContext* context = CContext::getCurrent();
649    CContextClient* client = context->client;
650    int nbServer = client->serverSize;
651
652    serverDescription.computeServerDistribution(nbServer);
653    std::vector<std::vector<int> > serverIndexBegin = serverDescription.getServerIndexBegin();
654    std::vector<std::vector<int> > serverDimensionSizes = serverDescription.getServerDimensionSizes();
655
656    CEventClient event(getType(),EVENT_ID_SERVER_ATTRIBUT);
657    if (client->isServerLeader())
658    {
659      std::list<CMessage> msgs;
660
661      const std::list<int>& ranks = client->getRanksServerLeader();
662      for (std::list<int>::const_iterator itRank = ranks.begin(), itRankEnd = ranks.end(); itRank != itRankEnd; ++itRank)
663      {
664        // Use const int to ensure CMessage holds a copy of the value instead of just a reference
665        const int ibegin_srv = serverIndexBegin[*itRank][0];
666        const int jbegin_srv = serverIndexBegin[*itRank][1];
667        const int ni_srv = serverDimensionSizes[*itRank][0];
668        const int nj_srv = serverDimensionSizes[*itRank][1];
669        const int iend_srv = ibegin_srv + ni_srv - 1;
670        const int jend_srv = jbegin_srv + nj_srv - 1;
671
672        msgs.push_back(CMessage());
673        CMessage& msg = msgs.back();
674        msg << this->getId() ;
675        msg << ni_srv << ibegin_srv << iend_srv << nj_srv << jbegin_srv << jend_srv;
676
677        event.push(*itRank,1,msg);
678      }
679      client->sendEvent(event);
680    }
681    else client->sendEvent(event);
682  }
683
684  void CDomain::computeConnectedServer(void)
685  {
686    ibegin_client=ibegin ; iend_client=iend ; ni_client=ni ;
687    jbegin_client=jbegin ; jend_client=jend ; nj_client=nj ;
688
689    CContext* context=CContext::getCurrent() ;
690    CContextClient* client=context->client ;
691    int nbServer=client->serverSize;
692    bool doComputeGlobalIndexServer = true;
693
694    int i,j,i_ind,j_ind ;
695    int zoom_iend=zoom_ibegin+zoom_ni-1 ;
696    int zoom_jend=zoom_jbegin+zoom_nj-1 ;
697
698    // Precompute number of index
699    int globalIndexCountZoom = 0;
700    for(j=0;j<nj;j++)
701      for(i=0;i<ni;i++)
702      {
703        i_ind=ibegin+i_index(i,j) ;
704        j_ind=jbegin+j_index(i,j) ;
705
706        if (i_ind >= zoom_ibegin && i_ind <= zoom_iend && j_ind >= zoom_jbegin && j_ind <= zoom_jend)
707        {
708          ++globalIndexCountZoom;
709        }
710      }
711
712    // Fill in index
713    CArray<size_t,1> globalIndexDomainZoom(globalIndexCountZoom);
714    CArray<size_t,1> globalIndexDomain(ni*nj);
715    size_t globalIndex;
716    int globalIndexCount = 0;
717    globalIndexCountZoom = 0;
718
719    for(j=0;j<nj;j++)
720      for(i=0;i<ni;i++)
721      {
722        i_ind=ibegin+i_index(i,j) ;
723        j_ind=jbegin+j_index(i,j) ;
724
725        globalIndex = i_ind + j_ind * ni_glo;
726        globalIndexDomain(globalIndexCount) = globalIndex;
727        ++globalIndexCount;
728        if (i_ind >= zoom_ibegin && i_ind <= zoom_iend && j_ind >= zoom_jbegin && j_ind <= zoom_jend)
729        {
730          globalIndexDomainZoom(globalIndexCountZoom) = globalIndex;
731          ++globalIndexCountZoom;
732        }
733      }
734
735     std::vector<int> nGlobDomain(2);
736     nGlobDomain[0] = ni_glo.getValue();
737     nGlobDomain[1] = nj_glo.getValue();
738     size_t globalSizeIndex = 1, indexBegin, indexEnd;
739     int range, clientSize = client->clientSize;
740     for (int i = 0; i < nGlobDomain.size(); ++i) globalSizeIndex *= nGlobDomain[i];
741     indexBegin = 0;
742     for (int i = 0; i < clientSize; ++i)
743     {
744       range = globalSizeIndex / clientSize;
745       if (i < (globalSizeIndex%clientSize)) ++range;
746       if (i == client->clientRank) break;
747       indexBegin += range;
748     }
749     indexEnd = indexBegin + range - 1;
750
751    CServerDistributionDescription serverDescription(nGlobDomain);
752    serverDescription.computeServerGlobalIndexInRange(nbServer, std::make_pair<size_t,size_t>(indexBegin, indexEnd));
753    CClientServerMapping* clientServerMap = new CClientServerMappingDistributed(serverDescription.getGlobalIndexRange(),
754                                                                                client->intraComm);
755    clientServerMap->computeServerIndexMapping(globalIndexDomain);
756    const std::map<int, std::vector<size_t> >& globalIndexDomainOnServer = clientServerMap->getGlobalIndexOnServer();
757
758    std::map<int, std::vector<size_t> >::const_iterator it = globalIndexDomainOnServer.begin(),
759                                                       ite = globalIndexDomainOnServer.end();
760    indSrv_.clear();
761    for (; it != ite; ++it)
762    {
763      int rank = it->first;
764      std::vector<size_t>::const_iterator itbVec  = (it->second).begin(),
765                                           iteVec = (it->second).end();
766      int nb = globalIndexDomainZoom.numElements();
767      for (int i = 0; i < nb; ++i)
768      {
769        if (iteVec != std::find(itbVec, iteVec, globalIndexDomainZoom(i)))
770        {
771          indSrv_[rank].push_back(globalIndexDomainZoom(i));
772        }
773      }
774    }
775
776    connectedServerRank_.clear();
777    for (it = globalIndexDomainOnServer.begin(); it != ite; ++it) {
778      connectedServerRank_.push_back(it->first);
779    }
780
781    if (!indSrv_.empty())
782    {
783      connectedServerRank_.clear();
784      for (it = indSrv_.begin(); it != indSrv_.end(); ++it)
785        connectedServerRank_.push_back(it->first);
786    }
787    nbConnectedClients_ = clientServerMap->computeConnectedClients(client->serverSize, client->clientSize, client->intraComm, connectedServerRank_);
788
789    delete clientServerMap;
790  }
791
792  void CDomain::sendLonLat(void)
793  {
794    int ns, n, i, j, ind, nv, idx;
795    CContext* context = CContext::getCurrent();
796    CContextClient* client=context->client;
797
798    // send lon lat for each connected server
799    CEventClient eventIndex(getType(), EVENT_ID_INDEX);
800    CEventClient eventLon(getType(), EVENT_ID_LON);
801    CEventClient eventLat(getType(), EVENT_ID_LAT);
802
803    list<CMessage> list_msgsIndex, list_msgsLon, list_msgsLat;
804    list<CArray<int,1> > list_indi, list_indj;
805    list<CArray<double,1> > list_lon, list_lat;
806    list<CArray<double,2> > list_boundslon, list_boundslat;
807
808    std::map<int, std::vector<size_t> >::const_iterator it, iteMap;
809    iteMap = indSrv_.end();
810    for (int k = 0; k < connectedServerRank_.size(); ++k)
811    {
812      int nbData = 0;
813      int rank = connectedServerRank_[k];
814      it = indSrv_.find(rank);
815      if (iteMap != it)
816        nbData = it->second.size();
817
818      list_indi.push_back(CArray<int,1>(nbData));
819      list_indj.push_back(CArray<int,1>(nbData));
820      list_lon.push_back(CArray<double,1>(nbData));
821      list_lat.push_back(CArray<double,1>(nbData));
822
823      if (hasBounds)
824      {
825        list_boundslon.push_back(CArray<double,2>(nvertex, nbData));
826        list_boundslat.push_back(CArray<double,2>(nvertex, nbData));
827      }
828
829      CArray<int,1>& indi = list_indi.back();
830      CArray<int,1>& indj = list_indj.back();
831      CArray<double,1>& lon = list_lon.back();
832      CArray<double,1>& lat = list_lat.back();
833
834      for (n = 0; n < nbData; ++n)
835      {
836        idx = static_cast<int>(it->second[n]);
837        i = idx % ni_glo;
838        j = idx / ni_glo;
839        ind = (i - zoom_ibegin_client) + (j - zoom_jbegin_client) * zoom_ni_client;
840
841        lon(n) = lonvalue(ind);
842        lat(n) = latvalue(ind);
843
844        if (hasBounds)
845        {
846          CArray<double,2>& boundslon = list_boundslon.back();
847          CArray<double,2>& boundslat = list_boundslat.back();
848
849          for (nv = 0; nv < nvertex; nv++)
850          {
851            boundslon(nv, n) = bounds_lon(nv, ind);
852            boundslat(nv, n) = bounds_lat(nv, ind);
853          }
854        }
855
856        indi(n) = ibegin + i_index(i - ibegin, j - jbegin);
857        indj(n) = jbegin + j_index(i - ibegin, j - jbegin);
858      }
859
860      list_msgsIndex.push_back(CMessage());
861
862      list_msgsIndex.back() << this->getId() << (int)type; // enum ne fonctionne pour les message => ToFix
863      list_msgsIndex.back() << isCurvilinear;
864      list_msgsIndex.back() << list_indi.back() << list_indj.back();
865
866      list_msgsLon.push_back(CMessage());
867      list_msgsLat.push_back(CMessage());
868
869      list_msgsLon.back() << this->getId() << list_lon.back();
870      list_msgsLat.back() << this->getId() << list_lat.back();
871
872      if (hasBounds)
873      {
874        list_msgsLon.back() << list_boundslon.back();
875        list_msgsLat.back() << list_boundslat.back();
876      }
877
878      eventIndex.push(rank, nbConnectedClients_[rank], list_msgsIndex.back());
879      eventLon.push(rank, nbConnectedClients_[rank], list_msgsLon.back());
880      eventLat.push(rank, nbConnectedClients_[rank], list_msgsLat.back());
881    }
882
883    client->sendEvent(eventIndex);
884    client->sendEvent(eventLon);
885    client->sendEvent(eventLat);
886  }
887
888  bool CDomain::dispatchEvent(CEventServer& event)
889  {
890    if (SuperClass::dispatchEvent(event)) return true;
891    else
892    {
893      switch(event.type)
894      {
895        case EVENT_ID_SERVER_ATTRIBUT:
896          recvServerAttribut(event);
897          return true;
898          break;
899        case EVENT_ID_INDEX:
900          recvIndex(event);
901          return true;
902          break;
903        case EVENT_ID_LON:
904          recvLon(event);
905          return true;
906          break;
907        case EVENT_ID_LAT:
908          recvLat(event);
909          return true;
910          break;
911        default:
912          ERROR("bool CContext::dispatchEvent(CEventServer& event)",
913                << "Unknown Event");
914          return false;
915       }
916    }
917  }
918
919  void CDomain::recvServerAttribut(CEventServer& event)
920  {
921    CBufferIn* buffer=event.subEvents.begin()->buffer;
922    string domainId ;
923    *buffer>>domainId ;
924    get(domainId)->recvServerAttribut(*buffer) ;
925  }
926
927  void CDomain::recvServerAttribut(CBufferIn& buffer)
928  {
929    int zoom_iend = zoom_ibegin.getValue() + zoom_ni.getValue() - 1;
930    int zoom_jend = zoom_jbegin.getValue() + zoom_nj.getValue() - 1;
931
932    buffer >> ni_srv >> ibegin_srv >> iend_srv >> nj_srv >> jbegin_srv >> jend_srv;
933
934    zoom_ibegin_srv = zoom_ibegin.getValue() > ibegin_srv ? zoom_ibegin.getValue() : ibegin_srv ;
935    zoom_iend_srv = zoom_iend < iend_srv ? zoom_iend : iend_srv ;
936    zoom_ni_srv=zoom_iend_srv-zoom_ibegin_srv+1 ;
937
938    zoom_jbegin_srv = zoom_jbegin.getValue() > jbegin_srv ? zoom_jbegin.getValue() : jbegin_srv ;
939    zoom_jend_srv = zoom_jend < jend_srv ? zoom_jend : jend_srv ;
940    zoom_nj_srv=zoom_jend_srv-zoom_jbegin_srv+1 ;
941
942    if (zoom_ni_srv<=0 || zoom_nj_srv<=0)
943    {
944      zoom_ibegin_srv=0 ; zoom_iend_srv=0 ; zoom_ni_srv=0 ;
945      zoom_jbegin_srv=0 ; zoom_jend_srv=0 ; zoom_nj_srv=0 ;
946    }
947    lonvalue_srv.resize(zoom_ni_srv*zoom_nj_srv) ;
948    lonvalue_srv = 0. ;
949    latvalue_srv.resize(zoom_ni_srv*zoom_nj_srv) ;
950    latvalue_srv = 0. ;
951    if (hasBounds)
952    {
953      bounds_lon_srv.resize(nvertex,zoom_ni_srv*zoom_nj_srv) ;
954      bounds_lon_srv = 0. ;
955      bounds_lat_srv.resize(nvertex,zoom_ni_srv*zoom_nj_srv) ;
956      bounds_lat_srv = 0. ;
957    }
958  }
959
960  void CDomain::recvIndex(CEventServer& event)
961  {
962    list<CEventServer::SSubEvent>::iterator it;
963    for (it = event.subEvents.begin(); it != event.subEvents.end(); ++it)
964    {
965      CBufferIn* buffer = it->buffer;
966      string domainId;
967      *buffer >> domainId;
968      get(domainId)->recvIndex(it->rank, *buffer);
969    }
970  }
971
972  void CDomain::recvIndex(int rank, CBufferIn& buffer)
973  {
974    int type_int;
975    buffer >> type_int >> isCurvilinear >> indiSrv[rank] >> indjSrv[rank];
976    type.setValue((type_attr::t_enum)type_int); // probleme des type enum avec les buffers : ToFix
977  }
978
979  void CDomain::recvLon(CEventServer& event)
980  {
981    list<CEventServer::SSubEvent>::iterator it;
982    for (it = event.subEvents.begin(); it != event.subEvents.end(); ++it)
983    {
984      CBufferIn* buffer = it->buffer;
985      string domainId;
986      *buffer >> domainId;
987      get(domainId)->recvLon(it->rank, *buffer);
988    }
989  }
990
991  void CDomain::recvLon(int rank, CBufferIn& buffer)
992  {
993    CArray<int,1> &indi = indiSrv[rank], &indj = indjSrv[rank];
994    CArray<double,1> lon;
995    CArray<double,2> boundslon;
996
997    buffer >> lon;
998    if (hasBounds) buffer >> boundslon;
999
1000    int i, j, ind_srv;
1001    for (int ind = 0; ind < indi.numElements(); ind++)
1002    {
1003      i = indi(ind); j = indj(ind);
1004      ind_srv = (i - zoom_ibegin_srv) + (j - zoom_jbegin_srv) * zoom_ni_srv;
1005      lonvalue_srv(ind_srv) = lon(ind);
1006      if (hasBounds)
1007      {
1008        for (int nv = 0; nv < nvertex; nv++)
1009          bounds_lon_srv(nv, ind_srv) = boundslon(nv, ind);
1010      }
1011    }
1012  }
1013
1014  void CDomain::recvLat(CEventServer& event)
1015  {
1016    list<CEventServer::SSubEvent>::iterator it;
1017    for (it = event.subEvents.begin(); it != event.subEvents.end(); ++it)
1018    {
1019      CBufferIn* buffer = it->buffer;
1020      string domainId;
1021      *buffer >> domainId;
1022      get(domainId)->recvLat(it->rank, *buffer);
1023    }
1024  }
1025
1026  void CDomain::recvLat(int rank, CBufferIn& buffer)
1027  {
1028    CArray<int,1> &indi = indiSrv[rank], &indj = indjSrv[rank];
1029    CArray<double,1> lat;
1030    CArray<double,2> boundslat;
1031
1032    buffer >> lat;
1033    if (hasBounds) buffer >> boundslat;
1034
1035    int i, j, ind_srv;
1036    for (int ind = 0; ind < indi.numElements(); ind++)
1037    {
1038      i = indi(ind); j = indj(ind);
1039      ind_srv = (i - zoom_ibegin_srv) + (j - zoom_jbegin_srv) * zoom_ni_srv;
1040      latvalue_srv(ind_srv) = lat(ind);
1041      if (hasBounds)
1042      {
1043        for (int nv = 0; nv < nvertex; nv++)
1044          bounds_lat_srv(nv, ind_srv) = boundslat(nv, ind);
1045      }
1046    }
1047  }
1048
1049   //----------------------------------------------------------------
1050
1051   DEFINE_REF_FUNC(Domain,domain)
1052
1053   ///---------------------------------------------------------------
1054
1055} // namespace xios
Note: See TracBrowser for help on using the repository browser.