1 // Copyright (C) 2020-2022 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, or (at your option) any later version.
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 (EDF R&D)
21 #include "vtkGroupAsMultiBlock.h"
22 #include "ExtractGroupHelper.h"
23 #include "vtkMEDReader.h"
24 #include "vtkUgSelectCellIds.h"
25 #include "MEDFileFieldRepresentationTree.hxx"
26 #include "vtkLongArray.h"
27 #include "VTKMEDTraits.hxx"
29 #include <vtkCellArray.h>
30 #include <vtkCellData.h>
31 #include "vtkMutableDirectedGraph.h"
32 #include "vtkInformationDataObjectMetaDataKey.h"
33 #include <vtkMergeBlocks.h>
34 #include <vtkMultiBlockDataGroupFilter.h>
35 #include <vtkDoubleArray.h>
36 #include <vtkInformation.h>
37 #include <vtkInformationVector.h>
39 #include <vtkPVGlyphFilter.h>
40 #include <vtkPointData.h>
41 #include <vtkPolyData.h>
42 #include <vtkUnstructuredGrid.h>
43 #include <vtkMultiBlockDataSet.h>
44 #include <vtkCellCenters.h>
45 #include <vtkGlyphSource2D.h>
47 class vtkGroupAsMultiBlockInternal : public ExtractGroupInternal
51 vtkStandardNewMacro(vtkGroupAsMultiBlock)
53 vtkGroupAsMultiBlock::vtkGroupAsMultiBlock():Internal(new ExtractGroupInternal)
57 vtkGroupAsMultiBlock::~vtkGroupAsMultiBlock()
59 delete this->Internal;
62 int vtkGroupAsMultiBlock::RequestInformation(vtkInformation */*request*/, vtkInformationVector **inputVector, vtkInformationVector */*outputVector*/)
64 //vtkInformation *outInfo(outputVector->GetInformationObject(0)); // todo: unused
65 vtkInformation *inputInfo(inputVector[0]->GetInformationObject(0));
66 if(!ExtractGroupInternal::IndependantIsInformationOK(vtkMEDReader::META_DATA(),inputInfo))
68 vtkErrorMacro("No SIL Data available ! The source of this filter must be MEDReader !");
71 vtkMutableDirectedGraph *mdg(vtkMutableDirectedGraph::SafeDownCast(inputInfo->Get(vtkMEDReader::META_DATA())));
72 this->Internal->loadFrom(mdg);
76 int vtkGroupAsMultiBlock::RequestData(vtkInformation *vtkNotUsed(request), vtkInformationVector **inputVector, vtkInformationVector *outputVector)
78 vtkInformation* inputInfo=inputVector[0]->GetInformationObject(0);
79 vtkMultiBlockDataSet *inputMB(vtkMultiBlockDataSet::SafeDownCast(inputInfo->Get(vtkDataObject::DATA_OBJECT())));
80 if(inputMB->GetNumberOfBlocks()!=1)
82 vtkErrorMacro("vtkGroupAsMultiBlock::RequestData : input has not the right number of parts ! Expected 1 !");
85 vtkDataSet *input(vtkDataSet::SafeDownCast(inputMB->GetBlock(0)));
86 //vtkInformation *info(input->GetInformation()); // todo: unused
87 vtkInformation *outInfo(outputVector->GetInformationObject(0));
88 vtkMultiBlockDataSet *output(vtkMultiBlockDataSet::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT())));
91 vtkErrorMacro("vtkGroupAsMultiBlock::RequestData : something wrong !");
95 vtkUnstructuredGrid *inputc(vtkUnstructuredGrid::SafeDownCast(input));
98 vtkErrorMacro("vtkGroupAsMultiBlock::RequestData : Only implemented for vtkUnstructuredGrid !");
102 vtkDataArray *famIdsGen(inputc->GetCellData()->GetScalars(MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_CELL_NAME));
105 vtkErrorMacro(<< "vtkGroupAsMultiBlock::RequestData : Fail to locate " << MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_CELL_NAME << " array !");
108 using vtkMCIdTypeArray = MEDFileVTKTraits<mcIdType>::VtkType;
109 vtkMCIdTypeArray *famIdsArr(vtkMCIdTypeArray::SafeDownCast(famIdsGen));
112 vtkErrorMacro(<< "vtkGroupAsMultiBlock::RequestData : cell array " << MEDFileFieldRepresentationLeavesArrays::FAMILY_ID_CELL_NAME << " is exepected to be IdType !");
116 vtkIdType inputNbCell(famIdsArr->GetNumberOfTuples());
117 std::vector< std::pair<std::string,std::vector<int> > > allGroups(this->Internal->getAllGroups());
118 output->SetNumberOfBlocks(allGroups.size());
120 for(const auto& grp : allGroups)
122 std::vector<vtkIdType> ids;
123 // iterate on all families on current group grp
124 for(const auto& famId : grp.second)
125 {// for each family of group grp locate cells in input lying on that group
127 std::for_each(famIdsArr->GetPointer(0),famIdsArr->GetPointer(inputNbCell),[&pos,&ids,famId](vtkIdType curFamId) { if(curFamId == famId) { ids.push_back(pos); } pos++; });
129 std::sort(ids.begin(),ids.end());
130 vtkNew<vtkIdTypeArray> idsVTK;
131 idsVTK->SetNumberOfComponents(1); idsVTK->SetNumberOfTuples(ids.size());
132 std::copy(ids.begin(),ids.end(),idsVTK->GetPointer(0));
133 vtkNew<vtkUgSelectCellIds> cellSelector;
134 cellSelector->SetIds(idsVTK);
135 cellSelector->SetInputData(inputc);
136 cellSelector->Update();
137 output->SetBlock(blockId++,cellSelector->GetOutput());