]> git.uio.no Git - u/mrichter/AliRoot.git/blame - HLT/exa/runhough.C
Renamed houghtest.C to runhough.C
[u/mrichter/AliRoot.git] / HLT / exa / runhough.C
CommitLineData
e0f350ab 1void houghtest(Int_t slice,Char_t *path,Int_t n_eta_segments)
2{
3
4
1772d17b 5 hough = new AliL3Hough();
6 hough->Init(path,kTRUE,n_eta_segments,kTRUE);
e0f350ab 7
1772d17b 8 hough->ReadData(slice);
9
e0f350ab 10 hough->Transform();
1772d17b 11
12 hough->SetPeakThreshold(1);
e0f350ab 13 hough->AddAllHistograms();
14 hough->FindTrackCandidates();
15
16 hough->WriteTracks(slice);
17
18 //hough->Evaluate();
19 tracks = (AliL3TrackArray*)hough->GetTracks(0);
20 for(int i=0; i<tracks->GetNTracks(); i++)
21 {
22 track = (AliL3HoughTrack*)tracks->GetCheckedTrack(i);
23 if(!track) continue;
24 cout<<"pt "<<track->GetPt()<<" psi "<<track->GetPsi()<<" etaindex "<<track->GetEtaIndex()<<" weight "<<track->GetWeight()<<endl;
25 }
26
27
28 display(hough,0);
29
30}
31
32void display(AliL3Hough *hough,Int_t eta_index)
33{
34 //Display the data/tracks in eta_index
35
36 hough->InitEvaluate();
37 digitd = new AliL3Histogram("Digits display","",250,0,250,250,-125,125);
38 trackd = new AliL3Histogram("Found tracks display","",250,0,250,250,-125,125);
39 for(int i=0; i<6; i++)
40 hough->GetEval(i)->DisplayEtaSlice(eta_index,digitd);
41
42 tracks = (AliL3TrackArray*)hough->GetTracks(0);
43 float xyz[3];
44 for(int i=0; i<tracks->GetNTracks(); i++)
45 {
46 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->GetCheckedTrack(i);
47 if(!track) continue;
48 if(track->GetEtaIndex() != eta_index) continue;
49
50 for(int j=0; j<176; j++)
51 {
52 track->GetCrossingPoint(j,xyz);
53 trackd->Fill(xyz[0],xyz[1],1);
54 }
55 }
56
57 //Draw the parameter space
58 c1 = new TCanvas("c1","",2);
1772d17b 59 hough->GetTransformer(0)->GetHistogram(eta_index)->Draw("lego");
60 return;
e0f350ab 61 //Draw the tracks
62 c2 = new TCanvas("c2","",2);
63 digitd->Draw();
64 trackd->Draw("same");
65 ((TH1F*)trackd->GetRootHisto())->SetMarkerColor(2);
66}