]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/AliITSComparisonV2.C
Transition to NewIO
[u/mrichter/AliRoot.git] / ITS / AliITSComparisonV2.C
index ee287dca77ad2fb15611607c55db672bcf1fe478..9dcdb9ff60aae6ba9a7a5d1240da446322974076 100644 (file)
@@ -1,17 +1,22 @@
 #ifndef __CINT__
-  #include <iostream.h>
+  #include <Riostream.h>
   #include <fstream.h>
 
   #include "AliRun.h"
+  #include "AliRunLoader.h"
+  #include "AliLoader.h"
+  #include "AliITSLoader.h"
   #include "AliITS.h"
   #include "AliITSgeom.h"
   #include "AliITStrackerV2.h"
   #include "AliITStrackV2.h"
   #include "AliITSclusterV2.h"
+  #include "AliMagF.h"
 
   #include "TFile.h"
   #include "TTree.h"
   #include "TH1.h"
+  #include "TH2.h"
   #include "TObjArray.h"
   #include "TStyle.h"
   #include "TCanvas.h"
   #include "TParticle.h"
 #endif
 
-struct GoodTrack {
+struct GoodTrackITS {
   Int_t lab;
   Int_t code;
   Float_t px,py,pz;
   Float_t x,y,z;
 };
-Int_t good_tracks(GoodTrack *gt, Int_t max);
 
-Int_t AliITSComparisonV2() {
+Int_t AliITSComparisonV2(Int_t event=0) {
    cerr<<"Doing comparison...\n";
+   if (gAlice) 
+    {
+      delete gAlice->GetRunLoader();
+      delete gAlice; 
+      gAlice=0;
+    }
+   
+   AliRunLoader *rl = AliRunLoader::Open("galice.root");
+   if (!rl) 
+     {
+       cerr<<"AliITSComparisonV2.C :Can't start sesion !\n";
+       return 1;
+     }
+   rl->LoadgAlice();
+   if (rl->GetAliRun()) 
+    AliKalmanTrack::SetConvConst(1000/0.299792458/rl->GetAliRun()->Field()->SolenoidField());
+   else
+    {
+       cerr<<"AliITSComparisonV2.C :Can't get AliRun !\n";
+       return 1;
+    }
+   rl->UnloadgAlice();
+    
+   AliITSLoader* itsl = (AliITSLoader*)rl->GetLoader("ITSLoader");
+   if (itsl == 0x0)
+     {
+       cerr<<"AliITSComparisonV2.C : Can not find TPCLoader\n";
+       delete rl;
+       return 3;
+     }
+   
+   const Int_t MAX=15000;
+   Int_t good_tracks_its(GoodTrackITS *gt, const Int_t max, const char* evfoldname = AliConfig::fgkDefaultEventFolderName);//declaration only
+
+   Int_t nentr=0; TObjArray tarray(2000);
+   {/* Load tracks */ 
+     itsl->LoadTracks();
+     TTree *tracktree=itsl->TreeT();
+     if (!tracktree) {cerr<<"Can't get a tree with ITS tracks !\n"; return 4;}
+     TBranch *tbranch=tracktree->GetBranch("tracks");
+     nentr=(Int_t)tracktree->GetEntries();
+
+     for (Int_t i=0; i<nentr; i++) {
+        AliITStrackV2 *iotrack=new AliITStrackV2;
+        tbranch->SetAddress(&iotrack);
+        tracktree->GetEvent(i);
+        tarray.AddLast(iotrack);
+     }
+     itsl->UnloadTracks();
+   }
 
-   TFile *cf=TFile::Open("AliITSclustersV2.root");
-   if (!cf->IsOpen()) {cerr<<"Can't open AliITSclustersV2.root !\n"; return 1;}
-   AliITSgeom *geom=(AliITSgeom*)cf->Get("AliITSgeom");
-   if (!geom) { cerr<<"Can't get the ITS geometry !\n"; return 2; }
-   AliITStrackerV2 tracker(geom);   
-
-// Load tracks
-   TFile *tf=TFile::Open("AliITStracksV2.root");
-   if (!tf->IsOpen()) {cerr<<"Can't open AliITStracksV2.root !\n"; return 3;}
-   TObjArray tarray(2000);
-   TTree *tracktree=(TTree*)tf->Get("TreeT");
-   TBranch *tbranch=tracktree->GetBranch("tracks");
-   Int_t nentr=(Int_t)tracktree->GetEntries(),i;
-   for (i=0; i<nentr; i++) {
-       AliITStrackV2 *iotrack=new AliITStrackV2;
-       tbranch->SetAddress(&iotrack);
-       tracktree->GetEvent(i);
-
-       Int_t tpcLabel=iotrack->GetLabel();
-       tracker.CookLabel(iotrack,0.);
-       Int_t itsLabel=iotrack->GetLabel();
-       if (itsLabel != tpcLabel) iotrack->SetLabel(-TMath::Abs(itsLabel));
-       if (tpcLabel < 0)         iotrack->SetLabel(-TMath::Abs(itsLabel));
-       /*
-       if (itsLabel==1234) {
-         Int_t nc=iotrack->GetNumberOfClusters();
-         for (Int_t k=0; k<nc; k++) {
-           Int_t index=iotrack->GetClusterIndex(k);
-           AliITSclusterV2 *c=tracker.GetCluster(index);
-           cout<<c->GetLabel(0)<<' '<<c->GetLabel(1)<<' '<<c->GetLabel(2)<<endl;
-         }
-       }
-       */
-       tarray.AddLast(iotrack);
-   }   
-   tf->Close();
-   cf->Close();
-
-/////////////////////////////////////////////////////////////////////////
-   GoodTrack gt[15000];
+   /* Generate a list of "good" tracks */
+   GoodTrackITS gt[MAX];
    Int_t ngood=0;
    ifstream in("good_tracks_its");
    if (in) {
@@ -80,7 +96,7 @@ Int_t AliITSComparisonV2() {
                  gt[ngood].x >>gt[ngood].y >>gt[ngood].z) {
          ngood++;
          cerr<<ngood<<'\r';
-         if (ngood==15000) {
+         if (ngood==MAX) {
             cerr<<"Too many good tracks !\n";
             break;
          }
@@ -88,17 +104,16 @@ Int_t AliITSComparisonV2() {
       if (!in.eof()) cerr<<"Read error (good_tracks_its) !\n";
    } else {
       cerr<<"Marking good tracks (this will take a while)...\n";
-      ngood=good_tracks(gt,15000);
+      ngood=good_tracks_its(gt,MAX,AliConfig::fgkDefaultEventFolderName);
       ofstream out("good_tracks_its");
       if (out) {
-         for (Int_t ngd=0; ngd<ngood; ngd++)            
-           out<<gt[ngd].lab<<' '<<gt[ngd].code<<' '<<
-                 gt[ngd].px<<' '<<gt[ngd].py<<' '<<gt[ngd].pz<<' '<<
-                 gt[ngd].x <<' '<<gt[ngd].y <<' '<<gt[ngd].z <<endl;
+       for (Int_t ngd=0; ngd<ngood; ngd++)
+         out<<gt[ngd].lab<<' '<<gt[ngd].code<<' '
+             <<gt[ngd].px<<' '<<gt[ngd].py<<' '<<gt[ngd].pz<<' '
+             <<gt[ngd].x <<' '<<gt[ngd].y <<' '<<gt[ngd].z <<endl;
       } else cerr<<"Can not open file (good_tracks_its) !\n";
       out.close();
    }
-   cerr<<"Number of good tracks : "<<ngood<<endl;
 
    TH1F *hp=new TH1F("hp","PHI resolution",50,-20.,20.); hp->SetFillColor(4);
    TH1F *hl=new TH1F("hl","LAMBDA resolution",50,-20,20);hl->SetFillColor(4);
@@ -107,41 +122,69 @@ Int_t AliITSComparisonV2() {
    TH1F *hmpt=new TH1F("hmpt","Transverse impact parameter",30,-300,300); 
    hmpt->SetFillColor(6);
    TH1F *hz=new TH1F("hz","Longitudinal impact parameter",30,-300,300); 
-   //hmpt->SetFillColor(6);
 
-   TH1F *hgood=new TH1F("hgood","Good tracks",30,0.1,6.1);    
-   TH1F *hfound=new TH1F("hfound","Found tracks",30,0.1,6.1);
-   TH1F *hfake=new TH1F("hfake","Fake tracks",30,0.1,6.1);
-   TH1F *hg=new TH1F("hg","",30,0.1,6.1); //efficiency for good tracks
+   AliITStrackV2 *trk=(AliITStrackV2*)tarray.UncheckedAt(0);
+   Double_t pmin=0.1*(100/0.299792458/0.2/trk->GetConvConst());
+   Double_t pmax=6.0+pmin;
+
+   TH1F *hgood=new TH1F("hgood","Good tracks",30,pmin,pmax);    
+   TH1F *hfound=new TH1F("hfound","Found tracks",30,pmin,pmax);
+   TH1F *hfake=new TH1F("hfake","Fake tracks",30,pmin,pmax);
+   TH1F *hg=new TH1F("hg","",30,pmin,pmax); //efficiency for good tracks
    hg->SetLineColor(4); hg->SetLineWidth(2);
-   TH1F *hf=new TH1F("hf","Efficiency for fake tracks",30,0.1,6.1);
+   TH1F *hf=new TH1F("hf","Efficiency for fake tracks",30,pmin,pmax);
    hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
 
-   TH1F *hptw=new TH1F("hptw","Weghted pt",30,0.1,6.1);
+   //TH1F *hptw=new TH1F("hptw","Weghted pt",30,pmax,pmin);
 
-   while (ngood--) {
-      Int_t lab=gt[ngood].lab, tlab=-1;
-      Double_t pxg=gt[ngood].px, pyg=gt[ngood].py, pzg=gt[ngood].pz;
+   TH1F *he =new TH1F("he","dE/dX for pions with 0.4<p<0.5 GeV/c",50,0.,100.);
+   TH2F *hep=new TH2F("hep","dE/dX vs momentum",50,0.,2.,50,0.,200.);
+   hep->SetMarkerStyle(8);
+   hep->SetMarkerSize(0.4);
+
+
+   Int_t notf[MAX], nnotf=0;
+   Int_t fake[MAX], nfake=0;
+   Int_t mult[MAX], numb[MAX], nmult=0;
+
+   Int_t ng;
+   for (ng=0; ng<ngood; ng++) {
+      Int_t lab=gt[ng].lab, tlab=-1;
+      Double_t pxg=gt[ng].px, pyg=gt[ng].py, pzg=gt[ng].pz;
       Double_t ptg=TMath::Sqrt(pxg*pxg+pyg*pyg);
 
-      hgood->Fill(ptg);
+
+      if (ptg>pmin) hgood->Fill(ptg);
 
       AliITStrackV2 *track=0;
+      Int_t cnt=0;
       Int_t j;
       for (j=0; j<nentr; j++) {
-          track=(AliITStrackV2*)tarray.UncheckedAt(j);
-          tlab=track->GetLabel();
-          if (lab==TMath::Abs(tlab)) break;
+          AliITStrackV2 *trk=(AliITStrackV2*)tarray.UncheckedAt(j);
+          Int_t lbl=trk->GetLabel();
+          if (lab==TMath::Abs(lbl)) {
+           if (cnt==0) {track=trk; tlab=lbl;}
+            cnt++;
+          }
       }
-      if (j==nentr) {
-       cerr<<"Track "<<lab<<" was not found !\n";
+      if (cnt==0) {
+        notf[nnotf++]=lab;
         continue;
+      } else if (cnt>1){
+        mult[nmult]=lab;
+        numb[nmult]=cnt; nmult++;        
       }
-      track->Propagate(track->GetAlpha(),3.,0.1/65.19*1.848,0.1*1.848);
-      track->PropagateToVertex();
 
-      if (lab==tlab) hfound->Fill(ptg);
-      else { hfake->Fill(ptg); cerr<<lab<<" fake\n";}
+      if (ptg>pmin) {
+        if (lab==tlab) hfound->Fill(ptg);
+        else {
+          fake[nfake++]=lab;
+          hfake->Fill(ptg); 
+        }
+      }
+
+      track->PropagateTo(3.,0.0028,65.19);
+      track->PropagateToVertex();
 
       Double_t xv,par[5]; track->GetExternalParameters(xv,par);
       Float_t phi=TMath::ASin(par[2]) + track->GetAlpha();
@@ -159,20 +202,45 @@ Int_t AliITSComparisonV2() {
       Double_t d=10000*track->GetD();
       hmpt->Fill(d);
 
-      hptw->Fill(ptg,TMath::Abs(d));
+      //hptw->Fill(ptg,TMath::Abs(d));
 
       Double_t z=10000*track->GetZ();
       hz->Fill(z);
 
-//if (TMath::Abs(gt[ngood].code)==11 && ptg>4.)
       hpt->Fill((pt_1 - 1/ptg)/(1/ptg)*100.);
 
+      Float_t mom=1./(pt_1*TMath::Cos(lam));
+      Float_t dedx=track->GetdEdx();
+      hep->Fill(mom,dedx,1.);
+      if (TMath::Abs(gt[ng].code)==211)
+         if (mom>0.4 && mom<0.5) he->Fill(dedx,1.);
+   }
+
+
+   cout<<"\nList of Not found tracks :\n";
+   for (ng=0; ng<nnotf; ng++){
+     cout<<notf[ng]<<"\t";
+     if ((ng%9)==8) cout<<"\n";
+   }
+   cout<<"\n\nList of fake  tracks :\n";
+   for (ng=0; ng<nfake; ng++){
+     cout<<fake[ng]<<"\t";
+     if ((ng%9)==8) cout<<"\n";
+   }
+   cout<<"\n\nList of multiple found tracks :\n";
+   for (ng=0; ng<nmult; ng++) {
+       cout<<"id.   "<<mult[ng]
+           <<"     found - "<<numb[ng]<<"times\n";
    }
+   cout<<endl;
+
+   cerr<<"Number of good tracks : "<<ngood<<endl;
+   cerr<<"Number of found tracks : "<<nentr<<endl;
 
-   Stat_t ng=hgood->GetEntries(); cerr<<"Good tracks "<<ng<<endl;
-   Stat_t nf=hfound->GetEntries();
-   if (ng!=0) 
-      cerr<<"Integral efficiency is about "<<nf/ng*100.<<" %\n";
+   ng=(Int_t)hgood->GetEntries(); //cerr<<"Good tracks "<<ng<<endl;
+   if (ng!=0)  
+   cerr<<"Integral efficiency is about "<<hfound->GetEntries()/ng*100.<<" %\n";
+   cerr<<endl;
 
    gStyle->SetOptStat(111110);
    gStyle->SetOptFit(1);
@@ -214,9 +282,9 @@ Int_t AliITSComparisonV2() {
    hg->SetXTitle("Pt (GeV/c)");
    hg->Draw();
 
-   TLine *line1 = new TLine(0,1.0,7,1.0); line1->SetLineStyle(4);
+   TLine *line1 = new TLine(pmin,1.0,pmax,1.0); line1->SetLineStyle(4);
    line1->Draw("same");
-   TLine *line2 = new TLine(0,0.9,7,0.9); line2->SetLineStyle(4);
+   TLine *line2 = new TLine(pmin,0.9,pmax,0.9); line2->SetLineStyle(4);
    line2->Draw("same");
 
    hf->SetFillColor(1);
@@ -231,39 +299,65 @@ Int_t AliITSComparisonV2() {
    text->SetTextSize(0.05);
    text->Draw();
 
+//     PH taken from v3-09-09, but not used
+//    TCanvas *c2=new TCanvas("c2","",320,32,530,590);
+
+//    TPad *p6=new TPad("p6","",0.,0.,1.,.5); p6->Draw();
+//    p6->cd(); p6->SetFillColor(42); p6->SetFrameFillColor(10); 
+//    he->SetFillColor(2); he->SetFillStyle(3005);  
+//    he->SetXTitle("Arbitrary Units"); 
+//    he->Fit("gaus"); c2->cd();
+
+//    TPad *p7=new TPad("p7","",0.,0.5,1.,1.); p7->Draw(); 
+//    p7->cd(); p7->SetFillColor(42); p7->SetFrameFillColor(10);
+//    hep->SetXTitle("p (Gev/c)"); hep->SetYTitle("dE/dX (Arb. Units)"); 
+//    hep->Draw(); c1->cd();
+   
+
    return 0;
 }
 
-Int_t good_tracks(GoodTrack *gt, Int_t max) {
-   if (gAlice) {delete gAlice; gAlice=0;}
+Int_t good_tracks_its(GoodTrackITS *gt, const Int_t max, const char* evfoldname) {
+
+   Int_t nt=0;
+
+   AliRunLoader* rl = AliRunLoader::GetRunLoader(evfoldname);
+   if (rl == 0x0)
+    {
+      ::Fatal("AliTPCComparison.C::good_tracks_its",
+              "Can not find Run Loader in Folder Named %s",
+              evfoldname);
+    }
+
+   AliITSLoader* itsl = (AliITSLoader*)rl->GetLoader("ITSLoader");
+   if (itsl == 0x0)
+     {
+       cerr<<"AliITSComparisonV2.C : Can not find TPCLoader\n";
+       delete rl;
+       return 3;
+     }
+   
+   rl->LoadgAlice();
+   rl->LoadHeader();
+   Int_t np = rl->GetHeader()->GetNtrack();
 
-   TFile *file=TFile::Open("galice.root");
-   if (!file->IsOpen()) {cerr<<"Can't open galice.root !\n"; exit(4);}
-   if (!(gAlice=(AliRun*)file->Get("gAlice"))) {
-     cerr<<"gAlice have not been found on galice.root !\n";
-     exit(5);
-   }
 
-   Int_t np=gAlice->GetEvent(0);
 
    Int_t *good=new Int_t[np];
    Int_t k;
    for (k=0; k<np; k++) good[k]=0;
 
-   AliITS *ITS=(AliITS*)gAlice->GetDetector("ITS");
+   AliITS *ITS=(AliITS*)rl->GetAliRun()->GetDetector("ITS");
    if (!ITS) {
       cerr<<"can't get ITS !\n"; exit(8);
    }
    AliITSgeom *geom=ITS->GetITSgeom();
    if (!geom) {
-      cerr<<"cen't get ITS geometry !\n"; exit(9);
+      cerr<<"can't get ITS geometry !\n"; exit(9);
    }
 
-   TFile *cf=TFile::Open("AliITSclustersV2.root");
-   if (!cf->IsOpen()){
-      cerr<<"Can't open AliITSclustersV2.root !\n"; exit(6);
-   }
-   TTree *cTree=(TTree*)cf->Get("cTree");
+   itsl->LoadRawClusters();
+   TTree *cTree=itsl->TreeC();
    if (!cTree) {
       cerr<<"Can't get cTree !\n"; exit(7);
    }
@@ -276,12 +370,12 @@ Int_t good_tracks(GoodTrack *gt, Int_t max) {
 
    Int_t entr=(Int_t)cTree->GetEntries();
    for (k=0; k<entr; k++) {
-     if (!cTree->GetEvent(k)) continue;
-     Int_t lay,lad,det;  geom->GetModuleId(k-1,lay,lad,det);
+     cTree->GetEvent(k);
+     Int_t ncl=clusters->GetEntriesFast(); if (ncl==0) continue;
+     Int_t lay,lad,det;  geom->GetModuleId(k,lay,lad,det);
      if (lay<1 || lay>6) {
        cerr<<"wrong layer !\n"; exit(10);
      }
-     Int_t ncl=clusters->GetEntriesFast();
      while (ncl--) {
         AliITSclusterV2 *pnt=(AliITSclusterV2*)clusters->UncheckedAt(ncl);
         Int_t l0=pnt->GetLabel(0);
@@ -294,18 +388,28 @@ Int_t good_tracks(GoodTrack *gt, Int_t max) {
      }
    }
    clusters->Delete(); delete clusters;
-   cf->Close();
-
+   itsl->UnloadRawClusters();
+   
    ifstream in("good_tracks_tpc");
    if (!in) {
      cerr<<"can't get good_tracks_tpc !\n"; exit(11);
    }
+   
+   rl->LoadKinematics();
+   AliStack* stack = rl->Stack();
    Int_t nt=0;
    Double_t px,py,pz,x,y,z;
    Int_t code,lab;
    while (in>>lab>>code>>px>>py>>pz>>x>>y>>z) {
       if (good[lab] != 0x3F) continue;
-      TParticle *p = (TParticle*)gAlice->Particle(lab);
+      TParticle *p = (TParticle*)stack->Particle(lab);
+      if (p == 0x0)
+       {
+         cerr<<"Can not get particle "<<lab<<endl;
+         nt++;
+         if (nt==max) {cerr<<"Too many good tracks !\n"; break;}
+         continue;
+       }
       gt[nt].lab=lab;
       gt[nt].code=p->GetPdgCode();
 //**** px py pz - in global coordinate system
@@ -317,8 +421,9 @@ Int_t good_tracks(GoodTrack *gt, Int_t max) {
 
    delete[] good;
 
-   delete gAlice; gAlice=0;
-   file->Close();
+   rl->UnloadKinematics();
+   rl->UnloadHeader();
+   rl->UnloadgAlice();
 
    return nt;
 }