1 // Copyright (C) 2010-2020 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
20 #include "vtkELNOSurfaceFilter.h"
21 #include "vtkInformation.h"
22 #include "vtkInformationVector.h"
23 #include "vtkObjectFactory.h"
24 #include "vtkPolyDataAlgorithm.h"
25 #include "vtkPolyData.h"
26 #include "vtkIdTypeArray.h"
27 #include "vtkInformationQuadratureSchemeDefinitionVectorKey.h"
28 #include "vtkQuadratureSchemeDefinition.h"
29 #include "vtkUnstructuredGrid.h"
30 #include "vtkPVGeometryFilter.h"
31 #include "vtkShrinkFilter.h"
32 #include "vtkSmartPointer.h"
33 #include "vtkPointData.h"
34 #include "vtkCellData.h"
35 #include "vtkIdList.h"
37 //vtkCxxRevisionMacro(vtkELNOSurfaceFilter, "$Revision$")
39 vtkStandardNewMacro(vtkELNOSurfaceFilter)
41 vtkELNOSurfaceFilter::vtkELNOSurfaceFilter()
45 vtkELNOSurfaceFilter::~vtkELNOSurfaceFilter()
49 int vtkELNOSurfaceFilter::RequestData(vtkInformation * /*request*/, vtkInformationVector **input, vtkInformationVector *output)
51 vtkUnstructuredGrid *usgIn=vtkUnstructuredGrid::SafeDownCast(
52 input[0]->GetInformationObject(0)->Get(vtkDataObject::DATA_OBJECT()));
54 vtkUnstructuredGrid *usgOut=vtkUnstructuredGrid::SafeDownCast(
55 output->GetInformationObject(0)->Get(vtkDataObject::DATA_OBJECT()));
57 vtkIdTypeArray* usg_offsets=vtkIdTypeArray::SafeDownCast(
58 this->GetInputArrayToProcess(0, input));
60 if(usgIn==NULL||usg_offsets==NULL||usgOut==NULL)
62 vtkDebugMacro("vtkELNOSurfaceFilter no correctly configured : offsets = " << usg_offsets);
66 // first shrink the input
67 vtkUnstructuredGrid* usgInClone=usgIn->NewInstance();
69 usgInClone->ShallowCopy(usgIn);
71 vtkSmartPointer<vtkPVGeometryFilter> geomFilter=vtkSmartPointer<
72 vtkPVGeometryFilter>::New();
73 geomFilter->SetInputData(usgInClone);
74 geomFilter->SetPassThroughCellIds(1);
75 geomFilter->SetPassThroughPointIds(1);
76 geomFilter->SetUseOutline(0);
79 vtkPolyData* surface=vtkPolyData::SafeDownCast(geomFilter->GetOutput());
80 vtkIdTypeArray* originalCellIds=vtkIdTypeArray::SafeDownCast(
81 surface->GetCellData()->GetArray("vtkOriginalCellIds"));
82 vtkIdTypeArray* originalPointIds=vtkIdTypeArray::SafeDownCast(
83 surface->GetPointData()->GetArray("vtkOriginalPointIds"));
85 if( originalCellIds == NULL )
87 vtkErrorMacro("vtkPVGeometryFilter return NULL 'vtkOriginalCellIds' array");
91 if(originalPointIds==NULL)
93 vtkErrorMacro("It appears that your dataset is not reduced using vtkPVGeometryFilter (NULL 'vtkOriginalPointIds).\n==================================================================================================\nProbably your dataset is not 3D.\nIf it is not a 3D dataset you are expected to use ELNO Mesh filter instead of ELNO Surface filter.\n==================================================================================================\n");
97 vtkSmartPointer<vtkShrinkFilter> shrink=
98 vtkSmartPointer<vtkShrinkFilter>::New();
99 shrink->SetInputConnection(geomFilter->GetOutputPort(0));
100 shrink->SetShrinkFactor(0.9999);
103 vtkUnstructuredGrid* shrinked=shrink->GetOutput();
105 usgInClone->Delete();
107 usgOut->ShallowCopy(shrinked);
109 vtkIdTypeArray* offsets=vtkIdTypeArray::SafeDownCast(
110 shrinked->GetCellData()->GetArray(usg_offsets->GetName()));
112 // now copy ELNO data. Start by verifying if it is possible to
113 // shallow copy the array.
114 vtkInformation *info=offsets->GetInformation();
115 vtkInformationQuadratureSchemeDefinitionVectorKey *key=
116 vtkQuadratureSchemeDefinition::DICTIONARY();
119 vtkDebugMacro("Dictionary is not present in array " << offsets->GetName()
120 << " " << offsets << " Aborting." );
123 int dictSize=key->Size(info);
124 vtkQuadratureSchemeDefinition **dict=
125 new vtkQuadratureSchemeDefinition *[dictSize];
126 key->GetRange(info, dict, 0, 0, dictSize);
128 vtkIdType ncell=shrinked->GetNumberOfCells();
130 vtkFieldData* fielddata=usgIn->GetFieldData();
131 vtkIdList *ids=vtkIdList::New();
132 vtkIdList *surfaceIds=vtkIdList::New();
133 vtkIdList *originalIds=vtkIdList::New();
134 for(int index=0; index<fielddata->GetNumberOfArrays(); index++)
136 vtkDataArray* data=fielddata->GetArray(index);
140 vtkInformation* info=data->GetInformation();
141 const char* arrayOffsetName=info->Get(
142 vtkQuadratureSchemeDefinition::QUADRATURE_OFFSET_ARRAY_NAME());
144 if(arrayOffsetName == NULL ||
145 strcmp(arrayOffsetName, offsets->GetName())!=0)
147 usgOut->GetFieldData()->AddArray(data);
152 vtkDataArray* newArray=data->NewInstance();
153 newArray->SetName(data->GetName());
154 usgOut->GetPointData()->AddArray(newArray);
155 newArray->SetNumberOfComponents(data->GetNumberOfComponents());
156 newArray->SetNumberOfTuples(usgOut->GetNumberOfPoints());
157 newArray->CopyComponentNames(data);
160 for(vtkIdType cellId=0; cellId<ncell; cellId++)
162 vtkIdType offset=offsets->GetValue(cellId);
164 vtkIdType originalCellId=originalCellIds->GetValue(cellId);
165 int originalCellType=usgIn->GetCellType(originalCellId);
167 shrinked->GetCellPoints(cellId, ids);
168 surface->GetCellPoints(cellId, surfaceIds);
170 for(int id=0; id<ids->GetNumberOfIds(); id++)
172 vtkIdType surfaceId=surfaceIds->GetId(id);
173 vtkIdType shrinkedId=ids->GetId(id);
174 vtkIdType originalPointId = originalPointIds->GetValue(surfaceId);
176 usgIn->GetCellPoints(originalCellId, originalIds);
177 int originalLocalId=-1;
178 for(int li=0; li<originalIds->GetNumberOfIds(); li++)
180 if(originalPointId==originalIds->GetId(li))
186 if(originalLocalId==-1)
189 vtkErrorMacro("cannot find original id");
192 const double * w=dict[originalCellType]->GetShapeFunctionWeights(
195 for(j=0; j<dict[originalCellType]->GetNumberOfNodes(); j++)
200 if(j==dict[originalCellType]->GetNumberOfNodes())
202 //vtkErrorMacro("cannot find elno weigth.");
205 newArray->SetTuple(shrinkedId, offset+j, data);
211 surfaceIds->FastDelete();
212 originalIds->FastDelete();
218 void vtkELNOSurfaceFilter::PrintSelf(ostream& os, vtkIndent indent)
220 this->Superclass::PrintSelf(os, indent);