168 lines
5.1 KiB
C++
168 lines
5.1 KiB
C++
//============================================================================
|
|
// 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.
|
|
//============================================================================
|
|
#ifndef vtk_m_cont_DataSetBuilderRectilinear_h
|
|
#define vtk_m_cont_DataSetBuilderRectilinear_h
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/CoordinateSystem.h>
|
|
#include <vtkm/cont/Assert.h>
|
|
#include <vtkm/cont/ArrayHandleCartesianProduct.h>
|
|
|
|
namespace vtkm {
|
|
namespace cont {
|
|
|
|
typedef vtkm::cont::DeviceAdapterAlgorithm<VTKM_DEFAULT_DEVICE_ADAPTER_TAG> DFA;
|
|
|
|
class DataSetBuilderRectilinear
|
|
{
|
|
public:
|
|
VTKM_CONT_EXPORT
|
|
DataSetBuilderRectilinear() {}
|
|
|
|
//2D grids.
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(vtkm::Id nx, vtkm::Id ny,
|
|
T *xvals, T *yvals,
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
{
|
|
T zvals = 0;
|
|
return Create(2, nx,ny, 1, xvals, yvals, &zvals, coordNm, cellNm);
|
|
}
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(const std::vector<T> &xvals, const std::vector<T> &yvals,
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
{
|
|
std::vector<T> zvals(1,0);
|
|
return Create(2, xvals, yvals, zvals, coordNm, cellNm);
|
|
}
|
|
|
|
//3D grids.
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(vtkm::Id nx, vtkm::Id ny, vtkm::Id nz,
|
|
T *xvals, T *yvals, T *zvals,
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
{
|
|
return Create(3, nx,ny,nz, xvals, yvals, zvals, coordNm, cellNm);
|
|
}
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(const std::vector<T> &xvals,
|
|
const std::vector<T> &yvals,
|
|
const std::vector<T> &zvals,
|
|
std::string coordNm="coords", std::string cellNm="cells")
|
|
{
|
|
return Create(3, xvals, yvals, zvals, coordNm, cellNm);
|
|
}
|
|
|
|
private:
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(int dim, vtkm::Id nx, vtkm::Id ny, vtkm::Id nz,
|
|
T *xvals, T *yvals, T *zvals,
|
|
std::string coordNm, std::string cellNm)
|
|
{
|
|
VTKM_ASSERT_CONT(nx>1 && ny>1 &&
|
|
((dim==2 && nz==1)||(dim==3 && nz>=1)));
|
|
|
|
vtkm::cont::ArrayHandle<T> X = vtkm::cont::make_ArrayHandle(nx,xvals);
|
|
vtkm::cont::ArrayHandle<T> Y = vtkm::cont::make_ArrayHandle(ny,yvals);
|
|
vtkm::cont::ArrayHandle<T> Z = vtkm::cont::make_ArrayHandle(nz,zvals);
|
|
|
|
return BuildDataSet(dim, X,Y,Z, coordNm, cellNm);
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
Create(int dim,
|
|
const std::vector<T> &xvals,
|
|
const std::vector<T> &yvals,
|
|
const std::vector<T> &zvals,
|
|
std::string coordNm, std::string cellNm)
|
|
{
|
|
VTKM_ASSERT_CONT(xvals.size()>1 && yvals.size()>1 &&
|
|
((dim==2 && zvals.size()==1)||(dim==3 && zvals.size()>=1)));
|
|
|
|
vtkm::cont::ArrayHandle<T> X = vtkm::cont::make_ArrayHandle(xvals);
|
|
vtkm::cont::ArrayHandle<T> Y = vtkm::cont::make_ArrayHandle(yvals);
|
|
vtkm::cont::ArrayHandle<T> Z = vtkm::cont::make_ArrayHandle(zvals);
|
|
|
|
return BuildDataSet(dim, X,Y,Z, coordNm, cellNm);
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
vtkm::cont::DataSet
|
|
BuildDataSet(int dim,
|
|
const vtkm::cont::ArrayHandle<T> &X,
|
|
const vtkm::cont::ArrayHandle<T> &Y,
|
|
const vtkm::cont::ArrayHandle<T> &Z,
|
|
std::string coordNm, std::string cellNm)
|
|
{
|
|
vtkm::cont::DataSet dataSet;
|
|
|
|
vtkm::cont::ArrayHandleCartesianProduct<
|
|
vtkm::cont::ArrayHandle<T>,
|
|
vtkm::cont::ArrayHandle<T>,
|
|
vtkm::cont::ArrayHandle<T> > coords;
|
|
|
|
vtkm::cont::ArrayHandle<T> Xc, Yc, Zc;
|
|
DFA::Copy(X, Xc);
|
|
DFA::Copy(Y, Yc);
|
|
DFA::Copy(Z, Zc);
|
|
|
|
coords = vtkm::cont::make_ArrayHandleCartesianProduct(Xc,Yc,Zc);
|
|
vtkm::cont::CoordinateSystem cs(coordNm, 1, coords);
|
|
dataSet.AddCoordinateSystem(cs);
|
|
|
|
if (dim == 2)
|
|
{
|
|
vtkm::cont::CellSetStructured<2> cellSet(cellNm);
|
|
cellSet.SetPointDimensions(vtkm::make_Vec(Xc.GetNumberOfValues(),
|
|
Yc.GetNumberOfValues()));
|
|
dataSet.AddCellSet(cellSet);
|
|
}
|
|
else
|
|
{
|
|
vtkm::cont::CellSetStructured<3> cellSet(cellNm);
|
|
cellSet.SetPointDimensions(vtkm::make_Vec(Xc.GetNumberOfValues(),
|
|
Yc.GetNumberOfValues(),
|
|
Zc.GetNumberOfValues()));
|
|
dataSet.AddCellSet(cellSet);
|
|
}
|
|
|
|
return dataSet;
|
|
|
|
}
|
|
};
|
|
|
|
} // namespace cont
|
|
} // namespace vtkm
|
|
|
|
#endif //vtk_m_cont_DataSetBuilderRectilinear_h
|