DataSetBuilderExplicit.h

//============================================================================
//  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_DataSetBuilderExplicit_h
#define vtk_m_cont_DataSetBuilderExplicit_h

#include <vtkm/cont/DataSet.h>
#include <vtkm/cont/CoordinateSystem.h>
#include <vtkm/cont/Assert.h>

namespace vtkm {
namespace cont {

typedef vtkm::cont::DeviceAdapterAlgorithm<VTKM_DEFAULT_DEVICE_ADAPTER_TAG> DFA;

//Coordinates builder??
//Need a singlecellset handler.

class DataSetBuilderExplicit
{
public:
    VTKM_CONT_EXPORT
    DataSetBuilderExplicit() {}

    //Single cell explicits.
    //TODO

    //Zoo explicit cell
    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet
    Create(const std::vector<T> &xVals,
           const std::vector<T> &yVals,
           const std::vector<vtkm::UInt8> &shapes,
           const std::vector<vtkm::IdComponent> &numIndices,
           const std::vector<vtkm::Id> &connectivity,
           const std::string &coordsNm="coords",
           const std::string &cellNm="cells")
    {
	std::vector<T> zVals(xVals.size(),0);
	return Create(xVals,yVals,zVals,
		      shapes,numIndices,connectivity,
		      coordsNm,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,
           const std::vector<vtkm::UInt8> &shapes,
           const std::vector<vtkm::IdComponent> &numIndices,
           const std::vector<vtkm::Id> &connectivity,
           const std::string &coordsNm="coords",
           const std::string &cellNm="cells");

    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet
    Create(const vtkm::cont::ArrayHandle<T> &xVals,
	   const vtkm::cont::ArrayHandle<T> &yVals,
	   const vtkm::cont::ArrayHandle<T> &zVals,
           const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
           const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
           const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
           const std::string &coordsNm="coords",
           const std::string &cellNm="cells")
    {
	return BuildDataSet(xVals,yVals,zVals,
			    shapes,numIndices,connectivity,
			    coordsNm,cellNm);
    }


    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet
    Create(const std::vector<vtkm::Vec<T,3> > &coords,
           const std::vector<vtkm::UInt8> &shapes,
           const std::vector<vtkm::IdComponent> &numIndices,
           const std::vector<vtkm::Id> &connectivity,
           const std::string &coordsNm="coords",
           const std::string &cellNm="cells");

    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet
    Create(const vtkm::cont::ArrayHandle<vtkm::Vec<T,3> > &coords,
           const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
           const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
           const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
           const std::string &coordsNm="coords",
           const std::string &cellNm="cells")
    {
	return BuildDataSet(coords, shapes, numIndices, connectivity,
			    coordsNm, cellNm);
    }
    
private:
    template<typename T>
    vtkm::cont::DataSet
    BuildDataSet(const vtkm::cont::ArrayHandle<T> &X,
		 const vtkm::cont::ArrayHandle<T> &Y,
		 const vtkm::cont::ArrayHandle<T> &Z,
		 const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
		 const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
		 const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
		 const std::string &coordsNm,
		 const std::string &cellNm);

    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet
    BuildDataSet(const vtkm::cont::ArrayHandle<vtkm::Vec<T,3> > &coords,
		 const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
		 const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
		 const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
		 const std::string &coordsNm,
		 const std::string &cellNm);
};

template<typename T>
vtkm::cont::DataSet
DataSetBuilderExplicit::Create(const std::vector<T> &xVals,
                               const std::vector<T> &yVals,
                               const std::vector<T> &zVals,
                               const std::vector<vtkm::UInt8> &shapes,
                               const std::vector<vtkm::IdComponent> &numIndices,
                               const std::vector<vtkm::Id> &connectivity,
                               const std::string &coordsNm,
                               const std::string &cellNm)
{
    VTKM_ASSERT_CONT(xVals.size() == yVals.size() &&
                     yVals.size() == zVals.size() &&
	             xVals.size() > 0);

    vtkm::cont::ArrayHandle<T> Xc, Yc, Zc;
    DFA::Copy(vtkm::cont::make_ArrayHandle(xVals), Xc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(yVals), Yc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(zVals), Zc);

    vtkm::cont::ArrayHandle<vtkm::UInt8> Sc;
    vtkm::cont::ArrayHandle<vtkm::IdComponent> Nc;
    vtkm::cont::ArrayHandle<vtkm::Id> Cc;
    DFA::Copy(vtkm::cont::make_ArrayHandle(shapes), Sc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(numIndices), Nc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(connectivity), Cc);

    return BuildDataSet(Xc,Yc,Zc, Sc,Nc,Cc, coordsNm, cellNm);
}

template<typename T>
vtkm::cont::DataSet
DataSetBuilderExplicit::BuildDataSet(const vtkm::cont::ArrayHandle<T> &X,
	                             const vtkm::cont::ArrayHandle<T> &Y,
	                             const vtkm::cont::ArrayHandle<T> &Z,
	                             const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
	                             const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
	                             const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
	                             const std::string &coordsNm,
	                             const std::string &cellNm)
{
    VTKM_ASSERT_CONT(X.GetNumberOfValues() == Y.GetNumberOfValues() &&
	             Y.GetNumberOfValues() == Z.GetNumberOfValues() &&
	             X.GetNumberOfValues() > 0 &&
		     shapes.GetNumberOfValues() == numIndices.GetNumberOfValues());

    vtkm::cont::DataSet dataSet;

    //Temp code until ArrayHandleCompositeVector is fully working...
    typedef vtkm::Vec<vtkm::Float32,3> CoordType;
    std::size_t nPts = X.GetNumberOfValues();
    std::vector<CoordType> coords(nPts);

    for (std::size_t i = 0; i < nPts; i++)
    {
	coords[i][0] = X.GetPortalConstControl().Get(i);
	coords[i][1] = Y.GetPortalConstControl().Get(i);
	coords[i][2] = Z.GetPortalConstControl().Get(i);
    }
    dataSet.AddCoordinateSystem(
        vtkm::cont::CoordinateSystem(coordsNm, 1, coords));
    vtkm::cont::CellSetExplicit<> cellSet(nPts, cellNm, 3);
    
    cellSet.Fill(shapes, numIndices, connectivity);
    dataSet.AddCellSet(cellSet);

    return dataSet;
}

template<typename T>
vtkm::cont::DataSet
DataSetBuilderExplicit::Create(const std::vector<vtkm::Vec<T,3> > &coords,
                               const std::vector<vtkm::UInt8> &shapes,
                               const std::vector<vtkm::IdComponent> &numIndices,
                               const std::vector<vtkm::Id> &connectivity,
                               const std::string &coordsNm,
                               const std::string &cellNm)
{
    vtkm::cont::DataSet dataSet;

    vtkm::cont::ArrayHandle<Vec<T,3> > coordsArray;
    DFA::Copy(vtkm::cont::make_ArrayHandle(coords), coordsArray);

    dataSet.AddCoordinateSystem(
        vtkm::cont::CoordinateSystem(coordsNm, 1, coordsArray));

    vtkm::cont::ArrayHandle<vtkm::UInt8> Sc;
    vtkm::cont::ArrayHandle<vtkm::IdComponent> Nc;
    vtkm::cont::ArrayHandle<vtkm::Id> Cc;
    DFA::Copy(vtkm::cont::make_ArrayHandle(shapes), Sc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(numIndices), Nc);
    DFA::Copy(vtkm::cont::make_ArrayHandle(connectivity), Cc);
    
    return Create(coordsArray, Sc, Nc, Cc, coordsNm, cellNm);
}

template<typename T>
VTKM_CONT_EXPORT
vtkm::cont::DataSet
DataSetBuilderExplicit::BuildDataSet(const vtkm::cont::ArrayHandle<vtkm::Vec<T,3> > &coords,
				     const vtkm::cont::ArrayHandle<vtkm::UInt8> &shapes,
				     const vtkm::cont::ArrayHandle<vtkm::IdComponent> &numIndices,
				     const vtkm::cont::ArrayHandle<vtkm::Id> &connectivity,
				     const std::string &coordsNm,
				     const std::string &cellNm)
{
    vtkm::cont::DataSet dataSet;

    dataSet.AddCoordinateSystem(vtkm::cont::CoordinateSystem(coordsNm,
							     1, coords));
    vtkm::Id nPts = static_cast<vtkm::Id>(coords.GetNumberOfValues());
    vtkm::cont::CellSetExplicit<> cellSet(nPts, cellNm, 3);
    
    cellSet.Fill(shapes, numIndices, connectivity);
    dataSet.AddCellSet(cellSet);
    
    return dataSet;
}

class DataSetBuilderExplicitIterative
{
public:
    VTKM_CONT_EXPORT
    DataSetBuilderExplicitIterative() {}

    VTKM_CONT_EXPORT
    void Begin(const std::string &_coordNm="coords",
	       const std::string &_cellNm="cells")
    {
	this->coordNm = _coordNm;
	this->cellNm = _cellNm;
        this->points.resize(0);
        this->shapes.resize(0);
        this->numIdx.resize(0);
        this->connectivity.resize(0);
    }

    //Define points.
    VTKM_CONT_EXPORT
    vtkm::cont::DataSet Create();

    VTKM_CONT_EXPORT
    vtkm::Id AddPoint(const vtkm::Vec<vtkm::Float32, 3> &pt)
    {
	points.push_back(pt);
	vtkm::Id id = static_cast<vtkm::Id>(points.size());
	return id;
    }
    VTKM_CONT_EXPORT
    vtkm::Id AddPoint(const vtkm::Float32 &x,
		      const vtkm::Float32 &y,
		      const vtkm::Float32 &z=0)
    {
	points.push_back(vtkm::make_Vec(x,y,z));
	vtkm::Id id = static_cast<vtkm::Id>(points.size());
	return id;
    }

    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::Id AddPoint(const T &x, const T &y, const T &z=0)
    {
	return AddPoint(static_cast<vtkm::Float32>(x),
			static_cast<vtkm::Float32>(y),
			static_cast<vtkm::Float32>(z));
    }

    template<typename T>
    VTKM_CONT_EXPORT
    vtkm::Id AddPoint(const vtkm::Vec<T,3> &pt)
    {
	return AddPoint(static_cast<vtkm::Vec<vtkm::Float32,3> >(pt));
    }

    //Define cells.
    VTKM_CONT_EXPORT
    void AddCell(vtkm::UInt8 shape)
    {
	this->shapes.push_back(shape);
	this->numIdx.push_back(0);
    }
    VTKM_CONT_EXPORT
    void AddCell(const vtkm::UInt8 &shape, const std::vector<vtkm::Id> &conn)
    {
	this->shapes.push_back(shape);
	this->numIdx.push_back(static_cast<vtkm::IdComponent>(conn.size()));
	connectivity.insert(connectivity.end(), conn.begin(), conn.end());
    }
    VTKM_CONT_EXPORT
    void AddCell(const vtkm::UInt8 &shape, const vtkm::Id *conn, const vtkm::IdComponent &n)
    {
        this->shapes.push_back(shape);
        this->numIdx.push_back(n);
        for (int i = 0; i < n; i++)
            connectivity.push_back(conn[i]);
    }
    VTKM_CONT_EXPORT
    void AddCellPoint(vtkm::Id pointIndex)
    {
	VTKM_ASSERT_CONT(this->numIdx.size() > 0);
	this->connectivity.push_back(pointIndex);
	this->numIdx.back() += 1;
    }
    
private:
    std::string coordNm, cellNm;

    std::vector<vtkm::Vec<vtkm::Float32,3> > points;
    std::vector<vtkm::UInt8> shapes;
    std::vector<vtkm::IdComponent> numIdx;
    std::vector<vtkm::Id> connectivity;
};

vtkm::cont::DataSet
DataSetBuilderExplicitIterative::Create()
{
    DataSetBuilderExplicit dsb;
    return dsb.Create(points, shapes, numIdx, connectivity, coordNm, cellNm);
}



#if 0
template<typename T, typename CellType>
vtkm::cont::DataSet
DataSetBuilderExplicit::Create(const std::vector<T> &xVals,
			       const std::vector<T> &yVals,
			       const std::vector<vtkm::Id> &connectivity,
			       const std::string &coordsNm,
			       const std::string &cellNm)
{
    VTKM_CONT_ASSERT(xVals.size() == yVals.size() && xVals.size() > 0);
    vtkm::cont::DataSet dataSet;

    typedef vtkm::Vec<vtkm::Float32,3> CoordType;
    std::vector<CoordType> coords(xVals.size());
    
    for (size_t i=0; i < coords.size(); i++)
    {
        coords[i][0] = xVals[i];
        coords[i][1] = yVals[i];
        coords[i][2] = 0;
    }
    dataSet.AddCoordinateSystem(
        vtkm::cont::CoordinateSystem(coordsNm, 1, coords));

    vtkm::cont::CellSetSingleType< > cellSet(CellType(), cellNm);
    cellSet.FillViaCopy(connectivity);
    dataSet.AddCellSet(cellSet);

    return dataSet;
}

template<typename T, typename CellType>
vtkm::cont::DataSet
DataSetBuilderExplicit::Create(const std::vector<T> &xVals,
			       const std::vector<T> &yVals,
			       const std::vector<T> &zVals,
			       const std::vector<vtkm::Id> &connectivity,
			       const std::string &coordsNm,
			       const std::string &cellNm)
{
    VTKM_CONT_ASSERT(xVals.size() == yVals.size() &&
                     yVals.size() == zVals.size() &&
                     xVals.size() > 0);
    vtkm::cont::DataSet dataSet;

    typedef vtkm::Vec<vtkm::Float32,3> CoordType;
    std::vector<CoordType> coords(xVals.size());
    
    vtkm::Id nPts = static_cast<vtkm::Id>(coords.size());
    for (vtkm::Id i=0; i < nPts; i++)
    {
        coords[i][0] = xVals[i];
        coords[i][1] = yVals[i];
        coords[i][2] = zVals[i];
    }
    dataSet.AddCoordinateSystem(
        vtkm::cont::CoordinateSystem(coordsNm, 1, coords));

    vtkm::cont::CellSetSingleType< > cellSet(CellType(), cellNm);
    cellSet.FillViaCopy(connectivity);
    dataSet.AddCellSet(cellSet);

    return dataSet;
}

template<typename T, typename CellType>
vtkm::cont::DataSet
DataSetBuilderExplicit::Create(const std::vector<vtkm::Vec<T,3> > &coords,
                               const std::vector<vtkm::Id> &connectivity,
                               const std::string &coordsNm,
                               const std::string &cellNm)
{
    vtkm::cont::DataSet dataSet;

    vtkm::cont::ArrayHandle<Vec<T,3> > coordsArray;
    DFA::Copy(vtkm::cont::make_ArrayHandle(coords), coordsArray);

    dataSet.AddCoordinateSystem(
        vtkm::cont::CoordinateSystem(coordsNm, 1, coordsArray));

    vtkm::cont::CellSetSingleType< > cellSet(CellType(), cellNm);
    cellSet.FillViaCopy(connectivity);
    dataSet.AddCellSet(cellSet);

    return dataSet;
}
#endif

}
}

#endif //vtk_m_cont_DataSetBuilderExplicit_h