imstkLevelSetCH.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 "imstkCollisionHandling.h"
#include "imstkMacros.h"

#include <unordered_set>
#include <vector>

namespace imstk
{
class CollisionData;
class LevelSetDeformableObject;
class RigidObject2;

class LevelSetCH : public CollisionHandling
{
public:
    LevelSetCH();
    ~LevelSetCH() override;

    IMSTK_TYPE_NAME(LevelSetCH)

public:
    void setInputLvlSetObj(std::shared_ptr<LevelSetDeformableObject> lvlSetObj);
    void setInputRigidObj(std::shared_ptr<RigidObject2> rbdObj);

    std::shared_ptr<LevelSetDeformableObject> getLvlSetObj();
    std::shared_ptr<RigidObject2> getRigidObj();

public:
    void addPoint(int id) { m_ptIdMask.insert(id); }

    void maskAllPoints();

    void unmaskAllPoints() { m_ptIdMask.clear(); }

    double getLevelSetVelocityScaling() const { return m_velocityScaling; }
    void setLevelSetVelocityScaling(const double velocityScaling) { m_velocityScaling = velocityScaling; }

    void setUseProportionalVelocity(const bool useProportionalForce) { m_useProportionalForce = useProportionalForce; }
    bool getUseProportionalVelocity() const { return m_useProportionalForce; }

    void setKernel(const int size, const double sigma = 1.0);
    int getKernelSize() const { return m_kernelSize; }
    double getKernelSigma() const { return m_kernelSigma; }

protected:
    void handle(
        const std::vector<CollisionElement>& elementsA,
        const std::vector<CollisionElement>& elementsB) override;

protected:
    std::unordered_set<int> m_ptIdMask;
    double  m_velocityScaling      = 0.1;
    bool    m_useProportionalForce = false;
    int     m_kernelSize    = 3;
    double  m_kernelSigma   = 1.0;
    double* m_kernelWeights = nullptr;
};
} // namespace imstk