a83978715f8b54ffba5e0f45aea07e55da395898
[occt.git] / src / IVtkTools / IVtkTools_SubPolyDataFilter.cxx
1 // Created on: 2011-10-27 
2 // Created by: Roman KOZLOV
3 // Copyright (c) 2011-2014 OPEN CASCADE SAS 
4 // 
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16 #include <IVtkTools_SubPolyDataFilter.hxx>
17 #include <IVtkVTK_ShapeData.hxx>
18
19 // prevent disabling some MSVC warning messages by VTK headers 
20 #ifdef _MSC_VER
21 #pragma warning(push)
22 #endif
23 #include <vtkCellData.h>
24 #include <vtkIdList.h>
25 #include <vtkIdTypeArray.h>
26 #include <vtkInformation.h>
27 #include <vtkInformationVector.h>
28 #include <vtkObjectFactory.h>
29 #ifdef _MSC_VER
30 #pragma warning(pop)
31 #endif
32
33 vtkStandardNewMacro(IVtkTools_SubPolyDataFilter)
34
35 //================================================================
36 // Function : Constructor
37 // Purpose  : 
38 //================================================================
39 IVtkTools_SubPolyDataFilter::IVtkTools_SubPolyDataFilter()
40 {
41   myIdsArrayName = IVtkVTK_ShapeData::ARRNAME_SUBSHAPE_IDS;
42   myDoFiltering = true;
43 }
44
45 //================================================================
46 // Function : Destructor
47 // Purpose  : 
48 //================================================================
49 IVtkTools_SubPolyDataFilter::~IVtkTools_SubPolyDataFilter() { }
50
51 //================================================================
52 // Function : RequestData
53 // Purpose  : Filter cells according to the given set of ids.
54 //================================================================
55 int IVtkTools_SubPolyDataFilter::RequestData (vtkInformation *vtkNotUsed(theRequest),
56                                               vtkInformationVector **theInputVector,
57                                               vtkInformationVector *theOutputVector)
58 {
59   // get the input and output
60   vtkSmartPointer<vtkInformation> anInInfo = theInputVector[0]->GetInformationObject(0);
61   vtkSmartPointer<vtkInformation> anOutInfo = theOutputVector->GetInformationObject(0);
62
63   vtkSmartPointer<vtkPolyData> anInput = vtkPolyData::SafeDownCast(
64     anInInfo->Get (vtkDataObject::DATA_OBJECT()));
65
66   vtkSmartPointer<vtkPolyData> anOutput = vtkPolyData::SafeDownCast(
67     anOutInfo->Get (vtkDataObject::DATA_OBJECT()));
68
69   anInput->Modified();
70
71   if (myDoFiltering)
72   {
73     vtkSmartPointer<vtkCellData> aCellData = anInput->GetCellData();
74     vtkIdType aSize = 0;
75     vtkSmartPointer<vtkIdTypeArray> aDataArray =
76       vtkIdTypeArray::SafeDownCast (aCellData->GetArray (myIdsArrayName));
77
78     // List of cell ids to be passed
79     vtkSmartPointer<vtkIdList> anIdList = vtkSmartPointer<vtkIdList>::New();
80     anIdList->Allocate(myIdsSet.Extent());  // Allocate the list of ids
81
82     if (aDataArray.GetPointer() != NULL)
83     {
84       aSize = aDataArray->GetNumberOfTuples();
85       anIdList->Allocate (aSize);  // Allocate the list of ids
86     }
87
88     // Prepare the list of ids from the set of ids.
89     // Iterate on input cells.
90     if (!myIdsSet.IsEmpty())
91     {
92       for (vtkIdType anI = 0; anI < aSize; anI++)
93       {
94         if (myIdsSet.Contains (aDataArray->GetValue (anI)))
95         {
96           // Add a cell id to output if it's value is in the set.
97           anIdList->InsertNextId (anI);
98         }
99       }
100     }
101
102     // Copy cells with their points according to the prepared list of cell ids.
103     anOutput->GetCellData()->AllocateArrays(anInput->GetCellData()->GetNumberOfArrays());
104     anOutput->Allocate(anInput, anIdList->GetNumberOfIds());  // Allocate output cells
105     // Pass data arrays.
106     // Create new arrays for output data 
107     vtkSmartPointer<vtkCellData> anInData = anInput->GetCellData();
108     vtkSmartPointer<vtkCellData> anOutData = anOutput->GetCellData();
109     vtkSmartPointer<vtkDataArray> anInArr, anOutArr;
110
111     for (Standard_Integer anI = 0; anI < anInData->GetNumberOfArrays(); anI++)
112     {
113       anInArr = anInData->GetArray (anI);
114       anOutArr = vtkSmartPointer<vtkDataArray>::Take(
115         vtkDataArray::CreateDataArray(anInArr->GetDataType()));
116       anOutArr->SetName(anInArr->GetName());
117       anOutArr->Allocate(anIdList->GetNumberOfIds() * anInArr->GetNumberOfComponents());
118       anOutArr->SetNumberOfTuples (anIdList->GetNumberOfIds());
119       anOutArr->SetNumberOfComponents (anInArr->GetNumberOfComponents());
120       anOutData->AddArray(anOutArr);
121     }
122
123     // Copy cells with ids from our list.
124     anOutput->CopyCells (anInput, anIdList);
125
126     // Copy filtered arrays data
127     vtkIdType anOutId, anInId;
128
129     for (Standard_Integer anI = 0; anI < anInData->GetNumberOfArrays(); anI++)
130     {
131       anInArr = anInData->GetArray (anI);
132       anOutArr = anOutData->GetArray(anI);
133       for (anOutId = 0; anOutId < anIdList->GetNumberOfIds(); anOutId++)
134       {
135         anInId = anIdList->GetId (anOutId);
136         anOutArr->SetTuple (anOutId, anInId, anInArr);
137       }
138     }
139   }
140   else
141   {
142     anOutput->CopyStructure (anInput);  // Copy points and cells
143     anOutput->CopyAttributes (anInput); // Copy data arrays (sub-shapes ids)
144   }
145
146   return 1; // Return non-zero value if success and pipeline is not failed.
147 }
148
149 //================================================================
150 // Function : SetDoFiltering
151 // Purpose  : 
152 //================================================================
153 void IVtkTools_SubPolyDataFilter::SetDoFiltering (const bool theDoFiltering)
154 {
155   myDoFiltering = theDoFiltering;
156 }
157
158 //================================================================
159 // Function : PrintSelf
160 // Purpose  : 
161 //================================================================
162 void IVtkTools_SubPolyDataFilter::PrintSelf (std::ostream& theOs, vtkIndent theIndent)
163 {
164   this->Superclass::PrintSelf (theOs,theIndent);
165   theOs << theIndent << "SubPolyData: " << "\n"; 
166   theOs << theIndent << "   Number of cells to pass: " << myIdsSet.Extent() << "\n";
167   theOs << theIndent << "   Cells ids to pass: {" ;
168   // Print the content of the set of ids.
169   IVtk_IdTypeMap::Iterator anIter(myIdsSet);
170   while (anIter.More())
171   {
172       theOs << " " << anIter.Value();
173       anIter.Next();
174       if (anIter.More())
175       {
176           theOs << "; ";
177       }
178   }
179   theOs << "}" << "\n";
180 }
181
182 //================================================================
183 // Function : Clear
184 // Purpose  : Clear ids set to be passed through this filter.
185 //================================================================
186 void IVtkTools_SubPolyDataFilter::Clear()
187 {
188   myIdsSet.Clear();
189 }
190
191 //================================================================
192 // Function : SetData
193 // Purpose  : Set ids to be passed through this filter.
194 //================================================================
195 void IVtkTools_SubPolyDataFilter::SetData (const IVtk_IdTypeMap theSet)
196 {
197   myIdsSet = theSet;
198 }
199
200 //================================================================
201 // Function : AddData
202 // Purpose  : Add ids to be passed through this filter.
203 //================================================================
204 void IVtkTools_SubPolyDataFilter::AddData (const IVtk_IdTypeMap theSet)
205 {
206   IVtk_IdTypeMap::Iterator anIt (theSet);
207   for (; anIt.More(); anIt.Next())
208   {
209     if (!myIdsSet.Contains (anIt.Value()))
210     {
211       myIdsSet.Add (anIt.Value());
212     }
213   }
214 }
215
216 //================================================================
217 // Function : SetData
218 // Purpose  : Set ids to be passed through this filter.
219 //================================================================
220 void IVtkTools_SubPolyDataFilter::SetData (const IVtk_ShapeIdList theIdList)
221 {
222   myIdsSet.Clear();
223   AddData (theIdList);
224 }
225
226 //================================================================
227 // Function : AddData
228 // Purpose  : Add ids to be passed through this filter.
229 //================================================================
230 void IVtkTools_SubPolyDataFilter::AddData (const IVtk_ShapeIdList theIdList)
231 {
232   IVtk_ShapeIdList::Iterator anIt (theIdList);
233   for (; anIt.More(); anIt.Next())
234   {
235     if (!myIdsSet.Contains (anIt.Value()))
236     {
237       myIdsSet.Add (anIt.Value());
238     }
239   }
240 }
241
242 //! Set ids to be passed through this filter.
243 //================================================================
244 // Function : SetIdsArrayName
245 // Purpose  : 
246 //================================================================
247 void IVtkTools_SubPolyDataFilter::SetIdsArrayName (const char* theArrayName)
248 {
249   myIdsArrayName = theArrayName;
250 }