imstkBurnable.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.
*/

#include "imstkComponent.h"
#include "imstkMath.h"
#include "imstkVecDataArray.h"
#include "imstkPbdObjectCellRemoval.h"

#pragma once

namespace imstk
{
class PbdConstraint;
class PbdObject;
class PbdObjectCellRemoval;
class TaskNode;

class Burnable : public SceneBehaviour
{
public:
    Burnable(const std::string& name = "BurnableBehaviour");

    std::shared_ptr<PbdObject> getPbdObject() const { return m_burnableObject; }

    void setTrackOnly(bool val) { m_trackOnly = val; }

    bool getTrackOnly() const { return m_trackOnly; }

    // Check state of mesh and delete cells that are burned away
    // void update(double dt);
    void visualUpdate(const double& dt) override;

    void setUpdateType(PbdObjectCellRemoval::OtherMeshUpdateType updateType)
    {
        m_updateType = updateType;
    }

    PbdObjectCellRemoval::OtherMeshUpdateType getUpdateType() const
    {
        return m_updateType;
    }

    std::shared_ptr<PbdObjectCellRemoval> getCellRemover() const
    {
        return m_cellRemover;
    }

protected:

    // Allocate state against mesh
    void init() override;

    bool m_trackOnly = false;

    PbdObjectCellRemoval::OtherMeshUpdateType m_updateType = PbdObjectCellRemoval::OtherMeshUpdateType::None;

    std::shared_ptr<PbdObject> m_burnableObject;             
    std::shared_ptr<PbdObjectCellRemoval> m_cellRemover;     

    std::shared_ptr<DataArray<double>> m_burnDamagePtr;
    std::shared_ptr<DataArray<double>> m_burnVisualPtr;
};
} // namespace imstk