c7b0ffb8a3
Added cont/testing/ExplicitTestData.h which includes several explicit datasets. These datasets come from VTK data generated in VisIt. The new unit tests build datasets in several different ways and do some basic validation. Add some new methods for DataSetFieldAdd class to improve usability.
144 lines
4.8 KiB
C++
144 lines
4.8 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 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_cont_DataSetFieldAdd_h
|
|
#define vtk_m_cont_DataSetFieldAdd_h
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/Field.h>
|
|
|
|
namespace vtkm {
|
|
namespace cont {
|
|
|
|
class DataSetFieldAdd
|
|
{
|
|
public:
|
|
VTKM_CONT_EXPORT
|
|
DataSetFieldAdd() {}
|
|
|
|
//Point centered fields.
|
|
template <typename T, typename Storage>
|
|
VTKM_CONT_EXPORT
|
|
void AddPointField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
vtkm::cont::ArrayHandle<T, Storage> &field)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_POINTS,
|
|
field));
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddPointField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const std::vector<T> &field)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_POINTS,
|
|
field));
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddPointField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const T *field, const vtkm::Id &n)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_POINTS,
|
|
field, n));
|
|
}
|
|
|
|
//Cell centered field
|
|
template <typename T, typename Storage>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
vtkm::cont::ArrayHandle<T, Storage> &field,
|
|
const std::string &cellSetName)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_CELL_SET,
|
|
cellSetName, field));
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const std::vector<T> &field,
|
|
const std::string &cellSetName)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_CELL_SET,
|
|
cellSetName, field));
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const T *field, const vtkm::Id &n,
|
|
const std::string &cellSetName)
|
|
{
|
|
dataSet.AddField(Field(fieldName, 1, vtkm::cont::Field::ASSOC_CELL_SET,
|
|
cellSetName, field, n));
|
|
}
|
|
|
|
template <typename T, typename Storage>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
vtkm::cont::ArrayHandle<T, Storage> &field,
|
|
vtkm::Id cellSetIndex = 0)
|
|
{
|
|
std::string cellSetName =
|
|
dataSet.GetCellSet(cellSetIndex).GetCellSet().GetName();
|
|
this->AddCellField(dataSet, fieldName, field, cellSetName);
|
|
}
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const std::vector<T> &field,
|
|
vtkm::Id cellSetIndex = 0)
|
|
{
|
|
std::string cellSetName =
|
|
dataSet.GetCellSet(cellSetIndex).GetCellSet().GetName();
|
|
this->AddCellField(dataSet, fieldName, field, cellSetName);
|
|
}
|
|
|
|
template<typename T>
|
|
VTKM_CONT_EXPORT
|
|
void AddCellField(vtkm::cont::DataSet &dataSet,
|
|
const std::string &fieldName,
|
|
const T *field, const vtkm::Id &n,
|
|
vtkm::Id cellSetIndex = 0)
|
|
{
|
|
std::string cellSetName =
|
|
dataSet.GetCellSet(cellSetIndex).GetCellSet().GetName();
|
|
this->AddCellField(dataSet, fieldName, field, n, cellSetName);
|
|
}
|
|
|
|
|
|
};
|
|
|
|
}
|
|
}//namespace vtkm::cont
|
|
|
|
|
|
#endif //vtk_m_cont_DataSetFieldAdd_h
|