Add parametrised response for preshower detector

Author: marcovanleeuwen

src/CMakeLists.txt

@@ -7,6 +7,7 @@ set(SOURCES
   TOFLayer.cc
   RICHdetector.cc
   MIDdetector.cc
+  PreShower.cc
   )
 
 set(HEADERS
@@ -16,6 +17,7 @@ set(HEADERS
   TOFLayer.hh
   RICHdetector.hh
   MIDdetector.hh 
+  PreShower.hh 
   )
 
 get_target_property(DELPHES_INCLUDE_DIRECTORIES

src/PreShower.cc

@@ -0,0 +1,69 @@
+/// @author: Roberto Preghenella
+/// @email: [email protected]
+
+/// @author: Antonio Uras
+/// @email: [email protected]
+
+/// @author: Marco van Leeuwen
+/// @email: [email protected]
+
+#include "PreShower.hh"
+#include "TDatabasePDG.h"
+#include "THnSparse.h"
+#include "TRandom.h"
+#include "TDatime.h"
+#include "TFile.h"
+#include "TVector3.h"
+#include "TMath.h"
+#include "TAxis.h"
+
+namespace o2 {
+  namespace delphes {
+    
+    //==========================================================================================================
+    
+    bool PreShower::setup() {
+
+      TDatime t;
+      gRandom->SetSeed(t.GetDate()+t.GetYear()*t.GetHour()*t.GetMinute()*t.GetSecond()); // NB: gRandom is a global pointer ?
+      for (Int_t iPart = 0; iPart < kNPart; iPart++) {
+	      mMomMin[iPart] = 0.1;
+	      mMomMax[iPart] = 20;
+      }
+
+      return kTRUE;
+	
+    }
+
+    //==========================================================================================================
+
+    bool PreShower::hasPreShower(const Track &track) {
+
+      auto pdg  = std::abs(track.PID);
+      auto part = pidmap[pdg];
+      return ((TMath::Abs(track.Eta) < mEtaMax) && (track.P > mMomMin[part]));
+
+    }
+
+    //==========================================================================================================
+
+    bool PreShower::isElectron(const Track &track, int multiplicity=1) {
+
+      auto pdg  = std::abs(track.PID);
+      auto part = pidmap[pdg];
+      if (part == kElectron) {
+			// Parametrisation of preshower detector studies without charge sharing
+         float eff = 0.8*(1.-exp(1.6*(track.P-0.05)));
+         return (gRandom->Uniform() < eff);
+      }
+      else {
+         const Float_t misTagProb = 0.001;
+         return (gRandom->Uniform() < misTagProb);
+      }
+    }
+
+    //==========================================================================================================
+   
+  } /** namespace delphes **/
+
+} /** namespace o2 **/

src/PreShower.hh

@@ -0,0 +1,49 @@
+/// @author: Roberto Preghenella
+/// @email: [email protected]
+
+/// @author: Antonio Uras
+/// @email: [email protected]
+
+/// @author: Marco van Leeuwen
+/// @email: [email protected]
+
+#ifndef _DelphesO2_PreShower_h_
+#define _DelphesO2_PreShower_h_
+
+#include "classes/DelphesClasses.h"
+#include "THnSparse.h"
+#include "TFile.h"
+
+#include <map>
+using namespace std;
+
+namespace o2 {
+  namespace delphes {
+
+    class PreShower {
+  
+    public:
+      PreShower() = default;
+      ~PreShower() = default;
+  
+      enum { kElectron, kMuon, kPion, kKaon, kProton, kNPart }; // primary particles with a non-zero muon PID probability
+      
+      bool setup();
+      bool hasPreShower(const Track &track);
+      bool isElectron(const Track &track, int multiplicity);
+
+    protected:
+
+      const double mEtaMax = 1.75;
+      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} };
+  
+    };
+  
+  } /** namespace delphes **/
+} /** namespace o2 **/
+
+#endif /** _DelphesO2_MIDLayer_h_ **/
+