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..0d04d7834d6 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,7 +365,9 @@ struct TreeCreatorMuonML { auto muonAtMP = propagateMuon(mchtrack, mchtrack, collision, propagationPoint::kToMatchingPlane, glMuonCutGroup.matchingZ, mBz, mZShift); // propagated to matching plane xMCHMIDatMP = muonAtMP.getX(); yMCHMIDatMP = muonAtMP.getY(); - signed1PtMCHMIDatMP = mftsaAtMP.getInvQPt(); + xErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2X()); + yErrMCHMIDatMP = std::sqrt(muonAtMP.getSigma2Y()); + signed1PtMCHMIDatMP = muonAtMP.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(),