OctreeDebugModel.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 "imstkDebugGeometryModel.h"

namespace imstk
{
class LooseOctree;
class OctreeNode;

class OctreeDebugModel : public DebugGeometryModel
{
public:
    OctreeDebugModel() : DebugGeometryModel(),
        m_looseOctree(nullptr),
        m_maxDisplayDepth(5),
        m_drawNonEmptyParents(true) { }
    ~OctreeDebugModel() override = default;

public:
    void debugUpdate(int maxDisplayDepth, bool drawNonEmptyParents = true);

    void setInputOctree(std::shared_ptr<LooseOctree> octree) { m_looseOctree = octree; }

protected:
    bool updateGeom(OctreeNode* node);

protected:
    std::shared_ptr<LooseOctree> m_looseOctree;

    int  m_maxDisplayDepth; 
    bool m_drawNonEmptyParents;
};
} // namespace imstk