SaitoSquaredDistanceTransform.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
// Inspired from bil_edt.cxx (VXL) by Ricardo Fabbri (rfabbri), Brown University
// (rfabbri@lems.brown.edu)
 
// Local
#include "Grid3D.h"
#include "MathTools.h"
 
namespace cloudViewer {
class GenericProgressCallback;
class NormalizedProgress;
class GenericIndexedMesh;
class GenericCloud;
 
class CV_CORE_LIB_API SaitoSquaredDistanceTransform : public Grid3D<unsigned>,
                                                      public MathTools {
public:
    SaitoSquaredDistanceTransform() = default;
 
 
    inline bool initGrid(const Tuple3ui& gridSize) {
        return Grid3D<GridElement>::init(
                gridSize.x, gridSize.y, gridSize.z, 0,
                0);  // margin = 0 (we need continuous memory)
    }
 
    inline bool initDT(GenericIndexedMesh* mesh,
                       PointCoordinateType cellLength,
                       const CCVector3& gridMinCorner,
                       GenericProgressCallback* progressCb = nullptr) {
        return intersectWith(mesh, cellLength, gridMinCorner, 1, progressCb);
    }
 
    inline bool initDT(GenericCloud* cloud,
                       PointCoordinateType cellLength,
                       const CCVector3& gridMinCorner,
                       GenericProgressCallback* progressCb = nullptr) {
        return intersectWith(cloud, cellLength, gridMinCorner, 1, progressCb);
    }
 
 
    inline bool propagateDistance(
            GenericProgressCallback* progressCb = nullptr) {
        return SDT_3D(*this, progressCb);
    }
 
protected:
    static bool EDT_1D(GridElement* slice, std::size_t r, std::size_t c);
    static bool SDT_2D(Grid3D<GridElement>& image,
                       std::size_t sliceIndex,
                       const std::vector<GridElement>& sq);
    static bool SDT_3D(Grid3D<GridElement>& image,
                       GenericProgressCallback* progressCb = nullptr);
};
 
}  // namespace cloudViewer