start adding FlatTrackSmearer

Author: aalkin

ALICE3/Core/FlatTrackSmearer.cxx

@@ -0,0 +1,66 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#include "FlatTrackSmearer.h"
+
+namespace o2::delphes {
+int TrackSmearer::getIndexPDG(int pdg) const
+{
+  switch (std::abs(pdg)) {
+    case 11:
+      return 0; // Electron
+    case 13:
+      return 1; // Muon
+    case 211:
+      return 2; // Pion
+    case 321:
+      return 3; // Kaon
+    case 2212:
+      return 4; // Proton
+    case 1000010020:
+      return 5; // Deuteron
+    case 1000010030:
+      return 6; // Triton
+    case 1000020030:
+      return 7; // Helium3
+    case 1000020040:
+      return 8; // Alphas
+    default:
+      return 2; // Default: pion
+  }
+}
+
+const char* TrackSmearer::getParticleName(int pdg) const
+{
+  switch (std::abs(pdg)) {
+    case 11:
+      return "electron";
+    case 13:
+      return "muon";
+    case 211:
+      return "pion";
+    case 321:
+      return "kaon";
+    case 2212:
+      return "proton";
+    case 1000010020:
+      return "deuteron";
+    case 1000010030:
+      return "triton";
+    case 1000020030:
+      return "helium3";
+    case 1000020040:
+      return "alpha";
+    default:
+      return "pion"; // Default: pion
+  }
+}
+} // namespace o2::delphes

ALICE3/Core/FlatTrackSmearer.h

@@ -0,0 +1,76 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#ifndef ALICE3_CORE_FLATTRACKSMEARER_H_
+#define ALICE3_CORE_FLATTRACKSMEARER_H_
+
+#include "FlatLutEntry.h"
+
+#include <CCDB/BasicCCDBManager.h>
+#include <ReconstructionDataFormats/Track.h>
+
+namespace o2::delphes {
+/**
+ * @brief Track smearing with flat LUT backend
+ */
+using O2Track = o2::track::TrackParCov;
+class TrackSmearer
+{
+ public:
+  TrackSmearer() = default;
+  ~TrackSmearer() = default;
+
+  /** LUT methods **/
+  bool loadTable(int pdg, const char* filename, bool forceReload = false);
+  bool hasTable(int pdg) const;
+
+  void useEfficiency(bool val) { mUseEfficiency = val; }
+  void interpolateEfficiency(bool val) { mInterpolateEfficiency = val; }
+  void skipUnreconstructed(bool val) { mSkipUnreconstructed = val; }
+  void setWhatEfficiency(int val) { mWhatEfficiency = val; }
+
+  const lutHeader_t* getLUTHeader(int pdg) const;
+  const lutEntry_t* getLUTEntry(int pdg, float nch, float radius, float eta, float pt, float& interpolatedEff) const;
+
+  bool smearTrack(O2Track& o2track, const lutEntry_t* lutEntry, float interpolatedEff);
+  bool smearTrack(O2Track& o2track, int pdg, float nch);
+
+  double getPtRes(int pdg, float nch, float eta, float pt) const;
+  double getEtaRes(int pdg, float nch, float eta, float pt) const;
+  double getAbsPtRes(int pdg, float nch, float eta, float pt) const;
+  double getAbsEtaRes(int pdg, float nch, float eta, float pt) const;
+  double getEfficiency(int pdg, float nch, float eta, float pt) const;
+
+  int getIndexPDG(int pdg) const;
+
+  const char* getParticleName(int pdg) const;
+
+  void setdNdEta(float val) { mdNdEta = val; }
+  void setCcdbManager(o2::ccdb::BasicCCDBManager* mgr) { mCcdbManager = mgr; }
+
+ protected:
+  static constexpr unsigned int nLUTs = 9; // Number of LUT available
+  lutHeader_t const* mHeaders[nLUTs]; // header references for quick access
+  FlatLutData mLUTData[nLUTs];  // NEW: Flat data storage
+
+  bool mUseEfficiency = true;
+  bool mInterpolateEfficiency = false;
+  bool mSkipUnreconstructed = true; // don't smear tracks that are not reco'ed
+  int mWhatEfficiency = 1;
+  float mdNdEta = 1600.f;
+
+ private:
+  o2::ccdb::BasicCCDBManager* mCcdbManager = nullptr;
+};
+
+}
+
+#endif // ALICE3_CORE_FLATTRACKSMEARER_H_