f7cefc2c570b641a150ebaa493e2d97a1967f1de
[u/mrichter/AliRoot.git] / 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   delete fAODEvent;
57   if(fFileA){
58     // is already handled in TerminateIO
59     fFileA->Close();
60     delete fFileA;
61   }
62   delete fTreeA;
63   delete fName;
64  // destructor
65 }
66
67
68 Bool_t AliAODHandler::Init(Option_t* opt)
69 {
70   // Initialize IO
71   //
72   // Create the AODevent object
73   if(!fAODEvent){
74     fAODEvent = new AliAODEvent();
75     fAODEvent->CreateStdContent();
76   }
77   //
78   // File opening according to execution mode
79   
80   if (!(strcmp(opt, "proof"))) {
81     // proof
82     CreateTree(0);
83   } else {
84     // local and grid
85     TDirectory *owd = gDirectory;
86     fFileA = new TFile(fName, "RECREATE");
87     CreateTree(1);
88     owd->cd();
89   }
90   return kTRUE;
91 }
92
93 Bool_t AliAODHandler::FinishEvent()
94 {
95     // Fill data structures
96     FillTree();
97     fAODEvent->ResetStd();
98     return kTRUE;
99 }
100
101 Bool_t AliAODHandler::Terminate()
102 {
103     // Terminate 
104     AddAODtoTreeUserInfo();
105     return kTRUE;
106 }
107
108 Bool_t AliAODHandler::TerminateIO()
109 {
110     // Terminate IO
111     if (fFileA) {
112         fFileA->Close();
113         delete fFileA;
114     }
115     return kTRUE;
116 }
117
118
119 void AliAODHandler::CreateTree(Int_t flag)
120 {
121     // Creates the AOD Tree
122     fTreeA = new TTree("aodTree", "AliAOD tree");
123     fTreeA->Branch(fAODEvent->GetList());
124     if (flag == 0) fTreeA->SetDirectory(0);
125 }
126
127 void AliAODHandler::FillTree()
128 {
129     // Fill the AOD Tree
130     fTreeA->Fill();
131 }
132
133
134 void AliAODHandler::AddAODtoTreeUserInfo()
135 {
136     // Add aod event to tree user info
137     fTreeA->GetUserInfo()->Add(fAODEvent);
138 }