DrawBothDigits.C

Go to the documentation of this file.

//____________________________________________________________________
//
// $Id: DrawBothDigits.C 20907 2007-09-25 08:44:03Z cholm $
//
// Script that contains a class to draw eloss from hits, versus ADC
// counts from digits, using the AliFMDInputHits class in the util library. 
//
// It draws the energy loss versus the p/(mq^2).  It can be overlayed
// with the Bethe-Bloc curve to show how the simulation behaves
// relative to the expected. 
//
// Use the script `Compile.C' to compile this class using ACLic. 
//
#include <TH2F.h>
#include <AliFMDDigit.h>
#include <AliFMDSDigit.h>
#include <AliFMDInput.h>
#include <AliFMDEdepMap.h>
#include <iostream>
#include <TStyle.h>
#include <TArrayF.h>
#include <AliLog.h>

class DrawBothDigits : public AliFMDInput
{
private:
  TH2F* fTrackNos; // Histogram 
  AliFMDEdepMap fCache;
public:
  //__________________________________________________________________
  DrawBothDigits(Int_t max=300) 
    : AliFMDInput("galice.root")
  { 
    AddLoad(kDigits);
    AddLoad(kSDigits);
    fTrackNos = new TH2F("trackNos", "Track numbers", 
       max+1, -1.5, max-.5, max+1, -1.5, max-.5);
    fTrackNos->SetXTitle("Digit track");
    fTrackNos->SetYTitle("SDigit track");
  }
  //__________________________________________________________________
  Bool_t Begin(Int_t evno)
  {
    fCache.Reset();
    return AliFMDInput::Begin(evno);
  }
  //__________________________________________________________________
  Bool_t ProcessSDigit(AliFMDSDigit* sdigit)
  {
    if (!sdigit) return kTRUE;
    AliFMDEdepHitPair& entry = fCache(sdigit->Detector(), 
              sdigit->Ring(), 
              sdigit->Sector(), 
              sdigit->Strip());
    entry.fLabels.Set(sdigit->GetNTrack());
    Info("ProcessSDigit", "Got %d SDigit tracks", sdigit->GetNTrack());
    for (size_t i = 0; i < sdigit->GetNTrack(); i++) 
      entry.fLabels.fArray[i] = sdigit->GetTrack(i);
    return kTRUE;
  }
  //__________________________________________________________________
  Bool_t ProcessDigit(AliFMDDigit* digit)
  {
    if (!digit) return kTRUE;
    AliFMDEdepHitPair& entry = fCache(digit->Detector(), 
              digit->Ring(), 
              digit->Sector(), 
              digit->Strip());
    TArrayI& stracks = entry.fLabels;
    Info("ProcessDigit", "Got %d SDigit tracks, and %d Digit tracks", 
   stracks.fN, digit->GetNTrack());
    for (Int_t i = 0; (i < stracks.fN || i < digit->GetNTrack()); i++) { 
      Int_t strack = (i < stracks.fN ? stracks.fArray[i] : -1);
      Int_t dtrack = (i < digit->GetNTrack() ? digit->GetTrack(i) : -1);
      fTrackNos->Fill(strack, dtrack);
    }
    return kTRUE;
  }
  //__________________________________________________________________
  Bool_t Finish()
  {
    gStyle->SetPalette(1);
    gStyle->SetOptTitle(0);
    gStyle->SetCanvasColor(0);
    gStyle->SetCanvasBorderSize(0);
    gStyle->SetPadColor(0);
    gStyle->SetPadBorderSize(0);
    fTrackNos->SetStats(kFALSE);
    // fTrackNos->Scale(1. / fAdc->GetEntries());
    fTrackNos->Draw("colz");
    return kTRUE;
  }

  ClassDef(DrawBothDigits,0);
};

//____________________________________________________________________
//
// EOF
//