]> git.uio.no Git - u/mrichter/AliRoot.git/blob - STEER/cloneAODTreeAndAddObject.C
Coding rule violation corrected.
[u/mrichter/AliRoot.git] / STEER / cloneAODTreeAndAddObject.C
1 void cloneAODTreeAndAddObject(const char *newFileName = "AliAOD_new.root", const char *orgFileName = "AliAOD.root") {
2   // This little macro takes an already created AOD file and clones it.
3   // After adding a new brach, the new TTree is written to a new file.
4   //
5   // To read back this data (including the newly updated branch), have
6   // a look at $ALICE_ROOT/STEER/ReadAOD.C and add one line
7   //       TClonesArray *esdTracks = 
8   //            (TClonesArray*)ev->GetList()->FindObject("esdTracks");
9   // after   
10   //       ev->ReadFromTree(aodTree);
11   //
12   // esdTracks will be 'filled' correctly after each call of
13   //       aodTree->GetEvent(nEv);
14   //
15
16
17   // open input file and get the TTree
18   TFile orgFile(orgFileName, "READ");
19   // get original TTree
20   TTree *orgAodTree = (TTree*)orgFile.Get("aodTree");
21
22   // open new output file
23   TFile *newFile = new TFile(newFileName, "RECREATE");
24   // clone old TTree
25   TTree *newAodTree = orgAodTree->CloneTree();
26
27   // do your gymnastics with the new TTree
28   AliAODEvent *evNew = new AliAODEvent();
29   evNew->ReadFromTree(newAodTree);
30     
31   // add new information to the list (we use AliESDtracks as an example)
32   TClonesArray *tracks = new TClonesArray("AliESDtrack", 0);
33   evNew->AddObject(tracks); // add new object to the list
34
35   // define a unique name
36   const char *name = "esdTracks";
37   tracks->SetName(name); // set this name to be the name of the TClonesArray
38   // create the new branch
39   TBranch *newBranch = newAodTree->Branch(name, &tracks); // the branch gets the same name
40
41   // loop over events
42   Int_t nEvent = newAodTree->GetEntries();
43
44   for(Int_t iEv = 0; iEv < nEvent; iEv++) {
45
46     /*
47     // read events (only necessary if you want to access the old data)
48     newAodTree->GetEvent(iEv);
49     */
50
51     tracks->Delete(); // for each event delete old entries of new TClonesArray
52
53     Int_t nTracks = gRandom->Rndm() * 50; // randomize size of TClonesArray (just for this example)
54     tracks->Expand(nTracks); // expand container (just for speed)
55     
56     // fill TClonesArray
57     TClonesArray &rTracks = *tracks;
58     for (Int_t iTr = 0; iTr< nTracks; iTr++) {
59       new(rTracks[iTr]) AliESDtrack();
60     }
61     
62     // fill the new branch
63     newBranch->Fill();
64   }
65
66   // delete old and write new UserInfo
67   newAodTree->GetUserInfo()->Clear();
68   newAodTree->GetUserInfo()->Add(evNew);
69
70   // write new TTree to file
71   newAodTree->Write();
72
73   // close files
74   newFile->Close();
75   delete newFile;
76
77   orgFile.Close();
78 }