Fix for coverity

[u/mrichter/AliRoot.git] / EVGEN / AliGenPileup.cxx

diff --git a/EVGEN/AliGenPileup.cxx b/EVGEN/AliGenPileup.cxx
index 3073da705b101f7cd1c75e445ac9cffc0811e0a3..6d0925ba9aeda4c4dc3dbf308fc5ee421f37350f 100644 (file)
--- a/EVGEN/AliGenPileup.cxx
+++ b/EVGEN/AliGenPileup.cxx
@@ -61,7 +61,8 @@ ClassImp(AliGenPileup)
 AliGenPileup::AliGenPileup():
   AliGenCocktail(),
   fBCMask("bcm","3564H"),
-  fGenTrig(kFALSE)
+  fGenTrig(kFALSE),
+  fFlag(kFALSE)
 {
 // Constructor
 // The pileup time window is by default
@@ -77,7 +78,7 @@ AliGenPileup::~AliGenPileup()
 // Destructor
 }
 
-void AliGenPileup::SetGenerator(AliGenerator *generator, Float_t rate)
+void AliGenPileup::SetGenerator(AliGenerator *generator, Float_t rate, Bool_t flag)
 {
   // The method sets the geenrator to be used
   // for pileup simulation.
@@ -91,6 +92,7 @@ void AliGenPileup::SetGenerator(AliGenerator *generator, Float_t rate)
     }
   }
   AddGenerator(generator,"pileup generator",rate);
+  fFlag = flag;
 }
 
 void AliGenPileup::AddGenerator(AliGenerator *Generator,
@@ -150,16 +152,24 @@ void AliGenPileup::Generate()
   Int_t *nIntBC = new Int_t[3*AliTriggerBCMask::kNBits];
   Int_t *indexBC = new Int_t[3*AliTriggerBCMask::kNBits];
   Int_t nTotBC = 0;
-  for(Int_t iBC = 0; iBC <  AliTriggerBCMask::kNBits; iBC++) {
+  while (nTotBC == 0) {
+    for(Int_t iBC = 0; iBC <  AliTriggerBCMask::kNBits; iBC++) {
 
-    if (!fBCMask.GetMask(iBC)) continue;
+      if (!fBCMask.GetMask(iBC)) continue;
 
-    Int_t nInteractions = gRandom->Poisson(rate);
-    if (nInteractions == 0) continue;
+      //      Int_t nInteractions = gRandom->Poisson(rate);
+      Int_t nInteractions;
+      if (!fFlag) 
+       nInteractions = gRandom->Poisson(rate);
+      else 
+       nInteractions = TMath::Nint(rate) + 1;
 
-    nIntBC[nTotBC] = nInteractions;
-    indexBC[nTotBC] = iBC;
-    nTotBC++;
+      if (nInteractions == 0) continue;
+
+      nIntBC[nTotBC] = nInteractions;
+      indexBC[nTotBC] = iBC;
+      nTotBC++;
+    }
   }
 
   // Select the bunch crossing for triggered event
@@ -168,7 +178,7 @@ void AliGenPileup::Generate()
   // generated within this bc (only in case
   // the user disabled the generation of the trigger
   // interaction)
-  if (fGenTrig) nIntBC[iTrgBC]--;
+  if (!fGenTrig) nIntBC[iTrgBC]--;
 
   // Remove bunch crossings outside pileup
   // time window
@@ -208,7 +218,7 @@ void AliGenPileup::Generate()
 
   // Loop over the generated collision times, call the generator
   // and correct the partcile times in the stack
-  AliStack *stack = AliRunLoader::GetRunLoader()->Stack();
+  AliStack *stack = AliRunLoader::Instance()->Stack();
   Int_t lastpart=0;
   entry->SetFirst(lastpart);
 
@@ -234,6 +244,7 @@ void AliGenPileup::Generate()
 
       // Store the interaction header in the container of the headers
       ((AliGenEventHeader*) fHeader->GetHeaders()->Last())->SetPrimaryVertex(eventVertex);
+      ((AliGenEventHeader*) fHeader->GetHeaders()->Last())->SetInteractionTime(vTime);
     }
   }
   delete [] nIntBC;