imstkLocalMarchingCubes.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 "imstkGeometryAlgorithm.h"
#include "imstkMath.h"

#include <list>
#include <unordered_set>

namespace imstk
{
class ImageData;
class SurfaceMesh;

class LocalMarchingCubes : public GeometryAlgorithm
{
public:
    LocalMarchingCubes();
    ~LocalMarchingCubes() override = default;

    std::shared_ptr<SurfaceMesh> getOutputMesh(const int i) const;

    void setInputImage(std::shared_ptr<ImageData> inputImage);

    void setIsoValue(double isovalue) { m_isoValue = isovalue; }

    void setAllModified(bool val) { m_allModified = val; }

    void setModified(const Vec3i& coord);

    void clearModified() { m_modifiedVoxels.clear(); }

    void setNumberOfChunks(const Vec3i& numChunks);

    const Vec3i& getNumberOfChunks() const { return m_numChunks; }

    std::unordered_map<int, Vec3i> getModifiedChunks() { return m_modifiedChunks; }

protected:
    void requestUpdate() override;

private:
    // Id + coordinate of the modified voxels
    std::unordered_map<int, Vec3i> m_modifiedVoxels;   // Id + coordinate

    bool m_allModified = true;

    double m_isoValue = 0.0;

    Vec3i  m_numChunks  = Vec3i(0, 0, 0); // Number of chunks along x, y, z
    size_t m_chunkCount = 0;              

    std::unordered_map<int, Vec3i> m_modifiedChunks;
};
} // namespace imstk