]>
Commit | Line | Data |
---|---|---|
8f62fbbb | 1 | //////////////////////////////////////////////////////////////////////// |
2 | // | |
3 | // name: CopyAli.C | |
4 | // date: 17.3.2002 | |
5 | // last update: 21.3.2002 | |
6 | // author: Jiri Chudoba | |
7 | // version: 1.0 | |
8 | // | |
9 | // description: | |
10 | // copy some alice objects from 1 file to another | |
11 | // | |
12 | // ToDo: | |
13 | // add support for more events in 1 file | |
14 | // | |
15 | // | |
16 | // Note: | |
17 | // copied objects are not deleted, I assume that the root | |
18 | // session is exited after this copy | |
19 | // | |
20 | // Example: | |
21 | // aliroot -b -q CopyAli.C\("galice.root","TreeK.root",0,1,1,1\) | |
22 | // | |
23 | // History: | |
24 | // | |
25 | // 21.3.02 - first version | |
26 | // | |
27 | //////////////////////////////////////////////////////////////////////// | |
28 | ||
29 | #if !defined(__CINT__) || defined(__MAKECINT__) | |
30 | #include "iostream.h" | |
31 | #include "TTree.h" | |
32 | #include "TBranch.h" | |
33 | #include "TDirectory.h" | |
34 | #include "TFile.h" | |
35 | #include "AliRun.h" | |
36 | #include "TParticle.h" | |
37 | #include "AliHeader.h" | |
38 | #include "TGeometry.h" | |
39 | #include "TObjArray.h" | |
40 | #include "TString.h" | |
41 | #endif | |
42 | ||
43 | void CopyAli(TString fnOrig="rfio:galice.root", TString fnNew="galice_new.root",Int_t iEvent = 0, Bool_t copygAlice=kTRUE,Bool_t copyTreeK = kFALSE) | |
44 | { | |
45 | ||
46 | TFile *fileOrig = TFile::Open(fnOrig); | |
47 | if (!fileOrig->IsOpen()) { | |
48 | cerr<<"Cannot open input file "<<fnOrig.Data()<<endl; | |
49 | return; | |
50 | } | |
51 | // AliRun *gAlice; | |
52 | if (gAlice) {delete gAlice; gAlice = 0;} | |
53 | gAlice = (AliRun*)(fileOrig->Get("gAlice")); | |
54 | if (!gAlice) { | |
55 | cerr<<"Cannot read gAlice from the input file"<<endl; | |
56 | return; | |
57 | } | |
58 | ||
59 | Int_t nAllTracks = gAlice->GetEvent(iEvent); | |
60 | cout<<"nAllTracks = "<<nAllTracks<<endl; | |
61 | ||
62 | // Open the new file | |
63 | ||
64 | TFile *fileNew = TFile::Open(fnNew,"update"); | |
65 | if (!fileNew->IsOpen()) { | |
66 | cerr<<"Cannot open output file "<<fnNew.Data()<<endl; | |
67 | return; | |
68 | } | |
69 | if (copygAlice) { | |
70 | cout<<"Copy gAlice: "; | |
71 | gAlice->Write(); | |
72 | cout<<"done"<<endl; | |
73 | ||
74 | TTree *treeE = gAlice->TreeE(); | |
75 | if (!treeE) { | |
76 | cerr<<"No TreeE found for event "<<iEvent<<endl; | |
77 | return; | |
78 | } | |
79 | cout<<"Copy TreeE: "; | |
80 | AliHeader *header = new AliHeader(); | |
81 | treeE->SetBranchAddress("Header", &header); | |
82 | treeE->SetBranchStatus("*",1); | |
83 | TTree *treeENew = treeE->CloneTree(); | |
84 | treeENew->Write(); | |
85 | cout<<"done"<<endl; | |
86 | ||
87 | cout<<"Copy AliceGeom: "; | |
88 | TGeometry *AliceGeom = static_cast<TGeometry*>(fileOrig->Get("AliceGeom")); | |
89 | if (!AliceGeom) { | |
90 | cerr<<"AliceGeom was not found in the input file "<<fnNew.Data()<<endl; | |
91 | return; | |
92 | } | |
93 | AliceGeom->Write(); | |
94 | cout<<"done"<<endl; | |
95 | } | |
96 | ||
97 | ||
98 | if (copyTreeK) { | |
99 | cout<<"Copy TreeK: "; | |
100 | TTree *treeK = gAlice->TreeK(); | |
101 | if (!treeK) { | |
102 | cerr<<"No TreeK found for event "<<iEvent<<endl; | |
103 | return; | |
104 | } | |
105 | TParticle *particle = new TParticle(); | |
106 | treeK->SetBranchAddress("Particles",&particle); | |
107 | treeK->SetBranchStatus("*",1); | |
108 | TTree *treeKNew = treeK->CloneTree(); | |
109 | treeKNew->Write(); | |
110 | cout<<"done"<<endl; | |
111 | } | |
112 | ||
113 | ||
114 | delete gAlice; | |
115 | gAlice = 0; | |
116 | fileNew->Close(); | |
117 | fileOrig->Close(); | |
118 | delete fileNew; | |
119 | delete fileOrig; | |
120 | } |