[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);    
  }

  Add2RecPointsList(
  new TH1F("SSD1 absolute residuals Y for Z<0 (cm)",
           "SSD1 absolute residuals Y for Z<0 (cm)",100,-2.,2.),40);
  Add2RecPointsList(
  new TH1F("SSD1 absolute residuals Z for Z<0 (cm)",
           "SSD1 absolute residuals Z for Z<0 (cm)",100,-2.,2.),41);
  Add2RecPointsList(
  new TH1F("SSD1 absolute residuals Y for Z>0 (cm)",
           "SSD1 absolute residuals Y for Z>0 (cm)",100,-2.,2.),42);
  Add2RecPointsList(
  new TH1F("SSD1 absolute residuals Z for Z>0 (cm)",
           "SSD1 absolute residuals Z for Z>0 (cm)",100,-2.,2.),43);


  Add2RecPointsList(
  new TH1F("SSD2 absolute residuals Y for Z<0 (cm)",
           "SSD2 absolute residuals Y for Z<0 (cm)",100,-3.,3.),44);
  Add2RecPointsList(
  new TH1F("SSD2 absolute residuals Z for Z<0 (cm)",
           "SSD2 absolute residuals Z for Z<0 (cm)",100,-3.,3.),45);
  Add2RecPointsList(
  new TH1F("SSD2 absolute residuals Y for Z>0 (cm)",
           "SSD2 absolute residuals Y for Z>0 (cm)",100,-3.,3.),46);
  Add2RecPointsList(
  new TH1F("SSD2 absolute residuals Z for Z>0 (cm)",
           "SSD2 absolute residuals Z for Z>0 (cm)",100,-3.,3.),47);
  
}

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)",               // kTrk2
    "TPC-ITS matched: track momentum distribution (GeV)",   // kTrk3
    "TPC-TOF matched: track momentum distribution (GeV)",   // 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: dEdx (ADC) for particles with momentum 0.4 - 0.5 (GeV)",
    "TPC: dEdx (ADC) for particles with momentum 0.4 - 0.5 (GeV)",
    "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->Eta();
	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	}
a935c6c0	60
49ab451d	61	Add2RecPointsList(
	62	new TH1F("SSD1 absolute residuals Y for Z<0 (cm)",
	63	"SSD1 absolute residuals Y for Z<0 (cm)",100,-2.,2.),40);
a935c6c0	64	Add2RecPointsList(
a935c6c0	65	new TH1F("SSD1 absolute residuals Z for Z<0 (cm)",
49ab451d	66	"SSD1 absolute residuals Z for Z<0 (cm)",100,-2.,2.),41);
	67	Add2RecPointsList(
	68	new TH1F("SSD1 absolute residuals Y for Z>0 (cm)",
	69	"SSD1 absolute residuals Y for Z>0 (cm)",100,-2.,2.),42);
a935c6c0	70	Add2RecPointsList(
a935c6c0	71	new TH1F("SSD1 absolute residuals Z for Z>0 (cm)",
49ab451d	72	"SSD1 absolute residuals Z for Z>0 (cm)",100,-2.,2.),43);
	73
	74
	75	Add2RecPointsList(
	76	new TH1F("SSD2 absolute residuals Y for Z<0 (cm)",
	77	"SSD2 absolute residuals Y for Z<0 (cm)",100,-3.,3.),44);
a935c6c0	78	Add2RecPointsList(
a935c6c0	79	new TH1F("SSD2 absolute residuals Z for Z<0 (cm)",
49ab451d	80	"SSD2 absolute residuals Z for Z<0 (cm)",100,-3.,3.),45);
	81	Add2RecPointsList(
	82	new TH1F("SSD2 absolute residuals Y for Z>0 (cm)",
	83	"SSD2 absolute residuals Y for Z>0 (cm)",100,-3.,3.),46);
a935c6c0	84	Add2RecPointsList(
a935c6c0	85	new TH1F("SSD2 absolute residuals Z for Z>0 (cm)",
49ab451d	86	"SSD2 absolute residuals Z for Z>0 (cm)",100,-3.,3.),47);
a935c6c0	87
c0af1be2	88	}
d7259b2e	89
	90	void AliGlobalQADataMaker::InitESDs() {
	91	//------------------------------------------------------
	92	// This function books the ESD QA histograms
	93	// as a part of global QA
	94	//------------------------------------------------------
542c6c71	95
542c6c71	96	{// Cluster related QA
d7259b2e	97	Char_t *name[]={
	98	"Fraction of the assigned clusters in ITS",
	99	"Fraction of the assigned clusters in TPC",
	100	"Fraction of the assigned clusters in TRD"
	101	};
542c6c71	102	Add2ESDsList(new TH1F(name[0],name[0],100,0.,2.),kClr0);
	103	Add2ESDsList(new TH1F(name[1],name[1],100,0.,2.),kClr1);
	104	Add2ESDsList(new TH1F(name[2],name[2],100,0.,2.),kClr2);
	105	}
	106
	107	{// Track related QA
	108	Char_t *name[]={
	109	"Track azimuthal distribution (rad)", // kTrk0
	110	"Track pseudo-rapidity distribution", // kTrk1
a0994cfd	111	"TPC: track momentum distribution (GeV)", // kTrk2
	112	"TPC-ITS matched: track momentum distribution (GeV)", // kTrk3
	113	"TPC-TOF matched: track momentum distribution (GeV)", // kTrk4
542c6c71	114	"TPC-ITS track-matching probability", // kTrk5
	115	"TPC-TOF track-matching probability" // kTrk6
	116	};
	117	Add2ESDsList(new TH1F(name[0],name[0],100,-0.02,6.30),kTrk0);
	118	Add2ESDsList(new TH1F(name[1],name[1],100,-2.00,2.00),kTrk1);
	119	Add2ESDsList(new TH1F(name[2],name[2],50, 0.20,5.00),kTrk2);
	120	Add2ESDsList(new TH1F(name[3],name[3],50, 0.20,5.00),kTrk3);
	121	Add2ESDsList(new TH1F(name[4],name[4],50, 0.20,5.00),kTrk4);
	122	Add2ESDsList(new TH1F(name[5],name[5],50, 0.20,5.00),kTrk5);
	123	Add2ESDsList(new TH1F(name[6],name[6],50, 0.20,5.00),kTrk6);
	124	}
	125
	126	{// V0 related QA
	127	Char_t *name[]={
	128	"K0s mass (GeV)",
	129	"Lambda0 + Lambda0Bar mass (GeV)"
	130	};
	131	Add2ESDsList(new TH1F(name[0],name[0],50, 0.4477,0.5477),kV0s0);
	132	Add2ESDsList(new TH1F(name[1],name[1],50, 1.0657,1.1657),kV0s1);
	133	}
	134
	135	{// PID related QA
	136	Char_t *name[]={
a0994cfd	137	"ITS: dEdx (ADC) for particles with momentum 0.4 - 0.5 (GeV)",
a0994cfd	138	"TPC: dEdx (ADC) for particles with momentum 0.4 - 0.5 (GeV)",
542c6c71	139	"TOF: tracking - measured (ps)"
	140	};
	141	Add2ESDsList(new TH1F(name[0],name[0],50,0.00,200.),kPid0);
	142	Add2ESDsList(new TH1F(name[1],name[1],50,0.00,100.),kPid1);
	143	Add2ESDsList(new TH1F(name[2],name[2],50,-3500.,3500.),kPid2);
	144	}
	145
d7259b2e	146	}
d7259b2e	147
542c6c71	148	void AliGlobalQADataMaker::MakeESDs(AliESDEvent * event) {
d7259b2e	149	//-----------------------------------------------------------
	150	// This function fills the ESD QA histograms
	151	// as a part of global QA
	152	//-----------------------------------------------------------
542c6c71	153	const AliESDEvent *esd=event;
542c6c71	154
d7259b2e	155	Int_t ntrk=esd->GetNumberOfTracks() ;
d7259b2e	156	for (Int_t i=0; i<ntrk; i++) {
542c6c71	157	const AliESDtrack *track=esd->GetTrack(i);
d7259b2e	158
542c6c71	159	// Cluster related QA
d7259b2e	160	if (track->IsOn(AliESDtrack::kITSrefit)) {
d7259b2e	161	Int_t n=track->GetITSclusters(0);
542c6c71	162	GetESDsData(kClr0)->Fill(Float_t(n)/6.); //6 is the number of ITS layers
d7259b2e	163	}
	164
	165	if (track->IsOn(AliESDtrack::kTPCrefit)) {
	166	Int_t n =track->GetTPCNcls();
	167	Int_t nf=track->GetTPCNclsF(); // number of crossed TPC pad rows
542c6c71	168	GetESDsData(kClr1)->Fill(Float_t(n)/nf);
d7259b2e	169	}
	170
	171	if (track->IsOn(AliESDtrack::kTRDrefit)) {
	172	Int_t n=track->GetTRDclusters(0);
e3bd5382	173	GetESDsData(kClr2)->Fill(Float_t(n)/(624));//(624) is the number of TRD time bins
542c6c71	174	}
	175
	176	Double_t p=track->GetP();
	177
	178	// Track related QA
	179	if (track->IsOn(AliESDtrack::kTPCrefit)) {
	180	Float_t dz[2];
	181	track->GetDZ(0.,0.,0.,esd->GetMagneticField(),dz);
	182	if ((TMath::Abs(dz[0])<3.) && (TMath::Abs(dz[1])<3.)) { // beam pipe
	183	Double_t phi=track->Phi();
	184	GetESDsData(kTrk0)->Fill(phi);
44f14df0	185	Double_t y=track->Eta();
542c6c71	186	GetESDsData(kTrk1)->Fill(y);
542c6c71	187
cecff4fc	188	if (TMath::Abs(y)<0.9) {
	189	GetESDsData(kTrk2)->Fill(p);
	190	if (track->IsOn(AliESDtrack::kITSrefit)) GetESDsData(kTrk3)->Fill(p);
	191	if (track->IsOn(AliESDtrack::kTOFout)) GetESDsData(kTrk4)->Fill(p);
	192	}
542c6c71	193	}
	194	}
	195
	196	// PID related QA
	197	if ((p>0.4) && (p<0.5)) {
	198	if (track->IsOn(AliESDtrack::kITSpid)) {
	199	Double_t dedx=track->GetITSsignal();
	200	GetESDsData(kPid0)->Fill(dedx);
	201	}
	202	if (track->IsOn(AliESDtrack::kTPCpid)) {
	203	Double_t dedx=track->GetTPCsignal();
	204	GetESDsData(kPid1)->Fill(dedx);
	205	}
d7259b2e	206	}
542c6c71	207	if (p>1.0) {
	208	if (track->IsOn(AliESDtrack::kTOFpid)) {
	209	Double_t times[10];
	210	track->GetIntegratedTimes(times);
	211	Double_t tof=track->GetTOFsignal();
	212	GetESDsData(kPid2)->Fill(times[2]-tof);
	213	}
	214	}
	215	}
d7259b2e	216
cecff4fc	217	TH1 *tpc=GetESDsData(kTrk2); tpc->Sumw2();
	218	TH1 *its=GetESDsData(kTrk3); its->Sumw2();
	219	TH1 *tof=GetESDsData(kTrk4); tof->Sumw2();
542c6c71	220	GetESDsData(kTrk5)->Divide(its,tpc,1,1.,"b");
	221	GetESDsData(kTrk6)->Divide(tof,tpc,1,1.,"b");
	222
	223	// V0 related QA
	224	Int_t nV0=esd->GetNumberOfV0s();
	225	for (Int_t i=0; i<nV0; i++) {
	226	Double_t mass;
	227	AliESDv0 v0(*esd->GetV0(i));
	228
	229	v0.ChangeMassHypothesis(kK0Short);
	230	mass=v0.GetEffMass();
	231	GetESDsData(kV0s0)->Fill(mass);
	232
	233	v0.ChangeMassHypothesis(kLambda0);
	234	mass=v0.GetEffMass();
	235	GetESDsData(kV0s1)->Fill(mass);
	236
	237	v0.ChangeMassHypothesis(kLambda0Bar);
	238	mass=v0.GetEffMass();
	239	GetESDsData(kV0s1)->Fill(mass);
d7259b2e	240	}
542c6c71	241
d7259b2e	242	}