[u/mrichter/AliRoot.git] / PWGCF / Correlations / macros / dptdptcorrelations / AddTaskDptQA.C

//
// Macro designed for use with the AliAnalysisTaskDptDptQA task.
//
// Author: Prabhat Pujahari & Claude Pruneau, Wayne State
// 
//           system:  0: PbPb                 1: pPb
//      singlesOnly:  0: full correlations    1: singles only
//       useWeights:  0: no                   1: yes
// centralityMethod:  3: track count  4: V0 centrality  7: V0A centrality for pPb
//        chargeSet:  0: ++    1: +-    2: -+    3: --
/////////////////////////////////////////////////////////////////////////////////

AliAnalysisTaskDptDptQA *AddTaskDptQA
(int    system                 = 0,
 int    singlesOnly            = 0,
 int    useWeights             = 0,
 int    centralityMethod       = 4,
 int    chargeSet              = 1,
 int    trackFilterBit         = 128,
 int    nClusterMin            = 80,
 double etaMin                 = -0.8,
 double etaMax                 = 0.8,
 double dcaZMin                = -3.2,
 double dcaZMax                =  3.2,
 double dcaXYMin               = -2.4,
 double dcaXYMax               =  2.4,
 int    nclus                  = 80,
 double chi2ndf                = 4,
 int nCentrality               =  2, 
 Bool_t trigger                = kFALSE,
 const char* taskname          = "dca3"
 )
  
{
    
  // Set Default Configuration of this analysis
  // ==========================================
  int    debugLevel             = 0;
  int    rejectPileup           = 1;
  int    rejectPairConversion   = 1;
  int    sameFilter             = 1;

  
  //int    nCentrality;
  double minCentrality[10];
  double maxCentrality[10];

  if (system==0) // PbPb
    {
    if (centralityMethod == 4)
      {
	
	minCentrality[0] = 0.0; maxCentrality[0] = 10.0; 
	minCentrality[1] = 30.0; maxCentrality[1] = 40.;
		
      }
    else
      {
      cout << "-F- AddTaskDptQA() system:" << system << ". centralityMethod:" << centralityMethod << " Option NOT AVAILABLE. ABORT."
      return 0;
      }
    }
  else if (system==1) // pPb
    {
    if (centralityMethod == 7)
      {
	//nCentrality = 4;
      minCentrality[0] = 0;   maxCentrality[0] = 20.0;
      minCentrality[1] = 20;   maxCentrality[1] = 40.;
      minCentrality[2] = 40.; maxCentrality[2] = 60.;
      minCentrality[3] = 0.; maxCentrality[3] = 100.;
      }
    else
      {
	//cout << "-F- AddTaskDptDptCorrelations() system:" << system << ". centralityMethod:" << centralityMethod << " Option NOT AVAILABLE. ABORT."
      return 0;
      }
    }
  else
    {
      //cout << "-F- AddTaskDptDptCorrelations() system:" << system << ". Option NOT CURRENTLY AVAILABLE. ABORT."
    return 0;
    }

  double zMin                   = -10.;
  double zMax                   =  10.;
  double ptMin                  =  0.2;
  double ptMax                  =  2.0;
  double dedxMin                =  0.0;
  double dedxMax                =  20000.0;
  int    requestedCharge1       =  1; //default
  int    requestedCharge2       = -1; //default
  
  
  // Get the pointer to the existing analysis manager via the static access method.
  // ==============================================================================
  AliAnalysisManager *analysisManager = AliAnalysisManager::GetAnalysisManager();
  
  if (!analysisManager) 
    {
    ::Error("AddTaskDptDptCorrelations", "No analysis manager to connect to.");
    return NULL;
    }  
  
  TString part1Name;
  TString part2Name;
  TString eventName;
  TString prefixName        = "Corr_";
  TString pileupRejecSuffix = "_PileupRejec";
  TString pairRejecSuffix   = "_PairRejec";
  TString calibSuffix       = "_calib";
  TString singlesOnlySuffix = "_SO";
  TString suffix;
  
  TString inputPath         = ".";
  TString outputPath        = ".";
  TString baseName;
  TString listName;
  TString taskName;
  TString inputHistogramFileName;
  TString outputHistogramFileName;
  
  // Create the task and add subtask.
  // ===========================================================================
  int iTask = 0; // task counter
  AliAnalysisDataContainer *taskInputContainer;
  AliAnalysisDataContainer *taskOutputContainer;
  AliAnalysisTaskDptDptQA* task;
  
  for (int iCentrality=0; iCentrality < nCentrality; ++iCentrality)
    {
      switch (chargeSet)
        {
          case 0: part1Name = "P_"; part2Name = "P_"; requestedCharge1 =  1; requestedCharge2 =  1; sameFilter = 1; break;
          case 1: part1Name = "P_"; part2Name = "M_"; requestedCharge1 =  1; requestedCharge2 = -1; sameFilter = 0;   break;
          case 2: part1Name = "M_"; part2Name = "P_"; requestedCharge1 = -1; requestedCharge2 =  1; sameFilter = 0;   break;
          case 3: part1Name = "M_"; part2Name = "M_"; requestedCharge1 = -1; requestedCharge2 = -1; sameFilter = 1;   break;
        }
      //part1Name += int(1000*etaMin);                                                                                        
      part1Name += "eta";
      part1Name += int(1000*etaMax);
      part1Name += "_";
      part1Name += int(1000*ptMin);
      part1Name += "pt";
      part1Name += int(1000*ptMax);
      part1Name += "_";
      part1Name += int(1000*dcaZMin);
      part1Name += "DCA";
      part1Name += int(1000*dcaZMax);
      part1Name += "_";


      //part2Name += int(1000*etaMin);                                                                                        
      part2Name += "eta";
      part2Name += int(1000*etaMax);
      part2Name += "_";
      part2Name += int(1000*ptMin);
      part2Name += "pt";
      part2Name += int(1000*ptMax);
      part2Name += "_";
      part2Name += int(1000*dcaZMin);
      part2Name += "DCA";
      part2Name += int(1000*dcaZMax);
      part2Name += "_";

      eventName =  "";
      eventName += int(10.*minCentrality[iCentrality] );
      eventName += "Vo";
      eventName += int(10.*maxCentrality[iCentrality] );
      

      baseName     =   prefixName;
      baseName     +=  part1Name;
      baseName     +=  part2Name;
      baseName     +=  eventName;
      listName     =   baseName;
      taskName     =   baseName;
      
      
      outputHistogramFileName = baseName;
      if (singlesOnly) outputHistogramFileName += singlesOnlySuffix;
      outputHistogramFileName += ".root";
      
    
      TFile  * inputFile  = 0;
      TList  * histoList  = 0;
      TH3F   * weight_1   = 0;
      TH3F   * weight_2   = 0;
      if (useWeights)
        {
        TGrid::Connect("alien:");
        inputFile = TFile::Open(inputHistogramFileName,"OLD");
        if (!inputFile)
          {
          cout << "Requested file:" << inputHistogramFileName << " was not opened. ABORT." << endl;
          return;
          }
        TString nameHistoBase = "correction_";
        TString nameHisto;
        nameHistoBase += eventName;
        if (requestedCharge1 == 1)
          {
          nameHisto = nameHistoBase + "_p";
          cout << "Input Histogram named: " << nameHisto << endl;
          weight_1 = (TH3F *) inputFile->Get(nameHisto);
          }
        else
          {
          nameHisto = nameHistoBase + "_m";
          cout << "Input Histogram named: " << nameHisto << endl;
          weight_1 = (TH3F *) inputFile->Get(nameHisto);
          }
        if (!weight_1) 
          {
          cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
          return 0;
          }
        
        if (!sameFilter)
          {
          weight_2 = 0;
          if (requestedCharge2 == 1)
            {
            nameHisto = nameHistoBase + "_p";
            cout << "Input Histogram named: " << nameHisto << endl;
            weight_2 = (TH3F *) inputFile->Get(nameHisto);
            }
          else
            {
            nameHisto = nameHistoBase + "_m";
            cout << "Input Histogram named: " << nameHisto << endl;
            weight_2 = (TH3F *) inputFile->Get(nameHisto);
            }
          if (!weight_2) 
            {
            cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
            return 0;
            }
          }  
        }
      task = new  AliAnalysisTaskDptDptQA(taskName);
      //configure my task
      task->SetDebugLevel(          debugLevel      ); 
      task->SetSameFilter(          sameFilter      );
      task->SetSinglesOnly(         singlesOnly     ); 
      task->SetUseWeights(          useWeights      ); 
      task->SetRejectPileup(        rejectPileup    ); 
      task->SetRejectPairConversion(rejectPairConversion); 
      task->SetVertexZMin(          zMin            ); 
      task->SetVertexZMax(          zMax            ); 
      task->SetVertexXYMin(         -1.            ); 
      task->SetVertexXYMax(          1.            ); 
      task->SetCentralityMethod(    centralityMethod);
      task->SetCentrality(          minCentrality[iCentrality], maxCentrality[iCentrality]);
      task->SetPtMin1(              ptMin           ); 
      task->SetPtMax1(              ptMax           ); 
      task->SetEtaMin1(             etaMin          ); 
      task->SetEtaMax1(             etaMax          ); 
      task->SetPtMin2(              ptMin           ); 
      task->SetPtMax2(              ptMax           ); 
      task->SetEtaMin2(             etaMin          ); 
      task->SetEtaMax2(             etaMax          ); 
      task->SetDcaZMin( dcaZMin ); 
      task->SetDcaZMax( dcaZMax ); 
      task->SetDcaXYMin(dcaXYMin); 
      task->SetDcaXYMax(dcaXYMax);
      task->SetTPCNclus(nclus);
      task->SetChi2PerNDF(chi2ndf);

      task->SetDedxMin(             dedxMin         ); 
      task->SetDedxMax(             dedxMax         ); 
      task->SetNClusterMin(         nClusterMin     ); 
      task->SetTrackFilterBit(      trackFilterBit  );
      task->SetRequestedCharge_1(   requestedCharge1); 
      task->SetRequestedCharge_2(   requestedCharge2); 
      task->SetWeigth_1(            weight_1        );
      task->SetWeigth_2(            weight_2        );
            
      
      if(trigger) task->SelectCollisionCandidates(AliVEvent::kINT7);
      else task->SelectCollisionCandidates(AliVEvent::kMB);
            
      cout << "Creating task output container" << endl;
      
      taskOutputContainer = analysisManager->CreateContainer(listName, 
                                                             TList::Class(),    
                                                             AliAnalysisManager::kOutputContainer, 
                                                             Form("%s:%s", AliAnalysisManager::GetCommonFileName(),taskname));
      cout << "Add task to analysis manager and connect it to input and output containers" << endl;
           

      analysisManager->AddTask(task);
      analysisManager->ConnectInput( task,  0, analysisManager->GetCommonInputContainer());
      analysisManager->ConnectOutput(task,  0, taskOutputContainer );
      cout << "Task added ...." << endl;
      
      iTask++;
    
    }
  
  return task;
}
Commit	Line	Data
143d78d1	1	//
	2	// Macro designed for use with the AliAnalysisTaskDptDptQA task.
	3	//
	4	// Author: Prabhat Pujahari & Claude Pruneau, Wayne State
	5	//
	6	// system: 0: PbPb 1: pPb
	7	// singlesOnly: 0: full correlations 1: singles only
	8	// useWeights: 0: no 1: yes
	9	// centralityMethod: 3: track count 4: V0 centrality 7: V0A centrality for pPb
	10	// chargeSet: 0: ++ 1: +- 2: -+ 3: --
	11	/////////////////////////////////////////////////////////////////////////////////
2542e3ae	12
	13	AliAnalysisTaskDptDptQA *AddTaskDptQA
	14	(int system = 0,
	15	int singlesOnly = 0,
	16	int useWeights = 0,
	17	int centralityMethod = 4,
	18	int chargeSet = 1,
	19	int trackFilterBit = 128,
143d78d1	20	int nClusterMin = 80,
2542e3ae	21	double etaMin = -0.8,
	22	double etaMax = 0.8,
	23	double dcaZMin = -3.2,
	24	double dcaZMax = 3.2,
	25	double dcaXYMin = -2.4,
	26	double dcaXYMax = 2.4,
143d78d1	27	int nclus = 80,
	28	double chi2ndf = 4,
	29	int nCentrality = 2,
	30	Bool_t trigger = kFALSE,
	31	const char* taskname = "dca3"
2542e3ae	32	)
	33
	34	{
143d78d1	35
2542e3ae	36	// Set Default Configuration of this analysis
	37	// ==========================================
	38	int debugLevel = 0;
	39	int rejectPileup = 1;
	40	int rejectPairConversion = 1;
	41	int sameFilter = 1;
	42
	43
	44	//int nCentrality;
	45	double minCentrality[10];
	46	double maxCentrality[10];
	47
	48	if (system==0) // PbPb
	49	{
	50	if (centralityMethod == 4)
	51	{
2542e3ae	52
143d78d1	53	minCentrality[0] = 0.0; maxCentrality[0] = 10.0;
	54	minCentrality[1] = 30.0; maxCentrality[1] = 40.;
	55
2542e3ae	56	}
	57	else
	58	{
143d78d1	59	cout << "-F- AddTaskDptQA() system:" << system << ". centralityMethod:" << centralityMethod << " Option NOT AVAILABLE. ABORT."
2542e3ae	60	return 0;
	61	}
	62	}
143d78d1	63	else if (system==1) // pPb
2542e3ae	64	{
143d78d1	65	if (centralityMethod == 7)
2542e3ae	66	{
143d78d1	67	//nCentrality = 4;
	68	minCentrality[0] = 0; maxCentrality[0] = 20.0;
	69	minCentrality[1] = 20; maxCentrality[1] = 40.;
	70	minCentrality[2] = 40.; maxCentrality[2] = 60.;
	71	minCentrality[3] = 0.; maxCentrality[3] = 100.;
2542e3ae	72	}
	73	else
	74	{
	75	//cout << "-F- AddTaskDptDptCorrelations() system:" << system << ". centralityMethod:" << centralityMethod << " Option NOT AVAILABLE. ABORT."
	76	return 0;
	77	}
	78	}
	79	else
	80	{
	81	//cout << "-F- AddTaskDptDptCorrelations() system:" << system << ". Option NOT CURRENTLY AVAILABLE. ABORT."
	82	return 0;
	83	}
	84
	85	double zMin = -10.;
	86	double zMax = 10.;
	87	double ptMin = 0.2;
	88	double ptMax = 2.0;
2542e3ae	89	double dedxMin = 0.0;
2542e3ae	90	double dedxMax = 20000.0;
2542e3ae	91	int requestedCharge1 = 1; //default
	92	int requestedCharge2 = -1; //default
	93
	94
	95	// Get the pointer to the existing analysis manager via the static access method.
	96	// ==============================================================================
	97	AliAnalysisManager *analysisManager = AliAnalysisManager::GetAnalysisManager();
	98
	99	if (!analysisManager)
	100	{
	101	::Error("AddTaskDptDptCorrelations", "No analysis manager to connect to.");
	102	return NULL;
	103	}
	104
	105	TString part1Name;
	106	TString part2Name;
	107	TString eventName;
	108	TString prefixName = "Corr_";
	109	TString pileupRejecSuffix = "_PileupRejec";
	110	TString pairRejecSuffix = "_PairRejec";
	111	TString calibSuffix = "_calib";
	112	TString singlesOnlySuffix = "_SO";
	113	TString suffix;
	114
	115	TString inputPath = ".";
	116	TString outputPath = ".";
	117	TString baseName;
	118	TString listName;
	119	TString taskName;
143d78d1	120	TString inputHistogramFileName;
2542e3ae	121	TString outputHistogramFileName;
	122
	123	// Create the task and add subtask.
	124	// ===========================================================================
	125	int iTask = 0; // task counter
	126	AliAnalysisDataContainer *taskInputContainer;
	127	AliAnalysisDataContainer *taskOutputContainer;
	128	AliAnalysisTaskDptDptQA* task;
	129
	130	for (int iCentrality=0; iCentrality < nCentrality; ++iCentrality)
	131	{
	132	switch (chargeSet)
	133	{
	134	case 0: part1Name = "P_"; part2Name = "P_"; requestedCharge1 = 1; requestedCharge2 = 1; sameFilter = 1; break;
	135	case 1: part1Name = "P_"; part2Name = "M_"; requestedCharge1 = 1; requestedCharge2 = -1; sameFilter = 0; break;
	136	case 2: part1Name = "M_"; part2Name = "P_"; requestedCharge1 = -1; requestedCharge2 = 1; sameFilter = 0; break;
	137	case 3: part1Name = "M_"; part2Name = "M_"; requestedCharge1 = -1; requestedCharge2 = -1; sameFilter = 1; break;
	138	}
143d78d1	139	//part1Name += int(1000*etaMin);
2542e3ae	140	part1Name += "eta";
	141	part1Name += int(1000*etaMax);
	142	part1Name += "_";
	143	part1Name += int(1000*ptMin);
	144	part1Name += "pt";
	145	part1Name += int(1000*ptMax);
	146	part1Name += "_";
143d78d1	147	part1Name += int(1000*dcaZMin);
	148	part1Name += "DCA";
	149	part1Name += int(1000*dcaZMax);
	150	part1Name += "_";
	151
	152
	153	//part2Name += int(1000*etaMin);
2542e3ae	154	part2Name += "eta";
	155	part2Name += int(1000*etaMax);
	156	part2Name += "_";
	157	part2Name += int(1000*ptMin);
	158	part2Name += "pt";
	159	part2Name += int(1000*ptMax);
	160	part2Name += "_";
143d78d1	161	part2Name += int(1000*dcaZMin);
	162	part2Name += "DCA";
	163	part2Name += int(1000*dcaZMax);
	164	part2Name += "_";
	165
2542e3ae	166	eventName = "";
	167	eventName += int(10.*minCentrality[iCentrality] );
	168	eventName += "Vo";
	169	eventName += int(10.*maxCentrality[iCentrality] );
143d78d1	170
143d78d1	171
2542e3ae	172	baseName = prefixName;
	173	baseName += part1Name;
	174	baseName += part2Name;
	175	baseName += eventName;
	176	listName = baseName;
	177	taskName = baseName;
2542e3ae	178
143d78d1	179
2542e3ae	180	outputHistogramFileName = baseName;
	181	if (singlesOnly) outputHistogramFileName += singlesOnlySuffix;
	182	outputHistogramFileName += ".root";
	183
143d78d1	184
143d78d1	185	TFile * inputFile = 0;
2542e3ae	186	TList * histoList = 0;
	187	TH3F * weight_1 = 0;
	188	TH3F * weight_2 = 0;
	189	if (useWeights)
	190	{
	191	TGrid::Connect("alien:");
	192	inputFile = TFile::Open(inputHistogramFileName,"OLD");
	193	if (!inputFile)
	194	{
143d78d1	195	cout << "Requested file:" << inputHistogramFileName << " was not opened. ABORT." << endl;
2542e3ae	196	return;
	197	}
	198	TString nameHistoBase = "correction_";
	199	TString nameHisto;
	200	nameHistoBase += eventName;
	201	if (requestedCharge1 == 1)
	202	{
	203	nameHisto = nameHistoBase + "_p";
143d78d1	204	cout << "Input Histogram named: " << nameHisto << endl;
2542e3ae	205	weight_1 = (TH3F *) inputFile->Get(nameHisto);
	206	}
	207	else
	208	{
	209	nameHisto = nameHistoBase + "_m";
143d78d1	210	cout << "Input Histogram named: " << nameHisto << endl;
2542e3ae	211	weight_1 = (TH3F *) inputFile->Get(nameHisto);
	212	}
	213	if (!weight_1)
	214	{
143d78d1	215	cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
2542e3ae	216	return 0;
	217	}
	218
	219	if (!sameFilter)
	220	{
	221	weight_2 = 0;
	222	if (requestedCharge2 == 1)
	223	{
	224	nameHisto = nameHistoBase + "_p";
143d78d1	225	cout << "Input Histogram named: " << nameHisto << endl;
2542e3ae	226	weight_2 = (TH3F *) inputFile->Get(nameHisto);
	227	}
	228	else
	229	{
	230	nameHisto = nameHistoBase + "_m";
143d78d1	231	cout << "Input Histogram named: " << nameHisto << endl;
2542e3ae	232	weight_2 = (TH3F *) inputFile->Get(nameHisto);
	233	}
	234	if (!weight_2)
	235	{
143d78d1	236	cout << "Requested histogram 'correction_p/m' was not found. ABORT." << endl;
2542e3ae	237	return 0;
	238	}
	239	}
	240	}
	241	task = new AliAnalysisTaskDptDptQA(taskName);
	242	//configure my task
	243	task->SetDebugLevel( debugLevel );
	244	task->SetSameFilter( sameFilter );
	245	task->SetSinglesOnly( singlesOnly );
	246	task->SetUseWeights( useWeights );
	247	task->SetRejectPileup( rejectPileup );
	248	task->SetRejectPairConversion(rejectPairConversion);
	249	task->SetVertexZMin( zMin );
	250	task->SetVertexZMax( zMax );
	251	task->SetVertexXYMin( -1. );
	252	task->SetVertexXYMax( 1. );
	253	task->SetCentralityMethod( centralityMethod);
	254	task->SetCentrality( minCentrality[iCentrality], maxCentrality[iCentrality]);
	255	task->SetPtMin1( ptMin );
	256	task->SetPtMax1( ptMax );
	257	task->SetEtaMin1( etaMin );
	258	task->SetEtaMax1( etaMax );
	259	task->SetPtMin2( ptMin );
	260	task->SetPtMax2( ptMax );
	261	task->SetEtaMin2( etaMin );
	262	task->SetEtaMax2( etaMax );
143d78d1	263	task->SetDcaZMin( dcaZMin );
	264	task->SetDcaZMax( dcaZMax );
	265	task->SetDcaXYMin(dcaXYMin);
	266	task->SetDcaXYMax(dcaXYMax);
	267	task->SetTPCNclus(nclus);
	268	task->SetChi2PerNDF(chi2ndf);
	269
2542e3ae	270	task->SetDedxMin( dedxMin );
	271	task->SetDedxMax( dedxMax );
	272	task->SetNClusterMin( nClusterMin );
	273	task->SetTrackFilterBit( trackFilterBit );
	274	task->SetRequestedCharge_1( requestedCharge1);
	275	task->SetRequestedCharge_2( requestedCharge2);
	276	task->SetWeigth_1( weight_1 );
	277	task->SetWeigth_2( weight_2 );
143d78d1	278
2542e3ae	279
143d78d1	280	if(trigger) task->SelectCollisionCandidates(AliVEvent::kINT7);
	281	else task->SelectCollisionCandidates(AliVEvent::kMB);
	282
	283	cout << "Creating task output container" << endl;
2542e3ae	284
2542e3ae	285	taskOutputContainer = analysisManager->CreateContainer(listName,
	286	TList::Class(),
	287	AliAnalysisManager::kOutputContainer,
143d78d1	288	Form("%s:%s", AliAnalysisManager::GetCommonFileName(),taskname));
	289	cout << "Add task to analysis manager and connect it to input and output containers" << endl;
	290
	291
2542e3ae	292	analysisManager->AddTask(task);
	293	analysisManager->ConnectInput( task, 0, analysisManager->GetCommonInputContainer());
	294	analysisManager->ConnectOutput(task, 0, taskOutputContainer );
143d78d1	295	cout << "Task added ...." << endl;
2542e3ae	296
	297	iTask++;
	298
	299	}
	300
2542e3ae	301	return task;
2542e3ae	302	}