]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PYTHIA6/AliDecayerPythia.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Coverity corrections.
[u/mrichter/AliRoot.git] / PYTHIA6 / AliDecayerPythia.cxx
diff --git a/PYTHIA6/AliDecayerPythia.cxx b/PYTHIA6/AliDecayerPythia.cxx
index 040def84383096cbe091af44a64dd3d1ab0dd17b..6cf6d84a1ff16b2dfc78401809381fff19614103 100644 (file)
--- a/PYTHIA6/AliDecayerPythia.cxx
+++ b/PYTHIA6/AliDecayerPythia.cxx
@@ -58,7 +58,8 @@ Bool_t AliDecayerPythia::fgInit = kFALSE;
 AliDecayerPythia::AliDecayerPythia():
     fPythia(AliPythia::Instance()),
     fDecay(kAll),
-    fHeavyFlavour(kTRUE)
+    fHeavyFlavour(kTRUE),
+    fLongLived(kFALSE)
 {
 // Default Constructor
     for (Int_t i=0; i< 501; i++) fBraPart[i]= 1.;
@@ -69,10 +70,12 @@ AliDecayerPythia::AliDecayerPythia(const AliDecayerPythia &decayer):
     AliDecayer(decayer),
     fPythia(0),
     fDecay(kAll),
-    fHeavyFlavour(kTRUE)
+    fHeavyFlavour(kTRUE),
+    fLongLived(kFALSE)
 {
     // Copy Constructor
     decayer.Copy(*this);
+    for (Int_t i = 0; i < 501; i++) fBraPart[i] = 0.;
 }
 
 void AliDecayerPythia::Init()
@@ -114,16 +117,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();
 }
@@ -282,6 +287,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 +379,8 @@ void AliDecayerPythia::ForceDecay()
        break;
     case kOmega:
        ForceOmega();
+    case kLambda:
+       ForceLambda();
     case kAll:
        break;
     case kNoDecay:
@@ -374,6 +389,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;
     }
 }
 
@@ -441,6 +483,8 @@ void AliDecayerPythia::ForceHadronicD(Int_t optUse4Bodies)
   ForceParticleDecay(iLambda_1520, 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] = 
@@ -486,10 +530,21 @@ 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;
       decayP3[1][iDau]=-1;
+      decayP4[1][iDau]=-1; 
     }
   }  
   
@@ -539,6 +594,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
+
              ))
          
          {
@@ -584,7 +646,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, Int_t* products, Int_t* mult, Int_t npart, Bool_t flag)
 {
 //
 //  Force decay of particle into products with multiplicity mult
@@ -608,9 +670,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);
        }
@@ -699,8 +761,36 @@ 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(){
+  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)
 {
 // Get branching ratio