[u/mrichter/AliRoot.git] / STEER / AliV0Comparison.C

/****************************************************************************
 *           Very important, delicate and rather obscure macro.             *
 *                                                                          *
 *                  Creates list of "findable" V0s,                         *
 *             calculates efficiency, resolutions etc.                      *
 *                                                                          *
 *   To get the list of the "findable" V0s, you should first get the list   *
 *   of "findable" tracks, which can be done by running                     *
 *   TPC/AliTPCComparison.C and ITS/AliITSComparisonV2.C macros.            *
 *                                                                          *
 *   Origin: I.Belikov, IReS, Strasbourg, Jouri.Belikov@cern.ch             *
 ****************************************************************************/

#if !defined(__CINT__) || defined(__MAKECINT__)
  #include <TMath.h>
  #include <TError.h>
  #include <Riostream.h>
  #include <TH1F.h>
  #include <TTree.h>
  #include <TParticle.h>
  #include <TPDGCode.h>
  #include <TCanvas.h>
  #include <TLine.h>
  #include <TText.h>
  #include <TBenchmark.h>
  #include <TStyle.h>
  #include <TFile.h>
  #include <TROOT.h>

  #include "AliStack.h"
  #include "AliHeader.h"
  #include "AliTrackReference.h"
  #include "AliRunLoader.h"
  #include "AliRun.h"
  #include "AliESDEvent.h"
  #include "AliESDv0.h"
#endif

Int_t GoodV0s(const Char_t *dir=".");

extern AliRun *gAlice;
extern TBenchmark *gBenchmark;
extern TROOT *gROOT;

static Int_t allgood=0;
static Int_t allfound=0;

Int_t AliV0Comparison(Int_t code=310, const Char_t *dir=".") { 
   //Lambda=3122, LambdaBar=-3122
   gBenchmark->Start("AliV0Comparison");

   ::Info("AliV0Comparison.C","Doing comparison...");


   const Double_t V0window=0.05;
   Double_t ptncut=0.13, ptpcut=0.13, kinecut=0.03; 
   Double_t V0mass=0.497672, V0width=0.020;
   switch (code) {
   case kK0Short: 
        break;
   case kLambda0:    
        V0mass=1.115683; V0width=0.015; ptpcut=0.50; kinecut=0.002; 
        break;
   case kLambda0Bar: 
        V0mass=1.115683; V0width=0.015; ptncut=0.50; kinecut=0.002;
        break;
   default: cerr<<"Invalid PDG code !\n"; return 1;
   }

   TH1F *hp=(TH1F*)gROOT->FindObject("hp");
   if (!hp) hp=new TH1F("hp","PHI Resolution",30,-30.,30.);
   hp->SetXTitle("(mrad)"); hp->SetFillColor(2);

   TH1F *hl=(TH1F*)gROOT->FindObject("hl");
   if (!hl) hl=new TH1F("hl","LAMBDA Resolution",30,-30,30);
   hl->SetXTitle("(mrad)"); hl->SetFillColor(1); hl->SetFillStyle(3013);
 
   TH1F *hpt=(TH1F*)gROOT->FindObject("hpt");
   if (!hpt) hpt=new TH1F("hpt","Relative Pt Resolution",30,-10.,10.); 
   hpt->SetXTitle("(%)"); hpt->SetFillColor(2); 

   TH1F *hx=(TH1F*)gROOT->FindObject("hx");
   if (!hx) hx=new TH1F("hx","Position Resolution (X)",30,-3.,3.);
   hx->SetXTitle("(mm)"); hx->SetFillColor(6);

   TH1F *hy=(TH1F*)gROOT->FindObject("hy");
   if (!hy) hy=new TH1F("hy","Position Resolution (Y)",30,-3.,3.); 
   hy->SetXTitle("(mm)"); hy->SetFillColor(1); hy->SetFillStyle(3013);

   TH1F *hz=(TH1F*)gROOT->FindObject("hz");
   if (!hz) hz=new TH1F("hz","Position Resolution (Z)",30,-3.,3.);
   hz->SetXTitle("(mm)"); hz->SetFillColor(6);


   Double_t pmin=0.2, pmax=4.2; Int_t nchan=20;
   TH1F *hgood=(TH1F*)gROOT->FindObject("hgood");    
   if (!hgood) hgood=new TH1F("hgood","Good Vertices",nchan,pmin,pmax);
    
   TH1F *hfound=(TH1F*)gROOT->FindObject("hfound");
   if (!hfound) hfound=new TH1F("hfound","Found Vertices",nchan,pmin,pmax);

   TH1F *hfake=(TH1F*)gROOT->FindObject("hfake");
   if (!hfake) hfake=new TH1F("hfake","Fake Vertices",nchan,pmin,pmax);

   TH1F *hg=(TH1F*)gROOT->FindObject("hg");
   if (!hg) hg=new TH1F("hg","Efficiency for Good Vertices",nchan,pmin,pmax);
   hg->SetLineColor(4); hg->SetLineWidth(2);

   TH1F *hf=(TH1F*)gROOT->FindObject("hf");
   if (!hf) hf=new TH1F("hf","Probability of Fake Vertices",nchan,pmin,pmax);
   hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);


   Double_t mmin=V0mass-V0window, mmax=V0mass+V0window;
   TH1F *v0s=(TH1F*)gROOT->FindObject("v0s");
   if (!v0s) v0s=new TH1F("v0s","V0s Effective Mass",40, mmin, mmax);
   v0s->SetXTitle("(GeV)"); v0s->SetLineColor(4); v0s->SetLineWidth(4);

   TH1F *v0sf=(TH1F*)gROOT->FindObject("v0sf");
   if (!v0sf) v0sf=new TH1F("v0sf","Fake V0s Effective Mass",40, mmin, mmax);
   v0sf->SetXTitle("(GeV)"); v0sf->SetFillColor(6);


   Char_t fname[100];
   sprintf(fname,"%s/GoodV0s.root",dir);

   TFile *refFile=TFile::Open(fname,"old");
   if (!refFile || !refFile->IsOpen()) {
   ::Info("AliV0Comparison.C","Marking good V0s (will take a while)...");
     if (GoodV0s(dir)) {
        ::Error("AliV0Comparison.C","Can't generate the reference file !");
        return 1;
     }
   }
   refFile=TFile::Open(fname,"old");
   if (!refFile || !refFile->IsOpen()) {
     ::Error("AliV0Comparison.C","Can't open the reference file !");
     return 1;
   }   
  
   TTree *v0Tree=(TTree*)refFile->Get("v0Tree");
   if (!v0Tree) {
     ::Error("AliV0Comparison.C","Can't get the reference tree !");
     return 2;
   }
   TBranch *pbranch=v0Tree->GetBranch("positive");
   if (!pbranch) {
     ::Error("AliV0Comparison.C","Can't get the positive daughter branch !");
     return 3;
   }
   TClonesArray dummy("AliTrackReference",1000), *prefs=&dummy;
   pbranch->SetAddress(&prefs);

   TBranch *nbranch=v0Tree->GetBranch("negative");
   if (!nbranch) {
     ::Error("AliV0Comparison.C","Can't get the negative daughter branch !");
     return 4;
   }
   TClonesArray dumm("AliTrackReference",1000), *nrefs=&dumm;
   nbranch->SetAddress(&nrefs);

   
   sprintf(fname,"%s/AliESDs.root",dir);
   TFile *ef=TFile::Open(fname);
   if ((!ef)||(!ef->IsOpen())) {
      sprintf(fname,"%s/AliESDv0.root",dir);
      ef=TFile::Open(fname);
      if ((!ef)||(!ef->IsOpen())) {
         ::Error("AliV0Comparison.C","Can't open AliESDv0.root !");
         return 5;
      }
   }
   AliESDEvent* event = new AliESDEvent();
   TTree* esdTree = (TTree*) ef->Get("esdTree");
   if (!esdTree) {
      ::Error("AliV0Comparison.C", "no ESD tree found");
      return 6;
   }
   event->ReadFromTree(esdTree);


   //******* Loop over events *********
   Int_t e=0;
   while (esdTree->GetEvent(e)) {
      cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
 
      Int_t nentr=event->GetNumberOfV0s();
      allfound+=nentr;

      if (v0Tree->GetEvent(e++)==0) {
         cerr<<"No reconstructable V0s !\n";
         continue;
      }

      Int_t ngood=prefs->GetEntriesFast(),ng=0; 

      Double_t pxg=0.,pyg=0.,pzg=0.,ptg=0.;
      Int_t nlab=-1, plab=-1;
      Int_t i;
      for (i=0; i<nentr; i++) {
          AliESDv0 *vertex=event->GetV0(i);

          if (vertex->GetOnFlyStatus()) continue;

          Int_t nidx=TMath::Abs(vertex->GetNindex());
          Int_t pidx=TMath::Abs(vertex->GetPindex());

          AliESDtrack *ntrack=event->GetTrack(nidx);
          AliESDtrack *ptrack=event->GetTrack(pidx);

          nlab=TMath::Abs(ntrack->GetLabel());
          plab=TMath::Abs(ptrack->GetLabel());

          /** Kinematical cuts **/
          Double_t pxn,pyn,pzn; vertex->GetNPxPyPz(pxn,pyn,pzn); 
          Double_t ptn=TMath::Sqrt(pxn*pxn + pyn*pyn);
          if (ptn < ptncut) continue;
          Double_t pxp,pyp,pzp; vertex->GetPPxPyPz(pxp,pyp,pzp); 
          Double_t ptp=TMath::Sqrt(pxp*pxp + pyp*pyp);
          if (ptp < ptpcut) continue;
          Double_t kine=vertex->ChangeMassHypothesis(code);
          //if (TMath::Abs(kine)>kinecut) continue;

          Double_t mass=vertex->GetEffMass();
          v0s->Fill(mass);
          v0sf->Fill(mass);

          AliTrackReference *nref=0, *pref=0;
          Int_t j;
          for (j=0; j<ngood; j++) {
              nref=(AliTrackReference*)nrefs->UncheckedAt(j);
              pref=(AliTrackReference*)prefs->UncheckedAt(j);
              if (nref->Label() == nlab)
              if (pref->Label() == plab) break;
          }

          if (TMath::Abs(mass-V0mass)>V0width) continue;

          Double_t px,py,pz; vertex->GetPxPyPz(px,py,pz);
          Double_t pt=TMath::Sqrt(px*px+py*py);

          if (j==ngood) {
             hfake->Fill(pt);
             cout<<"Fake vertex: ("<<nlab<<","<<plab<<")\n";
             continue;
          }
          v0sf->Fill(mass,-1);

          pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py();
          pzg=nref->Pz()+pref->Pz(); 
          ptg=TMath::Sqrt(pxg*pxg+pyg*pyg);
          Double_t phig=TMath::ATan2(pyg,pxg), phi=TMath::ATan2(py,px);
          Double_t lamg=TMath::ATan2(pzg,ptg), lam=TMath::ATan2(pz,pt);
          hp->Fill((phi - phig)*1000.);
          hl->Fill((lam - lamg)*1000.);
          hpt->Fill((1/pt - 1/ptg)/(1/ptg)*100.);

          Double_t x,y,z; vertex->GetXYZ(x,y,z);
          hx->Fill((x - nref->X())*10);
          hy->Fill((y - nref->Y())*10);
          hz->Fill((z - nref->Z())*10);

          hfound->Fill(ptg);
          nref->SetLabel(-1);

      }
      for (i=0; i<ngood; i++) {
         AliTrackReference *nref=(AliTrackReference*)nrefs->UncheckedAt(i);
         AliTrackReference *pref=(AliTrackReference*)prefs->UncheckedAt(i);
         Int_t pdg=(Int_t)nref->GetLength();//this is the mother's PDG !
         if (code!=pdg) continue;
         ng++;
         pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py(); 
         ptg=TMath::Sqrt(pxg*pxg+pyg*pyg);
         hgood->Fill(ptg);
         nlab=nref->Label(); plab=pref->Label();
         if (nlab < 0) continue;
         cout<<"Vertex ("<<nlab<<','<<plab<<") has not been found !\n";
      }
      allgood+=ng;

      cout<<"Number of found V0s : "<<nentr<<endl;
      cout<<"Number of \"good\" V0s : "<<ng<<endl;

      prefs->Clear();
      nrefs->Clear();

   } //**** End of the loop over events

   delete event;
   delete esdTree;
   ef->Close();
   
   delete v0Tree;
   refFile->Close();

   Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
   if (ng!=0) cout<<"Integral efficiency is about "<<nf/ng*100.<<" %\n";
   cout<<"Total number of found V0s: "<<allfound<<" ("<<nf<<" in the peak)\n";
   cout<<"Total number of \"good\" V0s: "<<allgood<<endl;

   gStyle->SetOptStat(111110);
   gStyle->SetOptFit(1);

   TCanvas *c1=(TCanvas*)gROOT->FindObject("c1");
   if (!c1) {
      c1=new TCanvas("c1","",0,0,580,610);
      c1->Divide(2,2);
   }

   Int_t minc=33; 

   c1->cd(1);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10); 
   if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus");
   hl->Draw("same"); c1->cd();

   c1->cd(2);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
   if (hpt->GetEntries()<minc) hpt->Draw(); else hpt->Fit("gaus");
   c1->cd();

   c1->cd(3);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10); 
   if (hx->GetEntries()<minc) hx->Draw(); else hx->Fit("gaus"); 
   hy->Draw("same"); c1->cd();

   c1->cd(4);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
   if (hz->GetEntries()<minc) hz->Draw(); else hz->Fit("gaus");

   c1->Update();   

   TCanvas *c2=(TCanvas*)gROOT->FindObject("c2");
   if (!c2) {
      c2=new TCanvas("c2","",600,0,580,610);
      c2->Divide(1,2);
   }

   c2->cd(1);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
   hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
   hg->Divide(hfound,hgood,1,1.,"b");
   hf->Divide(hfake,hgood,1,1.,"b");
   hg->SetMaximum(1.4);
   hg->SetYTitle("V0 reconstruction efficiency");
   hg->SetXTitle("Pt (GeV/c)");
   hg->Draw();

   TLine *line1 = new TLine(pmin,1.0,pmax,1.0); line1->SetLineStyle(4);
   line1->Draw("same");
   TLine *line2 = new TLine(pmin,0.9,pmax,0.9); line2->SetLineStyle(4);
   line2->Draw("same");

   hf->SetFillColor(1);
   hf->SetFillStyle(3013);
   hf->SetLineColor(2);
   hf->SetLineWidth(2);
   hf->Draw("histsame");
   TText *text = new TText(0.461176,0.248448,"Fake vertices");
   text->SetTextSize(0.05);
   text->Draw();
   text = new TText(0.453919,1.11408,"Good vertices");
   text->SetTextSize(0.05);
   text->Draw();


   c2->cd(2);
   gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
   if (v0s->GetEntries()<minc) v0s->Draw();
   else v0s->Fit("gaus","","",V0mass-V0width,V0mass+V0width);
   v0sf->Draw("same");
   Double_t max=v0s->GetMaximum();
   TLine *line3 = new TLine(V0mass-V0width,0.,V0mass-V0width,max);
   line3->Draw("same");
   TLine *line4 = new TLine(V0mass+V0width,0.,V0mass+V0width,max);
   line4->Draw("same");

   c2->Update();

   TFile fc("AliV0Comparison.root","RECREATE");
   c1->Write();
   c2->Write();
   fc.Close();

   gBenchmark->Stop("AliV0Comparison");
   gBenchmark->Show("AliV0Comparison");

   return 0;
}


Int_t GoodV0s(const Char_t *dir) {
   if (gAlice) { 
       delete AliRunLoader::Instance();
       delete gAlice;//if everything was OK here it is already NULL
       gAlice = 0x0;
   }

   Char_t fname[100];
   sprintf(fname,"%s/galice.root",dir);

   AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
   if (!rl) {
      ::Error("GoodTracksITS","Can't start session !");
      return 1;
   }

   rl->LoadgAlice();
   rl->LoadHeader();
   rl->LoadKinematics();


   Int_t nev=rl->GetNumberOfEvents();
   ::Info("GoodV0s","Number of events : %d\n",nev);  

   sprintf(fname,"%s/GoodTracksITS.root",dir);
   TFile *itsFile=TFile::Open(fname);
   if ((!itsFile)||(!itsFile->IsOpen())) {
       ::Error("GoodV0s","Can't open the GoodTracksITS.root !");
       delete rl;
       return 5; 
   }
   TClonesArray dum("AliTrackReference",1000), *itsRefs=&dum;
   TTree *itsTree=(TTree*)itsFile->Get("itsTree");
   if (!itsTree) {
       ::Error("GoodV0s","Can't get the ITS reference tree !");
       delete rl;
       return 6;
   }
   TBranch *itsBranch=itsTree->GetBranch("ITS");
   if (!itsBranch) {
      ::Error("GoodV0s","Can't get the ITS reference branch !");
      delete rl;
      return 7;
   }
   itsBranch->SetAddress(&itsRefs);


   sprintf(fname,"%s/GoodV0s.root",dir);
   TFile *v0File=TFile::Open(fname,"recreate");
   TClonesArray dummy("AliTrackReference",1000), *nrefs=&dummy;
   TClonesArray dumm("AliTrackReference",1000), *prefs=&dumm;
   TTree v0Tree("v0Tree","Tree with info about the reconstructable V0s");
   v0Tree.Branch("negative",&nrefs);
   v0Tree.Branch("positive",&prefs);


   /*** Some information about the cuts ***/
   Double_t r2min=0.9*0.9;
   Double_t r2max=2.9*2.9;


   //********  Loop over generated events
   for (Int_t e=0; e<nev; e++) {
      rl->GetEvent(e);  v0File->cd();

      Int_t np = rl->GetHeader()->GetNtrack();
      cout<<"Event "<<e<<" Number of particles: "<<np<<endl;

      itsTree->GetEvent(e);
      Int_t nk=itsRefs->GetEntriesFast();

      AliStack *stack=rl->Stack();

      AliTrackReference *nref=0, *pref=0;

      Int_t nv=0;
      while (np--) {
	//cerr<<np<<'\r';
         TParticle *p0=stack->Particle(np);

         /*** only these V0s are "good" ***/
         Int_t code=p0->GetPdgCode();
         if (code!=kK0Short)
         if (code!=kLambda0)
         if (code!=kLambda0Bar) continue; 

         /*** daughter tracks should be "good" ***/
         Int_t plab=p0->GetFirstDaughter(), nlab=p0->GetLastDaughter();
         if (nlab==plab) continue;
         if (nlab<0) continue;
         if (plab<0) continue;
         Int_t i;
	 TParticle * part = stack->Particle(plab);
	 if (part) {
	   TParticlePDG * partPDG = part->GetPDG();
	     if (partPDG && partPDG->Charge() < 0.) {
	       i=plab; plab=nlab; nlab=i;
	     }
	 }

         for (i=0; i<nk; i++) {
	     nref=(AliTrackReference*)itsRefs->UncheckedAt(i);
             if (nref->Label()==nlab) break;
         }
         if (i==nk) continue;
         for (i=0; i<nk; i++) {
	     pref=(AliTrackReference*)itsRefs->UncheckedAt(i);
             if (pref->Label()==plab) break;
         }
         if (i==nk) continue;

         /*** fiducial volume ***/
         TParticle *p=stack->Particle(nlab);
         Double_t x=p->Vx(), y=p->Vy(), z=p->Vz(), r2=x*x+y*y;
         if (r2<r2min) continue;
         if (r2>r2max) continue;
       
         Int_t pdg=p0->GetPdgCode();
         nref->SetLength(pdg);  //This will the V0's PDG !

         new((*nrefs)[nv]) AliTrackReference(*nref);
         new((*prefs)[nv]) AliTrackReference(*pref);

         nv++;
      }
      itsRefs->Clear();

      v0Tree.Fill();
      nrefs->Clear(); prefs->Clear();

   }  //**** end of the loop over generated events

   v0Tree.Write();
   v0File->Close();

   delete itsTree;
   itsFile->Close();

   delete rl;
   return 0;
}
Commit	Line	Data
ca28c5f5	1	/****************************************************************************
	2	* Very important, delicate and rather obscure macro. *
	3	* *
	4	* Creates list of "findable" V0s, *
	5	* calculates efficiency, resolutions etc. *
	6	* *
1330b1b0	7	* To get the list of the "findable" V0s, you should first get the list *
	8	* of "findable" tracks, which can be done by running *
	9	* TPC/AliTPCComparison.C and ITS/AliITSComparisonV2.C macros. *
	10	* *
ca28c5f5	11	* Origin: I.Belikov, IReS, Strasbourg, Jouri.Belikov@cern.ch *
	12	****************************************************************************/
	13
566abf75	14	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
4fa1e629	15	#include <TMath.h>
	16	#include <TError.h>
	17	#include <Riostream.h>
	18	#include <TH1F.h>
	19	#include <TTree.h>
	20	#include <TParticle.h>
1330b1b0	21	#include <TPDGCode.h>
4fa1e629	22	#include <TCanvas.h>
	23	#include <TLine.h>
	24	#include <TText.h>
	25	#include <TBenchmark.h>
	26	#include <TStyle.h>
8b462fd8	27	#include <TFile.h>
4fa1e629	28	#include <TROOT.h>
	29
	30	#include "AliStack.h"
566abf75	31	#include "AliHeader.h"
4fa1e629	32	#include "AliTrackReference.h"
566abf75	33	#include "AliRunLoader.h"
ca28c5f5	34	#include "AliRun.h"
86ad5fcb	35	#include "AliESDEvent.h"
1330b1b0	36	#include "AliESDv0.h"
ca28c5f5	37	#endif
ca28c5f5	38
4fa1e629	39	Int_t GoodV0s(const Char_t *dir=".");
ca28c5f5	40
566abf75	41	extern AliRun *gAlice;
4fa1e629	42	extern TBenchmark *gBenchmark;
4fa1e629	43	extern TROOT *gROOT;
566abf75	44
4fa1e629	45	static Int_t allgood=0;
4fa1e629	46	static Int_t allfound=0;
5102bab6	47
4fa1e629	48	Int_t AliV0Comparison(Int_t code=310, const Char_t *dir=".") {
	49	//Lambda=3122, LambdaBar=-3122
	50	gBenchmark->Start("AliV0Comparison");
	51
	52	::Info("AliV0Comparison.C","Doing comparison...");
5102bab6	53
5102bab6	54
04b2a5f1	55	const Double_t V0window=0.05;
	56	Double_t ptncut=0.13, ptpcut=0.13, kinecut=0.03;
	57	Double_t V0mass=0.497672, V0width=0.020;
ca28c5f5	58	switch (code) {
04b2a5f1	59	case kK0Short:
	60	break;
	61	case kLambda0:
	62	V0mass=1.115683; V0width=0.015; ptpcut=0.50; kinecut=0.002;
	63	break;
	64	case kLambda0Bar:
	65	V0mass=1.115683; V0width=0.015; ptncut=0.50; kinecut=0.002;
	66	break;
ca28c5f5	67	default: cerr<<"Invalid PDG code !\n"; return 1;
	68	}
	69
4fa1e629	70	TH1F hp=(TH1F)gROOT->FindObject("hp");
4fa1e629	71	if (!hp) hp=new TH1F("hp","PHI Resolution",30,-30.,30.);
566abf75	72	hp->SetXTitle("(mrad)"); hp->SetFillColor(2);
4fa1e629	73
	74	TH1F hl=(TH1F)gROOT->FindObject("hl");
	75	if (!hl) hl=new TH1F("hl","LAMBDA Resolution",30,-30,30);
	76	hl->SetXTitle("(mrad)"); hl->SetFillColor(1); hl->SetFillStyle(3013);
	77
	78	TH1F hpt=(TH1F)gROOT->FindObject("hpt");
	79	if (!hpt) hpt=new TH1F("hpt","Relative Pt Resolution",30,-10.,10.);
566abf75	80	hpt->SetXTitle("(%)"); hpt->SetFillColor(2);
566abf75	81
4fa1e629	82	TH1F hx=(TH1F)gROOT->FindObject("hx");
4fa1e629	83	if (!hx) hx=new TH1F("hx","Position Resolution (X)",30,-3.,3.);
566abf75	84	hx->SetXTitle("(mm)"); hx->SetFillColor(6);
4fa1e629	85
	86	TH1F hy=(TH1F)gROOT->FindObject("hy");
	87	if (!hy) hy=new TH1F("hy","Position Resolution (Y)",30,-3.,3.);
566abf75	88	hy->SetXTitle("(mm)"); hy->SetFillColor(1); hy->SetFillStyle(3013);
4fa1e629	89
	90	TH1F hz=(TH1F)gROOT->FindObject("hz");
	91	if (!hz) hz=new TH1F("hz","Position Resolution (Z)",30,-3.,3.);
566abf75	92	hz->SetXTitle("(mm)"); hz->SetFillColor(6);
	93
	94
	95	Double_t pmin=0.2, pmax=4.2; Int_t nchan=20;
4fa1e629	96	TH1F hgood=(TH1F)gROOT->FindObject("hgood");
	97	if (!hgood) hgood=new TH1F("hgood","Good Vertices",nchan,pmin,pmax);
	98
	99	TH1F hfound=(TH1F)gROOT->FindObject("hfound");
	100	if (!hfound) hfound=new TH1F("hfound","Found Vertices",nchan,pmin,pmax);
	101
	102	TH1F hfake=(TH1F)gROOT->FindObject("hfake");
	103	if (!hfake) hfake=new TH1F("hfake","Fake Vertices",nchan,pmin,pmax);
	104
	105	TH1F hg=(TH1F)gROOT->FindObject("hg");
	106	if (!hg) hg=new TH1F("hg","Efficiency for Good Vertices",nchan,pmin,pmax);
566abf75	107	hg->SetLineColor(4); hg->SetLineWidth(2);
4fa1e629	108
	109	TH1F hf=(TH1F)gROOT->FindObject("hf");
	110	if (!hf) hf=new TH1F("hf","Probability of Fake Vertices",nchan,pmin,pmax);
566abf75	111	hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
566abf75	112
4fa1e629	113
566abf75	114	Double_t mmin=V0mass-V0window, mmax=V0mass+V0window;
4fa1e629	115	TH1F v0s=(TH1F)gROOT->FindObject("v0s");
	116	if (!v0s) v0s=new TH1F("v0s","V0s Effective Mass",40, mmin, mmax);
	117	v0s->SetXTitle("(GeV)"); v0s->SetLineColor(4); v0s->SetLineWidth(4);
	118
	119	TH1F v0sf=(TH1F)gROOT->FindObject("v0sf");
	120	if (!v0sf) v0sf=new TH1F("v0sf","Fake V0s Effective Mass",40, mmin, mmax);
566abf75	121	v0sf->SetXTitle("(GeV)"); v0sf->SetFillColor(6);
	122
	123
4fa1e629	124	Char_t fname[100];
	125	sprintf(fname,"%s/GoodV0s.root",dir);
	126
	127	TFile *refFile=TFile::Open(fname,"old");
	128	if (!refFile \|\| !refFile->IsOpen()) {
	129	::Info("AliV0Comparison.C","Marking good V0s (will take a while)...");
	130	if (GoodV0s(dir)) {
	131	::Error("AliV0Comparison.C","Can't generate the reference file !");
	132	return 1;
	133	}
	134	}
	135	refFile=TFile::Open(fname,"old");
	136	if (!refFile \|\| !refFile->IsOpen()) {
	137	::Error("AliV0Comparison.C","Can't open the reference file !");
	138	return 1;
	139	}
	140
	141	TTree v0Tree=(TTree)refFile->Get("v0Tree");
	142	if (!v0Tree) {
	143	::Error("AliV0Comparison.C","Can't get the reference tree !");
	144	return 2;
	145	}
	146	TBranch *pbranch=v0Tree->GetBranch("positive");
	147	if (!pbranch) {
	148	::Error("AliV0Comparison.C","Can't get the positive daughter branch !");
	149	return 3;
	150	}
	151	TClonesArray dummy("AliTrackReference",1000), *prefs=&dummy;
	152	pbranch->SetAddress(&prefs);
ca28c5f5	153
4fa1e629	154	TBranch *nbranch=v0Tree->GetBranch("negative");
	155	if (!nbranch) {
	156	::Error("AliV0Comparison.C","Can't get the negative daughter branch !");
	157	return 4;
ca28c5f5	158	}
4fa1e629	159	TClonesArray dumm("AliTrackReference",1000), *nrefs=&dumm;
	160	nbranch->SetAddress(&nrefs);
	161
	162
	163	sprintf(fname,"%s/AliESDs.root",dir);
	164	TFile *ef=TFile::Open(fname);
	165	if ((!ef)\|\|(!ef->IsOpen())) {
	166	sprintf(fname,"%s/AliESDv0.root",dir);
	167	ef=TFile::Open(fname);
	168	if ((!ef)\|\|(!ef->IsOpen())) {
	169	::Error("AliV0Comparison.C","Can't open AliESDv0.root !");
	170	return 5;
	171	}
ca28c5f5	172	}
4514157e	173	AliESDEvent* event = new AliESDEvent();
8b462fd8	174	TTree* esdTree = (TTree*) ef->Get("esdTree");
	175	if (!esdTree) {
	176	::Error("AliV0Comparison.C", "no ESD tree found");
	177	return 6;
	178	}
86ad5fcb	179	event->ReadFromTree(esdTree);
4fa1e629	180
	181
	182	//***** Loop over events *******
	183	Int_t e=0;
8b462fd8	184	while (esdTree->GetEvent(e)) {
4fa1e629	185	cout<<endl<<endl<<"******* Processing event number: "<<e<<"*****\n";
4fa1e629	186
4fa1e629	187	Int_t nentr=event->GetNumberOfV0s();
4fa1e629	188	allfound+=nentr;
ca28c5f5	189
4fa1e629	190	if (v0Tree->GetEvent(e++)==0) {
	191	cerr<<"No reconstructable V0s !\n";
	192	continue;
	193	}
ca28c5f5	194
4fa1e629	195	Int_t ngood=prefs->GetEntriesFast(),ng=0;
ca28c5f5	196
4fa1e629	197	Double_t pxg=0.,pyg=0.,pzg=0.,ptg=0.;
	198	Int_t nlab=-1, plab=-1;
	199	Int_t i;
	200	for (i=0; i<nentr; i++) {
	201	AliESDv0 *vertex=event->GetV0(i);
	202
d6a49f20	203	if (vertex->GetOnFlyStatus()) continue;
d6a49f20	204
4fa1e629	205	Int_t nidx=TMath::Abs(vertex->GetNindex());
	206	Int_t pidx=TMath::Abs(vertex->GetPindex());
	207
	208	AliESDtrack *ntrack=event->GetTrack(nidx);
	209	AliESDtrack *ptrack=event->GetTrack(pidx);
	210
	211	nlab=TMath::Abs(ntrack->GetLabel());
	212	plab=TMath::Abs(ptrack->GetLabel());
	213
	214	/ Kinematical cuts /
	215	Double_t pxn,pyn,pzn; vertex->GetNPxPyPz(pxn,pyn,pzn);
	216	Double_t ptn=TMath::Sqrt(pxnpxn + pynpyn);
	217	if (ptn < ptncut) continue;
	218	Double_t pxp,pyp,pzp; vertex->GetPPxPyPz(pxp,pyp,pzp);
	219	Double_t ptp=TMath::Sqrt(pxppxp + pyppyp);
	220	if (ptp < ptpcut) continue;
	221	Double_t kine=vertex->ChangeMassHypothesis(code);
	222	//if (TMath::Abs(kine)>kinecut) continue;
	223
	224	Double_t mass=vertex->GetEffMass();
	225	v0s->Fill(mass);
	226	v0sf->Fill(mass);
	227
	228	AliTrackReference nref=0, pref=0;
	229	Int_t j;
	230	for (j=0; j<ngood; j++) {
	231	nref=(AliTrackReference*)nrefs->UncheckedAt(j);
	232	pref=(AliTrackReference*)prefs->UncheckedAt(j);
	233	if (nref->Label() == nlab)
	234	if (pref->Label() == plab) break;
	235	}
	236
	237	if (TMath::Abs(mass-V0mass)>V0width) continue;
	238
	239	Double_t px,py,pz; vertex->GetPxPyPz(px,py,pz);
	240	Double_t pt=TMath::Sqrt(pxpx+pypy);
	241
	242	if (j==ngood) {
	243	hfake->Fill(pt);
	244	cout<<"Fake vertex: ("<<nlab<<","<<plab<<")\n";
	245	continue;
	246	}
	247	v0sf->Fill(mass,-1);
	248
	249	pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py();
	250	pzg=nref->Pz()+pref->Pz();
	251	ptg=TMath::Sqrt(pxgpxg+pygpyg);
	252	Double_t phig=TMath::ATan2(pyg,pxg), phi=TMath::ATan2(py,px);
	253	Double_t lamg=TMath::ATan2(pzg,ptg), lam=TMath::ATan2(pz,pt);
	254	hp->Fill((phi - phig)*1000.);
	255	hl->Fill((lam - lamg)*1000.);
	256	hpt->Fill((1/pt - 1/ptg)/(1/ptg)*100.);
	257
	258	Double_t x,y,z; vertex->GetXYZ(x,y,z);
	259	hx->Fill((x - nref->X())*10);
	260	hy->Fill((y - nref->Y())*10);
	261	hz->Fill((z - nref->Z())*10);
	262
	263	hfound->Fill(ptg);
	264	nref->SetLabel(-1);
	265
	266	}
	267	for (i=0; i<ngood; i++) {
	268	AliTrackReference nref=(AliTrackReference)nrefs->UncheckedAt(i);
269	AliTrackReference pref=(AliTrackReference)prefs->UncheckedAt(i);
270	Int_t pdg=(Int_t)nref->GetLength();//this is the mother's PDG !
271	if (code!=pdg) continue;
272	ng++;
273	pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py();
274	ptg=TMath::Sqrt(pxgpxg+pygpyg);
275	hgood->Fill(ptg);
276	nlab=nref->Label(); plab=pref->Label();
277	if (nlab < 0) continue;
278	cout<<"Vertex ("<<nlab<<','<<plab<<") has not been found !\n";
279	}
280	allgood+=ng;
281
282	cout<<"Number of found V0s : "<<nentr<<endl;
283	cout<<"Number of \"good\" V0s : "<<ng<<endl;
284
285	prefs->Clear();
286	nrefs->Clear();
4fa1e629	287
	288	} //**** End of the loop over events
	289
8b462fd8	290	delete event;
4514157e	291	delete esdTree;
4fa1e629	292	ef->Close();
	293
	294	delete v0Tree;
	295	refFile->Close();
ca28c5f5	296
4fa1e629	297	Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
	298	if (ng!=0) cout<<"Integral efficiency is about "<<nf/ng*100.<<" %\n";
	299	cout<<"Total number of found V0s: "<<allfound<<" ("<<nf<<" in the peak)\n";
	300	cout<<"Total number of \"good\" V0s: "<<allgood<<endl;
ca28c5f5	301
	302	gStyle->SetOptStat(111110);
	303	gStyle->SetOptFit(1);
	304
4fa1e629	305	TCanvas c1=(TCanvas)gROOT->FindObject("c1");
	306	if (!c1) {
	307	c1=new TCanvas("c1","",0,0,580,610);
	308	c1->Divide(2,2);
	309	}
	310
	311	Int_t minc=33;
ca28c5f5	312
	313	c1->cd(1);
	314	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629	315	if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus");
ca28c5f5	316	hl->Draw("same"); c1->cd();
	317
	318	c1->cd(2);
	319	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629	320	if (hpt->GetEntries()<minc) hpt->Draw(); else hpt->Fit("gaus");
4fa1e629	321	c1->cd();
ca28c5f5	322
	323	c1->cd(3);
	324	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629	325	if (hx->GetEntries()<minc) hx->Draw(); else hx->Fit("gaus");
ca28c5f5	326	hy->Draw("same"); c1->cd();
	327
	328	c1->cd(4);
	329	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629	330	if (hz->GetEntries()<minc) hz->Draw(); else hz->Fit("gaus");
ca28c5f5	331
4fa1e629	332	c1->Update();
ca28c5f5	333
4fa1e629	334	TCanvas c2=(TCanvas)gROOT->FindObject("c2");
	335	if (!c2) {
	336	c2=new TCanvas("c2","",600,0,580,610);
	337	c2->Divide(1,2);
	338	}
ca28c5f5	339
	340	c2->cd(1);
	341	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
	342	hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
	343	hg->Divide(hfound,hgood,1,1.,"b");
	344	hf->Divide(hfake,hgood,1,1.,"b");
	345	hg->SetMaximum(1.4);
	346	hg->SetYTitle("V0 reconstruction efficiency");
	347	hg->SetXTitle("Pt (GeV/c)");
	348	hg->Draw();
	349
	350	TLine *line1 = new TLine(pmin,1.0,pmax,1.0); line1->SetLineStyle(4);
	351	line1->Draw("same");
	352	TLine *line2 = new TLine(pmin,0.9,pmax,0.9); line2->SetLineStyle(4);
	353	line2->Draw("same");
	354
	355	hf->SetFillColor(1);
	356	hf->SetFillStyle(3013);
	357	hf->SetLineColor(2);
	358	hf->SetLineWidth(2);
	359	hf->Draw("histsame");
	360	TText *text = new TText(0.461176,0.248448,"Fake vertices");
	361	text->SetTextSize(0.05);
	362	text->Draw();
	363	text = new TText(0.453919,1.11408,"Good vertices");
	364	text->SetTextSize(0.05);
	365	text->Draw();
	366
	367
	368	c2->cd(2);
	369	gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629	370	if (v0s->GetEntries()<minc) v0s->Draw();
4fa1e629	371	else v0s->Fit("gaus","","",V0mass-V0width,V0mass+V0width);
04b2a5f1	372	v0sf->Draw("same");
ca28c5f5	373	Double_t max=v0s->GetMaximum();
	374	TLine *line3 = new TLine(V0mass-V0width,0.,V0mass-V0width,max);
	375	line3->Draw("same");
	376	TLine *line4 = new TLine(V0mass+V0width,0.,V0mass+V0width,max);
	377	line4->Draw("same");
	378
4fa1e629	379	c2->Update();
	380
	381	TFile fc("AliV0Comparison.root","RECREATE");
	382	c1->Write();
	383	c2->Write();
	384	fc.Close();
	385
	386	gBenchmark->Stop("AliV0Comparison");
	387	gBenchmark->Show("AliV0Comparison");
ca28c5f5	388
	389	return 0;
	390	}
	391
ca28c5f5	392
4fa1e629	393	Int_t GoodV0s(const Char_t *dir) {
4fa1e629	394	if (gAlice) {
33c3c91a	395	delete AliRunLoader::Instance();
4fa1e629	396	delete gAlice;//if everything was OK here it is already NULL
4fa1e629	397	gAlice = 0x0;
ca28c5f5	398	}
ca28c5f5	399
4fa1e629	400	Char_t fname[100];
	401	sprintf(fname,"%s/galice.root",dir);
	402
	403	AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
5102bab6	404	if (!rl) {
4fa1e629	405	::Error("GoodTracksITS","Can't start session !");
4fa1e629	406	return 1;
ca28c5f5	407	}
5102bab6	408
566abf75	409	rl->LoadgAlice();
	410	rl->LoadHeader();
	411	rl->LoadKinematics();
4fa1e629	412
	413
	414	Int_t nev=rl->GetNumberOfEvents();
	415	::Info("GoodV0s","Number of events : %d\n",nev);
	416
	417	sprintf(fname,"%s/GoodTracksITS.root",dir);
	418	TFile *itsFile=TFile::Open(fname);
	419	if ((!itsFile)\|\|(!itsFile->IsOpen())) {
	420	::Error("GoodV0s","Can't open the GoodTracksITS.root !");
	421	delete rl;
	422	return 5;
	423	}
	424	TClonesArray dum("AliTrackReference",1000), *itsRefs=&dum;
	425	TTree itsTree=(TTree)itsFile->Get("itsTree");
	426	if (!itsTree) {
	427	::Error("GoodV0s","Can't get the ITS reference tree !");
	428	delete rl;
	429	return 6;
	430	}
	431	TBranch *itsBranch=itsTree->GetBranch("ITS");
	432	if (!itsBranch) {
	433	::Error("GoodV0s","Can't get the ITS reference branch !");
	434	delete rl;
	435	return 7;
	436	}
	437	itsBranch->SetAddress(&itsRefs);
	438
	439
	440	sprintf(fname,"%s/GoodV0s.root",dir);
	441	TFile *v0File=TFile::Open(fname,"recreate");
	442	TClonesArray dummy("AliTrackReference",1000), *nrefs=&dummy;
	443	TClonesArray dumm("AliTrackReference",1000), *prefs=&dumm;
	444	TTree v0Tree("v0Tree","Tree with info about the reconstructable V0s");
	445	v0Tree.Branch("negative",&nrefs);
	446	v0Tree.Branch("positive",&prefs);
	447
	448
	449	/* Some information about the cuts */
	450	Double_t r2min=0.9*0.9;
	451	Double_t r2max=2.9*2.9;
	452
	453
	454
	455	//******** Loop over generated events
	456	for (Int_t e=0; e<nev; e++) {
	457	rl->GetEvent(e); v0File->cd();
	458
	459	Int_t np = rl->GetHeader()->GetNtrack();
	460	cout<<"Event "<<e<<" Number of particles: "<<np<<endl;
	461
	462	itsTree->GetEvent(e);
	463	Int_t nk=itsRefs->GetEntriesFast();
	464
	465	AliStack *stack=rl->Stack();
	466
	467	AliTrackReference nref=0, pref=0;
	468
	469	Int_t nv=0;
	470	while (np--) {
	471	//cerr<<np<<'\r';
	472	TParticle *p0=stack->Particle(np);
	473
	474	/* only these V0s are "good" */
	475	Int_t code=p0->GetPdgCode();
476	if (code!=kK0Short)
477	if (code!=kLambda0)
478	if (code!=kLambda0Bar) continue;
479
480	/* daughter tracks should be "good" */
481	Int_t plab=p0->GetFirstDaughter(), nlab=p0->GetLastDaughter();
482	if (nlab==plab) continue;
483	if (nlab<0) continue;
484	if (plab<0) continue;
485	Int_t i;
60f349ea	486	TParticle * part = stack->Particle(plab);
	487	if (part) {
	488	TParticlePDG * partPDG = part->GetPDG();
	489	if (partPDG && partPDG->Charge() < 0.) {
	490	i=plab; plab=nlab; nlab=i;
	491	}
	492	}
4fa1e629	493
	494	for (i=0; i<nk; i++) {
	495	nref=(AliTrackReference*)itsRefs->UncheckedAt(i);
	496	if (nref->Label()==nlab) break;
	497	}
	498	if (i==nk) continue;
	499	for (i=0; i<nk; i++) {
	500	pref=(AliTrackReference*)itsRefs->UncheckedAt(i);
	501	if (pref->Label()==plab) break;
	502	}
	503	if (i==nk) continue;
	504
	505	/* fiducial volume */
	506	TParticle *p=stack->Particle(nlab);
	507	Double_t x=p->Vx(), y=p->Vy(), z=p->Vz(), r2=xx+yy;
	508	if (r2<r2min) continue;
	509	if (r2>r2max) continue;
ca28c5f5	510
4fa1e629	511	Int_t pdg=p0->GetPdgCode();
	512	nref->SetLength(pdg); //This will the V0's PDG !
	513
	514	new((nrefs)[nv]) AliTrackReference(nref);
	515	new((prefs)[nv]) AliTrackReference(pref);
	516
	517	nv++;
ca28c5f5	518	}
4fa1e629	519	itsRefs->Clear();
ca28c5f5	520
4fa1e629	521	v0Tree.Fill();
	522	nrefs->Clear(); prefs->Clear();
	523
	524	} //**** end of the loop over generated events
	525
	526	v0Tree.Write();
	527	v0File->Close();
	528
	529	delete itsTree;
	530	itsFile->Close();
	531
	532	delete rl;
	533	return 0;
ca28c5f5	534	}