update in MIDdetector class

Author: urasantonio

src/MIDdetector.cc

@@ -11,6 +11,7 @@
 #include "TDatime.h"
 #include "TFile.h"
 #include "TVector3.h"
+#include "TMath.h"
 
 namespace o2 {
   namespace delphes {
@@ -38,6 +39,8 @@ namespace o2 {
 	  printf("Object %s not found, quitting\n",Form("mAccEffMuonPID_%s",partLabel[iPart]));
 	  return kFALSE;
 	}
+	mMomMin[iPart] = TMath::Max(1.2, mAccEffMuonPID[iPart]->GetAxis(1)->GetXmin());
+	mMomMax[iPart] = mAccEffMuonPID[iPart]->GetAxis(1)->GetXmax();
       }
 
       printf("Setup of MIDdetector successfully completed\n");
@@ -49,8 +52,10 @@ namespace o2 {
 
     bool MIDdetector::hasMID(const Track &track) {
 
+      auto pdg  = std::abs(track.PID);
+      auto part = pidmap[pdg];
       TVector3 v(track.XOuter, track.YOuter, track.ZOuter);
-      return (TMath::Abs(v.Eta()) < mEtaMax);
+      return ((TMath::Abs(v.Eta()) < mEtaMax) && (track.P > mMomMin[part]));
 
     }
 
@@ -64,7 +69,9 @@ namespace o2 {
 
       auto particle = (GenParticle*) track.Particle.GetObject();
 
-      Double_t var[4] = {track.P, track.Eta, particle->Z, double(multiplicity)};
+      Double_t mom = TMath::Min(track.P, mMomMax[part]);
+      
+      Double_t var[4] = {track.Eta, mom, particle->Z, double(multiplicity)};
       Double_t probMuonPID = mAccEffMuonPID[part]->GetBinContent(mAccEffMuonPID[part]->GetBin(var));
       return (gRandom->Uniform() < probMuonPID);
 

src/MIDdetector.hh

@@ -34,6 +34,8 @@ namespace o2 {
       TFile *mFileAccEffMuonPID;
       THnSparse *mAccEffMuonPID[kNPart];
       const double mEtaMax = 1.6;
+      double mMomMin[kNPart];
+      double mMomMax[kNPart];
       const char *partLabel[kNPart] = {"electron","muon","pion","kaon","proton"};
       std::map<int, int> pidmap = { {11, kElectron}, {13, kMuon}, {211, kPion}, {321, kKaon}, {2212, kProton} };