DataSetBuilderExplicit.h

Go to the documentation of this file.

//============================================================================
//  The contents of this file are covered by the Viskores license. See
//  LICENSE.txt for details.
//
//  By contributing to this file, all contributors agree to the Developer
//  Certificate of Origin Version 1.1 (DCO 1.1) as stated in DCO.txt.
//============================================================================
 
//============================================================================
//  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.
//============================================================================
#ifndef viskores_cont_DataSetBuilderExplicit_h
#define viskores_cont_DataSetBuilderExplicit_h
 
#include <viskores/cont/ArrayHandleCast.h>
#include <viskores/cont/ArrayHandleCompositeVector.h>
#include <viskores/cont/ArrayPortalToIterators.h>
#include <viskores/cont/ConvertNumComponentsToOffsets.h>
#include <viskores/cont/CoordinateSystem.h>
#include <viskores/cont/DataSet.h>
 
namespace viskores
{
namespace cont
{
 
//Coordinates builder??
 
class VISKORES_CONT_EXPORT DataSetBuilderExplicit
{
public:
  VISKORES_CONT
  DataSetBuilderExplicit() {}
 
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const std::vector<T>& xVals,
    const std::vector<viskores::UInt8>& shapes,
    const std::vector<viskores::IdComponent>& numIndices,
    const std::vector<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords")
  {
    std::vector<T> yVals(xVals.size(), 0), zVals(xVals.size(), 0);
    return DataSetBuilderExplicit::Create(
      xVals, yVals, zVals, shapes, numIndices, connectivity, coordsNm);
  }
 
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const std::vector<T>& xVals,
    const std::vector<T>& yVals,
    const std::vector<viskores::UInt8>& shapes,
    const std::vector<viskores::IdComponent>& numIndices,
    const std::vector<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords")
  {
    std::vector<T> zVals(xVals.size(), 0);
    return DataSetBuilderExplicit::Create(
      xVals, yVals, zVals, shapes, numIndices, connectivity, coordsNm);
  }
 
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const std::vector<T>& xVals,
    const std::vector<T>& yVals,
    const std::vector<T>& zVals,
    const std::vector<viskores::UInt8>& shapes,
    const std::vector<viskores::IdComponent>& numIndices,
    const std::vector<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords");
 
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const std::vector<viskores::Vec<T, 3>>& coords,
    const std::vector<viskores::UInt8>& shapes,
    const std::vector<viskores::IdComponent>& numIndices,
    const std::vector<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords");
 
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
    const viskores::cont::ArrayHandle<viskores::UInt8>& shapes,
    const viskores::cont::ArrayHandle<viskores::IdComponent>& numIndices,
    const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords")
  {
    auto offsets = viskores::cont::ConvertNumComponentsToOffsets(numIndices);
    return DataSetBuilderExplicit::BuildDataSet(coords, shapes, offsets, connectivity, coordsNm);
  }
 
  template <typename T, typename CellShapeTag>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const std::vector<viskores::Vec<T, 3>>& coords,
    CellShapeTag tag,
    viskores::IdComponent numberOfPointsPerCell,
    const std::vector<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords");
 
  template <typename T, typename CellShapeTag>
  VISKORES_CONT static viskores::cont::DataSet Create(
    const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
    CellShapeTag tag,
    viskores::IdComponent numberOfPointsPerCell,
    const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
    const std::string& coordsNm = "coords")
  {
    return DataSetBuilderExplicit::BuildDataSet(
      coords, tag, numberOfPointsPerCell, connectivity, coordsNm);
  }
 
private:
  template <typename T>
  VISKORES_CONT static viskores::cont::DataSet BuildDataSet(
    const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
    const viskores::cont::ArrayHandle<viskores::UInt8>& shapes,
    const viskores::cont::ArrayHandle<viskores::Id>& offsets,
    const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
    const std::string& coordsNm);
 
  template <typename T, typename CellShapeTag>
  VISKORES_CONT static viskores::cont::DataSet BuildDataSet(
    const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
    CellShapeTag tag,
    viskores::IdComponent numberOfPointsPerCell,
    const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
    const std::string& coordsNm);
};
 
template <typename T>
inline VISKORES_CONT viskores::cont::DataSet DataSetBuilderExplicit::Create(
  const std::vector<T>& xVals,
  const std::vector<T>& yVals,
  const std::vector<T>& zVals,
  const std::vector<viskores::UInt8>& shapes,
  const std::vector<viskores::IdComponent>& numIndices,
  const std::vector<viskores::Id>& connectivity,
  const std::string& coordsNm)
{
  VISKORES_ASSERT(xVals.size() == yVals.size() && yVals.size() == zVals.size() && xVals.size() > 0);
 
  viskores::cont::ArrayHandle<viskores::Vec3f> coordsArray;
  coordsArray.Allocate(static_cast<viskores::Id>(xVals.size()));
  auto coordsPortal = coordsArray.WritePortal();
  for (std::size_t index = 0; index < xVals.size(); ++index)
  {
    coordsPortal.Set(static_cast<viskores::Id>(index),
                     viskores::make_Vec(static_cast<viskores::FloatDefault>(xVals[index]),
                                        static_cast<viskores::FloatDefault>(yVals[index]),
                                        static_cast<viskores::FloatDefault>(zVals[index])));
  }
 
  auto shapesArray = viskores::cont::make_ArrayHandle(shapes, viskores::CopyFlag::On);
  auto connArray = viskores::cont::make_ArrayHandle(connectivity, viskores::CopyFlag::On);
 
  auto offsetsArray = viskores::cont::ConvertNumComponentsToOffsets(
    viskores::cont::make_ArrayHandle(numIndices, viskores::CopyFlag::Off));
 
  return DataSetBuilderExplicit::BuildDataSet(
    coordsArray, shapesArray, offsetsArray, connArray, coordsNm);
}
 
template <typename T>
inline VISKORES_CONT viskores::cont::DataSet DataSetBuilderExplicit::Create(
  const std::vector<viskores::Vec<T, 3>>& coords,
  const std::vector<viskores::UInt8>& shapes,
  const std::vector<viskores::IdComponent>& numIndices,
  const std::vector<viskores::Id>& connectivity,
  const std::string& coordsNm)
{
  auto coordsArray = viskores::cont::make_ArrayHandle(coords, viskores::CopyFlag::On);
  auto shapesArray = viskores::cont::make_ArrayHandle(shapes, viskores::CopyFlag::On);
  auto connArray = viskores::cont::make_ArrayHandle(connectivity, viskores::CopyFlag::On);
 
  auto offsetsArray = viskores::cont::ConvertNumComponentsToOffsets(
    viskores::cont::make_ArrayHandle(numIndices, viskores::CopyFlag::Off));
 
  return DataSetBuilderExplicit::BuildDataSet(
    coordsArray, shapesArray, offsetsArray, connArray, coordsNm);
}
 
template <typename T>
inline VISKORES_CONT viskores::cont::DataSet DataSetBuilderExplicit::BuildDataSet(
  const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
  const viskores::cont::ArrayHandle<viskores::UInt8>& shapes,
  const viskores::cont::ArrayHandle<viskores::Id>& offsets,
  const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
  const std::string& coordsNm)
{
  viskores::cont::DataSet dataSet;
 
  dataSet.AddCoordinateSystem(viskores::cont::CoordinateSystem(coordsNm, coords));
  viskores::Id nPts = static_cast<viskores::Id>(coords.GetNumberOfValues());
  viskores::cont::CellSetExplicit<> cellSet;
 
  cellSet.Fill(nPts, shapes, connectivity, offsets);
  dataSet.SetCellSet(cellSet);
 
  return dataSet;
}
 
template <typename T, typename CellShapeTag>
inline VISKORES_CONT viskores::cont::DataSet DataSetBuilderExplicit::Create(
  const std::vector<viskores::Vec<T, 3>>& coords,
  CellShapeTag tag,
  viskores::IdComponent numberOfPointsPerCell,
  const std::vector<viskores::Id>& connectivity,
  const std::string& coordsNm)
{
  auto coordsArray = viskores::cont::make_ArrayHandle(coords, viskores::CopyFlag::On);
  auto connArray = viskores::cont::make_ArrayHandle(connectivity, viskores::CopyFlag::On);
 
  return DataSetBuilderExplicit::Create(
    coordsArray, tag, numberOfPointsPerCell, connArray, coordsNm);
}
 
template <typename T, typename CellShapeTag>
inline VISKORES_CONT viskores::cont::DataSet DataSetBuilderExplicit::BuildDataSet(
  const viskores::cont::ArrayHandle<viskores::Vec<T, 3>>& coords,
  CellShapeTag tag,
  viskores::IdComponent numberOfPointsPerCell,
  const viskores::cont::ArrayHandle<viskores::Id>& connectivity,
  const std::string& coordsNm)
{
  (void)tag; //C4100 false positive workaround
  viskores::cont::DataSet dataSet;
 
  dataSet.AddCoordinateSystem(viskores::cont::CoordinateSystem(coordsNm, coords));
  viskores::cont::CellSetSingleType<> cellSet;
 
  cellSet.Fill(coords.GetNumberOfValues(), tag.Id, numberOfPointsPerCell, connectivity);
  dataSet.SetCellSet(cellSet);
 
  return dataSet;
}
 
class VISKORES_CONT_EXPORT DataSetBuilderExplicitIterative
{
public:
  VISKORES_CONT DataSetBuilderExplicitIterative();
 
  VISKORES_CONT void Begin(const std::string& coordName = "coords");
 
  VISKORES_CONT viskores::Id AddPoint(const viskores::Vec3f& pt);
 
  template <typename T>
  VISKORES_CONT viskores::Id AddPoint(const viskores::Vec<T, 3>& pt)
  {
    return AddPoint(static_cast<viskores::Vec3f>(pt));
  }
 
  VISKORES_CONT viskores::Id AddPoint(const viskores::FloatDefault& x,
                                      const viskores::FloatDefault& y,
                                      const viskores::FloatDefault& z = 0);
 
  template <typename T>
  VISKORES_CONT viskores::Id AddPoint(const T& x, const T& y, const T& z = 0)
  {
    return AddPoint(static_cast<viskores::FloatDefault>(x),
                    static_cast<viskores::FloatDefault>(y),
                    static_cast<viskores::FloatDefault>(z));
  }
 
  VISKORES_CONT void AddCell(const viskores::UInt8& shape, const std::vector<viskores::Id>& conn);
 
  VISKORES_CONT void AddCell(const viskores::UInt8& shape,
                             const viskores::Id* conn,
                             const viskores::IdComponent& n);
 
  VISKORES_CONT void AddCell(viskores::UInt8 shape);
 
  VISKORES_CONT void AddCellPoint(viskores::Id pointIndex);
 
  VISKORES_CONT viskores::cont::DataSet Create();
 
private:
  std::string coordNm;
 
  std::vector<viskores::Vec3f> points;
  std::vector<viskores::UInt8> shapes;
  std::vector<viskores::IdComponent> numIdx;
  std::vector<viskores::Id> connectivity;
};
}
}
 
#endif //viskores_cont_DataSetBuilderExplicit_h