imstkSurfaceMeshDistanceTransform.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 <vtkSmartPointer.h>

class vtkImplicitPolyDataDistance;

namespace imstk
{
class ImageData;
class SurfaceMesh;

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

public:
    void setInputMesh(std::shared_ptr<SurfaceMesh> surfMesh);

    std::shared_ptr<ImageData> getOutputImage();

    void setupDistFunc();

    Vec3d getNearestPoint(const Vec3d& pos);

    imstkSetMacro(Dimensions, const Vec3i&);
    void setDimensions(int dimX, int dimY, int dimZ) { setDimensions(Vec3i(dimX, dimY, dimZ)); }
    imstkGetMacro(Dimensions, const Vec3i&);

    void setBounds(const Vec3d& min, const Vec3d& max);
    void setBounds(const Vec6d&);
    imstkGetMacro(Bounds, const Vec6d&);

    imstkSetMacro(NarrowBanded, bool);
    imstkGetMacro(NarrowBanded, bool);

    imstkSetMacro(DilateSize, int);
    imstkGetMacro(DilateSize, int);

    imstkSetMacro(Tolerance, double);
    imstkGetMacro(Tolerance, double);

protected:
    void requestUpdate() override;

private:
    Vec3i  m_Dimensions = Vec3i::Zero();
    Vec6d  m_Bounds     = Vec6d::Zero();
    double m_Tolerance  = 1.0e-10;

    bool m_NarrowBanded = false;
    int  m_DilateSize   = 4; 

    vtkSmartPointer<vtkImplicitPolyDataDistance> m_distFunc;
};
} // namespace imstk