Add hitset, module row and column to NtupleDumper

Added the module row and column and a 24 size bitset, where each bit corresponds to a side of a layer and is 1 if the track has a hit on this layer and 0 otherwise. The bitset is then cast to an unsigned long, so that we only have to write out a single variable.

PhysicsAnalysis/NtupleDumper/src/NtupleDumperAlg.cxx

@@ -225,6 +225,9 @@ StatusCode NtupleDumperAlg::initialize()
   m_tree->Branch("longTracks", &m_longTracks, "longTracks/I");
   m_tree->Branch("Track_Chi2", &m_Chi2);
   m_tree->Branch("Track_nDoF", &m_DoF);
+  m_tree->Branch("Track_eta", &m_eta);
+  m_tree->Branch("Track_hitSet", &m_hitSet);
+  m_tree->Branch("Track_phi", &m_phi);
   m_tree->Branch("Track_x0", &m_xup);
   m_tree->Branch("Track_y0", &m_yup);
   m_tree->Branch("Track_z0", &m_zup);
@@ -912,16 +915,20 @@ StatusCode NtupleDumperAlg::execute(const EventContext &ctx) const
     std::set<std::pair<int, int>> layerMap;
     std::set<int> stationMap;
     // Check for hit in the three downstream stations
+    HitSet hitSet {24};
     for (auto measurement : *(track->measurementsOnTrack())) {
         const Tracker::FaserSCT_ClusterOnTrack* cluster = dynamic_cast<const Tracker::FaserSCT_ClusterOnTrack*>(measurement);
         if (cluster != nullptr) {
             Identifier id = cluster->identify();
             int station = m_sctHelper->station(id);
             int layer = m_sctHelper->layer(id);
+            int side = m_sctHelper->side(id);
             stationMap.emplace(station);
             layerMap.emplace(station, layer);
+            hitSet.set(23 - (station * 6 + layer * 2 + side));
         }
     }
+    m_hitSet.push_back(hitSet.to_ulong());
     if (stationMap.count(1) == 0 || stationMap.count(2) == 0 || stationMap.count(3) == 0) continue;
 
     const Trk::TrackParameters* upstreamParameters = track->trackParameters()->front();
@@ -933,6 +940,12 @@ StatusCode NtupleDumperAlg::execute(const EventContext &ctx) const
 
     m_Chi2.push_back(track->fitQuality()->chiSquared());
     m_DoF.push_back(track->fitQuality()->numberDoF());
+    const Trk::MeasurementBase *measurement = track->measurementsOnTrack()->front();
+    const Tracker::FaserSCT_ClusterOnTrack* cluster =
+      dynamic_cast<const Tracker::FaserSCT_ClusterOnTrack*>(measurement);
+    Identifier id = cluster->identify();
+    m_eta.push_back(m_sctHelper->eta_module(id));
+    m_phi.push_back(m_sctHelper->phi_module(id));
 
     m_nHit0.push_back(stationMap.count(0));
     m_nHit1.push_back(stationMap.count(1));
@@ -1418,6 +1431,9 @@ NtupleDumperAlg::clearTree() const
 
   m_Chi2.clear();
   m_DoF.clear();
+  m_eta.clear();
+  m_phi.clear();
+  m_hitSet.clear();
   m_charge.clear();
   m_nLayers.clear();
   m_longTracks = 0;