VolumeTreeNode.hpp

/*
 * (c) Copyright 2020 CORSIKA Project, corsika-project@lists.kit.edu
 *
 * This software is distributed under the terms of the GNU General Public
 * Licence version 3 (GPL Version 3). See file LICENSE for a full version of
 * the license.
 */

#pragma once

#include <corsika/framework/geometry/IVolume.hpp>
#include <corsika/media/IEmpty.hpp>
#include <memory>
#include <vector>

namespace corsika {

  template <typename TModelProperties = IEmpty>
  class VolumeTreeNode {

  public:
    using IModelProperties = TModelProperties;
    using VTN_type = VolumeTreeNode<IModelProperties>;
    using VTNUPtr = std::unique_ptr<VTN_type>;
    using IMPSharedPtr = std::shared_ptr<IModelProperties>;
    using VolUPtr = std::unique_ptr<IVolume>;

    VolumeTreeNode(VolUPtr pVolume = nullptr)
        : geoVolume_(std::move(pVolume)) {}

    bool contains(Point const& p) const;

    VolumeTreeNode<IModelProperties> const* excludes(Point const& p) const;

    VolumeTreeNode<IModelProperties> const* getContainingNode(Point const& p) const;

    template <typename TCallable, bool preorder = true>
    void walk(TCallable func) const;

    void addChild(VTNUPtr pChild);

    void excludeOverlapWith(VTNUPtr const& pNode);

    VTN_type const* getParent() const { return parentNode_; };

    auto const& getChildNodes() const { return childNodes_; }

    auto const& getExcludedNodes() const { return excludedNodes_; }

    auto const& getVolume() const { return *geoVolume_; }

    auto const& getModelProperties() const { return *modelProperties_; }

    bool hasModelProperties() const { return modelProperties_.get() != nullptr; }

    template <typename ModelProperties, typename... Args>
    auto setModelProperties(Args&&... args) {
      // static_assert(std::is_base_of_v<IModelProperties, ModelProperties>,
      //            "unusable type provided");
      modelProperties_ = std::make_shared<ModelProperties>(std::forward<Args>(args)...);
      return modelProperties_;
    }

    void setModelProperties(IMPSharedPtr ptr) { modelProperties_ = ptr; }

    // template <class MediumType, typename... Args>
    // static auto createMedium(Args&&... args);

  private:
    std::vector<VTNUPtr> childNodes_;
    std::vector<VTN_type const*> excludedNodes_;
    VTN_type const* parentNode_ = nullptr;
    VolUPtr geoVolume_;
    IMPSharedPtr modelProperties_;
  };

} // namespace corsika

#include <corsika/detail/media/VolumeTreeNode.inl>