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 //