[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / charmFlow / AddTaskFlowD2H.C

void AddTaskFlowD2H( int nDmeson, TString fileName, int RPsource,
                     int EtaGapSP=3, double aMax=-0.2, double bMin=+0.2,
                     int minCent=30, int maxCent=50, TString suffixName="C" )
{
  suffixName += Form("%d%d_%s",minCent,maxCent,fileName.Data());
  //-R-P---c-u-t-s--------------------------------------------------------------
  AliFlowTrackCuts* cutsRP;
  double aMin, aMax, bMin, bMax;    // For SP method
  switch (RPsource) {
    case (0):
      cutsRP = (AliFlowTrackCuts*) 
               AliFlowTrackCuts::GetStandardTPCStandaloneTrackCuts2010();
      cutsRP->SetEtaRange( -0.8, +0.8 );
      cutsRP->SetPtRange( 0.2, 5.0 );
      cutsRP->SetAODfilterBit(1);
      cutsRP->SetName( Form("rp_cuts_%s",suffixName.Data()) );
      aMin=-0.8, bMax=0.8;
      break;
    case (1):
      cutsRP = (AliFlowTrackCuts*) 
               AliFlowTrackCuts::GetStandardVZEROOnlyTrackCuts();
      aMin = -3.7; aMax = -2.2 ; bMin = 3.4; bMax = +5.1;
      break;
  }
  fileName.Append(".root");
  //-P-O-I---c-u-t-s------------------------------------------------------------
  double flowBands[5][2];
  switch (nDmeson) { //thanks to ROOT for allowing this
    case (AliRDHFCuts::kD0toKpiCuts):
      AliRDHFCutsD0toKpi *cutsPOI = new AliRDHFCutsD0toKpi( Form("D0toKpi_%s",suffixName.Data()) );
      flowBands[0][0]=1.700; flowBands[0][1]=1.750;
      flowBands[1][0]=1.750; flowBands[1][1]=1.800;
      flowBands[2][0]=1.830; flowBands[2][1]=1.900;
      flowBands[3][0]=1.930; flowBands[3][1]=1.970;
      flowBands[4][0]=1.970; flowBands[4][1]=2.030;
      break;
    case (AliRDHFCuts::kDstarCuts):
      AliRDHFCutsDStartoKpipi *cutsPOI = new AliRDHFCutsDStartoKpipi( Form("DStartoKpipi_%s",suffixName.Data()) );
      flowBands[0][0]=0.140; flowBands[0][1]=0.142;
      flowBands[1][0]=0.144; flowBands[1][1]=0.147;
      flowBands[2][0]=0.150; flowBands[2][1]=0.155;
      flowBands[3][0]=0.155; flowBands[3][1]=0.160;
      flowBands[4][0]=0.160; flowBands[4][1]=0.165;
      break;
  }
  cutsPOI->SetStandardCutsPbPb2010();
  cutsPOI->SetUseCentrality(AliRDHFCuts::kCentV0M);
  cutsPOI->SetMinCentrality(minCent); cutsPOI->SetMaxCentrality(maxCent);
  cutsPOI->SetUseAOD049();

  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();

  //-----------------FLOWD2H TASK----------------------------
  AliAnalysisTaskFlowD2H *taskSel = new AliAnalysisTaskFlowD2H(
                                    Form("FlowD2H_%s",suffixName.Data()),
                                    cutsRP, cutsPOI, nDmeson);
  taskSel->SelectCollisionCandidates(AliVEvent::kMB);
  taskSel->SetDebug();
  taskSel->SetFlowEtaRangeAB( aMin, aMax, bMin, bMax );
  TString Qvector;
  switch (EtaGapSP) {
    case 0:
      taskSel->SetPOIEtaRange( -1.5, aMax );
      Qvector = "Qb";
      break;
    case 1:
      taskSel->SetPOIEtaRange( bMin, +1.5 );
      Qvector = "Qa";
      break;
    case 3:
      taskSel->SetPOIEtaRange( -1.5, 1.5 );
      Qvector = "QaQb"; // dummy
      break;
  }
  taskSel->SetFlowPtRange( 2, 16 );
  for(int i=0; i!=5; ++i)
    taskSel->SetFlowBandRange( i, flowBands[i][0], flowBands[i][1] );
  mgr->AddTask(taskSel);
  AliAnalysisDataContainer *coutputCandidatesQA = mgr->CreateContainer(
     Form("QA_%s",suffixName.Data()),TList::Class(),
     AliAnalysisManager::kOutputContainer, fileName+":QACandidates");
  mgr->ConnectOutput(taskSel,1,coutputCandidatesQA);
  mgr->ConnectInput (taskSel,0,cinput1);
  AliAnalysisDataContainer *coutputCandidates[5];
  for(int r=0; r!=5; ++r) {
    coutputCandidates[r] = mgr->CreateContainer(
       Form("Flow_MassBand%d_%s",r,suffixName.Data()),
       AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
    mgr->ConnectOutput(taskSel,2+r,coutputCandidates[r]);
  }
  // Scalar Product
  AliAnalysisTaskScalarProduct *taskSP[5];
  AliAnalysisDataContainer *coutputSP[5];
  for(int r=0; r!=5; ++r) {
    taskSP[r] = new AliAnalysisTaskScalarProduct(
             Form("SP_MassBand%d_%s",r,suffixName.Data()),kFALSE);
    taskSP[r]->SetRelDiffMsub(1.0);
    taskSP[r]->SetApplyCorrectionForNUA(kTRUE);
    taskSP[r]->SetTotalQvector( Qvector.Data() );
    mgr->AddTask( taskSP[r] );
    TString outputSP = fileName + Form(":outputSP_MassBand%d",r);
    coutputSP[r] = mgr->CreateContainer(
         Form("cobjSP_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
         AliAnalysisManager::kOutputContainer,outputSP.Data());
    mgr->ConnectInput(taskSP[r],0,coutputCandidates[r]);
    mgr->ConnectOutput(taskSP[r],1,coutputSP[r]);
  }
  if(RPsource!=0)
    return;
  if(EtaGapSP!=3)
    return;
  cout << "HERE!\n";
  // Event Plane
  AliAnalysisTaskEventPlane *taskEP[5];
  AliAnalysisDataContainer *coutputEP[5];
  for(int r=0; r!=5; ++r) {
    taskEP[r] = new AliAnalysisTaskEventPlane(
             Form("EP_MassBand%d_%s",r,suffixName.Data()),kFALSE);
    taskEP[r]->SetApplyCorrectionForNUA(kTRUE);
    taskEP[r]->SetNormalizationType( 0 );
    mgr->AddTask( taskEP[r] );
    TString outputEP = fileName + Form(":outputEP_MassBand%d",r);
    coutputEP[r] = mgr->CreateContainer(
         Form("cobjEP_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
         AliAnalysisManager::kOutputContainer,outputEP.Data());
    mgr->ConnectInput(taskEP[r],0,coutputCandidates[r]);
    mgr->ConnectOutput(taskEP[r],1,coutputEP[r]);
  }
  // Event Plane 2
  AliAnalysisTaskEventPlane *taskEP2[5];
  AliAnalysisDataContainer *coutputEP2[5];
  for(int r=0; r!=5; ++r) {
    taskEP2[r] = new AliAnalysisTaskEventPlane(
             Form("EP2_MassBand%d_%s",r,suffixName.Data()),kFALSE);
    taskEP2[r]->SetApplyCorrectionForNUA(kTRUE);
    taskEP2[r]->SetNormalizationType( 0 );
    mgr->AddTask( taskEP2[r] );
    TString outputEP2 = fileName + Form(":outputEP2_MassBand%d",r);
    coutputEP2[r] = mgr->CreateContainer(
         Form("cobjEP2_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
         AliAnalysisManager::kOutputContainer,outputEP2.Data());
    mgr->ConnectInput(taskEP2[r],0,coutputCandidates[r]);
    mgr->ConnectOutput(taskEP2[r],1,coutputEP2[r]);
  }
  // Q-Cumulants
  AliAnalysisTaskQCumulants *taskQC[5];
  AliAnalysisDataContainer *coutputQC[5];
  for(int r=0; r!=5; ++r) {
    taskQC[r] = new AliAnalysisTaskQCumulants(
             Form("QC_MassBand%d_%s",r,suffixName.Data()),kFALSE);
    taskQC[r]->SetCalculateCumulantsVsM(kFALSE);
    taskQC[r]->SetnBinsMult(10000);
    taskQC[r]->SetMinMult(0.);
    taskQC[r]->SetMaxMult(10000.);
    taskQC[r]->SetApplyCorrectionForNUA(kTRUE);
    taskQC[r]->SetFillMultipleControlHistograms(kFALSE);
    mgr->AddTask( taskQC[r] );
    TString outputQC = fileName + Form(":outputQC_MassBand%d",r);
    coutputQC[r] = mgr->CreateContainer(
         Form("cobjQC_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
         AliAnalysisManager::kOutputContainer,outputQC.Data());
    mgr->ConnectInput(taskQC[r],0,coutputCandidates[r]);
    mgr->ConnectOutput(taskQC[r],1,coutputQC[r]);
  }
}
Commit	Line	Data
a8f6c03f	1	void AddTaskFlowD2H( int nDmeson, TString fileName, int RPsource,
	2	int EtaGapSP=3, double aMax=-0.2, double bMin=+0.2,
	3	int minCent=30, int maxCent=50, TString suffixName="C" )
	4	{
	5	suffixName += Form("%d%d_%s",minCent,maxCent,fileName.Data());
	6	//-R-P---c-u-t-s--------------------------------------------------------------
	7	AliFlowTrackCuts* cutsRP;
	8	double aMin, aMax, bMin, bMax; // For SP method
	9	switch (RPsource) {
	10	case (0):
	11	cutsRP = (AliFlowTrackCuts*)
	12	AliFlowTrackCuts::GetStandardTPCStandaloneTrackCuts2010();
	13	cutsRP->SetEtaRange( -0.8, +0.8 );
	14	cutsRP->SetPtRange( 0.2, 5.0 );
	15	cutsRP->SetAODfilterBit(1);
	16	cutsRP->SetName( Form("rp_cuts_%s",suffixName.Data()) );
	17	aMin=-0.8, bMax=0.8;
	18	break;
	19	case (1):
	20	cutsRP = (AliFlowTrackCuts*)
	21	AliFlowTrackCuts::GetStandardVZEROOnlyTrackCuts();
	22	aMin = -3.7; aMax = -2.2 ; bMin = 3.4; bMax = +5.1;
	23	break;
	24	}
	25	fileName.Append(".root");
	26	//-P-O-I---c-u-t-s------------------------------------------------------------
	27	double flowBands[5][2];
	28	switch (nDmeson) { //thanks to ROOT for allowing this
	29	case (AliRDHFCuts::kD0toKpiCuts):
	30	AliRDHFCutsD0toKpi *cutsPOI = new AliRDHFCutsD0toKpi( Form("D0toKpi_%s",suffixName.Data()) );
	31	flowBands[0][0]=1.700; flowBands[0][1]=1.750;
	32	flowBands[1][0]=1.750; flowBands[1][1]=1.800;
	33	flowBands[2][0]=1.830; flowBands[2][1]=1.900;
	34	flowBands[3][0]=1.930; flowBands[3][1]=1.970;
	35	flowBands[4][0]=1.970; flowBands[4][1]=2.030;
	36	break;
	37	case (AliRDHFCuts::kDstarCuts):
	38	AliRDHFCutsDStartoKpipi *cutsPOI = new AliRDHFCutsDStartoKpipi( Form("DStartoKpipi_%s",suffixName.Data()) );
	39	flowBands[0][0]=0.140; flowBands[0][1]=0.142;
	40	flowBands[1][0]=0.144; flowBands[1][1]=0.147;
	41	flowBands[2][0]=0.150; flowBands[2][1]=0.155;
	42	flowBands[3][0]=0.155; flowBands[3][1]=0.160;
	43	flowBands[4][0]=0.160; flowBands[4][1]=0.165;
	44	break;
	45	}
	46	cutsPOI->SetStandardCutsPbPb2010();
	47	cutsPOI->SetUseCentrality(AliRDHFCuts::kCentV0M);
	48	cutsPOI->SetMinCentrality(minCent); cutsPOI->SetMaxCentrality(maxCent);
	49	cutsPOI->SetUseAOD049();
	50
	51	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	52	AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
	53
	54	//-----------------FLOWD2H TASK----------------------------
	55	AliAnalysisTaskFlowD2H *taskSel = new AliAnalysisTaskFlowD2H(
	56	Form("FlowD2H_%s",suffixName.Data()),
	57	cutsRP, cutsPOI, nDmeson);
	58	taskSel->SelectCollisionCandidates(AliVEvent::kMB);
	59	taskSel->SetDebug();
	60	taskSel->SetFlowEtaRangeAB( aMin, aMax, bMin, bMax );
	61	TString Qvector;
	62	switch (EtaGapSP) {
	63	case 0:
	64	taskSel->SetPOIEtaRange( -1.5, aMax );
65	Qvector = "Qb";
66	break;
67	case 1:
68	taskSel->SetPOIEtaRange( bMin, +1.5 );
69	Qvector = "Qa";
70	break;
71	case 3:
72	taskSel->SetPOIEtaRange( -1.5, 1.5 );
73	Qvector = "QaQb"; // dummy
74	break;
75	}
76	taskSel->SetFlowPtRange( 2, 16 );
77	for(int i=0; i!=5; ++i)
78	taskSel->SetFlowBandRange( i, flowBands[i][0], flowBands[i][1] );
79	mgr->AddTask(taskSel);
80	AliAnalysisDataContainer *coutputCandidatesQA = mgr->CreateContainer(
81	Form("QA_%s",suffixName.Data()),TList::Class(),
82	AliAnalysisManager::kOutputContainer, fileName+":QACandidates");
83	mgr->ConnectOutput(taskSel,1,coutputCandidatesQA);
84	mgr->ConnectInput (taskSel,0,cinput1);
85	AliAnalysisDataContainer *coutputCandidates[5];
86	for(int r=0; r!=5; ++r) {
87	coutputCandidates[r] = mgr->CreateContainer(
88	Form("Flow_MassBand%d_%s",r,suffixName.Data()),
89	AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
90	mgr->ConnectOutput(taskSel,2+r,coutputCandidates[r]);
91	}
92	// Scalar Product
93	AliAnalysisTaskScalarProduct *taskSP[5];
94	AliAnalysisDataContainer *coutputSP[5];
95	for(int r=0; r!=5; ++r) {
96	taskSP[r] = new AliAnalysisTaskScalarProduct(
97	Form("SP_MassBand%d_%s",r,suffixName.Data()),kFALSE);
98	taskSP[r]->SetRelDiffMsub(1.0);
99	taskSP[r]->SetApplyCorrectionForNUA(kTRUE);
100	taskSP[r]->SetTotalQvector( Qvector.Data() );
101	mgr->AddTask( taskSP[r] );
102	TString outputSP = fileName + Form(":outputSP_MassBand%d",r);
103	coutputSP[r] = mgr->CreateContainer(
104	Form("cobjSP_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
105	AliAnalysisManager::kOutputContainer,outputSP.Data());
106	mgr->ConnectInput(taskSP[r],0,coutputCandidates[r]);
107	mgr->ConnectOutput(taskSP[r],1,coutputSP[r]);
108	}
109	if(RPsource!=0)
110	return;
111	if(EtaGapSP!=3)
112	return;
113	cout << "HERE!\n";
114	// Event Plane
115	AliAnalysisTaskEventPlane *taskEP[5];
116	AliAnalysisDataContainer *coutputEP[5];
117	for(int r=0; r!=5; ++r) {
118	taskEP[r] = new AliAnalysisTaskEventPlane(
119	Form("EP_MassBand%d_%s",r,suffixName.Data()),kFALSE);
120	taskEP[r]->SetApplyCorrectionForNUA(kTRUE);
121	taskEP[r]->SetNormalizationType( 0 );
122	mgr->AddTask( taskEP[r] );
123	TString outputEP = fileName + Form(":outputEP_MassBand%d",r);
124	coutputEP[r] = mgr->CreateContainer(
125	Form("cobjEP_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
126	AliAnalysisManager::kOutputContainer,outputEP.Data());
127	mgr->ConnectInput(taskEP[r],0,coutputCandidates[r]);
128	mgr->ConnectOutput(taskEP[r],1,coutputEP[r]);
129	}
130	// Event Plane 2
131	AliAnalysisTaskEventPlane *taskEP2[5];
132	AliAnalysisDataContainer *coutputEP2[5];
133	for(int r=0; r!=5; ++r) {
134	taskEP2[r] = new AliAnalysisTaskEventPlane(
135	Form("EP2_MassBand%d_%s",r,suffixName.Data()),kFALSE);
136	taskEP2[r]->SetApplyCorrectionForNUA(kTRUE);
137	taskEP2[r]->SetNormalizationType( 0 );
138	mgr->AddTask( taskEP2[r] );
139	TString outputEP2 = fileName + Form(":outputEP2_MassBand%d",r);
140	coutputEP2[r] = mgr->CreateContainer(
141	Form("cobjEP2_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
142	AliAnalysisManager::kOutputContainer,outputEP2.Data());
143	mgr->ConnectInput(taskEP2[r],0,coutputCandidates[r]);
144	mgr->ConnectOutput(taskEP2[r],1,coutputEP2[r]);
145	}
146	// Q-Cumulants
147	AliAnalysisTaskQCumulants *taskQC[5];
148	AliAnalysisDataContainer *coutputQC[5];
149	for(int r=0; r!=5; ++r) {
150	taskQC[r] = new AliAnalysisTaskQCumulants(
151	Form("QC_MassBand%d_%s",r,suffixName.Data()),kFALSE);
152	taskQC[r]->SetCalculateCumulantsVsM(kFALSE);
153	taskQC[r]->SetnBinsMult(10000);
154	taskQC[r]->SetMinMult(0.);
155	taskQC[r]->SetMaxMult(10000.);
156	taskQC[r]->SetApplyCorrectionForNUA(kTRUE);
157	taskQC[r]->SetFillMultipleControlHistograms(kFALSE);
158	mgr->AddTask( taskQC[r] );
159	TString outputQC = fileName + Form(":outputQC_MassBand%d",r);
160	coutputQC[r] = mgr->CreateContainer(
161	Form("cobjQC_MassBand%d_%s",r,suffixName.Data()),TList::Class(),
162	AliAnalysisManager::kOutputContainer,outputQC.Data());
163	mgr->ConnectInput(taskQC[r],0,coutputCandidates[r]);
164	mgr->ConnectOutput(taskQC[r],1,coutputQC[r]);
165	}
166	}
167