forked from Kitware/VTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vtkPointDataToCellData.cxx
125 lines (102 loc) · 3.75 KB
/
vtkPointDataToCellData.cxx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*=========================================================================
Program: Visualization Toolkit
Module: vtkPointDataToCellData.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkPointDataToCellData.h"
#include "vtkCellData.h"
#include "vtkDataSet.h"
#include "vtkIdList.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPointData.h"
vtkStandardNewMacro(vtkPointDataToCellData);
//----------------------------------------------------------------------------
// Instantiate object so that point data is not passed to output.
vtkPointDataToCellData::vtkPointDataToCellData()
{
this->PassPointData = 0;
}
//----------------------------------------------------------------------------
int vtkPointDataToCellData::RequestData(
vtkInformation*,
vtkInformationVector** inputVector,
vtkInformationVector* outputVector)
{
vtkInformation* info = outputVector->GetInformationObject(0);
vtkDataSet *output = vtkDataSet::SafeDownCast(
info->Get(vtkDataObject::DATA_OBJECT()));
vtkInformation* inInfo = inputVector[0]->GetInformationObject(0);
vtkDataSet *input = vtkDataSet::SafeDownCast(
inInfo->Get(vtkDataObject::DATA_OBJECT()));
vtkIdType cellId, ptId;
vtkIdType numCells, numPts;
vtkPointData *inPD=input->GetPointData();
vtkCellData *outCD=output->GetCellData();
int maxCellSize=input->GetMaxCellSize();
vtkIdList *cellPts;
double weight;
double *weights;
vtkDebugMacro(<<"Mapping point data to cell data");
// First, copy the input to the output as a starting point
output->CopyStructure( input );
if ( (numCells=input->GetNumberOfCells()) < 1 )
{
vtkDebugMacro(<<"No input cells!");
return 1;
}
weights=new double[maxCellSize];
cellPts = vtkIdList::New();
cellPts->Allocate(maxCellSize);
// Pass the cell data first. The fields and attributes
// which also exist in the point data of the input will
// be over-written during CopyAllocate
output->GetCellData()->CopyGlobalIdsOff();
output->GetCellData()->PassData(input->GetCellData());
output->GetCellData()->CopyFieldOff("vtkGhostLevels");
// notice that inPD and outCD are vtkPointData and vtkCellData; respectively.
// It's weird, but it works.
outCD->InterpolateAllocate(inPD,numCells);
int abort=0;
vtkIdType progressInterval=numCells/20 + 1;
for (cellId=0; cellId < numCells && !abort; cellId++)
{
if ( !(cellId % progressInterval) )
{
this->UpdateProgress((double)cellId/numCells);
abort = GetAbortExecute();
}
input->GetCellPoints(cellId, cellPts);
numPts = cellPts->GetNumberOfIds();
if ( numPts > 0 )
{
weight = 1.0 / numPts;
for (ptId=0; ptId < numPts; ptId++)
{
weights[ptId] = weight;
}
outCD->InterpolatePoint(inPD, cellId, cellPts, weights);
}
}
if ( !this->PassPointData )
{
output->GetPointData()->CopyAllOff();
output->GetPointData()->CopyFieldOn("vtkGhostLevels");
}
output->GetPointData()->PassData(input->GetPointData());
cellPts->Delete();
delete [] weights;
return 1;
}
//----------------------------------------------------------------------------
void vtkPointDataToCellData::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os,indent);
os << indent << "Pass Point Data: " << (this->PassPointData ? "On\n" : "Off\n");
}