[u/mrichter/AliRoot.git] / STEER / AliGlobalQADataMaker.cxx

/*
 The class for calculating the global (not detector specific) quality assurance.
 It reuses the following TLists from its base class 
    AliQADataMaker::fRecPointsQAList (for keeping the track residuals)
    AliQADataMaker::fESDsQAList      (for keeping global ESD QA data)
*/

#include <TPDGCode.h>
#include <TH1F.h>

#include "AliGlobalQADataMaker.h"
#include "AliGeomManager.h"
#include "AliESDEvent.h"
#include "AliESDv0.h"

ClassImp(AliGlobalQADataMaker)
 
void AliGlobalQADataMaker::InitRecPoints() {
  //------------------------------------------------------
  // This function books the histograms of *track*residuals*
  // as a part of global QA
  //------------------------------------------------------
  Char_t *name[]={
    "SPD1 residuals Y","SPD1 residuals Z",
    "SPD2 residuals Y","SPD2 residuals Z",
    "SDD1 residuals Y","SDD1 residuals Z",
    "SDD2 residuals Y","SDD2 residuals Z",
    "SSD1 residuals Y","SSD1 residuals Z",
    "SSD2 residuals Y","SSD2 residuals Z",

    "TPC1 residuals Y","TPC1 residuals Z",
    "TPC2 residuals Y","TPC2 residuals Z",

    "TRD1 residuals Y","TRD1 residuals Z",
    "TRD2 residuals Y","TRD2 residuals Z",
    "TRD3 residuals Y","TRD3 residuals Z",
    "TRD4 residuals Y","TRD4 residuals Z",
    "TRD5 residuals Y","TRD5 residuals Z",
    "TRD6 residuals Y","TRD6 residuals Z",

    "TOF residuals Y","TOF residuals Z",

    "PHOS1 residuals Y","PHOS1 residuals Z",
    "PHOS2 residuals Y","PHOS2 residuals Z",

    "HMPID residuals Y","HMPID residuals Z",

    "MUON residuals Y","MUON residuals Z",

    "EMCAL residuals Y","EMCAL residuals Z"
  };

  for (Int_t m=1; m<AliGeomManager::kLastLayer; m++) {
    Int_t i=2*m-2;
    TH1F *h=new TH1F(name[i],name[i],100,-5.,5.);
    Add2RecPointsList(h,i);    
    h=new TH1F(name[i+1],name[i+1],100,-5.,5.);
    Add2RecPointsList(h,i+1);    
  }
}

void AliGlobalQADataMaker::InitESDs() {
  //------------------------------------------------------
  // This function books the ESD QA histograms
  // as a part of global QA
  //------------------------------------------------------

  {// Cluster related QA
  Char_t *name[]={
    "Fraction of the assigned clusters in ITS",
    "Fraction of the assigned clusters in TPC",
    "Fraction of the assigned clusters in TRD"
  };
  Add2ESDsList(new TH1F(name[0],name[0],100,0.,2.),kClr0);
  Add2ESDsList(new TH1F(name[1],name[1],100,0.,2.),kClr1);
  Add2ESDsList(new TH1F(name[2],name[2],100,0.,2.),kClr2);
  }

  {// Track related QA
  Char_t *name[]={
    "Track azimuthal distribution (rad)",                   // kTrk0
    "Track pseudo-rapidity distribution",                   // kTrk1
    "TPC: track momentum distribution (GeV/c)",             // kTrk2
    "TPC-ITS matched: track momentum distribution (GeV/c)", // kTrk3
    "TPC-TOF matched: track momentum distribution (GeV/c)", // kTrk4
    "TPC-ITS track-matching probability",                   // kTrk5
    "TPC-TOF track-matching probability"                    // kTrk6
  };
  Add2ESDsList(new TH1F(name[0],name[0],100,-0.02,6.30),kTrk0);
  Add2ESDsList(new TH1F(name[1],name[1],100,-2.00,2.00),kTrk1);
  Add2ESDsList(new TH1F(name[2],name[2],50,  0.20,5.00),kTrk2);
  Add2ESDsList(new TH1F(name[3],name[3],50,  0.20,5.00),kTrk3);
  Add2ESDsList(new TH1F(name[4],name[4],50,  0.20,5.00),kTrk4);
  Add2ESDsList(new TH1F(name[5],name[5],50,  0.20,5.00),kTrk5);
  Add2ESDsList(new TH1F(name[6],name[6],50,  0.20,5.00),kTrk6);
  }

  {// V0 related QA
  Char_t *name[]={
    "K0s mass (GeV)",
    "Lambda0 + Lambda0Bar mass (GeV)"
  };
  Add2ESDsList(new TH1F(name[0],name[0],50,  0.4477,0.5477),kV0s0);
  Add2ESDsList(new TH1F(name[1],name[1],50,  1.0657,1.1657),kV0s1);
  }

  {// PID related QA
  Char_t *name[]={
    "ITS: dE/dx (ADC) for particles with momentum 0.4 - 0.5 (GeV/c)",
    "TPC: dE/dx (ADC) for particles with momentum 0.4 - 0.5 (GeV/c)",
    "TOF: tracking - measured (ps)"
  };
  Add2ESDsList(new TH1F(name[0],name[0],50,0.00,200.),kPid0);
  Add2ESDsList(new TH1F(name[1],name[1],50,0.00,100.),kPid1);
  Add2ESDsList(new TH1F(name[2],name[2],50,-3500.,3500.),kPid2);
  }

}

void AliGlobalQADataMaker::MakeESDs(AliESDEvent * event) {
  //-----------------------------------------------------------
  // This function fills the ESD QA histograms
  // as a part of global QA
  //-----------------------------------------------------------
  const AliESDEvent *esd=event;

  Int_t ntrk=esd->GetNumberOfTracks() ; 
  for (Int_t i=0; i<ntrk; i++) {
    const AliESDtrack *track=esd->GetTrack(i);

    // Cluster related QA
    if (track->IsOn(AliESDtrack::kITSrefit)) {
      Int_t n=track->GetITSclusters(0);
      GetESDsData(kClr0)->Fill(Float_t(n)/6.); //6 is the number of ITS layers
    }

    if (track->IsOn(AliESDtrack::kTPCrefit)) {
      Int_t n =track->GetTPCNcls();
      Int_t nf=track->GetTPCNclsF();      // number of crossed TPC pad rows
      GetESDsData(kClr1)->Fill(Float_t(n)/nf);
    }

    if (track->IsOn(AliESDtrack::kTRDrefit)) {
      Int_t n=track->GetTRDclusters(0);
      GetESDsData(kClr2)->Fill(Float_t(n)/(6*24));//(6*24) is the number of TRD time bins
    }

    Double_t p=track->GetP();

    // Track related QA
    if (track->IsOn(AliESDtrack::kTPCrefit)) {
      Float_t dz[2]; 
      track->GetDZ(0.,0.,0.,esd->GetMagneticField(),dz); 
      if ((TMath::Abs(dz[0])<3.) && (TMath::Abs(dz[1])<3.)) { // beam pipe
        Double_t phi=track->Phi();
	GetESDsData(kTrk0)->Fill(phi);
	Double_t y=track->Y();
	GetESDsData(kTrk1)->Fill(y);

        if (TMath::Abs(y)<0.9) {
	   GetESDsData(kTrk2)->Fill(p);
	   if (track->IsOn(AliESDtrack::kITSrefit)) GetESDsData(kTrk3)->Fill(p);
	   if (track->IsOn(AliESDtrack::kTOFout)) GetESDsData(kTrk4)->Fill(p);
	}
      }
    }

    // PID related QA
    if ((p>0.4) && (p<0.5)) {
      if (track->IsOn(AliESDtrack::kITSpid)) {
	Double_t dedx=track->GetITSsignal();
        GetESDsData(kPid0)->Fill(dedx);
      }
      if (track->IsOn(AliESDtrack::kTPCpid)) {
	Double_t dedx=track->GetTPCsignal();
        GetESDsData(kPid1)->Fill(dedx);
      }
    }
    if (p>1.0) {
      if (track->IsOn(AliESDtrack::kTOFpid)) {
        Double_t times[10];
        track->GetIntegratedTimes(times);
        Double_t tof=track->GetTOFsignal();
        GetESDsData(kPid2)->Fill(times[2]-tof);
      }
    }
  }

  TH1 *tpc=GetESDsData(kTrk2); tpc->Sumw2();
  TH1 *its=GetESDsData(kTrk3); its->Sumw2();
  TH1 *tof=GetESDsData(kTrk4); tof->Sumw2();
  GetESDsData(kTrk5)->Divide(its,tpc,1,1.,"b");
  GetESDsData(kTrk6)->Divide(tof,tpc,1,1.,"b");

  // V0 related QA
  Int_t nV0=esd->GetNumberOfV0s();
  for (Int_t i=0; i<nV0; i++) {
    Double_t mass;
    AliESDv0 v0(*esd->GetV0(i));

    v0.ChangeMassHypothesis(kK0Short);
    mass=v0.GetEffMass();
    GetESDsData(kV0s0)->Fill(mass);

    v0.ChangeMassHypothesis(kLambda0);
    mass=v0.GetEffMass();
    GetESDsData(kV0s1)->Fill(mass);

    v0.ChangeMassHypothesis(kLambda0Bar);
    mass=v0.GetEffMass();
    GetESDsData(kV0s1)->Fill(mass);
  }

}
Commit	Line	Data
c0af1be2	1	/*
	2	The class for calculating the global (not detector specific) quality assurance.
	3	It reuses the following TLists from its base class
	4	AliQADataMaker::fRecPointsQAList (for keeping the track residuals)
	5	AliQADataMaker::fESDsQAList (for keeping global ESD QA data)
	6	*/
	7
542c6c71	8	#include <TPDGCode.h>
c0af1be2	9	#include <TH1F.h>
	10
	11	#include "AliGlobalQADataMaker.h"
	12	#include "AliGeomManager.h"
d7259b2e	13	#include "AliESDEvent.h"
542c6c71	14	#include "AliESDv0.h"
c0af1be2	15
	16	ClassImp(AliGlobalQADataMaker)
	17
	18	void AliGlobalQADataMaker::InitRecPoints() {
	19	//------------------------------------------------------
d7259b2e	20	// This function books the histograms of trackresiduals*
c0af1be2	21	// as a part of global QA
	22	//------------------------------------------------------
	23	Char_t *name[]={
	24	"SPD1 residuals Y","SPD1 residuals Z",
	25	"SPD2 residuals Y","SPD2 residuals Z",
	26	"SDD1 residuals Y","SDD1 residuals Z",
	27	"SDD2 residuals Y","SDD2 residuals Z",
	28	"SSD1 residuals Y","SSD1 residuals Z",
	29	"SSD2 residuals Y","SSD2 residuals Z",
	30
	31	"TPC1 residuals Y","TPC1 residuals Z",
	32	"TPC2 residuals Y","TPC2 residuals Z",
	33
	34	"TRD1 residuals Y","TRD1 residuals Z",
	35	"TRD2 residuals Y","TRD2 residuals Z",
	36	"TRD3 residuals Y","TRD3 residuals Z",
	37	"TRD4 residuals Y","TRD4 residuals Z",
	38	"TRD5 residuals Y","TRD5 residuals Z",
	39	"TRD6 residuals Y","TRD6 residuals Z",
	40
	41	"TOF residuals Y","TOF residuals Z",
	42
	43	"PHOS1 residuals Y","PHOS1 residuals Z",
	44	"PHOS2 residuals Y","PHOS2 residuals Z",
	45
	46	"HMPID residuals Y","HMPID residuals Z",
	47
	48	"MUON residuals Y","MUON residuals Z",
	49
	50	"EMCAL residuals Y","EMCAL residuals Z"
	51	};
	52
	53	for (Int_t m=1; m<AliGeomManager::kLastLayer; m++) {
	54	Int_t i=2*m-2;
	55	TH1F *h=new TH1F(name[i],name[i],100,-5.,5.);
	56	Add2RecPointsList(h,i);
	57	h=new TH1F(name[i+1],name[i+1],100,-5.,5.);
	58	Add2RecPointsList(h,i+1);
	59	}
	60	}
d7259b2e	61
	62	void AliGlobalQADataMaker::InitESDs() {
	63	//------------------------------------------------------
	64	// This function books the ESD QA histograms
	65	// as a part of global QA
	66	//------------------------------------------------------
542c6c71	67
542c6c71	68	{// Cluster related QA
d7259b2e	69	Char_t *name[]={
	70	"Fraction of the assigned clusters in ITS",
	71	"Fraction of the assigned clusters in TPC",
	72	"Fraction of the assigned clusters in TRD"
	73	};
542c6c71	74	Add2ESDsList(new TH1F(name[0],name[0],100,0.,2.),kClr0);
	75	Add2ESDsList(new TH1F(name[1],name[1],100,0.,2.),kClr1);
	76	Add2ESDsList(new TH1F(name[2],name[2],100,0.,2.),kClr2);
	77	}
	78
	79	{// Track related QA
	80	Char_t *name[]={
	81	"Track azimuthal distribution (rad)", // kTrk0
	82	"Track pseudo-rapidity distribution", // kTrk1
	83	"TPC: track momentum distribution (GeV/c)", // kTrk2
	84	"TPC-ITS matched: track momentum distribution (GeV/c)", // kTrk3
	85	"TPC-TOF matched: track momentum distribution (GeV/c)", // kTrk4
	86	"TPC-ITS track-matching probability", // kTrk5
	87	"TPC-TOF track-matching probability" // kTrk6
	88	};
	89	Add2ESDsList(new TH1F(name[0],name[0],100,-0.02,6.30),kTrk0);
	90	Add2ESDsList(new TH1F(name[1],name[1],100,-2.00,2.00),kTrk1);
	91	Add2ESDsList(new TH1F(name[2],name[2],50, 0.20,5.00),kTrk2);
	92	Add2ESDsList(new TH1F(name[3],name[3],50, 0.20,5.00),kTrk3);
	93	Add2ESDsList(new TH1F(name[4],name[4],50, 0.20,5.00),kTrk4);
	94	Add2ESDsList(new TH1F(name[5],name[5],50, 0.20,5.00),kTrk5);
	95	Add2ESDsList(new TH1F(name[6],name[6],50, 0.20,5.00),kTrk6);
	96	}
	97
	98	{// V0 related QA
	99	Char_t *name[]={
	100	"K0s mass (GeV)",
	101	"Lambda0 + Lambda0Bar mass (GeV)"
	102	};
	103	Add2ESDsList(new TH1F(name[0],name[0],50, 0.4477,0.5477),kV0s0);
	104	Add2ESDsList(new TH1F(name[1],name[1],50, 1.0657,1.1657),kV0s1);
	105	}
	106
	107	{// PID related QA
	108	Char_t *name[]={
	109	"ITS: dE/dx (ADC) for particles with momentum 0.4 - 0.5 (GeV/c)",
	110	"TPC: dE/dx (ADC) for particles with momentum 0.4 - 0.5 (GeV/c)",
	111	"TOF: tracking - measured (ps)"
	112	};
	113	Add2ESDsList(new TH1F(name[0],name[0],50,0.00,200.),kPid0);
	114	Add2ESDsList(new TH1F(name[1],name[1],50,0.00,100.),kPid1);
	115	Add2ESDsList(new TH1F(name[2],name[2],50,-3500.,3500.),kPid2);
	116	}
	117
d7259b2e	118	}
d7259b2e	119
542c6c71	120	void AliGlobalQADataMaker::MakeESDs(AliESDEvent * event) {
d7259b2e	121	//-----------------------------------------------------------
	122	// This function fills the ESD QA histograms
	123	// as a part of global QA
	124	//-----------------------------------------------------------
542c6c71	125	const AliESDEvent *esd=event;
542c6c71	126
d7259b2e	127	Int_t ntrk=esd->GetNumberOfTracks() ;
d7259b2e	128	for (Int_t i=0; i<ntrk; i++) {
542c6c71	129	const AliESDtrack *track=esd->GetTrack(i);
d7259b2e	130
542c6c71	131	// Cluster related QA
d7259b2e	132	if (track->IsOn(AliESDtrack::kITSrefit)) {
d7259b2e	133	Int_t n=track->GetITSclusters(0);
542c6c71	134	GetESDsData(kClr0)->Fill(Float_t(n)/6.); //6 is the number of ITS layers
d7259b2e	135	}
	136
	137	if (track->IsOn(AliESDtrack::kTPCrefit)) {
	138	Int_t n =track->GetTPCNcls();
	139	Int_t nf=track->GetTPCNclsF(); // number of crossed TPC pad rows
542c6c71	140	GetESDsData(kClr1)->Fill(Float_t(n)/nf);
d7259b2e	141	}
	142
	143	if (track->IsOn(AliESDtrack::kTRDrefit)) {
	144	Int_t n=track->GetTRDclusters(0);
e3bd5382	145	GetESDsData(kClr2)->Fill(Float_t(n)/(624));//(624) is the number of TRD time bins
542c6c71	146	}
	147
	148	Double_t p=track->GetP();
	149
	150	// Track related QA
	151	if (track->IsOn(AliESDtrack::kTPCrefit)) {
	152	Float_t dz[2];
	153	track->GetDZ(0.,0.,0.,esd->GetMagneticField(),dz);
	154	if ((TMath::Abs(dz[0])<3.) && (TMath::Abs(dz[1])<3.)) { // beam pipe
	155	Double_t phi=track->Phi();
	156	GetESDsData(kTrk0)->Fill(phi);
	157	Double_t y=track->Y();
	158	GetESDsData(kTrk1)->Fill(y);
	159
cecff4fc	160	if (TMath::Abs(y)<0.9) {
	161	GetESDsData(kTrk2)->Fill(p);
	162	if (track->IsOn(AliESDtrack::kITSrefit)) GetESDsData(kTrk3)->Fill(p);
	163	if (track->IsOn(AliESDtrack::kTOFout)) GetESDsData(kTrk4)->Fill(p);
	164	}
542c6c71	165	}
	166	}
	167
	168	// PID related QA
	169	if ((p>0.4) && (p<0.5)) {
	170	if (track->IsOn(AliESDtrack::kITSpid)) {
	171	Double_t dedx=track->GetITSsignal();
	172	GetESDsData(kPid0)->Fill(dedx);
	173	}
	174	if (track->IsOn(AliESDtrack::kTPCpid)) {
	175	Double_t dedx=track->GetTPCsignal();
	176	GetESDsData(kPid1)->Fill(dedx);
	177	}
d7259b2e	178	}
542c6c71	179	if (p>1.0) {
	180	if (track->IsOn(AliESDtrack::kTOFpid)) {
	181	Double_t times[10];
	182	track->GetIntegratedTimes(times);
	183	Double_t tof=track->GetTOFsignal();
	184	GetESDsData(kPid2)->Fill(times[2]-tof);
	185	}
	186	}
	187	}
d7259b2e	188
cecff4fc	189	TH1 *tpc=GetESDsData(kTrk2); tpc->Sumw2();
	190	TH1 *its=GetESDsData(kTrk3); its->Sumw2();
	191	TH1 *tof=GetESDsData(kTrk4); tof->Sumw2();
542c6c71	192	GetESDsData(kTrk5)->Divide(its,tpc,1,1.,"b");
	193	GetESDsData(kTrk6)->Divide(tof,tpc,1,1.,"b");
	194
	195	// V0 related QA
	196	Int_t nV0=esd->GetNumberOfV0s();
	197	for (Int_t i=0; i<nV0; i++) {
	198	Double_t mass;
	199	AliESDv0 v0(*esd->GetV0(i));
	200
	201	v0.ChangeMassHypothesis(kK0Short);
	202	mass=v0.GetEffMass();
	203	GetESDsData(kV0s0)->Fill(mass);
	204
	205	v0.ChangeMassHypothesis(kLambda0);
	206	mass=v0.GetEffMass();
	207	GetESDsData(kV0s1)->Fill(mass);
	208
	209	v0.ChangeMassHypothesis(kLambda0Bar);
	210	mass=v0.GetEffMass();
	211	GetESDsData(kV0s1)->Fill(mass);
d7259b2e	212	}
542c6c71	213
d7259b2e	214	}