diff --git a/PWGMM/Lumi/Tasks/LumiFDDFT0.cxx b/PWGMM/Lumi/Tasks/LumiFDDFT0.cxx index 878e0482f7c..c67e36672b8 100644 --- a/PWGMM/Lumi/Tasks/LumiFDDFT0.cxx +++ b/PWGMM/Lumi/Tasks/LumiFDDFT0.cxx @@ -10,41 +10,39 @@ // or submit itself to any jurisdiction. // author: akhuntia@cern.ch -#include -#include -#include -#include - #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 +#include +#include +#include +#include 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 rowEventInfo; Produces rowEventInfofdd; Produces rowEventInfoft0; + Produces rowEventInfofv0; Service 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::iterator const& collision, aod::FDDs const& /*fdds*/, aod::FT0s const& /*ft0s*/, aod::BCsWithTimestamps const&, + void processFull(soa::Join::iterator const& collision, aod::FDDs const& /*fdds*/, aod::FT0s const& /*ft0s*/, aod::FV0As const& /*fv0s*/, aod::BCsWithTimestamps const&, o2::soa::Join 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(); + 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);