DecayTree.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Thomas Keck                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/utility/DecayNode.h>
 
#include <vector>
#include <string>
 
namespace Belle2 {
  class DecayTree {
 
  public:
    explicit DecayTree(const std::string& decaystring, bool removeRadiativeGammaFlag = false);
 
    DecayTree(const DecayTree& tree);
 
    DecayTree& operator=(const DecayTree& tree);
 
    std::vector<DecayNode> build_tree(const std::string& decaystring, bool removeRadiativeGammaFlag);
 
    bool find_decay(const DecayTree& tree) const;
 
    std::string to_string() const;
 
    int getMatchSymbolPosition() const { return m_match_symbol_position; }
 
    bool isValid() const { return m_valid; }
 
    const DecayNode& getDecayNode(unsigned int n) const;
 
    DecayNode& getDecayNode(unsigned int n);
 
    size_t getNumberOfDecayNodes() const { return m_nodes_cache.size(); }
 
  private:
    void build_cache(DecayNode& node);
 
  private:
    bool m_valid; 
    size_t m_i; 
    unsigned int m_token_count; 
    int m_match_symbol_position; 
    DecayNode m_root_node; 
    std::vector<DecayNode*>
    m_nodes_cache; 
  };
}