Filling the trigger classes names into the ESDRun object. Initial version, could...

[u/mrichter/AliRoot.git] / STEER / AliReconstruction.cxx

diff --git a/STEER/AliReconstruction.cxx b/STEER/AliReconstruction.cxx
index 0cc8a74c04445ecb9b90ff1ad8870672d4ed6e67..0fb34bd3936e22d0fa728b0a71fb8e8809dfe57c 100644 (file)
--- a/STEER/AliReconstruction.cxx
+++ b/STEER/AliReconstruction.cxx
@@ -168,6 +168,8 @@
 #include "AliAlignObj.h"
 
 #include "AliCentralTrigger.h"
+#include "AliTriggerConfiguration.h"
+#include "AliTriggerClass.h"
 #include "AliCTPRawStream.h"
 
 #include "AliAODEvent.h"
@@ -1646,6 +1648,8 @@ Bool_t AliReconstruction::FillTriggerESD(AliESDEvent*& esd)
   
   AliInfo("Filling trigger information into the ESD");
 
+  AliCentralTrigger *aCTP = NULL;
+
   if (fRawReader) {
     AliCTPRawStream input(fRawReader);
     if (!input.Next()) {
@@ -1654,6 +1658,13 @@ Bool_t AliReconstruction::FillTriggerESD(AliESDEvent*& esd)
     }
     esd->SetTriggerMask(input.GetClassMask());
     esd->SetTriggerCluster(input.GetClusterMask());
+
+    aCTP = new AliCentralTrigger();
+    TString configstr("");
+    if (!aCTP->LoadConfiguration(configstr)) { // Load CTP config from OCDB
+      AliError("No trigger configuration found in OCDB! The trigger classes information will no be stored in ESD!");
+      return kFALSE;
+    }
   }
   else {
     AliRunLoader *runloader = AliRunLoader::GetRunLoader();
@@ -1674,6 +1685,21 @@ Bool_t AliReconstruction::FillTriggerESD(AliESDEvent*& esd)
     }
   }
 
+  // Now fill the trigger class names into AliESDRun object
+  AliTriggerConfiguration *config = aCTP->GetConfiguration();
+  if (!config) {
+    AliError("No trigger configuration has been found! The trigger classes information will no be stored in ESD!");
+    return kFALSE;
+  }
+
+  const TObjArray& classesArray = config->GetClasses();
+  Int_t nclasses = classesArray.GetEntriesFast();
+  for( Int_t j=0; j<nclasses; j++ ) {
+    AliTriggerClass* trclass = (AliTriggerClass*)classesArray.At( j );
+    Int_t trindex = (Int_t)TMath::Log2(trclass->GetMask());
+    esd->SetTriggerClass(trclass->GetName(),trindex);
+  }
+
   return kTRUE;
 }