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

#include "smtk/CoreExports.h"
#include "smtk/PublicPointerDefs.h"

#include "smtk/mesh/core/CellTraits.h"
#include "smtk/mesh/core/CellTypes.h"
#include "smtk/mesh/core/DimensionTypes.h"
#include "smtk/mesh/core/FieldTypes.h"
#include "smtk/mesh/core/Handle.h"
#include "smtk/mesh/core/TypeSet.h"

#include <array>
#include <vector>

namespace smtk
{
namespace mesh
{

//forward declare classes we use
class ContainsFunctor;
class PointConnectivity;

class SMTKCORE_EXPORT Allocator
{
public:
  virtual ~Allocator() = default;

  virtual bool allocatePoints(
    std::size_t numPointsToAlloc,
    smtk::mesh::Handle& firstVertexHandle,
    std::vector<double*>& coordinateMemory) = 0;

  virtual bool allocateCells(
    smtk::mesh::CellType cellType,
    std::size_t numCellsToAlloc,
    int numVertsPerCell,
    smtk::mesh::HandleRange& createdCellIds,
    smtk::mesh::Handle*& connectivityArray) = 0;

  template<int CellType>
  bool allocateCells(
    std::size_t numCellsToAlloc,
    smtk::mesh::HandleRange& createdCellIds,
    smtk::mesh::Handle*& connectivityArray)
  {
    typedef typename smtk::mesh::CellEnumToType<CellType>::Traits Traits;
    smtk::mesh::CellType cellType = static_cast<smtk::mesh::CellType>(CellType);

    return this->allocateCells(
      cellType, numCellsToAlloc, Traits::NUM_VERTICES, createdCellIds, connectivityArray);
  }

  virtual bool connectivityModified(
    const smtk::mesh::HandleRange& cellsToUpdate,
    int numVertsPerCell,
    const smtk::mesh::Handle* connectivityArray) = 0;
};

// BufferedCellAllocator allows for the allocation of meshes by
// 1. reserving memory for all of your points
// 2. filling your points and cell connectivities by point index.
// Point indices are assumed to be contiguous with first index at 0.
class SMTKCORE_EXPORT BufferedCellAllocator
{
public:
  BufferedCellAllocator() = default;

  virtual ~BufferedCellAllocator() = default;

  virtual bool reserveNumberOfCoordinates(std::size_t nCoordinates) = 0;
  virtual bool setCoordinate(std::size_t coord, double* xyz) = 0;

  virtual bool
  addCell(smtk::mesh::CellType ctype, long long int* pointIds, std::size_t nCoordinates = 0) = 0;
  virtual bool
  addCell(smtk::mesh::CellType ctype, long int* pointIds, std::size_t nCoordinates = 0) = 0;
  virtual bool addCell(smtk::mesh::CellType ctype, int* pointIds, std::size_t nCoordinates = 0) = 0;

  virtual bool flush() = 0;

  virtual smtk::mesh::HandleRange cells() = 0;

  bool setCoordinate(std::size_t coord, double x, double y, double z)
  {
    double xyz[3] = { x, y, z };
    return this->setCoordinate(coord, xyz);
  }
  bool setCoordinate(std::size_t coord, float* xyz)
  {
    return this->setCoordinate(coord, xyz[0], xyz[1], xyz[2]);
  }
  bool setCoordinate(std::size_t coord, float x, float y, float z)
  {
    double xyz[3] = { x, y, z };
    return this->setCoordinate(coord, xyz);
  }

  bool isValid() const { return m_validState; }

protected:
  bool m_validState{ false };
};

// IncrementalAllocator allows for the allocation of meshes by incrementally
// filling your points and cell connectivities by point index. This allocator is
// the least efficient for storage and data retrieval, but provides the most
// flexible API.
class SMTKCORE_EXPORT IncrementalAllocator
{
public:
  IncrementalAllocator() = default;

  virtual ~IncrementalAllocator() = default;

  virtual std::size_t addCoordinate(double* xyz) = 0;
  virtual bool setCoordinate(std::size_t coord, double* xyz) = 0;

  virtual bool
  addCell(smtk::mesh::CellType ctype, long long int* pointIds, std::size_t nCoordinates = 0) = 0;
  virtual bool
  addCell(smtk::mesh::CellType ctype, long int* pointIds, std::size_t nCoordinates = 0) = 0;
  virtual bool addCell(smtk::mesh::CellType ctype, int* pointIds, std::size_t nCoordinates = 0) = 0;

  virtual bool flush() = 0;

  virtual smtk::mesh::HandleRange cells() = 0;

  std::size_t addCoordinate(double x, double y, double z)
  {
    double xyz[3] = { x, y, z };
    return this->addCoordinate(xyz);
  }
  std::size_t addCoordinate(float* xyz) { return this->addCoordinate(xyz[0], xyz[1], xyz[2]); }
  std::size_t addCoordinate(float x, float y, float z)
  {
    double xyz[3] = { x, y, z };
    return this->addCoordinate(xyz);
  }

  bool setCoordinate(std::size_t coord, double x, double y, double z)
  {
    double xyz[3] = { x, y, z };
    return this->setCoordinate(coord, xyz);
  }
  bool setCoordinate(std::size_t coord, float* xyz)
  {
    return this->setCoordinate(coord, xyz[0], xyz[1], xyz[2]);
  }
  bool setCoordinate(std::size_t coord, float x, float y, float z)
  {
    double xyz[3] = { x, y, z };
    return this->setCoordinate(coord, xyz);
  }

  virtual bool isValid() const = 0;
};

class SMTKCORE_EXPORT ConnectivityStorage
{
public:
  virtual ~ConnectivityStorage() = default;

  //struct that holds the required information to compute what is the
  //current cell when we are iterating.
  struct IterationState
  {
    IterationState() = default;
    std::size_t whichConnectivityVector{ 0 };
    std::size_t ptrOffsetInVector{ 0 };
  };

  virtual void initTraversal(IterationState& state) = 0;

  virtual bool fetchNextCell(
    IterationState& state,
    smtk::mesh::CellType& cellType,
    int& numPts,
    const smtk::mesh::Handle*& points) = 0;

  virtual bool equal(ConnectivityStorage* other) const = 0;

  virtual std::size_t cellSize() const = 0;

  virtual std::size_t vertSize() const = 0;
};

class SMTKCORE_EXPORT PointLocatorImpl
{
public:
  //copy of vector on return, ugggh.
  struct Results
  {
    Results() = default;

    std::vector<std::size_t> pointIds;
    std::vector<double> sqDistances;
    std::vector<double> x_s, y_s, z_s;
    bool want_sqDistances{ false };
    bool want_Coordinates{ false };
  };

  virtual ~PointLocatorImpl() = default;

  //returns all the point ids that are inside the locator
  virtual smtk::mesh::HandleRange range() const = 0;

  virtual void
  locatePointsWithinRadius(double x, double y, double z, double radius, Results& results) = 0;
};

class SMTKCORE_EXPORT Interface
{
public:
  Interface() = default;

  virtual ~Interface() = default;

  //get back a string that contains the pretty name for the interface class.
  //Requirements: The string must be all lower-case.
  //For example, the moab interface would return "moab"
  virtual std::string name() const = 0;

  //returns if the underlying data has been modified since the mesh was loaded
  //from disk. If the mesh has no underlying file, it will always be considered
  //modified. Once the mesh is written to disk, we will reset the modified
  //flag.
  virtual bool isModified() const = 0;

  //get back a lightweight interface around allocating memory into the given
  //interface. This is generally used to create new coordinates or cells that
  //are than assigned to an existing mesh or new mesh.
  //
  //If the current interface is read-only, the AllocatorPtr that is returned
  //will be nullptr.
  //
  //Note: Merely fetching a valid allocator will mark the resource as
  //modified. This is done instead of on a per-allocation basis so that
  //modification state changes don't impact performance.
  virtual smtk::mesh::AllocatorPtr allocator() = 0;

  //get back a lightweight interface around incrementally allocating memory into
  //the given interface. This is generally used to create new coordinates or
  //cells that are than assigned to an existing mesh or new mesh.
  //
  //If the current interface is read-only, the BufferedCellAllocatorPtr that is
  //returned will be nullptr.
  //
  //Note: Merely fetching a valid allocator will mark the resource as
  //modified. This is done instead of on a per-allocation basis so that
  //modification state changes don't impact performance.
  virtual smtk::mesh::BufferedCellAllocatorPtr bufferedCellAllocator() = 0;

  //get back a lightweight interface around incrementally allocating memory into
  //the given interface. This is generally used to create new coordinates or
  //cells that are than assigned to an existing mesh or new mesh.
  //
  //If the current interface is read-only, the IncrementalAllocatorPtr that is
  //returned will be nullptr.
  //
  //Note: Merely fetching a valid allocator will mark the resource as
  //modified. This is done instead of on a per-allocation basis so that
  //modification state changes don't impact performance.
  virtual smtk::mesh::IncrementalAllocatorPtr incrementalAllocator() = 0;

  //get back an efficient storage mechanism for a range of cells point
  //connectivity. This allows for efficient iteration of cell connectivity, and
  //conversion to other formats
  virtual smtk::mesh::ConnectivityStoragePtr connectivityStorage(
    const smtk::mesh::HandleRange& cells) = 0;

  //get back an efficient point locator for a range of points
  //This allows for efficient point locator on a per interface basis.
  virtual smtk::mesh::PointLocatorImplPtr pointLocator(const smtk::mesh::HandleRange& points) = 0;
  virtual smtk::mesh::PointLocatorImplPtr pointLocator(
    std::size_t numPoints,
    const std::function<std::array<double, 3>(std::size_t)>& coordinates) = 0;

  virtual smtk::mesh::Handle getRoot() const = 0;

  virtual void registerQueries(smtk::mesh::Resource&) const = 0;

  //creates a mesh with that contains the input cells.
  //the mesh will have the root as its parent.
  //Will fail if the HandleRange is empty or doesn't contain valid
  //cell handles.
  //Note: Will mark the interface as modified when successful
  virtual bool createMesh(const smtk::mesh::HandleRange& cells, smtk::mesh::Handle& meshHandle) = 0;

  //the number of meshes that are children of this mesh.
  virtual std::size_t numMeshes(smtk::mesh::Handle handle) const = 0;

  virtual smtk::mesh::HandleRange getMeshsets(smtk::mesh::Handle handle) const = 0;

  virtual smtk::mesh::HandleRange getMeshsets(smtk::mesh::Handle handle, int dimension) const = 0;

  //find all entity sets that have this exact name tag
  virtual smtk::mesh::HandleRange getMeshsets(smtk::mesh::Handle handle, const std::string& name)
    const = 0;

  //find all entity sets that have this exact domain tag
  virtual smtk::mesh::HandleRange getMeshsets(
    smtk::mesh::Handle handle,
    const smtk::mesh::Domain& domain) const = 0;

  //find all entity sets that have this exact dirichlet tag
  virtual smtk::mesh::HandleRange getMeshsets(
    smtk::mesh::Handle handle,
    const smtk::mesh::Dirichlet& dirichlet) const = 0;

  //find all entity sets that have this exact neumann tag
  virtual smtk::mesh::HandleRange getMeshsets(
    smtk::mesh::Handle handle,
    const smtk::mesh::Neumann& neumann) const = 0;

  //get all cells held by this range
  virtual smtk::mesh::HandleRange getCells(const smtk::mesh::HandleRange& meshsets) const = 0;

  //get all cells held by this range handle of a given cell type
  virtual smtk::mesh::HandleRange getCells(
    const smtk::mesh::HandleRange& meshsets,
    smtk::mesh::CellType cellType) const = 0;

  //get all cells held by this range handle of a given cell type(s)
  virtual smtk::mesh::HandleRange getCells(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::CellTypes& cellTypes) const = 0;

  //get all cells held by this range handle of a given dimension
  virtual smtk::mesh::HandleRange getCells(
    const smtk::mesh::HandleRange& meshsets,
    smtk::mesh::DimensionType dim) const = 0;

  //get all points held by this range of handle of a given dimension. If
  //boundary_only is set to true, ignore the higher order points of the
  //cells
  virtual smtk::mesh::HandleRange getPoints(
    const smtk::mesh::HandleRange& cells,
    bool boundary_only = false) const = 0;

  //get all the coordinates for the points in this range
  //xyz needs to be allocated to 3*points.size()
  virtual bool getCoordinates(const smtk::mesh::HandleRange& points, double* xyz) const = 0;

  //get all the coordinates for the points in this range
  //xyz needs to be allocated to 3*points.size()
  //Floats are not how we store the coordinates internally, so asking for
  //the coordinates in such a manner could cause data inaccuracies to appear
  //so generally this is only used if you fully understand the input domain
  virtual bool getCoordinates(const smtk::mesh::HandleRange& points, float* xyz) const = 0;

  //set all the coordinates for the points in this range
  //xyz needs to be allocated to 3*points.size()
  virtual bool setCoordinates(const smtk::mesh::HandleRange& points, const double* xyz) = 0;

  //set all the coordinates for the points in this range
  //xyz needs to be allocated to 3*points.size()
  virtual bool setCoordinates(const smtk::mesh::HandleRange& points, const float* xyz) = 0;

  virtual std::string name(const smtk::mesh::Handle& meshset) const = 0;
  virtual bool setName(const smtk::mesh::Handle& meshset, const std::string& name) = 0;

  virtual std::vector<std::string> computeNames(const smtk::mesh::HandleRange& meshsets) const = 0;

  virtual std::vector<smtk::mesh::Domain> computeDomainValues(
    const smtk::mesh::HandleRange& meshsets) const = 0;

  virtual std::vector<smtk::mesh::Dirichlet> computeDirichletValues(
    const smtk::mesh::HandleRange& meshsets) const = 0;

  virtual std::vector<smtk::mesh::Neumann> computeNeumannValues(
    const smtk::mesh::HandleRange& meshsets) const = 0;

  virtual smtk::common::UUIDArray computeModelEntities(
    const smtk::mesh::HandleRange& meshsets) const = 0;

  virtual smtk::mesh::TypeSet computeTypes(const smtk::mesh::HandleRange& range) const = 0;

  //compute the cells that make the shell/skin of the set of meshes
  virtual bool computeShell(const smtk::mesh::HandleRange& meshes, smtk::mesh::HandleRange& shell)
    const = 0;

  //compute adjacencies of a given dimension, creating them if necessary
  virtual bool computeAdjacenciesOfDimension(
    const smtk::mesh::HandleRange& meshes,
    int dimension,
    smtk::mesh::HandleRange& adj) const = 0;

  //given a handle to a cell, return its parent handle and canonical index.
  virtual bool
  canonicalIndex(const smtk::mesh::Handle& cell, smtk::mesh::Handle& parent, int& index) const = 0;

  //merge any duplicate points used by the cells that have been passed
  //Note: Will mark the interface as modified when successful
  virtual bool mergeCoincidentContactPoints(
    const smtk::mesh::HandleRange& meshes,
    double tolerance) = 0;

  //given a handle to a cell, return its dimension-equivalent neighbors.
  virtual smtk::mesh::HandleRange neighbors(const smtk::mesh::Handle& cell) const = 0;

  // Note: Will mark the interface as modified when successful
  virtual bool setDomain(const smtk::mesh::HandleRange& meshsets, const smtk::mesh::Domain& domain)
    const = 0;

  // Note: Will mark the interface as modified when successful
  virtual bool setDirichlet(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::Dirichlet& dirichlet) const = 0;

  // Note: Will mark the interface as modified when successful
  virtual bool setNeumann(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::Neumann& neumann) const = 0;

  virtual bool setId(const smtk::mesh::Handle& meshset, const smtk::common::UUID& id) const = 0;

  virtual smtk::common::UUID getId(const smtk::mesh::Handle& meshset) const = 0;

  virtual bool findById(
    const smtk::mesh::Handle& root,
    const smtk::common::UUID& id,
    smtk::mesh::Handle& meshset) const = 0;

  // Specify for a given sets of handles what the associated model entity is.
  // This allows for a model region, face, or edge to be associated with a
  // given set of meshes.
  // Note: Only MeshSets can have associations applied. Other elements will cause
  // undefined behavior
  // Note: Will mark the interface as modified when successful
  virtual bool setAssociation(
    const smtk::common::UUID& modelUUID,
    const smtk::mesh::HandleRange& meshsets) const = 0;

  // For a given handle root, find all meshsets that have an association to
  // the given model uuid
  virtual smtk::mesh::HandleRange findAssociations(
    const smtk::mesh::Handle& root,
    const smtk::common::UUID& modelUUID) const = 0;

  // Specify the model uuid that the root of this interface is associated too
  // This represents generally is the MODEL_ENTITY that owns all associations
  // with this interface
  // Note: Will mark the interface as modified when successful
  virtual bool setRootAssociation(const smtk::common::UUID& modelUUID) const = 0;

  virtual smtk::common::UUID rootAssociation() const = 0;

  virtual bool createCellField(
    const smtk::mesh::HandleRange& meshsets,
    const std::string& name,
    std::size_t dimension,
    const smtk::mesh::FieldType& type,
    const void* data) = 0;

  virtual int getCellFieldDimension(const smtk::mesh::CellFieldTag& cfTag) const = 0;

  virtual smtk::mesh::FieldType getCellFieldType(const smtk::mesh::CellFieldTag& cfTag) const = 0;

  virtual smtk::mesh::HandleRange getMeshsets(
    smtk::mesh::Handle handle,
    const smtk::mesh::CellFieldTag& cfTag) const = 0;

  virtual bool hasCellField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::CellFieldTag& cfTag) const = 0;

  virtual bool getCellField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::CellFieldTag& cfTag,
    void* data) const = 0;

  virtual bool setCellField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::CellFieldTag& cfTag,
    const void* data) = 0;

  virtual bool getField(
    const smtk::mesh::HandleRange& cells,
    const smtk::mesh::CellFieldTag& cfTag,
    void* data) const = 0;

  virtual bool setField(
    const smtk::mesh::HandleRange& cells,
    const smtk::mesh::CellFieldTag& cfTag,
    const void* data) = 0;

  virtual std::set<smtk::mesh::CellFieldTag> computeCellFieldTags(
    const smtk::mesh::Handle& handle) const = 0;

  virtual bool deleteCellField(
    const smtk::mesh::CellFieldTag& dsTag,
    const smtk::mesh::HandleRange& meshsets) = 0;

  virtual bool createPointField(
    const smtk::mesh::HandleRange& meshsets,
    const std::string& name,
    std::size_t dimension,
    const smtk::mesh::FieldType& type,
    const void* data) = 0;

  virtual int getPointFieldDimension(const smtk::mesh::PointFieldTag& pfTag) const = 0;

  virtual smtk::mesh::FieldType getPointFieldType(const smtk::mesh::PointFieldTag& pfTag) const = 0;

  virtual smtk::mesh::HandleRange getMeshsets(
    smtk::mesh::Handle handle,
    const smtk::mesh::PointFieldTag& pfTag) const = 0;

  virtual bool hasPointField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::PointFieldTag& pfTag) const = 0;

  virtual bool getPointField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::PointFieldTag& pfTag,
    void* data) const = 0;

  virtual bool setPointField(
    const smtk::mesh::HandleRange& meshsets,
    const smtk::mesh::PointFieldTag& pfTag,
    const void* data) = 0;

  virtual bool getField(
    const smtk::mesh::HandleRange& points,
    const smtk::mesh::PointFieldTag& pfTag,
    void* data) const = 0;

  virtual bool setField(
    const smtk::mesh::HandleRange& points,
    const smtk::mesh::PointFieldTag& pfTag,
    const void* data) = 0;

  virtual std::set<smtk::mesh::PointFieldTag> computePointFieldTags(
    const smtk::mesh::Handle& handle) const = 0;

  virtual bool deletePointField(
    const smtk::mesh::PointFieldTag& dsTag,
    const smtk::mesh::HandleRange& meshsets) = 0;

  virtual smtk::mesh::HandleRange pointIntersect(
    const smtk::mesh::HandleRange& a,
    const smtk::mesh::HandleRange& b,
    smtk::mesh::PointConnectivity& bpc,
    smtk::mesh::ContainmentType containmentType) const = 0;

  virtual smtk::mesh::HandleRange pointDifference(
    const smtk::mesh::HandleRange& a,
    const smtk::mesh::HandleRange& b,
    smtk::mesh::PointConnectivity& bpc,
    smtk::mesh::ContainmentType containmentType) const = 0;

  virtual void pointForEach(const HandleRange& points, smtk::mesh::PointForEach& filter) const = 0;

  virtual void cellForEach(
    const HandleRange& cells,
    smtk::mesh::PointConnectivity& a,
    smtk::mesh::CellForEach& filter) const = 0;

  virtual void meshForEach(const HandleRange& meshes, smtk::mesh::MeshForEach& filter) const = 0;

  //The handles must be all mesh or cell elements. Mixed ranges wil
  //not be deleted and will return false. Empty ranges will be ignored
  //and return true.
  //When deleting meshes if the range contains the root handle (getRoot()) the
  //request will fail, nothing will be deleted, and we will return false.
  //Note: Will mark the interface as modified when successful
  virtual bool deleteHandles(const smtk::mesh::HandleRange& toDel) = 0;

  //Manually modify the modified state. This is only done to set the modified
  //state to be proper after serialization / deserialization.
  virtual void setModifiedState(bool state) = 0;
};
} // namespace mesh
} // namespace smtk

#endif
