bug fix
[u/mrichter/AliRoot.git] / FMD / scripts / DrawHitsSDigits.C
1 //____________________________________________________________________
2 //
3 // $Id: DrawHitsSDigits.C 20907 2007-09-25 08:44:03Z cholm $
4 //
5 // Script that contains a class to draw eloss from hits, versus ADC
6 // counts from sdigits, using the AliFMDInputHits class in the util library. 
7 //
8 // It draws the energy loss versus the p/(mq^2).  It can be overlayed
9 // with the Bethe-Bloc curve to show how the simulation behaves
10 // relative to the expected. 
11 //
12 // Use the script `Compile.C' to compile this class using ACLic. 
13 //
14 #include <TH2D.h>
15 #include <AliFMDHit.h>
16 #include <AliFMDSDigit.h>
17 #include <AliFMDInput.h>
18 #include <AliFMDEdepMap.h>
19 #include <iostream>
20 #include <TStyle.h>
21 #include <TArrayF.h>
22 #include <AliLog.h>
23 #include <TMath.h>
24
25 /** @class DrawHitsSDigits
26     @brief Draw hit energy loss versus sdigit ADC
27     @code 
28     Root> .L Compile.C
29     Root> Compile("DrawHitsSDigits.C")
30     Root> DrawHitsSDigits c
31     Root> c.Run();
32     @endcode
33     @ingroup FMD_script
34  */
35 class DrawHitsSDigits : public AliFMDInput
36 {
37 private:
38   TH2D* fElossVsAdc; // Histogram 
39   AliFMDEdepMap fMap;
40 public:
41   //__________________________________________________________________
42   DrawHitsSDigits(Int_t n=900, Double_t emin=1e-3, Double_t emax=10, 
43                  Int_t m=1100, Double_t amin=-0.5, Double_t amax=1099.5) 
44   { 
45     AddLoad(kSDigits);
46     AddLoad(kHits);
47     TArrayF eloss(MakeLogScale(n, emin, emax));
48     TArrayF adcs(m+1);
49     adcs[0] = amin;
50     for (Int_t i = 1; i < m+1; i++) adcs[i] = adcs[i-1] + (amax-amin)/m;
51     fElossVsAdc = new TH2D("bad", "#Delta E vs. ADC", 
52                            eloss.fN-1, eloss.fArray, adcs.fN-1, adcs.fArray);
53     fElossVsAdc->SetXTitle("#Delta E/#Delta x [MeV/cm]");
54     fElossVsAdc->SetYTitle("ADC value");
55   }
56   //__________________________________________________________________
57   /** Begining of event
58       @param ev Event number
59       @return @c false on error */
60   Bool_t Begin(Int_t ev) 
61   {
62     fMap.Reset();
63     return AliFMDInput::Begin(ev);
64   }
65   //__________________________________________________________________
66   Bool_t ProcessHit(AliFMDHit* hit, TParticle*) 
67   {
68     Info("ProcessHit", "Processing hit %p", hit);
69     // Cache the energy loss 
70     if (!hit) return kFALSE;
71     UShort_t det = hit->Detector();
72     Char_t   rng = hit->Ring();
73     UShort_t sec = hit->Sector();
74     UShort_t str = hit->Strip();
75     if (str > 511) {
76       AliWarning(Form("Bad strip number %d in hit", str));
77       return kTRUE;
78     }
79     fMap(det, rng, sec, str).fEdep += hit->Edep();
80     fMap(det, rng, sec, str).fN++;
81     hit->Print();
82     return kTRUE;
83   }
84   //__________________________________________________________________
85   Bool_t ProcessSDigit(AliFMDSDigit* sdigit)
86   {
87     Info("ProcessSDigit", "Processing sdigit %p", sdigit);
88     if (!sdigit) return kFALSE;
89     UShort_t det = sdigit->Detector();
90     Char_t   rng = sdigit->Ring();
91     UShort_t sec = sdigit->Sector();
92     UShort_t str = sdigit->Strip();
93     if (str > 511) {
94       AliWarning(Form("Bad strip number %d in sdigit", str));
95       return kFALSE;
96     }
97     fElossVsAdc->Fill(fMap(det, rng, sec, str).fEdep, sdigit->Counts());
98     sdigit->Print();
99     return kTRUE;
100   }
101   //__________________________________________________________________
102   Bool_t Finish()
103   {
104     gStyle->SetPalette(1);
105     gStyle->SetOptTitle(0);
106     gStyle->SetCanvasColor(0);
107     gStyle->SetCanvasBorderSize(0);
108     gStyle->SetPadColor(0);
109     gStyle->SetPadBorderSize(0);
110     fElossVsAdc->SetStats(kFALSE);
111     fElossVsAdc->Draw("COLZ");
112     return kTRUE;
113   }
114
115   ClassDef(DrawHitsSDigits,0);
116 };
117
118 //____________________________________________________________________
119 //
120 // EOF
121 //