NeighborhoodGenerator.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <memory>
 
#include "vc/core/types/NDArray.hpp"
#include "vc/core/types/Volume.hpp"
 
namespace volcart
{
 
using Neighborhood = NDArray<std::uint16_t>;
 
enum class Direction {
    Negative = -1,
    Bidirectional = 0,
    Positive = 1
};
 
class NeighborhoodGenerator
{
public:
    using Pointer = std::shared_ptr<NeighborhoodGenerator>;
 
    std::size_t dim() { return dim_; }
 
    virtual Neighborhood::Extent extents() const = 0;
    void setSamplingRadius(double r, std::size_t axis = 0)
    {
        radius_[axis] = r;
    }
 
    void setSamplingRadius(double r0, double r1, double r2)
    {
        radius_ = {r0, r1, r2};
    }
 
    void setSamplingRadius(const cv::Vec3d& radii) { radius_ = radii; }
 
    void setSamplingInterval(double i) { interval_ = i; }
 
    void setSamplingDirection(Direction d) { direction_ = d; }
 
    void setAutoGenAxes(bool b) { autoGenAxes_ = b; }
    virtual Neighborhood compute(
        const Volume::Pointer& v,
        const cv::Vec3d& pt,
        const std::vector<cv::Vec3d>& axes) = 0;
protected:
    explicit NeighborhoodGenerator(std::size_t dim) : dim_{dim} {}
 
    virtual ~NeighborhoodGenerator() = default;
 
    const std::size_t dim_{0};
 
    cv::Vec3d radius_{1.0, 1.0, 1.0};
 
    double interval_{1.0};
 
    Direction direction_{Direction::Bidirectional};
 
    bool autoGenAxes_{true};
};
 
}  // namespace volcart