src/MEDCoupling/MEDCouplingMultiFields.cxx

   1 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 // Author : Anthony Geay (CEA/DEN)
  20
  21 #include "MEDCouplingMultiFields.hxx"
  22 #include "MEDCouplingFieldTemplate.hxx"
  23 #include "MEDCouplingFieldDouble.hxx"
  24 #include "MEDCouplingMesh.hxx"
  25 #include "MEDCouplingAutoRefCountObjectPtr.hxx"
  26
  27 #include <sstream>
  28 #include <algorithm>
  29
  30 using namespace ParaMEDMEM;
  31
  32 MEDCouplingMultiFields *MEDCouplingMultiFields::New(const std::vector<MEDCouplingFieldDouble *>& fs) throw(INTERP_KERNEL::Exception)
  33 {
  34   return new MEDCouplingMultiFields(fs);
  35 }
  36
  37 MEDCouplingMultiFields *MEDCouplingMultiFields::New()
  38 {
  39   return new MEDCouplingMultiFields;
  40 }
  41
  42 MEDCouplingMultiFields *MEDCouplingMultiFields::deepCpy() const
  43 {
  44   return new MEDCouplingMultiFields(*this);
  45 }
  46
  47 bool MEDCouplingMultiFields::isEqual(const MEDCouplingMultiFields *other, double meshPrec, double valsPrec) const
  48 {
  49   std::size_t sz=_fs.size();
  50   if(sz!=other->_fs.size())
  51     return false;
  52   for(std::size_t i=0;i<sz;i++)
  53     {
  54       const MEDCouplingFieldDouble *f1=_fs[i];
  55       const MEDCouplingFieldDouble *f2=other->_fs[i];
  56       if(f1!=f2)
  57         {
  58           if(f1==0 || f2==0)
  59             return false;
  60           if(!_fs[i]->isEqual(other->_fs[i],meshPrec,valsPrec))
  61             return false;
  62         }
  63     }
  64   std::vector<int> refs1,refs2;
  65   std::vector<MEDCouplingMesh *> ms1=getDifferentMeshes(refs1);
  66   std::vector<MEDCouplingMesh *> ms2=other->getDifferentMeshes(refs2);
  67   if(ms1.size()!=ms2.size())
  68     return false;
  69   if(refs1!=refs2)
  70     return false;
  71   std::vector< std::vector<int> > refs3,refs4;
  72   std::vector<DataArrayDouble *> das1=getDifferentArrays(refs3);
  73   std::vector<DataArrayDouble *> das2=getDifferentArrays(refs4);
  74   if(das1.size()!=das2.size())
  75     return false;
  76   if(refs3!=refs4)
  77     return false;
  78   return true;
  79 }
  80
  81 std::string MEDCouplingMultiFields::getName() const
  82 {
  83   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
  84   for(;it!=_fs.end();it++)
  85     if((const MEDCouplingFieldDouble *)(*it))
  86       return (*it)->getName();
  87   return std::string();
  88 }
  89
  90 std::string MEDCouplingMultiFields::getDescription() const
  91 {
  92   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
  93   for(;it!=_fs.end();it++)
  94     if((const MEDCouplingFieldDouble *)(*it))
  95       return (*it)->getDescription();
  96   return std::string();
  97 }
  98
  99 std::string MEDCouplingMultiFields::getTimeUnit() const
 100 {
 101   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
 102   for(;it!=_fs.end();it++)
 103     if((const MEDCouplingFieldDouble *)(*it))
 104       return (*it)->getTimeUnit();
 105   return std::string();
 106 }
 107
 108 double MEDCouplingMultiFields::getTimeResolution() const throw(INTERP_KERNEL::Exception)
 109 {
 110   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
 111   for(;it!=_fs.end();it++)
 112     if((const MEDCouplingFieldDouble *)(*it))
 113       return (*it)->getTimeTolerance();
 114   throw INTERP_KERNEL::Exception("MEDCouplingMultiFields::getTimeResolution : no not null field !");
 115 }
 116
 117 std::string MEDCouplingMultiFields::simpleRepr() const
 118 {
 119   std::ostringstream ret;
 120   ret << "MEDCouplingMultiFields with name : \"" << getName() << "\"\n";
 121   ret << "Description of MEDCouplingMultiFields is : \"" << getDescription() << "\"\n";
 122   ret << "Number of discretization : " << _fs.size() << "\n";
 123   ret << "Number of different meshes : ";
 124   std::vector<MEDCouplingMesh *> ms;
 125   std::vector<int> refms;
 126   try
 127     {
 128       ms=getDifferentMeshes(refms);
 129       ret << ms.size() << "\n";
 130     }
 131   catch(INTERP_KERNEL::Exception& e)
 132     { ret << "Current instance is INVALID !\n"; }
 133   return ret.str();
 134 }
 135
 136 std::string MEDCouplingMultiFields::advancedRepr() const
 137 {
 138   return simpleRepr();
 139 }
 140
 141 bool MEDCouplingMultiFields::isEqualWithoutConsideringStr(const MEDCouplingMultiFields *other, double meshPrec, double valsPrec) const
 142 {
 143   std::size_t sz=_fs.size();
 144   if(sz!=other->_fs.size())
 145     return false;
 146   for(std::size_t i=0;i<sz;i++)
 147     if(!_fs[i]->isEqualWithoutConsideringStr(other->_fs[i],meshPrec,valsPrec))
 148       return false;
 149   return true;
 150 }
 151
 152 const MEDCouplingFieldDouble *MEDCouplingMultiFields::getFieldWithId(int id) const throw(INTERP_KERNEL::Exception)
 153 {
 154   if(id>=(int)_fs.size() || id < 0)
 155     throw INTERP_KERNEL::Exception("MEDCouplingMultiFields::getFieldWithId : invalid id outside boundaries !");
 156   return _fs[id];
 157 }
 158
 159 std::vector<const MEDCouplingFieldDouble *> MEDCouplingMultiFields::getFields() const
 160 {
 161   std::vector<const MEDCouplingFieldDouble *> ret(_fs.size());
 162   std::copy(_fs.begin(),_fs.end(),ret.begin());
 163   return ret;
 164 }
 165
 166 int MEDCouplingMultiFields::getNumberOfFields() const
 167 {
 168   return (int)_fs.size();
 169 }
 170
 171 const MEDCouplingFieldDouble *MEDCouplingMultiFields::getFieldAtPos(int id) const throw(INTERP_KERNEL::Exception)
 172 {
 173   if(id<0 || id>=(int)_fs.size())
 174     {
 175       std::ostringstream oss; oss << "MEDCouplingMultiFields::getFieldAtPos : Invalid given pos : should be >=0 and < " << _fs.size() << " !";
 176       throw INTERP_KERNEL::Exception(oss.str().c_str());
 177     }
 178   return _fs[id];
 179 }
 180
 181 void MEDCouplingMultiFields::updateTime() const
 182 {
 183   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
 184   for(;it!=_fs.end();it++)
 185     if((const MEDCouplingFieldDouble *)(*it))
 186       (*it)->updateTime();
 187   it=_fs.begin();
 188   for(;it!=_fs.end();it++)
 189     if((const MEDCouplingFieldDouble *)(*it))
 190       updateTimeWith(*(*it));
 191 }
 192
 193 std::vector<MEDCouplingMesh *> MEDCouplingMultiFields::getMeshes() const throw(INTERP_KERNEL::Exception)
 194 {
 195   std::vector<MEDCouplingMesh *> ms;
 196   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++)
 197     {
 198       const MEDCouplingMesh *m=0;
 199       if((const MEDCouplingFieldDouble *)(*it))
 200         m=(*it)->getMesh();
 201       ms.push_back(const_cast<MEDCouplingMesh *>(m));
 202     }
 203   return ms;
 204 }
 205
 206 std::vector<MEDCouplingMesh *> MEDCouplingMultiFields::getDifferentMeshes(std::vector<int>& refs) const throw(INTERP_KERNEL::Exception)
 207 {
 208   refs.resize(_fs.size());
 209   std::vector<MEDCouplingMesh *> ms;
 210   int id=0;
 211   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++,id++)
 212     {
 213       const MEDCouplingMesh *m=0;
 214       if((const MEDCouplingFieldDouble *)(*it))
 215         m=(*it)->getMesh();
 216       if(m)
 217         {
 218           std::vector<MEDCouplingMesh *>::iterator it2=std::find(ms.begin(),ms.end(),m);
 219           if(it2==ms.end())
 220             {
 221               ms.push_back(const_cast<MEDCouplingMesh *>(m));
 222               refs[id]=(int)ms.size()-1;
 223             }
 224           else
 225             refs[id]=(int)std::distance(ms.begin(),it2);
 226         }
 227       else
 228         refs[id]=-1;
 229     }
 230   return ms;
 231 }
 232
 233 std::vector<DataArrayDouble *> MEDCouplingMultiFields::getArrays() const throw(INTERP_KERNEL::Exception)
 234 {
 235   std::vector<DataArrayDouble *> tmp;
 236   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++)
 237     {
 238       std::vector<DataArrayDouble *> tmp2=(*it)->getArrays();
 239       tmp.insert(tmp.end(),tmp2.begin(),tmp2.end());
 240     }
 241   return tmp;
 242 }
 243
 244 std::vector<DataArrayDouble *> MEDCouplingMultiFields::getDifferentArrays(std::vector< std::vector<int> >& refs) const throw(INTERP_KERNEL::Exception)
 245 {
 246   refs.resize(_fs.size());
 247   int id=0;
 248   std::vector<DataArrayDouble *> ret;
 249   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++,id++)
 250     {
 251       std::vector<DataArrayDouble *> tmp2;
 252       if((const MEDCouplingFieldDouble *)(*it))
 253         {
 254           tmp2=(*it)->getArrays();
 255           refs[id].resize(tmp2.size());
 256         }
 257       else
 258         refs[id].clear();
 259       int id2=0;
 260       for(std::vector<DataArrayDouble *>::const_iterator it2=tmp2.begin();it2!=tmp2.end();it2++,id2++)
 261         {
 262           if(*it2)
 263             {
 264               std::vector<DataArrayDouble *>::iterator it3=std::find(ret.begin(),ret.end(),*it2);
 265               if(it3==ret.end())
 266                 {
 267                   ret.push_back(*it2);
 268                   refs[id][id2]=(int)ret.size()-1;
 269                 }
 270               else
 271                 refs[id][id2]=(int)std::distance(ret.begin(),it3);
 272             }
 273           else
 274             refs[id][id2]=-1;
 275         }
 276     }
 277   return ret;
 278 }
 279
 280 void MEDCouplingMultiFields::checkCoherency() const throw(INTERP_KERNEL::Exception)
 281 {
 282   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
 283   for(;it!=_fs.end();it++)
 284     {
 285       if((const MEDCouplingFieldDouble *)(*it)==0)
 286         throw INTERP_KERNEL::Exception("MEDCouplingMultiFields::checkCoherency : There is an empty Field in array...");
 287       (*it)->checkCoherency();
 288     }
 289 }
 290
 291 MEDCouplingMultiFields::MEDCouplingMultiFields(const std::vector<MEDCouplingFieldDouble *>& fs) throw(INTERP_KERNEL::Exception):_fs(fs.size())
 292 {
 293   int id=0;
 294   for(std::vector< MEDCouplingFieldDouble * >::const_iterator it=fs.begin();it!=fs.end();it++,id++)
 295     {
 296       if(*it)
 297         (*it)->incrRef();
 298       else
 299         throw INTERP_KERNEL::Exception("MEDCouplingMultiFields constructor : empty field found in vector !");
 300       (*it)->checkCoherency();
 301       _fs[id]=*it;
 302     }
 303 }
 304
 305
 306 /*!
 307  * Performs deepCpy.
 308  */
 309 MEDCouplingMultiFields::MEDCouplingMultiFields(const MEDCouplingMultiFields& other)
 310 {
 311   std::size_t sz=other._fs.size();
 312   _fs.resize(sz);
 313   std::vector<int> refs;
 314   std::vector< std::vector<int> > refs2;
 315   std::vector<MEDCouplingMesh *> ms=other.getDifferentMeshes(refs);
 316   std::size_t msLgh=ms.size();
 317   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingMesh> > ms2(msLgh);
 318   for(std::size_t i=0;i<msLgh;i++)
 319     ms2[i]=ms[i]->deepCpy();
 320   std::vector<DataArrayDouble *> das=other.getDifferentArrays(refs2);
 321   std::size_t dasLgth=das.size();
 322   std::vector< MEDCouplingAutoRefCountObjectPtr<DataArrayDouble> > das2(dasLgth);
 323   for(std::size_t i=0;i<dasLgth;i++)
 324     das2[i]=das[i]->deepCpy();
 325   for(std::size_t i=0;i<sz;i++)
 326     {
 327       if((const MEDCouplingFieldDouble *)other._fs[i])
 328         {
 329           MEDCouplingFieldTemplate *tmp=MEDCouplingFieldTemplate::New(other._fs[i]);
 330           _fs[i]=MEDCouplingFieldDouble::New(tmp,other._fs[i]->getTimeDiscretization());
 331           tmp->decrRef();
 332           if(refs[i]!=-1)
 333             _fs[i]->setMesh(ms2[refs[i]]);
 334           std::size_t nbOfArr=refs2[i].size();
 335           std::vector<DataArrayDouble *> tmp2(nbOfArr);
 336           for(std::size_t j=0;j<nbOfArr;j++)
 337             {
 338               if(refs2[i][j]!=-1)
 339                 tmp2[j]=das2[refs2[i][j]];
 340               else
 341                 tmp2[j]=0;
 342             }
 343           _fs[i]->setArrays(tmp2);
 344           std::vector<int> tinyInfo;
 345           std::vector<double> tinyInfo2;
 346           other._fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationIntInformation2(tinyInfo);
 347           other._fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationDbleInformation2(tinyInfo2);
 348           _fs[i]->getTimeDiscretizationUnderGround()->finishUnserialization2(tinyInfo,tinyInfo2);
 349         }
 350     }
 351 }
 352
 353 MEDCouplingMultiFields::MEDCouplingMultiFields()
 354 {
 355 }
 356
 357 void MEDCouplingMultiFields::getTinySerializationInformation(std::vector<int>& tinyInfo, std::vector<double>& tinyInfo2, int& nbOfDiffMeshes, int& nbOfDiffArr) const
 358 {
 359   std::vector<int> refs;
 360   std::vector<MEDCouplingMesh *> ms=getDifferentMeshes(refs);
 361   nbOfDiffMeshes=(int)ms.size();
 362   std::vector< std::vector<int> > refs2;
 363   std::vector<DataArrayDouble *> fs=getDifferentArrays(refs2);
 364   nbOfDiffArr=(int)fs.size();
 365   //
 366   std::size_t sz=refs.size();//==_fs.size()
 367   int sz2=0;
 368   for(std::size_t i=0;i<sz;i++)
 369     sz2+=(int)refs2[i].size();
 370   //
 371   tinyInfo2.clear();
 372   std::vector<int> doubleDaInd(sz);
 373   std::vector<int> timeDiscrInt;
 374   tinyInfo.resize(sz2+5*sz+3);
 375   tinyInfo[0]=(int)sz;
 376   tinyInfo[1]=sz2;
 377   for(std::size_t i=0;i<sz;i++)
 378     {
 379       std::vector<double> tmp;
 380       std::vector<int> tmp2;
 381       _fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationDbleInformation2(tmp);
 382       _fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationIntInformation2(tmp2);
 383       tinyInfo[3*sz+3+i]=(int)tmp.size();
 384       tinyInfo[4*sz+3+i]=(int)tmp2.size();
 385       tinyInfo2.insert(tinyInfo2.end(),tmp.begin(),tmp.end());
 386       timeDiscrInt.insert(timeDiscrInt.end(),tmp2.begin(),tmp2.end());
 387     }
 388   int sz3=(int)timeDiscrInt.size();
 389   tinyInfo[2]=sz3;
 390   //
 391   for(std::size_t i=0;i<sz;i++)
 392     tinyInfo[i+3]=refs[i];
 393   for(std::size_t i=0;i<sz;i++)
 394     tinyInfo[i+sz+3]=(int)refs2[i].size();
 395   for(std::size_t i=0;i<sz;i++)
 396     tinyInfo[i+2*sz+3]=(int)_fs[i]->getTimeDiscretization();
 397   int k=0;
 398   for(std::size_t i=0;i<sz;i++)
 399     for(std::vector<int>::const_iterator it=refs2[i].begin();it!=refs2[i].end();it++,k++)
 400       tinyInfo[5*sz+k+3]=*it;
 401   tinyInfo.insert(tinyInfo.end(),timeDiscrInt.begin(),timeDiscrInt.end());//tinyInfo has lgth==sz3+sz2+5*sz+3
 402 }
 403
 404 void MEDCouplingMultiFields::finishUnserialization(const std::vector<int>& tinyInfoI, const std::vector<double>& tinyInfoD,
 405                                                    const std::vector<MEDCouplingFieldTemplate *>& ft, const std::vector<MEDCouplingMesh *>& ms,
 406                                                    const std::vector<DataArrayDouble *>& das)
 407 {
 408   int sz=tinyInfoI[0];
 409   _fs.resize(sz);
 410   int sz2=tinyInfoI[1];
 411   // dealing with ft with no mesh set.
 412   for(int i=0;i<sz;i++)
 413     {
 414       int meshId=tinyInfoI[3+i];
 415       if(meshId!=-1)
 416         ft[i]->setMesh(ms[meshId]);
 417     }
 418   // dealing with fieldtemplate->fielddouble
 419   int k=0;
 420   int offI=0;
 421   int offD=0;
 422   for(int i=0;i<sz;i++)
 423     {
 424       _fs[i]=MEDCouplingFieldDouble::New(ft[i],(TypeOfTimeDiscretization)tinyInfoI[2*sz+3+i]);
 425       int sz3=tinyInfoI[sz+i+3];
 426       std::vector<DataArrayDouble *> tmp(sz3);
 427       for(int j=0;j<sz3;j++,k++)
 428         {
 429           int daId=tinyInfoI[5*sz+k+3];
 430           if(daId!=-1)
 431             tmp[j]=das[daId];
 432           else
 433             tmp[j]=0;
 434         }
 435       _fs[i]->setArrays(tmp);
 436       // time discr tiny info
 437       int lgthI=tinyInfoI[4*sz+3+i];
 438       int lgthD=tinyInfoI[3*sz+3+i];
 439       //
 440       std::vector<int> tdInfoI(tinyInfoI.begin()+sz2+5*sz+3+offI,tinyInfoI.begin()+sz2+5*sz+3+offI+lgthI);
 441       std::vector<double> tdInfoD(tinyInfoD.begin()+offD,tinyInfoD.begin()+offD+lgthD);
 442       _fs[i]->getTimeDiscretizationUnderGround()->finishUnserialization2(tdInfoI,tdInfoD);
 443       //
 444       offI+=lgthI;
 445       offD+=lgthD;
 446     }
 447 }