imstkSceneObject.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 "imstkEntity.h"
#include "imstkMacros.h"

#include <vector>

namespace imstk
{
class VisualModel;
class Geometry;
class TaskGraph;
class TaskNode;

class SceneObject : public Entity
{
public:
    SceneObject(const std::string& name = "SceneObject");
    ~SceneObject() override = default;

    IMSTK_TYPE_NAME(SceneObject)

    
    std::shared_ptr<TaskGraph> getTaskGraph() const { return m_taskGraph; }

    std::shared_ptr<Geometry> getVisualGeometry() const;
    void setVisualGeometry(std::shared_ptr<Geometry> geometry);

    std::shared_ptr<VisualModel> getVisualModel(const int index) const;
    void addVisualModel(std::shared_ptr<VisualModel> visualModel);
    void removeVisualModel(std::shared_ptr<VisualModel> visualModel);

    std::shared_ptr<TaskNode> getUpdateNode() const { return m_updateNode; }

    std::shared_ptr<TaskNode> getUpdateGeometryNode() const { return m_updateGeometryNode; }

    virtual void update() { }

    virtual void visualUpdate() { }

    virtual void updateGeometries() { }

    virtual bool initialize() { return true; }

    void initGraphEdges();

    virtual void reset() { }

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

    virtual void postModifiedAll();

    std::shared_ptr<TaskGraph> m_taskGraph = nullptr;         

private:
    // Dissallow reassignment of these in subclasses
    std::shared_ptr<TaskNode> m_updateNode = nullptr;
    std::shared_ptr<TaskNode> m_updateGeometryNode = nullptr;
};
} // namespace imstk