A new macro to show how to access and display ITS digits. Tested with

[u/mrichter/AliRoot.git] / ITS / AliITSDigitPlot.C

diff --git a/ITS/AliITSDigitPlot.C b/ITS/AliITSDigitPlot.C
new file mode 100644 (file)
index 0000000..9a87401
--- /dev/null
+++ b/ITS/AliITSDigitPlot.C
@@ -0,0 +1,176 @@
+//
+
+#if !defined(__CINT__) || defined(__MAKECINT__)
+  #include <Riostream.h>
+  #include <TClassTable.h>
+     //#include <fstream.h>
+
+  #include "TH2.h"
+  #include "TFile.h"
+  #include "TKey.h"
+  #include "TObjArray.h"
+  #include "TStyle.h"
+  #include "TCanvas.h"
+  #include "TLine.h"
+  #include "TText.h"
+  #include "TParticle.h"
+  #include "TStopwatch.h"
+  #include "TPDGCode.h"
+
+  #include "AliRun.h"
+  #include "AliESD.h"
+  #include "AliCascadeVertex.h"
+
+  #include "AliMC.h"
+  #include "AliHeader.h"
+  #include "AliConfig.h"
+  #include "AliRunLoader.h"
+  #include "AliITSLoader.h"
+  #include "AliITS.h"
+  #include "AliITSdigit.h"
+  #include "AliITSgeom.h"
+#endif
+
+    const Int_t ndets=5;
+    TH2I *BeamSpot[ndets];
+//----------------------------------------------------------------------
+void SetUPHistograms(){
+    // Define the histograms to be filled
+    Int_t i;
+    Char_t name[10],title[50];
+
+    for(i=0;i<ndets;i++){
+        sprintf(name,"BeamSpot%d",i);
+        sprintf(title,"Beam Spot for SPD %d detectors row-column values",i);
+        BeamSpot[i] = new TH2I(name,title,256,0.,256.,160,0.,160.);
+    } // end for i
+}
+//----------------------------------------------------------------------
+void FillHistograms(Int_t module,Int_t row,Int_t colm,Int_t signal){
+    // Fill the histograms
+
+    if(module<0 || module>=ndets) return;
+    BeamSpot[module]->Fill((Double_t) row,(Double_t) colm,(Double_t) signal);
+}
+//----------------------------------------------------------------------
+void DisplayHistograms(){
+    // Display the histograms
+
+    TCanvas *c0 = new TCanvas("c0","SPD Digits",400,10,600,700);
+    BeamSpot[2]->Draw();
+}
+//----------------------------------------------------------------------
+void AliITSDigitPlot(Int_t istart=0,Int_t iend=-1,
+                     const char *filename="galice.root"){
+    // Macro to plot digits from many events
+    // Inputs:
+    //   Int_t istart   Starting event number
+    //   Int_t iend     Last event number, =-1 all
+    // Outputs:
+    //   none.
+    // Return:
+    //   none.
+    if (gClassTable->GetID("AliRun") < 0) {
+        gROOT->ProcessLine(".x $(ALICE_ROOT)/macros/loadlibs.C");
+    }
+    if(gAlice){
+        delete gAlice->GetRunLoader();
+        delete gAlice;
+        gAlice=0;
+    } // end if gAlice
+
+    Int_t nevents=0,nmodules=0,retval=0;
+    Int_t i,j,module,dig,ndig,row,column,signal,det;
+    AliITS       *its    = 0;
+    AliITSgeom   *gm     = 0;
+    AliRunLoader *rl     = 0;
+    AliITSLoader *ld     = 0;
+    TTree        *treeD  = 0;
+    TBranch      *br     = 0;
+    TClonesArray *digits = 0;
+    AliITSdigit  *d      = 0;
+    TObjArray    *digDet = 0;
+    Char_t *branchname[3] = {"ITSDigitsSPD","ITSDigitsSDD","ITSDigitsSSD"};
+    //
+    rl = AliRunLoader::Open(filename);
+    if(!rl){
+        cerr<<"Error, can not open file "<<filename<<endl;
+        return;
+    } // end if !rl
+    retval = rl->LoadgAlice();
+    if (retval){
+        cerr<<"Error, LoadgAlice returned error"<<endl;
+        return;
+    }
+    gAlice = rl->GetAliRun();
+    retval = rl->LoadHeader();
+    if (retval){
+        cerr<<"Error, LoadHeader returned error"<<endl;
+        return;
+    } // end if
+    ld = (AliITSLoader*) rl->GetLoader("ITSLoader");
+    if(!ld){
+        cerr<<"Error, ITS loader not found"<<endl;
+        return;
+    } // end if
+    its = (AliITS*) gAlice->GetModule("ITS");
+    if(!its){
+        cerr <<"Error, No AliDetector ITS found on file"<<endl;
+        return;
+    } // end if
+    gm  = its->GetITSgeom();
+    if(!gm){
+        cerr <<"Error, AliITSgeom not initilized in module ITS"<<endl;
+        return;
+    } // end if
+    nevents = rl->GetNumberOfEvents();
+    if(iend>nevents) iend = nevents;
+    if(iend<0)       iend = nevents;
+    if(iend<=istart){delete rl; return;}
+    nmodules = gm->GetIndexMax();
+    ld->GetDigitsDataLoader()->Load("read");
+    treeD = ld->TreeD();
+    if(!treeD){
+        cerr <<"Error, could not get TreeD="<<treeD << endl;
+        return;
+    } // end if !treeD
+    digDet = new TObjArray(3);
+    for(det=0;det<3;det++){
+        digDet->AddAt(new TClonesArray((its->DetType(det)->
+                                        GetDigitClassName()).Data(),1000),det);
+        br = treeD->GetBranch(branchname[det]);
+        br->SetAddress(&((*digDet)[det]));
+    } // end for det
+
+    //
+    SetUPHistograms();
+    //
+    for(i=istart;i<iend;i++){
+        rl->GetEvent(i);
+        treeD = ld->TreeD();
+        for(det=0;det<3;det++){
+            ((TClonesArray*)(digDet->At(det)))->Clear();
+            br = treeD->GetBranch(branchname[det]);
+            br->SetAddress(&((*digDet)[det]));
+        } // end for det
+        for(module=0;module<nmodules;module++){
+            for(j=0;j<3;j++) ((TClonesArray*)(digDet->At(j)))->Clear();
+            treeD->GetEvent(module);
+            digits = (TClonesArray*) (digDet->At(0)); // SPD only.
+            ndig = digits->GetEntriesFast();
+            for(dig=0;dig<ndig;dig++){
+                d = (AliITSdigit*) digits->At(dig);
+                row    = d->GetCoord1();
+                column = d->GetCoord1();
+                signal = d->GetSignal();
+                     //cout <<"event="<<i<< " ndig="<< ndig<< " mod="
+                     //<<module<<" row="<<row<<" col="<<column<< " sig="
+                     //<<signal<<endl;
+                FillHistograms(module,row,column,signal);
+            } // end for mod
+        } // end for module
+    } // end for i
+    DisplayHistograms();
+    delete digits;
+    return;
+}