[u/mrichter/AliRoot.git] / PWG2 / AliAnalysisTaskProtonsQA.cxx

#include "TChain.h"
#include "TTree.h"
#include "TString.h"
#include "TList.h"
#include "TH2F.h"
#include "TH1I.h"
#include "TF1.h"
#include "TCanvas.h"
#include "TLorentzVector.h"

#include "AliAnalysisTask.h"
#include "AliAnalysisManager.h"

#include "AliESDEvent.h"
#include "AliESDInputHandler.h"
#include "AliMCEventHandler.h"
#include "AliMCEvent.h"
#include "AliStack.h"

#include "PWG2spectra/SPECTRA/AliProtonQAAnalysis.h"
#include "AliAnalysisTaskProtonsQA.h"

// Analysis task used for the QA of the (anti)proton analysis
// Author: Panos Cristakoglou

ClassImp(AliAnalysisTaskProtonsQA)

//________________________________________________________________________ 
AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA()
  : AliAnalysisTask(), fESD(0), fMC(0),
    fList0(0), fList1(0), fList2(0), fList3(0), fList4(0), fList5(0),
    fAnalysis(0) {
    //Dummy constructor
}

//________________________________________________________________________
AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA(const char *name) 
: AliAnalysisTask(name, ""), fESD(0), fMC(0),
  fList0(0), fList1(0), fList2(0), fList3(0), fList4(0), fList5(0),
  fAnalysis(0) {
  // Constructor

  // Define input and output slots here
  // Input slot #0 works with a TChain
  DefineInput(0, TChain::Class());
  // Output slot #0 writes into a TList container
  DefineOutput(0, TList::Class());
  DefineOutput(1, TList::Class());
  DefineOutput(2, TList::Class());
  DefineOutput(3, TList::Class());
  DefineOutput(4, TList::Class());
  DefineOutput(5, TList::Class());
}

//________________________________________________________________________
void AliAnalysisTaskProtonsQA::ConnectInputData(Option_t *) {
  // Connect ESD or AOD here
  // Called once

  TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
  if (!tree) {
    Printf("ERROR: Could not read chain from input slot 0");
  } else {
    AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
     
    if (!esdH) {
      Printf("ERROR: Could not get ESDInputHandler");
    } else
      fESD = esdH->GetEvent();
  }

  AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
  if (!mcH) {
    Printf("ERROR: Could not retrieve MC event handler");
  }
  else
    fMC = mcH->MCEvent();
}

//________________________________________________________________________
void AliAnalysisTaskProtonsQA::CreateOutputObjects() {
  // Create histograms
  // Called once
  //Prior probabilities
  Double_t partFrac[5] = {0.01, 0.01, 0.85, 0.10, 0.05};
  
  //proton analysis object
  fAnalysis = new AliProtonQAAnalysis();
  fAnalysis->SetRunMCAnalysis();
  fAnalysis->SetRunEfficiencyAnalysis();
  //fAnalysis->SetMCProcessId(4);//4: weak decay - 13: hadronic interaction
  //fAnalysis->SetMotherParticlePDGCode(3122);//3122: Lambda

  //Use of TPConly tracks
  fAnalysis->SetQAYPtBins(10, -0.5, 0.5, 12, 0.5, 0.9); //TPC only
  fAnalysis->UseTPCOnly();
  fAnalysis->SetMinTPCClusters(50);
  fAnalysis->SetMaxChi2PerTPCCluster(2.5);
  fAnalysis->SetMaxCov11(2.0);
  fAnalysis->SetMaxCov22(2.0);
  fAnalysis->SetMaxCov33(0.5);
  fAnalysis->SetMaxCov44(0.5);
  fAnalysis->SetMaxCov55(2.0);
  fAnalysis->SetMaxSigmaToVertexTPC(2.0);
  //fAnalysis->SetMaxDCAXYTPC(2.0);
  //fAnalysis->SetMaxDCAZTPC(2.0);
  fAnalysis->SetTPCpid();

  //Use of HybridTPC tracks
  /*fAnalysis->SetQAYPtBins(10, -0.5, 0.5, 12, 0.5, 0.9); //HybridTPC
  fAnalysis->UseHybridTPC();
  fAnalysis->SetMinTPCClusters(50);
  fAnalysis->SetMaxChi2PerTPCCluster(2.5);
  fAnalysis->SetMaxCov11(2.0);
  fAnalysis->SetMaxCov22(2.0);
  fAnalysis->SetMaxCov33(0.5);
  fAnalysis->SetMaxCov44(0.5);
  fAnalysis->SetMaxCov55(2.0);
  fAnalysis->SetMaxSigmaToVertexTPC(2.0);
  //fAnalysis->SetMaxDCAXY(2.0);
  //fAnalysis->SetMaxDCAZ(2.0);
  fAnalysis->SetTPCpid();
  fAnalysis->SetPointOnITSLayer1();
  fAnalysis->SetPointOnITSLayer2();
  fAnalysis->SetPointOnITSLayer3();
  fAnalysis->SetPointOnITSLayer4();
  fAnalysis->SetPointOnITSLayer5();
  fAnalysis->SetPointOnITSLayer6();
  fAnalysis->SetMinITSClusters(5);*/

  //Combined tracking
  /*fAnalysis->SetQAYPtBins(20, -1.0, 1.0, 27, 0.4, 3.1); //combined tracking
  fAnalysis->SetMinTPCClusters(50);
  fAnalysis->SetMaxChi2PerTPCCluster(3.5);
  fAnalysis->SetMaxCov11(2.0);
  fAnalysis->SetMaxCov22(2.0);
  fAnalysis->SetMaxCov33(0.5);
  fAnalysis->SetMaxCov44(0.5);
  fAnalysis->SetMaxCov55(2.0);
  fAnalysis->SetMaxSigmaToVertex(2.0);
  //fAnalysis->SetMaxDCAXY(2.0);
  //fAnalysis->SetMaxDCAZ(2.0);
  fAnalysis->SetTPCRefit();
  fAnalysis->SetPointOnITSLayer1();
  fAnalysis->SetPointOnITSLayer2();
  fAnalysis->SetPointOnITSLayer3();
  fAnalysis->SetPointOnITSLayer4();
  fAnalysis->SetPointOnITSLayer5();
  fAnalysis->SetPointOnITSLayer6();
  fAnalysis->SetMinITSClusters(1);
  fAnalysis->SetITSRefit();
  fAnalysis->SetESDpid();*/

  fAnalysis->SetPriorProbabilities(partFrac);

  fList0 = new TList();
  fList0 = fAnalysis->GetGlobalQAList();

  fList1 = new TList();
  fList1 = fAnalysis->GetPDGList();

  fList2 = new TList();
  fList2 = fAnalysis->GetMCProcessesList();

  fList3 = new TList();
  fList3 = fAnalysis->GetAcceptedCutList();

  fList4 = new TList();
  fList4 = fAnalysis->GetAcceptedDCAList();

  fList5 = new TList();
  fList5 = fAnalysis->GetEfficiencyQAList();
}

//________________________________________________________________________
void AliAnalysisTaskProtonsQA::Exec(Option_t *) {
  // Main loop
  // Called for each event
  
  if (!fESD) {
    Printf("ERROR: fESD not available");
    return;
  }

  if (!fMC) {
    Printf("ERROR: Could not retrieve MC event");
    return;
  }
  
  AliStack* stack = fMC->Stack();
  if (!stack) {
    Printf("ERROR: Could not retrieve the stack");
    return;
  }
  
  fAnalysis->RunQAAnalysis(stack, fESD);
  fAnalysis->RunMCAnalysis(stack);
  fAnalysis->RunEfficiencyAnalysis(stack, fESD);

  // Post output data.
  PostData(0, fList0);
  PostData(1, fList1);
  PostData(2, fList2);
  PostData(3, fList3);
  PostData(4, fList4);
  PostData(5, fList5);
}      

//________________________________________________________________________
void AliAnalysisTaskProtonsQA::Terminate(Option_t *) {
  // Draw result to the screen
  // Called once at the end of the query
  
  fList0 = dynamic_cast<TList*> (GetOutputData(0));
  if (!fList0) {
    Printf("ERROR: fList not available");
    return;
  }
}
Commit	Line	Data
7b59a00b	1	#include "TChain.h"
	2	#include "TTree.h"
	3	#include "TString.h"
	4	#include "TList.h"
	5	#include "TH2F.h"
	6	#include "TH1I.h"
	7	#include "TF1.h"
	8	#include "TCanvas.h"
	9	#include "TLorentzVector.h"
	10
	11	#include "AliAnalysisTask.h"
	12	#include "AliAnalysisManager.h"
	13
	14	#include "AliESDEvent.h"
	15	#include "AliESDInputHandler.h"
	16	#include "AliMCEventHandler.h"
	17	#include "AliMCEvent.h"
	18	#include "AliStack.h"
	19
3e6c06f4	20	#include "PWG2spectra/SPECTRA/AliProtonQAAnalysis.h"
7b59a00b	21	#include "AliAnalysisTaskProtonsQA.h"
	22
	23	// Analysis task used for the QA of the (anti)proton analysis
	24	// Author: Panos Cristakoglou
	25
	26	ClassImp(AliAnalysisTaskProtonsQA)
	27
	28	//________________________________________________________________________
	29	AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA()
	30	: AliAnalysisTask(), fESD(0), fMC(0),
9c0b9f24	31	fList0(0), fList1(0), fList2(0), fList3(0), fList4(0), fList5(0),
d4733690	32	fAnalysis(0) {
7b59a00b	33	//Dummy constructor
7b59a00b	34	}
	35
	36	//________________________________________________________________________
	37	AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA(const char *name)
f99662ca	38	: AliAnalysisTask(name, ""), fESD(0), fMC(0),
9c0b9f24	39	fList0(0), fList1(0), fList2(0), fList3(0), fList4(0), fList5(0),
d4733690	40	fAnalysis(0) {
7b59a00b	41	// Constructor
	42
	43	// Define input and output slots here
	44	// Input slot #0 works with a TChain
	45	DefineInput(0, TChain::Class());
	46	// Output slot #0 writes into a TList container
	47	DefineOutput(0, TList::Class());
d4733690	48	DefineOutput(1, TList::Class());
d4733690	49	DefineOutput(2, TList::Class());
5b8133c7	50	DefineOutput(3, TList::Class());
5b8133c7	51	DefineOutput(4, TList::Class());
9c0b9f24	52	DefineOutput(5, TList::Class());
7b59a00b	53	}
	54
	55	//________________________________________________________________________
	56	void AliAnalysisTaskProtonsQA::ConnectInputData(Option_t *) {
	57	// Connect ESD or AOD here
	58	// Called once
	59
	60	TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
	61	if (!tree) {
	62	Printf("ERROR: Could not read chain from input slot 0");
	63	} else {
7b59a00b	64	AliESDInputHandler esdH = dynamic_cast<AliESDInputHandler> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
	65
	66	if (!esdH) {
	67	Printf("ERROR: Could not get ESDInputHandler");
	68	} else
	69	fESD = esdH->GetEvent();
	70	}
	71
	72	AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
	73	if (!mcH) {
	74	Printf("ERROR: Could not retrieve MC event handler");
	75	}
	76	else
	77	fMC = mcH->MCEvent();
	78	}
	79
	80	//________________________________________________________________________
	81	void AliAnalysisTaskProtonsQA::CreateOutputObjects() {
	82	// Create histograms
	83	// Called once
	84	//Prior probabilities
	85	Double_t partFrac[5] = {0.01, 0.01, 0.85, 0.10, 0.05};
	86
	87	//proton analysis object
3e6c06f4	88	fAnalysis = new AliProtonQAAnalysis();
d4733690	89	fAnalysis->SetRunMCAnalysis();
9c0b9f24	90	fAnalysis->SetRunEfficiencyAnalysis();
5b8133c7	91	//fAnalysis->SetMCProcessId(4);//4: weak decay - 13: hadronic interaction
5b8133c7	92	//fAnalysis->SetMotherParticlePDGCode(3122);//3122: Lambda
7b59a00b	93
3a954fe7	94	//Use of TPConly tracks
5b8133c7	95	fAnalysis->SetQAYPtBins(10, -0.5, 0.5, 12, 0.5, 0.9); //TPC only
7b59a00b	96	fAnalysis->UseTPCOnly();
0326e385	97	fAnalysis->SetMinTPCClusters(50);
	98	fAnalysis->SetMaxChi2PerTPCCluster(2.5);
	99	fAnalysis->SetMaxCov11(2.0);
	100	fAnalysis->SetMaxCov22(2.0);
	101	fAnalysis->SetMaxCov33(0.5);
	102	fAnalysis->SetMaxCov44(0.5);
	103	fAnalysis->SetMaxCov55(2.0);
	104	fAnalysis->SetMaxSigmaToVertexTPC(2.0);
	105	//fAnalysis->SetMaxDCAXYTPC(2.0);
	106	//fAnalysis->SetMaxDCAZTPC(2.0);
5b8133c7	107	fAnalysis->SetTPCpid();
7b59a00b	108
0326e385	109	//Use of HybridTPC tracks
	110	/*fAnalysis->SetQAYPtBins(10, -0.5, 0.5, 12, 0.5, 0.9); //HybridTPC
	111	fAnalysis->UseHybridTPC();
	112	fAnalysis->SetMinTPCClusters(50);
	113	fAnalysis->SetMaxChi2PerTPCCluster(2.5);
	114	fAnalysis->SetMaxCov11(2.0);
	115	fAnalysis->SetMaxCov22(2.0);
	116	fAnalysis->SetMaxCov33(0.5);
	117	fAnalysis->SetMaxCov44(0.5);
	118	fAnalysis->SetMaxCov55(2.0);
	119	fAnalysis->SetMaxSigmaToVertexTPC(2.0);
	120	//fAnalysis->SetMaxDCAXY(2.0);
	121	//fAnalysis->SetMaxDCAZ(2.0);
	122	fAnalysis->SetTPCpid();
	123	fAnalysis->SetPointOnITSLayer1();
	124	fAnalysis->SetPointOnITSLayer2();
	125	fAnalysis->SetPointOnITSLayer3();
	126	fAnalysis->SetPointOnITSLayer4();
	127	fAnalysis->SetPointOnITSLayer5();
	128	fAnalysis->SetPointOnITSLayer6();
	129	fAnalysis->SetMinITSClusters(5);*/
	130
7b59a00b	131	//Combined tracking
0326e385	132	/*fAnalysis->SetQAYPtBins(20, -1.0, 1.0, 27, 0.4, 3.1); //combined tracking
	133	fAnalysis->SetMinTPCClusters(50);
	134	fAnalysis->SetMaxChi2PerTPCCluster(3.5);
	135	fAnalysis->SetMaxCov11(2.0);
	136	fAnalysis->SetMaxCov22(2.0);
	137	fAnalysis->SetMaxCov33(0.5);
	138	fAnalysis->SetMaxCov44(0.5);
	139	fAnalysis->SetMaxCov55(2.0);
	140	fAnalysis->SetMaxSigmaToVertex(2.0);
	141	//fAnalysis->SetMaxDCAXY(2.0);
	142	//fAnalysis->SetMaxDCAZ(2.0);
	143	fAnalysis->SetTPCRefit();
	144	fAnalysis->SetPointOnITSLayer1();
	145	fAnalysis->SetPointOnITSLayer2();
	146	fAnalysis->SetPointOnITSLayer3();
	147	fAnalysis->SetPointOnITSLayer4();
	148	fAnalysis->SetPointOnITSLayer5();
	149	fAnalysis->SetPointOnITSLayer6();
	150	fAnalysis->SetMinITSClusters(1);
	151	fAnalysis->SetITSRefit();
	152	fAnalysis->SetESDpid();*/
7b59a00b	153
	154	fAnalysis->SetPriorProbabilities(partFrac);
	155
d4733690	156	fList0 = new TList();
	157	fList0 = fAnalysis->GetGlobalQAList();
	158
	159	fList1 = new TList();
	160	fList1 = fAnalysis->GetPDGList();
	161
	162	fList2 = new TList();
	163	fList2 = fAnalysis->GetMCProcessesList();
5b8133c7	164
	165	fList3 = new TList();
	166	fList3 = fAnalysis->GetAcceptedCutList();
	167
	168	fList4 = new TList();
	169	fList4 = fAnalysis->GetAcceptedDCAList();
9c0b9f24	170
	171	fList5 = new TList();
	172	fList5 = fAnalysis->GetEfficiencyQAList();
7b59a00b	173	}
	174
	175	//________________________________________________________________________
	176	void AliAnalysisTaskProtonsQA::Exec(Option_t *) {
	177	// Main loop
	178	// Called for each event
	179
	180	if (!fESD) {
	181	Printf("ERROR: fESD not available");
	182	return;
	183	}
	184
	185	if (!fMC) {
	186	Printf("ERROR: Could not retrieve MC event");
	187	return;
	188	}
	189
	190	AliStack* stack = fMC->Stack();
	191	if (!stack) {
	192	Printf("ERROR: Could not retrieve the stack");
	193	return;
	194	}
	195
9c0b9f24	196	fAnalysis->RunQAAnalysis(stack, fESD);
d4733690	197	fAnalysis->RunMCAnalysis(stack);
9c0b9f24	198	fAnalysis->RunEfficiencyAnalysis(stack, fESD);
d4733690	199
7b59a00b	200	// Post output data.
d4733690	201	PostData(0, fList0);
	202	PostData(1, fList1);
	203	PostData(2, fList2);
5b8133c7	204	PostData(3, fList3);
5b8133c7	205	PostData(4, fList4);
9c0b9f24	206	PostData(5, fList5);
7b59a00b	207	}
	208
	209	//________________________________________________________________________
	210	void AliAnalysisTaskProtonsQA::Terminate(Option_t *) {
	211	// Draw result to the screen
	212	// Called once at the end of the query
	213
d4733690	214	fList0 = dynamic_cast<TList*> (GetOutputData(0));
d4733690	215	if (!fList0) {
7b59a00b	216	Printf("ERROR: fList not available");
	217	return;
	218	}
	219	}
	220
	221