FlavorTaggerInfoMap.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/RelationsObject.h>
 
#include <map>
#include <string>
 
namespace Belle2 {
  // forward declarations
 
  class Track;
 
  class FlavorTaggerInfoMap : public RelationsObject {
 
  public:
 
    FlavorTaggerInfoMap():
      m_targetTrackLevel(),
      m_probTrackLevel(),
      m_targetEventLevel(),
      m_probEventLevel(),
      m_qpCategory(),
      m_hasTrueTarget(),
      m_isTrueCategory(),
      m_qrCombined(-2),
      m_B0Probability(-2),
      m_B0barProbability(-2)
    {}
 
    void setTargetTrackLevel(const std::string& category, const Belle2::Track* track)
    {
      m_targetTrackLevel.insert(std::pair<std::string, const Track*>(category, track));
    }
 
    void setProbTrackLevel(const std::string& category, float probability)
    {
      m_probTrackLevel.insert(std::pair<std::string, float>(category, probability));
    }
 
    void setTargetEventLevel(const std::string& category, const Belle2::Track* track)
    {
      m_targetEventLevel.insert(std::pair<std::string, const Track*>(category, track));
    }
 
    void setProbEventLevel(const std::string& category, float probability)
    {
      m_probEventLevel.insert(std::pair<std::string, float>(category, probability));
    }
 
    void setQpCategory(const std::string& category, float qr)
    {
      m_qpCategory.insert(std::pair<std::string, float>(category, qr));
    }
 
    void setHasTrueTarget(const std::string& category, float isTrue)
    {
      m_hasTrueTarget.insert(std::pair<std::string, float>(category, isTrue));
    }
 
    void setIsTrueCategory(const std::string& category, float isTrue)
    {
      m_isTrueCategory.insert(std::pair<std::string, float>(category, isTrue));
    }
 
    void setQrCombined(float qr) { m_qrCombined = qr; }
 
    void setB0Probability(float B0Probability) { m_B0Probability = B0Probability; }
 
    void setB0barProbability(float B0barProbability) { m_B0barProbability = B0barProbability; }
 
    std::map<std::string, const Belle2::Track*> getTargetTrackLevel() const { return m_targetTrackLevel; }
 
    std::map<std::string, float> getProbTrackLevel() const { return m_probTrackLevel; }
 
    std::map<std::string, const Belle2::Track*> getTargetEventLevel() const { return m_targetEventLevel; }
 
    std::map<std::string, float> getProbEventLevel() const { return m_probEventLevel; }
 
    std::map<std::string, float> getQpCategory() const { return m_qpCategory; }
 
    std::map<std::string, float> getHasTrueTarget() const { return m_hasTrueTarget; }
 
    std::map<std::string, float> getIsTrueCategory() const { return m_isTrueCategory; }
 
    float getQrCombined() const { return m_qrCombined; }
 
    float getB0Probability() const { return m_B0Probability; }
 
    float getB0barProbability() const { return m_B0barProbability; }
 
  private:
 
    // persistent data members
 
    // Track, Event and Combiner Level Flavor Tagger Info
 
    std::map<std::string, const Belle2::Track*>
    m_targetTrackLevel; 
    std::map<std::string, float>
    m_probTrackLevel; 
    std::map<std::string, const Belle2::Track*>
    m_targetEventLevel; 
    std::map<std::string, float>
    m_probEventLevel; 
    std::map<std::string, float>
    m_qpCategory; 
    std::map<std::string, float>
    m_hasTrueTarget; 
    std::map<std::string, float>
    m_isTrueCategory; 
    float m_qrCombined; 
    float m_B0Probability; 
    float m_B0barProbability; 
    ClassDef(FlavorTaggerInfoMap, 3) 
  };
 
} // end namespace Belle2