Skip to content

PWGHF: add a lighter version of muon selection process in HFL single muon task #1980

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vkucera merged 23 commits into from
Feb 27, 2023
Merged

PWGHF: add a lighter version of muon selection process in HFL single muon task #1980

Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
ff67241
Update taskSingleMuon.cxx
MaolinZH Jan 14, 2023
5ee59c8
Merge pull request #4 from MaolinZH/MaolinZH-patch-1
MaolinZH Jan 14, 2023
439369e
Merge branch 'AliceO2Group:master' into master
MaolinZH Jan 14, 2023
bc7d045
Update taskSingleMuon.cxx
MaolinZH Jan 14, 2023
63b5cec
Please consider the following formatting changes
alibuild Jan 14, 2023
a895463
Merge pull request #5 from alibuild/alibot-cleanup-1786
MaolinZH Jan 14, 2023
e1d8016
Merge branch 'AliceO2Group:master' into master
MaolinZH Feb 5, 2023
400c373
Add chi2 of MCH-MFT matching histograms
MaolinZH Feb 5, 2023
99b3527
Merge branch 'AliceO2Group:master' into master
MaolinZH Feb 13, 2023
1de604f
Add a lighter muon selection process
MaolinZH Feb 13, 2023
66905e9
Update taskSingleMuon.cxx
MaolinZH Feb 13, 2023
143eb4a
Please consider the following formatting changes
alibuild Feb 13, 2023
22dd197
Merge pull request #6 from alibuild/alibot-cleanup-1980
MaolinZH Feb 13, 2023
5e14e42
Merge branch 'AliceO2Group:master' into master
MaolinZH Feb 18, 2023
fdb68ad
Update taskSingleMuon.cxx
MaolinZH Feb 18, 2023
dd8b4b5
Update taskSingleMuon.cxx
MaolinZH Feb 21, 2023
34e7b3c
Update taskSingleMuon.cxx
MaolinZH Feb 22, 2023
543a7b0
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
dec1cbd
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
4f5db0a
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
91b448b
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
43bef7d
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
e9285fd
Update taskSingleMuon.cxx
MaolinZH Feb 23, 2023
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
176 changes: 119 additions & 57 deletions PWGHF/HFL/Tasks/taskSingleMuon.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,21 +13,21 @@
/// \brief Task derived from the DQ framework and used to extract the observables on single muons needed for the HF-muon analysis.
/// \author Maolin Zhang <maolin.zhang@cern.ch>, CCNU

#include "Framework/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Common/DataModel/EventSelection.h"
#include "Common/DataModel/TrackSelectionTables.h"
#include "Framework/ASoAHelpers.h"
#include "Framework/AnalysisDataModel.h"
#include "PWGDQ/Core/VarManager.h"
#include "PWGDQ/Core/HistogramManager.h"
#include "PWGDQ/Core/AnalysisCut.h"
#include "Framework/AnalysisTask.h"
#include "Framework/HistogramRegistry.h"
#include "Framework/runDataProcessing.h"
#include "PWGDQ/Core/AnalysisCompositeCut.h"
#include "PWGDQ/Core/AnalysisCut.h"
#include "PWGDQ/Core/CutsLibrary.h"
#include "PWGDQ/Core/HistogramManager.h"
#include "PWGDQ/Core/HistogramsLibrary.h"
#include "Framework/HistogramRegistry.h"
#include "PWGDQ/Core/VarManager.h"
#include "PWGDQ/DataModel/ReducedInfoTables.h"
#include "ReconstructionDataFormats/TrackFwd.h"
#include "Common/DataModel/EventSelection.h"
#include "Common/DataModel/TrackSelectionTables.h"

using namespace o2;
using namespace o2::aod;
Expand Down Expand Up @@ -136,6 +136,8 @@ struct HfTaskSingleMuonEventSelection {

struct HfTaskSingleMuonSelection {
Configurable<std::string> muonCuts{"muonCuts", "muonQualityCuts", "muon selection"};
Configurable<bool> fillLiteHist{"fillLiteHist", true, "fill lite version muon histograms"};
Configurable<bool> fillMcHist{"fillMcHist", false, "fill MC-related muon histograms"};

float* values;
AnalysisCompositeCut* trackCut;
Expand Down Expand Up @@ -171,15 +173,23 @@ struct HfTaskSingleMuonSelection {
HistogramConfigSpec hTHnPt{HistType::kTHnSparseD, {axispT, axispTGen, axispTDif, axisTrkType}, 4};
HistogramConfigSpec hTHnEta{HistType::kTHnSparseD, {axisEta, axisEtaGen, axisEtaDif, axisTrkType}, 4};
HistogramConfigSpec hTHnP{HistType::kTHnSparseD, {axisP, axisPGen, axisPDif, axisTrkType}, 4};
HistogramConfigSpec hTHnMuLite{HistType::kTHnSparseD, {axispT, axisEta, axisDCA, axisTrkType, axisChi2MatchMCHMFT, axisMCmask}, 6};

registry.add("hMuBcuts", "", hTHnMu);
registry.add("hMuAcuts", "", hTHnMu);
registry.add("hPTBcuts", "", hTHnPt);
registry.add("hPTAcuts", "", hTHnPt);
registry.add("hEtaBcuts", "", hTHnEta);
registry.add("hEtaAcuts", "", hTHnEta);
registry.add("hPBcuts", "", hTHnP);
registry.add("hPAcuts", "", hTHnP);
if (fillLiteHist) {
registry.add("hMuLiteBeforeCuts", "", hTHnMuLite);
registry.add("hMuLiteAfterCuts", "", hTHnMuLite);
} else {
registry.add("hMuBeforeCuts", "", hTHnMu);
registry.add("hMuAfterCuts", "", hTHnMu);
}
if (fillMcHist) {
registry.add("hPtBeforeCuts", "", hTHnPt);
registry.add("hPtAfterCuts", "", hTHnPt);
registry.add("hEtaBeforeCuts", "", hTHnEta);
registry.add("hEtaAfterCuts", "", hTHnEta);
registry.add("hPBeforeCuts", "", hTHnP);
registry.add("hPAfterCuts", "", hTHnP);
}

VarManager::SetDefaultVarNames();
values = new float[VarManager::kNVars];
Expand All @@ -194,8 +204,9 @@ struct HfTaskSingleMuonSelection {
void runMuonSel(TEvent const& event, aod::BCs const& bcs, TMuons const& tracks)
{
// select muons in data
if (event.isEventSelected() == 0)
if (event.isEventSelected() == 0) {
return;
}

VarManager::ResetValues(0, VarManager::kNMuonTrackVariables, values);
VarManager::FillEvent<gEventFillMap>(event, values);
Expand All @@ -205,34 +216,50 @@ struct HfTaskSingleMuonSelection {
VarManager::FillTrack<TMuonFillMap>(track, values);

// compute DCAXY
// With DQ's muon DCA calculation applied, we no longer need calculate it ourselve.
// Same for MC
const auto dcaXY(std::sqrt(values[VarManager::kMuonDCAx] * values[VarManager::kMuonDCAx] + values[VarManager::kMuonDCAy] * values[VarManager::kMuonDCAy]));

// Before Muon Cuts
registry.fill(HIST("hMuBcuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT], 0);
// After Muon Cuts
if (trackCut->IsSelected(values))
registry.fill(HIST("hMuAcuts"),
if (fillLiteHist) {
// Before Muon Cuts
registry.fill(HIST("hMuLiteBeforeCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT], 0);
// After Muon Cuts
if (trackCut->IsSelected(values)) {
registry.fill(HIST("hMuLiteAfterCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT], 0);
}
} else {
// Before Muon Cuts
registry.fill(HIST("hMuBeforeCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT], 0);
// After Muon Cuts
if (trackCut->IsSelected(values)) {
registry.fill(HIST("hMuAfterCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT], 0);
}
}
} // end loop over muon tracks
}

template <uint32_t TEventFillMap, uint32_t TMuonFillMap, uint32_t TTrackMCFillMap, typename TEvent, typename TMuons, typename TMC>
void runMuonSelMC(TEvent const& event, aod::BCs const& bcs, TMuons const& tracks, TMC const& mc)
void runMuonSelMc(TEvent const& event, aod::BCs const& bcs, TMuons const& tracks, TMC const& mc)
{
// select muons in MC
if (event.isEventSelected() == 0)
if (event.isEventSelected() == 0) {
return;
}

VarManager::ResetValues(0, VarManager::kNMuonTrackVariables, values);
VarManager::FillEvent<gEventFillMap>(event, values);
Expand All @@ -251,37 +278,72 @@ struct HfTaskSingleMuonSelection {
// compute DCAXY
const auto dcaXY(std::sqrt(values[VarManager::kMuonDCAx] * values[VarManager::kMuonDCAx] + values[VarManager::kMuonDCAy] * values[VarManager::kMuonDCAy]));

// Before Muon Cuts
registry.fill(HIST("hMuBcuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT],
track.mcMask());

registry.fill(HIST("hPTBcuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaBcuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPBcuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
// After Muon Cuts
if (trackCut->IsSelected(values)) {
registry.fill(HIST("hMuAcuts"),
if (fillLiteHist) {
// Before Muon Cuts
registry.fill(HIST("hMuLiteBeforeCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT],
track.mcMask());
if (fillMcHist) {
registry.fill(HIST("hPtBeforeCuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaBeforeCuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPBeforeCuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
}
// After Muon Cuts
if (trackCut->IsSelected(values)) {
registry.fill(HIST("hMuLiteAfterCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT],
track.mcMask());
if (fillMcHist) {
registry.fill(HIST("hPtAfterCuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaAfterCuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPAfterCuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
}
}
} else {
// Before Muon Cuts
registry.fill(HIST("hMuBeforeCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT],
track.mcMask());

registry.fill(HIST("hPTAcuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaAcuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPAcuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
if (fillMcHist) {
registry.fill(HIST("hPtBeforeCuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaBeforeCuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPBeforeCuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
}
// After Muon Cuts
if (trackCut->IsSelected(values)) {
registry.fill(HIST("hMuAfterCuts"),
values[VarManager::kPt],
values[VarManager::kEta], dcaXY,
values[VarManager::kCharge], track.p(),
values[VarManager::kVtxZ],
values[VarManager::kMuonTrackType], values[VarManager::kMuonChi2MatchMCHMFT],
track.mcMask());
if (fillMcHist) {
registry.fill(HIST("hPtAfterCuts"),
values[VarManager::kPt], values[VarManager::kMCPt], values[VarManager::kMCPt] - values[VarManager::kPt], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hEtaAfterCuts"),
values[VarManager::kEta], values[VarManager::kMCEta], values[VarManager::kMCEta] - values[VarManager::kEta], values[VarManager::kMuonTrackType]);
registry.fill(HIST("hPAfterCuts"),
values[VarManager::kP], values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]), values[VarManager::kMCPt] * std::cosh(values[VarManager::kMCEta]) - values[VarManager::kP], values[VarManager::kMuonTrackType]);
}
}
}
} // end loop over muon tracks
}
Expand All @@ -300,11 +362,11 @@ struct HfTaskSingleMuonSelection {
void processMuonMc(MyMcEventsSelected::iterator const& event, aod::BCs const& bcs,
MyMcMuons const& tracks, aod::McParticles const& mc)
{
runMuonSelMC<gEventFillMap, gMuonFillMap, gTrackMCFillMap>(event, bcs, tracks, mc);
runMuonSelMc<gEventFillMap, gMuonFillMap, gTrackMCFillMap>(event, bcs, tracks, mc);
}

PROCESS_SWITCH(HfTaskSingleMuonSelection, processDummy, "do nothing", false);
PROCESS_SWITCH(HfTaskSingleMuonSelection, processMuon, "run muon selection with real data", true);
PROCESS_SWITCH(HfTaskSingleMuonSelection, processMuon, "run muon selection with real data", false);
PROCESS_SWITCH(HfTaskSingleMuonSelection, processMuonMc, "run muon selection with MC data", false);
};

Expand Down