src/MEDCoupling/MEDCouplingMultiFields.cxx

   1 // Copyright (C) 2007-2014  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, or (at your option) any later version.
   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)
  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
 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
 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
 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::size_t MEDCouplingMultiFields::getHeapMemorySizeWithoutChildren() const
 194 {
 195   return 0;
 196 }
 197
 198 std::vector<const BigMemoryObject *> MEDCouplingMultiFields::getDirectChildren() const
 199 {
 200   std::vector<const BigMemoryObject *> ret;
 201   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++)
 202     {
 203       const MEDCouplingFieldDouble *curF(*it);
 204       if(curF)
 205         ret.push_back(curF);
 206     }
 207   return ret;
 208 }
 209
 210 std::vector<MEDCouplingMesh *> MEDCouplingMultiFields::getMeshes() const
 211 {
 212   std::vector<MEDCouplingMesh *> ms;
 213   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++)
 214     {
 215       const MEDCouplingMesh *m=0;
 216       if((const MEDCouplingFieldDouble *)(*it))
 217         m=(*it)->getMesh();
 218       ms.push_back(const_cast<MEDCouplingMesh *>(m));
 219     }
 220   return ms;
 221 }
 222
 223 std::vector<MEDCouplingMesh *> MEDCouplingMultiFields::getDifferentMeshes(std::vector<int>& refs) const
 224 {
 225   refs.resize(_fs.size());
 226   std::vector<MEDCouplingMesh *> ms;
 227   int id=0;
 228   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++,id++)
 229     {
 230       const MEDCouplingMesh *m=0;
 231       if((const MEDCouplingFieldDouble *)(*it))
 232         m=(*it)->getMesh();
 233       if(m)
 234         {
 235           std::vector<MEDCouplingMesh *>::iterator it2=std::find(ms.begin(),ms.end(),m);
 236           if(it2==ms.end())
 237             {
 238               ms.push_back(const_cast<MEDCouplingMesh *>(m));
 239               refs[id]=(int)ms.size()-1;
 240             }
 241           else
 242             refs[id]=(int)std::distance(ms.begin(),it2);
 243         }
 244       else
 245         refs[id]=-1;
 246     }
 247   return ms;
 248 }
 249
 250 std::vector<DataArrayDouble *> MEDCouplingMultiFields::getArrays() const
 251 {
 252   std::vector<DataArrayDouble *> tmp;
 253   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++)
 254     {
 255       std::vector<DataArrayDouble *> tmp2=(*it)->getArrays();
 256       tmp.insert(tmp.end(),tmp2.begin(),tmp2.end());
 257     }
 258   return tmp;
 259 }
 260
 261 std::vector<DataArrayDouble *> MEDCouplingMultiFields::getDifferentArrays(std::vector< std::vector<int> >& refs) const
 262 {
 263   refs.resize(_fs.size());
 264   int id=0;
 265   std::vector<DataArrayDouble *> ret;
 266   for(std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();it!=_fs.end();it++,id++)
 267     {
 268       std::vector<DataArrayDouble *> tmp2;
 269       if((const MEDCouplingFieldDouble *)(*it))
 270         {
 271           tmp2=(*it)->getArrays();
 272           refs[id].resize(tmp2.size());
 273         }
 274       else
 275         refs[id].clear();
 276       int id2=0;
 277       for(std::vector<DataArrayDouble *>::const_iterator it2=tmp2.begin();it2!=tmp2.end();it2++,id2++)
 278         {
 279           if(*it2)
 280             {
 281               std::vector<DataArrayDouble *>::iterator it3=std::find(ret.begin(),ret.end(),*it2);
 282               if(it3==ret.end())
 283                 {
 284                   ret.push_back(*it2);
 285                   refs[id][id2]=(int)ret.size()-1;
 286                 }
 287               else
 288                 refs[id][id2]=(int)std::distance(ret.begin(),it3);
 289             }
 290           else
 291             refs[id][id2]=-1;
 292         }
 293     }
 294   return ret;
 295 }
 296
 297 void MEDCouplingMultiFields::checkCoherency() const
 298 {
 299   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingFieldDouble> >::const_iterator it=_fs.begin();
 300   for(;it!=_fs.end();it++)
 301     {
 302       if((const MEDCouplingFieldDouble *)(*it)==0)
 303         throw INTERP_KERNEL::Exception("MEDCouplingMultiFields::checkCoherency : There is an empty Field in array...");
 304       (*it)->checkCoherency();
 305     }
 306 }
 307
 308 MEDCouplingMultiFields::MEDCouplingMultiFields(const std::vector<MEDCouplingFieldDouble *>& fs):_fs(fs.size())
 309 {
 310   int id=0;
 311   for(std::vector< MEDCouplingFieldDouble * >::const_iterator it=fs.begin();it!=fs.end();it++,id++)
 312     {
 313       if(*it)
 314         (*it)->incrRef();
 315       else
 316         throw INTERP_KERNEL::Exception("MEDCouplingMultiFields constructor : empty field found in vector !");
 317       (*it)->checkCoherency();
 318       _fs[id]=*it;
 319     }
 320 }
 321
 322
 323 /*!
 324  * Performs deepCpy.
 325  */
 326 MEDCouplingMultiFields::MEDCouplingMultiFields(const MEDCouplingMultiFields& other):RefCountObject(other)
 327 {
 328   std::size_t sz=other._fs.size();
 329   _fs.resize(sz);
 330   std::vector<int> refs;
 331   std::vector< std::vector<int> > refs2;
 332   std::vector<MEDCouplingMesh *> ms=other.getDifferentMeshes(refs);
 333   std::size_t msLgh=ms.size();
 334   std::vector< MEDCouplingAutoRefCountObjectPtr<MEDCouplingMesh> > ms2(msLgh);
 335   for(std::size_t i=0;i<msLgh;i++)
 336     ms2[i]=ms[i]->deepCpy();
 337   std::vector<DataArrayDouble *> das=other.getDifferentArrays(refs2);
 338   std::size_t dasLgth=das.size();
 339   std::vector< MEDCouplingAutoRefCountObjectPtr<DataArrayDouble> > das2(dasLgth);
 340   for(std::size_t i=0;i<dasLgth;i++)
 341     das2[i]=das[i]->deepCpy();
 342   for(std::size_t i=0;i<sz;i++)
 343     {
 344       if((const MEDCouplingFieldDouble *)other._fs[i])
 345         {
 346           MEDCouplingFieldTemplate *tmp=MEDCouplingFieldTemplate::New(*other._fs[i]);
 347           _fs[i]=MEDCouplingFieldDouble::New(*tmp,other._fs[i]->getTimeDiscretization());
 348           tmp->decrRef();
 349           if(refs[i]!=-1)
 350             _fs[i]->setMesh(ms2[refs[i]]);
 351           std::size_t nbOfArr=refs2[i].size();
 352           std::vector<DataArrayDouble *> tmp2(nbOfArr);
 353           for(std::size_t j=0;j<nbOfArr;j++)
 354             {
 355               if(refs2[i][j]!=-1)
 356                 tmp2[j]=das2[refs2[i][j]];
 357               else
 358                 tmp2[j]=0;
 359             }
 360           _fs[i]->setArrays(tmp2);
 361           std::vector<int> tinyInfo;
 362           std::vector<double> tinyInfo2;
 363           other._fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationIntInformation2(tinyInfo);
 364           other._fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationDbleInformation2(tinyInfo2);
 365           _fs[i]->getTimeDiscretizationUnderGround()->finishUnserialization2(tinyInfo,tinyInfo2);
 366         }
 367     }
 368 }
 369
 370 MEDCouplingMultiFields::MEDCouplingMultiFields()
 371 {
 372 }
 373
 374 void MEDCouplingMultiFields::getTinySerializationInformation(std::vector<int>& tinyInfo, std::vector<double>& tinyInfo2, int& nbOfDiffMeshes, int& nbOfDiffArr) const
 375 {
 376   std::vector<int> refs;
 377   std::vector<MEDCouplingMesh *> ms=getDifferentMeshes(refs);
 378   nbOfDiffMeshes=(int)ms.size();
 379   std::vector< std::vector<int> > refs2;
 380   std::vector<DataArrayDouble *> fs=getDifferentArrays(refs2);
 381   nbOfDiffArr=(int)fs.size();
 382   //
 383   std::size_t sz=refs.size();//==_fs.size()
 384   int sz2=0;
 385   for(std::size_t i=0;i<sz;i++)
 386     sz2+=(int)refs2[i].size();
 387   //
 388   tinyInfo2.clear();
 389   std::vector<int> doubleDaInd(sz);
 390   std::vector<int> timeDiscrInt;
 391   tinyInfo.resize(sz2+5*sz+3);
 392   tinyInfo[0]=(int)sz;
 393   tinyInfo[1]=sz2;
 394   for(std::size_t i=0;i<sz;i++)
 395     {
 396       std::vector<double> tmp;
 397       std::vector<int> tmp2;
 398       _fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationDbleInformation2(tmp);
 399       _fs[i]->getTimeDiscretizationUnderGround()->getTinySerializationIntInformation2(tmp2);
 400       tinyInfo[3*sz+3+i]=(int)tmp.size();
 401       tinyInfo[4*sz+3+i]=(int)tmp2.size();
 402       tinyInfo2.insert(tinyInfo2.end(),tmp.begin(),tmp.end());
 403       timeDiscrInt.insert(timeDiscrInt.end(),tmp2.begin(),tmp2.end());
 404     }
 405   int sz3=(int)timeDiscrInt.size();
 406   tinyInfo[2]=sz3;
 407   //
 408   for(std::size_t i=0;i<sz;i++)
 409     tinyInfo[i+3]=refs[i];
 410   for(std::size_t i=0;i<sz;i++)
 411     tinyInfo[i+sz+3]=(int)refs2[i].size();
 412   for(std::size_t i=0;i<sz;i++)
 413     tinyInfo[i+2*sz+3]=(int)_fs[i]->getTimeDiscretization();
 414   int k=0;
 415   for(std::size_t i=0;i<sz;i++)
 416     for(std::vector<int>::const_iterator it=refs2[i].begin();it!=refs2[i].end();it++,k++)
 417       tinyInfo[5*sz+k+3]=*it;
 418   tinyInfo.insert(tinyInfo.end(),timeDiscrInt.begin(),timeDiscrInt.end());//tinyInfo has lgth==sz3+sz2+5*sz+3
 419 }
 420
 421 void MEDCouplingMultiFields::finishUnserialization(const std::vector<int>& tinyInfoI, const std::vector<double>& tinyInfoD,
 422                                                    const std::vector<MEDCouplingFieldTemplate *>& ft, const std::vector<MEDCouplingMesh *>& ms,
 423                                                    const std::vector<DataArrayDouble *>& das)
 424 {
 425   int sz=tinyInfoI[0];
 426   _fs.resize(sz);
 427   int sz2=tinyInfoI[1];
 428   // dealing with ft with no mesh set.
 429   for(int i=0;i<sz;i++)
 430     {
 431       int meshId=tinyInfoI[3+i];
 432       if(meshId!=-1)
 433         ft[i]->setMesh(ms[meshId]);
 434     }
 435   // dealing with fieldtemplate->fielddouble
 436   int k=0;
 437   int offI=0;
 438   int offD=0;
 439   for(int i=0;i<sz;i++)
 440     {
 441       _fs[i]=MEDCouplingFieldDouble::New(*ft[i],(TypeOfTimeDiscretization)tinyInfoI[2*sz+3+i]);
 442       int sz3=tinyInfoI[sz+i+3];
 443       std::vector<DataArrayDouble *> tmp(sz3);
 444       for(int j=0;j<sz3;j++,k++)
 445         {
 446           int daId=tinyInfoI[5*sz+k+3];
 447           if(daId!=-1)
 448             tmp[j]=das[daId];
 449           else
 450             tmp[j]=0;
 451         }
 452       _fs[i]->setArrays(tmp);
 453       // time discr tiny info
 454       int lgthI=tinyInfoI[4*sz+3+i];
 455       int lgthD=tinyInfoI[3*sz+3+i];
 456       //
 457       std::vector<int> tdInfoI(tinyInfoI.begin()+sz2+5*sz+3+offI,tinyInfoI.begin()+sz2+5*sz+3+offI+lgthI);
 458       std::vector<double> tdInfoD(tinyInfoD.begin()+offD,tinyInfoD.begin()+offD+lgthD);
 459       _fs[i]->getTimeDiscretizationUnderGround()->finishUnserialization2(tdInfoI,tdInfoD);
 460       //
 461       offI+=lgthI;
 462       offD+=lgthD;
 463     }
 464 }