pass acceptance and parton pdg as parameters optionally

Author: gconesab

MC/config/PWGGAJE/trigger/prompt_gamma.C

@@ -16,9 +16,9 @@ R__ADD_INCLUDE_PATH($O2DPG_ROOT)
 /// \author Gustavo Conesa Balbastre (LPSC-IN2P3-CNRS)
 /// =================================================
 
-o2::eventgen::Trigger prompt_gamma( )
+o2::eventgen::Trigger prompt_gamma( int acceptanceIn = 0, int partonpdgIn = 0 )
 {
-  return [](const std::vector<TParticle>& particles) -> bool {
+  return [acceptanceIn,partonpdgIn](const std::vector<TParticle>& particles) -> bool {
 
 //    for(Int_t ipart = 3; ipart < 10; ipart++)
 //    {
@@ -34,25 +34,43 @@ o2::eventgen::Trigger prompt_gamma( )
 //    }
 
     // Get the outgoing 2->2 partons. 
-    // The photon and the associated outgoing parton are in position 5 or 6.
-    // Note that in PYTHIA6 they are at positions 7 or 8.
-    TParticle gamma  = particles.at(5);
-    TParticle parton = particles.at(6);
+    // The photon and the associated outgoing parton are in position 4 or 5.
+    // Note that in PYTHIA6 they are at positions 6 or 7.
+    int ig = 4;
+    int ip = 5;
+    TParticle gamma  = particles.at(ig);
     if ( gamma.GetPdgCode() != 22 ) {
-      gamma  = particles.at(6);
-      parton = particles.at(5);
+      ig = 5;
+      ip = 4;
+      gamma  = particles.at(ig);
     }
-    
+
+    TParticle parton = particles.at(ip);
+
     if ( gamma.GetPdgCode() != 22 )
     {
       printf("No direct photon found in the parton list!\n");
+
+//      for(Int_t ipart = 3; ipart < 10; ipart++)
+//      {
+//        TParticle part = particles.at(ipart);
+//        printf("\t parton %d, PDG %d, status %d, mother %d, E %2.2f, pT %2.2f, eta %2.2f, phi %2.2f\n", ipart,
+//               part.GetPdgCode(),
+//               part.GetStatusCode(),
+//               part.GetFirstMother(),
+//               part.Energy(),
+//               part.Pt(),
+//               part.Eta(),
+//               part.Phi()*TMath::RadToDeg());
+//      }
+
       return false;
     }
 
     // Select the flavour of the outgoing parton
     //
-    Int_t partonpdg = 0;
-    if ( gSystem->Getenv("CONFIG_OUTPARTON_PDG") )
+    int partonpdg = partonpdgIn;
+    if ( partonpdg <= 0 && gSystem->Getenv("CONFIG_OUTPARTON_PDG") )
       partonpdg = atoi(gSystem->Getenv("CONFIG_OUTPARTON_PDG"));
 
     if ( partonpdg > 0 && partonpdg <= 22 ) 
@@ -70,26 +88,26 @@ o2::eventgen::Trigger prompt_gamma( )
 
     // Select photons within acceptance
     //
-    Int_t acceptance = 0;
-    if ( gSystem->Getenv("PARTICLE_ACCEPTANCE") )
+    int acceptance = acceptanceIn;
+    if ( acceptance <= 0 && gSystem->Getenv("PARTICLE_ACCEPTANCE") )
       acceptance = atoi(gSystem->Getenv("PARTICLE_ACCEPTANCE"));
     //printf("Requested acceptance %d\n",acceptance);
 
     if ( detector_acceptance(acceptance, gamma.Phi(),gamma.Eta()) ) 
     {
-      printf("+++ Accepted event +++ \n");
-      printf("gamma, PDG %d, status %d, mother %d, E %2.2f, pT %2.2f, eta %2.2f, phi %2.2f\n", 
+      //printf("+++ Accepted event +++ \n");
+      printf("Selected gamma, pos %d, PDG %d, status %d, mother %d, E %2.2f, pT %2.2f, eta %2.2f, phi %2.2f\n",ig,
              gamma.GetPdgCode(), gamma.GetStatusCode(), gamma.GetFirstMother(),
              gamma.Energy()    , gamma.Pt(),
              gamma.Eta()       , gamma.Phi()*TMath::RadToDeg());
 
-      printf("parton, PDG %d, status %d, mother %d, E %2.2f, pT %2.2f, eta %2.2f, phi %2.2f\n", 
-             parton.GetPdgCode(), parton.GetStatusCode(), parton.GetFirstMother(),
-             parton.Energy()    , parton.Pt(),
-             parton.Eta()       , parton.Phi()*TMath::RadToDeg());
-      
+//      printf("Back-to-back parton, pos %d, PDG %d, status %d, mother %d, E %2.2f, pT %2.2f, eta %2.2f, phi %2.2f\n",ip,
+//             parton.GetPdgCode(), parton.GetStatusCode(), parton.GetFirstMother(),
+//             parton.Energy()    , parton.Pt(),
+//             parton.Eta()       , parton.Phi()*TMath::RadToDeg());
+//
 //      // Check difference in pT and azimuthal angle, it should be 0 and +-180 degrees, respectively.
-//      printf("parton-photon, Delta E %2.2f, Delta pT %2.2f, Delta eta %2.2f, Delta phi %2.2f\n", 
+//      printf("parton-photon, Delta E %2.2f, Delta pT %2.2f, Delta eta %2.2f,  Delta phi %2.2f\n",
 //             parton.Energy()-gamma.Energy(), parton.Pt() - gamma.Pt(),
 //             parton.Eta()   -gamma.Eta()   , parton.Phi()*TMath::RadToDeg()-gamma.Phi()*TMath::RadToDeg());