Possibility to create an independent par file containing the new analysis framework...
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTask.cxx
index 68cd15b927afd758ca97609210035473a830ab38..ed917c59336828f08d3e934716ae9e720c3d1707 100644 (file)
 //                                    AliAnalysisDataContainer *cont)
 // To connect a slot to a data container, the data types declared by both must
 // match.
+//
+// The method Init will be called once per session at the moment when the data is
+// available at all input slots. 
+// The method Init() has to be overloaded by the derived class in order to:
+// 1. Define objects that should be created only once per session (e.g. output
+//    histograms)
+// 2. Set the branch address or connect to a branch address in case the input
+//    slots are connected to trees.
+//
+// Example:
+// MyAnalysisTask::Init(Option_t *)
+// {
+//    if (!fHist1) fHist1 = new TH1F("h1", ....);
+//    if (!fHist2) fHist2 = new TH1F("h2", ....);
+//    fESD = GetBranchAddress(islot=0, "ESD");
+//    if (!fESD) SetBranchAddress(islot=0, "ESD", &fESD);
+// }
+//
+// The method Terminate() will be called by the framework once at the end of
+// data processing. Overload this if needed.
+//
 //==============================================================================
 
+#include "Riostream.h"
+
 #include "TClass.h"
 
-#include "AliLog.h"
+//#include "AliLog.h"
 #include "AliAnalysisTask.h"
 #include "AliAnalysisDataSlot.h"
 #include "AliAnalysisDataContainer.h"
@@ -58,39 +81,48 @@ ClassImp(AliAnalysisTask)
 
 //______________________________________________________________________________
 AliAnalysisTask::AliAnalysisTask()
+                :fReady(kFALSE),
+                 fInitialized(kFALSE),
+                 fNinputs(0),
+                 fNoutputs(0),
+                 fOutputReady(NULL),
+                 fPublishedData(NULL),
+                 fInputs(NULL),
+                 fOutputs(NULL)
 {
 // Default constructor.
-   fReady       = kFALSE;
-   fNinputs     = 0;
-   fNoutputs    = 0;
-   fOutputReady = 0;
-   fPublishedData = 0;
-   fInputs      = 0;
-   fOutputs     = 0;
 }
 
 //______________________________________________________________________________
 AliAnalysisTask::AliAnalysisTask(const char *name, const char *title)
-                :TTask(name,title)
+                :TTask(name,title),
+                 fReady(kFALSE),
+                 fInitialized(kFALSE),
+                 fNinputs(0),
+                 fNoutputs(0),
+                 fOutputReady(NULL),
+                 fPublishedData(NULL),
+                 fInputs(NULL),
+                 fOutputs(NULL)                 
 {
-// Named constructor.
-   fReady       = kFALSE;
-   fNinputs     = 0;
-   fNoutputs    = 0;
-   fOutputReady = 0;
-   fPublishedData = 0;
+// Constructor.
    fInputs      = new TObjArray(2);
    fOutputs     = new TObjArray(2);
 }
 
 //______________________________________________________________________________
 AliAnalysisTask::AliAnalysisTask(const AliAnalysisTask &task)
-                :TTask(task)
+                :TTask(task),
+                 fReady(task.fReady),
+                 fInitialized(task.fInitialized),
+                 fNinputs(task.fNinputs),
+                 fNoutputs(task.fNoutputs),                 
+                 fOutputReady(NULL),
+                 fPublishedData(NULL),
+                 fInputs(NULL),
+                 fOutputs(NULL)                 
 {
 // Copy ctor.
-   fReady       = task.IsReady();
-   fNinputs     = task.GetNinputs();
-   fNoutputs    = task.GetNoutputs();
    fInputs      = new TObjArray((fNinputs)?fNinputs:2);
    fOutputs     = new TObjArray((fNoutputs)?fNoutputs:2);
    fPublishedData = 0;
@@ -115,22 +147,22 @@ AliAnalysisTask::~AliAnalysisTask()
 AliAnalysisTask& AliAnalysisTask::operator=(const AliAnalysisTask& task)
 {
 // Assignment
-   if (&task != this) {
-      TTask::operator=(task);
-      fReady       = task.IsReady();
-      fNinputs     = task.GetNinputs();
-      fNoutputs    = task.GetNoutputs();
-      fInputs      = new TObjArray((fNinputs)?fNinputs:2);
-      fOutputs     = new TObjArray((fNoutputs)?fNoutputs:2);
-      fPublishedData = 0;
-      Int_t i;
-      for (i=0; i<fNinputs; i++) fInputs->AddAt(task.GetInputSlot(i),i);
-      fOutputReady = new Bool_t[(fNoutputs)?fNoutputs:2];
-      for (i=0; i<fNoutputs; i++) {
-         fOutputReady[i] = IsOutputReady(i);
-         fOutputs->AddAt(task.GetOutputSlot(i),i);
-      }         
-   }
+   if (&task == this) return *this;
+   TTask::operator=(task);
+   fReady       = task.IsReady();
+   fInitialized = task.IsInitialized();
+   fNinputs     = task.GetNinputs();
+   fNoutputs    = task.GetNoutputs();
+   fInputs      = new TObjArray((fNinputs)?fNinputs:2);
+   fOutputs     = new TObjArray((fNoutputs)?fNoutputs:2);
+   fPublishedData = 0;
+   Int_t i;
+   for (i=0; i<fNinputs; i++) fInputs->AddAt(new AliAnalysisDataSlot(*task.GetInputSlot(i)),i);
+   fOutputReady = new Bool_t[(fNoutputs)?fNoutputs:2];
+   for (i=0; i<fNoutputs; i++) {
+      fOutputReady[i] = IsOutputReady(i);
+      fOutputs->AddAt(new AliAnalysisDataSlot(*task.GetOutputSlot(i)),i);
+   }         
    return *this;
 }
 
@@ -145,7 +177,8 @@ Bool_t AliAnalysisTask::AreSlotsConnected()
    for (i=0; i<fNinputs; i++) {
       slot = (AliAnalysisDataSlot*)fInputs->At(i);
       if (!slot) {
-         AliError(Form("Input slot %i of task %s not defined !",i,GetName()));
+       cout<<"Input slot "<<i<<" of task "<<GetName()<<" not defined !"<<endl;
+       //AliError(Form("Input slot %i of task %s not defined !",i,GetName()));
          return kFALSE;
       }   
       if (!slot->IsConnected()) return kFALSE;
@@ -153,7 +186,8 @@ Bool_t AliAnalysisTask::AreSlotsConnected()
    for (i=0; i<fNoutputs; i++) {
       slot = (AliAnalysisDataSlot*)fOutputs->At(i);
       if (!slot) {
-         AliError(Form("Output slot %i of task %s not defined !",i,GetName()));
+       cout<<"Output slot "<<i<<" of task "<<GetName()<<" not defined !"<<endl;
+       //AliError(Form("Output slot %i of task %s not defined !",i,GetName()));
          return kFALSE;
       }   
       if (!slot->IsConnected()) return kFALSE;
@@ -163,11 +197,12 @@ Bool_t AliAnalysisTask::AreSlotsConnected()
 }
 
 //______________________________________________________________________________
-void AliAnalysisTask::CheckNotify()
+void AliAnalysisTask::CheckNotify(Bool_t init)
 {
 // Check if data is available from all inputs. Change the status of the task
 // accordingly. This method is called automatically for all tasks connected
 // to a container where the data was published.
+   if (init) fInitialized = kFALSE;
    for (Int_t islot=0; islot<fNinputs; islot++) {
       if (!GetInputData(islot)) {
          SetActive(kFALSE);
@@ -175,6 +210,11 @@ void AliAnalysisTask::CheckNotify()
       }   
    }   
    SetActive(kTRUE);
+   if (fInitialized) return;
+   TDirectory *cursav = gDirectory;
+   Init();
+   if (cursav) cursav->cd();
+   fInitialized = kTRUE;
 }
 
 //______________________________________________________________________________
@@ -183,13 +223,14 @@ Bool_t AliAnalysisTask::ConnectInput(Int_t islot, AliAnalysisDataContainer *cont
 // Connect an input slot to a data container.
    AliAnalysisDataSlot *input = GetInputSlot(islot);
    if (!input) {
-      AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
+     cout<<"Input slot "<<islot<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
       return kFALSE;
    }
    // Check type matching          
    if (!input->GetType()->InheritsFrom(cont->GetType())) {
-      AliError(Form("Data type %s for input %i of task %s not matching container %s of type %s",
-               input->GetType()->GetName(), islot, GetName(), cont->GetName(), cont->GetType()->GetName()));
+     cout<<"Data type "<<input->GetType()->GetName()<<" for input "<<islot<<" of task "<<GetName()<<" not matching container "<<cont->GetName()<<" of type "<<cont->GetType()->GetName()<<endl;
+     //AliError(Form("Data type %s for input %i of task %s not matching container %s of type %s",input->GetType()->GetName(), islot, GetName(), cont->GetName(), cont->GetType()->GetName()));
       return kFALSE;
    }  
    // Connect the slot to the container as input          
@@ -206,13 +247,14 @@ Bool_t AliAnalysisTask::ConnectOutput(Int_t islot, AliAnalysisDataContainer *con
 // Connect an output slot to a data container.
    AliAnalysisDataSlot *output = GetOutputSlot(islot);
    if (!output) {
-      AliError(Form("Output slot %i not defined for analysis task %s", islot, GetName()));
+     cout<<"Output slot "<<islot<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Output slot %i not defined for analysis task %s", islot, GetName()));
       return kFALSE;
    }
    // Check type matching          
    if (!output->GetType()->InheritsFrom(cont->GetType())) {
-      AliError(Form("Data type %s for output %i of task %s not matching container %s of type %s",
-               output->GetType()->GetName(), islot, GetName(), cont->GetName(), cont->GetType()->GetName()));
+     cout<<"Data type "<<output->GetType()->GetName()<<" for output "<<islot<<" of task "<<GetName()<<" not matching container "<<cont->GetName()<<" of type "<<cont->GetType()->GetName()<<endl;
+     //AliError(Form("Data type %s for output %i of task %s not matching container %s of type %s",output->GetType()->GetName(), islot, GetName(), cont->GetName(), cont->GetType()->GetName()));
       return kFALSE;
    }            
    // Connect the slot to the container as output         
@@ -229,7 +271,7 @@ void AliAnalysisTask::DefineInput(Int_t islot, TClass *type)
 // Define an input slot and its type.
    AliAnalysisDataSlot *input = new AliAnalysisDataSlot(type, this);
    if (fNinputs<islot+1) fNinputs = islot+1;
-   fInputs->AddAt(input, islot);
+   fInputs->AddAtAndExpand(input, islot);
 }
 
 //______________________________________________________________________________
@@ -237,7 +279,8 @@ void AliAnalysisTask::DefineOutput(Int_t islot, TClass *type)
 {
 // Define an output slot and its type.
    if (islot<0) {
-      AliError(Form("Cannot define negative output slot number for task %s", GetName()));
+     cout<<"Cannot define negative output slot number for task "<<GetName()<<endl;
+     //AliError(Form("Cannot define negative output slot number for task %s", GetName()));
       return;
    }   
    AliAnalysisDataSlot *output = new AliAnalysisDataSlot(type, this);
@@ -247,7 +290,7 @@ void AliAnalysisTask::DefineOutput(Int_t islot, TClass *type)
       fOutputReady = new Bool_t[fNoutputs];
       memset(fOutputReady, 0, fNoutputs*sizeof(Bool_t));
    } 
-   fOutputs->AddAt(output, islot);
+   fOutputs->AddAtAndExpand(output, islot);
 }
 
 //______________________________________________________________________________
@@ -256,7 +299,8 @@ TClass *AliAnalysisTask::GetInputType(Int_t islot) const
 // Retreive type of a given input slot.
    AliAnalysisDataSlot *input = GetInputSlot(islot);
    if (!input) {
-      AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
+     cout<<"Input slot "<<islot<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
       return NULL;
    }
    return (input->GetType());
@@ -268,7 +312,8 @@ TClass *AliAnalysisTask::GetOutputType(Int_t islot) const
 // Retreive type of a given output slot.
    AliAnalysisDataSlot *output = GetOutputSlot(islot);
    if (!output) {
-      AliError(Form("Output slot %i not defined for analysis task %s", islot, GetName()));
+     cout<<"Output slot "<<islot<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Output slot %i not defined for analysis task %s", islot, GetName()));
       return NULL;
    }
    return (output->GetType());
@@ -281,12 +326,48 @@ TObject *AliAnalysisTask::GetInputData(Int_t islot) const
 // the object has to be statically cast to the appropriate type.
    AliAnalysisDataSlot *input = GetInputSlot(islot);
    if (!input) {
-      AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
+     cout<<"Input slot "<<islot<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Input slot %i not defined for analysis task %s", islot, GetName()));
       return NULL;
    }
    return (input->GetData()); 
 }
 
+//______________________________________________________________________________
+char *AliAnalysisTask::GetBranchAddress(Int_t islot, const char *branch) const
+{
+// Check if a branch with a given name from the specified input is connected
+// to some address. Call this in Init() before trying to call SetBranchAddress()
+// since the adress may be set by other task.
+   return (char *)GetInputSlot(islot)->GetBranchAddress(branch);
+}
+
+//______________________________________________________________________________
+Bool_t AliAnalysisTask::SetBranchAddress(Int_t islot, const char *branch, void *address) const
+{
+// Connect an object address to a branch of the specified input.
+   return GetInputSlot(islot)->SetBranchAddress(branch, address);
+}   
+
+//______________________________________________________________________________
+void AliAnalysisTask::Init(Option_t *)
+{
+// Branch address initialization.
+}
+
+//______________________________________________________________________________
+void AliAnalysisTask::Terminate(Option_t *)
+{
+// Method called by the framework at the end of data processing.
+}
+
+//______________________________________________________________________________
+void AliAnalysisTask::OpenFile(Int_t iout, const char *name, Option_t *option) const
+{
+// Set data at output iout to be written in the specified file.
+   GetOutputSlot(iout)->GetContainer()->OpenFile(name, option);
+}   
+
 //______________________________________________________________________________
 Bool_t AliAnalysisTask::PostData(Int_t iout, TObject *data, Option_t *option)
 {
@@ -297,11 +378,13 @@ Bool_t AliAnalysisTask::PostData(Int_t iout, TObject *data, Option_t *option)
    fPublishedData = 0;
    AliAnalysisDataSlot *output = GetOutputSlot(iout);
    if (!output) {
-      AliError(Form("Output slot %i not defined for analysis task %s", iout, GetName()));
+     cout<<"Output slot "<<iout<<" not defined for analysis task "<<GetName()<<endl;
+     //AliError(Form("Output slot %i not defined for analysis task %s", iout, GetName()));
       return kFALSE;
    }
    if (!output->IsConnected()) {
-      AliError(Form("Output slot %i of analysis task %s not connected to any data container", iout, GetName()));
+     cout<<"Output slot "<<iout<<" of analysis task "<<GetName()<<" not connected to any data container"<<endl;
+     //AliError(Form("Output slot %i of analysis task %s not connected to any data container", iout, GetName()));
       return kFALSE;
    }
    if (!fOutputReady) {