From 238d7e4d7b6b9a63efa745b0dba9e8211df1561d Mon Sep 17 00:00:00 2001 From: ALICE Action Bot Date: Tue, 28 Jan 2025 00:30:46 +0000 Subject: [PATCH] Please consider the following formatting changes --- .../femtoUniverseProducerTask.cxx | 18 ++-- .../Tasks/femtoUniversePairTaskTrackD0.cxx | 96 ++++++++++++------- 2 files changed, 73 insertions(+), 41 deletions(-) diff --git a/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx b/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx index 605fa2c7a94..04bf3d325c2 100644 --- a/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx +++ b/PWGCF/FemtoUniverse/TableProducer/femtoUniverseProducerTask.cxx @@ -80,7 +80,7 @@ using FemtoFullTracks = aod::pidTPCMu, aod::pidTPCPi, aod::pidTPCKa, aod::pidTPCPr, aod::pidTPCDe, aod::pidTOFEl, aod::pidTOFMu, aod::pidTOFPi, aod::pidTOFKa, aod::pidTOFPr, aod::pidTOFDe>; -using SelectedCandidatesDataMl = +using SelectedCandidatesDataMl = soa::Filtered>; // using FilteredFullV0s = soa::Filtered; /// predefined Join @@ -1355,7 +1355,7 @@ struct FemtoUniverseProducerTask { double invMassD0bar = 0.0; bool isD0D0bar = false; std::vector outputMlD0D0bar = {-1., -1., -1.}; // this vector keeps the probabilities from the ML model for D0/D0bar - uint8_t daughFlag = 0; // flag = 0 (daugh of D0 or D0bar), 1 (daug of D0), -1 (daugh of D0bar) + uint8_t daughFlag = 0; // flag = 0 (daugh of D0 or D0bar), 1 (daug of D0), -1 (daugh of D0bar) for (const auto& hfCand : hfCands) { @@ -1383,7 +1383,7 @@ struct FemtoUniverseProducerTask { if (hfCand.isSelD0() == 1 && hfCand.isSelD0bar() == 0) { invMassD0 = hfHelper.invMassD0ToPiK(hfCand); invMassD0bar = -hfHelper.invMassD0barToKPi(hfCand); - if(ConfD0Selection.applyMLSelD0D0bar) { + if (ConfD0Selection.applyMLSelD0D0bar) { for (unsigned int iClass = 0; iClass < ConfD0Selection.classMlD0D0bar->size(); iClass++) { outputMlD0D0bar[iClass] = hfCand.mlProbD0()[ConfD0Selection.classMlD0D0bar->at(iClass)]; } @@ -1393,7 +1393,7 @@ struct FemtoUniverseProducerTask { } else if (hfCand.isSelD0() == 0 && hfCand.isSelD0bar() == 1) { invMassD0 = -hfHelper.invMassD0ToPiK(hfCand); invMassD0bar = hfHelper.invMassD0barToKPi(hfCand); - if(ConfD0Selection.applyMLSelD0D0bar) { + if (ConfD0Selection.applyMLSelD0D0bar) { for (unsigned int iClass = 0; iClass < ConfD0Selection.classMlD0D0bar->size(); iClass++) { outputMlD0D0bar[iClass] = hfCand.mlProbD0bar()[ConfD0Selection.classMlD0D0bar->at(iClass)]; } @@ -1461,8 +1461,8 @@ struct FemtoUniverseProducerTask { hfCand.eta(), hfCand.phi(), aod::femtouniverseparticle::ParticleType::kD0, - -999, // cut, CutContainerType - -999, // PID, CutContainerType + -999, // cut, CutContainerType + -999, // PID, CutContainerType outputMlD0D0bar[0], // saving only the probaility for store class 1 - background indexChildID, invMassD0, // D0 mass (mLambda) @@ -1939,9 +1939,9 @@ struct FemtoUniverseProducerTask { "Provide experimental data for track D0 meson", false); void processTrackD0DataML(aod::FemtoFullCollision const& col, - aod::BCsWithTimestamps const&, - soa::Filtered const& tracks, - aod::SelectedCandidatesDataMl const& candidates) + aod::BCsWithTimestamps const&, + soa::Filtered const& tracks, + aod::SelectedCandidatesDataMl const& candidates) { // get magnetic field for run getMagneticFieldTesla(col.bc_as()); diff --git a/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx b/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx index 916ff30cc09..2eb3cd6eb74 100644 --- a/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx +++ b/PWGCF/FemtoUniverse/Tasks/femtoUniversePairTaskTrackD0.cxx @@ -450,41 +450,73 @@ struct FemtoUniversePairTaskTrackD0 { for (auto const& charmCand : groupD0D0barCands) { // D0 candidates if (charmCand.mLambda() > 0.0f && charmCand.mAntiLambda() < 0.0f) { - if (charmCand.tempFitVar() < 0.01) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt001"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.015) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0015"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.02) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt002"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.025) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0025"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.03) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt003"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.04) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt004"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.05) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt005"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.06) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt006"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.07) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt007"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.08) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt008"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.09) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt009"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.1) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt01"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.15) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt015"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.2) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt02"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.25) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt025"), charmCand.mLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.3) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt03"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.01) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt001"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.015) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0015"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.02) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt002"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.025) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0025"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.03) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt003"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.04) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt004"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.05) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt005"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.06) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt006"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.07) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt007"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.08) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt008"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.09) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt009"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.1) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt01"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.15) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt015"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.2) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt02"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.25) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt025"), charmCand.mLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.3) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt03"), charmCand.mLambda(), charmCand.pt()); } // DObar candidates if (charmCand.mLambda() < 0.0f && charmCand.mAntiLambda() > 0.0f) { - if (charmCand.tempFitVar() < 0.01) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt001"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.015) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0015"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.02) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt002"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.025) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0025"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.03) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt003"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.04) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt004"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.05) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt005"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.06) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt006"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.07) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt007"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.08) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt008"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.09) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt009"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.1) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt01"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.15) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt015"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.2) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt02"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.25) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt025"), charmCand.mAntiLambda(), charmCand.pt()); - if (charmCand.tempFitVar() < 0.3) registry.fill(HIST("D0D0bar_MLSel/hMassVsPt03"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.01) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt001"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.015) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0015"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.02) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt002"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.025) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt0025"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.03) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt003"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.04) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt004"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.05) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt005"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.06) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt006"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.07) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt007"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.08) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt008"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.09) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt009"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.1) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt01"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.15) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt015"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.2) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt02"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.25) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt025"), charmCand.mAntiLambda(), charmCand.pt()); + if (charmCand.tempFitVar() < 0.3) + registry.fill(HIST("D0D0bar_MLSel/hMassVsPt03"), charmCand.mAntiLambda(), charmCand.pt()); } } }