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

class AliAnalysisDataContainer;
class AliFlowEventCuts;
class AliFlowEventTrackCuts;
class AliFlowEventTrackSimpleCuts;
class AliRDHFCutsD0toKpi;
class AliRDHFCutsDStartoKpipi; 

void AddTaskFlowD2H(TString fileNameCuts, TString folderName, Int_t nDmeson, Int_t myHarmonic, 
		    Bool_t bDoQC, Bool_t bDoSPTPC, Bool_t bDoSPVZERO, Bool_t bDoEPTPC, Bool_t bDoEPVZERO, 
		    Int_t ptBinWidth, Double_t gapTPC, Double_t etaVZERO1, Double_t etaVZERO2, Double_t etaVZERO3, Double_t etaVZERO4,
		    Bool_t bOldApproach=kFALSE, Bool_t shrinkSP=kFALSE, Bool_t swapAssumption=kFALSE, Int_t nTPCClusters=70 ) {
  TFile *filecuts = TFile::Open( fileNameCuts.Data() );
  if( (!filecuts) || ( filecuts && !filecuts->IsOpen()) ){
    AliFatal("Could not open cuts file.");
  }

  TString fileName = AliAnalysisManager::GetCommonFileName();
  fileName.ReplaceAll(".root","");

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

  TString thecuts = folderName;
  TString sgn = DMesonName( nDmeson ) + Form("w%d",ptBinWidth);

  //**********************************************************************
  // FLOW TRACK CUTS
  AliFlowTrackCuts* cutsRFPTPC = new AliFlowTrackCuts( "GlobalRFPTPC" );
  cutsRFPTPC->SetParamType(AliFlowTrackCuts::kGlobal);
  cutsRFPTPC->SetPtRange(0.2,5.);
  cutsRFPTPC->SetEtaRange(-0.8,0.8);
  cutsRFPTPC->SetMinNClustersTPC(nTPCClusters);
  cutsRFPTPC->SetMinChi2PerClusterTPC(0.2);
  cutsRFPTPC->SetMaxChi2PerClusterTPC(4.0);
  cutsRFPTPC->SetAcceptKinkDaughters(kFALSE);
  cutsRFPTPC->SetMinimalTPCdedx(10.);
  cutsRFPTPC->SetAODfilterBit(1);
  //  cutsRFPTPC->SetQA(kTRUE);
  AliFlowTrackCuts* cutsRFPVZE = new AliFlowTrackCuts( "GlobalRFPVZE" );
  cutsRFPVZE->SetParamType(AliFlowTrackCuts::kVZERO);
  cutsRFPVZE->SetEtaRange(-10,+10);
  cutsRFPVZE->SetPhiMin(0);
  cutsRFPVZE->SetPhiMax(TMath::TwoPi());
  //  cutsRFPVZE->SetQA(kTRUE);
  AliFlowTrackCuts* cutsPOI = new AliFlowTrackCuts( "GlobalPOI" );
  cutsPOI->SetParamType(AliFlowTrackCuts::kGlobal);
  cutsPOI->SetPtRange(0.2,-5.);
  cutsPOI->SetEtaRange(0.8,-0.8);

  //**********************************************************************
  // POI FILTER CUTS
  AliFlowTrackSimpleCuts *filterPOIQC[50]; // MASS BANDS
  AliFlowTrackSimpleCuts *filterPOISP[50][2]; // MASS BANDS || ETA || SUBEVENT GAP
  int myMassBands = MassBands( nDmeson, bOldApproach );
  for(int mb=0; mb!=myMassBands; ++mb) {
    filterPOISP[mb][0] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETANEG",mb) );
    filterPOISP[mb][0]->SetEtaMin( -0.8 );
    filterPOISP[mb][0]->SetEtaMax( 0. );
    filterPOISP[mb][0]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
    filterPOISP[mb][0]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
    filterPOISP[mb][1] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETAPOS",mb) );
    filterPOISP[mb][1]->SetEtaMin( 0. );
    filterPOISP[mb][1]->SetEtaMax( +0.8 );
    filterPOISP[mb][1]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
    filterPOISP[mb][1]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
    filterPOIQC[mb] = new AliFlowTrackSimpleCuts( Form("FilterPOIQC_MB%d",mb) );
    filterPOIQC[mb]->SetEtaMin( -0.8 );
    filterPOIQC[mb]->SetEtaMax( +0.8 );
    filterPOIQC[mb]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
    filterPOIQC[mb]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
  }

  // * DMESON SELECTOR *************************************************
  AliAnalysisTaskFlowD2H *taskSel;
  switch (nDmeson) {
  case ( AliRDHFCuts::kD0toKpiCuts ):
    AliRDHFCutsD0toKpi *myCutsD0 = (AliRDHFCutsD0toKpi*)filecuts->Get("D0toKpiCuts");
    if(!myCutsD0) {
      AliFatal("Problems reaching D0toKpiCuts");
    }
    taskSel = new AliAnalysisTaskFlowD2H( Form("TaskD0Selector_%s",thecuts.Data()), 
					  cutsRFPTPC, cutsRFPVZE, myCutsD0, nDmeson );
    break;
  case ( AliRDHFCuts::kDstarCuts ):
    AliRDHFCutsDStartoKpipi *myCutsDStar = (AliRDHFCutsDStartoKpipi*)filecuts->Get("DStartoKpipiCuts");
    if(!myCutsDStar) {
      AliFatal("Problems reaching DStarToKpipiCuts");
    }
    taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDStarSelector_%s",thecuts.Data()),
					  cutsRFPTPC, cutsRFPVZE, myCutsDStar, nDmeson);
    break;
  case (AliRDHFCuts::kDplusCuts):
    AliRDHFCutsDplustoKpipi *myCutsDplus = (AliRDHFCutsDplustoKpipi*)filecuts->Get("AnalysisCuts");
    if(!myCutsDplus) {
      AliFatal("Problems reaching AnalysisCuts");
    }
    taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDplusSelector_%s",thecuts.Data()), 
					  cutsRFPTPC, cutsRFPVZE, myCutsDplus, nDmeson );
    break;
  }
  taskSel->SetCommonConstants( MassBins(nDmeson), MinMass(nDmeson), MaxMass(nDmeson), ptBinWidth );
  if (swapAssumption)
    taskSel->SetSwapAsumption();

  //taskSel->SelectCollisionCandidates(trigger);
  //taskSel->SetDebug();
  AliAnalysisDataContainer *coutHist = mgr->CreateContainer( Form("%sSelector_%s",sgn.Data(),thecuts.Data()),
							     TList::Class(),AliAnalysisManager::kOutputContainer,
							     Form("%s.root:FlowD2H_%s",fileName.Data(),thecuts.Data()) );
  AliAnalysisDataContainer *exc_TPC = mgr->CreateContainer( Form("TPCEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
							    AliFlowEventSimple::Class(),
							    AliAnalysisManager::kExchangeContainer );
  AliAnalysisDataContainer *exc_VZE = mgr->CreateContainer( Form("VZEEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
							    AliFlowEventSimple::Class(),
							    AliAnalysisManager::kExchangeContainer );
  mgr->AddTask(taskSel);
  mgr->ConnectOutput(taskSel,1,coutHist);
  mgr->ConnectOutput(taskSel,2,exc_TPC);
  mgr->ConnectOutput(taskSel,3,exc_VZE);
  mgr->ConnectInput (taskSel,0,cinput1);

  // * HANGING ANALYSIS TASKS ********************************************
  int harm = myHarmonic;
  for(int mb=0; mb!=myMassBands; ++mb) {
    if(bDoQC) {
      AddQCmethod( Form("%sQCTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), harm, exc_TPC, filterPOIQC[mb]);
    }
    if(bDoSPVZERO) {
      AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
		   "Qa", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
      AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
		   "Qb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
    }
    if(bDoEPVZERO) {
      AddSPmethod( Form("%sEPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
		   "QaQb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, true, shrinkSP );
    }
    if(bDoSPTPC) {
      for(int eg=0; eg!=2; ++eg) {
	AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPC*eg, +gapTPC*eg, +0.8,
		     "Qa", harm, exc_TPC, eg, filterPOISP[mb][1], NULL, false, shrinkSP );
	AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPC*eg, +gapTPC*eg, +0.8,
		     "Qb", harm, exc_TPC, eg, filterPOISP[mb][0], NULL, false, shrinkSP );
      }
    }
    if(bDoEPTPC) {
      AddSPmethod( Form("%sEPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -0.0, +0.0, +0.8,
		   "QaQb", harm, exc_TPC, 0, filterPOIQC[mb], NULL, true, shrinkSP );
    }
  }
}

void AddSPmethod(char *name, char *fileName, char *thecuts,
		 double minEtaA, double maxEtaA, double minEtaB, double maxEtaB,
		 char *Qvector, int harmonic, AliAnalysisDataContainer *flowEvent, int eg,
		 AliFlowTrackSimpleCuts *cutsPOI=NULL, AliFlowTrackSimpleCuts *cutsRFP=NULL,
		 bool bEP, bool shrink=false ) {
  TString myFolder = Form("v%d_%s",harmonic,thecuts);
  TString myNameSP = Form("%sSPv%d%sGAP%d_%s",name,harmonic,Qvector,eg,thecuts);
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myNameSP.Data()),
							       AliFlowEventSimple::Class(),
							       AliAnalysisManager::kExchangeContainer );
  AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myNameSP.Data()),
								    cutsRFP, cutsPOI);
  tskFilter->SetSubeventEtaRange( minEtaA, maxEtaA, minEtaB, maxEtaB );
  mgr->AddTask(tskFilter);
  mgr->ConnectInput( tskFilter,0,flowEvent);
  mgr->ConnectOutput(tskFilter,1,flowEvent2);
  AliAnalysisDataContainer *outSP = mgr->CreateContainer( myNameSP.Data(),
							  TList::Class(),AliAnalysisManager::kOutputContainer,
							  Form("%s.root:FlowD2H_SP_%s",fileName,myFolder.Data()) );
  AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct( Form("TaskScalarProduct_%s",
									       myNameSP.Data()),kFALSE);
  tskSP->SetApplyCorrectionForNUA(kTRUE);
  tskSP->SetHarmonic(harmonic);
  tskSP->SetTotalQvector(Qvector);
  if(bEP) tskSP->SetBehaveAsEP();
  if(shrink) tskSP->SetBookOnlyBasicCCH(kTRUE);
  mgr->AddTask(tskSP);
  mgr->ConnectInput( tskSP,0,flowEvent2);
  mgr->ConnectOutput(tskSP,1,outSP);
}

void AddQCmethod(char *name, char *fileName, char *thecuts,
		 int harmonic, AliAnalysisDataContainer *flowEvent,
		 AliFlowTrackSimpleCuts *cutsPOI=NULL, AliFlowTrackSimpleCuts *cutsRFP=NULL) {
  TString myFolder = Form("v%d_%s",harmonic,thecuts);
  TString myName = Form("%sv%d_%s",name,harmonic,thecuts);
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myName.Data()),
							       AliFlowEventSimple::Class(),
							       AliAnalysisManager::kExchangeContainer );
  AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myName.Data()),
								    cutsRFP, cutsPOI);
  mgr->AddTask(tskFilter);
  mgr->ConnectInput( tskFilter,0,flowEvent);
  mgr->ConnectOutput(tskFilter,1,flowEvent2);
  AliAnalysisDataContainer *outQC = mgr->CreateContainer( myName.Data(),
							  TList::Class(),AliAnalysisManager::kOutputContainer,
							  Form("%s.root:FlowD2H_QC_%s",fileName,myFolder.Data()) );
  AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants( Form("TaskQCumulants_%s",
									 myName.Data()),kFALSE);
  tskQC->SetApplyCorrectionForNUA(kTRUE);
  tskQC->SetHarmonic(harmonic);
  tskQC->SetBookOnlyBasicCCH(kTRUE);
  mgr->AddTask(tskQC);
  mgr->ConnectInput( tskQC,0,flowEvent2);
  mgr->ConnectOutput(tskQC,1,outQC);
}

int MassBands( int nDmeson, bool bOldApproach=false ) {
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
  case ( AliRDHFCuts::kD0toKpiCuts ):
    if(bOldApproach) return 5;
    else return 26;
  case ( AliRDHFCuts::kDstarCuts ):
    return 13; 
  }
}

double MassBandLowEdge( int nDmeson, int mb, bool bOldApproach=false ) {
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
  case ( AliRDHFCuts::kD0toKpiCuts ): // 2 + 20 + 4
    double lowEdgeMinimal[5+1] = {1.75,1.80,1.83,1.90,1.93,2.03};
    double lowEdge[26+1] = { 1.66, 1.71, 1.76, 1.77, 1.78, 1.79, 1.80, 1.81, 1.82, 1.83,
			     1.84, 1.85, 1.86, 1.87, 1.88, 1.89, 1.90, 1.91, 1.92, 1.93,
			     1.94, 1.95, 1.96, 2.01, 2.06, 2.11, 2.16 };
    if(bOldApproach) return lowEdgeMinimal[mb];
    else return lowEdge[mb];
  case ( AliRDHFCuts::kDstarCuts ): // 2 + 10 + 3
    double lowEdge[13+1]={0.138, 0.139, 0.141, 0.143, 0.144, 0.145, 0.146, 0.147, 0.148, 0.150, 0.152,  0.154, 0.156, 0.158}; 
    return lowEdge[mb];
  }
}

double MinMass( int nDmeson ) {
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
  case ( AliRDHFCuts::kD0toKpiCuts ):
    return 1.66;
  case ( AliRDHFCuts::kDstarCuts ):
    return 0.138;
  }
}

double MaxMass( int nDmeson ) {
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
  case ( AliRDHFCuts::kD0toKpiCuts ):
    return 2.16;
  case ( AliRDHFCuts::kDstarCuts ):
    return 0.158;
  }
}

int MassBins( int nDmeson ) {
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
  case ( AliRDHFCuts::kD0toKpiCuts ):
    return 50;
  case ( AliRDHFCuts::kDstarCuts ):
    return 40; 
  }
}

TString DMesonName( int nDmeson ) {
  TString toReturn;
  switch (nDmeson) {
  case ( AliRDHFCuts::kDplusCuts ):
    toReturn = "DPlus"; break;
  case ( AliRDHFCuts::kD0toKpiCuts ):
    toReturn = "D0"; break;
  case ( AliRDHFCuts::kDstarCuts ):
    toReturn = "DStar"; break;
  }
  return toReturn;
}
Commit	Line	Data
fcfcc352	1	class AliAnalysisDataContainer;
	2	class AliFlowEventCuts;
	3	class AliFlowEventTrackCuts;
fc4003c2	4	class AliFlowEventTrackSimpleCuts;
fcfcc352	5	class AliRDHFCutsD0toKpi;
	6	class AliRDHFCutsDStartoKpipi;
	7
fc4003c2	8	void AddTaskFlowD2H(TString fileNameCuts, TString folderName, Int_t nDmeson, Int_t myHarmonic,
	9	Bool_t bDoQC, Bool_t bDoSPTPC, Bool_t bDoSPVZERO, Bool_t bDoEPTPC, Bool_t bDoEPVZERO,
	10	Int_t ptBinWidth, Double_t gapTPC, Double_t etaVZERO1, Double_t etaVZERO2, Double_t etaVZERO3, Double_t etaVZERO4,
b57db0d7	11	Bool_t bOldApproach=kFALSE, Bool_t shrinkSP=kFALSE, Bool_t swapAssumption=kFALSE, Int_t nTPCClusters=70 ) {
fcfcc352	12	TFile *filecuts = TFile::Open( fileNameCuts.Data() );
	13	if( (!filecuts) \|\| ( filecuts && !filecuts->IsOpen()) ){
	14	AliFatal("Could not open cuts file.");
a8f6c03f	15	}
fc4003c2	16
fcfcc352	17	TString fileName = AliAnalysisManager::GetCommonFileName();
fcfcc352	18	fileName.ReplaceAll(".root","");
a8f6c03f	19
	20	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	21	AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
	22
bde814f5	23	TString thecuts = folderName;
fcfcc352	24	TString sgn = DMesonName( nDmeson ) + Form("w%d",ptBinWidth);
	25
	26	//**********************************************************************
	27	// FLOW TRACK CUTS
	28	AliFlowTrackCuts* cutsRFPTPC = new AliFlowTrackCuts( "GlobalRFPTPC" );
	29	cutsRFPTPC->SetParamType(AliFlowTrackCuts::kGlobal);
	30	cutsRFPTPC->SetPtRange(0.2,5.);
	31	cutsRFPTPC->SetEtaRange(-0.8,0.8);
b57db0d7	32	cutsRFPTPC->SetMinNClustersTPC(nTPCClusters);
fcfcc352	33	cutsRFPTPC->SetMinChi2PerClusterTPC(0.2);
	34	cutsRFPTPC->SetMaxChi2PerClusterTPC(4.0);
	35	cutsRFPTPC->SetAcceptKinkDaughters(kFALSE);
	36	cutsRFPTPC->SetMinimalTPCdedx(10.);
	37	cutsRFPTPC->SetAODfilterBit(1);
	38	// cutsRFPTPC->SetQA(kTRUE);
	39	AliFlowTrackCuts* cutsRFPVZE = new AliFlowTrackCuts( "GlobalRFPVZE" );
9579a93f	40	cutsRFPVZE->SetParamType(AliFlowTrackCuts::kVZERO);
fcfcc352	41	cutsRFPVZE->SetEtaRange(-10,+10);
	42	cutsRFPVZE->SetPhiMin(0);
	43	cutsRFPVZE->SetPhiMax(TMath::TwoPi());
	44	// cutsRFPVZE->SetQA(kTRUE);
	45	AliFlowTrackCuts* cutsPOI = new AliFlowTrackCuts( "GlobalPOI" );
	46	cutsPOI->SetParamType(AliFlowTrackCuts::kGlobal);
	47	cutsPOI->SetPtRange(0.2,-5.);
	48	cutsPOI->SetEtaRange(0.8,-0.8);
	49
	50	//**********************************************************************
	51	// POI FILTER CUTS
	52	AliFlowTrackSimpleCuts *filterPOIQC[50]; // MASS BANDS
	53	AliFlowTrackSimpleCuts *filterPOISP[50][2]; // MASS BANDS \|\| ETA \|\| SUBEVENT GAP
fc4003c2	54	int myMassBands = MassBands( nDmeson, bOldApproach );
8c5f45b8	55	for(int mb=0; mb!=myMassBands; ++mb) {
fcfcc352	56	filterPOISP[mb][0] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETANEG",mb) );
	57	filterPOISP[mb][0]->SetEtaMin( -0.8 );
	58	filterPOISP[mb][0]->SetEtaMax( 0. );
fc4003c2	59	filterPOISP[mb][0]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
fc4003c2	60	filterPOISP[mb][0]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
fcfcc352	61	filterPOISP[mb][1] = new AliFlowTrackSimpleCuts( Form("FilterPOISP_MB%d_ETAPOS",mb) );
	62	filterPOISP[mb][1]->SetEtaMin( 0. );
	63	filterPOISP[mb][1]->SetEtaMax( +0.8 );
fc4003c2	64	filterPOISP[mb][1]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
fc4003c2	65	filterPOISP[mb][1]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
fcfcc352	66	filterPOIQC[mb] = new AliFlowTrackSimpleCuts( Form("FilterPOIQC_MB%d",mb) );
	67	filterPOIQC[mb]->SetEtaMin( -0.8 );
	68	filterPOIQC[mb]->SetEtaMax( +0.8 );
fc4003c2	69	filterPOIQC[mb]->SetMassMin( MassBandLowEdge(nDmeson,mb,bOldApproach) );
fc4003c2	70	filterPOIQC[mb]->SetMassMax( MassBandLowEdge(nDmeson,mb+1,bOldApproach) );
fcfcc352	71	}
fcfcc352	72
8c5f45b8	73	// * DMESON SELECTOR *************************************************
	74	AliAnalysisTaskFlowD2H *taskSel;
	75	switch (nDmeson) {
	76	case ( AliRDHFCuts::kD0toKpiCuts ):
fc4003c2	77	AliRDHFCutsD0toKpi myCutsD0 = (AliRDHFCutsD0toKpi)filecuts->Get("D0toKpiCuts");
8c5f45b8	78	if(!myCutsD0) {
	79	AliFatal("Problems reaching D0toKpiCuts");
	80	}
fc4003c2	81	taskSel = new AliAnalysisTaskFlowD2H( Form("TaskD0Selector_%s",thecuts.Data()),
fc4003c2	82	cutsRFPTPC, cutsRFPVZE, myCutsD0, nDmeson );
8c5f45b8	83	break;
8c5f45b8	84	case ( AliRDHFCuts::kDstarCuts ):
fc4003c2	85	AliRDHFCutsDStartoKpipi myCutsDStar = (AliRDHFCutsDStartoKpipi)filecuts->Get("DStartoKpipiCuts");
8c5f45b8	86	if(!myCutsDStar) {
8c5f45b8	87	AliFatal("Problems reaching DStarToKpipiCuts");
fcfcc352	88	}
fc4003c2	89	taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDStarSelector_%s",thecuts.Data()),
fc4003c2	90	cutsRFPTPC, cutsRFPVZE, myCutsDStar, nDmeson);
8c5f45b8	91	break;
8c5f45b8	92	case (AliRDHFCuts::kDplusCuts):
fc4003c2	93	AliRDHFCutsDplustoKpipi myCutsDplus = (AliRDHFCutsDplustoKpipi)filecuts->Get("AnalysisCuts");
8c5f45b8	94	if(!myCutsDplus) {
	95	AliFatal("Problems reaching AnalysisCuts");
	96	}
fc4003c2	97	taskSel = new AliAnalysisTaskFlowD2H( Form("TaskDplusSelector_%s",thecuts.Data()),
fc4003c2	98	cutsRFPTPC, cutsRFPVZE, myCutsDplus, nDmeson );
8c5f45b8	99	break;
8c5f45b8	100	}
fc4003c2	101	taskSel->SetCommonConstants( MassBins(nDmeson), MinMass(nDmeson), MaxMass(nDmeson), ptBinWidth );
85b252b1	102	if (swapAssumption)
85b252b1	103	taskSel->SetSwapAsumption();
fc4003c2	104
fc4003c2	105	//taskSel->SelectCollisionCandidates(trigger);
8c5f45b8	106	//taskSel->SetDebug();
fc4003c2	107	AliAnalysisDataContainer *coutHist = mgr->CreateContainer( Form("%sSelector_%s",sgn.Data(),thecuts.Data()),
8c5f45b8	108	TList::Class(),AliAnalysisManager::kOutputContainer,
fc4003c2	109	Form("%s.root:FlowD2H_%s",fileName.Data(),thecuts.Data()) );
	110	AliAnalysisDataContainer *exc_TPC = mgr->CreateContainer( Form("TPCEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
	111	AliFlowEventSimple::Class(),
	112	AliAnalysisManager::kExchangeContainer );
	113	AliAnalysisDataContainer *exc_VZE = mgr->CreateContainer( Form("VZEEventWithCandidates_%s_%s",sgn.Data(),thecuts.Data()),
	114	AliFlowEventSimple::Class(),
	115	AliAnalysisManager::kExchangeContainer );
8c5f45b8	116	mgr->AddTask(taskSel);
8c5f45b8	117	mgr->ConnectOutput(taskSel,1,coutHist);
fc4003c2	118	mgr->ConnectOutput(taskSel,2,exc_TPC);
fc4003c2	119	mgr->ConnectOutput(taskSel,3,exc_VZE);
8c5f45b8	120	mgr->ConnectInput (taskSel,0,cinput1);
fc4003c2	121
8c5f45b8	122	// * HANGING ANALYSIS TASKS ********************************************
	123	int harm = myHarmonic;
	124	for(int mb=0; mb!=myMassBands; ++mb) {
	125	if(bDoQC) {
fc4003c2	126	AddQCmethod( Form("%sQCTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), harm, exc_TPC, filterPOIQC[mb]);
8c5f45b8	127	}
8c5f45b8	128	if(bDoSPVZERO) {
fc4003c2	129	AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
	130	"Qa", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
	131	AddSPmethod( Form("%sSPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
	132	"Qb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, false, shrinkSP );
	133	}
	134	if(bDoEPVZERO) {
2a9a9101	135	AddSPmethod( Form("%sEPVZEMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), etaVZERO1, etaVZERO2, etaVZERO3, etaVZERO4,
fc4003c2	136	"QaQb", harm, exc_VZE, 0, filterPOIQC[mb], NULL, true, shrinkSP );
8c5f45b8	137	}
	138	if(bDoSPTPC) {
	139	for(int eg=0; eg!=2; ++eg) {
fc4003c2	140	AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPCeg, +gapTPCeg, +0.8,
	141	"Qa", harm, exc_TPC, eg, filterPOISP[mb][1], NULL, false, shrinkSP );
	142	AddSPmethod( Form("%sSPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -gapTPCeg, +gapTPCeg, +0.8,
	143	"Qb", harm, exc_TPC, eg, filterPOISP[mb][0], NULL, false, shrinkSP );
fcfcc352	144	}
8c5f45b8	145	}
fc4003c2	146	if(bDoEPTPC) {
fc4003c2	147	AddSPmethod( Form("%sEPTPCMB%d",sgn.Data(),mb), fileName.Data(), thecuts.Data(), -0.8, -0.0, +0.0, +0.8,
2a9a9101	148	"QaQb", harm, exc_TPC, 0, filterPOIQC[mb], NULL, true, shrinkSP );
fc4003c2	149	}
a8f6c03f	150	}
fcfcc352	151	}
fcfcc352	152
fc4003c2	153	void AddSPmethod(char name, char fileName, char *thecuts,
fcfcc352	154	double minEtaA, double maxEtaA, double minEtaB, double maxEtaB,
fcfcc352	155	char Qvector, int harmonic, AliAnalysisDataContainer flowEvent, int eg,
bde814f5	156	AliFlowTrackSimpleCuts cutsPOI=NULL, AliFlowTrackSimpleCuts cutsRFP=NULL,
fc4003c2	157	bool bEP, bool shrink=false ) {
	158	TString myFolder = Form("v%d_%s",harmonic,thecuts);
	159	TString myNameSP = Form("%sSPv%d%sGAP%d_%s",name,harmonic,Qvector,eg,thecuts);
fcfcc352	160	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	161	AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myNameSP.Data()),
	162	AliFlowEventSimple::Class(),
	163	AliAnalysisManager::kExchangeContainer );
	164	AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myNameSP.Data()),
	165	cutsRFP, cutsPOI);
	166	tskFilter->SetSubeventEtaRange( minEtaA, maxEtaA, minEtaB, maxEtaB );
	167	mgr->AddTask(tskFilter);
	168	mgr->ConnectInput( tskFilter,0,flowEvent);
	169	mgr->ConnectOutput(tskFilter,1,flowEvent2);
fcfcc352	170	AliAnalysisDataContainer *outSP = mgr->CreateContainer( myNameSP.Data(),
	171	TList::Class(),AliAnalysisManager::kOutputContainer,
	172	Form("%s.root:FlowD2H_SP_%s",fileName,myFolder.Data()) );
	173	AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct( Form("TaskScalarProduct_%s",
	174	myNameSP.Data()),kFALSE);
	175	tskSP->SetApplyCorrectionForNUA(kTRUE);
	176	tskSP->SetHarmonic(harmonic);
	177	tskSP->SetTotalQvector(Qvector);
fc4003c2	178	if(bEP) tskSP->SetBehaveAsEP();
fc4003c2	179	if(shrink) tskSP->SetBookOnlyBasicCCH(kTRUE);
fcfcc352	180	mgr->AddTask(tskSP);
	181	mgr->ConnectInput( tskSP,0,flowEvent2);
	182	mgr->ConnectOutput(tskSP,1,outSP);
fcfcc352	183	}
fcfcc352	184
fc4003c2	185	void AddQCmethod(char name, char fileName, char *thecuts,
fcfcc352	186	int harmonic, AliAnalysisDataContainer *flowEvent,
fcfcc352	187	AliFlowTrackSimpleCuts cutsPOI=NULL, AliFlowTrackSimpleCuts cutsRFP=NULL) {
fc4003c2	188	TString myFolder = Form("v%d_%s",harmonic,thecuts);
fc4003c2	189	TString myName = Form("%sv%d_%s",name,harmonic,thecuts);
fcfcc352	190	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	191	AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myName.Data()),
	192	AliFlowEventSimple::Class(),
	193	AliAnalysisManager::kExchangeContainer );
	194	AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myName.Data()),
	195	cutsRFP, cutsPOI);
	196	mgr->AddTask(tskFilter);
	197	mgr->ConnectInput( tskFilter,0,flowEvent);
	198	mgr->ConnectOutput(tskFilter,1,flowEvent2);
	199	AliAnalysisDataContainer *outQC = mgr->CreateContainer( myName.Data(),
	200	TList::Class(),AliAnalysisManager::kOutputContainer,
	201	Form("%s.root:FlowD2H_QC_%s",fileName,myFolder.Data()) );
	202	AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants( Form("TaskQCumulants_%s",
	203	myName.Data()),kFALSE);
	204	tskQC->SetApplyCorrectionForNUA(kTRUE);
	205	tskQC->SetHarmonic(harmonic);
fc4003c2	206	tskQC->SetBookOnlyBasicCCH(kTRUE);
fcfcc352	207	mgr->AddTask(tskQC);
	208	mgr->ConnectInput( tskQC,0,flowEvent2);
	209	mgr->ConnectOutput(tskQC,1,outQC);
	210	}
	211
fc4003c2	212	int MassBands( int nDmeson, bool bOldApproach=false ) {
fc4003c2	213	switch (nDmeson) {
30121a82	214	case ( AliRDHFCuts::kDplusCuts ):
fc4003c2	215	case ( AliRDHFCuts::kD0toKpiCuts ):
	216	if(bOldApproach) return 5;
	217	else return 26;
	218	case ( AliRDHFCuts::kDstarCuts ):
e079069f	219	return 13;
fc4003c2	220	}
fc4003c2	221	}
fcfcc352	222
fc4003c2	223	double MassBandLowEdge( int nDmeson, int mb, bool bOldApproach=false ) {
fc4003c2	224	switch (nDmeson) {
30121a82	225	case ( AliRDHFCuts::kDplusCuts ):
fc4003c2	226	case ( AliRDHFCuts::kD0toKpiCuts ): // 2 + 20 + 4
	227	double lowEdgeMinimal[5+1] = {1.75,1.80,1.83,1.90,1.93,2.03};
	228	double lowEdge[26+1] = { 1.66, 1.71, 1.76, 1.77, 1.78, 1.79, 1.80, 1.81, 1.82, 1.83,
	229	1.84, 1.85, 1.86, 1.87, 1.88, 1.89, 1.90, 1.91, 1.92, 1.93,
	230	1.94, 1.95, 1.96, 2.01, 2.06, 2.11, 2.16 };
	231	if(bOldApproach) return lowEdgeMinimal[mb];
	232	else return lowEdge[mb];
2a9a9101	233	case ( AliRDHFCuts::kDstarCuts ): // 2 + 10 + 3
e079069f	234	double lowEdge[13+1]={0.138, 0.139, 0.141, 0.143, 0.144, 0.145, 0.146, 0.147, 0.148, 0.150, 0.152, 0.154, 0.156, 0.158};
2a9a9101	235	return lowEdge[mb];
fc4003c2	236	}
fc4003c2	237	}
fcfcc352	238
fc4003c2	239	double MinMass( int nDmeson ) {
fc4003c2	240	switch (nDmeson) {
30121a82	241	case ( AliRDHFCuts::kDplusCuts ):
fc4003c2	242	case ( AliRDHFCuts::kD0toKpiCuts ):
	243	return 1.66;
	244	case ( AliRDHFCuts::kDstarCuts ):
2a9a9101	245	return 0.138;
fc4003c2	246	}
fcfcc352	247	}
fcfcc352	248
fc4003c2	249	double MaxMass( int nDmeson ) {
fcfcc352	250	switch (nDmeson) {
30121a82	251	case ( AliRDHFCuts::kDplusCuts ):
fcfcc352	252	case ( AliRDHFCuts::kD0toKpiCuts ):
fc4003c2	253	return 2.16;
fcfcc352	254	case ( AliRDHFCuts::kDstarCuts ):
2a9a9101	255	return 0.158;
a8f6c03f	256	}
fcfcc352	257	}
fcfcc352	258
fc4003c2	259	int MassBins( int nDmeson ) {
fcfcc352	260	switch (nDmeson) {
30121a82	261	case ( AliRDHFCuts::kDplusCuts ):
fcfcc352	262	case ( AliRDHFCuts::kD0toKpiCuts ):
	263	return 50;
	264	case ( AliRDHFCuts::kDstarCuts ):
e079069f	265	return 40;
a8f6c03f	266	}
	267	}
	268
fcfcc352	269	TString DMesonName( int nDmeson ) {
	270	TString toReturn;
	271	switch (nDmeson) {
30121a82	272	case ( AliRDHFCuts::kDplusCuts ):
30121a82	273	toReturn = "DPlus"; break;
fcfcc352	274	case ( AliRDHFCuts::kD0toKpiCuts ):
	275	toReturn = "D0"; break;
	276	case ( AliRDHFCuts::kDstarCuts ):
	277	toReturn = "DStar"; break;
	278	}
	279	return toReturn;
	280	}