section.h

#pragma once

#include <functional>

#include <morphio/properties.h>
#include <morphio/section.h>
#include <morphio/types.h>

#include <morphio/section_iterators.hpp>

namespace morphio {
namespace mut {

using upstream_iterator = morphio::upstream_iterator_t<std::shared_ptr<Section>>;
using breadth_iterator = morphio::breadth_iterator_t<std::shared_ptr<Section>, Morphology>;
using depth_iterator = morphio::depth_iterator_t<std::shared_ptr<Section>, Morphology>;

class Section: public std::enable_shared_from_this<Section>
{
  public:
    ~Section() = default;

    inline uint32_t id() const noexcept;

    inline SectionType& type() noexcept;
    inline const SectionType& type() const noexcept;
    inline std::vector<Point>& points() noexcept;
    inline const std::vector<Point>& points() const noexcept;
    inline std::vector<morphio::floatType>& diameters() noexcept;
    inline const std::vector<morphio::floatType>& diameters() const noexcept;
    inline std::vector<morphio::floatType>& perimeters() noexcept;
    inline const std::vector<morphio::floatType>& perimeters() const noexcept;
    inline Property::PointLevel& properties() noexcept;
    inline const Property::PointLevel& properties() const noexcept;
    //
    // Methods that were previously in mut::Morphology
    //

    const std::shared_ptr<Section>& parent() const;

    bool isRoot() const;

    const std::vector<std::shared_ptr<Section>>& children() const;

    depth_iterator depth_begin() const;
    depth_iterator depth_end() const;

    breadth_iterator breadth_begin() const;
    breadth_iterator breadth_end() const;

    upstream_iterator upstream_begin() const;
    upstream_iterator upstream_end() const;

    std::shared_ptr<Section> appendSection(const morphio::Section&, bool recursive = false);

    std::shared_ptr<Section> appendSection(std::shared_ptr<Section> original_section,
                                           bool recursive = false);

    std::shared_ptr<Section> appendSection(
        const Property::PointLevel&, SectionType sectionType = SectionType::SECTION_UNDEFINED);

  private:
    friend class Morphology;

    Section(Morphology*, unsigned int id, SectionType type, const Property::PointLevel&);
    Section(Morphology*, unsigned int id, const morphio::Section& section);
    Section(Morphology*, unsigned int id, const Section&);


    void throwIfNoOwningMorphology() const;

    Morphology* getOwningMorphologyOrThrow() const;

    Morphology* _morphology;
    Property::PointLevel _pointProperties;
    uint32_t _id;
    SectionType _sectionType;
};

std::ostream& operator<<(std::ostream&, const std::shared_ptr<Section>&);

inline uint32_t Section::id() const noexcept {
    return _id;
}

inline SectionType& Section::type() noexcept {
    return _sectionType;
}

inline const SectionType& Section::type() const noexcept {
    return _sectionType;
}

inline std::vector<Point>& Section::points() noexcept {
    return _pointProperties._points;
}

inline const std::vector<Point>& Section::points() const noexcept {
    return _pointProperties._points;
}

inline std::vector<morphio::floatType>& Section::diameters() noexcept {
    return _pointProperties._diameters;
}

inline const std::vector<morphio::floatType>& Section::diameters() const noexcept {
    return _pointProperties._diameters;
}

inline std::vector<morphio::floatType>& Section::perimeters() noexcept {
    return _pointProperties._perimeters;
}

inline const std::vector<morphio::floatType>& Section::perimeters() const noexcept {
    return _pointProperties._perimeters;
}

inline Property::PointLevel& Section::properties() noexcept {
    return _pointProperties;
}

inline const Property::PointLevel& Section::properties() const noexcept {
    return _pointProperties;
}

}  // namespace mut
}  // namespace morphio

std::ostream& operator<<(std::ostream&, const morphio::mut::Section&);