extra cuts

[u/mrichter/AliRoot.git] / PWGCF / FLOW / macros / AddTaskFlowStrange.C

void AddTaskFlowStrange(int trigger, int centrMin, int centrMax, int harmonic=2) {\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"K0",Form("K%d%dc0",centrMin,centrMax),   0,0,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"K0",Form("K%d%dc1",centrMin,centrMax),   0,1,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"K0",Form("K%d%dc2",centrMin,centrMax),   0,2,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"K0",Form("K%d%dc3",centrMin,centrMax),   0,3,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"K0",Form("K%d%dc2TRK",centrMin,centrMax),0,2,"TRK",harmonic);\r
\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"L0",Form("L%d%dc0",centrMin,centrMax),   1,0,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"L0",Form("L%d%dc1",centrMin,centrMax),   1,1,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"L0",Form("L%d%dc2",centrMin,centrMax),   1,2,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"L0",Form("L%d%dc3",centrMin,centrMax),   1,3,"V0M",harmonic);\r
  AddTaskFlowStrange(trigger,centrMin,centrMax,"L0",Form("L%d%dc2TRK",centrMin,centrMax),1,2,"TRK",harmonic);\r
}\r
void AddTaskFlowStrange(int trigger, float centrMin, float centrMax, TString folderName="myFolder", TString suffixName="mySuffix", \r
			int specie=0, int cuts=1, char* MULT="V0M", int harmonic=2) {\r
  TString fileName = AliAnalysisManager::GetCommonFileName();\r
  fileName.ReplaceAll(".root","");\r
\r
  //-E-V-E-N-T- -c-u-t-s--------------------------------------------------------\r
  AliFlowEventCuts* cutsEvent = new AliFlowEventCuts(Form("event_cuts_%s",suffixName.Data()));\r
  cutsEvent->SetCentralityPercentileRange(centrMin,centrMax);\r
  if(MULT=="V0M")\r
    cutsEvent->SetCentralityPercentileMethod(AliFlowEventCuts::kV0);\r
  else\r
    cutsEvent->SetCentralityPercentileMethod(AliFlowEventCuts::kTPConly);\r
  cutsEvent->SetRefMultMethod(AliFlowEventCuts::kSPDtracklets);\r
  cutsEvent->SetNContributorsRange(2);\r
  cutsEvent->SetPrimaryVertexZrange(-9.5,+9.5);\r
\r
  //-R-P---c-u-t-s--------------------------------------------------------------\r
  AliFlowTrackCuts *cutsRPTPC = AliFlowTrackCuts::GetStandardTPCStandaloneTrackCuts();\r
  cutsRPTPC->SetParamType( AliFlowTrackCuts::kGlobal );\r
  cutsRPTPC->SetAODfilterBit(1); // for AOD compatibility\r
  AliFlowTrackCuts *cutsRPVZE = AliFlowTrackCuts::GetStandardVZEROOnlyTrackCuts();\r
\r
  //-D-A-U-G-H-T-E-R-S---c-u-t-s------------------------------------------------\r
  AliESDtrackCuts* cutsDaughter = new AliESDtrackCuts(Form("daughter_cuts_%s",suffixName.Data()) );\r
  cutsDaughter->SetPtRange(0.2,10.0);\r
  cutsDaughter->SetEtaRange(-0.8, 0.8 );\r
  cutsDaughter->SetMinNClustersTPC(80);\r
  cutsDaughter->SetMaxChi2PerClusterTPC(4.0);\r
  cutsDaughter->SetRequireTPCRefit(kTRUE);\r
  cutsDaughter->SetAcceptKinkDaughters(kFALSE);\r
\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();\r
\r
  //-----------------STRANGE TASK----------------------------\r
  AliAnalysisTaskFlowStrange *taskSel = new AliAnalysisTaskFlowStrange(Form("Strange_%s",suffixName.Data()),\r
								       cutsEvent, cutsRPTPC, cutsRPVZE,\r
								       cutsDaughter );\r
  taskSel->SelectCollisionCandidates(trigger);\r
  //taskSel->SetDebug();\r
  taskSel->SetCuts2010(cuts);\r
  taskSel->SetK0L0(specie);\r
  //printf( "CMM %d %f %f\n", MassBins(specie), MinMass(specie), MaxMass(specie) );\r
  taskSel->SetCommonConstants( MassBins(specie), MinMass(specie), MaxMass(specie) );\r
  AliAnalysisDataContainer *cOutHist = mgr->CreateContainer(Form("OutHistos_%s",suffixName.Data()),\r
							    TList::Class(),\r
							    AliAnalysisManager::kOutputContainer,\r
							    Form("%s.root:Selector_%s",fileName.Data(),\r
								 folderName.Data()));\r
  AliAnalysisDataContainer *exc_TPC = mgr->CreateContainer( Form("TPCEventWithCandidates_%s",suffixName.Data()),\r
                                                            AliFlowEventSimple::Class(),\r
                                                            AliAnalysisManager::kExchangeContainer );\r
  AliAnalysisDataContainer *exc_VZE = mgr->CreateContainer( Form("VZEEventWithCandidates_%s",suffixName.Data()),\r
                                                            AliFlowEventSimple::Class(),\r
                                                            AliAnalysisManager::kExchangeContainer );\r
  mgr->AddTask(taskSel);\r
  mgr->ConnectInput (taskSel,0,cinput1);\r
  mgr->ConnectOutput(taskSel,1,exc_TPC);\r
  mgr->ConnectOutput(taskSel,2,exc_VZE);\r
  mgr->ConnectOutput(taskSel,3,cOutHist);\r
\r
  //-------------------FLOW TASKS----------------------------\r
  AliFlowTrackSimpleCuts *filter[15], *filterhf[15][2]; // MASS BANDS\r
  for(int mb=0; mb!=MassBands(0); ++mb) {\r
    filter[mb] = new AliFlowTrackSimpleCuts( Form("Filter_MB%d",mb) );\r
    filter[mb]->SetEtaMin( -0.8 ); filter[mb]->SetEtaMax( +0.8 );\r
    filter[mb]->SetMassMin( MassBandLowEdge(specie,mb) ); filter[mb]->SetMassMax( MassBandLowEdge(specie,mb+1) );\r
\r
    filterhf[mb][0] = new AliFlowTrackSimpleCuts( Form("Filterhf0_MB%d",mb) );\r
    filterhf[mb][0]->SetEtaMin( 0.0 ); filterhf[mb][0]->SetEtaMax( +0.8 );\r
    filterhf[mb][0]->SetMassMin( MassBandLowEdge(specie,mb) ); filterhf[mb][0]->SetMassMax( MassBandLowEdge(specie,mb+1) );\r
\r
    filterhf[mb][1] = new AliFlowTrackSimpleCuts( Form("Filterhf1_MB%d",mb) );\r
    filterhf[mb][1]->SetEtaMin( -0.8 ); filterhf[mb][1]->SetEtaMax( 0.0 );\r
    filterhf[mb][1]->SetMassMin( MassBandLowEdge(specie,mb) ); filterhf[mb][1]->SetMassMax( MassBandLowEdge(specie,mb+1) );\r
\r
    AddQCmethod( Form("QCTPCMB%d",mb), folderName.Data(), suffixName.Data(), harmonic, exc_TPC, filter[mb]); // QC TPC\r
    AddSPmethod( Form("SPTPCMB%d",mb), folderName.Data(), suffixName.Data(), harmonic, exc_TPC, filterhf[mb][0], "Qa" ); // SP TPC Qa\r
    AddSPmethod( Form("SPTPCMB%d",mb), folderName.Data(), suffixName.Data(), harmonic, exc_TPC, filterhf[mb][1], "Qb" ); // SP TPC Qb\r
    AddSPmethod( Form("SPVZEMB%d",mb), folderName.Data(), suffixName.Data(), harmonic, exc_VZE, filter[mb], "Qa" ); // SP VZE Qa\r
    AddSPmethod( Form("SPVZEMB%d",mb), folderName.Data(), suffixName.Data(), harmonic, exc_VZE, filter[mb], "Qb" ); // SP VZE Qa\r
  }\r
}\r
\r
void AddQCmethod(char *name, TString myFolder, char *thecuts, int harmonic, \r
		 AliAnalysisDataContainer *flowEvent, AliFlowTrackSimpleCuts *cutsPOI=NULL) {\r
  TString fileName = AliAnalysisManager::GetCommonFileName();\r
  myFolder.Append( Form("v%d",harmonic) );\r
  TString myName = Form("%sv%d_%s",name,harmonic,thecuts);\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myName.Data()),\r
                                                               AliFlowEventSimple::Class(),\r
                                                               AliAnalysisManager::kExchangeContainer );\r
  AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myName.Data()),\r
                                                                    NULL, cutsPOI);\r
  mgr->AddTask(tskFilter);\r
  mgr->ConnectInput( tskFilter,0,flowEvent);\r
  mgr->ConnectOutput(tskFilter,1,flowEvent2);\r
  AliAnalysisDataContainer *outQC = mgr->CreateContainer( myName.Data(),TList::Class(),AliAnalysisManager::kOutputContainer,\r
                                                          Form("%s:FlowStrange_QC_%s",fileName.Data(),myFolder.Data()) );\r
  AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants( Form("TaskQCumulants_%s",myName.Data()),kFALSE );\r
  tskQC->SetApplyCorrectionForNUA(kTRUE);\r
  tskQC->SetHarmonic(harmonic);\r
  tskQC->SetBookOnlyBasicCCH(kTRUE);\r
  mgr->AddTask(tskQC);\r
  mgr->ConnectInput( tskQC,0,flowEvent2);\r
  mgr->ConnectOutput(tskQC,1,outQC);\r
}\r
\r
\r
void AddSPmethod(char *name, TString myFolder, char *thecuts, int harmonic,\r
		 AliAnalysisDataContainer *flowEvent, AliFlowTrackSimpleCuts *cutsPOI=NULL,\r
                 char *Qvector) {\r
  TString fileName = AliAnalysisManager::GetCommonFileName();\r
  myFolder.Append( Form("v%d",harmonic) );\r
  TString myNameSP = Form("%sv%d%s_%s",name,harmonic,Qvector,thecuts);\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer( Form("Filter_%s", myNameSP.Data()),\r
                                                               AliFlowEventSimple::Class(),\r
                                                               AliAnalysisManager::kExchangeContainer );\r
  AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE( Form("TaskFilter_%s",myNameSP.Data()),\r
                                                                    NULL, cutsPOI);\r
  tskFilter->SetSubeventEtaRange( -5.0, 0.0, 0.0, +5.0 );\r
  mgr->AddTask(tskFilter);\r
  mgr->ConnectInput( tskFilter,0,flowEvent);\r
  mgr->ConnectOutput(tskFilter,1,flowEvent2);\r
  AliAnalysisDataContainer *outSP = mgr->CreateContainer( myNameSP.Data(),TList::Class(),AliAnalysisManager::kOutputContainer,\r
                                                          Form("%s:FlowStrange_SP_%s",fileName.Data(),myFolder.Data()) );\r
  AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct( Form("TaskScalarProduct_%s",myNameSP.Data()),kFALSE);\r
  tskSP->SetApplyCorrectionForNUA(kTRUE);\r
  tskSP->SetHarmonic(harmonic);\r
  tskSP->SetTotalQvector(Qvector);\r
  tskSP->SetBookOnlyBasicCCH(kTRUE);\r
  mgr->AddTask(tskSP);\r
  mgr->ConnectInput( tskSP,0,flowEvent2);\r
  mgr->ConnectOutput(tskSP,1,outSP);\r
}\r
\r
double MassBandLowEdge( int nv0, int mb ) {\r
  if(nv0==0) {\r
    double lowEdge[15+1]={ 0.452, 0.462, 0.472, 0.482, 0.489, 0.492, 0.495, 0.498,\r
			   0.501, 0.504, 0.507, 0.514, 0.524, 0.534, 0.544, 0.554 };\r
  } if(nv0==1) {\r
    double lowEdge[15+1]={ 1.094, 1.099, 1.104, 1.109, 1.112, 1.114, 1.115, 1.116, \r
			   1.117, 1.118, 1.120, 1.123, 1.128, 1.133, 1.138, 1.143 };\r
  }\r
  return lowEdge[mb];\r
}\r
\r
int MassBands( int nv0 ) {\r
  if(nv0==0) {\r
    return 15;\r
  } else if(nv0==1) {\r
    return 15;\r
  }\r
}\r
\r
int MassBins( int nv0 ) {\r
  if(nv0==0) {\r
    return 102;\r
  } else if(nv0==1) {\r
    return 49;\r
  }\r
}\r
\r
double MinMass( int nv0 ) {\r
  return MassBandLowEdge( nv0, 0 );\r
}\r
\r
double MaxMass( int nv0 ) {\r
  return MassBandLowEdge( nv0, MassBands(nv0) );\r
}\r