imstkDebugGeometryModel.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 "imstkComponent.h"
#include "imstkColor.h"
#include "imstkMath.h"

namespace imstk
{
class LineMesh;
class PointSet;
class RenderMaterial;
class SurfaceMesh;
template<typename T, int N> class VecDataArray;
class VisualModel;

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

protected:
    void init() override;

public:
    void addLine(const Vec3d& a, const Vec3d& b);

    void addLine(const Vec3d& a, const Vec3d& b, const Color& color);

    void addTriangle(const Vec3d& a, const Vec3d& b, const Vec3d& c);

    void addTriangle(const Vec3d& a, const Vec3d& b, const Vec3d& c, const Color& color);

    void addPoint(const Vec3d& a);

    void addPoint(const Vec3d& a, const Color& color);

    void addArrow(const Vec3d& start, const Vec3d& end);

    void addArrow(const Vec3d& start, const Vec3d& end, const Color& color);

    void clear();

    std::shared_ptr<RenderMaterial> getPointMaterial() const;
    std::shared_ptr<RenderMaterial> getLineMaterial() const;
    std::shared_ptr<RenderMaterial> getFaceMaterial() const;

    void visualUpdate(const double& dt) override;

    void setLineWidth(const double width);

    void setTriColor(const Color& color);
    void setLineColor(const Color& color);
    void setPointColor(const Color& color);
    void setArrowColor(const Color& color);

    void setPointSize(const double size);

    void setArrowScale(const double arrowScale) { m_arrowScale = arrowScale; }

    int getNumPoints() const;
    int getNumLines() const;
    int getNumTriangles() const;

protected:
    double m_arrowScale;
    Color  m_arrowColor;

    std::shared_ptr<VisualModel> m_debugLineModel;
    std::shared_ptr<VisualModel> m_debugPointModel;
    std::shared_ptr<VisualModel> m_debugSurfModel;

    std::shared_ptr<LineMesh>    m_debugLineMesh;
    std::shared_ptr<PointSet>    m_debugPointSet;
    std::shared_ptr<SurfaceMesh> m_debugSurfMesh;

    std::shared_ptr<VecDataArray<double, 3>> m_triVerticesPtr;
    std::shared_ptr<VecDataArray<int, 3>>    m_triIndicesPtr;
    std::shared_ptr<VecDataArray<unsigned char, 3>> m_triColorsPtr;
    bool m_trianglesChanged;

    std::shared_ptr<VecDataArray<double, 3>> m_lineVerticesPtr;
    std::shared_ptr<VecDataArray<int, 2>>    m_lineIndicesPtr;
    std::shared_ptr<VecDataArray<unsigned char, 3>> m_lineColorsPtr;
    bool m_linesChanged;

    std::shared_ptr<VecDataArray<double, 3>> m_pointVerticesPtr;
    std::shared_ptr<VecDataArray<unsigned char, 3>> m_pointColorsPtr;
    bool m_ptsChanged;
};
} // namespace imstk