NoKickRTSel.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.                  *
 **************************************************************************/
 
#include <TFile.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/dataobjects/hitXP.h>
#include <TObject.h>
#include <tracking/trackFindingVXD/sectorMapTools/NoKickCuts.h>
#include <string>
#include <TLatex.h>
#include <TH1F.h>
#include <TTree.h>
 
#pragma once
 
 
namespace Belle2 {

 
  class NoKickRTSel: public TObject {
 
  public:
    std::vector<hitXP> m_hitXP; 
    std::set<hitXP, hitXP::timeCompare> m_setHitXP; 
    std::vector<hitXP> m_8hitTrack; 
    NoKickCuts m_trackCuts; 
    double m_pmax = 10.; 
    double m_pMag; 
    double m_pt; 
    double m_pdgID; 
 
    TFile* m_noKickOutputTFile; 
    TTree* m_noKickTree; 
    TH1F* m_momSel; 
    TH1F* m_momCut; 
    TH1F* m_momEff; 
    TH1F* m_PDGIDCut; 
    TH1F* m_PDGIDSel; 
    TH1F* m_PDGIDEff; 
    TH1F* m_nCutHit; 
 
    int m_numberOfCuts; 
    int m_Ncuts; 
    bool m_outputFlag; 
    bool m_isCutted; 

    NoKickRTSel(const std::string& fileName, bool outputHisto) :
      m_trackCuts(fileName)
    {
      m_outputFlag = false;
      initNoKickRTSel();
      initHistoNoKick(outputHisto);
    }

    NoKickRTSel() :
      m_trackCuts()
    {
      initNoKickRTSel();
    }

    void initNoKickRTSel()
    {
      m_hitXP.clear();
      m_setHitXP.clear();
      m_8hitTrack.clear();
      m_numberOfCuts = 0;
    }

    void hitXPBuilder(const RecoTrack& track);

    void hit8TrackBuilder(const RecoTrack& track);
 

    bool trackSelector(const RecoTrack& track);

    bool segmentSelector(const hitXP& hit1, const hitXP& hit2, const std::vector<double>& selCut, NoKickCuts::EParameters par,
                         bool is0 = false);

    bool globalCut(const std::vector<hitXP>& track8);

    void initHistoNoKick(bool outHisto)
    {
      if (outHisto) {
        m_noKickOutputTFile = new TFile("TrackSelection_NoKick.root", "RECREATE");
        m_momSel = new TH1F("m_momSel", "m_momSel", 100, 0, 4);
        m_momCut = new TH1F("m_momCut", "m_momCut", 100, 0, 4);
        m_momEff = new TH1F("m_momEff", "m_momEff", 100, 0, 4);
 
        m_PDGIDSel = new TH1F("m_PDGIDSel", "m_PDGIDSel", 6000, -3000, 3000);
        m_PDGIDCut = new TH1F("m_PDGIDCut", "m_PDGIDCut", 6000, -3000, 3000);
        m_PDGIDEff = new TH1F("m_PDGIDEff", "m_PDGIDEff", 6000, -3000, 3000);
 
        m_nCutHit = new TH1F("m_nCutHit", "m_nCutHit", 30, 0, 30);
 
 
        m_noKickTree = new TTree("noKickTree", "noKickTree");
        m_noKickTree->Branch("is_rejected", &m_isCutted);
        m_noKickTree->Branch("p_mag", &m_pMag);
        m_noKickTree->Branch("pt", &m_pt);
        m_noKickTree->Branch("pdgID", &m_pdgID);
        m_noKickTree->Branch("number_of_rejected_SP", &m_Ncuts);
 
        m_outputFlag = true;
      }
 
    }

    void produceHistoNoKick();

    ClassDef(NoKickRTSel, 2);
  };

}