NoKickRTSel Class Reference

This class implement some methods useful for the application of cuts evaluated in NoKickCutsEval module. More...

#include <NoKickRTSel.h>

Inheritance diagram for NoKickRTSel:

Public Member Functions
	NoKickRTSel (const std::string &fileName, bool outputHisto)
	Constructor with input file for use specific cuts file and allows validation.
	NoKickRTSel ()
	Empty Constructor that uses the defaults cuts file.
void	initNoKickRTSel ()
	Initialize the class cleaning the member vectors.
void	hitXPBuilder (const RecoTrack &track)
	this method build a vector of hitXP from a track.
void	hit8TrackBuilder (const RecoTrack &track)
	this method build a vector of hitXP from a track selecting the first hit on each layer of VXD (8 hit for SVD only, counting overlaps).
bool	trackSelector (const RecoTrack &track)
	This method return true if every segment (see segmentSelector) of the input track respects the cuts constraints.
bool	segmentSelector (const hitXP &hit1, const hitXP &hit2, const std::vector< double > &selCut, NoKickCuts::EParameters par, bool is0=false)
	This method return true if a couple of hits resects the cuts constraints.
bool	globalCut (const std::vector< hitXP > &track8)
	This method make some global cuts on the tracks (layer 3 and 6 required, d0 and z0 inside beam pipe).
void	initHistoNoKick (bool outHisto)
	This method initialize some validation histograms of the Training Sample Selection.
void	produceHistoNoKick ()
	This method produce the validation histograms (to be used the endrun combined with the filling in trackSelector method)
	ClassDef (NoKickRTSel, 2)
	Making this class a ROOT class.

Public Attributes
std::vector< hitXP >	m_hitXP
	vector of hit, to convert the track
std::set< hitXP, hitXP::timeCompare >	m_setHitXP
	set of hit to order the hit in time
std::vector< hitXP >	m_8hitTrack
	vector of selected hit
NoKickCuts	m_trackCuts
	auxiliary member to apply the cuts
double	m_pmax = 10.
	range analyzed with cuts
double	m_pMag
	momentum magnitut
double	m_pt
	transverse momentum
double	m_pdgID
	pdg Code
TFile *	m_noKickOutputTFile
	validation output TFile
TTree *	m_noKickTree
	TTree to which the information is written.
TH1F *	m_momSel
	histogram of selected tracks
TH1F *	m_momCut
	histogram of cut tracks
TH1F *	m_momEff
	histogram for efficiency
TH1F *	m_PDGIDCut
	histogram for PDGID of cut track
TH1F *	m_PDGIDSel
	histogram for PDGID of selected track
TH1F *	m_PDGIDEff
	histogram for efficiency for each PDGID
TH1F *	m_nCutHit
	histogram for number of cut hist per track
int	m_numberOfCuts
	number of catastrophic interaction for each track
int	m_Ncuts
	number of times the cut is applied on a particle
bool	m_outputFlag
	true=produce validation output
bool	m_isCutted
	Indicator if cut is applied.

Detailed Description

This class implement some methods useful for the application of cuts evaluated in NoKickCutsEval module.

Use and auxiliary class (NoKickCuts) that contains the cuts used in selection.

Definition at line 33 of file NoKickRTSel.h.

Constructor & Destructor Documentation

NoKickRTSel() [1/2]

NoKickRTSel ( const std::string & fileName, bool outputHisto )

inline

Constructor with input file for use specific cuts file and allows validation.

Definition at line 61 of file NoKickRTSel.h.

                                                             :
      m_trackCuts(fileName)
    {
      m_outputFlag = false;
      initNoKickRTSel();
      initHistoNoKick(outputHisto);
    }

NoKickRTSel() [2/2]

NoKickRTSel ( )

inline

Empty Constructor that uses the defaults cuts file.

Definition at line 70 of file NoKickRTSel.h.

                  :
      m_trackCuts()
    {
      initNoKickRTSel();
    }

Member Function Documentation

globalCut()

bool globalCut

(

const std::vector< hitXP > &

track8

)

This method make some global cuts on the tracks (layer 3 and 6 required, d0 and z0 inside beam pipe).

Return false if this filter fails. input (the selected hit of the track)

Definition at line 111 of file NoKickRTSel.cc.

{
  int flagd0 = 1;
  int flagz0 = 1;
  int lay3 = 0;
  int lay4 = 0;
  int lay5 = 0;
  int lay6 = 0;
  for (const hitXP& XP : track8) {
    if (XP.getSensorLayer() == 3) lay3 = 1;
    if (XP.getSensorLayer() == 4) lay4 = 1;
    if (XP.getSensorLayer() == 5) lay5 = 1;
    if (XP.getSensorLayer() == 6) lay6 = 1;
    //    if (fabs(XP.getD0Entry()) > 1.) flagd0 = 0;
    //    if (fabs(XP.getZ0Entry()) > 1.) flagz0 = 0;
  }
  int N_lay = lay3 + lay4 + lay5 + lay6;
  if (N_lay >= 3) N_lay = 1;
  else N_lay = 0;
  int flagTot = flagd0 * flagz0 * N_lay;
  if (flagTot == 1) return true;
  else return false;
}

hit8TrackBuilder()

void hit8TrackBuilder

(

const RecoTrack &

track

)

this method build a vector of hitXP from a track selecting the first hit on each layer of VXD (8 hit for SVD only, counting overlaps).

The output is the member of the class. input (one reconstructed track)

Definition at line 96 of file NoKickRTSel.cc.

{
  hitXPBuilder(track);
  for (const hitXP& XP : m_hitXP) {
    if (m_8hitTrack.size() < 1) {
      m_8hitTrack.push_back(XP);
    } else if (XP.m_sensorLayer != m_8hitTrack.back().m_sensorLayer ||
               XP.m_sensorLadder != m_8hitTrack.back().m_sensorLadder ||
               XP.m_sensorSensor != m_8hitTrack.back().m_sensorSensor) {
      m_8hitTrack.push_back(XP);
    }
  }
 
}

hitXPBuilder()

void hitXPBuilder

(

const RecoTrack &

track

)

this method build a vector of hitXP from a track.

The output is the member of the class. input (one reconstructed track)

Definition at line 21 of file NoKickRTSel.cc.

{
  StoreArray<SVDCluster> SVDClusters;
  StoreArray<SVDTrueHit> SVDTrueHits;
  StoreArray<MCParticle> MCParticles;
  StoreArray<RecoTrack> recoTracks;
 
  if (track.getRelationsTo<MCParticle>().size() > 0) {
 
    const MCParticle* particle = track.getRelationsTo<MCParticle>()[0];
 
    std::vector<Belle2::RecoHitInformation::UsedSVDHit*> clusterListSVD = track.getSVDHitList();
    for (const SVDCluster* cluster : clusterListSVD) {
      for (const SVDTrueHit& hit : cluster->getRelationsTo<SVDTrueHit>()) {
        RelationVector<MCParticle> hitFromParticle = hit.getRelationsFrom<MCParticle>();
        if (hitFromParticle.size() > 0) {
          if (hitFromParticle[0]->getIndex() != particle->getIndex()) {
            continue;
          }
        } else continue;
        VxdID trueHitSensorID = hit.getSensorID();
        const VXD::SensorInfoBase& sensorInfo = VXD::GeoCache::getInstance().getSensorInfo(trueHitSensorID);
        hitXPDerivate entry(hit, *cluster, *particle, sensorInfo);
        int NClusterU = 0;
        int NClusterV = 0;
        for (const SVDCluster& Ncluster : hit.getRelationsFrom<SVDCluster>()) {
          if (Ncluster.isUCluster()) NClusterU++;
          else NClusterV++;
        }
        entry.setClusterU(NClusterU);
        entry.setClusterV(NClusterV);
 
        bool isReconstructed(false);
        for (const RecoTrack& aRecoTrack : particle->getRelationsFrom<RecoTrack>())
          isReconstructed |= aRecoTrack.hasSVDHits();
        entry.setReconstructed(isReconstructed);
 
        m_setHitXP.insert(entry);
      }
    }
 
 
    StoreArray<PXDCluster> PXDClusters;
    StoreArray<PXDTrueHit> PXDTrueHits;
 
    std::vector<Belle2::RecoHitInformation::UsedPXDHit*> clusterListPXD = track.getPXDHitList();
    for (const PXDCluster* cluster : clusterListPXD) {
      for (const PXDTrueHit& hit : cluster->getRelationsTo<PXDTrueHit>()) {
        RelationVector<MCParticle> hitFromParticle = hit.getRelationsFrom<MCParticle>();
        if (hitFromParticle.size() > 0) {
          if (hitFromParticle[0]->getIndex() != particle->getIndex()) {
            continue;
          }
        } else continue;
        VxdID trueHitSensorID = hit.getSensorID();
        const VXD::SensorInfoBase& sensorInfo = VXD::GeoCache::getInstance().getSensorInfo(trueHitSensorID);
        hitXPDerivate entry(hit, *particle, sensorInfo);
 
        bool isReconstructed(false);
        for (const RecoTrack& aRecoTrack : particle->getRelationsFrom<RecoTrack>())
          isReconstructed |= aRecoTrack.hasPXDHits();
        entry.setReconstructed(isReconstructed);
 
        m_setHitXP.insert(entry);
      }
    }
 
    for (const auto& element : m_setHitXP) {
      m_hitXP.push_back(element);
    }
 
  }
 
}

initHistoNoKick()

void initHistoNoKick ( bool outHisto )

inline

This method initialize some validation histograms of the Training Sample Selection.

The input boolean allows the initialization, otherwise the method is empty (no validation)

Definition at line 120 of file NoKickRTSel.h.

    {
      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;
      }
 
    }

initNoKickRTSel()

void initNoKickRTSel ( )

inline

Initialize the class cleaning the member vectors.

Definition at line 77 of file NoKickRTSel.h.

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

produceHistoNoKick()

void produceHistoNoKick

(

)

This method produce the validation histograms (to be used the endrun combined with the filling in trackSelector method)

Definition at line 278 of file NoKickRTSel.cc.

{
  if (m_outputFlag) {
    m_noKickOutputTFile->cd();
    m_momSel->Write();
    m_momCut->Write();
 
    m_momEff->Add(m_momSel, 1);
    m_momEff->Add(m_momCut, 1);
    m_momEff->Divide(m_momSel, m_momEff, 1, 1);
    m_momEff->Write();
 
    m_PDGIDSel->Write();
    m_PDGIDCut->Write();
 
    m_PDGIDEff->Add(m_PDGIDSel, 1);
    m_PDGIDEff->Add(m_PDGIDCut, 1);
    m_PDGIDEff->Divide(m_PDGIDSel, m_PDGIDEff, 1, 1);
    m_PDGIDEff->Write();
 
    m_nCutHit->Write();
 
    m_noKickTree->Write();
 
    delete m_noKickOutputTFile;
  }
}

segmentSelector()

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

This method return true if a couple of hits resects the cuts constraints.

input (first hit, second hit, selected cut to apply, track parameter, it is first hit the IP?)

Definition at line 135 of file NoKickRTSel.cc.

{
  if (hit2.m_sensorLayer - hit1.m_sensorLayer > 1) return true;
  else {
    double deltaPar = 0;
    switch (par) {
      case NoKickCuts::c_Omega:
        deltaPar = fabs(hit1.getOmegaEntry() - hit2.getOmegaEntry());
        if (is0) deltaPar = fabs(hit1.getOmega0() - hit2.getOmegaEntry());
        //    selCutPXD =0.4;
        break;
 
      case NoKickCuts::c_D0:
        deltaPar = hit1.getD0Entry() - hit2.getD0Entry();
        if (is0) deltaPar = hit1.getD00() - hit2.getD0Entry();
        //  selCutPXD =1;
        break;
 
      case NoKickCuts::c_Phi0:
        deltaPar = asin(sin(hit1.getPhi0Entry())) - asin(sin(hit2.getPhi0Entry()));
        if (is0) deltaPar = asin(sin(hit1.getPhi00())) - asin(sin(hit2.getPhi0Entry()));
        //  selCutPXD =0.3;
        break;
 
      case NoKickCuts::c_Z0:
        deltaPar = hit1.getZ0Entry() - hit2.getZ0Entry();
        if (is0) deltaPar = hit1.getZ00() - hit2.getZ0Entry();
        //  selCutPXD =1;
        break;
 
      case NoKickCuts::c_Tanlambda:
        deltaPar = hit1.getTanLambdaEntry() - hit2.getTanLambdaEntry();
        if (is0) deltaPar = hit1.getTanLambda0() - hit2.getTanLambdaEntry();
        //  selCutPXD =0.3;
        break;
    }
 
    double usedCut = 0;
    if (fabs(selCut.at(0)) > fabs(selCut.at(1))) {
      usedCut = fabs(selCut.at(0));
    } else usedCut = fabs(selCut.at(1));
 
    if (deltaPar > -usedCut && deltaPar < usedCut) return true;
    else {
      B2DEBUG(20, "--------------------------");
      B2DEBUG(20, "lay1=" << hit1.m_sensorLayer);
      B2DEBUG(20, "lay2=" << hit2.m_sensorLayer);
      B2DEBUG(20, "parameter=" << par);
      B2DEBUG(20, "Min=" << selCut.at(0));
      B2DEBUG(20, "Max=" << selCut.at(1));
      B2DEBUG(20, "deltaPar=" << deltaPar);
      B2DEBUG(20, "momentum=" << hit1.m_momentum0.R());
      return false;
    }
  }
}

trackSelector()

bool trackSelector

(

const RecoTrack &

track

)

This method return true if every segment (see segmentSelector) of the input track respects the cuts constraints.

input (one reconstructed track)

Definition at line 195 of file NoKickRTSel.cc.

{
  initNoKickRTSel();
  hit8TrackBuilder(track);
 
  if (m_outputFlag) {
    m_pdgID = m_8hitTrack[0].getPDGID();
    m_pMag = track.getMomentumSeed().R();
    m_pt = sqrt(track.getMomentumSeed().X() * track.getMomentumSeed().X() + track.getMomentumSeed().Y() * track.getMomentumSeed().Y());
  }
 
  bool good = globalCut(m_8hitTrack);
  if (good == false) {
    if (m_outputFlag) {
      m_numberOfCuts = -1; //it means "no specific cuts applied, but rejected for global cuts"
      m_Ncuts = m_numberOfCuts;
      m_isCutted = true;
      m_nCutHit->Fill(m_numberOfCuts);
      m_momCut->Fill(track.getMomentumSeed().R());
      m_PDGIDCut->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
      m_noKickTree->Fill();
    }
    return false;
  }
 
  if (track.getMomentumSeed().R() > m_pmax) {
    if (m_outputFlag) {
      m_Ncuts = m_numberOfCuts;
      m_isCutted = false;
      m_nCutHit->Fill(m_numberOfCuts);
      m_momSel->Fill(track.getMomentumSeed().R());
      m_PDGIDSel->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
      m_noKickTree->Fill();
    }
    return good;
  }
  for (int i = 0; i < (int)(m_8hitTrack.size() - 2); i++) {
    double sinTheta = fabs(m_8hitTrack.at(i).m_momentum0.Y()) /
                      sqrt(pow(m_8hitTrack.at(i).m_momentum0.Y(), 2) +
                           pow(m_8hitTrack.at(i).m_momentum0.Z(), 2));
 
    double momentum =
      sqrt(pow(m_8hitTrack.at(i).m_momentum0.X(), 2) +
           pow(m_8hitTrack.at(i).m_momentum0.Y(), 2) +
           pow(m_8hitTrack.at(i).m_momentum0.Z(), 2));
 
    for (int j = NoKickCuts::c_Omega; j <= NoKickCuts::c_Tanlambda; j++) { //track parameters loop
      std::vector<double> selCut = m_trackCuts.cutSelector(sinTheta, momentum, m_8hitTrack.at(i).m_sensorLayer,
                                                           m_8hitTrack.at(i + 1).m_sensorLayer, (NoKickCuts::EParameters) j);
      bool goodSeg = segmentSelector(m_8hitTrack.at(i), m_8hitTrack.at(i + 1), selCut, (NoKickCuts::EParameters) j);
 
      if (!goodSeg) {
        good = false;
        m_numberOfCuts++;
      }
      if (i == 0) { //beampipe crossing
        bool goodSeg0 = segmentSelector(m_8hitTrack.at(i), m_8hitTrack.at(i), selCut, (NoKickCuts::EParameters) j, true);
        if (!goodSeg0) {
          good = false;
          m_numberOfCuts++;
        }
      }
 
    }
  }
  if (m_outputFlag) {
    m_Ncuts = m_numberOfCuts;
    m_nCutHit->Fill(m_numberOfCuts);
    if (good) {
      m_isCutted = false;
      m_momSel->Fill(track.getMomentumSeed().R());
      m_PDGIDSel->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
    } else {
      m_isCutted = true;
      m_momCut->Fill(track.getMomentumSeed().R());
      m_PDGIDCut->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
    }
    m_noKickTree->Fill();
  }
  return good;
}

Member Data Documentation

m_8hitTrack

std::vector<hitXP> m_8hitTrack

vector of selected hit

Definition at line 38 of file NoKickRTSel.h.

m_hitXP

std::vector<hitXP> m_hitXP

vector of hit, to convert the track

Definition at line 36 of file NoKickRTSel.h.

m_isCutted

bool m_isCutted

Indicator if cut is applied.

Definition at line 58 of file NoKickRTSel.h.

m_momCut

TH1F* m_momCut

histogram of cut tracks

Definition at line 48 of file NoKickRTSel.h.

m_momEff

TH1F* m_momEff

histogram for efficiency

Definition at line 49 of file NoKickRTSel.h.

m_momSel

TH1F* m_momSel

histogram of selected tracks

Definition at line 47 of file NoKickRTSel.h.

m_nCutHit

TH1F* m_nCutHit

histogram for number of cut hist per track

Definition at line 53 of file NoKickRTSel.h.

m_Ncuts

int m_Ncuts

number of times the cut is applied on a particle

Definition at line 56 of file NoKickRTSel.h.

m_noKickOutputTFile

TFile* m_noKickOutputTFile

validation output TFile

Definition at line 45 of file NoKickRTSel.h.

m_noKickTree

TTree* m_noKickTree

TTree to which the information is written.

Definition at line 46 of file NoKickRTSel.h.

m_numberOfCuts

int m_numberOfCuts

number of catastrophic interaction for each track

Definition at line 55 of file NoKickRTSel.h.

m_outputFlag

bool m_outputFlag

true=produce validation output

Definition at line 57 of file NoKickRTSel.h.

m_pdgID

double m_pdgID

pdg Code

Definition at line 43 of file NoKickRTSel.h.

m_PDGIDCut

TH1F* m_PDGIDCut

histogram for PDGID of cut track

Definition at line 50 of file NoKickRTSel.h.

m_PDGIDEff

TH1F* m_PDGIDEff

histogram for efficiency for each PDGID

Definition at line 52 of file NoKickRTSel.h.

m_PDGIDSel

TH1F* m_PDGIDSel

histogram for PDGID of selected track

Definition at line 51 of file NoKickRTSel.h.

m_pMag

double m_pMag

momentum magnitut

Definition at line 41 of file NoKickRTSel.h.

m_pmax

double m_pmax = 10.

range analyzed with cuts

Definition at line 40 of file NoKickRTSel.h.

m_pt

double m_pt

transverse momentum

Definition at line 42 of file NoKickRTSel.h.

m_setHitXP

std::set<hitXP, hitXP::timeCompare> m_setHitXP

set of hit to order the hit in time

Definition at line 37 of file NoKickRTSel.h.

m_trackCuts

NoKickCuts m_trackCuts

auxiliary member to apply the cuts

Definition at line 39 of file NoKickRTSel.h.

---

The documentation for this class was generated from the following files:

• tracking/trackFindingVXD/sectorMapTools/include/NoKickRTSel.h
• tracking/trackFindingVXD/sectorMapTools/src/NoKickRTSel.cc