Possibility to create an independent par file containing the new analysis framework...
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTask.cxx
index a61b2d8d1d614e86d0342f3395ec652f339a54c8..ed917c59336828f08d3e934716ae9e720c3d1707 100644 (file)
 //                                    AliAnalysisDataContainer *cont)
 // To connect a slot to a data container, the data types declared by both must
 // match.
 //                                    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 "TClass.h"
 
-#include "AliLog.h"
+//#include "AliLog.h"
 #include "AliAnalysisTask.h"
 #include "AliAnalysisDataSlot.h"
 #include "AliAnalysisDataContainer.h"
 #include "AliAnalysisTask.h"
 #include "AliAnalysisDataSlot.h"
 #include "AliAnalysisDataContainer.h"
@@ -59,6 +82,7 @@ ClassImp(AliAnalysisTask)
 //______________________________________________________________________________
 AliAnalysisTask::AliAnalysisTask()
                 :fReady(kFALSE),
 //______________________________________________________________________________
 AliAnalysisTask::AliAnalysisTask()
                 :fReady(kFALSE),
+                 fInitialized(kFALSE),
                  fNinputs(0),
                  fNoutputs(0),
                  fOutputReady(NULL),
                  fNinputs(0),
                  fNoutputs(0),
                  fOutputReady(NULL),
@@ -73,6 +97,7 @@ AliAnalysisTask::AliAnalysisTask()
 AliAnalysisTask::AliAnalysisTask(const char *name, const char *title)
                 :TTask(name,title),
                  fReady(kFALSE),
 AliAnalysisTask::AliAnalysisTask(const char *name, const char *title)
                 :TTask(name,title),
                  fReady(kFALSE),
+                 fInitialized(kFALSE),
                  fNinputs(0),
                  fNoutputs(0),
                  fOutputReady(NULL),
                  fNinputs(0),
                  fNoutputs(0),
                  fOutputReady(NULL),
@@ -89,6 +114,7 @@ AliAnalysisTask::AliAnalysisTask(const char *name, const char *title)
 AliAnalysisTask::AliAnalysisTask(const AliAnalysisTask &task)
                 :TTask(task),
                  fReady(task.fReady),
 AliAnalysisTask::AliAnalysisTask(const AliAnalysisTask &task)
                 :TTask(task),
                  fReady(task.fReady),
+                 fInitialized(task.fInitialized),
                  fNinputs(task.fNinputs),
                  fNoutputs(task.fNoutputs),                 
                  fOutputReady(NULL),
                  fNinputs(task.fNinputs),
                  fNoutputs(task.fNoutputs),                 
                  fOutputReady(NULL),
@@ -124,6 +150,7 @@ AliAnalysisTask& AliAnalysisTask::operator=(const AliAnalysisTask& task)
    if (&task == this) return *this;
    TTask::operator=(task);
    fReady       = task.IsReady();
    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);
    fNinputs     = task.GetNinputs();
    fNoutputs    = task.GetNoutputs();
    fInputs      = new TObjArray((fNinputs)?fNinputs:2);
@@ -150,7 +177,8 @@ Bool_t AliAnalysisTask::AreSlotsConnected()
    for (i=0; i<fNinputs; i++) {
       slot = (AliAnalysisDataSlot*)fInputs->At(i);
       if (!slot) {
    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;
          return kFALSE;
       }   
       if (!slot->IsConnected()) return kFALSE;
@@ -158,7 +186,8 @@ Bool_t AliAnalysisTask::AreSlotsConnected()
    for (i=0; i<fNoutputs; i++) {
       slot = (AliAnalysisDataSlot*)fOutputs->At(i);
       if (!slot) {
    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;
          return kFALSE;
       }   
       if (!slot->IsConnected()) return kFALSE;
@@ -168,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.
 {
 // 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);
    for (Int_t islot=0; islot<fNinputs; islot++) {
       if (!GetInputData(islot)) {
          SetActive(kFALSE);
@@ -180,6 +210,11 @@ void AliAnalysisTask::CheckNotify()
       }   
    }   
    SetActive(kTRUE);
       }   
    }   
    SetActive(kTRUE);
+   if (fInitialized) return;
+   TDirectory *cursav = gDirectory;
+   Init();
+   if (cursav) cursav->cd();
+   fInitialized = kTRUE;
 }
 
 //______________________________________________________________________________
 }
 
 //______________________________________________________________________________
@@ -188,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) {
 // 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())) {
       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          
       return kFALSE;
    }  
    // Connect the slot to the container as input          
@@ -211,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) {
 // 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())) {
       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         
       return kFALSE;
    }            
    // Connect the slot to the container as output         
@@ -234,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;
 // 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);
 }
 
 //______________________________________________________________________________
 }
 
 //______________________________________________________________________________
@@ -242,7 +279,8 @@ void AliAnalysisTask::DefineOutput(Int_t islot, TClass *type)
 {
 // Define an output slot and its type.
    if (islot<0) {
 {
 // 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);
       return;
    }   
    AliAnalysisDataSlot *output = new AliAnalysisDataSlot(type, this);
@@ -252,7 +290,7 @@ void AliAnalysisTask::DefineOutput(Int_t islot, TClass *type)
       fOutputReady = new Bool_t[fNoutputs];
       memset(fOutputReady, 0, fNoutputs*sizeof(Bool_t));
    } 
       fOutputReady = new Bool_t[fNoutputs];
       memset(fOutputReady, 0, fNoutputs*sizeof(Bool_t));
    } 
-   fOutputs->AddAt(output, islot);
+   fOutputs->AddAtAndExpand(output, islot);
 }
 
 //______________________________________________________________________________
 }
 
 //______________________________________________________________________________
@@ -261,7 +299,8 @@ TClass *AliAnalysisTask::GetInputType(Int_t islot) const
 // Retreive type of a given input slot.
    AliAnalysisDataSlot *input = GetInputSlot(islot);
    if (!input) {
 // 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());
       return NULL;
    }
    return (input->GetType());
@@ -273,7 +312,8 @@ TClass *AliAnalysisTask::GetOutputType(Int_t islot) const
 // Retreive type of a given output slot.
    AliAnalysisDataSlot *output = GetOutputSlot(islot);
    if (!output) {
 // 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());
       return NULL;
    }
    return (output->GetType());
@@ -286,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) {
 // 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()); 
 }
 
       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)
 {
 //______________________________________________________________________________
 Bool_t AliAnalysisTask::PostData(Int_t iout, TObject *data, Option_t *option)
 {
@@ -302,11 +378,13 @@ Bool_t AliAnalysisTask::PostData(Int_t iout, TObject *data, Option_t *option)
    fPublishedData = 0;
    AliAnalysisDataSlot *output = GetOutputSlot(iout);
    if (!output) {
    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()) {
       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) {
       return kFALSE;
    }
    if (!fOutputReady) {