[PWGEM/Dileton] update studyDCAFitter.cxx (#16574)

Author: dsekihat

PWGEM/Dilepton/DataModel/lmeeMLTables.h

@@ -344,6 +344,58 @@ DECLARE_SOA_TABLE(EMMLLCascPairs, "AOD", "EMMLLCPAIR", //!
 // iterators
 using EMMLLCascPair = EMMLLCascPairs::iterator;
 
+namespace emmldilepton
+{
+DECLARE_SOA_INDEX_COLUMN(EMMLEvent, emmlevent);    //! index to event table
+DECLARE_SOA_COLUMN(Signed1Pt1, signed1Pt1, float); //! q/pt of lepton1 at PV
+DECLARE_SOA_COLUMN(Eta1, eta1, float);             //! eta of lepton1 at PV
+DECLARE_SOA_COLUMN(ImpParXY1, impParXY1, float);   //! impact parameter for lepton1 in XY plane
+DECLARE_SOA_COLUMN(ImpParZ1, impParZ1, float);     //! impact parameter for lepton1 in Z plane
+DECLARE_SOA_COLUMN(ImpParCYY1, impParCYY1, float); //! sigma of impact parameter for lepton1 in XY
+DECLARE_SOA_COLUMN(ImpParCZY1, impParCZY1, float); //! sigma of impact parameter for lepton1, correlaion term
+DECLARE_SOA_COLUMN(ImpParCZZ1, impParCZZ1, float); //! sigma of impact parameter for lepton1 in Z
+
+DECLARE_SOA_COLUMN(Signed1Pt2, signed1Pt2, float); //! q/pt of lepton2 at PV
+DECLARE_SOA_COLUMN(Eta2, eta2, float);             //! eta of lepton1 at PV
+DECLARE_SOA_COLUMN(ImpParXY2, impParXY2, float);   //! impact parameter for lepton2 in XY plane
+DECLARE_SOA_COLUMN(ImpParZ2, impParZ2, float);     //! impact parameter for lepton2 in Z plane
+DECLARE_SOA_COLUMN(ImpParCYY2, impParCYY2, float); //! sigma of impact parameter for lepton2 in XY
+DECLARE_SOA_COLUMN(ImpParCZY2, impParCZY2, float); //! sigma of impact parameter for lepton2, correlaion term
+DECLARE_SOA_COLUMN(ImpParCZZ2, impParCZZ2, float); //! sigma of impact parameter for lepton2 in Z
+
+DECLARE_SOA_COLUMN(Mass, mass, float);         //! invariant mass of dilepton
+DECLARE_SOA_COLUMN(Pt, pt, float);             //! dilepton pair pt
+DECLARE_SOA_COLUMN(Rapidity, rapidity, float); //! dilepton pair rapidity
+DECLARE_SOA_COLUMN(Chi2PCA, chi2PCA, float);   //! chi2 at PCA of dilepton
+DECLARE_SOA_COLUMN(CPA, cpa, float);           //! cosine of pointing angle of dilepton
+DECLARE_SOA_COLUMN(CPAXY, cpaXY, float);       //! cosine of pointing angle of dilepton in XY
+DECLARE_SOA_COLUMN(CPARZ, cpaRZ, float);       //! cosine of pointing angle of dilepton in RZ
+
+DECLARE_SOA_COLUMN(Lxy, lxy, float);         //! decay length of dilepton
+DECLARE_SOA_COLUMN(Lz, lz, float);           //! decay length of dilepton
+DECLARE_SOA_COLUMN(Lxyz, lxyz, float);       //! decay length of dilepton
+DECLARE_SOA_COLUMN(LxyErr, lxyErr, float);   //! decay length resolution of dilepton
+DECLARE_SOA_COLUMN(LzErr, lzErr, float);     //! decay length resolution of dilepton
+DECLARE_SOA_COLUMN(LxyzErr, lxyzErr, float); //! decay length resolution of dilepton
+
+DECLARE_SOA_COLUMN(DileptonType, dileptonType, uint8_t);           //! dilepton type: 0=combinatorial bkg, 1=prompt SM, 2=nonprompt SM, 3=c2l_c2l, 4=b2l_b2l, 5=b2bc2l_b2c2l, 6=b2c2e_b2e_sameb, 7=b2c2e_b2e_diffb
+DECLARE_SOA_COLUMN(PdgCodeCommonMother, pdgCodeCommonMother, int); //! pdg code of common mother. If it is not found, -1.
+} // namespace emmldilepton
+
+DECLARE_SOA_TABLE(EMMLDielectronsAtSV, "AOD", "EMMLEESV", //!
+                  emmldilepton::EMMLEventId,
+                  emmldilepton::Signed1Pt1, emmldilepton::Eta1, emmldilepton::ImpParXY1, emmldilepton::ImpParZ1, emmldilepton::ImpParCYY1, emmldilepton::ImpParCZY1, emmldilepton::ImpParCZZ1,
+                  emmldilepton::Signed1Pt2, emmldilepton::Eta2, emmldilepton::ImpParXY2, emmldilepton::ImpParZ2, emmldilepton::ImpParCYY2, emmldilepton::ImpParCZY2, emmldilepton::ImpParCZZ2,
+                  emmldilepton::Mass, emmldilepton::Pt, emmldilepton::Rapidity,
+                  emmldilepton::Chi2PCA,
+                  emmldilepton::CPA, emmldilepton::CPAXY, emmldilepton::CPARZ,
+                  emmldilepton::Lxy, emmldilepton::Lz, emmldilepton::Lxyz,
+                  emmldilepton::LxyErr, emmldilepton::LzErr, emmldilepton::LxyzErr,
+                  emmldilepton::DileptonType, emmldilepton::PdgCodeCommonMother);
+
+// iterators
+using EMMLDielectronAtSV = EMMLDielectronsAtSV::iterator;
+
 } // namespace o2::aod
 
 #endif // PWGEM_DILEPTON_DATAMODEL_LMEEMLTABLES_H_