FlavorTaggerInfoMap.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Fernando Abudinen                                        *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/RelationsObject.h>
#include <analysis/dataobjects/Particle.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);
 
    void setProbTrackLevel(const std::string& category, float probability);
 
    void setTargetEventLevel(const std::string& category, const Belle2::Track* track);
 
    void setProbEventLevel(const std::string& category, float probability);
 
    void setQpCategory(const std::string& category, float qr);
 
    void setHasTrueTarget(const std::string& category, float isTrue);
 
    void setIsTrueCategory(const std::string& category, float isTrue);
 
    void setQrCombined(float qr);
 
    void setB0Probability(float B0Probability);
 
    void setB0barProbability(float B0barProbability);
 
 
 
    std::map<std::string, const Belle2::Track*> getTargetTrackLevel();
 
    std::map<std::string, float> getProbTrackLevel();
 
    std::map<std::string, const Belle2::Track*> getTargetEventLevel();
 
    std::map<std::string, float> getProbEventLevel();
 
    std::map<std::string, float> getQpCategory();
 
    std::map<std::string, float> getHasTrueTarget();
 
    std::map<std::string, float> getIsTrueCategory();
 
    float getQrCombined();
 
    float getB0Probability();
 
    float getB0barProbability();
 
 
  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