Make the macro compatible with the new AliRun
[u/mrichter/AliRoot.git] / ITS / ITSHitsToDigits.C
1 #include "iostream.h"
2
3 void ITSHitsToDigits (Int_t evNumber1=0,Int_t evNumber2=0,Int_t nsignal  =25, Int_t size=-1) 
4 {
5 /////////////////////////////////////////////////////////////////////////
6 //   This macro is a small example of a ROOT macro
7 //   illustrating how to read the output of GALICE
8 //   and do some analysis.
9 //   
10 /////////////////////////////////////////////////////////////////////////
11
12 // Dynamically link some shared libs
13
14    if (gClassTable->GetID("AliRun") < 0) {
15       gROOT->LoadMacro("loadlibs.C");
16       loadlibs();
17    } else {
18       delete gAlice;
19       gAlice=0;
20    }
21
22
23 // Connect the Root Galice file containing Geometry, Kine and Hits
24
25    TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject("galice.root");
26    printf("file %p\n",file);
27    if (file) file->Close(); 
28    if (!file) file = new TFile("galice.root","UPDATE");
29    file->ls();
30
31    printf ("I'm after Map \n");
32
33 // Get AliRun object from file or create it if not on file
34
35    if (!gAlice) {
36        gAlice = (AliRun*)file->Get("gAlice");
37        if (gAlice) printf("AliRun object found on file\n");
38        if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
39    }
40    printf ("I'm after gAlice \n");
41    
42    AliITS *ITS  = (AliITS*) gAlice->GetModule("ITS");
43    if (!ITS) return;
44
45
46    // Set the simulation models
47
48    AliITSgeom *geom = ITS->GetITSgeom();
49
50    // SDD
51    // SDD compression param: 2 fDecrease, 2fTmin, 2fTmax or disable, 2 fTolerance
52    Float_t baseline = 10.;
53    Float_t noise = 1.75;
54
55                 
56    AliITSDetType *iDetType=ITS->DetType(1);
57    AliITSresponseSDD *res1 = (AliITSresponseSDD*)iDetType->GetResponseModel();
58    if (!res1) {
59          res1=new AliITSresponseSDD();
60          ITS->SetResponseModel(1,res1);
61    }
62    res1->SetMagicValue(900.);
63
64    Float_t maxadc = res1->MaxAdc();    
65    Float_t topValue = res1->MagicValue();
66    Float_t norm = maxadc/topValue;
67
68    Float_t fCutAmp = baseline + 2.*noise;
69    fCutAmp *= norm;
70    Int_t cp[8]={0,0,fCutAmp,fCutAmp,0,0,0,0}; //1D
71
72    //res1->SetZeroSupp("2D");
73    res1->SetZeroSupp("1D");
74    res1->SetNoiseParam(noise,baseline);
75    res1->SetDo10to8(kTRUE);
76    res1->SetMinVal(4);
77    res1->SetCompressParam(cp);
78    res1->SetDiffCoeff(3.6,40.);
79
80    AliITSsegmentationSDD *seg1=(AliITSsegmentationSDD*)iDetType->GetSegmentationModel();
81    if (!seg1) {
82        seg1 = new AliITSsegmentationSDD(geom,res1);
83        ITS->SetSegmentationModel(1,seg1);
84    }
85
86    AliITSsimulationSDD *sim1=new AliITSsimulationSDD(seg1,res1);
87    sim1->SetDoFFT(1);
88    sim1->SetCheckNoise(kFALSE);
89
90    ITS->SetSimulationModel(1,sim1);
91    
92    
93
94    // SPD
95
96    AliITSDetType *iDetType=ITS->DetType(0);
97    AliITSsegmentationSPD *seg0=(AliITSsegmentationSPD*)iDetType->GetSegmentationModel();
98    AliITSresponseSPD *res0 = (AliITSresponseSPD*)iDetType->GetResponseModel();
99    AliITSsimulationSPD *sim0=new AliITSsimulationSPD(seg0,res0);
100    ITS->SetSimulationModel(0,sim0);
101    // test
102    //printf("SPD dimensions %f %f \n",seg0->Dx(),seg0->Dz());
103    //printf("SPD npixels %d %d \n",seg0->Npz(),seg0->Npx());
104    //printf("SPD pitches %d %d \n",seg0->Dpz(0),seg0->Dpx(0));
105    // end test
106
107
108    // SSD
109
110    AliITSDetType *iDetType=ITS->DetType(2);
111    AliITSsegmentationSSD *seg2=(AliITSsegmentationSSD*)iDetType->GetSegmentationModel();
112    AliITSresponseSSD *res2 = (AliITSresponseSSD*)iDetType->GetResponseModel();
113    res2->SetSigmaSpread(3.,2.);
114    AliITSsimulationSSD *sim2=new AliITSsimulationSSD(seg2,res2);
115    ITS->SetSimulationModel(2,sim2);
116
117
118 //
119 // Event Loop
120 //
121
122
123    // create the TreeD 
124
125    Int_t nparticles=gAlice->GetEvent(0);
126    printf("Create TreeD \n");
127    gAlice->MakeTree("D");
128    printf("TreeD %p\n",gAlice->TreeD());
129    //make branch
130    ITS->MakeBranch("D");
131
132    Int_t nbgr_ev=0;
133    for (Int_t nev=evNumber1; nev<= evNumber2; nev++) {
134        cout << "nev         " <<nev<<endl;
135        if(nev>0) {
136          nparticles = gAlice->GetEvent(nev);
137          gAlice->SetEvent(nev);
138          gAlice-> MakeTree("D");
139          ITS->MakeBranch("D");
140        }
141        cout << "nparticles  " <<nparticles<<endl;
142        if (nev < evNumber1) continue;
143        if (nparticles <= 0) return;
144
145        Int_t nbgr_ev=0;
146        if(nsignal) nbgr_ev=Int_t(nev/nsignal);
147        ITS->HitsToDigits(nev,nbgr_ev,size," ","All"," ");
148    } // event loop 
149
150    delete sim0;
151    delete sim1;
152    delete sim2;
153
154
155    file->Close();
156 }
157
158
159
160
161
162
163
164
165
166
167
168
169
170