-add pt^2 variable

diff --git a/PWGDQ/dielectron/AliDielectronVarManager.cxx b/PWGDQ/dielectron/AliDielectronVarManager.cxx
index 4300065a47af84dbad8c9b7ae62ccb03fba3e446..c95de43aed79c74236828492acc4fd9a86ede34e 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronVarManager.cxx
+++ b/PWGDQ/dielectron/AliDielectronVarManager.cxx
@@ -31,6 +31,7 @@ const char* AliDielectronVarManager::fgkParticleNames[AliDielectronVarManager::k
   {"Py",                     "#it{p}_{y}",                                         "(GeV/#it{c})"},
   {"Pz",                     "#it{p}_{z}",                                         "(GeV/#it{c})"},
   {"Pt",                     "#it{p}_{T}",                                         "(GeV/#it{c})"},
   {"Py",                     "#it{p}_{y}",                                         "(GeV/#it{c})"},
   {"Pz",                     "#it{p}_{z}",                                         "(GeV/#it{c})"},
   {"Pt",                     "#it{p}_{T}",                                         "(GeV/#it{c})"},

+  {"PtSq",                   "#it{p}_{T}^{2}",                                     "((GeV/#it{c})^{2})"},

   {"P",                      "#it{p}",                                             "(GeV/#it{c})"},
   {"Xv",                     "x_{vtx}",                                            "(cm)"},
   {"Yv",                     "y_{vtx}",                                            "(cm)"},
   {"P",                      "#it{p}",                                             "(GeV/#it{c})"},
   {"Xv",                     "x_{vtx}",                                            "(cm)"},
   {"Yv",                     "y_{vtx}",                                            "(cm)"},

diff --git a/PWGDQ/dielectron/AliDielectronVarManager.h b/PWGDQ/dielectron/AliDielectronVarManager.h
index ddf2264f490b5d238a61c3a2aa46f25e88c41326..fbe258df97bbf2ef3b17de985b532c621feffa9a 100644 (file)
--- a/PWGDQ/dielectron/AliDielectronVarManager.h
+++ b/PWGDQ/dielectron/AliDielectronVarManager.h
@@ -84,6 +84,7 @@ public:
     kPy,                     // py
     kPz,                     // pz
     kPt,                     // transverse momentum
     kPy,                     // py
     kPz,                     // pz
     kPt,                     // transverse momentum

+    kPtSq,                   // transverse momentum squared

     kP,                      // momentum
     kXv,                     // vertex position in x
     kYv,                     // vertex position in y
     kP,                      // momentum
     kXv,                     // vertex position in x
     kYv,                     // vertex position in y


@@ -516,6 +517,7 @@ inline void AliDielectronVarManager::FillVarVParticle(const AliVParticle *partic
   values[AliDielectronVarManager::kPy]        = particle->Py();
   values[AliDielectronVarManager::kPz]        = particle->Pz();
   values[AliDielectronVarManager::kPt]        = particle->Pt();
   values[AliDielectronVarManager::kPy]        = particle->Py();
   values[AliDielectronVarManager::kPz]        = particle->Pz();
   values[AliDielectronVarManager::kPt]        = particle->Pt();

+  values[AliDielectronVarManager::kPtSq]      = particle->Pt()*particle->Pt();

   values[AliDielectronVarManager::kP]         = particle->P();
 
   values[AliDielectronVarManager::kXv]        = particle->Xv();
   values[AliDielectronVarManager::kP]         = particle->P();
 
   values[AliDielectronVarManager::kXv]        = particle->Xv();


@@ -1088,6 +1090,7 @@ inline void AliDielectronVarManager::FillVarMCParticle2(const AliVParticle *p1,
                                                      values[AliDielectronVarManager::kPx]+
                                                      values[AliDielectronVarManager::kPy]*
                                                      values[AliDielectronVarManager::kPy]);
                                                      values[AliDielectronVarManager::kPx]+
                                                      values[AliDielectronVarManager::kPy]*
                                                      values[AliDielectronVarManager::kPy]);

+  values[AliDielectronVarManager::kPtSq]      = values[AliDielectronVarManager::kPt] * values[AliDielectronVarManager::kPt];

   values[AliDielectronVarManager::kP]         = TMath::Sqrt(values[AliDielectronVarManager::kPt]*
                                                      values[AliDielectronVarManager::kPt]+
                                                      values[AliDielectronVarManager::kPz]*
   values[AliDielectronVarManager::kP]         = TMath::Sqrt(values[AliDielectronVarManager::kPt]*
                                                      values[AliDielectronVarManager::kPt]+
                                                      values[AliDielectronVarManager::kPz]*


@@ -1293,6 +1296,7 @@ inline void AliDielectronVarManager::FillVarDielectronPair(const AliDielectronPa
        values[AliDielectronVarManager::kPz]        = (lv1+lv2).Pz();
 
        values[AliDielectronVarManager::kPt]        =  (lv1+lv2).Pt();
        values[AliDielectronVarManager::kPz]        = (lv1+lv2).Pz();
 
        values[AliDielectronVarManager::kPt]        =  (lv1+lv2).Pt();

+       values[AliDielectronVarManager::kPtSq]      = values[AliDielectronVarManager::kPt] * values[AliDielectronVarManager::kPt];

 
        values[AliDielectronVarManager::kP]         =  (lv1+lv2).P();
 
 
        values[AliDielectronVarManager::kP]         =  (lv1+lv2).P();
 


@@ -1498,6 +1502,7 @@ inline void AliDielectronVarManager::FillVarKFParticle(const AliKFParticle *part
   values[AliDielectronVarManager::kPy]        = particle->GetPy();
   values[AliDielectronVarManager::kPz]        = particle->GetPz();
   values[AliDielectronVarManager::kPt]        = particle->GetPt();
   values[AliDielectronVarManager::kPy]        = particle->GetPy();
   values[AliDielectronVarManager::kPz]        = particle->GetPz();
   values[AliDielectronVarManager::kPt]        = particle->GetPt();

+  values[AliDielectronVarManager::kPtSq]      = particle->GetPt() * particle->GetPt();

   values[AliDielectronVarManager::kP]         = particle->GetP();
   
   values[AliDielectronVarManager::kXv]        = particle->GetX();
   values[AliDielectronVarManager::kP]         = particle->GetP();
   
   values[AliDielectronVarManager::kXv]        = particle->GetX();