Adapt add macro and particle/cluster containers for the analysis on jets

[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTask.cxx

diff --git a/ANALYSIS/AliAnalysisTask.cxx b/ANALYSIS/AliAnalysisTask.cxx
index 47edc46945ae2d97e1e153579054e85c3d1a5e3d..a49b1d88b3d2e6307ba6e666fd85a953d4ace78a 100644 (file)
--- a/ANALYSIS/AliAnalysisTask.cxx
+++ b/ANALYSIS/AliAnalysisTask.cxx
@@ -100,6 +100,7 @@
 #include <Riostream.h>
 #include <TFile.h>
 #include <TClass.h>
+#include <TCollection.h>
 #include <TTree.h>
 #include <TROOT.h>
 
@@ -254,6 +255,48 @@ void AliAnalysisTask::CheckNotify(Bool_t init)
    fInitialized = kTRUE;
 }
 
+//______________________________________________________________________________
+Bool_t AliAnalysisTask::CheckPostData() const
+{
+// Checks if data was posted to all outputs defined by the task. If task does
+// not have output slots this returns always kTRUE.
+   Bool_t dataPosted = kTRUE;
+   AliAnalysisDataContainer *coutput;
+   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+   for (Int_t islot=0; islot<fNoutputs; islot++) {
+      coutput = GetOutputSlot(islot)->GetContainer();
+      if (!coutput) continue;
+      if (coutput->IsExchange() || !mgr->GetOutputs()->FindObject(coutput) || coutput==mgr->GetCommonOutputContainer()) continue;
+      if (!coutput->GetData()) {
+         Error("CheckPostData", "Data not posted for slot #%d of task %s (%s)", 
+               islot, GetName(), ClassName());
+         dataPosted = kFALSE;
+      }   
+   }
+   CheckOwnership();
+   return dataPosted;
+}
+
+//______________________________________________________________________________
+Bool_t AliAnalysisTask::CheckOwnership() const
+{
+// Check ownership of containers posted on output slots (1 level only)
+   TObject *outdata;
+   for (Int_t islot=0; islot<fNoutputs; islot++) {
+      outdata = GetOutputData(islot);
+      if (outdata && outdata->InheritsFrom(TCollection::Class())) {
+         TCollection *coll = (TCollection*)outdata;
+         if (!coll->IsOwner()) {
+            Error("CheckOwnership","####### IMPORTANT! ####### \n\n\n\
+                Task %s (%s) posts a container that is not owner at output #%d. This may apply for other embedded containers. \n\n\
+                ####### FIX YOUR CODE, THIS WILL PRODUCE A FATAL ERROR IN FUTURE! ##########", GetName(), ClassName(), islot);
+             return kFALSE;   
+         }
+      }
+   }
+   return kTRUE;
+}
+      
 //______________________________________________________________________________
 Bool_t AliAnalysisTask::ConnectInput(Int_t islot, AliAnalysisDataContainer *cont)
 {
@@ -532,6 +575,21 @@ void AliAnalysisTask::SetUsed(Bool_t flag)
    }
 }   
 
+//______________________________________________________________________________
+void AliAnalysisTask::Reset()
+{
+// Clear activity flag. Reset data for exchange containers.
+   fActive = kFALSE;
+   fHasExecuted = kFALSE;
+   // Call PostData(islot, 0) for all slots connected to exchange containers
+   Int_t islot;
+   AliAnalysisDataContainer *cont;
+   for (islot=0; islot<fNinputs; islot++) {
+      cont = GetInputSlot(islot)->GetContainer();
+      if (cont && cont->IsExchange()) cont->Reset();
+   }   
+}
+   
 //______________________________________________________________________________
 Bool_t AliAnalysisTask::CheckCircularDeps()
 {
@@ -549,23 +607,39 @@ Bool_t AliAnalysisTask::CheckCircularDeps()
    SetChecked(kFALSE);
    return kFALSE;
 }   
-   
+
+//______________________________________________________________________________
+Bool_t AliAnalysisTask::ProducersTouched() const
+{
+// Check if all producer containers are in the "touched" state.
+   Int_t islot;
+   AliAnalysisDataContainer *cont;
+   for (islot=0; islot<fNinputs; islot++) {
+      cont = GetInputSlot(islot)->GetContainer();
+      // Simulate the data flow so the tasks are printed only when all inputs
+      // are touched
+      if (cont && !cont->IsTouched()) return kFALSE;
+   }
+   return kTRUE;
+}
+
 //______________________________________________________________________________
 void AliAnalysisTask::PrintTask(Option_t *option, Int_t indent) const
 {
 // Print task info.
-   AliAnalysisTask *thistask = (AliAnalysisTask*)this;
+   Int_t islot;
+   AliAnalysisDataContainer *cont;
+   if (fActive) return;
+   if (!ProducersTouched()) return;
    TString opt(option);
    opt.ToLower();
    Bool_t dep = (opt.Contains("dep"))?kTRUE:kFALSE;
    TString ind;
-   Int_t islot;
-   AliAnalysisDataContainer *cont;
    for (Int_t i=0; i<indent; i++) ind += " ";
    if (!dep || (dep && IsChecked())) {
       printf("______________________________________________________________________________\n");
       printf("%s\n", Form("%stask: %s  ACTIVE=%i POST_LOOP=%i", ind.Data(), GetName(),IsActive(),IsPostEventLoop()));
-      if (dep) thistask->SetChecked(kFALSE);
+      if (dep) const_cast<AliAnalysisTask*>(this)->SetChecked(kFALSE);
       else {
          for (islot=0; islot<fNinputs; islot++) {
             printf("%s", Form("%s   INPUT #%i: %s <- ",ind.Data(),islot, GetInputType(islot)->GetName()));
@@ -581,8 +655,9 @@ void AliAnalysisTask::PrintTask(Option_t *option, Int_t indent) const
          }            
       }
    }
+   ((AliAnalysisTask*)this)->SetActive(kTRUE);
    PrintContainers(option, indent+3);
-   if (!fBranchNames.IsNull()) printf("Requested branches:   %s\n", fBranchNames.Data());
+   if (!fBranchNames.IsNull()) printf("%sRequested branches:   %s\n", ind.Data(), fBranchNames.Data());
 }      
 
 //______________________________________________________________________________