FlavorTaggerInfo.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Christian Roca and Fernando Abudinen                     *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/RelationsObject.h>
#include <analysis/dataobjects/Particle.h>
#include <analysis/dataobjects/FlavorTaggerInfoMap.h>
 
#include <vector>
 
namespace Belle2 {
  // forward declarations
 
  class Track;
  class MCParticle;
  class FlavorTaggerInfoMap;
 
  class FlavorTaggerInfo : public RelationsObject {
 
  public:
 
    FlavorTaggerInfo()
    {
      m_goodTracksPurityFT = 0;
      m_goodTracksPurityROE = 0;
      m_badTracksROE = 0;
      m_goodTracksROE = 0;
      m_badTracksFT = 0;
      m_goodTracksFT = 0;
    };
 
 
    void setUseModeFlavorTagger(const std::string& mode);
 
    void addMethodMap(const std::string& method);
 
    void setTrack(Belle2::Track* track);
 
    void setParticle(Particle* Particle);
 
    void setMCParticle(Belle2::MCParticle* MCParticle);
 
    void setMCParticleMother(Belle2::MCParticle* MCParticle);
 
    void setCatProb(float catProb);
 
    void setTargProb(float);
 
    void setP(float momentum);
 
    void setIsFromB(int isFromB);
 
    void setCategories(const std::string& catName);
 
    void setD0(double D0);
 
    void setZ0(double Z0);
 
    void setProdPointResolutionZ(float deltaProdZ);
 
//      /**
//      * Vector filler: Set the momentum absolute value of the corresponding category's track
//      * @param momentum momentum
//      */
//     void setTracks(std::vector<Belle2::Track*>); // Save a vector of tracks
 
 
    void setGoodTracksPurityFT(float goodTracksPurity);
 
    void setGoodTracksPurityROE(float goodTracksPurity);
 
    void setGoodTracksROE(int numTracks);
 
    void setBadTracksROE(int numTracks);
 
    void setGoodTracksFT(int numTracks);
 
    void setBadTracksFT(int numTracks);
 
 
 
 
    std::string getUseModeFlavorTagger();
 
    FlavorTaggerInfoMap* getMethodMap(const std::string& method);
 
    bool isMethodInMap(const std::string& method);
 
    std::vector<Belle2::Track*> getTracks();
 
    Belle2::Track* getTrack(int position);
 
    std::vector<float> getCatProb();
 
    std::vector<float> getTargProb();
 
    std::vector<float> getP();
 
    std::vector<Particle*> getParticle();
 
    std::vector<Belle2::MCParticle*> getMCParticle();
 
    std::vector<Belle2::MCParticle*> getMCParticleMother();
 
    std::vector<float> getD0();
 
    std::vector<float> getZ0();
 
    std::vector<int> getIsFromB();
 
    std::vector<std::string> getCategories();
 
    std::vector<float> getProdPointResolutionZ();
 
    float getGoodTracksPurityFT();
 
    float getGoodTracksPurityROE();
 
    int getBadTracksROE();
 
    int getGoodTracksROE();
 
    int getBadTracksFT();
 
    int getGoodTracksFT();
 
 
  private:
 
    // persistent data members
 
    // Track Level Flavor Tagger Info
 
    std::string m_useModeFlavorTagger; 
    std::map<std::string, FlavorTaggerInfoMap*>
    m_methodMap; 
    std::vector<Belle2::Track*> m_tracks;  
    std::vector<Particle*> m_particle; 
    std::vector<Belle2::MCParticle*> m_MCparticle; 
    std::vector<Belle2::MCParticle*> m_MCparticle_mother; 
    std::vector<float> m_D0; 
    std::vector<float> m_Z0; 
    std::vector<std::string> m_categories;  
    std::vector<float> m_prodPointResZ;  
    std::vector<float> m_categoryProb; 
    std::vector<float> m_targetProb; 
    std::vector<float> m_P; 
    std::vector<int> m_isFromB; 
    float m_goodTracksPurityFT; 
    float m_goodTracksPurityROE; 
    int m_badTracksROE; 
    int m_goodTracksROE; 
    int m_badTracksFT; 
    int m_goodTracksFT; 
    ClassDef(FlavorTaggerInfo, 3) 
  };
 
 
} // end namespace Belle2