TreeNode.hpp

Go to the documentation of this file.

// Copyright (c) 2013 The ComPWA Team.
// This file is part of the ComPWA framework, check
// https://github.com/ComPWA/ComPWA/license.txt for details.


#ifndef COMPWA_FUNCTIONTREE_TREENODE_HPP_
#define COMPWA_FUNCTIONTREE_TREENODE_HPP_

#include <complex>
#include <memory>
#include <string>

#include "Core/FunctionTree/Functions.hpp"
#include "Core/FunctionTree/ParObserver.hpp"
#include "Core/FunctionTree/ParameterList.hpp"

namespace ComPWA {
namespace FunctionTree {

class TreeNode : public std::enable_shared_from_this<TreeNode>,
                 public ParObserver {
public:
  TreeNode(std::shared_ptr<Strategy> strategy);
  TreeNode(std::shared_ptr<Parameter> parameter,
           std::shared_ptr<Strategy> strategy = nullptr);

  virtual ~TreeNode();

  void removeExpiredParents();

  void addNode(std::shared_ptr<TreeNode> node);
  void addNodes(std::vector<std::shared_ptr<TreeNode>> nodes);
  void addParent(std::weak_ptr<TreeNode> node);

  std::shared_ptr<Parameter> parameter();

  void fillParameters(ParameterList &list);

  void update();

  friend std::ostream &operator<<(std::ostream &os,
                                  std::shared_ptr<TreeNode> p) {
    return os << p->print(-1);
  }

  std::string print(int level = -1, std::string prefix = "");

private:
  std::vector<std::weak_ptr<TreeNode>> Parents;

  std::vector<std::shared_ptr<TreeNode>> ChildNodes;

  std::shared_ptr<ComPWA::FunctionTree::Parameter> OutputParameter;

  bool HasChanged;

  std::shared_ptr<Strategy> Strat;

  std::shared_ptr<ComPWA::FunctionTree::Parameter> recalculate();
};

std::shared_ptr<TreeNode> createLeaf(std::shared_ptr<Parameter> parameter);

template <typename T,
          typename = std::enable_if_t<
              (std::is_same<int, T>::value || std::is_same<double, T>::value ||
               std::is_same<std::complex<double>, T>::value ||
               std::is_same<std::vector<int>, T>::value ||
               std::is_same<std::vector<double>, T>::value ||
               std::is_same<std::vector<std::complex<double>>, T>::value)>>
std::shared_ptr<TreeNode> createLeaf(const T &value, std::string name = "") {
  auto leaf =
      std::make_shared<TreeNode>(std::make_shared<Value<T>>(name, value));
  return leaf;
}

} // namespace FunctionTree
} // namespace ComPWA

#endif