Update timestamps for new AMANDA simulation (17/02/2015)
[u/mrichter/AliRoot.git] / TPC / AliTPCDisplayDigits.C
1 /// \file AliTPCDisplayDigits.C
2 ///
3 /// \author I.Belikov, CERN, Jouri.Belikov@cern.ch
4
5 #ifndef __CINT__
6 #include <Riostream.h>
7 #include <TFile.h>
8 #include <TTree.h>
9 #include <TCanvas.h>
10 #include <TStyle.h>
11 #include <TH2.h>
12
13 #include "AliTPCParam.h"
14 #include "AliSimDigits.h"
15 #endif
16
17 Int_t AliTPCDisplayDigits(Int_t eventn, int sec, int row, int lab=-1,
18                  int max_t_chan=500, float min_t=0., float max_t=500.,
19                  int max_p_chan=150, float min_p=0., float max_p=150.)
20 {
21    cerr<<"Displaying digits...\n";
22
23 // Connect the Root Galice file containing Geometry, Kine and Hits
24    TFile *f = TFile::Open("rfio:galice.root");
25    if (!f->IsOpen()) {cerr<<"Can't open galice.root !\n"; return 1;}
26
27    
28    
29    AliTPCParam *par=(AliTPCParam *)f->Get("75x40_100x60_150x60");
30    if (!par) { cerr<<"TPC parameters have not been found !\n"; return 2; }
31
32    Char_t s[80];
33    sprintf(s,"Sector %d   Row %d\n",sec,row);
34    TH2F *h = new TH2F("h",s,max_t_chan,min_t,max_t,max_p_chan,min_p,max_p);
35    
36    Char_t  name[100];
37    sprintf(name,"TreeD_75x40_100x60_150x60_%d",eventn);
38
39    TTree *t=(TTree*)f->Get(name);
40    if (!t) { cerr<<"TPC digits have not been found !\n"; return 3; }
41    AliSimDigits dummy, *digit=&dummy;
42    t->GetBranch("Segment")->SetAddress(&digit);
43    Int_t sbr=(Int_t)t->GetEntries();
44    for (Int_t i=0; i<sbr; i++) {
45        if (!t->GetEvent(i)) continue;
46        Int_t s,r;
47        par->AdjustSectorRow(digit->GetID(),s,r);
48        if (s==sec && r==row) goto ok;
49    }
50    return 4;
51    
52 ok:
53
54    Int_t imax=0, jmax=0, qmax=0;
55    digit->First();
56    do {
57      //Short_t dig=digit->CurrentDigit();
58       Int_t i=digit->CurrentRow(), j=digit->CurrentColumn();
59       Short_t dig=digit->GetDigit(i,j);
60       if (lab >= 0) {
61          Int_t lab0=digit->GetTrackID(i,j,0);
62          Int_t lab1=digit->GetTrackID(i,j,1);
63          Int_t lab2=digit->GetTrackID(i,j,2);
64          if (lab0!=lab) if (lab1!=lab) if (lab2!=lab) continue;
65          if (dig>qmax) {imax=i; jmax=j; qmax=dig;}
66          cerr<<lab0<<' '<<lab1<<' '<<lab2<<endl;
67       }
68       h->Fill(i,j,dig);
69    } while (digit->Next());
70    if (qmax>0){cerr<<"Peak (time,pad,q) : "<<imax<<' '<<jmax<<' '<<qmax<<endl;}
71
72    h->SetMaximum(100);
73    gStyle->SetOptStat(0);
74    TCanvas *c1=new TCanvas("ddisplay","TPC digits display",0,0,1110,680);
75    TPad *p1=new TPad("p1","",0,0,1,0.5);
76    p1->Draw();
77    TPad *p2=new TPad("p2","",0,0.5,1,1);
78    p2->Draw();
79    p2->cd();
80    h->DrawCopy("lego");
81    p1->cd();
82    h->DrawCopy("colz");
83
84    c1->Modified(); c1->Update(); 
85
86    f->Close();
87    return 0;
88 }
89