MorphologicalTreeFactory.hpp

#pragma once
 
#include "MorphologicalTree.hpp"
#include "TreeAltitudeAlgorithms.hpp"
#include "WeightedMorphologicalTree.hpp"
#include "detail/MorphologicalTreeConstructionTag.hpp"
#include "sdrt/SelfDualResidualTreeBuilder.hpp"
 
#include <optional>
#include <span>
#include <utility>
 
namespace mmcfilters {
 
class MorphologicalTreeFactory {
private:
    static constexpr detail::MorphologicalTreeConstructionTag tag() noexcept {
        return detail::MorphologicalTreeConstructionTag{};
    }
 
public:
    template<AltitudeValue T>
    [[nodiscard]] static WeightedMorphologicalTree<T> createMaxTree(ImagePtr<T> img, double radius = 1.5) {
        TreeAltitudeAlgorithms::validateFiniteImageAltitudes(img, "MorphologicalTreeFactory::createMaxTree image");
        auto topology = MorphologicalTree(tag(), img, ComponentTreeKind::MAX_TREE, radius);
        return WeightedMorphologicalTree<T>(tag(), std::move(topology), std::move(img));
    }
 
    template<AltitudeValue T>
    [[nodiscard]] static WeightedMorphologicalTree<T> createMinTree(ImagePtr<T> img, double radius = 1.5) {
        TreeAltitudeAlgorithms::validateFiniteImageAltitudes(img, "MorphologicalTreeFactory::createMinTree image");
        auto topology = MorphologicalTree(tag(), img, ComponentTreeKind::MIN_TREE, radius);
        return WeightedMorphologicalTree<T>(tag(), std::move(topology), std::move(img));
    }
 
    [[nodiscard]] static WeightedMorphologicalTree<std::uint8_t> createTreeOfShapes(
        ImageUInt8Ptr img,
        ToSInterpolation interpolation = ToSInterpolation::SelfDual,
        int infinitySeedRow = ToSDefaultInfinityRow,
        int infinitySeedCol = ToSDefaultInfinityCol) {
        auto topology = MorphologicalTree(tag(), img, interpolation, infinitySeedRow, infinitySeedCol);
        return WeightedMorphologicalTree<std::uint8_t>(tag(), std::move(topology), std::move(img));
    }
 
    template<AltitudeValue T>
    [[nodiscard]] static WeightedMorphologicalTree<T> createFromHigraParent(std::span<const NodeId> higraParent, std::span<const T> higraAltitude, int rows, int cols, MorphologicalTreeKind kind, std::optional<AdjacencyRelation> adjacency = std::nullopt) {
        auto topology = MorphologicalTree(tag(), higraParent, rows, cols, kind, std::move(adjacency));
        return WeightedMorphologicalTree<T>(tag(), std::move(topology), detail::AltitudeInput<T>{higraAltitude, detail::AltitudeDomain::HigraNodeIds});
    }
 
    template<AltitudeValue T>
    [[nodiscard]] static WeightedMorphologicalTree<T> createSelfDualResidualTree(ImagePtr<T> img, double radius = 1.5) {
        auto minTree = createMinTree(img, radius);
        auto maxTree = createMaxTree(img, radius);
        sdrt::SelfDualResidualTreeBuilder<T> builder(radius);
        builder.build(img, std::move(minTree), std::move(maxTree));
        auto topology = MorphologicalTree(tag(), builder.getNodeParent(), builder.getProperPartOwner(), builder.getRoot(), builder.getRows(), builder.getCols());
        return WeightedMorphologicalTree<T>(tag(), std::move(topology), detail::AltitudeInput<T>{builder.getAltitude(), detail::AltitudeDomain::InternalNodeSlots});
    }
};
 
} // namespace mmcfilters