[PWGLF] [PWGMM] FV0 info added to Lumi task (#12101)

Co-authored-by: Arvind Khuntia <arvind.khuntia@cern.ch>
Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: 3 people

PWGMM/Lumi/Tasks/LumiFDDFT0.cxx

@@ -10,41 +10,39 @@
 // or submit itself to any jurisdiction.
 // author: akhuntia@cern.ch
 
-#include <array>
-#include <cmath>
-#include <iostream>
-#include <vector>
-
 #include "Common/Core/TrackSelection.h"
 #include "Common/Core/trackUtilities.h"
 #include "Common/DataModel/EventSelection.h"
-#include "Common/DataModel/TrackSelectionTables.h"
 #include "Common/DataModel/Multiplicity.h"
-#include "CommonUtils/NameConf.h"
+#include "Common/DataModel/TrackSelectionTables.h"
 
+#include "CCDB/BasicCCDBManager.h"
+#include "CCDB/CcdbApi.h"
+#include "CommonConstants/GeomConstants.h"
+#include "CommonUtils/NameConf.h"
+#include "DataFormatsCalibration/MeanVertexObject.h"
+#include "DataFormatsFDD/Digit.h"
+#include "DataFormatsFIT/Triggers.h"
+#include "DataFormatsParameters/GRPMagField.h"
+#include "DataFormatsParameters/GRPObject.h"
+#include "DetectorsBase/GeometryManager.h"
+#include "DetectorsBase/Propagator.h"
+#include "DetectorsVertexing/PVertexer.h"
 #include "Framework/ASoAHelpers.h"
+#include "Framework/AnalysisDataModel.h"
 #include "Framework/AnalysisTask.h"
 #include "Framework/HistogramRegistry.h"
 #include "Framework/RunningWorkflowInfo.h"
 #include "Framework/runDataProcessing.h"
-#include "Framework/AnalysisDataModel.h"
-
-#include "DetectorsVertexing/PVertexer.h"
 #include "ReconstructionDataFormats/DCA.h"
 #include "ReconstructionDataFormats/PrimaryVertex.h"
 #include "ReconstructionDataFormats/Vertex.h"
 
-#include "DataFormatsFDD/Digit.h"
-#include "DataFormatsFIT/Triggers.h"
-#include "DataFormatsParameters/GRPObject.h"
-#include "DataFormatsParameters/GRPMagField.h"
-
-#include "DetectorsBase/GeometryManager.h"
-#include "DetectorsBase/Propagator.h"
-#include "CommonConstants/GeomConstants.h"
-#include "CCDB/BasicCCDBManager.h"
-#include "CCDB/CcdbApi.h"
-#include "DataFormatsCalibration/MeanVertexObject.h"
+#include <array>
+#include <cmath>
+#include <iostream>
+#include <map>
+#include <vector>
 
 using namespace o2;
 using namespace o2::framework;
@@ -85,6 +83,12 @@ DECLARE_SOA_COLUMN(TimeCFT0, timeCft0, double);
 DECLARE_SOA_COLUMN(ChargeAFT0, chargeAft0, double);
 DECLARE_SOA_COLUMN(ChargeCFT0, chargeCft0, double);
 
+// information for FV0
+DECLARE_SOA_COLUMN(isFV0, isfv0, bool);
+DECLARE_SOA_COLUMN(TCMTriggerFV0, tcmTriggerfv0, uint8_t);
+DECLARE_SOA_COLUMN(TimeAFV0, timeAfv0, double);     // Only FV0-A time
+DECLARE_SOA_COLUMN(ChargeAFV0, chargeAfv0, double); // Only FV0-A charge
+
 } // namespace full
 DECLARE_SOA_TABLE(EventInfo, "AOD", "EventInfo", full::TimeStamp, full::VertexX,
                   full::VertexY, full::VertexZ, full::GlobalBC,
@@ -94,7 +98,8 @@ DECLARE_SOA_TABLE(EventInfo, "AOD", "EventInfo", full::TimeStamp, full::VertexX,
                   full::ChargeAFDD, full::ChargeCFDD,
                   full::isFT0, full::TCMTriggerFT0,
                   full::TimeAFT0, full::TimeCFT0,
-                  full::ChargeAFT0, full::ChargeCFT0);
+                  full::ChargeAFT0, full::ChargeCFT0, full::isFV0,
+                  full::TCMTriggerFV0, full::TimeAFV0, full::ChargeAFV0);
 
 DECLARE_SOA_TABLE(EventInfoFDD, "AOD", "EventInfoFDD",
                   full::TimeStamp, full::GlobalBC,
@@ -109,12 +114,18 @@ DECLARE_SOA_TABLE(EventInfoFT0, "AOD", "EventInfoFT0",
                   full::TimeCFT0, full::ChargeAFT0,
                   full::ChargeCFT0);
 
+DECLARE_SOA_TABLE(EventInfoFV0, "AOD", "EventInfoFV0",
+                  full::TimeStamp, full::GlobalBC,
+                  full::TCMTriggerFV0, full::TimeAFV0,
+                  full::ChargeAFV0);
+
 } // namespace o2::aod
 
 struct LumiFDDFT0 {
   Produces<o2::aod::EventInfo> rowEventInfo;
   Produces<o2::aod::EventInfoFDD> rowEventInfofdd;
   Produces<o2::aod::EventInfoFT0> rowEventInfoft0;
+  Produces<o2::aod::EventInfoFV0> rowEventInfofv0;
   Service<o2::ccdb::BasicCCDBManager> ccdb;
   const char* ccdbpath_grp = "GLO/Config/GRPMagField";
   const char* ccdburl = "http://alice-ccdb.cern.ch";
@@ -147,10 +158,10 @@ struct LumiFDDFT0 {
     {
       {"BCFDD", "", {HistType::kTH1F, {{nBCsPerOrbit + 1, -0.5f, nBCsPerOrbit + 0.5f, "x"}}}}, //
       {"BCFT0", "", {HistType::kTH1F, {{nBCsPerOrbit + 1, -0.5f, nBCsPerOrbit + 0.5f, "x"}}}}, //
+      {"BCFV0", "", {HistType::kTH1F, {{nBCsPerOrbit + 1, -0.5f, nBCsPerOrbit + 0.5f, "x"}}}}, //
     }};
 
   bool doPVrefit = true;
-
   void init(InitContext&)
   {
     if (doprocessLite == true && doprocessFull == true) {
@@ -165,7 +176,7 @@ struct LumiFDDFT0 {
     mRunNumber = 0;
   }
 
-  void processFull(soa::Join<aod::Collisions, aod::EvSels>::iterator const& collision, aod::FDDs const& /*fdds*/, aod::FT0s const& /*ft0s*/, aod::BCsWithTimestamps const&,
+  void processFull(soa::Join<aod::Collisions, aod::EvSels>::iterator const& collision, aod::FDDs const& /*fdds*/, aod::FT0s const& /*ft0s*/, aod::FV0As const& /*fv0s*/, aod::BCsWithTimestamps const&,
                    o2::soa::Join<o2::aod::Tracks, o2::aod::TracksCov,
                                  o2::aod::TracksExtra> const& unfiltered_tracks)
   {
@@ -239,6 +250,10 @@ struct LumiFDDFT0 {
     double chargecFT0 = 0.;
     uint8_t mTriggerFT0 = 0;
 
+    double timeaFV0 = -999.;
+    double chargeaFV0 = 0.;
+    uint8_t mTriggerFV0 = 0;
+
     if (doPVrefit && PVrefit_doable) {
       auto Pvtx_refitted = vertexer.refitVertex(vec_useTrk_PVrefit, Pvtx);
       chi2 = Pvtx_refitted.getChi2();
@@ -249,38 +264,49 @@ struct LumiFDDFT0 {
       // refitYY = Pvtx_refitted.getSigmaY2();
       // refitXY = Pvtx_refitted.getSigmaXY();
 
-      // now get information for FDD
-      if (collision.has_foundFDD()) {
-        auto fdd = collision.foundFDD();
-        mTriggerFDD = fdd.triggerMask();
-        timeaFDD = fdd.timeA();
-        timecFDD = fdd.timeC();
-        for (auto amplitude : fdd.chargeA()) {
-          chargeaFDD += amplitude;
-        }
-        for (auto amplitude : fdd.chargeC()) {
-          chargecFDD += amplitude;
-        }
-      } // fdd
-
-      if (collision.has_foundFT0()) {
-        auto ft0 = collision.foundFT0();
-        mTriggerFT0 = ft0.triggerMask();
-        timeaFT0 = ft0.timeA();
-        timecFT0 = ft0.timeC();
-        for (auto amplitude : ft0.amplitudeA()) {
-          chargeaFT0 += amplitude;
-        }
+    } // pv refit
 
-        for (auto amplitude : ft0.amplitudeC()) {
-          chargecFT0 += amplitude;
-        }
-      } // ft0
+    // now get information for FDD
+    if (collision.has_foundFDD()) {
+      auto fdd = collision.foundFDD();
+      mTriggerFDD = fdd.triggerMask();
+      timeaFDD = fdd.timeA();
+      timecFDD = fdd.timeC();
+      for (auto amplitude : fdd.chargeA()) {
+        chargeaFDD += amplitude;
+      }
+      for (auto amplitude : fdd.chargeC()) {
+        chargecFDD += amplitude;
+      }
+    } // fdd
+
+    if (collision.has_foundFT0()) {
+      auto ft0 = collision.foundFT0();
+      mTriggerFT0 = ft0.triggerMask();
+      timeaFT0 = ft0.timeA();
+      timecFT0 = ft0.timeC();
+      for (auto amplitude : ft0.amplitudeA()) {
+        chargeaFT0 += amplitude;
+      }
+
+      for (auto amplitude : ft0.amplitudeC()) {
+        chargecFT0 += amplitude;
+      }
+    } // ft0
+
+    // FV0
+    if (collision.has_foundFV0()) {
+      auto fv0 = collision.foundFV0();
+      mTriggerFV0 = fv0.triggerMask();
+      timeaFV0 = fv0.time();
+      for (auto amplitude : fv0.amplitude()) {
+        chargeaFV0 += amplitude;
+      }
+    } // fv0
 
-    } // pv refit
     rowEventInfo(relTS, refitX, refitY, refitZ, globalBC, chi2, nContrib, collision.has_foundFDD(),
                  mTriggerFDD, timeaFDD, timecFDD, chargeaFDD, chargecFDD, collision.has_foundFT0(), mTriggerFT0, timeaFT0,
-                 timecFT0, chargeaFT0, chargecFT0);
+                 timecFT0, chargeaFT0, chargecFT0, collision.has_foundFV0(), mTriggerFV0, timeaFV0, chargeaFV0);
 
     histos.fill(HIST("chisquare_Refitted"), chi2);
     if (nContrib > nContribMin && nContrib < nContribMax &&
@@ -313,7 +339,7 @@ struct LumiFDDFT0 {
   };
   PROCESS_SWITCH(LumiFDDFT0, processFull, "Process FDD", true);
 
-  void processLite(aod::FDDs const& fdds, aod::FT0s const& ft0s, aod::BCsWithTimestamps const&)
+  void processLite(aod::FDDs const& fdds, aod::FT0s const& ft0s, aod::FV0As const& fv0s, aod::BCsWithTimestamps const&)
   {
 
     // Scan over the FDD table and store charge and time along with globalBC
@@ -384,6 +410,23 @@ struct LumiFDDFT0 {
       }
       rowEventInfoft0(relTS, globalBC, ft0.triggerMask(), ft0.timeA(), ft0.timeC(), chargeaFT0, chargecFT0);
     } // end of ft0 table
+
+    // Scan over the FV0 table and store charge and time along with globalBC
+    for (auto& fv0 : fv0s) {
+      auto bc = fv0.bc_as<BCsWithTimestamps>();
+      if (!bc.timestamp())
+        continue;
+      Long64_t relTS = bc.timestamp() - fttimestamp;
+      Long64_t globalBC = bc.globalBC();
+      int localBC = globalBC % nBCsPerOrbit;
+      histoslite.fill(HIST("BCFV0"), localBC);
+
+      double chargeaFV0 = 0.;
+      for (auto amplitude : fv0.amplitude()) {
+        chargeaFV0 += amplitude;
+      }
+      rowEventInfofv0(relTS, globalBC, fv0.triggerMask(), fv0.time(), chargeaFV0);
+    } // end of fv0 table
   };
   PROCESS_SWITCH(LumiFDDFT0, processLite, "Process FDD and FT0 info", false);