FemurObject.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 "imstkLevelSetDeformableObject.h"

#include <unordered_set>

namespace imstk
{
class LocalMarchingCubes;
class TaskNode;
} // namespace imstk

using namespace imstk;

class FemurObject : public LevelSetDeformableObject
{
public:
    FemurObject();
    ~FemurObject() override = default;

    bool initialize() override;

    void visualUpdate() override;

    void createVisualModels();

    void setUseRandomChunkColors(const bool useRandom) { m_useRandomChunkColors = useRandom; }
    bool getUseRandomChunkColors() const { return m_useRandomChunkColors; }

protected:
    void updateModifiedVoxels();

    virtual void initGraphEdges(std::shared_ptr<TaskNode> source, std::shared_ptr<TaskNode> sink) override;

protected:
    std::shared_ptr<LocalMarchingCubes> m_isoExtract;
    std::unordered_set<int>   m_chunksGenerated;               // Lazy generation of chunks
    std::shared_ptr<TaskNode> m_forwardModifiedVoxels;
    bool m_useRandomChunkColors = false;
};