]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PYTHIA6/AliDecayerPythia.cxx
Updating macros for LEGO train
[u/mrichter/AliRoot.git] / PYTHIA6 / AliDecayerPythia.cxx
index 125956ee2cd5e50157ca8e9e948010791e0ea330..0e7e5c9948dba7cea9ad4b0cc1029091d1de48bf 100644 (file)
@@ -58,7 +58,9 @@ Bool_t AliDecayerPythia::fgInit = kFALSE;
 AliDecayerPythia::AliDecayerPythia():
     fPythia(AliPythia::Instance()),
     fDecay(kAll),
-    fHeavyFlavour(kTRUE)
+    fHeavyFlavour(kTRUE),
+    fLongLived(kFALSE),
+    fPatchOmegaDalitz(0)
 {
 // Default Constructor
     for (Int_t i=0; i< 501; i++) fBraPart[i]= 1.;
@@ -69,10 +71,13 @@ AliDecayerPythia::AliDecayerPythia(const AliDecayerPythia &decayer):
     AliDecayer(decayer),
     fPythia(0),
     fDecay(kAll),
-    fHeavyFlavour(kTRUE)
+    fHeavyFlavour(kTRUE),
+    fLongLived(kFALSE),
+    fPatchOmegaDalitz(0)
 {
     // Copy Constructor
     decayer.Copy(*this);
+    for (Int_t i = 0; i < 501; i++) fBraPart[i] = 0.;
 }
 
 void AliDecayerPythia::Init()
@@ -114,16 +119,18 @@ void AliDecayerPythia::Init()
 */
 
     fPythia->SetMDCY(fPythia->Pycomp(111) ,1, 1);
-
-    fPythia->SetMDCY(fPythia->Pycomp(310) ,1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3122),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3112),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3212),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3222),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3312),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3322),1,0);
-    fPythia->SetMDCY(fPythia->Pycomp(3334),1,0);
-
+    Int_t isw = 0;
+    if (fLongLived) isw = 1;
+    
+    fPythia->SetMDCY(fPythia->Pycomp(310) ,1, isw);
+    fPythia->SetMDCY(fPythia->Pycomp(3122),1, isw);
+    fPythia->SetMDCY(fPythia->Pycomp(3112),1, isw);
+//    fPythia->SetMDCY(fPythia->Pycomp(3212),1, isw); // Sigma0 decays elem.
+    fPythia->SetMDCY(fPythia->Pycomp(3222),1, isw);
+    fPythia->SetMDCY(fPythia->Pycomp(3312),1, isw);
+    fPythia->SetMDCY(fPythia->Pycomp(3322),1, isw);
+    fPythia->SetMDCY(fPythia->Pycomp(3334),1, isw);
+    
 // .. Force decay channels
     ForceDecay();
 }
@@ -135,8 +142,15 @@ void AliDecayerPythia::Decay(Int_t idpart, TLorentzVector* p)
     Float_t energy = p->Energy();
     Float_t theta  = p->Theta();
     Float_t phi    = p->Phi();
-    
-    Lu1Ent(0, idpart, energy, theta, phi);
+    if (!fPatchOmegaDalitz) {
+       Lu1Ent(0, idpart, energy, theta, phi);
+    } else {
+       fPythia->SetMDCY(fPythia->Pycomp(111) ,1, 0);
+       Lu1Ent(0, idpart, energy, theta, phi);
+       fPythia->DalitzDecays();
+       fPythia->SetMDCY(fPythia->Pycomp(111) ,1, 1);
+       fPythia->Pyexec();
+    }
     fPythia->GetPrimaries();
 }
 
@@ -156,7 +170,6 @@ void AliDecayerPythia::ForceDecay()
 //
     Decay_t decay=fDecay;
     fPythia->SetMSTJ(21,2);
-    if (decay == kNoDecayHeavy) return;
 
 //
 // select mode    
@@ -246,6 +259,9 @@ void AliDecayerPythia::ForceDecay()
        ForceParticleDecay(100553,13,2);// Upsilon'
        ForceParticleDecay(200553,13,2);// Upsilon''
        break;
+    case kJpsiDiMuon:
+       ForceParticleDecay(  443,13,2); // J/Psi
+       break;
     case kBSemiElectronic:
        ForceParticleDecay(  511,11,1); // B0     
        ForceParticleDecay(  521,11,1); // B+/-     
@@ -282,6 +298,14 @@ void AliDecayerPythia::ForceDecay()
        ForceParticleDecay(100553,11,2);// Upsilon'
        ForceParticleDecay(200553,11,2);// Upsilon''
        break;
+    case kPsiPrimeJpsiDiElectron:
+       products[0] =    443;
+        products[1] =    211;
+        mult[0] = 1;
+        mult[1] = 2;
+        ForceParticleDecay( 100443, products, mult, 2, 1);
+        ForceParticleDecay(    443,11,2);             
+        break; 
     case kBJpsiDiMuon:
 
        products[0] =    443;
@@ -366,6 +390,10 @@ void AliDecayerPythia::ForceDecay()
        break;
     case kOmega:
        ForceOmega();
+       break;
+    case kLambda:
+       ForceLambda();
+       break;
     case kAll:
        break;
     case kNoDecay:
@@ -374,6 +402,33 @@ void AliDecayerPythia::ForceDecay()
     case kNoDecayHeavy:
     case kNeutralPion:
        break;
+    case kNoDecayBeauty:                                                    
+        SwitchOffBDecay(); 
+       break;
+    case kElectronEM:
+        ForceParticleDecay(  111,11,1); // pi^0     
+       ForceParticleDecay(  221,11,1); // eta     
+       ForceParticleDecay(  113,11,1); // rho     
+       ForceParticleDecay(  223,11,1); // omega     
+       ForceParticleDecay(  331,11,1); // etaprime     
+       ForceParticleDecay(  333,11,1); // phi     
+       break;
+    case kDiElectronEM:
+       ForceParticleDecay(  111,11,2); // pi^0     
+       ForceParticleDecay(  221,11,2); // eta     
+       ForceParticleDecay(  113,11,2); // rho     
+       ForceParticleDecay(  223,11,2); // omega     
+       ForceParticleDecay(  331,11,2); // etaprime     
+       ForceParticleDecay(  333,11,2); // phi     
+       break;
+    case kGammaEM:
+       ForceParticleDecay(  111,22,1); // pi^0     
+       ForceParticleDecay(  221,22,1); // eta     
+       ForceParticleDecay(  113,22,1); // rho     
+       ForceParticleDecay(  223,22,1); // omega     
+       ForceParticleDecay(  331,22,1); // etaprime     
+       ForceParticleDecay(  333,22,1); // phi     
+       break;
     }
 }
 
@@ -436,11 +491,13 @@ void AliDecayerPythia::ForceHadronicD(Int_t optUse4Bodies)
   Int_t productsD[2] = {kProton, kPiPlus}, multD[2] = {1, 1};
   ForceParticleDecay(iDeltaPP, productsD, multD, 2);
   // for Lambda_c -> Lambda(1520) pi+ -> p K- pi+ 
-  Int_t iLambda_1520 = 3124;
+  Int_t iLambda1520 = 3124;
   Int_t productsL[2] = {kProton, kKMinus}, multL[2] = {1, 1};
-  ForceParticleDecay(iLambda_1520, productsL, multL, 2);
+  ForceParticleDecay(iLambda1520, productsL, multL, 2);
   // for Lambda_c -> Lambda pi+
   Int_t iLambda=3122;
+  //for Lambda_c -> antiK0 p
+  Int_t iK0bar=-311;
 
 
   Int_t decayP1[kNHadrons][4] = 
@@ -474,7 +531,7 @@ void AliDecayerPythia::ForceHadronicD(Int_t optUse4Bodies)
       {iKstarbar0  , kPiPlus , kPiMinus,  0},
       {-1          , -1      , -1      , -1},
       {-1          , -1      , -1      , -1},
-      {iLambda_1520, kPiPlus ,  0      ,  0}
+      {iLambda1520 , kPiPlus ,  0      ,  0}
     };
   // for Lambda_c -> Lambda pi+ 
     Int_t decayP5[kNHadrons][4] =
@@ -486,6 +543,16 @@ void AliDecayerPythia::ForceHadronicD(Int_t optUse4Bodies)
       {iLambda   , kPiPlus,  0      ,  0}
     };
 
+    // for Lambda_c -> K0bar p
+    Int_t decayP6[kNHadrons][4] =
+    {
+      {-1        , -1     , -1      , -1},
+      {-1        , -1     , -1      , -1},
+      {-1        , -1     , -1      , -1},
+      {-1        , -1     , -1      , -1},
+      {kProton    , iK0bar,  0      ,  0}
+    };
+
   if(optUse4Bodies==0){
     for(Int_t iDau=0;iDau<4;iDau++){
       decayP2[1][iDau]=-1;
@@ -540,6 +607,13 @@ void AliDecayerPythia::ForceHadronicD(Int_t optUse4Bodies)
               fPythia->GetKFDP(channel,3) == decayP5[ihadron][2] &&
               fPythia->GetKFDP(channel,4) == decayP5[ihadron][3] &&
               fPythia->GetKFDP(channel,5) == 0
+        ) || (
+              fPythia->GetKFDP(channel,1) == decayP6[ihadron][0] &&
+              fPythia->GetKFDP(channel,2) == decayP6[ihadron][1] &&
+              fPythia->GetKFDP(channel,3) == decayP6[ihadron][2] &&
+              fPythia->GetKFDP(channel,4) == decayP6[ihadron][3] &&
+              fPythia->GetKFDP(channel,5) == 0
+
              ))
          
          {
@@ -585,7 +659,7 @@ void AliDecayerPythia::ForceParticleDecay(Int_t particle, Int_t product, Int_t m
     if (norm > 0.) fBraPart[kc] /= norm;
 }
 
-void AliDecayerPythia::ForceParticleDecay(Int_t particle, Int_t* products, Int_t* mult, Int_t npart)
+void AliDecayerPythia::ForceParticleDecay(Int_t particle, const Int_t* products, Int_t* mult, Int_t npart, Bool_t flag)
 {
 //
 //  Force decay of particle into products with multiplicity mult
@@ -609,9 +683,9 @@ void AliDecayerPythia::ForceParticleDecay(Int_t particle, Int_t* products, Int_t
        for (Int_t i = 0; i < npart; i++) {
            nprod += (CountProducts(channel, products[i]) >= mult[i]);
        }
-       if (nprod) {
+       if ((nprod && !flag) || ((nprod == npart) && flag)) {
            fPythia->SetMDME(channel,1,1);
-       } else {
+       } else {                // 
            fPythia->SetMDME(channel,1,0);
            fBraPart[kc] -= fPythia->GetBRAT(channel);
        }
@@ -700,7 +774,37 @@ void  AliDecayerPythia::ForceOmega()
     } // decay channels
 }
 
+void  AliDecayerPythia::ForceLambda()
+{
+    // Force Lambda -> p pi-
+    Int_t kc=fPythia->Pycomp(3122);
+    fPythia->SetMDCY(kc,1,1);  
+    Int_t ifirst = fPythia->GetMDCY(kc,2);
+    Int_t ilast  = ifirst + fPythia->GetMDCY(kc,3)-1;
+    for (Int_t channel = ifirst; channel <= ilast; channel++) {
+       if (
+           fPythia->GetKFDP(channel,1) == kProton   &&
+           fPythia->GetKFDP(channel,2) == kPiMinus  &&
+           fPythia->GetKFDP(channel,3) == 0
+           )
+       {
+           fPythia->SetMDME(channel,1,1);
+       } else {
+           fPythia->SetMDME(channel,1,0);
+       } // selected channel ?
+    } // decay channels
+}
+
 
+void AliDecayerPythia::SwitchOffBDecay()
+{
+// Switch off B-decays
+  Int_t heavyB[]={511,521,531,5122,5132,5232,5332};
+  for(int i=0;i<4;i++)
+    {
+      fPythia->SetMDCY(fPythia->Pycomp(heavyB[i]),1,0);
+    }
+}
 
 Float_t  AliDecayerPythia::GetPartialBranchingRatio(Int_t kf)
 {
@@ -734,27 +838,6 @@ void AliDecayerPythia::ReadDecayTable()
     
 }
 
-#ifdef never
-void AliDecayerPythia::Streamer(TBuffer &R__b)
-{
-   // Stream an object of class AliDecayerPythia.
-
-   if (R__b.IsReading()) {
-      Version_t R__v = R__b.ReadVersion(); if (R__v) { }
-      AliDecayer::Streamer(R__b);
-      (AliPythia::Instance())->Streamer(R__b);
-      R__b >> (Int_t&)fDecay;
-      R__b.ReadStaticArray(fBraPart);
-   } else {
-      R__b.WriteVersion(AliDecayerPythia::IsA());
-      AliDecayer::Streamer(R__b);
-      R__b << fPythia;
-      R__b << (Int_t)fDecay;
-      R__b.WriteArray(fBraPart, 501);
-   }
-}
-#endif
-
 void AliDecayerPythia::Copy(TObject &) const
 {
   //