]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ANALYSIS/AliAnalysisManager.cxx
Added AddTrackParams() method for convenience + some comments
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisManager.cxx
index 15d3b1ed7f27efec747671de246d73855f0af70a..04531c7fc7280eccd9b43fe446c85825cfc8ee24 100644 (file)
@@ -29,6 +29,7 @@
 
 #include <TClass.h>
 #include <TFile.h>
+#include <TKey.h>
 #include <TMethodCall.h>
 #include <TChain.h>
 #include <TSystem.h>
@@ -36,6 +37,7 @@
 #include <TCanvas.h>
 
 #include "AliAnalysisSelector.h"
+#include "AliAnalysisGrid.h"
 #include "AliAnalysisTask.h"
 #include "AliAnalysisDataContainer.h"
 #include "AliAnalysisDataSlot.h"
@@ -55,7 +57,7 @@ AliAnalysisManager::AliAnalysisManager(const char *name, const char *title)
                     fInputEventHandler(NULL),
                     fOutputEventHandler(NULL),
                     fMCtruthEventHandler(NULL),
-                   fEventPool(NULL),
+                    fEventPool(NULL),
                     fCurrentEntry(-1),
                     fNSysInfo(0),
                     fMode(kLocalAnalysis),
@@ -68,7 +70,10 @@ AliAnalysisManager::AliAnalysisManager(const char *name, const char *title)
                     fContainers(NULL),
                     fInputs(NULL),
                     fOutputs(NULL),
-                    fSelector(NULL)
+                    fCommonInput(NULL),
+                    fCommonOutput(NULL),
+                    fSelector(NULL),
+                    fGridHandler(NULL)
 {
 // Default constructor.
    fgAnalysisManager = this;
@@ -101,7 +106,10 @@ AliAnalysisManager::AliAnalysisManager(const AliAnalysisManager& other)
                     fContainers(NULL),
                     fInputs(NULL),
                     fOutputs(NULL),
-                    fSelector(NULL)
+                    fCommonInput(NULL),
+                    fCommonOutput(NULL),
+                    fSelector(NULL),
+                    fGridHandler(NULL)
 {
 // Copy constructor.
    fTasks      = new TObjArray(*other.fTasks);
@@ -135,7 +143,10 @@ AliAnalysisManager& AliAnalysisManager::operator=(const AliAnalysisManager& othe
       fContainers = new TObjArray(*other.fContainers);
       fInputs     = new TObjArray(*other.fInputs);
       fOutputs    = new TObjArray(*other.fOutputs);
+      fCommonInput = NULL;
+      fCommonOutput = NULL;
       fSelector   = NULL;
+      fGridHandler = NULL;
       fgAnalysisManager = this;
    }
    return *this;
@@ -151,6 +162,7 @@ AliAnalysisManager::~AliAnalysisManager()
    if (fContainers) {fContainers->Delete(); delete fContainers;}
    if (fInputs) delete fInputs;
    if (fOutputs) delete fOutputs;
+   if (fGridHandler) delete fGridHandler;
    if (fgAnalysisManager==this) fgAnalysisManager = NULL;
 }
 
@@ -164,60 +176,79 @@ Int_t AliAnalysisManager::GetEntry(Long64_t entry, Int_t getall)
 }
    
 //______________________________________________________________________________
-void AliAnalysisManager::Init(TTree *tree)
+Bool_t AliAnalysisManager::Init(TTree *tree)
 {
   // The Init() function is called when the selector needs to initialize
   // a new tree or chain. Typically here the branch addresses of the tree
   // will be set. It is normaly not necessary to make changes to the
   // generated code, but the routine can be extended by the user if needed.
   // Init() will be called many times when running with PROOF.
-   if (!tree) return;
+   Bool_t init = kFALSE;
+   if (!tree) return kFALSE; // Should not happen - protected in selector caller
    if (fDebug > 0) {
       printf("->AliAnalysisManager::Init(%s)\n", tree->GetName());
    }
-
    // Call InitTree of EventHandler
    if (fOutputEventHandler) {
       if (fMode == kProofAnalysis) {
-         fOutputEventHandler->Init(0x0, "proof");
+         init = fOutputEventHandler->Init(0x0, "proof");
       } else {
-         fOutputEventHandler->Init(0x0, "local");
+         init = fOutputEventHandler->Init(0x0, "local");
       }
+      if (!init) {
+         Error("Init", "Output event handler failed to initialize");
+         return kFALSE;
+      }         
    }
-
+   
    if (fInputEventHandler) {
       if (fMode == kProofAnalysis) {
-         fInputEventHandler->Init(tree, "proof");
+         init = fInputEventHandler->Init(tree, "proof");
       } else {
-         fInputEventHandler->Init(tree, "local");
+         init = fInputEventHandler->Init(tree, "local");
       }
+      if (!init) {
+         Error("Init", "Input event handler failed to initialize tree"); 
+         return kFALSE;
+      }         
    } else {
       // If no input event handler we need to get the tree once
       // for the chain
-      if(!tree->GetTree()) tree->LoadTree(0);
+      if(!tree->GetTree()) {
+         Long64_t readEntry = tree->LoadTree(0);
+         if (readEntry == -2) {
+            Error("Init", "Input tree has no entry. Aborting");
+            return kFALSE;
+         }
+      }   
    }
-   
 
    if (fMCtruthEventHandler) {
       if (fMode == kProofAnalysis) {
-         fMCtruthEventHandler->Init(0x0, "proof");
+         init = fMCtruthEventHandler->Init(0x0, "proof");
       } else {
-         fMCtruthEventHandler->Init(0x0, "local");
+         init = fMCtruthEventHandler->Init(0x0, "local");
       }
+      if (!init) {
+         Error("Init", "MC event handler failed to initialize"); 
+         return kFALSE;
+      }         
    }
 
    if (!fInitOK) InitAnalysis();
-   if (!fInitOK) return;
+   if (!fInitOK) return kFALSE;
    fTree = tree;
-   AliAnalysisDataContainer *top = (AliAnalysisDataContainer*)fInputs->At(0);
+   AliAnalysisDataContainer *top = fCommonInput;
+   if (!top) top = (AliAnalysisDataContainer*)fInputs->At(0);
    if (!top) {
       Error("Init","No top input container !");
-      return;
+      return kFALSE;
    }
    top->SetData(tree);
    if (fDebug > 0) {
       printf("<-AliAnalysisManager::Init(%s)\n", tree->GetName());
    }
+   return kTRUE;
 }
 
 //______________________________________________________________________________
@@ -228,44 +259,76 @@ void AliAnalysisManager::SlaveBegin(TTree *tree)
   // The tree argument is deprecated (on PROOF 0 is passed).
    if (fDebug > 0) printf("->AliAnalysisManager::SlaveBegin()\n");
    static Bool_t isCalled = kFALSE;
+   Bool_t init = kFALSE;
+   Bool_t initOK = kTRUE;
+   TString msg;
+   TDirectory *curdir = gDirectory;
    // Call SlaveBegin only once in case of mixing
    if (isCalled && fMode==kMixingAnalysis) return;
    // Call Init of EventHandler
    if (fOutputEventHandler) {
       if (fMode == kProofAnalysis) {
-         fOutputEventHandler->Init("proof");
+         TIter nextout(fOutputs);
+         AliAnalysisDataContainer *c_aod;
+         while ((c_aod=(AliAnalysisDataContainer*)nextout())) if (!strcmp(c_aod->GetFileName(),"default")) break;
+         if (c_aod && c_aod->IsSpecialOutput()) {
+            // Merging via files
+            if (fDebug > 1) printf("   Initializing special output file %s...\n", fOutputEventHandler->GetOutputFileName());
+            OpenProofFile(fOutputEventHandler->GetOutputFileName(), "RECREATE");
+            c_aod->SetFile(gFile);
+            init = fOutputEventHandler->Init("proofspecial");
+            if (!init) msg = "Failed to initialize output handler on worker using special proof output";
+         } else {
+            // Merging in memory
+            init = fOutputEventHandler->Init("proof");
+            if (!init) msg = "Failed to initialize output handler on worker";
+         }   
       } else {
-         fOutputEventHandler->Init("local");
+         init = fOutputEventHandler->Init("local");
+         if (!init) msg = "Failed to initialize output handler on worker";
       }
+      initOK &= init;
+      if (!fSelector) Error("SlaveBegin", "Selector not set");
+      else if (!init) {fSelector->Abort(msg); fSelector->SetStatus(-1);}
    }
 
    if (fInputEventHandler) {
       fInputEventHandler->SetInputTree(tree);
       if (fMode == kProofAnalysis) {
-         fInputEventHandler->Init("proof");
+         init = fInputEventHandler->Init("proof");
+         if (!init) msg = "Failed to initialize input handler on worker";
       } else {
-         fInputEventHandler->Init("local");
+         init = fInputEventHandler->Init("local");
+         if (!init) msg = "Failed to initialize input handler";
       }
+      initOK &= init;
+      if (!fSelector) Error("SlaveBegin", "Selector not set");      
+      else if (!init) {fSelector->Abort(msg); fSelector->SetStatus(-1);}
    }
 
    if (fMCtruthEventHandler) {
       if (fMode == kProofAnalysis) {
-         fMCtruthEventHandler->Init("proof");
+         init = fMCtruthEventHandler->Init("proof");
+         if (!init) msg = "Failed to initialize MC handler on worker";
       } else {
-         fMCtruthEventHandler->Init("local");
+         init = fMCtruthEventHandler->Init("local");
+         if (!init) msg = "Failed to initialize MC handler";
       }
+      initOK &= init;
+      if (!fSelector) Error("SlaveBegin", "Selector not set");      
+      else if (!init) {fSelector->Abort(msg); fSelector->SetStatus(-1);}
    }
-
+   if (curdir) curdir->cd();
+   isCalled = kTRUE;
+   if (!initOK) return;   
    TIter next(fTasks);
    AliAnalysisTask *task;
    // Call CreateOutputObjects for all tasks
    while ((task=(AliAnalysisTask*)next())) {
-      TDirectory *curdir = gDirectory;
+      curdir = gDirectory;
       task->CreateOutputObjects();
       if (curdir) curdir->cd();
    }
-   isCalled = kTRUE;
-
    if (fDebug > 0) printf("<-AliAnalysisManager::SlaveBegin()\n");
 }
 
@@ -277,10 +340,8 @@ Bool_t AliAnalysisManager::Notify()
    // is started when using PROOF. It is normaly not necessary to make changes
    // to the generated code, but the routine can be extended by the
    // user if needed. The return value is currently not used.
-   if (!fTree) {
-      Error("Notify","No current tree.");
-      return kFALSE;
-   }   
+   if (!fTree) return kFALSE;
+
    TFile *curfile = fTree->GetCurrentFile();
    if (!curfile) {
       Error("Notify","No current file");
@@ -371,22 +432,25 @@ void AliAnalysisManager::PackOutput(TList *target)
    if (fMode == kProofAnalysis) {
       TIter next(fOutputs);
       AliAnalysisDataContainer *output;
+      Bool_t isManagedByHandler = kFALSE;
       while ((output=(AliAnalysisDataContainer*)next())) {
          // Do not consider outputs of post event loop tasks
+         isManagedByHandler = kFALSE;
          if (output->GetProducer()->IsPostEventLoop()) continue;
+         const char *filename = output->GetFileName();
+         if (!(strcmp(filename, "default")) && fOutputEventHandler) {
+            isManagedByHandler = kTRUE;
+            filename = fOutputEventHandler->GetOutputFileName();
+         }
          // Check if data was posted to this container. If not, issue an error.
-         if (!output->GetData() ) {
+         if (!output->GetData() && !isManagedByHandler) {
             Error("PackOutput", "No data for output container %s. Forgot to PostData ?\n", output->GetName());
             continue;
          }   
          if (!output->IsSpecialOutput()) {
             // Normal outputs
-            const char *filename = output->GetFileName();
-            if (!(strcmp(filename, "default"))) {
-               if (fOutputEventHandler) filename = fOutputEventHandler->GetOutputFileName();
-            }      
-            if (strlen(filename)) {
-            // File resident outputs
+            if (strlen(filename) && !isManagedByHandler) {
+               // File resident outputs
                TFile *file = output->GetFile();
                // Backup current folder
                TDirectory *opwd = gDirectory;
@@ -408,7 +472,13 @@ void AliAnalysisManager::PackOutput(TList *target)
                   coll->SetName(output->GetName());
                   coll->Write(output->GetName(), TObject::kSingleKey);
                } else {
-                  output->GetData()->Write();
+                  if (output->GetData()->InheritsFrom(TTree::Class())) {
+                     TTree *tree = (TTree*)output->GetData();
+                     tree->SetDirectory(file);
+                     tree->AutoSave();
+                  } else {
+                     output->GetData()->Write();
+                  }   
                }      
                if (fDebug > 1) printf("PackOutput %s: memory merge, file resident output\n", output->GetName());
                if (fDebug > 2) {
@@ -420,17 +490,33 @@ void AliAnalysisManager::PackOutput(TList *target)
                if (opwd) opwd->cd();
             } else {
                // Memory-resident outputs   
-               if (fDebug > 1) printf("PackOutput %s: memory merge memory resident output\n", output->GetName());
+               if (fDebug > 1) printf("PackOutput %s: memory merge memory resident output\n", filename);
             }   
-            AliAnalysisDataWrapper *wrap = output->ExportData();
-            // Output wrappers must delete data after merging (AG 13/11/07)
-            wrap->SetDeleteData(kTRUE);
+            AliAnalysisDataWrapper *wrap = 0;
+            if (isManagedByHandler) {
+               wrap = new AliAnalysisDataWrapper(fOutputEventHandler->GetTree());
+               wrap->SetName(output->GetName());
+            }   
+            else                    wrap =output->ExportData();
+            // Output wrappers must NOT delete data after merging - the user owns them
+            wrap->SetDeleteData(kFALSE);
             target->Add(wrap);
-         }   
+         } else {
          // Special outputs
-         if (output->IsSpecialOutput()) {
             TDirectory *opwd = gDirectory;
             TFile *file = output->GetFile();
+            if (fDebug > 1 && file) printf("PackOutput %s: file merge, special output\n", output->GetName());
+            if (isManagedByHandler) {
+               // Terminate IO for files managed by the output handler
+               if (file) file->Write();
+               if (file && fDebug > 2) {
+                  printf("   handled file %s listing content:\n", file->GetName());
+                  file->ls();
+               }   
+               fOutputEventHandler->TerminateIO();
+               continue;
+            }   
+            
             if (!file) {
                AliAnalysisTask *producer = output->GetProducer();
                Error("PackOutput", 
@@ -447,13 +533,20 @@ void AliAnalysisManager::PackOutput(TList *target)
                coll->SetName(output->GetName());
                coll->Write(output->GetName(), TObject::kSingleKey);
             } else {
-               output->GetData()->Write();
+               if (output->GetData()->InheritsFrom(TTree::Class())) {
+                  TTree *tree = (TTree*)output->GetData();
+                  tree->SetDirectory(file);
+                  tree->AutoSave();
+               } else {
+                  output->GetData()->Write();
+               }   
             }      
             file->Clear();
             if (fDebug > 2) {
                printf("   file %s listing content:\n", output->GetFileName());
                file->ls();
-            }   
+            }
+            TString outFilename = file->GetName();
             file->Close();
             // Restore current directory
             if (opwd) opwd->cd();
@@ -464,7 +557,7 @@ void AliAnalysisManager::PackOutput(TList *target)
                Int_t gid = gROOT->ProcessLine("gProofServ->GetGroupId();");
                remote += Form("%s_%d_", gSystem->HostName(), gid);
                remote += output->GetFileName();
-               TFile::Cp(output->GetFileName(), remote.Data());
+               TFile::Cp ( outFilename.Data(), remote.Data() );
             } else {
             // No special location specified-> use TProofOutputFile as merging utility
             // The file at this output slot must be opened in CreateOutputObjects
@@ -485,28 +578,73 @@ void AliAnalysisManager::ImportWrappers(TList *source)
    AliAnalysisDataContainer *cont;
    AliAnalysisDataWrapper   *wrap;
    Int_t icont = 0;
+   Bool_t inGrid = (fMode == kGridAnalysis)?kTRUE:kFALSE;
    while ((cont=(AliAnalysisDataContainer*)next())) {
       wrap = 0;
-      if (cont->GetProducer()->IsPostEventLoop()) continue;
-      if (cont->IsSpecialOutput()) {
-         if (strlen(fSpecialOutputLocation.Data())) continue;
-         // Copy merged file from PROOF scratch space
-         if (fDebug > 1) 
-            printf("   Copying file %s from PROOF scratch space\n", cont->GetFileName());
-         Bool_t gotit = TFile::Cp(Form("root://lxb6045.cern.ch:11094//pool/scratch/%s",cont->GetFileName()),
-                   cont->GetFileName()); 
-         if (!gotit) {
-            Error("ImportWrappers", "Could not get file %s from proof scratch space", cont->GetFileName());
-         }
+      if (cont->GetProducer()->IsPostEventLoop() && !inGrid) continue;
+      const char *filename = cont->GetFileName();
+      Bool_t isManagedByHandler = kFALSE;
+      if (!(strcmp(filename, "default")) && fOutputEventHandler) {
+         isManagedByHandler = kTRUE;
+         filename = fOutputEventHandler->GetOutputFileName();
+      }
+      if (cont->IsSpecialOutput() || inGrid) {
+         if (strlen(fSpecialOutputLocation.Data()) && !isManagedByHandler) continue;
+         // Copy merged file from PROOF scratch space. 
+         // In case of grid the files are already in the current directory.
+         if (!inGrid) {
+            char full_path[512];
+            char ch_url[512];
+            TObject *pof =  source->FindObject(filename);
+            if (!pof || !pof->InheritsFrom("TProofOutputFile")) {
+               Error("ImportWrappers", "TProofOutputFile object not found in output list for container %s", cont->GetName());
+               continue;
+            }
+            gROOT->ProcessLine(Form("sprintf((char*)0x%lx, \"%%s\", ((TProofOutputFile*)0x%lx)->GetOutputFileName();)", full_path, pof));
+            gROOT->ProcessLine(Form("sprintf((char*)0x%lx, \"%%s\", gProof->GetUrl();)", ch_url));
+            TString clientUrl(ch_url);
+            TString full_path_str(full_path);
+            if (clientUrl.Contains("localhost")){
+               TObjArray* array = full_path_str.Tokenize ( "//" );
+               TObjString *strobj = ( TObjString *)array->At(1);
+               TObjArray* arrayPort = strobj->GetString().Tokenize ( ":" );
+               TObjString *strobjPort = ( TObjString *) arrayPort->At(1);
+               full_path_str.ReplaceAll(strobj->GetString().Data(),"localhost:PORT");
+               full_path_str.ReplaceAll(":PORT",Form(":%s",strobjPort->GetString().Data()));
+               if (fDebug > 1) Info("ImportWrappers","Using tunnel from %s to %s",full_path_str.Data(),filename);
+            }
+            if (fDebug > 1) 
+               printf("   Copying file %s from PROOF scratch space\n", full_path_str.Data());
+            Bool_t gotit = TFile::Cp(full_path_str.Data(), filename); 
+            if (!gotit) {
+               Error("ImportWrappers", "Could not get file %s from proof scratch space", cont->GetFileName());
+               continue;
+            }
+         }   
          // Normally we should connect data from the copied file to the
          // corresponding output container, but it is not obvious how to do this
          // automatically if several objects in file...
-         TFile *f = new TFile(cont->GetFileName(), "READ");
-         TObject *obj = f->Get(cont->GetName());
-         if (!obj) {
-            Error("ImportWrappers", "Could not find object %s in file %s", cont->GetName(), cont->GetFileName());
+         TFile *f = TFile::Open(filename, "READ");
+         if (!f) {
+            Error("ImportWrappers", "Cannot open file %s in read-only mode", filename);
             continue;
+         }   
+         TObject *obj = 0;
+         // Try to fetch first a list object having the container name.
+         obj = f->Get(cont->GetName());
+         if (!obj) {
+         // Fetch first object from file having the container type.
+            TIter nextkey(f->GetListOfKeys());
+            TKey *key;
+            while ((key=(TKey*)nextkey())) {
+               obj = f->Get(key->GetName());
+               if (obj && obj->IsA()->InheritsFrom(cont->GetType())) break;
+            }                     
          }
+         if (!obj) {
+            Error("ImportWrappers", "Could not find object for container %s in file %s", cont->GetName(), filename);
+            continue;
+         }  
          wrap = new AliAnalysisDataWrapper(obj);
          wrap->SetDeleteData(kFALSE);
       }   
@@ -518,7 +656,7 @@ void AliAnalysisManager::ImportWrappers(TList *source)
       icont++;
       if (fDebug > 1) {
          printf("   Importing data for container %s", cont->GetName());
-         if (strlen(cont->GetFileName())) printf("    -> file %s\n", cont->GetFileName());
+         if (strlen(filename)) printf("    -> file %s\n", filename);
          else printf("\n");
       }   
       cont->ImportData(wrap);
@@ -578,9 +716,9 @@ void AliAnalysisManager::Terminate()
    TIter next1(fOutputs);
    AliAnalysisDataContainer *output;
    while ((output=(AliAnalysisDataContainer*)next1())) {
-      // Close all files at output
-      // Special outputs have the files already closed and written.
-      if (output->IsSpecialOutput()) continue;
+      // Special outputs or grid files have the files already closed and written.
+      if (fMode == kGridAnalysis) continue;
+      if (output->IsSpecialOutput()&&(fMode == kProofAnalysis)) continue;
       const char *filename = output->GetFileName();
       if (!(strcmp(filename, "default"))) {
          if (fOutputEventHandler) filename = fOutputEventHandler->GetOutputFileName();
@@ -595,13 +733,11 @@ void AliAnalysisManager::Terminate()
       if (!output->GetData()) continue;
       TFile *file = output->GetFile();
       TDirectory *opwd = gDirectory;
-      if (file) {
-         file->cd();
-      } else {
-         file = new TFile(filename, "RECREATE");
-         if (file->IsZombie()) continue;
-         output->SetFile(file);
-      }   
+      file = (TFile*)gROOT->GetListOfFiles()->FindObject(filename);
+      if (!file) file = new TFile(filename, "RECREATE");
+      if (file->IsZombie()) continue;
+      output->SetFile(file);
+      file->cd();
       if (fDebug > 1) printf("   writing output data %s to file %s\n", output->GetData()->GetName(), file->GetName());
       if (output->GetData()->InheritsFrom(TCollection::Class())) {
       // If data is a collection, we set the name of the collection 
@@ -610,9 +746,21 @@ void AliAnalysisManager::Terminate()
          coll->SetName(output->GetName());
          coll->Write(output->GetName(), TObject::kSingleKey);
       } else {
-         output->GetData()->Write();
+         if (output->GetData()->InheritsFrom(TTree::Class())) {
+            TTree *tree = (TTree*)output->GetData();
+            tree->SetDirectory(file);
+            tree->AutoSave();
+         } else {
+            output->GetData()->Write();
+         }   
       }      
-      file->Close();
+      if (opwd) opwd->cd();
+   }   
+   next1.Reset();
+   while ((output=(AliAnalysisDataContainer*)next1())) {
+      // Close all files at output
+      TDirectory *opwd = gDirectory;
+      if (output->GetFile()) output->GetFile()->Close();
       if (opwd) opwd->cd();
    }   
 
@@ -631,14 +779,9 @@ void AliAnalysisManager::Terminate()
          tree->SetMarkerSize(0.5);
          if (!gROOT->IsBatch()) {
             tree->SetAlias("event", "id0");
-            tree->SetAlias("memUSED", "pI.fMemVirtual");
-            tree->SetAlias("userCPU", "pI.fCpuUser");
-            TCanvas *c = new TCanvas("SysInfo","SysInfo",10,10,800,600);
-            c->Divide(2,1,0.01,0.01);
-            c->cd(1);
+            tree->SetAlias("memUSED", "mi.fMemUsed");
+            new TCanvas("SysInfo","SysInfo",10,10,800,600);
             tree->Draw("memUSED:event","","", 1234567890, 0);
-            c->cd(2);
-            tree->Draw("userCPU:event","","", 1234567890, 0);
          }   
          tree->Write();
          f.Close();
@@ -705,6 +848,10 @@ Bool_t AliAnalysisManager::ConnectInput(AliAnalysisTask *task, Int_t islot,
                                         AliAnalysisDataContainer *cont)
 {
 // Connect input of an existing task to a data container.
+   if (!task) {
+      Error("ConnectInput", "Task pointer is NULL");
+      return kFALSE;
+   }   
    if (!fTasks->FindObject(task)) {
       AddTask(task);
       Info("ConnectInput", "Task %s was not registered. Now owned by analysis manager", task->GetName());
@@ -718,6 +865,10 @@ Bool_t AliAnalysisManager::ConnectOutput(AliAnalysisTask *task, Int_t islot,
                                         AliAnalysisDataContainer *cont)
 {
 // Connect output of an existing task to a data container.
+   if (!task) {
+      Error("ConnectOutput", "Task pointer is NULL");
+      return kFALSE;
+   }   
    if (!fTasks->FindObject(task)) {
       AddTask(task);
       Warning("ConnectOutput", "Task %s not registered. Now owned by analysis manager", task->GetName());
@@ -877,14 +1028,40 @@ void AliAnalysisManager::StartAnalysis(const char *type, TTree *tree, Long64_t n
    TString anaType = type;
    anaType.ToLower();
    fMode = kLocalAnalysis;
-   if (tree) {
-      if (anaType.Contains("proof"))     fMode = kProofAnalysis;
-      else if (anaType.Contains("grid")) fMode = kGridAnalysis;
-      else if (anaType.Contains("mix"))  fMode = kMixingAnalysis;
-   }
+   Bool_t runlocalinit = kTRUE;
+   if (anaType.Contains("file")) runlocalinit = kFALSE;
+   if (anaType.Contains("proof"))     fMode = kProofAnalysis;
+   else if (anaType.Contains("grid")) fMode = kGridAnalysis;
+   else if (anaType.Contains("mix"))  fMode = kMixingAnalysis;
+
    if (fMode == kGridAnalysis) {
-      Warning("StartAnalysis", "GRID analysis mode not implemented. Running local.");
-      fMode = kLocalAnalysis;
+      if (!fGridHandler) {
+         Error("StartAnalysis", "Cannot start grid analysis without a grid handler.");
+         Info("===", "Add an AliAnalysisAlien object as plugin for this manager and configure it.");
+         return;
+      }
+      // Write analysis manager in the analysis file
+      cout << "===== RUNNING GRID ANALYSIS: " << GetName() << endl;
+      // run local task configuration
+      TIter nextTask(fTasks);
+      AliAnalysisTask *task;
+      while ((task=(AliAnalysisTask*)nextTask())) {
+         task->LocalInit();
+      }
+      fGridHandler->StartAnalysis(nentries, firstentry);
+
+      // Terminate grid analysis
+      if (fSelector && fSelector->GetStatus() == -1) return;
+      if (fGridHandler->GetRunMode() == AliAnalysisGrid::kOffline) return;
+      cout << "===== MERGING OUTPUTS REGISTERED BY YOUR ANALYSIS JOB: " << GetName() << endl;
+      if (!fGridHandler->MergeOutputs()) {
+         // Return if outputs could not be merged or if it alien handler
+         // was configured for offline mode or local testing.
+         return;
+      }
+      ImportWrappers(NULL);
+      Terminate();
+      return;
    }
    char line[256];
    SetEventLoop(kFALSE);
@@ -893,7 +1070,7 @@ void AliAnalysisManager::StartAnalysis(const char *type, TTree *tree, Long64_t n
 
    TChain *chain = 0;
    TString ttype = "TTree";
-   if (tree->IsA() == TChain::Class()) {
+   if (tree && tree->IsA() == TChain::Class()) {
       chain = (TChain*)tree;
       if (!chain || !chain->GetListOfFiles()->First()) {
          Error("StartAnalysis", "Cannot process null or empty chain...");
@@ -905,9 +1082,11 @@ void AliAnalysisManager::StartAnalysis(const char *type, TTree *tree, Long64_t n
    // Initialize locally all tasks (happens for all modes)
    TIter next(fTasks);
    AliAnalysisTask *task;
-   while ((task=(AliAnalysisTask*)next())) {
-      task->LocalInit();
-   }
+   if (runlocalinit) {
+      while ((task=(AliAnalysisTask*)next())) {
+         task->LocalInit();
+      }
+   }   
    
    switch (fMode) {
       case kLocalAnalysis:
@@ -980,7 +1159,7 @@ void AliAnalysisManager::StartAnalysis(const char *type, const char *dataset, Lo
    TString anaType = type;
    anaType.ToLower();
    if (!anaType.Contains("proof")) {
-      Error("Cannot process datasets in %s mode. Try PROOF.", type);
+      Error("StartAnalysis", "Cannot process datasets in %s mode. Try PROOF.", type);
       return;
    }   
    fMode = kProofAnalysis;
@@ -1056,7 +1235,8 @@ void AliAnalysisManager::ExecAnalysis(Option_t *option)
       TIter next(fTasks);
    // De-activate all tasks
       while ((task=(AliAnalysisTask*)next())) task->SetActive(kFALSE);
-      AliAnalysisDataContainer *cont = (AliAnalysisDataContainer*)fInputs->At(0);
+      AliAnalysisDataContainer *cont = fCommonInput;
+      if (!cont) cont = (AliAnalysisDataContainer*)fInputs->At(0);
       if (!cont) {
              Error("ExecAnalysis","Cannot execute analysis in TSelector mode without at least one top container");
          return;
@@ -1116,3 +1296,21 @@ void AliAnalysisManager::FinishAnalysis()
 {
 // Finish analysis.
 }
+
+//______________________________________________________________________________
+void AliAnalysisManager::SetInputEventHandler(AliVEventHandler*  handler)
+{
+// Set the input event handler and create a container for it.
+   fInputEventHandler   = handler;
+   fCommonInput = CreateContainer("cAUTO_INPUT", TChain::Class(), AliAnalysisManager::kInputContainer);
+   Warning("SetInputEventHandler", " An automatic input container for the input chain was created.\nPlease use: mgr->GetCommonInputContainer() to access it.");
+}
+
+//______________________________________________________________________________
+void AliAnalysisManager::SetOutputEventHandler(AliVEventHandler*  handler)
+{
+// Set the input event handler and create a container for it.
+   fOutputEventHandler   = handler;
+   fCommonOutput = CreateContainer("cAUTO_OUTPUT", TTree::Class(), AliAnalysisManager::kOutputContainer, "default");
+   Warning("SetOutputEventHandler", " An automatic output container for the output tree was created.\nPlease use: mgr->GetCommonOutputContainer() to access it.");
+}