From 8b7ebb39115eed0fccc22b4d0a7e57a8dfdd41d4 Mon Sep 17 00:00:00 2001 From: Daiki Sekihata Date: Tue, 21 Apr 2026 21:20:27 +0200 Subject: [PATCH 1/2] [PWGEM/Dilepton] add err of x,y in treeCreatorMuonML.cxx --- PWGEM/Dilepton/DataModel/lmeeMLTables.h | 4 ++-- PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx | 9 +++++++++ 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/PWGEM/Dilepton/DataModel/lmeeMLTables.h b/PWGEM/Dilepton/DataModel/lmeeMLTables.h index 9fdb761a883..17eef48afa7 100644 --- a/PWGEM/Dilepton/DataModel/lmeeMLTables.h +++ b/PWGEM/Dilepton/DataModel/lmeeMLTables.h @@ -163,9 +163,9 @@ DECLARE_SOA_TABLE_VERSIONED(EMFwdTracksForML_000, "AOD", "EMFWDTRKML", 0, //! emmlfwdtrack::Signed1PtMCHMIDatMP, emmlfwdtrack::EtaMCHMIDatMP, emmlfwdtrack::PhiMCHMIDatMP, emmlfwdtrack::Signed1PtMFTatMP, emmlfwdtrack::EtaMFTatMP, emmlfwdtrack::PhiMFTatMP, emmlfwdtrack::XMCHMIDatMP, emmlfwdtrack::YMCHMIDatMP, - // emmlfwdtrack::XErrMCHMIDatMP, emmlfwdtrack::YErrMCHMIDatMP, + emmlfwdtrack::XErrMCHMIDatMP, emmlfwdtrack::YErrMCHMIDatMP, emmlfwdtrack::XMFTatMP, emmlfwdtrack::YMFTatMP, - // emmlfwdtrack::XErrMFTatMP, emmlfwdtrack::YErrMFTatMP, + emmlfwdtrack::XErrMFTatMP, emmlfwdtrack::YErrMFTatMP, fwdtrack::FwdDcaX, fwdtrack::FwdDcaY, fwdtrack::NClusters, fwdtrack::PDca, fwdtrack::RAtAbsorberEnd, diff --git a/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx b/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx index 06c778e2c40..deb64fe77fa 100644 --- a/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx +++ b/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx @@ -344,6 +344,9 @@ struct TreeCreatorMuonML { float xMFTatMP = 999.f, yMFTatMP = 999.f; float xMCHMIDatMP = 999.f, yMCHMIDatMP = 999.f; + + float xErrMFTatMP = 999.f, yErrMFTatMP = 999.f; + float xErrMCHMIDatMP = 999.f, yErrMCHMIDatMP = 999.f; float signed1PtMFTatMP = 999.f, etaMFTatMP = 999.f, phiMFTatMP = 999.f; float signed1PtMCHMIDatMP = 999.f, etaMCHMIDatMP = 999.f, phiMCHMIDatMP = 999.f; @@ -353,6 +356,8 @@ struct TreeCreatorMuonML { mftsaAtMP.propagateToZhelix(glMuonCutGroup.matchingZ, mBz); // propagated to matching plane xMFTatMP = mftsaAtMP.getX(); yMFTatMP = mftsaAtMP.getY(); + xErrMFTatMP = std::sqrt(mftsaAtMP.getSigma2X()); + yErrMFTatMP = std::sqrt(mftsaAtMP.getSigma2Y()); signed1PtMFTatMP = mftsaAtMP.getInvQPt(); etaMFTatMP = mftsaAtMP.getEta(); phiMFTatMP = RecoDecay::constrainAngle(mftsaAtMP.getPhi(), 0, 1U); @@ -360,6 +365,8 @@ struct TreeCreatorMuonML { auto muonAtMP = propagateMuon(mchtrack, mchtrack, collision, propagationPoint::kToMatchingPlane, glMuonCutGroup.matchingZ, mBz, mZShift); // propagated to matching plane xMCHMIDatMP = muonAtMP.getX(); yMCHMIDatMP = muonAtMP.getY(); + xErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2X()); + yErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2Y()); signed1PtMCHMIDatMP = mftsaAtMP.getInvQPt(); etaMCHMIDatMP = muonAtMP.getEta(); phiMCHMIDatMP = RecoDecay::constrainAngle(muonAtMP.getPhi(), 0, 1U); @@ -374,7 +381,9 @@ struct TreeCreatorMuonML { signed1PtMFTatMP, etaMFTatMP, phiMFTatMP, signed1PtMCHMIDatMP, etaMCHMIDatMP, phiMCHMIDatMP, xMCHMIDatMP, yMCHMIDatMP, + xErrMCHMIDatMP, yErrMCHMIDatMP, xMFTatMP, yMFTatMP, + xErrMFTatMP, yErrMFTatMP, dcaX, dcaY, fwdtrack.nClusters(), pDCA, rAtAbsorberEnd, fwdtrack.chi2MatchMCHMID(), fwdtrack.chi2MatchMCHMFT(), mfttrack.mftClusterSizesAndTrackFlags(), chi2, mchtrack.chi2(), chi2mft, mfttrack.nClusters(), From 1d374b09bd0fd9b18fa52abdc3bbd06821e4a3cb Mon Sep 17 00:00:00 2001 From: Daiki Sekihata Date: Tue, 21 Apr 2026 21:44:55 +0200 Subject: [PATCH 2/2] Fix signed1PtMCHMIDatMP assignment to use muonAtMP --- PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx b/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx index deb64fe77fa..0d04d7834d6 100644 --- a/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx +++ b/PWGEM/Dilepton/TableProducer/treeCreatorMuonML.cxx @@ -367,7 +367,7 @@ struct TreeCreatorMuonML { yMCHMIDatMP = muonAtMP.getY(); xErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2X()); yErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2Y()); - signed1PtMCHMIDatMP = mftsaAtMP.getInvQPt(); + signed1PtMCHMIDatMP = muonAtMP.getInvQPt(); etaMCHMIDatMP = muonAtMP.getEta(); phiMCHMIDatMP = RecoDecay::constrainAngle(muonAtMP.getPhi(), 0, 1U); }