Added AddTrackParams() method for convenience + some comments

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

diff --git a/STEER/AliHeader.cxx b/STEER/AliHeader.cxx
index 25f867190d88685e3f56e0defd30ca89a9783a8f..622eae9c55c01fbdb417eb5a35dff577d93ec700 100644 (file)
--- a/STEER/AliHeader.cxx
+++ b/STEER/AliHeader.cxx
@@ -22,39 +22,47 @@
 //     Author:
 //-----------------------------------------------------------------------
 
+
 #include <stdio.h>
+#include <TObjArray.h>
 
+#include "AliLog.h"
 #include "AliHeader.h"
+#include "AliDetectorEventHeader.h"
+#include "AliGenEventHeader.h"
+    
  
 ClassImp(AliHeader)
 
 //_______________________________________________________________________
 AliHeader::AliHeader():
-  fRun(0),
+  fRun(-1),
   fNvertex(0),
   fNprimary(0),
   fNtrack(0),
   fEvent(0),
   fEventNrInRun(0),
   fStack(0),
-  fGenHeader(0)
+  fGenHeader(0),
+  fDetHeaders(0)
 {
   //
   // Default constructor
   //
-}
+ }
 
 //_______________________________________________________________________
 AliHeader::AliHeader(const AliHeader& head):
   TObject(head),
-  fRun(0),
+  fRun(-1),
   fNvertex(0),
   fNprimary(0),
   fNtrack(0),
   fEvent(0),
   fEventNrInRun(0),
   fStack(0),
-  fGenHeader(0)
+  fGenHeader(0),
+  fDetHeaders(0)
 {
   //
   // Copy constructor
@@ -71,7 +79,8 @@ AliHeader::AliHeader(Int_t run, Int_t event):
   fEvent(event),
   fEventNrInRun(0),
   fStack(0),
-  fGenHeader(0)
+  fGenHeader(0), 
+  fDetHeaders(0) 
 {
   //
   // Standard constructor
@@ -87,13 +96,28 @@ AliHeader::AliHeader(Int_t run, Int_t event, Int_t evNumber):
   fEvent(event),
   fEventNrInRun(evNumber),
   fStack(0),
-  fGenHeader(0)
+  fGenHeader(0), 
+  fDetHeaders(0) 
 {
   //
   // Standard constructor
   //
 }
 
+AliHeader::~AliHeader()
+{
+    //
+    // Destructor
+    //
+    if (fDetHeaders) {
+       fDetHeaders->Delete();
+       delete fDetHeaders;
+    }
+    delete fGenHeader;
+}
+
+
+
 //_______________________________________________________________________
 void AliHeader::Reset(Int_t run, Int_t event)
 {
@@ -105,6 +129,7 @@ void AliHeader::Reset(Int_t run, Int_t event)
   fNprimary=0;
   fNtrack=0;
   fEvent=event;
+  if (fDetHeaders) fDetHeaders->Delete();
 }
 
 //_______________________________________________________________________
@@ -119,6 +144,7 @@ void AliHeader::Reset(Int_t run, Int_t event, Int_t evNumber)
   fNtrack=0;
   fEvent=event;
   fEventNrInRun=evNumber;
+  if (fDetHeaders) fDetHeaders->Clear();
 }
 
 //_______________________________________________________________________
@@ -160,6 +186,49 @@ void AliHeader::SetGenEventHeader(AliGenEventHeader* header)
     fGenHeader = header;
 }
 
+void AliHeader::AddDetectorEventHeader(AliDetectorEventHeader* header)
+{
+// Add a detector specific header
+//
+//  Create the array of headers
+    if (!fDetHeaders) fDetHeaders = new TObjArray(77);
+
+//  Some basic checks
+
+    if (!header) {
+       Warning("AddDetectorEventHeader","Detector %s tries to add empty header \n", header->GetName());
+       return;
+    }
+    
+    if (strlen(header->GetName()) == 0) {
+       Warning("AddDetectorEventHeader","Detector %s tries to add header without name \n", header->GetName());
+       return;
+    }
+    
+    TObject *mod=fDetHeaders->FindObject(header->GetName());
+    if(mod) {
+       Warning("AddDetectorEventHeader","Detector %s tries to add more than one header \n", header->GetName());
+       return;
+    }
+    
+
+//  Add the header to the list 
+    fDetHeaders->Add(header);
+}
+
+AliDetectorEventHeader* AliHeader::GetDetectorEventHeader(const char *name) const
+{
+//
+// Returns detector specific event header
+//
+    if (!fDetHeaders) {
+       Warning("GetDetectorEventHeader","There are no  detector specific headers for this event");
+       return 0x0;
+    }
+    return  dynamic_cast<AliDetectorEventHeader*>(fDetHeaders->FindObject(name)) ;
+}
+
+
 //_______________________________________________________________________
 AliGenEventHeader*  AliHeader::GenEventHeader() const
 {
@@ -170,7 +239,7 @@ AliGenEventHeader*  AliHeader::GenEventHeader() const
 //_______________________________________________________________________
 void AliHeader::Copy(TObject&) const
 {
-  Fatal("Copy","Not implemented\n");
+  AliFatal("Not implemented");
 }