MantisBT - VTK
Diff:
/*=========================================================================
Program: Visualization Toolkit
Module: vtkDataObjectToTable.h
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.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
// .NAME vtkDataObjectToTable - extract field data as a table
//
// .SECTION Description
// This filter is used to extract either the field, cell or point data of
// any data object as a table.
#ifndef __vtkDataObjectToTable_h
#define __vtkDataObjectToTable_h
#include "vtkTableAlgorithm.h"
class VTK_INFOVIS_EXPORT vtkDataObjectToTable : public vtkTableAlgorithm
{
public:
static vtkDataObjectToTable* New();
vtkTypeMacro(vtkDataObjectToTable,vtkTableAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent);
//BTX
enum
{
FIELD_DATA = 0,
POINT_DATA = 1,
CELL_DATA = 2,
VERTEX_DATA = 3,
EDGE_DATA = 4
};
//ETX
// Description:
// The field type to copy into the output table.
// Should be one of FIELD_DATA, POINT_DATA, CELL_DATA, VERTEX_DATA, EDGE_DATA.
vtkGetMacro(FieldType, int);
vtkSetClampMacro(FieldType, int, 0, 4);
+ // Description:
+ // Turn on/off the transfere of points to the output.
+ vtkGetMacro(IsTransferPoints, bool);
+ vtkSetMacro(IsTransferPoints, bool);
+ vtkBooleanMacro(IsTransferPoints, bool);
protected:
vtkDataObjectToTable();
~vtkDataObjectToTable();
int FillInputPortInformation(int port, vtkInformation* info);
+ void AddPointsColumn(vtkTable *output, vtkDataObject *input);
int RequestData(
vtkInformation*,
vtkInformationVector**,
vtkInformationVector*);
int FieldType;
+ bool IsTransferPoints;
private:
vtkDataObjectToTable(const vtkDataObjectToTable&); // Not implemented
void operator=(const vtkDataObjectToTable&); // Not implemented
};
#endif
/*=========================================================================
Program: Visualization Toolkit
Module: vtkDataObjectToTable.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.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
#include "vtkDataObjectToTable.h"
#include "vtkCellData.h"
#include "vtkDataObject.h"
#include "vtkDataSet.h"
#include "vtkDataSetAttributes.h"
#include "vtkGraph.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPointData.h"
+ #include "vtkPointSet.h"
#include "vtkTable.h"
+ #include "vtkPoints.h"
+ #include "vtkDoubleArray.h"
vtkStandardNewMacro(vtkDataObjectToTable);
//---------------------------------------------------------------------------
vtkDataObjectToTable::vtkDataObjectToTable()
{
this->FieldType = POINT_DATA;
}
//---------------------------------------------------------------------------
vtkDataObjectToTable::~vtkDataObjectToTable()
{
}
//---------------------------------------------------------------------------
int vtkDataObjectToTable::FillInputPortInformation(
int vtkNotUsed(port), vtkInformation* info)
{
info->Remove(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE());
info->Append(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkDataSet");
info->Append(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkGraph");
info->Append(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkTable");
return 1;
}
//---------------------------------------------------------------------------
int vtkDataObjectToTable::RequestData(
vtkInformation*,
vtkInformationVector** inputVector,
vtkInformationVector* outputVector)
{
// Get input data
vtkInformation* inputInfo = inputVector[0]->GetInformationObject(0);
vtkDataObject* input = inputInfo->Get(vtkDataObject::DATA_OBJECT());
// Get output table
vtkInformation* outputInfo = outputVector->GetInformationObject(0);
vtkTable* output = vtkTable::SafeDownCast(
outputInfo->Get(vtkDataObject::DATA_OBJECT()));
// If the input is a table, just copy it into the output.
if (vtkTable::SafeDownCast(input))
{
output->ShallowCopy(input);
return 1;
}
vtkDataSetAttributes* data = vtkDataSetAttributes::New();
switch(this->FieldType)
{
case FIELD_DATA:
if(input->GetFieldData())
{
data->ShallowCopy(input->GetFieldData());
}
break;
case POINT_DATA:
if(vtkDataSet* const dataset = vtkDataSet::SafeDownCast(input))
{
if(dataset->GetPointData())
{
data->ShallowCopy(dataset->GetPointData());
}
}
break;
case CELL_DATA:
if(vtkDataSet* const dataset = vtkDataSet::SafeDownCast(input))
{
if(dataset->GetCellData())
{
data->ShallowCopy(dataset->GetCellData());
}
}
break;
case VERTEX_DATA:
if(vtkGraph* const graph = vtkGraph::SafeDownCast(input))
{
if(graph->GetVertexData())
{
data->ShallowCopy(graph->GetVertexData());
}
}
break;
case EDGE_DATA:
if(vtkGraph* const graph = vtkGraph::SafeDownCast(input))
{
if(graph->GetEdgeData())
{
data->ShallowCopy(graph->GetEdgeData());
}
}
break;
}
output->SetRowData(data);
+ if(GetIsTransferPoints())
+ AddPointsColumn(output, input);
data->Delete();
return 1;
}
+
+//---------------------------------------------------------------------------
+void vtkDataObjectToTable::AddPointsColumn(vtkTable* output, vtkDataObject* input)
+{
+ if(vtkPointSet* const pointsSet = vtkPointSet::SafeDownCast(input))
+ {
+ vtkPoints* points = pointsSet->GetPoints();
+
+ if(points->GetNumberOfPoints() > 0)
+ {
+ vtkDoubleArray* xArray = vtkDoubleArray::New();
+ vtkDoubleArray* yArray = vtkDoubleArray::New();
+ vtkDoubleArray* zArray = vtkDoubleArray::New();
+
+ xArray->SetName("X");
+ yArray->SetName("Y");
+ zArray->SetName("Z");
+
+ for(int i=0; i<points->GetNumberOfPoints(); ++i)
+ {
+ double* currentPoint = points->GetPoint(i);
+ xArray->InsertNextValue(currentPoint[0]);
+ yArray->InsertNextValue(currentPoint[1]);
+ zArray->InsertNextValue(currentPoint[2]);
+ }
+ output->AddColumn(xArray);
+ output->AddColumn(yArray);
+ output->AddColumn(zArray);
+ }
+ }
+}
//---------------------------------------------------------------------------
void vtkDataObjectToTable::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "FieldType: " << this->FieldType << endl;
}
vtkDataObjectToTable.zip (2,849) 2012-03-09 10:37https://www.vtk.org/Bug/file/9129/vtkDataObjectToTable.zip