imstkPointwiseMap.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkGeometryMap.h"
#include "imstkMacros.h"
#include "imstkMath.h"
#include "imstkTypes.h"

namespace imstk
{
template<typename T, int N> class VecDataArray;

class PointwiseMap : public GeometryMap
{
public:
    PointwiseMap();
    PointwiseMap(
        std::shared_ptr<Geometry> parent,
        std::shared_ptr<Geometry> child);
    ~PointwiseMap() override = default;

    IMSTK_TYPE_NAME(PointwiseMap)

    
    void compute() override;

    void computeMap(std::unordered_map<int, int>& tetVertToSurfVertMap);

    void setMap(const std::unordered_map<int, int>& sourceMap);
    const std::unordered_map<int, int>& getMap() const { return m_oneToOneMap; }

    void addNewUniquePoint(int sourceId, int targetId)
    {
        m_oneToOneMap[sourceId] = targetId;
        m_oneToOneMapVector.push_back({ sourceId, targetId });
    }

    int getParentVertexId(const int childVertexId) const;

    void setTolerance(const double tolerance) { m_epsilon = tolerance; }
    double getTolerance() const { return m_epsilon; }

protected:
    int findMatchingVertex(const VecDataArray<double, 3>& parentMesh, const Vec3d& p);

    void requestUpdate() override;

public:
    // A map and vector are maintained. The vector for parallel processing, the map for fast lookup
    std::unordered_map<int, int>     m_oneToOneMap;       
    std::vector<std::pair<int, int>> m_oneToOneMapVector; 

    double m_epsilon = 0.00000001;
};
} // namespace imstk