1 // Copyright (C) 2010-2014 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // Author : Anthony Geay
21 #include "vtkExtractGroup.h"
22 #include "MEDFileFieldRepresentationTree.hxx"
24 #include "vtkAdjacentVertexIterator.h"
25 #include "vtkIntArray.h"
26 #include "vtkCellData.h"
27 #include "vtkPointData.h"
29 #include "vtkStreamingDemandDrivenPipeline.h"
30 #include "vtkUnstructuredGrid.h"
31 #include "vtkMultiBlockDataSet.h"
33 #include "vtkInformationStringKey.h"
34 #include "vtkAlgorithmOutput.h"
35 #include "vtkObjectFactory.h"
36 #include "vtkMutableDirectedGraph.h"
37 #include "vtkMultiBlockDataSet.h"
38 #include "vtkDataSet.h"
39 #include "vtkInformationVector.h"
40 #include "vtkInformation.h"
41 #include "vtkDataArraySelection.h"
42 #include "vtkTimeStamp.h"
43 #include "vtkInEdgeIterator.h"
44 #include "vtkInformationDataObjectKey.h"
45 #include "vtkExecutive.h"
46 #include "vtkVariantArray.h"
47 #include "vtkStringArray.h"
48 #include "vtkDoubleArray.h"
49 #include "vtkCharArray.h"
50 #include "vtkUnsignedCharArray.h"
51 #include "vtkDataSetAttributes.h"
52 #include "vtkDemandDrivenPipeline.h"
53 #include "vtkDataObjectTreeIterator.h"
54 #include "vtkThreshold.h"
59 vtkStandardNewMacro(vtkExtractGroup);
61 vtkCxxSetObjectMacro(vtkExtractGroup, SIL, vtkMutableDirectedGraph);
65 class ExtractGroupStatus
68 ExtractGroupStatus():_status(false) { }
69 ExtractGroupStatus(const char *name);
70 bool getStatus() const { return _status; }
71 void setStatus(bool status) { _status=status; }
72 void cpyStatusFrom(const ExtractGroupStatus& other) { _status=other._status; }
73 std::string getName() const { return _name; }
74 const char *getKeyOfEntry() const { return _ze_key_name.c_str(); }
75 virtual void printMySelf(std::ostream& os) const;
76 virtual bool isSameAs(const ExtractGroupStatus& other) const;
80 std::string _ze_key_name;
83 class ExtractGroupGrp : public ExtractGroupStatus
86 ExtractGroupGrp(const char *name):ExtractGroupStatus(name) { std::ostringstream oss; oss << START << name; _ze_key_name=oss.str(); }
87 void setFamilies(const std::vector<std::string>& fams) { _fams=fams; }
88 const std::vector<std::string>& getFamiliesLyingOn() const { return _fams; }
89 bool isSameAs(const ExtractGroupGrp& other) const;
91 static const char START[];
92 std::vector<std::string> _fams;
95 class ExtractGroupFam : public ExtractGroupStatus
98 ExtractGroupFam(const char *name);
99 void printMySelf(std::ostream& os) const;
100 void fillIdsToKeep(std::set<int>& s) const;
101 int getId() const { return _id; }
102 bool isSameAs(const ExtractGroupFam& other) const;
104 static const char START[];
109 class vtkExtractGroup::vtkExtractGroupInternal
112 void loadFrom(vtkMutableDirectedGraph *sil);
113 int getNumberOfEntries() const;
114 const char *getKeyOfEntry(int i) const;
115 bool getStatusOfEntryStr(const char *entry) const;
116 void setStatusOfEntryStr(const char *entry, bool status);
117 void printMySelf(std::ostream& os) const;
118 std::set<int> getIdsToKeep() const;
119 int getIdOfFamily(const std::string& famName) const;
121 std::map<std::string,int> computeFamStrIdMap() const;
122 const ExtractGroupStatus& getEntry(const char *entry) const;
123 ExtractGroupStatus& getEntry(const char *entry);
125 std::vector<ExtractGroupGrp> _groups;
126 std::vector<ExtractGroupFam> _fams;
129 const char ExtractGroupGrp::START[]="GRP_";
131 const char ExtractGroupFam::START[]="FAM_";
133 ExtractGroupStatus::ExtractGroupStatus(const char *name):_status(false),_name(name)
137 void ExtractGroupStatus::printMySelf(std::ostream& os) const
139 os << " -" << _ze_key_name << "(";
144 os << ")" << std::endl;
147 bool ExtractGroupStatus::isSameAs(const ExtractGroupStatus& other) const
149 return _name==other._name && _ze_key_name==other._ze_key_name;
152 bool ExtractGroupGrp::isSameAs(const ExtractGroupGrp& other) const
154 bool ret(ExtractGroupStatus::isSameAs(other));
156 return _fams==other._fams;
161 void vtkExtractGroup::vtkExtractGroupInternal::loadFrom(vtkMutableDirectedGraph *sil)
163 std::vector<ExtractGroupGrp> oldGrps(_groups); _groups.clear();
164 std::vector<ExtractGroupFam> oldFams(_fams); _fams.clear();
166 vtkAbstractArray *verticesNames(sil->GetVertexData()->GetAbstractArray("Names",idNames));
167 vtkStringArray *verticesNames2(vtkStringArray::SafeDownCast(verticesNames));
170 for(int i=0;i<verticesNames2->GetNumberOfValues();i++)
172 vtkStdString &st(verticesNames2->GetValue(i));
173 if(st=="MeshesFamsGrps")
180 throw INTERP_KERNEL::Exception("There is an internal error ! The tree on server side has not the expected look !");
181 vtkAdjacentVertexIterator *it0(vtkAdjacentVertexIterator::New());
182 sil->GetAdjacentVertices(id0,it0);
184 while(it0->HasNext())
186 vtkIdType id1(it0->Next());
187 std::string meshName((const char *)verticesNames2->GetValue(id1));
188 vtkAdjacentVertexIterator *it1(vtkAdjacentVertexIterator::New());
189 sil->GetAdjacentVertices(id1,it1);
190 vtkIdType idZeGrps(it1->Next());//zeGroups
191 vtkAdjacentVertexIterator *itGrps(vtkAdjacentVertexIterator::New());
192 sil->GetAdjacentVertices(idZeGrps,itGrps);
193 while(itGrps->HasNext())
195 vtkIdType idg(itGrps->Next());
196 ExtractGroupGrp grp((const char *)verticesNames2->GetValue(idg));
197 vtkAdjacentVertexIterator *itGrps2(vtkAdjacentVertexIterator::New());
198 sil->GetAdjacentVertices(idg,itGrps2);
199 std::vector<std::string> famsOnGroup;
200 while(itGrps2->HasNext())
202 vtkIdType idgf(itGrps2->Next());
203 famsOnGroup.push_back(std::string((const char *)verticesNames2->GetValue(idgf)));
205 grp.setFamilies(famsOnGroup);
207 _groups.push_back(grp);
210 vtkIdType idZeFams(it1->Next());//zeFams
212 vtkAdjacentVertexIterator *itFams(vtkAdjacentVertexIterator::New());
213 sil->GetAdjacentVertices(idZeFams,itFams);
214 while(itFams->HasNext())
216 vtkIdType idf(itFams->Next());
217 ExtractGroupFam fam((const char *)verticesNames2->GetValue(idf));
218 _fams.push_back(fam);
224 std::size_t szg(_groups.size()),szf(_fams.size());
225 if(szg==oldGrps.size() && szf==oldFams.size())
228 for(std::size_t i=0;i<szg && isSame;i++)
229 isSame=_groups[i].isSameAs(oldGrps[i]);
230 for(std::size_t i=0;i<szf && isSame;i++)
231 isSame=_fams[i].isSameAs(oldFams[i]);
234 for(std::size_t i=0;i<szg;i++)
235 _groups[i].cpyStatusFrom(oldGrps[i]);
236 for(std::size_t i=0;i<szf;i++)
237 _fams[i].cpyStatusFrom(oldFams[i]);
242 int vtkExtractGroup::vtkExtractGroupInternal::getNumberOfEntries() const
244 std::size_t sz0(_groups.size()),sz1(_fams.size());
245 return (int)(sz0+sz1);
248 const char *vtkExtractGroup::vtkExtractGroupInternal::getKeyOfEntry(int i) const
250 int sz0((int)_groups.size());
252 return _groups[i].getKeyOfEntry();
254 return _fams[i-sz0].getKeyOfEntry();
257 bool vtkExtractGroup::vtkExtractGroupInternal::getStatusOfEntryStr(const char *entry) const
259 const ExtractGroupStatus& elt(getEntry(entry));
260 return elt.getStatus();
263 void vtkExtractGroup::vtkExtractGroupInternal::setStatusOfEntryStr(const char *entry, bool status)
265 ExtractGroupStatus& elt(getEntry(entry));
266 elt.setStatus(status);
269 const ExtractGroupStatus& vtkExtractGroup::vtkExtractGroupInternal::getEntry(const char *entry) const
271 std::string entryCpp(entry);
272 for(std::vector<ExtractGroupGrp>::const_iterator it0=_groups.begin();it0!=_groups.end();it0++)
273 if(entryCpp==(*it0).getKeyOfEntry())
275 for(std::vector<ExtractGroupFam>::const_iterator it0=_fams.begin();it0!=_fams.end();it0++)
276 if(entryCpp==(*it0).getKeyOfEntry())
278 std::ostringstream oss; oss << "vtkExtractGroupInternal::getEntry : no such entry \"" << entry << "\"!";
279 throw INTERP_KERNEL::Exception(oss.str().c_str());
282 ExtractGroupStatus& vtkExtractGroup::vtkExtractGroupInternal::getEntry(const char *entry)
284 std::string entryCpp(entry);
285 for(std::vector<ExtractGroupGrp>::iterator it0=_groups.begin();it0!=_groups.end();it0++)
286 if(entryCpp==(*it0).getKeyOfEntry())
288 for(std::vector<ExtractGroupFam>::iterator it0=_fams.begin();it0!=_fams.end();it0++)
289 if(entryCpp==(*it0).getKeyOfEntry())
291 std::ostringstream oss; oss << "vtkExtractGroupInternal::getEntry : no such entry \"" << entry << "\"!";
292 throw INTERP_KERNEL::Exception(oss.str().c_str());
295 void vtkExtractGroup::vtkExtractGroupInternal::printMySelf(std::ostream& os) const
297 os << "Groups :" << std::endl;
298 for(std::vector<ExtractGroupGrp>::const_iterator it0=_groups.begin();it0!=_groups.end();it0++)
299 (*it0).printMySelf(os);
300 os << "Families :" << std::endl;
301 for(std::vector<ExtractGroupFam>::const_iterator it0=_fams.begin();it0!=_fams.end();it0++)
302 (*it0).printMySelf(os);
305 int vtkExtractGroup::vtkExtractGroupInternal::getIdOfFamily(const std::string& famName) const
307 for(std::vector<ExtractGroupFam>::const_iterator it=_fams.begin();it!=_fams.end();it++)
309 if((*it).getName()==famName)
310 return (*it).getId();
314 ExtractGroupFam::ExtractGroupFam(const char *name):ExtractGroupStatus(name),_id(0)
316 std::size_t pos(_name.find(MEDFileFieldRepresentationLeavesArrays::ZE_SEP));
317 std::string name0(_name.substr(0,pos)),name1(_name.substr(pos+strlen(MEDFileFieldRepresentationLeavesArrays::ZE_SEP)));
318 std::istringstream iss(name1);
320 std::ostringstream oss; oss << START << name; _ze_key_name=oss.str(); _name=name0;
323 bool ExtractGroupFam::isSameAs(const ExtractGroupFam& other) const
325 bool ret(ExtractGroupStatus::isSameAs(other));
327 return _id==other._id;
332 void ExtractGroupFam::printMySelf(std::ostream& os) const
334 os << " -" << _ze_key_name << " famName : \"" << _name << "\" id : " << _id << " (";
339 os << ")" << std::endl;
342 void ExtractGroupFam::fillIdsToKeep(std::set<int>& s) const
347 std::set<int> vtkExtractGroup::vtkExtractGroupInternal::getIdsToKeep() const
349 std::map<std::string,int> m(this->computeFamStrIdMap());
351 for(std::vector<ExtractGroupGrp>::const_iterator it0=_groups.begin();it0!=_groups.end();it0++)
353 if((*it0).getStatus())
355 const std::vector<std::string>& fams((*it0).getFamiliesLyingOn());
356 for(std::vector<std::string>::const_iterator it1=fams.begin();it1!=fams.end();it1++)
358 std::map<std::string,int>::iterator it2(m.find((*it1)));
360 s.insert((*it2).second);
364 for(std::vector<ExtractGroupFam>::const_iterator it0=_fams.begin();it0!=_fams.end();it0++)
365 if((*it0).getStatus())
366 (*it0).fillIdsToKeep(s);
370 std::map<std::string,int> vtkExtractGroup::vtkExtractGroupInternal::computeFamStrIdMap() const
372 std::map<std::string,int> ret;
373 for(std::vector<ExtractGroupFam>::const_iterator it0=_fams.begin();it0!=_fams.end();it0++)
374 ret[(*it0).getName()]=(*it0).getId();
380 vtkExtractGroup::vtkExtractGroup():SIL(NULL),Internal(new vtkExtractGroupInternal),InsideOut(0)
384 vtkExtractGroup::~vtkExtractGroup()
386 delete this->Internal;
389 void vtkExtractGroup::SetInsideOut(int val)
391 if(this->InsideOut!=val)
398 int vtkExtractGroup::RequestInformation(vtkInformation *request, vtkInformationVector **inputVector, vtkInformationVector *outputVector)
402 //std::cerr << "########################################## vtkExtractGroup::RequestInformation ##########################################" << std::endl;
403 vtkInformation *outInfo(outputVector->GetInformationObject(0));
404 vtkInformation *inputInfo(inputVector[0]->GetInformationObject(0));//unfortunately inputInfo->Has(vtkDataObject::SIL) returns false... use executive to find it !
406 vtkExecutive *exe(GetExecutive());
407 vtkAlgorithm *alg(this);
408 vtkInformation *infoOnSIL(alg->GetOutputInformation(0));
409 while(!infoOnSIL->Has(vtkDataObject::SIL()))// skipping vtkPVPostFilter
411 if(exe->GetNumberOfInputConnections(0)<1)
413 vtkErrorMacro("No SIL Data available ! The source of this filter must be MEDReader !");
416 vtkExecutive *exe2(exe->GetInputExecutive(0,0));
418 alg=exe2->GetAlgorithm(); exe=exe2; infoOnSIL=alg->GetOutputInformation(0);
421 this->SetSIL(vtkMutableDirectedGraph::SafeDownCast(infoOnSIL->Get(vtkDataObject::SIL())));
422 this->Internal->loadFrom(this->SIL);
423 //this->Internal->printMySelf(std::cerr);
424 outInfo->Set(vtkDataObject::SIL(),this->SIL);
426 catch(INTERP_KERNEL::Exception& e)
428 std::cerr << "Exception has been thrown in vtkExtractGroup::RequestInformation : " << e.what() << std::endl;
434 template<class CellPointExtractor>
435 vtkDataSet *FilterFamilies(vtkDataSet *input, const std::set<int>& idsToKeep, bool insideOut, const char *arrNameOfFamilyField,
436 const char *associationForThreshold, bool& catchAll, bool& catchSmth)
438 static const int VTK_DATA_ARRAY_DELETE=vtkDataArrayTemplate<double>::VTK_DATA_ARRAY_DELETE;
439 static const char ZE_SELECTION_ARR_NAME[]="@@ZeSelection@@";
440 vtkDataSet *output(input->NewInstance());
441 output->ShallowCopy(input);
442 vtkSmartPointer<vtkThreshold> thres(vtkSmartPointer<vtkThreshold>::New());
443 thres->SetInputData(output);
444 vtkDataSetAttributes *dscIn(input->GetCellData()),*dscIn2(input->GetPointData());
445 vtkDataSetAttributes *dscOut(output->GetCellData()),*dscOut2(output->GetPointData());
447 double vMin(insideOut==0?1.:0.),vMax(insideOut==0?2.:1.);
448 thres->ThresholdBetween(vMin,vMax);
451 CellPointExtractor cpe2(input);
452 vtkDataArray *da(cpe2.Get()->GetScalars(arrNameOfFamilyField));
455 std::string daName(da->GetName());
456 vtkIntArray *dai(vtkIntArray::SafeDownCast(da));
457 if(daName!=arrNameOfFamilyField || !dai)
460 int nbOfTuples(dai->GetNumberOfTuples());
461 vtkCharArray *zeSelection(vtkCharArray::New());
462 zeSelection->SetName(ZE_SELECTION_ARR_NAME);
463 zeSelection->SetNumberOfComponents(1);
464 char *pt(new char[nbOfTuples]);
465 zeSelection->SetArray(pt,nbOfTuples,0,VTK_DATA_ARRAY_DELETE);
466 const int *inPtr(dai->GetPointer(0));
467 std::fill(pt,pt+nbOfTuples,0);
468 catchAll=true; catchSmth=false;
469 std::vector<bool> pt2(nbOfTuples,false);
470 for(std::set<int>::const_iterator it=idsToKeep.begin();it!=idsToKeep.end();it++)
472 bool catchFid(false);
473 for(int i=0;i<nbOfTuples;i++)
475 { pt2[i]=true; catchFid=true; }
481 for(int ii=0;ii<nbOfTuples;ii++)
484 CellPointExtractor cpe3(output);
485 int idx(cpe3.Get()->AddArray(zeSelection));
486 cpe3.Get()->SetActiveAttribute(idx,vtkDataSetAttributes::SCALARS);
487 cpe3.Get()->CopyScalarsOff();
488 zeSelection->Delete();
490 thres->SetInputArrayToProcess(idx,0,0,associationForThreshold,ZE_SELECTION_ARR_NAME);
492 vtkUnstructuredGrid *zeComputedOutput(thres->GetOutput());
493 CellPointExtractor cpe(zeComputedOutput);
494 cpe.Get()->RemoveArray(idx);
496 zeComputedOutput->Register(0);
497 return zeComputedOutput;
503 CellExtractor(vtkDataSet *ds):_ds(ds) { }
504 vtkDataSetAttributes *Get() { return _ds->GetCellData(); }
512 PointExtractor(vtkDataSet *ds):_ds(ds) { }
513 vtkDataSetAttributes *Get() { return _ds->GetPointData(); }
518 int vtkExtractGroup::RequestData(vtkInformation *request, vtkInformationVector **inputVector, vtkInformationVector *outputVector)
522 //std::cerr << "########################################## vtkExtractGroup::RequestData ##########################################" << std::endl;
523 vtkInformation* inputInfo=inputVector[0]->GetInformationObject(0);
524 vtkDataSet *input(vtkDataSet::SafeDownCast(inputInfo->Get(vtkDataObject::DATA_OBJECT())));
525 vtkInformation *info(input->GetInformation());
526 vtkInformation *outInfo(outputVector->GetInformationObject(0));
527 vtkDataSet *output(vtkDataSet::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT())));
528 std::set<int> idsToKeep(this->Internal->getIdsToKeep());
529 // first shrink the input
530 bool catchAll,catchSmth;
531 vtkDataSet *tryOnCell(FilterFamilies<CellExtractor>(input,idsToKeep,this->InsideOut,
532 MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_CELL_NAME,"vtkDataObject::FIELD_ASSOCIATION_CELLS",catchAll,catchSmth));
537 output->ShallowCopy(tryOnCell);
538 tryOnCell->Delete();//
545 vtkDataSet *tryOnNode(FilterFamilies<PointExtractor>(tryOnCell,idsToKeep,this->InsideOut,
546 MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_NODE_NAME,"vtkDataObject::FIELD_ASSOCIATION_POINTS",catchAll,catchSmth));
547 if(tryOnNode && catchSmth)
549 output->ShallowCopy(tryOnNode);
551 tryOnNode->Delete();//
558 output->ShallowCopy(tryOnCell);
565 vtkDataSet *tryOnNode(FilterFamilies<PointExtractor>(input,idsToKeep,this->InsideOut,
566 MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_NODE_NAME,"vtkDataObject::FIELD_ASSOCIATION_POINTS",catchAll,catchSmth));
570 output->ShallowCopy(tryOnNode);
576 output->ShallowCopy(tryOnCell);
585 vtkDataSet *tryOnNode(FilterFamilies<PointExtractor>(tryOnCell,idsToKeep,this->InsideOut,
586 MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_NODE_NAME,"vtkDataObject::FIELD_ASSOCIATION_POINTS",catchAll,catchSmth));
589 output->ShallowCopy(tryOnNode);
590 tryOnNode->Delete();//
595 std::ostringstream oss; oss << "vtkExtractGroup::RequestData : The integer array with name \""<< MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_CELL_NAME;
596 oss << "\" or \"" << MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_NODE_NAME << "\" does not exist ! The extraction of group and/or family is not possible !";
597 if(this->HasObserver("ErrorEvent") )
598 this->InvokeEvent("ErrorEvent",const_cast<char *>(oss.str().c_str()));
600 vtkOutputWindowDisplayErrorText(const_cast<char *>(oss.str().c_str()));
601 vtkObject::BreakOnError();
606 catch(INTERP_KERNEL::Exception& e)
608 std::cerr << "Exception has been thrown in vtkExtractGroup::RequestData : " << e.what() << std::endl;
613 int vtkExtractGroup::GetSILUpdateStamp()
615 return this->SILTime;
618 void vtkExtractGroup::PrintSelf(ostream& os, vtkIndent indent)
620 this->Superclass::PrintSelf(os, indent);
623 int vtkExtractGroup::GetNumberOfGroupsFlagsArrays()
625 int ret(this->Internal->getNumberOfEntries());
626 //std::cerr << "vtkExtractGroup::GetNumberOfFieldsTreeArrays() -> " << ret << std::endl;
630 const char *vtkExtractGroup::GetGroupsFlagsArrayName(int index)
632 const char *ret(this->Internal->getKeyOfEntry(index));
633 //std::cerr << "vtkExtractGroup::GetFieldsTreeArrayName(" << index << ") -> " << ret << std::endl;
637 int vtkExtractGroup::GetGroupsFlagsArrayStatus(const char *name)
639 int ret((int)this->Internal->getStatusOfEntryStr(name));
640 //std::cerr << "vtkExtractGroup::GetGroupsFlagsArrayStatus(" << name << ") -> " << ret << std::endl;
644 void vtkExtractGroup::SetGroupsFlagsStatus(const char *name, int status)
646 //std::cerr << "vtkExtractGroup::SetFieldsStatus(" << name << "," << status << ")" << std::endl;
647 this->Internal->setStatusOfEntryStr(name,(bool)status);
648 if(std::string(name)==GetGroupsFlagsArrayName(GetNumberOfGroupsFlagsArrays()-1))
650 //this->Internal->printMySelf(std::cerr);