2015-11-03 03:24:57 +00:00
|
|
|
//============================================================================
|
|
|
|
// 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 2015 Sandia Corporation.
|
|
|
|
// Copyright 2015 UT-Battelle, LLC.
|
|
|
|
// Copyright 2015 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.
|
|
|
|
//============================================================================
|
2016-01-19 22:54:05 +00:00
|
|
|
#ifndef vtk_m_cont_DataSetBuilderUniform_h
|
|
|
|
#define vtk_m_cont_DataSetBuilderUniform_h
|
2015-11-03 03:24:57 +00:00
|
|
|
|
|
|
|
#include <vtkm/cont/CoordinateSystem.h>
|
2016-04-20 21:41:14 +00:00
|
|
|
#include <vtkm/cont/DataSet.h>
|
2015-11-03 03:24:57 +00:00
|
|
|
|
|
|
|
namespace vtkm {
|
|
|
|
namespace cont {
|
|
|
|
|
2016-01-19 22:54:05 +00:00
|
|
|
class DataSetBuilderUniform
|
2015-11-03 03:24:57 +00:00
|
|
|
{
|
2016-11-07 19:12:32 +00:00
|
|
|
typedef vtkm::Vec<vtkm::FloatDefault,3> VecType;
|
2015-11-03 03:24:57 +00:00
|
|
|
public:
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-01-19 22:54:05 +00:00
|
|
|
DataSetBuilderUniform() {}
|
2015-11-03 03:24:57 +00:00
|
|
|
|
2016-09-23 13:00:40 +00:00
|
|
|
//1D uniform grid
|
2016-11-04 15:43:30 +00:00
|
|
|
template <typename T>
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-09-23 13:00:40 +00:00
|
|
|
static
|
|
|
|
vtkm::cont::DataSet
|
|
|
|
Create(const vtkm::Id &dimension,
|
2016-11-04 15:43:30 +00:00
|
|
|
const T &origin, const T &spacing,
|
2016-09-23 13:00:40 +00:00
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return DataSetBuilderUniform::CreateDataSet(1,
|
|
|
|
dimension,1,1,
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(origin),
|
|
|
|
0,0),
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(spacing),
|
|
|
|
1,1),
|
|
|
|
coordNm, cellNm);
|
2016-11-04 15:43:30 +00:00
|
|
|
}
|
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-11-04 15:43:30 +00:00
|
|
|
static
|
|
|
|
vtkm::cont::DataSet
|
|
|
|
Create(const vtkm::Id &dimension,
|
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return CreateDataSet(1, dimension, 1, 1,
|
|
|
|
VecType(0), VecType(1), coordNm, cellNm);
|
2016-11-04 15:43:30 +00:00
|
|
|
}
|
2016-09-23 13:00:40 +00:00
|
|
|
|
2016-01-19 22:54:05 +00:00
|
|
|
//2D uniform grids.
|
2016-11-04 15:43:30 +00:00
|
|
|
template <typename T>
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-01-21 17:45:33 +00:00
|
|
|
static
|
2015-11-03 03:24:57 +00:00
|
|
|
vtkm::cont::DataSet
|
2015-12-24 16:46:07 +00:00
|
|
|
Create(const vtkm::Id2 &dimensions,
|
2016-11-04 15:43:30 +00:00
|
|
|
const vtkm::Vec<T,2> &origin, const vtkm::Vec<T,2> &spacing,
|
2015-11-03 03:24:57 +00:00
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return DataSetBuilderUniform::CreateDataSet(2,
|
|
|
|
dimensions[0],dimensions[1],1,
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(origin[0]),
|
|
|
|
static_cast<vtkm::FloatDefault>(origin[1]),
|
|
|
|
0),
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(spacing[0]),
|
|
|
|
static_cast<vtkm::FloatDefault>(spacing[1]),
|
|
|
|
1),
|
|
|
|
coordNm, cellNm);
|
2015-11-03 03:24:57 +00:00
|
|
|
}
|
2016-11-04 15:43:30 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-11-04 15:43:30 +00:00
|
|
|
static
|
|
|
|
vtkm::cont::DataSet
|
|
|
|
Create(const vtkm::Id2 &dimensions,
|
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return CreateDataSet(2, dimensions[0], dimensions[1], 1,
|
|
|
|
VecType(0), VecType(1), coordNm, cellNm);
|
2016-11-04 15:43:30 +00:00
|
|
|
}
|
2015-11-03 03:24:57 +00:00
|
|
|
|
2016-01-19 22:54:05 +00:00
|
|
|
//3D uniform grids.
|
2016-11-04 15:43:30 +00:00
|
|
|
template <typename T>
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-01-21 17:45:33 +00:00
|
|
|
static
|
2015-11-03 03:24:57 +00:00
|
|
|
vtkm::cont::DataSet
|
2015-12-24 16:46:07 +00:00
|
|
|
Create(const vtkm::Id3 &dimensions,
|
2016-11-04 15:43:30 +00:00
|
|
|
const vtkm::Vec<T,3> &origin, const vtkm::Vec<T,3> &spacing,
|
2015-11-03 03:24:57 +00:00
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return DataSetBuilderUniform::CreateDataSet(3,
|
|
|
|
dimensions[0],dimensions[1],dimensions[2],
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(origin[0]),
|
|
|
|
static_cast<vtkm::FloatDefault>(origin[1]),
|
|
|
|
static_cast<vtkm::FloatDefault>(origin[2])),
|
|
|
|
VecType(static_cast<vtkm::FloatDefault>(spacing[0]),
|
|
|
|
static_cast<vtkm::FloatDefault>(spacing[1]),
|
|
|
|
static_cast<vtkm::FloatDefault>(spacing[2])),
|
|
|
|
coordNm, cellNm);
|
2015-11-03 03:24:57 +00:00
|
|
|
}
|
2016-11-04 15:43:30 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-11-04 15:43:30 +00:00
|
|
|
static
|
|
|
|
vtkm::cont::DataSet
|
|
|
|
Create(const vtkm::Id3 &dimensions,
|
|
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
|
|
{
|
2016-11-07 16:27:00 +00:00
|
|
|
return CreateDataSet(3, dimensions[0], dimensions[1], dimensions[2],
|
|
|
|
VecType(0), VecType(1), coordNm, cellNm);
|
2016-11-04 15:43:30 +00:00
|
|
|
}
|
2015-11-03 03:24:57 +00:00
|
|
|
|
|
|
|
private:
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2016-01-21 17:45:33 +00:00
|
|
|
static
|
2015-11-03 03:24:57 +00:00
|
|
|
vtkm::cont::DataSet
|
2016-11-07 16:27:00 +00:00
|
|
|
CreateDataSet(int dim, vtkm::Id nx, vtkm::Id ny, vtkm::Id nz,
|
|
|
|
const vtkm::Vec<vtkm::FloatDefault,3> &origin,
|
|
|
|
const vtkm::Vec<vtkm::FloatDefault,3> &spacing,
|
|
|
|
std::string coordNm, std::string cellNm)
|
2015-11-03 03:24:57 +00:00
|
|
|
{
|
2016-09-23 13:00:40 +00:00
|
|
|
VTKM_ASSERT((dim==1 && nx>1 && ny==1 && nz==1) ||
|
|
|
|
(dim==2 && nx>1 && ny>1 && nz==1) ||
|
|
|
|
(dim==3 && nx>1 && ny>1 && nz>1));
|
2016-04-20 21:41:14 +00:00
|
|
|
VTKM_ASSERT(spacing[0]>0 && spacing[1]>0 && spacing[2]>0);
|
2016-09-23 13:00:40 +00:00
|
|
|
|
2015-11-03 03:24:57 +00:00
|
|
|
vtkm::cont::DataSet dataSet;
|
|
|
|
vtkm::cont::ArrayHandleUniformPointCoordinates
|
|
|
|
coords(vtkm::Id3(nx, ny, nz),
|
2016-09-23 13:00:40 +00:00
|
|
|
origin, spacing);
|
2016-03-23 14:38:14 +00:00
|
|
|
vtkm::cont::CoordinateSystem cs(coordNm, coords);
|
2015-11-03 03:24:57 +00:00
|
|
|
dataSet.AddCoordinateSystem(cs);
|
2016-01-19 22:54:05 +00:00
|
|
|
|
2016-09-23 13:00:40 +00:00
|
|
|
if (dim == 1)
|
|
|
|
{
|
|
|
|
vtkm::cont::CellSetStructured<1> cellSet(cellNm);
|
|
|
|
cellSet.SetPointDimensions(nx);
|
|
|
|
dataSet.AddCellSet(cellSet);
|
|
|
|
}
|
|
|
|
else if (dim == 2)
|
2015-11-03 03:24:57 +00:00
|
|
|
{
|
|
|
|
vtkm::cont::CellSetStructured<2> cellSet(cellNm);
|
|
|
|
cellSet.SetPointDimensions(vtkm::make_Vec(nx,ny));
|
|
|
|
dataSet.AddCellSet(cellSet);
|
|
|
|
}
|
2016-09-23 13:00:40 +00:00
|
|
|
else if (dim == 3)
|
2015-11-03 03:24:57 +00:00
|
|
|
{
|
|
|
|
vtkm::cont::CellSetStructured<3> cellSet(cellNm);
|
|
|
|
cellSet.SetPointDimensions(vtkm::make_Vec(nx,ny,nz));
|
|
|
|
dataSet.AddCellSet(cellSet);
|
|
|
|
}
|
2016-09-23 13:00:40 +00:00
|
|
|
else
|
|
|
|
throw vtkm::cont::ErrorControlBadValue("Invalid cell set dimension");
|
2015-11-03 03:24:57 +00:00
|
|
|
|
|
|
|
return dataSet;
|
|
|
|
}
|
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace cont
|
|
|
|
} // namespace vtkm
|
|
|
|
|
|
|
|
|
2016-01-19 22:54:05 +00:00
|
|
|
#endif //vtk_m_cont_DataSetBuilderUniform_h
|