Adding scripts directory including:
[u/mrichter/AliRoot.git] / test / vmctest / scripts / digitsSSD.C
1 /**************************************************************************
2  * Copyright(c) 1998-1999, 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 // Macro to generate histograms from digits
19 // By E. Sicking, CERN
20
21 TDirectoryFile* GetDirectory(Int_t ievent, const TString& detName, Int_t nfiles) 
22 {
23   for (Int_t file =0; file<nfiles; file++) {
24     TString filename(detName);
25     if ( file == 0 ) {
26       filename += ".Digits.root";
27     }  
28     else { 
29       filename += TString(Form(".Digits%d.root",file));
30     }
31
32     TFile* file0 = TFile::Open(filename.Data());
33
34     TDirectoryFile* dir = (TDirectoryFile*)file0->Get(Form("Event%d",ievent));
35     if (dir)  return dir;
36   }
37   return 0;
38 }  
39     
40 void digitsSSD(Int_t nevents, Int_t nfiles)
41 {
42
43   TH1F * hadc = new TH1F ("hadc", "hadc",200, 0, 4500);   
44   TH1F * hadclog = new TH1F ("hadclog", "hadclog",200, -1, 4);   
45
46   TDirectoryFile *tdf[100];     
47   TDirectoryFile *tdfKine[100] ;
48
49   TTree *ttree[100];     
50   TTree *ttreeKine[100]; 
51
52
53   TClonesArray *arr= NULL; // 
54   
55
56
57   //Run loader------------
58   TString name;
59   name = "galice.root";
60   AliRunLoader* rlSig = AliRunLoader::Open(name.Data());
61
62   // gAlice
63   rlSig->LoadgAlice();
64   gAlice = rlSig->GetAliRun();
65
66   // Now load kinematics and event header
67   rlSig->LoadKinematics();
68   rlSig->LoadHeader();
69   cout <<  rlSig->GetNumberOfEvents()<< endl;
70   //----------------------
71
72     //loop over events in the files
73     for(Int_t event=0; event<nevents; event++){
74       //printf("###event= %d\n", event +file*100);
75       printf("###event= %d\n", event);
76
77       // tdf[event] = (TDirectoryFile*)file0->Get(Form("Event%d",event+file*100));
78     tdf[event] = GetDirectory(event, "ITS", nfiles);
79     if ( ! tdf[event] ) {
80       cerr << "Event directory not found in " << nfiles <<  " files" << endl;
81       exit(1);
82     }      
83
84       ttree[event] = (TTree*)tdf[event]->Get("TreeD");
85           
86       arr = NULL;
87       ttree[event]->SetBranchAddress("ITSDigitsSSD", &arr);
88    
89
90       // Runloader -> gives particle Stack
91       rlSig->GetEvent(event);
92       AliStack * stack = rlSig->Stack(); 
93       //stack->DumpPStack();
94
95
96       // loop over tracks
97       Int_t NumberPrim=0;
98       for(Int_t iev=0; iev<ttree[event]->GetEntries(); iev++){
99         ttree[event]->GetEntry(iev);
100
101         
102         for (Int_t j = 0; j < arr->GetEntries(); j++) {
103           AliITSdigit* digit = dynamic_cast<AliITSdigit*> (arr->At(j));
104           if (digit){
105
106             hadc->Fill(digit->GetSignal());
107             hadclog->Fill(TMath::Log10(digit->GetSignal()));
108           }
109         }
110       }
111
112
113
114     }
115   
116   TFile fc("digits.ITS.SSD.root","RECREATE");
117   fc.cd();
118    
119   hadc->Write();
120   hadclog->Write();
121
122   fc.Close();
123
124 }