STEER/AliAODHandler.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 //-------------------------------------------------------------------------
  19 //     Implementation of the Virtual Event Handler Interface for AOD
  20 //     Author: Andreas Morsch, CERN
  21 //-------------------------------------------------------------------------
  22
  23
  24 #include <TTree.h>
  25 #include <TFile.h>
  26
  27 #include "AliAODHandler.h"
  28 #include "AliAODEvent.h"
  29
  30 ClassImp(AliAODHandler)
  31
  32 //______________________________________________________________________________
  33 AliAODHandler::AliAODHandler() :
  34     AliVEventHandler(),
  35     fAODEvent(NULL),
  36     fTreeA(NULL),
  37     fFileA(NULL),
  38     fName("")
  39 {
  40   // default constructor
  41 }
  42
  43 //______________________________________________________________________________
  44 AliAODHandler::AliAODHandler(const char* name, const char* title):
  45     AliVEventHandler(name, title),
  46     fAODEvent(NULL),
  47     fTreeA(NULL),
  48     fFileA(NULL),
  49     fName("")
  50 {
  51 }
  52
  53 //______________________________________________________________________________
  54 AliAODHandler::~AliAODHandler()
  55 {
  56 // destructor
  57 }
  58
  59
  60 Bool_t AliAODHandler::InitIO(Option_t* opt)
  61 {
  62     // Initialize IO
  63     //
  64     // Create the AODevent object
  65     fAODEvent = new AliAODEvent();
  66     fAODEvent->CreateStdContent();
  67     //
  68     // File opening according to execution mode
  69
  70     if (!(strcmp(opt, "proof"))) {
  71         // proof
  72         CreateTree(0);
  73     } else {
  74         // local and grid
  75         fFileA = new TFile(fName, "RECREATE");
  76         CreateTree(1);
  77     }
  78     return kTRUE;
  79 }
  80
  81 Bool_t AliAODHandler::FinishEvent()
  82 {
  83     // Fill data structures
  84     FillTree();
  85     fAODEvent->ClearStd();
  86
  87     return kTRUE;
  88 }
  89
  90 Bool_t AliAODHandler::Terminate()
  91 {
  92     // Terminate
  93     AddAODtoTreeUserInfo();
  94     return kTRUE;
  95 }
  96
  97 Bool_t AliAODHandler::TerminateIO()
  98 {
  99     // Terminate IO
 100     if (fFileA) {
 101         fFileA->Close();
 102         delete fFileA;
 103     }
 104     return kTRUE;
 105 }
 106
 107
 108 void AliAODHandler::CreateTree(Int_t flag)
 109 {
 110     // Creates the AOD Tree
 111     fTreeA = new TTree("aodTree", "AliAOD tree");
 112     fTreeA->Branch(fAODEvent->GetList());
 113     if (flag == 0) fTreeA->SetDirectory(0);
 114 }
 115
 116 void AliAODHandler::FillTree()
 117 {
 118     // Fill the AOD Tree
 119     fTreeA->Fill();
 120 }
 121
 122
 123 void AliAODHandler::AddAODtoTreeUserInfo()
 124 {
 125     // Add aod event to tree user info
 126     fTreeA->GetUserInfo()->Add(fAODEvent);
 127 }