/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/MVAUtils/MVAUtils/Node.h

// dear emacs, this is really -*- C++ -*-
#ifndef MVAUtils_Node_H
#define MVAUtils_Node_H

#include <vector>
#include <cstdint>

namespace MVAUtils
{
  class Node
  {
  public:

    typedef int32_t index_t; 
    typedef int8_t var_t; 

    Node(const int ivar, const float val, const index_t right): 
      m_var(ivar), m_right(right), m_val(val) {}
        
    bool IsLeaf() const { return m_var < 0; } 

    index_t GetNext(const float value, index_t index) const ;
    
    var_t GetVar() const { return m_var; }

    float GetVal() const { return m_val; }

    // User should not really need these, but useful for debugging

    index_t GetLeft(index_t index) const {return index + 1; }
    
    index_t GetRight(index_t index) const {return index + m_right; }

    void Print(index_t index) const;

  private:

    var_t m_var; 

    // note: this doesn't need to be the same type as the external index
    int16_t m_right; 
    float m_val; 
  };
}

#endif

---