Modified and updated to be used with this latest code.
[u/mrichter/AliRoot.git] / ITS / ITSDigitsToClusters.C
1 #include "iostream.h"
2
3 void ITSDigitsToClusters (Int_t evNumber1=0,Int_t evNumber2=0) 
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    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    AliITSgeom *geom = ITS->GetITSgeom();
46
47
48    // Set the models for cluster finding
49
50    // SPD
51
52    AliITSDetType *iDetType=ITS->DetType(0);
53    AliITSsegmentationSPD *seg0=(AliITSsegmentationSPD*)iDetType->GetSegmentationModel();
54    TClonesArray *dig0  = ITS->DigitsAddress(0);
55    TClonesArray *recp0  = ITS->ClustersAddress(0);
56    AliITSClusterFinderSPD *rec0=new AliITSClusterFinderSPD(seg0,dig0,recp0);
57    ITS->SetReconstructionModel(0,rec0);
58    // test
59    //printf("SPD dimensions %f %f \n",seg0->Dx(),seg0->Dz());
60    //printf("SPD npixels %d %d \n",seg0->Npz(),seg0->Npx());
61
62
63    // SDD
64
65    Float_t baseline = 10.;
66    Float_t noise = 1.67;
67    Int_t thres = (Int_t)(baseline+3*noise);
68    printf("thresh %d\n",thres);
69
70    AliITSDetType *iDetType=ITS->DetType(1);
71    AliITSgeom *geom = ITS->GetITSgeom();
72
73    AliITSsegmentationSDD *seg1=(AliITSsegmentationSDD*)iDetType->GetSegmentationModel();
74    if (!seg1) seg1 = new AliITSsegmentationSDD(geom);
75    AliITSresponseSDD *res1 = (AliITSresponseSDD*)iDetType->GetResponseModel();
76    if (!res1) res1=new AliITSresponseSDD();
77    res1->SetNoiseParam(noise,baseline);
78    Float_t n,b;
79    res1->GetNoiseParam(n,b);
80     printf("SDD: noise baseline %f %f zs option %s data type %s\n",n,b,res1->ZeroSuppOption(),res1->DataType());
81    printf("SDD: DriftSpeed %f TopValue %f\n",res1->DriftSpeed(),res1->MagicValue());
82    Float_t dif0,dif1;
83    res1->DiffCoeff(dif0,dif1);
84    printf("SDD: dif0 %f dif1 %f\n",dif0,dif1);
85    TClonesArray *dig1  = ITS->DigitsAddress(1);
86    TClonesArray *recp1  = ITS->ClustersAddress(1);
87    AliITSClusterFinderSDD *rec1=new AliITSClusterFinderSDD(seg1,res1,dig1,recp1);
88    rec1->SetMinNCells(6);
89    rec1->SetTimeCorr(70.);
90    rec1->SetCutAmplitude(thres);
91    ITS->SetReconstructionModel(1,rec1);
92
93    // SSD
94
95    AliITSDetType *iDetType=ITS->DetType(2);
96    AliITSsegmentationSSD *seg2=(AliITSsegmentationSSD*)iDetType->GetSegmentationModel();
97    TClonesArray *dig2  = ITS->DigitsAddress(2);
98    TClonesArray *recp2  = ITS->ClustersAddress(2);
99    AliITSClusterFinderSSD *rec2=new AliITSClusterFinderSSD(seg2,dig2,recp2);
100    ITS->SetReconstructionModel(2,rec2);
101    // test
102    //printf("SSD dimensions %f %f \n",seg2->Dx(),seg2->Dz());
103    //printf("SSD nstrips %d %d \n",seg2->Npz(),seg2->Npx());
104
105
106
107 //
108 // Event Loop
109 //
110
111    for (int nev=evNumber1; nev<= evNumber2; nev++) {
112        Int_t nparticles = gAlice->GetEvent(nev);
113        cout << "nev         " <<nev<<endl;
114        cout << "nparticles  " <<nparticles<<endl;
115        if (nev < evNumber1) continue;
116        if (nparticles <= 0) return;
117
118        TTree *TD = gAlice->TreeD();
119        Int_t nent=TD->GetEntries();
120        printf("Found %d entries in the tree (must be one per module per event!)\n",nent);
121        //Int_t nmodules=geom->GetLastSSD();
122        //Int_t last_entry=nent-(nmodules+1);
123        Int_t last_entry=1;
124        ITS->DigitsToRecPoints(nev,last_entry,"All");
125    } // event loop 
126
127    delete rec0;
128    delete rec1;
129    delete rec2;
130
131    file->Close();
132 }
133
134
135
136
137
138
139
140
141
142
143
144
145
146