Adding a time-acceptance window for good MC vertices

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

diff --git a/ITS/AliITSPrintRecPoints.C b/ITS/AliITSPrintRecPoints.C
index 523379e7630dc1d1f01dae7ebbd28f86b391b076..625bfce669f04f37f71c6d28a4f0c30f09efe35f 100644 (file)
--- a/ITS/AliITSPrintRecPoints.C
+++ b/ITS/AliITSPrintRecPoints.C
@@ -1,24 +1,70 @@
-void AliITSPrintRecPoints(TString rfn="galice.root",Int_t mod=-1,
-                         Int_t evnt=-1){
+#if !defined(__CINT__) || defined(__MAKECINT__)
+#include<Riostream.h>
+#include<TROOT.h>
+#include<TClassTable.h>
+#include<TClonesArray.h>
+#include<TGeoManager.h>
+#include <TInterpreter.h>
+#include<TTree.h>
+#include "AliRun.h"
+#include "AliITSgeom.h"
+#include "AliGeomManager.h"
+#include "AliITSDetTypeRec.h"
+#include "AliITSRecPoint.h"
+#include "AliITSdigit.h"
+#include "AliITSdigitSSD.h"
+#include "AliITShit.h"
+#include "AliITSmodule.h" 
+#include "AliITSsegmentation.h"
+#include "AliITSsegmentationSPD.h" 
+#include "AliITSsegmentationSDD.h"
+#include "AliITSsegmentationSSD.h"
+#include "AliRunLoader.h"
+#include "AliITSLoader.h"
+#include "AliHeader.h"
+#include "AliCDBManager.h"
+#include "AliCDBStorage.h"
+#endif
+void AliITSPrintRecPoints(Int_t outtype=1,TString rfn="galice.root",
+                          Int_t mod=-1,Int_t evnt=-1){
   // Macro to print out the recpoints for all or a specific module
 
   // Dynamically link some shared libs
   if (gClassTable->GetID("AliRun") < 0) {
-    gROOT->LoadMacro("loadlibs.C");
-    loadlibs();
-  } 
+    gInterpreter->ExecuteMacro("loadlibs.C");
+  }
   else {
     if(gAlice){
-      delete gAlice->GetRunLoader();
+      delete AliRunLoader::Instance();
       delete gAlice;
       gAlice=0;
     }
   }
 
-  gROOT->LoadMacro("$(ALICE_ROOT)/ITS/AliITSstandard.C");
+  // Set OCDB if needed
+  AliCDBManager* man = AliCDBManager::Instance();
+  if (!man->IsDefaultStorageSet()) {
+    printf("Setting a local default storage and run number 0\n");
+    man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
+    man->SetRun(0);
+  }
+  else {
+    printf("Using deafult storage \n");
+  }
 
-  AliRunLoader *rl = AccessFile(rfn); // Set up to read in Data
-  Int_t retval = rl->LoadHeader();
+  AliRunLoader* rl = AliRunLoader::Open(rfn.Data());
+  if (rl == 0x0){
+    cerr<<"AliITSPrintRecPoints.C : Can not open session RL=NULL"<< endl;
+    return;
+  }
+  Int_t retval = rl->LoadgAlice();
+  if (retval){
+    cerr<<"AliITSPrintRecPoints.C : LoadgAlice returned error"<<endl;
+    return;
+  }
+  gAlice=rl->GetAliRun();
+
+  retval = rl->LoadHeader();
   if (retval){
     cerr<<"AliITSPrintRecPoints.C : LoadHeader returned error"<<endl;
     return;
@@ -30,48 +76,69 @@ void AliITSPrintRecPoints(TString rfn="galice.root",Int_t mod=-1,
     cerr<<"AliITSPrintRecPoints.C :  ITS loader not found"<<endl;
     return;
   }
+  cout <<"ITSloader ok"<<endl;
 
+  if(!gGeoManager){
+    AliGeomManager::LoadGeometry("geometry.root");
+  } 
   ITSloader->LoadHits("read");
   ITSloader->LoadDigits("read");
   ITSloader->LoadRecPoints("read");
-  AliITS *ITS = (AliITS*)gAlice->GetDetector("ITS");
-  if(!ITS){
-    cout << "Error: no ITS found. Aborting"<<endl;
-    return;
-  } // end if !ITS
+  cout << "loaded hits, digits, and RecPoints"<< endl;
+ 
+  AliITSgeom *gm=0;
+  gm = ITSloader->GetITSgeom();
 
   Int_t evNumber1 = 0;
-  Int_t evNumber2 = gAlice->GetEventsPerRun();
+  Int_t evNumber2 = AliRunLoader::GetNumberOfEvents();
   if(evnt>=0){
     evNumber1 = evnt;
     evNumber2 = evnt+1;
   } // end if evnt>=0
   Int_t mod1 = 0;
-  Int_t mod2 = ITS->GetITSgeom()->GetIndexMax();
+  Int_t mod2 = gm->GetIndexMax();
   if(mod>=0){
     mod1 = mod;
     mod2 = mod+1;
   } // end if mod>=0
   TClonesArray *rpa;
   AliITSRecPoint *rp = 0;
+  AliITSDetTypeRec* rec = new AliITSDetTypeRec();
+  rec->SetITSgeom(gm);
+  rec->SetDefaults();
 
   Int_t event,m,i,i2;
+  Float_t xyz[3];
   for(event = evNumber1; event < evNumber2; event++){
     rl->GetEvent(event);
-    ITS->SetTreeAddress();
+    //    rec->SetTreeAddress();
     for(m=mod1;m<mod2;m++){
-      ITS->ResetRecPoints();
+      rec->ResetRecPoints();
       TTree *TR = ITSloader->TreeR();
+      rec->SetTreeAddressR(TR);
       TR->GetEvent(m);
-      rpa = ITS->RecPoints();
+      rpa = rec->RecPoints();
       i2 = rpa->GetEntriesFast();
       cout <<  "Event=" << event << " module=" << m <<
        " Number of Recpoints=" << i2 <<endl;
       for(i=0;i<i2;i++){
-       rp = (AliITSRecPoint*)(rpa->At(i));
-       cout << i << " ";
-       rp->Print((ostream*)cout);
-       cout << endl;
+          rp = (AliITSRecPoint*)(rpa->At(i));
+          switch(outtype){
+          case 1:
+              rp->GetGlobalXYZ(xyz);
+              cout << i << " lx=" << rp->GetDetLocalX() 
+                        << " lz=" << rp->GetDetLocalZ() 
+                   << " x=" << rp->GetX() 
+                   << " y=" << rp->GetY()<< " z=" << rp->GetZ() 
+                   <<" gx=" << xyz[0] << " gy="<< xyz[1] <<" gz="<<xyz[2]
+                   << endl;
+              break;
+          default:
+              cout << i << " ";
+              rp->Print();
+              cout << endl;
+              break;
+          } // end switch
       } // end for i
     } // end for m
   } // end for event