TestEMCALData.C

Go to the documentation of this file.

#if !defined(__CINT__) || defined(__MAKECINT__)

//Root include files 
#include <TH1I.h>
#include <TH1F.h>
#include <TCanvas.h>
#include <TClonesArray.h>
#include <Riostream.h>

//AliRoot include files 
#include "AliStack.h"
#include "AliRunLoader.h"
#include "AliEMCALLoader.h"
#include "AliEMCAL.h"
#include "AliEMCALHit.h"
#include "AliEMCALDigit.h"
#include "AliEMCALRecPoint.h"

#endif

void TestEMCALData() {
  
  TH1F* hE = new TH1F("hEmcalHits",    "Hits energy distribution in EMCAL",       100, 0., 10.) ;
  hE->Sumw2() ;
  TH1I* hM = new TH1I("hEmcalHitsMul", "Hits multiplicity distribution in EMCAL", 500, 0., 10000) ;
  hM->Sumw2() ;
  
  TH1I * dA = new TH1I("hEmcalDigits",    "Digits amplitude distribution in EMCAL",    500, 0, 5000) ;
  dA->Sumw2() ;
  TH1I * dM = new TH1I("hEmcalDigitsMul", "Digits multiplicity distribution in EMCAL", 500, 0, 1000) ;
  dM->Sumw2() ;
  
  TH1F * sE = new TH1F("hEmcalSDigits",    "SDigits energy distribution in EMCAL",    200, 0, 1000) ;
  sE->Sumw2() ;
  TH1I * sM = new TH1I("hEmcalSDigitsMul", "SDigits multiplicity distribution in EMCAL", 500, 0, 1000) ;
  sM->Sumw2() ;
  
  TH1F * cE = new TH1F("hEmcalRecPoints",    "RecPoints energy distribution in EMCAL",    100, 0, 20) ;
  cE->Sumw2() ;
  TH1I * cM = new TH1I("hEmcalRecPointsMul", "RecPoints multiplicity distribution in EMCAL", 500, 0, 1000) ;
  cM->Sumw2() ;
  
  AliRunLoader *rl = AliRunLoader::Open("galice.root",AliConfig::GetDefaultEventFolderName(),"read");
  if (rl == 0x0)
    cout<<"Can not instatiate the Run Loader"<<endl;
  
  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>
    (rl->GetDetectorLoader("EMCAL"));
  
  // Load Hits
  rl->LoadHits("EMCAL");
  //Load Digits
  rl->LoadDigits("EMCAL");
  //Load SDigits
  rl->LoadSDigits("EMCAL");
  //Load RecPoints
  rl->LoadRecPoints("EMCAL");
  
  // One way to do it that works for all three
  AliEMCALHit* hit;
  AliEMCALDigit* dig;
  AliEMCALRecPoint* rp;
  
  rl->GetEvent(0);

  TClonesArray *hits = 0;  
  // Fill array of digits                                                                        
  TClonesArray *sdigits = emcalLoader->SDigits();
  // Fill array of digits                                                                        
  TClonesArray *digits = emcalLoader->Digits();
  // Fill array of clusters/rec-points                                                                        
  // TObjArray *clusters = emcalLoader->RecPoints(); //It should work, need to FIX
  TObjArray *clusters = 0;
  
  //
  // Get hits from the list  
  // Hits are stored in different branches in the hits Tree, 
  // first get the branch and then access the hits in the branch
  //
  Int_t nHit = 0;
  TTree *treeH = emcalLoader->TreeH();  
  if (treeH) 
  {
    // TreeH exists, get the branch
    Int_t nTrack = treeH->GetEntries();  // TreeH has array of hits for every primary
    TBranch * branchH = treeH->GetBranch("EMCAL");
    branchH->SetAddress(&hits);
    
    // Now get the hits in this branch
    for (Int_t iTrack = 0; iTrack < nTrack; iTrack++)
    {
      branchH->GetEntry(iTrack);
      Int_t nHit = hits->GetEntriesFast();
      for(Int_t ihit = 0; ihit< nHit;ihit++)
      {
        hit = static_cast<AliEMCALHit *> (hits->At(ihit));//hits->At(ihit);
        if(!hit) continue;
        
        nHit++;
        hE->Fill(hit->GetEnergy());
        cout<<"Hit Info "<<hit->GetId()<<" ELoss "<<hit->GetEnergy()<<endl;
      }//hit loop
    }// track loop
  }//treeH?
  
  hM->Fill(nHit);
  
  //
  // Get sdigits from the list 
  //
  if(sdigits)
  {
    sM->Fill(sdigits->GetEntries());
    for(Int_t isdig = 0; isdig< sdigits->GetEntries();isdig++)
    {
      //cout<<">> idig "<<idig<<endl;                                                             
      dig = static_cast<AliEMCALDigit *>(sdigits->At(isdig)) ;
      
      if(!dig) continue;
      
      sE->Fill(dig->GetAmplitude());
      cout << "SDigit info " << dig->GetId() << " " << dig->GetAmplitude() << endl;
    }
  }
  else printf("No Sdigits available\n");
  
  //
  // Get digits from the list
  //
  if(digits)
  {
    dM->Fill(digits->GetEntries());
    for(Int_t idig = 0; idig< digits->GetEntries();idig++)
    {
      //cout<<">> idig "<<idig<<endl;                                                             
      dig = static_cast<AliEMCALDigit *>(digits->At(idig)) ;
      
      if(!dig) continue;
      dA->Fill(dig->GetAmplitude());
      cout << "Digit info " << dig->GetId() << " " << dig->GetAmplitude() << endl;
    }
  }
  else printf("No digits available\n");
  
  // Get clusters from the list 
  TTree *treeR = emcalLoader->TreeR();
  TBranch * branchR = treeR->GetBranch("EMCALECARP"); 
  branchR->SetAddress(&clusters);
  branchR->GetEntry(0);
  
  if(clusters)
  {
    cM->Fill(clusters->GetEntries());
    for(Int_t iclu = 0; iclu< clusters->GetEntries();iclu++)
    {
      //cout<<">> idig "<<idig<<endl;                                                             
      rp = static_cast<AliEMCALRecPoint *>(clusters->At(iclu)) ;
      
      if(!rp) continue;
      cE->Fill(rp->GetEnergy());
      cout << "RecPoint info " << rp->GetAbsId(0) << " " << rp->GetEnergy() << endl;
    }
  }
  else printf("No recpoints available\n");
  
  /*
   //another way to do it
   TTree *hTree = rl->GetTreeH("EMCAL",false);
   TTree *dTree = rl->GetTreeD("EMCAL",false);
   TTree *sTree = rl->GetTreeS("EMCAL",false);
   TTree *cTree = rl->GetTreeR("EMCAL",false);
   
   TObjArray *harr=NULL;
   TBranch *hbranch=hTree->GetBranch("EMCAL");
   hbranch->SetAddress(&harr);
   
   TObjArray *darr=NULL;
   TBranch *dbranch=dTree->GetBranch("EMCAL");
   dbranch->SetAddress(&darr);
   
   TObjArray *sarr=NULL;
   TBranch *sbranch=sTree->GetBranch("EMCAL");
   sbranch->SetAddress(&sarr);
   
   TObjArray *carr=NULL;
   TBranch *cbranch=cTree->GetBranch("EMCALECARP");
   cbranch->SetAddress(&carr);
   
   
   if(hbranch->GetEvent(0)) {
   for(Int_t ih = 0; ih < harr->GetEntriesFast(); ih++) {
   hM->Fill(harr->GetEntriesFast());
   AliEMCALHit* hit =(AliEMCALHit*)harr->UncheckedAt(ih);
   if(hit != 0){
   hE->Fill(hit->GetEnergy());
   cout << "Hit info " << hit->GetId() << " " << hit->GetEnergy()*10.5 << endl;
   }
   }
   }
   
   if(dbranch->GetEvent(0)) {
   for(Int_t id = 0; id < darr->GetEntriesFast(); id++) {
   dM->Fill(darr->GetEntriesFast());
   AliEMCALDigit* dig =(AliEMCALDigit*)darr->UncheckedAt(id);
   if(dig != 0){
   dA->Fill(dig->GetAmp());
   cout << "Digit info " << dig->GetId() << " " << dig->GetAmp() << endl;
   }
   }
   }
   
   if(sbranch->GetEvent(0)) {
   for(Int_t id = 0; id < sarr->GetEntriesFast(); id++) {
   sM->Fill(sarr->GetEntriesFast());
   AliEMCALDigit* sdig =(AliEMCALDigit*)sarr->UncheckedAt(id);
   if(sdig != 0){
   sE->Fill(sdig->GetAmp()/1.e+6);
   cout << "SDigit info " << sdig->GetId() << " " << sdig->GetAmp()/1.e+6 << endl;
   }
   }
   }
   
   if(cbranch->GetEvent(0)) {
   for(Int_t ic = 0; ic < carr->GetEntriesFast(); ic++) {
   cE->Fill(carr->GetEntriesFast());
   AliEMCALRecPoint* rp =(AliEMCALRecPoint*)carr->UncheckedAt(ic);
   if(rp != 0){
   cE->Fill(rp->GetEnergy());
   cout << "RecPoint info " << rp->GetAbsId() << " " << rp->GetEnergy() << endl;
   }
   }
   }
   
   */
  
  TCanvas *chits = new TCanvas("chits","Hits",20,20,800,400);
  chits->Divide(2,1);
  chits->cd(1);
  hE->Draw();
  chits->cd(2);
  hM->Draw();
  
  TCanvas *cdig = new TCanvas("cdig","Digits",20,40,800,400);
  cdig->Divide(2,1);
  cdig->cd(1);
  dA->Draw();
  cdig->cd(2);
  dM->Draw();
  
  
  TCanvas *csdig = new TCanvas("csdig","SDigits",20,60,800,400);
  csdig->Divide(2,1);
  csdig->cd(1);
  sE->Draw();
  csdig->cd(2);
  sM->Draw();
  
  TCanvas *cclu = new TCanvas("cclu","Clusters",20,60,800,400);
  cclu->Divide(2,1);
  cclu->cd(1);
  cE->Draw();
  cclu->cd(2);
  cM->Draw();
}