PWGGA/GammaConv/macros/AddTask_GammaConvDalitzV1_PbPb.C

   1 void AddTask_GammaConvDalitzV1_PbPb(   Int_t trainConfig = 1,
   2                                        Bool_t isMC   = kFALSE, //run MC
   3                                        Bool_t enableQAMesonTask = kFALSE, //enable QA in AliAnalysisTaskGammaConvDalitzV1
   4                                        Bool_t enableDoMesonChic = kFALSE, // enable additional Chic analysis
   5                                        TString fileNameInputForWeighting = "MCSpectraInput.root", // path to file for weigting input
   6                                        Bool_t doWeighting = kFALSE,  //enable Weighting
   7                                        TString cutnumberAODBranch = "0000000060084001001500000"
   8                                   ) {
   9
  10
  11
  12    cout<<"Entro -1"<<endl;
  13
  14   // ================= Load Librariers =================================
  15    gSystem->Load("libCore.so");
  16    gSystem->Load("libTree.so");
  17    gSystem->Load("libGeom.so");
  18    gSystem->Load("libVMC.so");
  19    gSystem->Load("libPhysics.so");
  20    gSystem->Load("libMinuit");
  21    gSystem->Load("libSTEERBase");
  22    gSystem->Load("libESD");
  23    gSystem->Load("libAOD");
  24    gSystem->Load("libANALYSIS");
  25    gSystem->Load("libANALYSISalice");
  26    gSystem->Load("libPWGGAGammaConv.so");
  27    gSystem->Load("libCDB.so");
  28    gSystem->Load("libSTEER.so");
  29    gSystem->Load("libSTEERBase.so");
  30    gSystem->Load("libTENDER.so");
  31    gSystem->Load("libTENDERSupplies.so");
  32
  33
  34    cout<<"Entro 0"<<endl;
  35
  36    // ================== GetAnalysisManager ===============================
  37    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  38    if (!mgr) {
  39       Error(Form("AddTask_GammaConvDalitzV1_PbPb_%i",trainConfig), "No analysis manager found.");
  40       return ;
  41    }
  42
  43    // ================== GetInputEventHandler =============================
  44    AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
  45
  46    //========= Add PID Reponse to ANALYSIS manager ====
  47    if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
  48       gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
  49       AddTaskPIDResponse(isMC);
  50    }
  51
  52    //=========  Set Cutnumber for V0Reader ================================
  53    TString ConvCutnumber = "108000000008400100150000000";   //Online  V0 finder
  54    TString ElecCuts      = "9031620005500202100";            //Electron Cuts
  55                             //903162000550020210
  56                             //900054000000020000
  57
  58
  59    AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
  60
  61    //========= Add V0 Reader to  ANALYSIS manager if not yet existent =====
  62    if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
  63       AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
  64
  65       fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
  66       fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
  67       fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
  68
  69       if (!mgr) {
  70          Error("AddTask_V0ReaderV1", "No analysis manager found.");
  71          return;
  72       }
  73
  74       // Set AnalysisCut Number
  75       AliConversionCuts *fCuts=NULL;
  76       if( ConvCutnumber !=""){
  77          fCuts= new AliConversionCuts(ConvCutnumber.Data(),ConvCutnumber.Data());
  78          fCuts->SetPreSelectionCutFlag(kTRUE);
  79          if(fCuts->InitializeCutsFromCutString(ConvCutnumber.Data())){
  80             fV0ReaderV1->SetConversionCuts(fCuts);
  81             fCuts->SetFillCutHistograms("",kTRUE);
  82          }
  83       }
  84       if(inputHandler->IsA()==AliAODInputHandler::Class()){
  85       // AOD mode
  86          fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
  87       }
  88       fV0ReaderV1->Init();
  89
  90       AliLog::SetGlobalLogLevel(AliLog::kInfo);
  91
  92       //connect input V0Reader
  93       mgr->AddTask(fV0ReaderV1);
  94       mgr->ConnectInput(fV0ReaderV1,0,cinput);
  95
  96    }
  97
  98    //================================================
  99    //========= Add Electron Selector ================
 100
 101
 102    if( !(AliDalitzElectronSelector*)mgr->GetTask("ElectronSelector") ){
 103
 104       AliDalitzElectronSelector *fElectronSelector = new AliDalitzElectronSelector("ElectronSelector");
 105
 106       // Set AnalysisCut Number
 107       AliDalitzElectronCuts *fElecCuts=0;
 108
 109       if( ElecCuts!=""){
 110          fElecCuts= new AliDalitzElectronCuts(ElecCuts.Data(),ElecCuts.Data());
 111          if(fElecCuts->InitializeCutsFromCutString(ElecCuts.Data())){
 112             fElectronSelector->SetDalitzElectronCuts(fElecCuts);
 113             fElecCuts->SetFillCutHistograms("",kTRUE);
 114          }
 115       }
 116
 117       fElectronSelector->Init();
 118       mgr->AddTask(fElectronSelector);
 119
 120       AliAnalysisDataContainer *cinput1  = mgr->GetCommonInputContainer();
 121       //connect input V0Reader
 122       mgr->ConnectInput (fElectronSelector,0,cinput1);
 123    }
 124
 125
 126
 127     cout<<"Entro"<<endl;
 128    //================================================
 129    //========= Add task to the ANALYSIS manager =====
 130    //================================================
 131    //            find input container
 132
 133
 134
 135    AliAnalysisTaskGammaConvDalitzV1 *task=NULL;
 136    task= new AliAnalysisTaskGammaConvDalitzV1(Form("GammaConvDalitzV1_%i",trainConfig));
 137    task->SetIsHeavyIon(1);
 138    task->SetIsMC(isMC);
 139
 140
 141
 142    // Cut Numbers to use in Analysis
 143    Int_t numberOfCuts = 3;
 144
 145    TString *ConvCutarray    = new TString[numberOfCuts];
 146    TString *ElecCutarray    = new TString[numberOfCuts];
 147    TString *MesonCutarray   = new TString[numberOfCuts];
 148
 149    if( trainConfig == 1 ) {
 150         ConvCutarray[0]  = "124000104209297100322000000"; MesonCutarray[0] = "01522045009000"; ElecCutarray[0]  = "9051620025530252170"; //PbPb 20-40% kAny
 151         ConvCutarray[1]  = "146000104209297100322000000"; MesonCutarray[1] = "01522045009000"; ElecCutarray[1]  = "9051620025530252170"; //PbPb 40-60% kAny
 152         ConvCutarray[2]  = "168000104209297100322000000"; MesonCutarray[2] = "01522045009000"; ElecCutarray[2]  = "9051620025530252170"; //PbPb 60-80% kAny
 153    } else if ( trainConfig == 2 ) {
 154         ConvCutarray[0]  = "524000104209297100322000000"; MesonCutarray[0] = "01522045009000"; ElecCutarray[0]  = "9051620025530252170"; //PbPb 20-40% kAny
 155         ConvCutarray[1]  = "546000104209297100322000000"; MesonCutarray[1] = "01522045009000"; ElecCutarray[1]  = "9051620025530252170"; //PbPb 40-60% kAny
 156         ConvCutarray[2]  = "568000104209297100322000000"; MesonCutarray[2] = "01522045009000"; ElecCutarray[2]  = "9051620025530252170"; //PbPb 60-80% kAny
 157    } else if ( trainConfig == 3 ) {
 158         ConvCutarray[0]  = "524000104209297100322000000"; MesonCutarray[0] = "01522045009000"; ElecCutarray[0]  = "9051620025530252171"; //PbPb 20-40% kAny
 159         ConvCutarray[1]  = "546000104209297100322000000"; MesonCutarray[1] = "01522045009000"; ElecCutarray[1]  = "9051620025530252171"; //PbPb 40-60% kAny
 160         ConvCutarray[2]  = "568000104209297100322000000"; MesonCutarray[2] = "01522045009000"; ElecCutarray[2]  = "9051620025530252171"; //PbPb 60-80% kAny
 161    }
 162
 163
 164    TList *ConvCutList  = new TList();
 165    TList *MesonCutList = new TList();
 166    TList *ElecCutList  = new TList();
 167
 168    TList *HeaderList = new TList();
 169    TObjString *Header1 = new TObjString("pi0_1");
 170    HeaderList->Add(Header1);
 171
 172   //TObjString *Header3 = new TObjString("eta_2");
 173   //HeaderList->Add(Header3);
 174
 175    ConvCutList->SetOwner(kTRUE);
 176    AliConversionCuts **analysisCuts             = new AliConversionCuts*[numberOfCuts];
 177
 178    MesonCutList->SetOwner(kTRUE);
 179    AliConversionMesonCuts **analysisMesonCuts   = new AliConversionMesonCuts*[numberOfCuts];
 180
 181    ElecCutList->SetOwner(kTRUE);
 182    AliDalitzElectronCuts **analysisElecCuts     = new AliDalitzElectronCuts*[numberOfCuts];
 183
 184
 185
 186    for(Int_t i = 0; i<numberOfCuts; i++){
 187       analysisCuts[i] = new AliConversionCuts();
 188       if( ! analysisCuts[i]->InitializeCutsFromCutString(ConvCutarray[i].Data()) ) {
 189             cout<<"ERROR: analysisCuts [" <<i<<"]"<<endl;
 190             return 0;
 191       } else {
 192          ConvCutList->Add(analysisCuts[i]);
 193          analysisCuts[i]->SetFillCutHistograms("",kFALSE);
 194          if( trainConfig == 1){
 195             if (i == 0 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_2040V0M", "", "","Pi0_Fit_Data_PbPb_2760GeV_2040V0M");
 196             if (i == 1 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_4060V0M", "", "","Pi0_Fit_Data_PbPb_2760GeV_4060V0M");
 197             if (i == 2 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_6080V0M", "", "","Pi0_Fit_Data_PbPb_2760GeV_6080V0M");
 198          } else if ( trainConfig == 3 || trainConfig == 2) {
 199             if (i == 0 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_2040TPC", "", "","Pi0_Fit_Data_PbPb_2760GeV_2040V0M");
 200             if (i == 1 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_4060TPC", "", "","Pi0_Fit_Data_PbPb_2760GeV_4060V0M");
 201             if (i == 2 && doWeighting) analysisCuts[i]->SetUseReweightingWithHistogramFromFile(kTRUE, kFALSE, kFALSE, fileNameInputForWeighting, "Pi0_Hijing_LHC13d2_PbPb_2760GeV_6080TPC", "", "","Pi0_Fit_Data_PbPb_2760GeV_6080V0M");
 202          }
 203       }
 204
 205       analysisMesonCuts[i] = new AliConversionMesonCuts();
 206       if( ! analysisMesonCuts[i]->InitializeCutsFromCutString(MesonCutarray[i].Data()) ) {
 207             cout<<"ERROR: analysisMesonCuts [ " <<i<<" ] "<<endl;
 208             return 0;
 209       } else {
 210         MesonCutList->Add(analysisMesonCuts[i]);
 211         analysisMesonCuts[i]->SetFillCutHistograms("");
 212       }
 213
 214       TString cutName( Form("%s_%s_%s",ConvCutarray[i].Data(),ElecCutarray[i].Data(),MesonCutarray[i].Data() ) );
 215       analysisElecCuts[i] = new AliDalitzElectronCuts();
 216       if( !analysisElecCuts[i]->InitializeCutsFromCutString(ElecCutarray[i].Data())) {
 217             cout<< "ERROR:  analysisElecCuts [ " <<i<<" ] "<<endl;
 218             return 0;
 219       }  else {
 220          ElecCutList->Add(analysisElecCuts[i]);
 221          analysisElecCuts[i]->SetFillCutHistograms("",kFALSE,cutName);
 222       }
 223       analysisCuts[i]->SetAcceptedHeader(HeaderList);
 224
 225    }
 226
 227
 228    task->SetConversionCutList(numberOfCuts,ConvCutList);
 229    task->SetMesonCutList(MesonCutList);
 230    task->SetElectronCutList(ElecCutList);
 231
 232    task->SetMoveParticleAccordingToVertex(kTRUE);
 233
 234
 235    if(enableQAMesonTask) task->SetDoMesonQA(kTRUE);
 236    if(enableDoMesonChic) task->SetDoChicAnalysis(kTRUE);
 237
 238    //connect containers
 239    AliAnalysisDataContainer *coutput =
 240    mgr->CreateContainer(Form("GammaConvDalitzV1_%i",trainConfig), TList::Class(),
 241                            AliAnalysisManager::kOutputContainer,Form("GammaConvV1Dalitz_%i.root",trainConfig));
 242
 243    mgr->AddTask(task);
 244    mgr->ConnectInput(task,0,cinput);
 245    mgr->ConnectOutput(task,1,coutput);
 246
 247    return;
 248
 249 }