Add a protection against division by 0 (which may occur when a track exit from the...

[u/mrichter/AliRoot.git] / TRD / AliTRDdigitsManager.cxx

diff --git a/TRD/AliTRDdigitsManager.cxx b/TRD/AliTRDdigitsManager.cxx
index 08d2b6bce12c66c49ed9f57f4984316956a1d3c8..6a6e339e17587193c18f5ea2039f9e185fc00b75 100644 (file)
--- a/TRD/AliTRDdigitsManager.cxx
+++ b/TRD/AliTRDdigitsManager.cxx
@@ -239,8 +239,6 @@ void AliTRDdigitsManager::CreateArrays()
       delete fDigitsParam;
     }
   fDigitsParam = new AliTRDdigitsParam();
-  fDigitsParam->SetNTimeBins(AliTRDSimParam::Instance()->GetNTimeBins());
-  fDigitsParam->SetADCbaseline(AliTRDSimParam::Instance()->GetADCbaseline());
 
 }
 
@@ -455,8 +453,8 @@ Bool_t AliTRDdigitsManager::ReadDigits(TTree * const tree)
     AliWarning(Form("Create default version of digits parameter (NTimeBin=%d).\n"
                   ,AliTRDSimParam::Instance()->GetNTimeBins()));
     fDigitsParam = new AliTRDdigitsParam();
-    fDigitsParam->SetNTimeBins(AliTRDSimParam::Instance()->GetNTimeBins());
-    fDigitsParam->SetADCbaseline(AliTRDSimParam::Instance()->GetADCbaseline());
+    fDigitsParam->SetNTimeBinsAll(AliTRDSimParam::Instance()->GetNTimeBins());
+    fDigitsParam->SetADCbaselineAll(AliTRDSimParam::Instance()->GetADCbaseline());
   }
 
   return status;
@@ -467,7 +465,7 @@ Bool_t AliTRDdigitsManager::ReadDigits(TTree * const tree)
 Bool_t AliTRDdigitsManager::WriteDigits()
 {
   //
-  // Writes out the TRD-digits and the dictionaries
+  // Writes out the TRD-digits, the dictionaries and the digitsPaam
   //
 
   if (!StoreArrayDigits())
@@ -484,6 +482,12 @@ Bool_t AliTRDdigitsManager::WriteDigits()
          return kFALSE;
        }
     }
+
+  if (!StoreDigitsParam())
+    {
+      AliError("Error while storing digitsParam\n");
+      return kFALSE;
+    }
   
   // Write the new tree to the output file
   fTree->AutoSave();