1818// / profile and/or jet momentum fraction for charmed hadrons
1919
2020#include " PWGJE/Core/JetDerivedDataUtilities.h"
21- #include " PWGJE/Core/JetUtilities.h"
2221#include " PWGJE/Core/JetHFUtilities.h"
22+ #include " PWGJE/Core/JetUtilities.h"
2323#include " PWGJE/DataModel/Jet.h"
2424#include " PWGJE/DataModel/JetReducedData.h"
25- //
25+
2626#include " PWGHF/Core/DecayChannels.h"
2727
2828#include " Common/Core/RecoDecay.h"
@@ -292,15 +292,17 @@ struct HfFragmentationFunction {
292292 void processD0DataCharged (aod::JetCollision const & collision,
293293 soa::Join<aod::D0ChargedJets, aod::D0ChargedJetConstituents> const & jets,
294294 aod::CandidatesD0Data const & candidates,
295- aod::JetTracks const & jettracks) {
295+ aod::JetTracks const & jettracks)
296+ {
296297 analyzeData<soa::Join<aod::D0ChargedJets, aod::D0ChargedJetConstituents>, aod::CandidatesD0Data>(collision, jets, candidates, jettracks);
297298 }
298299 PROCESS_SWITCH (HfFragmentationFunction, processD0DataCharged, " Store kinematic charged D0 jet information from measured DATA" , false );
299300
300301 void processLcDataCharged (aod::JetCollision const & collision,
301302 soa::Join<aod::LcChargedJets, aod::LcChargedJetConstituents> const & jets,
302303 aod::CandidatesLcData const & candidates,
303- aod::JetTracks const & jettracks) {
304+ aod::JetTracks const & jettracks)
305+ {
304306 analyzeData<soa::Join<aod::LcChargedJets, aod::LcChargedJetConstituents>, aod::CandidatesLcData>(collision, jets, candidates, jettracks);
305307 }
306308 PROCESS_SWITCH (HfFragmentationFunction, processLcDataCharged, " Store kinematic charged Lc jet information from measured DATA" , false );
@@ -404,7 +406,8 @@ struct HfFragmentationFunction {
404406 TCandidatesMCD const &,
405407 TCandidatesMCP const &,
406408 aod::JetTracks const &,
407- aod::JetParticles const &) {
409+ aod::JetParticles const &)
410+ {
408411 for (const auto & mccollision : mccollisions) {
409412 registry.fill (HIST (" h_collision_counter" ), 0.0 );
410413 // skip collisions outside of |z| < vertexZCut
@@ -430,7 +433,6 @@ struct HfFragmentationFunction {
430433 registry.fill (HIST (" h_jet_counter" ), 2.0 );
431434
432435 // apply collision sel8 selection on detector level jet's collision
433- // const auto& collision = mcdjet.get<aod::JetCollisionsMCD>();
434436 const auto & collision = collisions.iteratorAt (mcdjet.collisionId ());
435437 registry.fill (HIST (" h_collision_counter" ), 2.0 );
436438 if (!jetderiveddatautilities::selectCollision (collision, eventSelectionBits) || !(std::abs (collision.posZ ()) < vertexZCut)) {
@@ -464,21 +466,21 @@ struct HfFragmentationFunction {
464466 }
465467
466468 // store matched particle and detector level data in one single table (calculate angular distance in eta-phi plane on the fly)
467- matchJetTable (jetutilities::deltaR (mcpjet, mcpcand), mcpjet.pt (), mcpjet.eta (), mcpjet.phi (), mcpjet.template tracks_as <aod::JetParticles>().size (), // particle level jet
468- mcpcand.pt (), mcpcand.eta (), mcpcand.phi (), mcpcand.y (), (mcpcand.originMcGen () == RecoDecay::OriginType::Prompt), // particle level HF
469- jetutilities::deltaR (mcdjet, mcdcand), mcdjet.pt (), mcdjet.eta (), mcdjet.phi (), mcdjet.template tracks_as <aod::JetTracks>().size (), // detector level jet
470- mcdcand.pt (), mcdcand.eta (), mcdcand.phi (), mcdcand.m (), mcdcand.y (), (mcdcand.originMcRec () == RecoDecay::OriginType::Prompt), // detector level HF
471- mcdcand.mlScores ()[0 ], mcdcand.mlScores ()[1 ], mcdcand.mlScores ()[2 ], // Machine Learning PID scores: background, prompt, non-prompt
472- matchedFrom, selectedAs); // HF = +1, HFbar = -1, neither = 0
469+ matchJetTable (jetutilities::deltaR (mcpjet, mcpcand), mcpjet.pt (), mcpjet.eta (), mcpjet.phi (), mcpjet.template tracks_as <aod::JetParticles>().size (), // particle level jet
470+ mcpcand.pt (), mcpcand.eta (), mcpcand.phi (), mcpcand.y (), (mcpcand.originMcGen () == RecoDecay::OriginType::Prompt), // particle level HF
471+ jetutilities::deltaR (mcdjet, mcdcand), mcdjet.pt (), mcdjet.eta (), mcdjet.phi (), mcdjet.template tracks_as <aod::JetTracks>().size (), // detector level jet
472+ mcdcand.pt (), mcdcand.eta (), mcdcand.phi (), mcdcand.m (), mcdcand.y (), (mcdcand.originMcRec () == RecoDecay::OriginType::Prompt), // detector level HF
473+ mcdcand.mlScores ()[0 ], mcdcand.mlScores ()[1 ], mcdcand.mlScores ()[2 ], // Machine Learning PID scores: background, prompt, non-prompt
474+ matchedFrom, selectedAs); // HF = +1, HFbar = -1, neither = 0
473475 }
474476 } else {
475477 // store matched particle and detector level data in one single table (calculate angular distance in eta-phi plane on the fly)
476- matchJetTable (jetutilities::deltaR (mcpjet, mcpcand), mcpjet.pt (), mcpjet.eta (), mcpjet.phi (), mcpjet.template tracks_as <aod::JetParticles>().size (), // particle level jet
477- mcpcand.pt (), mcpcand.eta (), mcpcand.phi (), mcpcand.y (), (mcpcand.originMcGen () == RecoDecay::OriginType::Prompt), // particle level HF
478- -2 , -2 , -2 , -2 , -2 , // detector level jet
479- -2 , -2 , -2 , -2 , -2 , -2 , // detector level HF
480- -2 , -2 , -2 , // Machine Learning PID scores: background, prompt, non-prompt
481- -2 , -2 ); // HF = +1, HFbar = -1, neither = 0
478+ matchJetTable (jetutilities::deltaR (mcpjet, mcpcand), mcpjet.pt (), mcpjet.eta (), mcpjet.phi (), mcpjet.template tracks_as <aod::JetParticles>().size (), // particle level jet
479+ mcpcand.pt (), mcpcand.eta (), mcpcand.phi (), mcpcand.y (), (mcpcand.originMcGen () == RecoDecay::OriginType::Prompt), // particle level HF
480+ -2 , -2 , -2 , -2 , -2 , // detector level jet
481+ -2 , -2 , -2 , -2 , -2 , -2 , // detector level HF
482+ -2 , -2 , -2 , // Machine Learning PID scores: background, prompt, non-prompt
483+ -2 , -2 ); // HF = +1, HFbar = -1, neither = 0
482484 }
483485 } // end of mcpjets loop
484486 } // end of mccollisions loop
@@ -491,7 +493,8 @@ struct HfFragmentationFunction {
491493 aod::CandidatesD0MCD const & mcdcands,
492494 aod::CandidatesD0MCP const & mcpcands,
493495 aod::JetTracks const & jettracks,
494- aod::JetParticles const & jetparticles) {
496+ aod::JetParticles const & jetparticles)
497+ {
495498 analyzeMC<Preslice<JetD0MCPTable>, JetD0MCDTable, JetD0MCPTable, aod::CandidatesD0MCD, aod::CandidatesD0MCP>(d0MCPJetsPerMCCollisionPreslice, mccollisions, collisions, mcdjets, mcpjets, mcdcands, mcpcands, jettracks, jetparticles);
496499 }
497500 PROCESS_SWITCH (HfFragmentationFunction, processD0MC, " Store all simulated D0 jets information with matched candidate (if any found)" , false );
@@ -503,11 +506,11 @@ struct HfFragmentationFunction {
503506 aod::CandidatesLcMCD const & mcdcands,
504507 aod::CandidatesLcMCP const & mcpcands,
505508 aod::JetTracks const & jettracks,
506- aod::JetParticles const & jetparticles) {
509+ aod::JetParticles const & jetparticles)
510+ {
507511 analyzeMC<Preslice<JetLcMCPTable>, JetLcMCDTable, JetLcMCPTable, aod::CandidatesLcMCD, aod::CandidatesLcMCP>(lcMCPJetsPerMCCollisionPreslice, mccollisions, collisions, mcdjets, mcpjets, mcdcands, mcpcands, jettracks, jetparticles);
508512 }
509513 PROCESS_SWITCH (HfFragmentationFunction, processLcMC, " Store all simulated Lc jets information with matched candidate (if any found)" , false );
510-
511514};
512515
513516WorkflowSpec defineDataProcessing (ConfigContext const & cfgc)
0 commit comments