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 126 127 128 129 130 131 132
|
/*=========================================================================
Program: Visualization Toolkit
Module: $RCSfile: vtkCellDataToPointData.cxx,v $
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 "vtkCellDataToPointData.h"
#include "vtkCellData.h"
#include "vtkDataSet.h"
#include "vtkIdList.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPointData.h"
vtkCxxRevisionMacro(vtkCellDataToPointData, "$Revision: 1.36 $");
vtkStandardNewMacro(vtkCellDataToPointData);
//----------------------------------------------------------------------------
// Instantiate object so that cell data is not passed to output.
vtkCellDataToPointData::vtkCellDataToPointData()
{
this->PassCellData = 0;
}
#define VTK_MAX_CELLS_PER_POINT 4096
//----------------------------------------------------------------------------
int vtkCellDataToPointData::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;
vtkCellData *inPD=input->GetCellData();
vtkPointData *outPD=output->GetPointData();
vtkIdList *cellIds;
double weight;
double *weights;
vtkDebugMacro(<<"Mapping cell data to point data");
// First, copy the input to the output as a starting point
output->CopyStructure( input );
cellIds = vtkIdList::New();
cellIds->Allocate(VTK_MAX_CELLS_PER_POINT);
if ( (numPts=input->GetNumberOfPoints()) < 1 )
{
vtkDebugMacro(<<"No input point data!");
cellIds->Delete();
return 1;
}
weights = new double[VTK_MAX_CELLS_PER_POINT];
// Pass the point data first. The fields and attributes
// which also exist in the cell data of the input will
// be over-written during CopyAllocate
output->GetPointData()->CopyGlobalIdsOff();
output->GetPointData()->PassData(input->GetPointData());
output->GetPointData()->CopyFieldOff("vtkGhostLevels");
// notice that inPD and outPD are vtkCellData and vtkPointData; respectively.
// It's weird, but it works.
outPD->InterpolateAllocate(inPD,numPts);
int abort=0;
vtkIdType progressInterval=numPts/20 + 1;
for (ptId=0; ptId < numPts && !abort; ptId++)
{
if ( !(ptId % progressInterval) )
{
this->UpdateProgress((double)ptId/numPts);
abort = GetAbortExecute();
}
input->GetPointCells(ptId, cellIds);
numCells = cellIds->GetNumberOfIds();
if ( numCells > 0 )
{
weight = 1.0 / numCells;
for (cellId=0; cellId < numCells; cellId++)
{
weights[cellId] = weight;
}
outPD->InterpolatePoint(inPD, ptId, cellIds, weights);
}
else
{
outPD->NullPoint(ptId);
}
}
if ( !this->PassCellData )
{
output->GetCellData()->CopyAllOff();
output->GetCellData()->CopyFieldOn("vtkGhostLevels");
}
output->GetCellData()->PassData(input->GetCellData());
cellIds->Delete();
delete [] weights;
return 1;
}
//----------------------------------------------------------------------------
void vtkCellDataToPointData::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os,indent);
os << indent << "Pass Cell Data: " << (this->PassCellData ? "On\n" : "Off\n");
}
|