Update DelphesO2LutWriter.h

Author: njacazio

ALICE3/Core/DelphesO2LutWriter.h

@@ -24,8 +24,11 @@
 
 #include "ALICE3/Core/DelphesO2TrackSmearer.h"
 #include "ALICE3/Core/FastTracker.h"
+
 #include "TGraph.h"
 
+#include <string>
+
 namespace o2::fastsim
 {
 class DelphesO2LutWriter
@@ -34,34 +37,56 @@ class DelphesO2LutWriter
   DelphesO2LutWriter() = default;
   virtual ~DelphesO2LutWriter() = default;
 
-  o2::fastsim::FastTracker fat;
-  void diagonalise(lutEntry_t& lutEntry);
-  float etaMaxBarrel = 1.75f;
-  bool usePara = true;        // use fwd parameterisation
-  bool useDipole = false;     // use dipole i.e. flat parametrization for efficiency and momentum resolution
-  bool useFlatDipole = false; // use dipole i.e. flat parametrization outside of the barrel
-
-  int mAtLeastHits = 4;
-  int mAtLeastCorr = 4;
-  int mAtLeastFake = 0;
+  // Setters
+  void setBinningNch(bool log, int nbins, float min, float max) { mNchBinning = {log, nbins, min, max}; }
+  void setBinningRadius(bool log, int nbins, float min, float max) { mRadiusBinning = {log, nbins, min, max}; }
+  void setBinningEta(bool log, int nbins, float min, float max) { mEtaBinning = {log, nbins, min, max}; }
+  void setBinningPt(bool log, int nbins, float min, float max) { mPtBinning = {log, nbins, min, max}; }
+  void setEtaMaxBarrel(float eta) { etaMaxBarrel = eta; }
   void SetAtLeastHits(int n) { mAtLeastHits = n; }
   void SetAtLeastCorr(int n) { mAtLeastCorr = n; }
   void SetAtLeastFake(int n) { mAtLeastFake = n; }
-
-  void printLutWriterConfiguration();
   bool fatSolve(lutEntry_t& lutEntry,
                 float pt = 0.1,
                 float eta = 0.0,
                 const float mass = 0.13957000,
-                int itof = 0,
-                int otof = 0,
+                size_t itof = 0,
+                size_t otof = 0,
                 int q = 1,
                 const float nch = 1);
+
+  void Print() const;
   bool fwdSolve(float* covm, float pt = 0.1, float eta = 0.0, float mass = 0.13957000);
   bool fwdPara(lutEntry_t& lutEntry, float pt = 0.1, float eta = 0.0, float mass = 0.13957000, float Bfield = 0.5);
-  void lutWrite(const char* filename = "lutCovm.dat", int pdg = 211, float field = 0.2, int itof = 0, int otof = 0);
+  void lutWrite(const char* filename = "lutCovm.dat", int pdg = 211, float field = 0.2, size_t itof = 0, size_t otof = 0);
   TGraph* lutRead(const char* filename, int pdg, int what, int vs, float nch = 0., float radius = 0., float eta = 0., float pt = 0.);
 
+  o2::fastsim::FastTracker fat;
+
+ private:
+  void diagonalise(lutEntry_t& lutEntry);
+  float etaMaxBarrel = 1.75f;
+  bool usePara = true;        // use fwd parameterisation
+  bool useDipole = false;     // use dipole i.e. flat parametrization for efficiency and momentum resolution
+  bool useFlatDipole = false; // use dipole i.e. flat parametrization outside of the barrel
+
+  int mAtLeastHits = 4;
+  int mAtLeastCorr = 4;
+  int mAtLeastFake = 0;
+
+  // Binning of the LUT to make
+  struct LutBinning {
+    bool log;
+    int nbins;
+    float min;
+    float max;
+    std::string toString() const;
+  };
+  LutBinning mNchBinning = {true, 20, 0.5f, 3.5f};
+  LutBinning mRadiusBinning = {false, 1, 0.0f, 100.0f};
+  LutBinning mEtaBinning = {false, 80, -4.0f, 4.0f};
+  LutBinning mPtBinning = {true, 200, -2.0f, 2.0f};
+
   ClassDef(DelphesO2LutWriter, 1);
 };
 } // namespace o2::fastsim