src/MEDLoader/MEDFileBlowStrEltUp.cxx

   1 // Copyright (C) 2007-2017  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 (EDF R&D)
  20
  21 #include "MEDFileBlowStrEltUp.hxx"
  22 #include "MEDCouplingFieldDouble.hxx"
  23 #include "MEDFileFieldVisitor.hxx"
  24 #include "MEDCouplingPartDefinition.hxx"
  25 #include "MCAuto.txx"
  26
  27 using namespace MEDCoupling;
  28
  29 const char MEDFileBlowStrEltUp::MED_BALL_STR[]="MED_BALL";
  30
  31 MEDFileBlowStrEltUp::MEDFileBlowStrEltUp(const MEDFileFields *fsOnlyOnSE, const MEDFileMeshes *ms, const MEDFileStructureElements *ses)
  32 {
  33   if(!fsOnlyOnSE || !ms || !ses)
  34     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp constructor : NULL input pointer !");
  35   _ms.takeRef(ms); _ses.takeRef(ses);
  36   std::vector< std::pair<std::string,std::string> > ps;
  37   fsOnlyOnSE->getMeshSENames(ps);
  38   std::size_t sz(ps.size());
  39   _elts.resize(sz);
  40   for(std::size_t i=0;i<sz;i++)
  41     {
  42       const std::pair<std::string,std::string>& p(ps[i]);
  43       MCAuto<MEDFileFields> f(fsOnlyOnSE->partOfThisLyingOnSpecifiedMeshSEName(p.first,p.second));
  44       _elts[i]=f;
  45     }
  46   for(std::size_t i=0;i<sz;i++)
  47     {
  48       const std::pair<std::string,std::string>& p(ps[i]);
  49       MEDFileMesh *mesh(_ms->getMeshWithName(p.first));
  50       if(!mesh)
  51         throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp : NULL mesh !");
  52       MEDFileUMesh *umesh(dynamic_cast<MEDFileUMesh *>(mesh));
  53       if(!umesh)
  54         throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp : Blow up of Stru Elt not managed yet for unstructured meshes !");
  55     }
  56 }
  57
  58 /*!
  59  * \param [in] mesh - The mesh containing structure element called \a seName. After the call of this method the Structure elements parts will be removed.
  60  * \param [out] mOut - the physical mesh of the structure element \a seName in mesh \a mesh
  61  * \param [out] fsOut - the list of var attribute of structure element \a seName - \b WARNING no time steps here
  62  */
  63 MCAuto<MEDFileEltStruct4Mesh> MEDFileBlowStrEltUp::dealWithSEInMesh(const std::string& seName, MEDFileUMesh *mesh, MCAuto<MEDFileUMesh>& mOut, MCAuto<MEDFileFields>& fsOut) const
  64 {
  65   if(!mesh)
  66     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithSEInMesh : null pointer !");
  67   if(seName==MED_BALL_STR)
  68     {
  69       MCAuto<MEDFileEltStruct4Mesh> ret(dealWithMEDBALLInMesh(mesh,mOut,fsOut));
  70       mesh->killStructureElements();
  71       return ret;
  72     }
  73   throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithSEInMesh : only MED_BALL is managed for the moment, but if you are interested please send spec to anthony.geay@edf.fr !");
  74 }
  75
  76 MCAuto<MEDFileEltStruct4Mesh> MEDFileBlowStrEltUp::dealWithMEDBALLInMesh(const MEDFileUMesh *mesh, MCAuto<MEDFileUMesh>& mOut, MCAuto<MEDFileFields>& fsOut) const
  77 {
  78   mOut=MEDFileUMesh::New(); fsOut=MEDFileFields::New();
  79   const std::vector< MCAuto<MEDFileEltStruct4Mesh> >& strs(mesh->getAccessOfUndergroundEltStrs());
  80   MCAuto<MEDFileEltStruct4Mesh> zeStr;
  81   for(std::vector< MCAuto<MEDFileEltStruct4Mesh> >::const_iterator it=strs.begin();it!=strs.end();it++)
  82     {
  83       if((*it)->getGeoTypeName()==MED_BALL_STR)
  84         {
  85           zeStr=*it;
  86           break;
  87         }
  88     }
  89   if(zeStr.isNull())
  90     {
  91       std::ostringstream oss; oss << "MEDFileBlowStrEltUp::dealWithMEDBALLInMesh : no geo type with name " <<  MED_BALL_STR << " in " << mesh->getName() << " !";
  92       throw INTERP_KERNEL::Exception(oss.str());
  93     }
  94   const DataArrayDouble *coo(mesh->getCoords());
  95   if(!coo)
  96     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLInMesh : null coords !");
  97   MCAuto<DataArrayInt> conn(zeStr->getConn());
  98   if(conn.isNull())
  99     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLInMesh : null connectivity !");
 100   conn->checkAllocated();
 101   if(conn->getNumberOfComponents()!=1)
 102     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLInMesh : excepted to be single compo !");
 103   int nbCells(conn->getNumberOfTuples());
 104   MCAuto<DataArrayDouble> connOut(coo->selectByTupleIdSafe(conn->begin(),conn->end()));
 105   MCAuto<MEDCouplingUMesh> mcOut(MEDCouplingUMesh::Build0DMeshFromCoords(connOut));
 106   mcOut->setName(BuildNewMeshName(mesh->getName(),MED_BALL_STR));
 107   mOut->setMeshAtLevel(0,mcOut);
 108   const DataArrayInt *ff1(mesh->getFamilyFieldAtLevel(1));
 109   if(ff1)
 110     {
 111       MCAuto<DataArrayInt> ff1o(ff1->selectByTupleIdSafe(conn->begin(),conn->end()));
 112       mOut->setFamilyFieldArr(1,ff1o);
 113     }
 114   const DataArrayInt *nf1(mesh->getNumberFieldAtLevel(1));
 115   if(nf1)
 116     {
 117       MCAuto<DataArrayInt> nf1o(nf1->selectByTupleIdSafe(conn->begin(),conn->end()));
 118       mOut->setRenumFieldArr(1,nf1o);
 119     }
 120   MCAuto<MEDFileUMeshPerTypeCommon> md(zeStr->getMeshDef());
 121   const DataArrayInt *ff0(md->getFam());
 122   if(ff0)
 123     mOut->setFamilyFieldArr(0,const_cast<DataArrayInt *>(ff0));
 124   const DataArrayInt *nf0(md->getNum());
 125   if(nf0)
 126     mOut->setRenumFieldArr(0,const_cast<DataArrayInt *>(nf0));
 127   mOut->copyFamGrpMapsFrom(*mesh);
 128   const std::vector< MCAuto<DataArray> >& vars(zeStr->getVars());
 129   for(std::vector< MCAuto<DataArray> >::const_iterator it=vars.begin();it!=vars.end();it++)
 130     {
 131       const DataArray *elt(*it);
 132       if(!elt)
 133         continue;
 134       {
 135         const DataArrayDouble *eltC(dynamic_cast<const DataArrayDouble *>(elt));
 136         if(eltC)
 137           {
 138             MCAuto<MEDFileFieldMultiTS> fmts(MEDFileFieldMultiTS::New());
 139             MCAuto<MEDFileField1TS> f1ts(MEDFileField1TS::New());
 140             MCAuto<MEDCouplingFieldDouble> f(MEDCouplingFieldDouble::New(ON_NODES));
 141             f->setMesh(mcOut);
 142             f->setArray(const_cast<DataArrayDouble *>(eltC));
 143             f->setName(eltC->getName());
 144             f1ts->setFieldNoProfileSBT(f);
 145             fmts->pushBackTimeStep(f1ts);
 146             fsOut->pushField(fmts);
 147           }
 148       }
 149     }
 150   return zeStr;
 151 }
 152
 153 /*!
 154  * \param [in] fs - fields lying all on same mesh and on same structure element
 155  * \param [in] zeStr - ze structure of current structure element
 156  * \param [in] varAtt - fields containing var att of current structure element. WARNING at this stage the number of iteration are equal to one for each field in \a varAtt
 157  * \param [out] zeOutputs - ze fields that are the concatenation of fields in \a fs transformed and those in \a varAtt normalized in time space
 158  */
 159 void MEDFileBlowStrEltUp::dealWithSEInFields(const std::string& seName, const MEDFileFields *fs, const MEDFileEltStruct4Mesh *zeStr, const MEDFileFields *varAtt, MEDFileFields *zeOutputs) const
 160 {
 161   if(!fs)
 162     throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithSEInFields : null pointer !");
 163   if(seName==MED_BALL_STR)
 164     {
 165       dealWithMEDBALLSInFields(fs,zeStr,varAtt,zeOutputs);
 166       return ;
 167     }
 168   throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithSEInFields : only MED_BALL is managed for the moment, but if you are interested please send spec to anthony.geay@edf.fr !");
 169 }
 170
 171 void MEDFileBlowStrEltUp::dealWithMEDBALLSInFields(const MEDFileFields *fs, const MEDFileEltStruct4Mesh *zeStr, const MEDFileFields *varAtt, MEDFileFields *zeOutputs) const
 172 {
 173   int nbf(fs->getNumberOfFields());
 174   std::vector< MCAuto<MEDFileAnyTypeFieldMultiTS> > elts0;
 175   std::vector< MEDFileAnyTypeFieldMultiTS * > elts1;
 176   std::string zeMeshName;
 177   for(int i=0;i<nbf;i++)
 178     {
 179       MCAuto<MEDFileAnyTypeFieldMultiTS> elt(fs->getFieldAtPos(i));
 180       MCAuto<MEDFileAnyTypeFieldMultiTS> eltOut(elt->buildNewEmpty());
 181       int nbTS(elt->getNumberOfTS());
 182       for(int j=0;j<nbTS;j++)
 183         {
 184           MCAuto<MEDFileAnyTypeField1TS> eltt(elt->getTimeStepAtPos(j));
 185           MCAuto<MEDFileAnyTypeField1TS> elttOut(eltt->deepCopy());
 186           std::string meshName(eltt->getMeshName());
 187           zeMeshName=BuildNewMeshName(meshName,MED_BALL_STR);
 188           elttOut->setMeshName(zeMeshName);
 189           elttOut->convertMedBallIntoClassic();
 190           eltOut->pushBackTimeStep(elttOut);
 191         }
 192       elts0.push_back(eltOut); elts1.push_back(eltOut);
 193     }
 194   //
 195   const MEDFileMesh *zeCurrentMesh(_ms->getMeshWithName(zeMeshName));
 196   //
 197   std::size_t ii(0);
 198   std::vector< std::vector<MEDFileAnyTypeFieldMultiTS *> > sp(MEDFileAnyTypeFieldMultiTS::SplitIntoCommonTimeSeries(elts1));
 199   for(std::vector< std::vector<MEDFileAnyTypeFieldMultiTS *> >::const_iterator it0=sp.begin();it0!=sp.end();it0++,ii++)
 200     {
 201       std::vector< MCAuto<MEDFileFastCellSupportComparator> > fsc;
 202       std::vector< std::vector<MEDFileAnyTypeFieldMultiTS *> > sp2(MEDFileAnyTypeFieldMultiTS::SplitPerCommonSupport(*it0,zeCurrentMesh,fsc));
 203       std::size_t jj(0);
 204       for(std::vector< std::vector<MEDFileAnyTypeFieldMultiTS *> >::const_iterator it1=sp2.begin();it1!=sp2.end();it1++,jj++)
 205         {
 206           for(std::vector<MEDFileAnyTypeFieldMultiTS *>::const_iterator it2=(*it1).begin();it2!=(*it1).end();it2++)
 207             zeOutputs->pushField(*it2);
 208           // The most exciting part. Users that put profiles on struct elements part of fields. Reduce var att.
 209           if((*it1).size()<1)
 210             throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : take a deep breath !");
 211           MCAuto<MEDFileAnyTypeField1TS> zeGuideForPfl;// This var is the reference for pfl management.
 212           {
 213             if(!(*it1)[0])
 214               throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : take a deep breath 2 !");
 215             int pdm((*it1)[0]->getNumberOfTS());
 216             if(pdm<1)
 217               throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : take a deep breath 3 !");
 218             zeGuideForPfl=(*it1)[0]->getTimeStepAtPos(0);
 219           }
 220           if(zeGuideForPfl.isNull())
 221             throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : take a deep breath 4 !");
 222           std::vector<std::string> pfls(zeGuideForPfl->getPflsReallyUsed());
 223           if(pfls.size()>=2)
 224             throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : drink less coffee");
 225           MCAuto<DataArrayInt> pflMyLove;
 226           if(pfls.size()==1)
 227             pflMyLove.takeRef(zeGuideForPfl->getProfile(pfls[0]));
 228           // Yeah we have pfls
 229           std::vector<double> t2s;
 230           std::vector< std::pair<int,int> > t1s((*it1)[0]->getTimeSteps(t2s));
 231           std::size_t nbTS3(t2s.size());
 232           int nbf2(varAtt->getNumberOfFields());
 233           for(int i=0;i<nbf2;i++)
 234             {
 235               MCAuto<MEDFileAnyTypeFieldMultiTS> elt(varAtt->getFieldAtPos(i));
 236               int nbTS2(elt->getNumberOfTS());
 237               if(nbTS2!=1)
 238                 throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::dealWithMEDBALLSInFields : internal error ! The dealWithMEDBALLInMesh is expected to return a single TS !");
 239               MCAuto<MEDFileAnyTypeField1TS> elt2(elt->getTimeStepAtPos(0));
 240               MCAuto<MEDFileAnyTypeFieldMultiTS> elt4(elt->buildNewEmpty());
 241               for(std::size_t j=0;j<nbTS3;j++)
 242                 {
 243                   MCAuto<MEDFileAnyTypeField1TS> elt3(elt2->deepCopy());
 244                   elt3->setTime(t1s[j].first,t1s[j].second,t2s[j]);
 245                   elt3->setName(BuildVarAttName(ii,sp.size(),jj,sp2.size(),elt3->getName()));
 246                   if(pflMyLove.isNotNull())
 247                     elt3->makeReduction(INTERP_KERNEL::NORM_ERROR,ON_NODES,pflMyLove);
 248                   elt4->pushBackTimeStep(elt3);
 249                 }
 250               zeOutputs->pushField(elt4);
 251             }
 252         }
 253     }
 254 }
 255
 256 void MEDFileBlowStrEltUp::generate(MEDFileMeshes *msOut, MEDFileFields *allZeOutFields)
 257 {
 258   for(std::vector< MCAuto<MEDFileFields> >::iterator elt=_elts.begin();elt!=_elts.end();elt++)
 259     {
 260       std::vector< std::pair<std::string,std::string> > ps;
 261       (*elt)->getMeshSENames(ps);
 262       if(ps.size()!=1)
 263         throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::generateMeshes : internal error !");
 264       MEDFileMesh *mesh(_ms->getMeshWithName(ps[0].first));
 265       if(!mesh)
 266         throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::generateMeshes : NULL mesh !");
 267       MEDFileUMesh *umesh(dynamic_cast<MEDFileUMesh *>(mesh));
 268       if(!umesh)
 269         throw INTERP_KERNEL::Exception("MEDFileBlowStrEltUp::generateMeshes : Blow up of Stru Elt not managed yet for unstructured meshes !");
 270       //
 271       MCAuto<MEDFileFields> classicalSEFields(splitFieldsPerLoc(*elt,umesh,msOut,allZeOutFields));
 272       if(classicalSEFields.isNotNull())
 273         {
 274           MCAuto<MEDFileUMesh> mOut;
 275           MCAuto<MEDFileFields> fsOut1;
 276           MCAuto<MEDFileEltStruct4Mesh> zeStr(dealWithSEInMesh(ps[0].second,umesh,mOut,fsOut1));
 277           msOut->pushMesh(mOut);
 278           dealWithSEInFields(ps[0].second,classicalSEFields,zeStr,fsOut1,allZeOutFields);
 279         }
 280     }
 281 }
 282
 283 std::string MEDFileBlowStrEltUp::BuildNewMeshName(const std::string& meshName, const std::string& seName)
 284 {
 285   std::ostringstream mNameOut;
 286   mNameOut << meshName << "_" << seName;
 287   return mNameOut.str();
 288 }
 289
 290 std::string MEDFileBlowStrEltUp::BuildVarAttName(std::size_t iPart, std::size_t totINbParts, std::size_t jPart, std::size_t totJNbParts, const std::string& name)
 291 {
 292   if(totINbParts==1 && totJNbParts==1)
 293     return name;
 294   std::ostringstream oss;
 295   oss << name << "@" << iPart << "@" << jPart;
 296   return oss.str();
 297 }
 298
 299 void MEDFileBlowStrEltUp::DealWithSE(MEDFileFields *fs, MEDFileMeshes *ms, const MEDFileStructureElements *ses)
 300 {
 301   MCAuto<MEDFileFields> fsSEOnly(fs->partOfThisOnStructureElements());
 302   fs->killStructureElements();
 303   MEDFileBlowStrEltUp bu(fsSEOnly,ms,ses);
 304   bu.generate(ms,fs);
 305   fs->killStructureElementsInGlobs();
 306 }
 307
 308 //
 309
 310 class FieldWalker2
 311 {
 312 public:
 313   FieldWalker2(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd);
 314   std::string getLoc() const { return _loc; }
 315   std::string getPfl() const { return _pfl; }
 316   bool isClassic() const { return _is_classic; }
 317   bool operator!=(const FieldWalker2& other) const;
 318   bool operator==(const FieldWalker2& other) const;
 319   const SlicePartDefinition *getPartDef() const { return _pd; }
 320 private:
 321   std::string _loc;
 322   std::string _pfl;
 323   bool _is_classic;
 324   MCAuto<SlicePartDefinition> _pd;
 325 };
 326
 327 class LocInfo
 328 {
 329 public:
 330   LocInfo() { }
 331   LocInfo(const std::vector<FieldWalker2>& fw);
 332   bool operator==(const LocInfo& other) const { return _locs==other._locs && _pfl==other._pfl; }
 333   void push(const std::string& loc, const std::string& pfl) { checkUniqueLoc(loc); _locs.push_back(loc); _pfl.push_back(pfl); }
 334   MCAuto<MEDFileUMesh> generateNonClassicalData(int zePos, const MEDFileUMesh *mesh, const MEDFileFieldGlobsReal *globs) const;
 335   const PartDefinition *getPartDef() const { return _pd; }
 336 private:
 337   void checkUniqueLoc(const std::string& loc) const;
 338   static MCAuto<DataArrayDouble> BuildMeshFromAngleVrille(INTERP_KERNEL::NormalizedCellType gt, const DataArrayDouble *angleDeVrille, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs);
 339   static MCAuto<DataArrayDouble> BuildMeshFromEpaisseur(INTERP_KERNEL::NormalizedCellType gt, const DataArrayDouble *thikness, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs);
 340   static MCAuto<DataArrayDouble> BuildMeshPipeSEG3(const DataArrayDouble *angle, const DataArrayDouble *scale, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs);
 341   static MCAuto<MEDCouplingUMesh> BuildMeshCommon(INTERP_KERNEL::NormalizedCellType gt, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs, MCAuto<DataArrayDouble>& ptsForLoc);
 342   static MCAuto<DataArrayDouble> BuildMeshFromStructure(INTERP_KERNEL::NormalizedCellType gt, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs);
 343 public:
 344   static const char ANGLE_DE_VRILLE[];
 345   static const char ANGLE[];
 346   static const char SCALE[];
 347   static const char EPAISSEUR[];
 348 private:
 349   std::vector<std::string> _locs;
 350   std::vector<std::string> _pfl;
 351   MCAuto<PartDefinition> _pd;
 352 };
 353
 354 const char LocInfo::ANGLE_DE_VRILLE[]="ANGLE DE VRILLE";
 355
 356 const char LocInfo::ANGLE[]="ANGLE";
 357
 358 const char LocInfo::SCALE[]="SCALE";
 359
 360 const char LocInfo::EPAISSEUR[]="EPAISSEUR";
 361
 362 LocInfo::LocInfo(const std::vector<FieldWalker2>& fw)
 363 {
 364   std::size_t sz(fw.size());
 365   _locs.resize(sz); _pfl.resize(sz);
 366   if(sz>0)
 367     _pd=fw[0].getPartDef()->deepCopy();
 368   for(std::size_t i=0;i<sz;i++)
 369     {
 370       _locs[i]=fw[i].getLoc();
 371       _pfl[i]=fw[i].getPfl();
 372       if(i>0)
 373         _pd=(*_pd)+(*(fw[i].getPartDef()));
 374     }
 375 }
 376
 377 void LocInfo::checkUniqueLoc(const std::string& loc) const
 378 {
 379   if(std::find(_locs.begin(),_locs.end(),loc)!=_locs.end())
 380     {
 381       std::ostringstream oss; oss << "LocInfo::checkUniqueLoc : loc \"" << loc << "\" already exists !";
 382       throw INTERP_KERNEL::Exception(oss.str());
 383     }
 384 }
 385
 386 MCAuto<MEDCouplingUMesh> LocInfo::BuildMeshCommon(INTERP_KERNEL::NormalizedCellType gt, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs, MCAuto<DataArrayDouble>& ptsForLoc)
 387 {
 388   MCAuto<DataArrayInt> conn(zeStr->getConn());
 389   conn=conn->deepCopy(); conn->rearrange(1);
 390   MCAuto<MEDCouplingUMesh> geoMesh;
 391   {
 392     MCAuto<MEDCoupling1SGTUMesh> umesh(MEDCoupling1SGTUMesh::New("",gt));
 393     umesh->setCoords(mesh->getCoords());
 394     umesh->setNodalConnectivity(conn);
 395     geoMesh=umesh->buildUnstructured();
 396   }
 397   //
 398   if(!pfl.empty())
 399     {
 400       const DataArrayInt *pflArr(globs->getProfile(pfl));
 401       geoMesh=geoMesh->buildPartOfMySelf(pflArr->begin(),pflArr->end(),true);
 402     }
 403   //
 404   MCAuto<MEDCouplingFieldDouble> fakeF(MEDCouplingFieldDouble::New(ON_GAUSS_PT));
 405   fakeF->setMesh(geoMesh);
 406   fakeF->setGaussLocalizationOnType(gt,loc.getRefCoords(),loc.getGaussCoords(),loc.getGaussWeights());
 407   ptsForLoc=fakeF->getLocalizationOfDiscr();
 408   //
 409   return geoMesh;
 410 }
 411
 412 MCAuto<DataArrayDouble> LocInfo::BuildMeshFromAngleVrille(INTERP_KERNEL::NormalizedCellType gt, const DataArrayDouble *angleDeVrille, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs)
 413 {
 414   MCAuto<DataArrayDouble> ptsForLoc;
 415   MCAuto<MEDCouplingUMesh> geoMesh(BuildMeshCommon(gt,pfl,loc,zeStr,mesh,section,globs,ptsForLoc));
 416   //
 417   MCConstAuto<DataArrayDouble> angleVrille;
 418   if(!pfl.empty())
 419     {
 420       const DataArrayInt *pflArr(globs->getProfile(pfl));
 421       angleVrille=angleDeVrille->selectByTupleIdSafe(pflArr->begin(),pflArr->end());
 422     }
 423   else
 424     angleVrille.takeRef(angleDeVrille);
 425   //
 426   MCAuto<MEDCouplingFieldDouble> dir(geoMesh->buildDirectionVectorField());
 427   MCAuto<DataArrayDouble> rot(dir->getArray()->fromCartToSpher());
 428   int nbCompo(ptsForLoc->getNumberOfComponents());
 429   MCAuto<DataArrayDouble> secPts(section->getCoords()->changeNbOfComponents(nbCompo,0.));
 430   int nbSecPts(secPts->getNumberOfTuples()),nbCells(geoMesh->getNumberOfCells()),nbg(loc.getGaussWeights().size());
 431   {
 432     const int TAB[3]={2,0,1};
 433     std::vector<int> v(TAB,TAB+3);
 434     secPts=secPts->keepSelectedComponents(v);
 435   }
 436   const double CENTER[3]={0.,0.,0.},AX0[3]={0.,0.,1.};
 437   double AX1[3]; AX1[2]=0.;
 438   std::vector< MCAuto<DataArrayDouble> > arrs(nbCells*nbg);
 439   for(int j=0;j<nbCells;j++)
 440     {
 441       MCAuto<DataArrayDouble> p(secPts->deepCopy());
 442       double ang0(rot->getIJ(j,2));
 443       DataArrayDouble::Rotate3DAlg(CENTER,AX0,ang0,nbSecPts,p->begin(),p->getPointer());
 444       AX1[0]=-sin(ang0); AX1[1]=cos(ang0);// rot Oy around OZ
 445       double ang1(M_PI/2.-rot->getIJ(j,1));
 446       DataArrayDouble::Rotate3DAlg(CENTER,AX1,-ang1,nbSecPts,p->begin(),p->getPointer());
 447       DataArrayDouble::Rotate3DAlg(CENTER,dir->getArray()->begin()+j*3,angleVrille->getIJ(j,0),nbSecPts,p->begin(),p->getPointer());
 448       for(int l=0;l<nbg;l++)
 449         {
 450           MCAuto<DataArrayDouble> p2(p->deepCopy());
 451           for(int k=0;k<nbCompo;k++)
 452             p2->applyLin(1.,ptsForLoc->getIJ(j*nbg+l,k),k);
 453           arrs[j*nbg+l]=p2;
 454         }
 455     }
 456   std::vector<const DataArrayDouble *> arrs2(VecAutoToVecOfCstPt(arrs));
 457   MCAuto<DataArrayDouble> resu(DataArrayDouble::Aggregate(arrs2));
 458   return resu;
 459 }
 460
 461 MCAuto<DataArrayDouble> LocInfo::BuildMeshFromEpaisseur(INTERP_KERNEL::NormalizedCellType gt, const DataArrayDouble *thikness, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs)
 462 {
 463   MCAuto<DataArrayDouble> ptsForLoc;
 464   MCAuto<MEDCouplingUMesh> geoMesh(BuildMeshCommon(gt,pfl,loc,zeStr,mesh,section,globs,ptsForLoc));
 465   int nbSecPts(section->getNumberOfNodes()),nbCells(geoMesh->getNumberOfCells()),nbg(loc.getGaussWeights().size());
 466   MCConstAuto<DataArrayDouble> zeThikness;
 467   if(!pfl.empty())
 468     {
 469       const DataArrayInt *pflArr(globs->getProfile(pfl));
 470       geoMesh=geoMesh->buildPartOfMySelf(pflArr->begin(),pflArr->end(),true);
 471       zeThikness=thikness->selectByTupleIdSafe(pflArr->begin(),pflArr->end());
 472     }
 473   else
 474     zeThikness.takeRef(thikness);
 475   MCAuto<DataArrayDouble> orthoArr;
 476   {
 477     MCAuto<MEDCouplingFieldDouble> ortho(geoMesh->buildOrthogonalField());
 478     orthoArr.takeRef(ortho->getArray());
 479   }
 480   int nbCompo(orthoArr->getNumberOfComponents());
 481   MCAuto<DataArrayDouble> secPts(section->getCoords()->duplicateEachTupleNTimes(nbCompo));
 482   secPts->rearrange(nbCompo);
 483   std::vector< MCAuto<DataArrayDouble> > arrs(nbCells*nbg);
 484   for(int j=0;j<nbCells;j++)
 485     {
 486       double thck(zeThikness->getIJ(j,0));
 487       MCAuto<DataArrayDouble> fact(DataArrayDouble::New()); fact->alloc(1,nbCompo);
 488       std::copy(orthoArr->begin()+j*nbCompo,orthoArr->begin()+(j+1)*nbCompo,fact->getPointer());
 489       std::transform(fact->begin(),fact->end(),fact->getPointer(),std::bind2nd(std::multiplies<double>(),thck/2.));
 490       MCAuto<DataArrayDouble> p(DataArrayDouble::Multiply(secPts,fact));
 491       for(int l=0;l<nbg;l++)
 492         {
 493           MCAuto<DataArrayDouble> p2(p->deepCopy());
 494           for(int k=0;k<nbCompo;k++)
 495             p2->applyLin(1.,ptsForLoc->getIJ(j*nbg+l,k),k);
 496           arrs[j*nbg+l]=p2;
 497         }
 498     }
 499   std::vector<const DataArrayDouble *> arrs2(VecAutoToVecOfCstPt(arrs));
 500   MCAuto<DataArrayDouble> resu(DataArrayDouble::Aggregate(arrs2));
 501   return resu;
 502 }
 503
 504 MCAuto<DataArrayDouble> LocInfo::BuildMeshPipeSEG3(const DataArrayDouble *angle, const DataArrayDouble *scale, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs)
 505 {
 506   static const char MSG1[]="BuildMeshPipeSEG3 : not recognized pattern ! Send mail to anthony.geay@edf.fr with corresponding MED file !";
 507   MCAuto<DataArrayDouble> ptsForLoc;
 508   MCAuto<MEDCouplingUMesh> geoMesh(BuildMeshCommon(INTERP_KERNEL::NORM_SEG3,pfl,loc,zeStr,mesh,section,globs,ptsForLoc));
 509   int nbSecPts(section->getNumberOfNodes()),nbCells(geoMesh->getNumberOfCells()),nbg(loc.getGaussWeights().size());
 510   MCConstAuto<DataArrayDouble> zeAngle,zeScale;
 511   if(!pfl.empty())
 512     {
 513       const DataArrayInt *pflArr(globs->getProfile(pfl));
 514       geoMesh=geoMesh->buildPartOfMySelf(pflArr->begin(),pflArr->end(),true);
 515       zeAngle=angle->selectByTupleIdSafe(pflArr->begin(),pflArr->end());
 516       zeScale=scale->selectByTupleIdSafe(pflArr->begin(),pflArr->end());
 517     }
 518   else
 519     {
 520       zeAngle.takeRef(angle);
 521       zeScale.takeRef(scale);
 522     }
 523   if(zeAngle->getNumberOfComponents()!=3 || zeScale->getNumberOfComponents()!=2 || nbg!=3)
 524     throw INTERP_KERNEL::Exception(MSG1);
 525   MCAuto<MEDCouplingFieldDouble> dir;
 526   {
 527     MCAuto<MEDCouplingUMesh> geoMesh2(geoMesh->deepCopy());
 528     geoMesh2->convertQuadraticCellsToLinear();
 529     dir=geoMesh2->buildDirectionVectorField();
 530   }
 531   MCAuto<DataArrayDouble> rot(dir->getArray()->fromCartToSpher());
 532   int nbCompo(ptsForLoc->getNumberOfComponents());
 533   MCAuto<DataArrayDouble> secPts(section->getCoords()->changeNbOfComponents(nbCompo,0.));
 534   {
 535     const int TAB[3]={2,0,1};
 536     std::vector<int> v(TAB,TAB+3);
 537     secPts=secPts->keepSelectedComponents(v);
 538   }
 539   const double CENTER[3]={0.,0.,0.},AX0[3]={0.,0.,1.};
 540   double AX1[3]; AX1[2]=0.;
 541   std::vector< MCAuto<DataArrayDouble> > arrs(nbCells*nbg);
 542   for(int j=0;j<nbCells;j++)
 543     {
 544       MCAuto<DataArrayDouble> p(secPts->deepCopy());
 545       double ang0(rot->getIJ(j,2));
 546       DataArrayDouble::Rotate3DAlg(CENTER,AX0,ang0,nbSecPts,p->begin(),p->getPointer());
 547       AX1[0]=-sin(ang0); AX1[1]=cos(ang0);// rot Oy around OZ
 548       double ang1(M_PI/2.-rot->getIJ(j,1));
 549       DataArrayDouble::Rotate3DAlg(CENTER,AX1,-ang1,nbSecPts,p->begin(),p->getPointer());
 550       for(int l=0;l<3;l++)
 551         {
 552           MCAuto<DataArrayDouble> p3(p->deepCopy());
 553           DataArrayDouble::Rotate3DAlg(CENTER,dir->getArray()->begin()+j*3,zeAngle->getIJ(j,l),nbSecPts,p3->begin(),p3->getPointer());
 554           MCAuto<DataArrayDouble> p2(p3->deepCopy());
 555           for(int k=0;k<nbCompo;k++)
 556             p2->applyLin(1.,ptsForLoc->getIJ(j*nbg+l,k),k);
 557           arrs[j*nbg+l]=p2;
 558         }
 559     }
 560   std::vector<const DataArrayDouble *> arrs2(VecAutoToVecOfCstPt(arrs));
 561   MCAuto<DataArrayDouble> resu(DataArrayDouble::Aggregate(arrs2));
 562   return resu;
 563 }
 564
 565 MCAuto<DataArrayDouble> LocInfo::BuildMeshFromStructure(INTERP_KERNEL::NormalizedCellType gt, const std::string& pfl, const MEDFileFieldLoc& loc, const MEDFileEltStruct4Mesh *zeStr, const MEDFileUMesh *mesh, const MEDFileUMesh *section, const MEDFileFieldGlobsReal *globs)
 566 {
 567   static const char MSG1[]="BuildMeshFromStructure : not recognized pattern ! Send mail to anthony.geay@edf.fr with corresponding MED file !";
 568   const std::vector< MCAuto<DataArray> >& vars(zeStr->getVars());
 569   if(vars.size()==1)
 570     {
 571       MCAuto<DataArray> zeArr(vars[0]);
 572       if(zeArr.isNull())
 573         throw INTERP_KERNEL::Exception(MSG1);
 574       MCAuto<DataArrayDouble> zeArr2(DynamicCast<DataArray,DataArrayDouble>(zeArr));
 575       if(zeArr2.isNull())
 576         throw INTERP_KERNEL::Exception(MSG1);
 577       if(zeArr2->getName()==ANGLE_DE_VRILLE || zeArr2->getName()==ANGLE)
 578         return BuildMeshFromAngleVrille(gt,zeArr2,pfl,loc,zeStr,mesh,section,globs);
 579       if(zeArr2->getName()==EPAISSEUR || zeArr2->getName()==SCALE)
 580         return BuildMeshFromEpaisseur(gt,zeArr2,pfl,loc,zeStr,mesh,section,globs);
 581     }
 582   if(vars.size()==2)
 583     {
 584       MCAuto<DataArray> zeArr0(vars[0]),zeArr1(vars[1]);
 585       if(zeArr0.isNull() || zeArr1.isNull())
 586         throw INTERP_KERNEL::Exception(MSG1);
 587       MCAuto<DataArrayDouble> zeArr00(DynamicCastSafe<DataArray,DataArrayDouble>(zeArr0)),zeArr11(DynamicCastSafe<DataArray,DataArrayDouble>(zeArr1));
 588       switch(gt)
 589       {
 590         case INTERP_KERNEL::NORM_SEG3:
 591           {
 592             MCAuto<DataArrayDouble> angle,scale;
 593             if(zeArr00->getName()==ANGLE)
 594               angle=zeArr00;
 595             if(zeArr00->getName()==SCALE)
 596               scale=zeArr00;
 597             if(zeArr11->getName()==ANGLE)
 598               angle=zeArr11;
 599             if(zeArr11->getName()==SCALE)
 600               scale=zeArr11;
 601             if(angle.isNull() || scale.isNull())
 602               throw INTERP_KERNEL::Exception(MSG1);
 603             return BuildMeshPipeSEG3(angle,scale,pfl,loc,zeStr,mesh,section,globs);
 604           }
 605         default:
 606           throw INTERP_KERNEL::Exception(MSG1);
 607       }
 608     }
 609   throw INTERP_KERNEL::Exception(MSG1);
 610 }
 611
 612 MCAuto<MEDFileUMesh> LocInfo::generateNonClassicalData(int zePos, const MEDFileUMesh *mesh, const MEDFileFieldGlobsReal *globs) const
 613 {
 614   static const char MSG1[]="LocInfo::generateNonClassicalData : no spec for GAUSS on StructureElement with more than one cell !";
 615   std::size_t sz(_locs.size());
 616   std::vector< MCAuto<DataArrayDouble> > arrs(sz);
 617   for(std::size_t i=0;i<sz;i++)
 618     {
 619       const MEDFileFieldLoc& loc(globs->getLocalization(_locs[i]));
 620       const MEDFileGTKeeper *gtk(loc.getUndergroundGTKeeper());
 621       const MEDFileGTKeeperDyn *gtk2(dynamic_cast<const MEDFileGTKeeperDyn *>(gtk));
 622       if(!gtk2)
 623         throw INTERP_KERNEL::Exception("LocInfo::generateNonClassicalData : internal error !");
 624       const MEDFileUMesh *meshLoc(gtk2->getMesh()),*section(gtk2->getSection());
 625       const MEDFileStructureElement *se(gtk2->getSE());
 626       INTERP_KERNEL::NormalizedCellType gt;
 627       {
 628         std::vector<int> nel(meshLoc->getNonEmptyLevels());
 629         if(nel.size()!=1)
 630           throw INTERP_KERNEL::Exception(MSG1);
 631         if(nel[0]!=0)
 632           throw INTERP_KERNEL::Exception(MSG1);
 633         MCAuto<MEDCouplingUMesh> um(meshLoc->getMeshAtLevel(0));
 634         if(um->getNumberOfCells()!=1)
 635           throw INTERP_KERNEL::Exception(MSG1);
 636         gt=um->getTypeOfCell(0);
 637         std::vector<int> v;
 638         um->getNodeIdsOfCell(0,v);
 639         std::size_t sz2(v.size());
 640         for(std::size_t j=0;j<sz2;j++)
 641           if(v[j]!=j)
 642             throw INTERP_KERNEL::Exception(MSG1);
 643       }
 644       const std::vector< MCAuto<MEDFileEltStruct4Mesh> >& strs(mesh->getAccessOfUndergroundEltStrs());
 645       MCAuto<MEDFileEltStruct4Mesh> zeStr;
 646       for(std::vector< MCAuto<MEDFileEltStruct4Mesh> >::const_iterator it=strs.begin();it!=strs.end();it++)
 647         {
 648           if((*it)->getGeoTypeName()==se->getName())
 649             {
 650               zeStr=*it;
 651               break;
 652             }
 653         }
 654       if(zeStr.isNull())
 655         {
 656           std::ostringstream oss; oss << "LocInfo::generateNonClassicalData :  : no geo type with name " <<  se->getName() << " in " << mesh->getName() << " !";
 657           throw INTERP_KERNEL::Exception(oss.str());
 658         }
 659       arrs[i]=BuildMeshFromStructure(gt,_pfl[i],loc,zeStr,mesh,section,globs);
 660     }
 661   std::vector<const DataArrayDouble *> arrs2(VecAutoToVecOfCstPt(arrs));
 662   MCAuto<DataArrayDouble> resu(DataArrayDouble::Aggregate(arrs2));
 663   MCAuto<MEDFileUMesh> ret(MEDFileUMesh::New());
 664   ret->setCoords(resu);
 665   std::ostringstream meshName; meshName << mesh->getName() << "_on_" << sz << "_sections" << "_" << zePos;
 666   ret->setName(meshName.str());
 667   return ret;
 668 }
 669
 670 FieldWalker2::FieldWalker2(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd)
 671 {
 672   _loc=pmptpd->getLocalization();
 673   _pfl=pmptpd->getProfile();
 674   _is_classic=pmptpd->getType()!=ON_GAUSS_PT;
 675   _pd=SlicePartDefinition::New(pmptpd->getStart(),pmptpd->getEnd(),1);
 676 }
 677
 678 bool FieldWalker2::operator!=(const FieldWalker2& other) const
 679 {
 680   return !((*this)==other);
 681 }
 682
 683 bool FieldWalker2::operator==(const FieldWalker2& other) const
 684 {
 685   bool ret2(false);
 686   {
 687     std::string tmp;
 688     ret2=_pd->isEqual(other._pd,tmp);
 689   }
 690   bool ret(_loc==other._loc && _pfl==other._pfl && _is_classic==other._is_classic && ret2);
 691   return ret;
 692 }
 693
 694 class FieldWalker1
 695 {
 696 public:
 697   FieldWalker1(const MEDFileAnyTypeField1TSWithoutSDA *ts):_ts(ts),_pm_pt(0),_nb_mesh(0) { }
 698   void newMeshEntry(const MEDFileFieldPerMesh *fpm);
 699   void endMeshEntry(const MEDFileFieldPerMesh *fpm) { }
 700   void newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 701   void endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 702   void newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd);
 703   void checkOK(const FieldWalker1& other) const;
 704   bool isClassical() const;
 705   std::vector<FieldWalker2> getNonClassicalData() const { return _fw; }
 706 private:
 707   const MEDFileAnyTypeField1TSWithoutSDA *_ts;
 708   const MEDFileFieldPerMeshPerTypeDyn *_pm_pt;
 709   std::vector<FieldWalker2> _fw;
 710   int _nb_mesh;
 711 };
 712
 713 void FieldWalker1::newMeshEntry(const MEDFileFieldPerMesh *fpm)
 714 {
 715   if(_nb_mesh++==1)
 716     throw INTERP_KERNEL::Exception("FieldWalker1::newMeshEntry : multi mesh not supported !");
 717 }
 718
 719 void FieldWalker1::newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt)
 720 {
 721   if(_pm_pt)
 722     throw INTERP_KERNEL::Exception("FieldWalker1::newPerMeshPerTypeEntry : multi SE loc not managed yet !");
 723   const MEDFileFieldPerMeshPerTypeDyn *pmpt2(dynamic_cast<const MEDFileFieldPerMeshPerTypeDyn *>(pmpt));
 724   if(!pmpt2)
 725     throw INTERP_KERNEL::Exception("newPerMeshPerTypeEntry : internal error !");
 726   _pm_pt=pmpt2;
 727 }
 728
 729 void FieldWalker1::endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *)
 730 {
 731   isClassical();
 732 }
 733
 734 void FieldWalker1::newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd)
 735 {
 736   _fw.push_back(FieldWalker2(pmptpd));
 737 }
 738
 739 void FieldWalker1::checkOK(const FieldWalker1& other) const
 740 {
 741   std::size_t sz(_fw.size());
 742   if(other._fw.size()!=sz)
 743     throw INTERP_KERNEL::Exception("checkOK : not OK because size are not the same !");
 744   for(std::size_t i=0;i<sz;i++)
 745     if(_fw[i]!=other._fw[i])
 746       throw INTERP_KERNEL::Exception("checkOK : not OK because an element mismatches !");
 747 }
 748
 749 bool FieldWalker1::isClassical() const
 750 {
 751   if(_fw.empty())
 752     throw INTERP_KERNEL::Exception("FieldWalker1::endPerMeshPerTypeEntry : internal error !");
 753   std::size_t ic(0),inc(0);
 754   for(std::vector<FieldWalker2>::const_iterator it=_fw.begin();it!=_fw.end();it++)
 755     {
 756       if((*it).isClassic())
 757         ic++;
 758       else
 759         inc++;
 760     }
 761   if(ic!=0 && inc!=0)
 762     throw INTERP_KERNEL::Exception("FieldWalker1::endPerMeshPerTypeEntry : mix is not allowed yet !");
 763   return inc==0;
 764 }
 765
 766 class FieldWalker
 767 {
 768 public:
 769   FieldWalker(const MEDFileAnyTypeFieldMultiTSWithoutSDA *f):_f(f) { }
 770   void newTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts);
 771   void endTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts);
 772   void newMeshEntry(const MEDFileFieldPerMesh *fpm);
 773   void endMeshEntry(const MEDFileFieldPerMesh *fpm);
 774   void newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 775   void endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 776   void newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd);
 777 public:
 778   bool isEmpty() const;
 779   bool isClassical() const;
 780   const MEDFileAnyTypeFieldMultiTSWithoutSDA *field() const { return _f; }
 781   std::vector<FieldWalker2> getNonClassicalData() const { return _fw_prev->getNonClassicalData(); }
 782 private:
 783   const MEDFileAnyTypeFieldMultiTSWithoutSDA *_f;
 784   mutable INTERP_KERNEL::AutoCppPtr<FieldWalker1> _fw;
 785   mutable INTERP_KERNEL::AutoCppPtr<FieldWalker1> _fw_prev;
 786 };
 787
 788 bool FieldWalker::isEmpty() const
 789 {
 790   return _fw_prev.isNull();
 791 }
 792
 793 bool FieldWalker::isClassical() const
 794 {
 795   return _fw_prev->isClassical();
 796 }
 797
 798 void FieldWalker::newTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts)
 799 {
 800   _fw=new FieldWalker1(ts);
 801 }
 802
 803 void FieldWalker::endTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts)
 804 {
 805   if(_fw_prev.isNull())
 806     _fw_prev=new FieldWalker1(*_fw);
 807   else
 808     _fw_prev->checkOK(*_fw);
 809   _fw=0;
 810 }
 811
 812 void FieldWalker::newMeshEntry(const MEDFileFieldPerMesh *fpm)
 813 {
 814   _fw->newMeshEntry(fpm);
 815 }
 816
 817 void FieldWalker::endMeshEntry(const MEDFileFieldPerMesh *fpm)
 818 {
 819   _fw->endMeshEntry(fpm);
 820 }
 821
 822 void FieldWalker::newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt)
 823 {
 824   _fw->newPerMeshPerTypeEntry(pmpt);
 825 }
 826
 827 void FieldWalker::endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt)
 828 {
 829   _fw->endPerMeshPerTypeEntry(pmpt);
 830 }
 831
 832 void FieldWalker::newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd)
 833 {
 834   _fw->newPerMeshPerTypePerDisc(pmptpd);
 835 }
 836
 837 // this class splits fields into same
 838 class LocSpliter : public MEDFileFieldVisitor
 839 {
 840 public:
 841   LocSpliter(const MEDFileFieldGlobsReal *globs):_globs(globs),_fw(0) { }
 842   MCAuto<MEDFileFields> getClassical() const { return _classical; }
 843   void generateNonClassicalData(const MEDFileUMesh *mesh, std::vector< MCAuto<MEDFileFields> >& outFields, std::vector< MCAuto<MEDFileUMesh> >& outMeshes) const;
 844 private:
 845   void newFieldEntry(const MEDFileAnyTypeFieldMultiTSWithoutSDA *field);
 846   void endFieldEntry(const MEDFileAnyTypeFieldMultiTSWithoutSDA *field);
 847   //
 848   void newTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts);
 849   void endTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts);
 850   //
 851   void newMeshEntry(const MEDFileFieldPerMesh *fpm);
 852   void endMeshEntry(const MEDFileFieldPerMesh *fpm);
 853   //
 854   void newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 855   void endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt);
 856   //
 857   void newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd);
 858 private:
 859   const MEDFileFieldGlobsReal *_globs;
 860   std::vector< LocInfo > _locs;
 861   std::vector< MCAuto<MEDFileFields> > _fields_on_locs;//size of _locs== size of _fields_on_locs
 862   MCAuto<MEDFileFields> _classical;
 863 private:
 864   mutable INTERP_KERNEL::AutoCppPtr<FieldWalker> _fw;
 865 };
 866
 867 void LocSpliter::newFieldEntry(const MEDFileAnyTypeFieldMultiTSWithoutSDA *field)
 868 {
 869   _fw=new FieldWalker(field);
 870 }
 871
 872 void LocSpliter::endFieldEntry(const MEDFileAnyTypeFieldMultiTSWithoutSDA *field)
 873 {
 874   if(_fw->isEmpty())
 875     return ;
 876   MCAuto<MEDFileAnyTypeFieldMultiTS> f(MEDFileAnyTypeFieldMultiTS::BuildNewInstanceFromContent(const_cast<MEDFileAnyTypeFieldMultiTSWithoutSDA *>(field)));
 877   if(_fw->isClassical())
 878     {
 879       if(_classical.isNull())
 880         {
 881           _classical=MEDFileFields::New();
 882           _classical->shallowCpyGlobs(*_globs);
 883         }
 884       _classical->pushField(f);
 885     }
 886   else
 887     {
 888       std::vector<FieldWalker2> fw2(_fw->getNonClassicalData());
 889       LocInfo elt(fw2);
 890       std::vector< LocInfo >::iterator it(std::find(_locs.begin(),_locs.end(),elt));
 891       if(it==_locs.end())
 892         {
 893           _locs.push_back(elt);
 894           MCAuto<MEDFileFields> zeF(MEDFileFields::New());
 895           zeF->shallowCpyGlobs(*_globs);
 896           zeF->pushField(f);
 897           _fields_on_locs.push_back(zeF);
 898         }
 899       else
 900         {
 901           MCAuto<MEDFileFields> zeF(_fields_on_locs[std::distance(_locs.begin(),it)]);
 902           zeF->pushField(f);
 903         }
 904     }
 905 }
 906
 907 void LocSpliter::generateNonClassicalData(const MEDFileUMesh *mesh, std::vector< MCAuto<MEDFileFields> >& outFields, std::vector< MCAuto<MEDFileUMesh> >& outMeshes) const
 908 {
 909   int i(0);
 910   for(std::vector<LocInfo>::const_iterator it=_locs.begin();it!=_locs.end();it++,i++)
 911     {
 912       MCAuto<MEDFileUMesh> m((*it).generateNonClassicalData(i,mesh,_globs));
 913       outMeshes.push_back(m);
 914       MCAuto<MEDCouplingUMesh> mcm(MEDCouplingUMesh::Build0DMeshFromCoords(m->getCoords()));
 915       mcm->setName(m->getName());
 916       MCAuto<MEDFileFields> fs(_fields_on_locs[i]);
 917       MCAuto<MEDFileFields> outFs(MEDFileFields::New());
 918       for(int j=0;j<fs->getNumberOfFields();j++)
 919         {
 920           MCAuto<MEDFileAnyTypeFieldMultiTS> fmtsNC(fs->getFieldAtPos(j));
 921           MCAuto<MEDFileFieldMultiTS> fmts(DynamicCastSafe<MEDFileAnyTypeFieldMultiTS,MEDFileFieldMultiTS>(fmtsNC));
 922           MCAuto<MEDFileFieldMultiTS> outFmts(MEDFileFieldMultiTS::New());
 923           for(int k=0;k<fmts->getNumberOfTS();k++)
 924             {
 925               MCAuto<MEDFileField1TS> outF1t(MEDFileField1TS::New());
 926               MCAuto<MEDFileField1TS> f1ts(fmts->getTimeStepAtPos(k));
 927               int t2,t3;
 928               double t1(f1ts->getTime(t2,t3));
 929               MCAuto<MEDCouplingFieldDouble> mcf(MEDCouplingFieldDouble::New(ON_NODES));
 930               MCAuto<DataArrayDouble> arr,arr2;
 931               arr.takeRef(f1ts->getUndergroundDataArray());
 932               arr2=arr->selectPartDef((*it).getPartDef());
 933               mcf->setArray(arr2);
 934               mcf->setTime(t1,t2,t3);
 935               mcf->setName(f1ts->getName());
 936               mcf->setMesh(mcm);
 937               outF1t->setFieldNoProfileSBT(mcf);
 938               outFmts->pushBackTimeStep(outF1t);
 939             }
 940           outFs->pushField(outFmts);
 941         }
 942       outFields.push_back(outFs);
 943     }
 944 }
 945
 946 void LocSpliter::newTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts)
 947 {
 948   _fw->newTimeStepEntry(ts);
 949 }
 950
 951 void LocSpliter::endTimeStepEntry(const MEDFileAnyTypeField1TSWithoutSDA *ts)
 952 {
 953   _fw->endTimeStepEntry(ts);
 954 }
 955
 956 void LocSpliter::newMeshEntry(const MEDFileFieldPerMesh *fpm)
 957 {
 958   _fw->newMeshEntry(fpm);
 959 }
 960
 961 void LocSpliter::endMeshEntry(const MEDFileFieldPerMesh *fpm)
 962 {
 963   _fw->endMeshEntry(fpm);
 964 }
 965
 966 void LocSpliter::newPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt)
 967 {
 968   _fw->newPerMeshPerTypeEntry(pmpt);
 969 }
 970
 971 void LocSpliter::endPerMeshPerTypeEntry(const MEDFileFieldPerMeshPerTypeCommon *pmpt)
 972 {
 973   _fw->endPerMeshPerTypeEntry(pmpt);
 974 }
 975
 976 void LocSpliter::newPerMeshPerTypePerDisc(const MEDFileFieldPerMeshPerTypePerDisc *pmptpd)
 977 {
 978   _fw->newPerMeshPerTypePerDisc(pmptpd);
 979 }
 980
 981 void MEDFileBlowStrEltUp::DealWithConflictNames(MEDFileAnyTypeFieldMultiTS *fmtsToAdd, const MEDFileFields *fs)
 982 {
 983   std::vector<std::string> fnames(fs->getFieldsNames());
 984   for(int i=0;i<1000;i++)
 985     {
 986       std::ostringstream oss; oss << fmtsToAdd->getName();
 987       if(i>=1)
 988         oss << "_" << i-1;
 989       if(std::find(fnames.begin(),fnames.end(),oss.str())==fnames.end())
 990         {
 991           fmtsToAdd->setName(oss.str());
 992           return ;
 993         }
 994     }
 995   throw INTERP_KERNEL::Exception("DealWithConflictNames : Eh eh interesting !");
 996 }
 997
 998 MCAuto<MEDFileFields> MEDFileBlowStrEltUp::splitFieldsPerLoc(const MEDFileFields *fields, const MEDFileUMesh *mesh, MEDFileMeshes *msOut, MEDFileFields *allZeOutFields)
 999 {
1000   LocSpliter ls(fields);
1001   fields->accept(ls);
1002   std::vector< MCAuto<MEDFileFields> > outFields;
1003   std::vector< MCAuto<MEDFileUMesh> > outMeshes;
1004   ls.generateNonClassicalData(mesh,outFields,outMeshes);
1005   for(std::vector< MCAuto<MEDFileFields> >::iterator it=outFields.begin();it!=outFields.end();it++)
1006     {
1007       for(int j=0;j<(*it)->getNumberOfFields();j++)
1008         {
1009           MCAuto<MEDFileAnyTypeFieldMultiTS> fmts((*it)->getFieldAtPos(j));
1010           //DealWithConflictNames(fmts,allZeOutFields);// uncomment to have a writable data structure
1011           allZeOutFields->pushField(fmts);
1012         }
1013     }
1014   for(std::vector< MCAuto<MEDFileUMesh> >::iterator it=outMeshes.begin();it!=outMeshes.end();it++)
1015     msOut->pushMesh(*it);
1016   return ls.getClassical();
1017 }