]>
Commit | Line | Data |
---|---|---|
e0f350ab | 1 | void 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 | ||
32 | void 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 | } |