Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 10 additions & 9 deletions PWGHF/HFC/TableProducer/correlatorFlowCharmHadronsReduced.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -166,7 +166,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {

HistogramRegistry registry{"registry", {}, OutputObjHandlingPolicy::AnalysisObject};

std::shared_ptr<TObject> loadEfficiencyHistogram(const std::string& path, int histoType) {
std::shared_ptr<TObject> loadEfficiencyHistogram(const std::string& path, int histoType)
{
TObject* object = nullptr;

switch (histoType) {
Expand Down Expand Up @@ -600,8 +601,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
}

void processSECharmHadWMultMix(SameEvtPairsChHad::iterator const& pair,
aod::HfcRedTrigCharms const&,
aod::HfcRedCorrColls const&)
aod::HfcRedTrigCharms const&,
aod::HfcRedCorrColls const&)
{
BinningMultPosZ binPolicyPosZMult{{binsZPool, binsMultPool}};
auto trigCand = pair.template hfcRedTrigCharm_as<aod::HfcRedTrigCharms>();
Expand All @@ -620,8 +621,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSECharmHadWMultMixWEffInfo, "Process Same Event for Charm-Had with multiplicity pools", false);

void processSEHadHadWMultMix(SameEvtPairsHadHad::iterator const& pair,
aod::HfcRedTrigTracks const&,
aod::HfcRedCorrColls const&)
aod::HfcRedTrigTracks const&,
aod::HfcRedCorrColls const&)
{
BinningMultPosZ binPolicyPosZMult{{binsZPool, binsMultPool}};
auto trigCand = pair.template hfcRedTrigTrack_as<aod::HfcRedTrigTracks>();
Expand All @@ -630,8 +631,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSEHadHadWMultMix, "Process Same Event for Had-Had with multiplicity pools", false);

void processSECharmHadWCentMix(SameEvtPairsChHad::iterator const& pair,
aod::HfcRedTrigCharms const&,
aod::HfcRedCorrColls const&)
aod::HfcRedTrigCharms const&,
aod::HfcRedCorrColls const&)
{
BinningCentPosZ binPolicyPosZCent{{binsZPool, binsCentPool}};
auto trigCand = pair.template hfcRedTrigCharm_as<aod::HfcRedTrigCharms>();
Expand All @@ -650,8 +651,8 @@ struct HfCorrelatorFlowCharmHadronsReduced {
PROCESS_SWITCH(HfCorrelatorFlowCharmHadronsReduced, processSECharmHadWCentMixWEffInfo, "Process Same Event for Charm-Had with centrality pools", false);

void processSEHadHadWCentMix(SameEvtPairsHadHad::iterator const& pair,
aod::HfcRedTrigTracks const&,
aod::HfcRedCorrColls const&)
aod::HfcRedTrigTracks const&,
aod::HfcRedCorrColls const&)
{
BinningCentPosZ binPolicyPosZCent{{binsZPool, binsCentPool}};
auto trigCand = pair.template hfcRedTrigTrack_as<aod::HfcRedTrigTracks>();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -637,7 +637,6 @@ struct HfDerivedDataCreatorCorrelationsReduced {
}
PROCESS_SWITCH(HfDerivedDataCreatorCorrelationsReduced, processDataLcME, "Process Mixed Event for Lc candidates", false);


// CHARM-HADRON CORRELATIONS PROCESS FUNCTIONS --- MC
// Dplus MC
void processMcDplusSE(McCollsWithCentMult::iterator const& coll,
Expand Down Expand Up @@ -799,12 +798,11 @@ struct HfDerivedDataCreatorCorrelationsReduced {
}
PROCESS_SWITCH(HfDerivedDataCreatorCorrelationsReduced, processMcLcME, "Process Mixed Event for Lc candidates", false);


// HADRON-HADRON CORRELATIONS PROCESS FUNCTIONS --- DATA
// Hadron Hadron Same Event
void processDataHadronHadronSE(DataCollsWithCentMult::iterator const& coll,
TracksData const& tracks)
{
{
float cent{-1.}, mult{-1.};
if (!checkCollision(coll, cent, mult)) {
return;
Expand All @@ -817,7 +815,7 @@ struct HfDerivedDataCreatorCorrelationsReduced {
// Hadron Hadron Mixed Event
void processDataHadronHadronME(DataCollsWithCentMult::iterator const& coll,
TracksData const& tracks)
{
{
float cent{-1.}, mult{-1.};
if (!checkCollision(coll, cent, mult)) {
return;
Expand All @@ -827,12 +825,11 @@ struct HfDerivedDataCreatorCorrelationsReduced {
}
PROCESS_SWITCH(HfDerivedDataCreatorCorrelationsReduced, processDataHadronHadronME, "Process Mixed Event for hadron candidates", false);


// HADRON-HADRON CORRELATIONS PROCESS FUNCTIONS --- MC
// Hadron Hadron Same Event
void processMcHadronHadronSE(McCollsWithCentMult::iterator const& coll,
TracksMc const& tracks)
{
{
float cent{-1.}, mult{-1.};
if (!checkCollision(coll, cent, mult)) {
return;
Expand All @@ -845,7 +842,7 @@ struct HfDerivedDataCreatorCorrelationsReduced {
// Hadron Hadron Mixed Event
void processMcHadronHadronME(McCollsWithCentMult::iterator const& coll,
TracksMc const& tracks)
{
{
float cent{-1.}, mult{-1.};
if (!checkCollision(coll, cent, mult)) {
return;
Expand Down
19 changes: 8 additions & 11 deletions PWGHF/HFC/Tasks/taskCorrelationsEfficiencies.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@
#include "PWGHF/DataModel/CandidateReconstructionTables.h"
#include "PWGHF/DataModel/CandidateSelectionTables.h"
#include "PWGHF/HFC/DataModel/DerivedDataCorrelationTables.h"
#include "PWGHF/Utils/utilsEvSelHf.h"
#include "PWGHF/HFC/Utils/utilsFlow.h"
#include "PWGHF/Utils/utilsEvSelHf.h"

#include "Common/Core/RecoDecay.h"
#include "Common/DataModel/Centrality.h"
Expand Down Expand Up @@ -63,13 +63,13 @@ namespace o2::aod
namespace hf_correlations_eff
{
// Charm-hadron candidate columns
DECLARE_SOA_COLUMN(Pt, pt, float); //! Transverse momentum of candidate (GeV/c)
DECLARE_SOA_COLUMN(PvContrib, pvContrib, int); //! Number of primary vertex contributors
DECLARE_SOA_COLUMN(Eta, eta, float); //! Pseudorapidity of candidate
DECLARE_SOA_COLUMN(Phi, phi, float); //! Azimuthal angle of candidate
DECLARE_SOA_COLUMN(Origin, origin, int); //! Origin of the candidate (prompt or non-prompt)
DECLARE_SOA_COLUMN(MlScore0, mlScore0, float); //! ML score of the first configured index
DECLARE_SOA_COLUMN(MlScore1, mlScore1, float); //! ML score of the second configured index
DECLARE_SOA_COLUMN(Pt, pt, float); //! Transverse momentum of candidate (GeV/c)
DECLARE_SOA_COLUMN(PvContrib, pvContrib, int); //! Number of primary vertex contributors
DECLARE_SOA_COLUMN(Eta, eta, float); //! Pseudorapidity of candidate
DECLARE_SOA_COLUMN(Phi, phi, float); //! Azimuthal angle of candidate
DECLARE_SOA_COLUMN(Origin, origin, int); //! Origin of the candidate (prompt or non-prompt)
DECLARE_SOA_COLUMN(MlScore0, mlScore0, float); //! ML score of the first configured index
DECLARE_SOA_COLUMN(MlScore1, mlScore1, float); //! ML score of the second configured index

// Associated track columns
DECLARE_SOA_COLUMN(PtAssoc, ptAssoc, float); //! Transverse momentum of associated track (GeV/c)
Expand Down Expand Up @@ -219,7 +219,6 @@ struct HfTaskCorrelationsEfficiencies {
std::vector<AxisSpec> axesAssocTracks = {axisPtAssoc, axisEtaAssoc, axisPhiAssoc, axisZVtx, axisAssTrkSpecie, axisTrackOrigin};
registry.add("hAssTrackReco", "Reconstructed associated tracks;;", {HistType::kTHnSparseF}, axesAssocTracks);
registry.add("hAssTrackGen", "Generated associated tracks;;", {HistType::kTHnSparseF}, axesAssocTracks);

}

/// Match charm-hadron based on its reconstruction flag
Expand Down Expand Up @@ -428,10 +427,8 @@ struct HfTaskCorrelationsEfficiencies {
auto genCollision = genTrk.template mcCollision_as<aod::McCollisions>();
registry.fill(HIST("hAssTrackGen"), genTrk.pt(), genTrk.eta(), genTrk.phi(), genCollision.posZ(), matchedSpecie, genTrkOrigin);
}

}
PROCESS_SWITCH(HfTaskCorrelationsEfficiencies, processEffAssTracks, "Process efficiencies for associated tracks", true);

};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down
Loading