Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-09-19 02:25:42 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

adding VTK file exporter and test cases.

Browse Source
This commit is contained in:
Jeremy Meredith 2015-10-22 17:18:33 -04:00
parent 876ef118cc
commit a78e735b3c
7 changed files with 555 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vtkm/CMakeLists.txt
View File

@ -69,3 +69,6 @@ add_subdirectory(worklet)
#add the benchmarking folder
add_subdirectory(benchmarking)
#-----------------------------------------------------------------------------
#add the io folder
add_subdirectory(io)

5
vtkm/cont/CellSetStructured.h
View File

@ -65,6 +65,11 @@ public:
this->Structure.SetPointDimensions(dimensions);
}
SchedulingRangeType GetPointDimensions()
{
return this->Structure.GetPointDimensions();
}
VTKM_CONT_EXPORT
vtkm::IdComponent
GetNumberOfPointsInCell(vtkm::Id vtkmNotUsed(cellIndex)=0) const

23
vtkm/io/CMakeLists.txt Normal file
View File

@ -0,0 +1,23 @@
##============================================================================
## Copyright (c) Kitware, Inc.
## All rights reserved.
## See LICENSE.txt 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 2014 Sandia Corporation.
## Copyright 2014 UT-Battelle, LLC.
## Copyright 2014 Los Alamos National Security.
##
## Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
## the U.S. Government retains certain rights in this software.
##
## Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
## Laboratory (LANL), the U.S. Government retains certain rights in
## this software.
##============================================================================
add_subdirectory(exporters)

28
vtkm/io/exporters/CMakeLists.txt Normal file
View File

@ -0,0 +1,28 @@
##============================================================================
## Copyright (c) Kitware, Inc.
## All rights reserved.
## See LICENSE.txt 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 2014 Sandia Corporation.
## Copyright 2014 UT-Battelle, LLC.
## Copyright 2014 Los Alamos National Security.
##
## Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
## the U.S. Government retains certain rights in this software.
##
## Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
## Laboratory (LANL), the U.S. Government retains certain rights in
## this software.
##============================================================================
set(headers
VTKDataSetExporter.h
)
vtkm_declare_headers(${headers})
#-----------------------------------------------------------------------------
add_subdirectory(testing)

377
vtkm/io/exporters/VTKDataSetExporter.h Normal file
View File

@ -0,0 +1,377 @@
//============================================================================
// Copyright (c) Kitware, Inc.
// All rights reserved.
// See LICENSE.txt 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 2014 Sandia Corporation.
// Copyright 2014 UT-Battelle, LLC.
// Copyright 2014 Los Alamos National Security.
//
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
//
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
// Laboratory (LANL), the U.S. Government retains certain rights in
// this software.
//============================================================================
#ifndef vtk_m_DataSetExporter_h
#define vtk_m_DataSetExporter_h
#include <algorithm>
#include <cstring>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
#include <vtkm/CellShape.h>
#include <vtkm/cont/CellSetExplicit.h>
#include <vtkm/cont/CellSetStructured.h>
#include <vtkm/cont/CellSetSingleType.h>
#include <vtkm/cont/DataSet.h>
#include <vtkm/cont/Field.h>
namespace {
#define VTK_EMPTY_CELL 0
#define VTK_VERTEX 1
#define VTK_POLY_VERTEX 2
#define VTK_LINE 3
#define VTK_POLY_LINE 4
#define VTK_TRIANGLE 5
#define VTK_TRIANGLE_STRIP 6
#define VTK_POLYGON 7
#define VTK_PIXEL 8
#define VTK_QUAD 9
#define VTK_TETRA 10
#define VTK_VOXEL 11
#define VTK_HEXAHEDRON 12
#define VTK_WEDGE 13
#define VTK_PYRAMID 14
#define VTK_PENTAGONAL_PRISM 15
#define VTK_HEXAGONAL_PRISM 16
int CellShapeToVTK(vtkm::Id type)
{
switch(type)
{
case vtkm::CELL_SHAPE_VERTEX: return 1;
case vtkm::CELL_SHAPE_LINE: return 3;
case vtkm::CELL_SHAPE_TRIANGLE: return 5;
case vtkm::CELL_SHAPE_QUAD: return 9;
//case vtkm::CELL_SHAPE_PIXEL: return 8;
case vtkm::CELL_SHAPE_TETRA: return 10;
case vtkm::CELL_SHAPE_PYRAMID: return 14;
case vtkm::CELL_SHAPE_WEDGE: return 13;
case vtkm::CELL_SHAPE_HEXAHEDRON: return 12;
//case vtkm::CELL_SHAPE_VOXEL: return 11;
case vtkm::CELL_SHAPE_POLYGON: return 7;
}
return 0;
}
struct OutputPointsFunctor
{
private:
std::ostream &out;
template <typename PortalType>
VTKM_CONT_EXPORT
void Output(const PortalType &portal) const
{
for (vtkm::Id index = 0; index < portal.GetNumberOfValues(); index++)
{
const int VTKDims = 3; // VTK files always require 3 dims for points
typedef typename PortalType::ValueType ValueType;
typedef typename vtkm::VecTraits<ValueType> VecType;
const ValueType &value = portal.Get(index);
vtkm::IdComponent numComponents = VecType::GetNumberOfComponents(value);
for (vtkm::IdComponent c = 0; c < numComponents && c < VTKDims; c++)
{
out << (c==0 ? "" : " ") << VecType::GetComponent(value, c);
}
for (vtkm::IdComponent c = numComponents; c < VTKDims; c++)
{
out << " 0";
}
out << std::endl;
}
}
public:
VTKM_CONT_EXPORT
OutputPointsFunctor(std::ostream &o) : out(o)
{
}
template <typename T, typename Storage>
VTKM_CONT_EXPORT
void operator()(const vtkm::cont::ArrayHandle<T,Storage > &array) const
{
this->Output(array.GetPortalConstControl());
}
};
struct OutputFieldFunctor
{
private:
std::ostream &out;
template <typename PortalType>
VTKM_CONT_EXPORT
void Output(const PortalType &portal) const
{
for (vtkm::Id index = 0; index < portal.GetNumberOfValues(); index++)
{
typedef typename PortalType::ValueType ValueType;
typedef typename vtkm::VecTraits<ValueType> VecType;
const ValueType &value = portal.Get(index);
vtkm::IdComponent numComponents = VecType::GetNumberOfComponents(value);
for (vtkm::IdComponent c = 0; c < numComponents; c++)
{
out << (c==0 ? "" : " ") << VecType::GetComponent(value, c);
}
out << std::endl;
}
}
public:
VTKM_CONT_EXPORT
OutputFieldFunctor(std::ostream &o) : out(o)
{
}
template <typename T, typename Storage>
VTKM_CONT_EXPORT
void operator()(const vtkm::cont::ArrayHandle<T,Storage > &array) const
{
this->Output(array.GetPortalConstControl());
}
};
}
namespace vtkm
{
namespace io
{
namespace exporters
{
struct VTKDataSetExporter
{
private:
static void ExportPoints(std::ostream &out,
vtkm::cont::DataSet ds)
{
///\todo: support other coordinate systems
int cindex = 0;
vtkm::cont::CoordinateSystem coords = ds.GetCoordinateSystem(cindex);
vtkm::cont::DynamicArrayHandleCoordinateSystem cdata = coords.GetData();
vtkm::Id npoints = cdata.GetNumberOfValues();
out << "POINTS " << npoints << " float" << std::endl;
cdata.CastAndCall(OutputPointsFunctor(out));
}
template <class CellSetType>
static void ExportExplicitCells(std::ostream &out,
CellSetType cs)
{
vtkm::Id nCells = cs.GetNumberOfCells();
vtkm::Id conn_length = 0;
for (vtkm::Id i=0; i<nCells; ++i)
conn_length += 1 + cs.GetNumberOfPointsInCell(i);
out << "CELLS " << nCells << " " << conn_length << std::endl;
vtkm::Vec<vtkm::Id,8> ids;
for (vtkm::Id i=0; i<nCells; ++i)
{
vtkm::Id nids = cs.GetNumberOfPointsInCell(i);
cs.GetIndices(i, ids);
out << nids;
for (int j=0; j<nids; ++j)
out << " " << ids[j];
out << std::endl;
}
out << "CELL_TYPES " << nCells << std::endl;
for (vtkm::Id i=0; i<nCells; ++i)
{
vtkm::Id shape = cs.GetCellShape(i);
out << CellShapeToVTK(shape) << std::endl;
}
}
static void ExportVertexCells(std::ostream &out,
vtkm::cont::DataSet ds)
{
vtkm::Id n = ds.GetCoordinateSystem(0).GetData().GetNumberOfValues();
out << "CELLS " << n << " " << n*2 << std::endl;
for (int i = 0; i < n; i++)
{
out << "1 " << i << std::endl;
}
out << "CELL_TYPES " << n << std::endl;
for (int i = 0; i < n; i++)
{
out << CellShapeToVTK(CELL_SHAPE_VERTEX) << std::endl;
}
}
static void ExportPointFields(std::ostream &out,
vtkm::cont::DataSet ds)
{
bool wrote_header = false;
for (vtkm::Id f = 0; f < ds.GetNumberOfFields(); f++)
{
const vtkm::cont::Field field = ds.GetField(f);
if (field.GetAssociation() != vtkm::cont::Field::ASSOC_POINTS)
continue;
vtkm::Id npoints = field.GetData().GetNumberOfValues();
int ncomps = field.GetData().GetNumberOfComponents();
if (ncomps > 4)
continue;
if (!wrote_header)
out << "POINT_DATA " << npoints << std::endl;
wrote_header = true;
out << "SCALARS " << field.GetName() << " float "<< ncomps << std::endl;
out << "LOOKUP_TABLE default" << std::endl;
field.GetData().CastAndCall(OutputFieldFunctor(out));
}
}
static void ExportCellFields(std::ostream &out,
vtkm::cont::DataSet ds,
vtkm::cont::DynamicCellSet cs)
{
bool wrote_header = false;
for (vtkm::Id f = 0; f < ds.GetNumberOfFields(); f++)
{
const vtkm::cont::Field field = ds.GetField(f);
if (field.GetAssociation() != vtkm::cont::Field::ASSOC_CELL_SET)
continue;
if (field.GetAssocCellSet() != cs.GetCellSet().GetName())
continue;
vtkm::Id ncells = field.GetData().GetNumberOfValues();
int ncomps = field.GetData().GetNumberOfComponents();
if (ncomps > 4)
continue;
if (!wrote_header)
out << "CELL_DATA " << ncells << std::endl;
wrote_header = true;
out << "SCALARS " << field.GetName() << " float "<< ncomps << std::endl;
out << "LOOKUP_TABLE default" << std::endl;
field.GetData().CastAndCall(OutputFieldFunctor(out));
}
}
static void ExportDataSetAsPoints(std::ostream &out,
vtkm::cont::DataSet ds)
{
out << "DATASET UNSTRUCTURED_GRID" << std::endl;
ExportPoints(out, ds);
ExportVertexCells(out, ds);
}
template <class CellSetType>
static void ExportDataSetAsUnstructured(std::ostream &out,
vtkm::cont::DataSet ds,
CellSetType cs)
{
out << "DATASET UNSTRUCTURED_GRID" << std::endl;
ExportPoints(out, ds);
ExportExplicitCells(out, cs);
}
template <vtkm::IdComponent DIM>
static void ExportDataSetAsStructured(std::ostream &out,
vtkm::cont::DataSet ds,
vtkm::cont::CellSetStructured<DIM> cs)
{
///\todo: support uniform/rectilinear
out << "DATASET STRUCTURED_GRID" << std::endl;
out << "DIMENSIONS ";
out << cs.GetPointDimensions()[0] << " ";
out << (DIM>1 ? cs.GetPointDimensions()[1] : 1) << " ";
out << (DIM>2 ? cs.GetPointDimensions()[2] : 1) << std::endl;
ExportPoints(out, ds);
}
public:
static void Export(std::ostream &out, vtkm::cont::DataSet ds, int csindex=0)
{
VTKM_ASSERT_CONT(csindex < ds.GetNumberOfCellSets());
out << "# vtk DataFile Version 3.0" << std::endl;
out << "vtk output" << std::endl;
out << "ASCII" << std::endl;
if (csindex < 0)
{
ExportDataSetAsPoints(out, ds);
ExportPointFields(out, ds);
}
else
{
vtkm::cont::DynamicCellSet cs = ds.GetCellSet(csindex);
if (cs.IsType<vtkm::cont::CellSetExplicit<> >())
{
ExportDataSetAsUnstructured(out, ds,
cs.CastTo<vtkm::cont::CellSetExplicit<> >());
}
else if (cs.IsType<vtkm::cont::CellSetStructured<2> >())
{
ExportDataSetAsStructured(out, ds,
cs.CastTo<vtkm::cont::CellSetStructured<2> >());
}
else if (cs.IsType<vtkm::cont::CellSetStructured<3> >())
{
ExportDataSetAsStructured(out, ds,
cs.CastTo<vtkm::cont::CellSetStructured<3> >());
}
else if (cs.IsType<vtkm::cont::CellSetSingleType<> >())
{
// these function just like explicit cell sets
ExportDataSetAsUnstructured(out, ds,
cs.CastTo<vtkm::cont::CellSetSingleType<> >());
}
else
{
VTKM_ASSERT_CONT(false);
}
ExportPointFields(out, ds);
ExportCellFields(out, ds, cs);
}
}
}; //struct VTKDataSetExporter
}}} //namespace vtkm::io::importers
#endif //vtk_m_DataSetExporter_h

25
vtkm/io/exporters/testing/CMakeLists.txt Normal file
View File

@ -0,0 +1,25 @@
#============================================================================
## Copyright (c) Kitware, Inc.
## All rights reserved.
## See LICENSE.txt 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 2014 Sandia Corporation.
## Copyright 2014 UT-Battelle, LLC.
## Copyright 2014 Los Alamos National Security.
##
## Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
## the U.S. Government retains certain rights in this software.
##
## Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
## Laboratory (LANL), the U.S. Government retains certain rights in
## this software.
##============================================================================
set(unit_tests
UnitTestVTKDataSetExporter.cxx
)
vtkm_unit_tests(SOURCES ${unit_tests})

94
vtkm/io/exporters/testing/UnitTestVTKDataSetExporter.cxx Normal file
View File

@ -0,0 +1,94 @@
//============================================================================
// Copyright (c) Kitware, Inc.
// All rights reserved.
// See LICENSE.txt 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 2014 Sandia Corporation.
// Copyright 2014 UT-Battelle, LLC.
// Copyright 2014 Los Alamos National Security.
//
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
//
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
// Laboratory (LANL), the U.S. Government retains certain rights in
// this software.
//============================================================================
#include <algorithm>
#include <fstream>
#include <iostream>
#include <iterator>
#include <vector>
#include <vtkm/cont/ArrayPortalToIterators.h>
#include <vtkm/cont/DeviceAdapter.h>
#include <vtkm/cont/testing/Testing.h>
#include <vtkm/cont/testing/MakeTestDataSet.h>
#include <vtkm/io/exporters/VTKDataSetExporter.h>
namespace {
void TestVTKExplicitExport()
{
vtkm::cont::testing::MakeTestDataSet tds;
std::ofstream out1("fileA1.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out1,
tds.Make3DExplicitDataSet0());
out1.close();
// force it to output an explicit grid as points
std::ofstream out2("fileA2.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out2,
tds.Make3DExplicitDataSet0(), -1);
out2.close();
std::ofstream out3("fileA3.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out3,
tds.Make3DExplicitDataSet1());
out3.close();
std::ofstream out4("fileA4.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out4,
tds.Make3DExplicitDataSetCowNose());
out4.close();
}
void TestVTKRegularExport()
{
vtkm::cont::testing::MakeTestDataSet tds;
std::ofstream out1("fileB1.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out1,
tds.Make2DRegularDataSet0());
out1.close();
std::ofstream out2("fileB2.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out2,
tds.Make3DRegularDataSet0());
out2.close();
// force it to output an explicit grid as points
std::ofstream out3("fileB3.vtk");
vtkm::io::exporters::VTKDataSetExporter::Export(out3,
tds.Make3DRegularDataSet0(), -1);
out3.close();
}
void TestVTKExport()
{
TestVTKExplicitExport();
TestVTKRegularExport();
}
} //Anonymous namespace
int UnitTestVTKDataSetExporter(int, char *[])
{
return vtkm::cont::testing::Testing::Run(TestVTKExport);
}