]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGGA/GammaConv/macros/AddTask_GammaConvV1_pPb2.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / macros / AddTask_GammaConvV1_pPb2.C
index 145a4933e6f67fc9d5d5a23fe6a0d920ceae2225..c3f60c8221826d17c51415dfb8d6216080b81875 100644 (file)
@@ -8,169 +8,204 @@ void AddTask_GammaConvV1_pPb2(  Int_t trainConfig = 1,  //change different set o
                               TString cutnumberAODBranch = "8000000060084000001500000" // cutnumber for AOD branch
                            ) {
  
-   // ================= Load Librariers =================================
-   gSystem->Load("libCore.so");  
-   gSystem->Load("libTree.so");
-   gSystem->Load("libGeom.so");
-   gSystem->Load("libVMC.so");
-   gSystem->Load("libPhysics.so");
-   gSystem->Load("libMinuit");
-   gSystem->Load("libSTEERBase");
-   gSystem->Load("libESD");
-   gSystem->Load("libAOD");
-   gSystem->Load("libANALYSIS");
-   gSystem->Load("libANALYSISalice");  
-   gSystem->Load("libPWGGAGammaConv.so");
-   gSystem->Load("libCDB.so");
-   gSystem->Load("libSTEER.so");
-   gSystem->Load("libSTEERBase.so");
-   gSystem->Load("libTENDER.so");
-   gSystem->Load("libTENDERSupplies.so");
+       // ================= Load Librariers =================================
+       gSystem->Load("libCore");
+       gSystem->Load("libTree");
+       gSystem->Load("libGeom");
+       gSystem->Load("libVMC");
+       gSystem->Load("libPhysics");
+       gSystem->Load("libMinuit");
+       gSystem->Load("libSTEERBase");
+       gSystem->Load("libESD");
+       gSystem->Load("libAOD");
+       gSystem->Load("libANALYSIS");
+       gSystem->Load("libANALYSISalice");  
+       gSystem->Load("libCDB");
+       gSystem->Load("libSTEER");
+       gSystem->Load("libSTEERBase");
+       gSystem->Load("libTender");
+       gSystem->Load("libTenderSupplies");
+       gSystem->Load("libPWGflowBase");
+       gSystem->Load("libPWGflowTasks");
+       gSystem->Load("libPWGGAGammaConv");
      
-   // ================== GetAnalysisManager ===============================
-   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
-   if (!mgr) {
-      Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
-      return ;
-   }
-   // ================== GetInputEventHandler =============================
-   AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
-   
-   //========= Add PID Reponse to ANALYSIS manager ====
-   if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
-      gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
-      AddTaskPIDResponse(isMC);
-   }
-   
-   //=========  Set Cutnumber for V0Reader ================================
-   TString cutnumber = "8000000060084001001500000000";
-   Bool_t doEtaShift = kFALSE;
-   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
-   //========= Add V0 Reader to  ANALYSIS manager if not yet existent =====
-   if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
-      AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
-     
-      fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
-      fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
-      fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
-     
-      if (!mgr) {
-         Error("AddTask_V0ReaderV1", "No analysis manager found.");
-         return;
-      }
-      // Set AnalysisCut Number
-      AliConversionCuts *fCuts=NULL;
-      if(cutnumber!=""){
-         fCuts= new AliConversionCuts(cutnumber.Data(),cutnumber.Data());
-         fCuts->SetPreSelectionCutFlag(kTRUE);
-         if(fCuts->InitializeCutsFromCutString(cutnumber.Data())){
-            fCuts->DoEtaShift(doEtaShift);
-            fV0ReaderV1->SetConversionCuts(fCuts);
-            fCuts->SetFillCutHistograms("",kTRUE);
-         }
-      }
-      if(inputHandler->IsA()==AliAODInputHandler::Class()){
-      // AOD mode
-         cout << "AOD handler: adding " << cutnumberAODBranch.Data() << " as conversion branch" << endl;
-         fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
-      }
-      fV0ReaderV1->Init();
-      AliLog::SetGlobalLogLevel(AliLog::kInfo);
-      //connect input V0Reader
-      mgr->AddTask(fV0ReaderV1);
-      mgr->ConnectInput(fV0ReaderV1,0,cinput);
-   }
-   //================================================
-   //========= Add task to the ANALYSIS manager =====
-   //================================================
-   //            find input container
-   AliAnalysisTaskGammaConvV1 *task=NULL;
-   task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
-   task->SetIsHeavyIon(2);
-   task->SetIsMC(isMC);
-   // Cut Numbers to use in Analysis
-   Int_t numberOfCuts = 1;
-   TString *cutarray = new TString[numberOfCuts];
-   TString *mesonCutArray = new TString[numberOfCuts];
-   Bool_t doEtaShiftIndCuts = kFALSE;
-   TString stringShift = "";
-   if(trainConfig == 1){
-      // Shifting in pPb direction
-      cutarray[ 0] = "8000011002093172003290000000"; mesonCutArray[ 0] = "01621035009000"; 
-   } else if (trainConfig == 2) {
-      cutarray[ 0] = "8000011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";
-   } else if (trainConfig == 3) {   
-      cutarray[ 0] = "8020011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";      
-   } else if (trainConfig == 4) {   
-      cutarray[ 0] = "8240011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
-   } else if (trainConfig == 5) {   
-      cutarray[ 0] = "8460011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
-   } else if (trainConfig == 6) {   
-      cutarray[ 0] = "8680011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";      
-   } else if (trainConfig == 7) {   
-      cutarray[ 0] = "8600011002092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
-   } else {
-      Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
-      return;
-   }
-   TList *ConvCutList = new TList();
-   TList *MesonCutList = new TList();
-   TList *HeaderList = new TList();
-   TObjString *Header1 = new TObjString("pi0_1");
-   HeaderList->Add(Header1);
-   TObjString *Header3 = new TObjString("eta_2");
-   HeaderList->Add(Header3);
-   
-   ConvCutList->SetOwner(kTRUE);
-   AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts];
-   MesonCutList->SetOwner(kTRUE);
-   AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];
-   for(Int_t i = 0; i<numberOfCuts; i++){
-     
-      analysisCuts[i] = new AliConversionCuts();   
-      analysisCuts[i]->InitializeCutsFromCutString(cutarray[i].Data());
-      if (doEtaShiftIndCuts) {
-         analysisCuts[i]->DoEtaShift(doEtaShiftIndCuts);
-         analysisCuts[i]->SetEtaShift(stringShift);
-      }
-     
-      ConvCutList->Add(analysisCuts[i]);
-         
-      analysisCuts[i]->SetFillCutHistograms("",kFALSE);
-      analysisMesonCuts[i] = new AliConversionMesonCuts();
-      analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
-      MesonCutList->Add(analysisMesonCuts[i]);
-      analysisMesonCuts[i]->SetFillCutHistograms("");
-      analysisCuts[i]->SetAcceptedHeader(HeaderList);
-   }
-   task->SetConversionCutList(numberOfCuts,ConvCutList);
-   task->SetMesonCutList(numberOfCuts,MesonCutList);
-   task->SetMoveParticleAccordingToVertex(kTRUE);
-   task->SetDoMesonAnalysis(kTRUE);
-   task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
-   task->SetDoPhotonQA(enableQAMesonTask);  //Attention new switch small for Photon QA
-   //connect containers
-   AliAnalysisDataContainer *coutput =
-      mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(),
-                           AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig));
-   mgr->AddTask(task);
-   mgr->ConnectInput(task,0,cinput);
-   mgr->ConnectOutput(task,1,coutput);
-   return;
+       Int_t isHeavyIon = 2;
+               
+       // ================== GetAnalysisManager ===============================
+       AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+       if (!mgr) {
+               Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
+               return ;
+       }
+       
+       // ================== GetInputEventHandler =============================
+       AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
+       
+       //========= Add PID Reponse to ANALYSIS manager ====
+       if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
+               gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
+               AddTaskPIDResponse(isMC);
+       }
+       
+       //=========  Set Cutnumber for V0Reader ================================
+       TString cutnumberPhoton = "060000084001001500000000";
+       TString cutnumberEvent = "8000000";
+       Bool_t doEtaShift = kFALSE;
+       AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
+       //========= Add V0 Reader to  ANALYSIS manager if not yet existent =====
+       if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
+               AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
+               
+               fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
+               fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
+               fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
+               
+               if (!mgr) {
+                       Error("AddTask_V0ReaderV1", "No analysis manager found.");
+                       return;
+               }
+       
+               AliConvEventCuts *fEventCuts=NULL;
+               if(cutnumberEvent!=""){
+                       fEventCuts= new AliConvEventCuts(cutnumberEvent.Data(),cutnumberEvent.Data());
+                       fEventCuts->SetPreSelectionCutFlag(kTRUE);
+                       if(fEventCuts->InitializeCutsFromCutString(cutnumberEvent.Data())){
+                               fEventCuts->DoEtaShift(doEtaShift);
+                               fV0ReaderV1->SetEventCuts(fEventCuts);
+                               fEventCuts->SetFillCutHistograms("",kTRUE);
+                       }
+               }
+
+               // Set AnalysisCut Number
+               AliConversionPhotonCuts *fCuts=NULL;
+               if(cutnumberPhoton!=""){
+                       fCuts= new AliConversionPhotonCuts(cutnumberPhoton.Data(),cutnumberPhoton.Data());
+                       fCuts->SetPreSelectionCutFlag(kTRUE);
+                       fCuts->SetIsHeavyIon(isHeavyIon);
+                       if(fCuts->InitializeCutsFromCutString(cutnumberPhoton.Data())){
+                               fV0ReaderV1->SetConversionCuts(fCuts);
+                               fCuts->SetFillCutHistograms("",kTRUE);
+                       }
+               }
+               if(inputHandler->IsA()==AliAODInputHandler::Class()){
+               // AOD mode
+                       cout << "AOD handler: adding " << cutnumberAODBranch.Data() << " as conversion branch" << endl;
+                       fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
+               }
+               fV0ReaderV1->Init();
+       
+               AliLog::SetGlobalLogLevel(AliLog::kInfo);
+       
+               //connect input V0Reader
+               mgr->AddTask(fV0ReaderV1);
+               mgr->ConnectInput(fV0ReaderV1,0,cinput);
+       
+       }
+       
+       //================================================
+       //========= Add task to the ANALYSIS manager =====
+       //================================================
+       //            find input container
+       AliAnalysisTaskGammaConvV1 *task=NULL;
+       task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
+       task->SetIsHeavyIon(isHeavyIon);
+       task->SetIsMC(isMC);
+       // Cut Numbers to use in Analysis
+       Int_t numberOfCuts = 1;
+       
+       TString *eventCutArray = new TString[numberOfCuts];
+       TString *photonCutArray = new TString[numberOfCuts];
+       TString *mesonCutArray = new TString[numberOfCuts];
+       Bool_t doEtaShiftIndCuts = kFALSE;
+       TString stringShift = "";
+       
+       if(trainConfig == 1){
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "002000093172003290000000"; mesonCutArray[ 0] = "01621035009000"; 
+       } else if (trainConfig == 2) {
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";
+       } else if (trainConfig == 3) {   
+               eventCutArray[ 0] = "8020011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";      
+       } else if (trainConfig == 4) {   
+               eventCutArray[ 0] = "8240011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
+       } else if (trainConfig == 5) {   
+               eventCutArray[ 0] = "8460011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
+       } else if (trainConfig == 6) {   
+               eventCutArray[ 0] = "8680011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";      
+       } else if (trainConfig == 7) {   
+               eventCutArray[ 0] = "8600011"; photonCutArray[ 0] = "002000092170008260400000"; mesonCutArray[ 0] = "01621035009000";         
+       } else if (trainConfig == 8) {   
+           eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "009000092170008260400000"; mesonCutArray[ 0] = "01621035009000";    //RCut 7.5cm   
+       } else if (trainConfig == 9) {   
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "005000092170008260400000"; mesonCutArray[ 0] = "01621035009000";    //RCut 10cm     
+       } else if (trainConfig == 10) {   
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "008000092170008260400000"; mesonCutArray[ 0] = "01621035009000";    //RCut 12.5cm    
+       } else if (trainConfig == 11) {   
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "006000092170008260400000"; mesonCutArray[ 0] = "01621035009000";    //RCut 20cm    
+       } else if (trainConfig == 12) {   
+               eventCutArray[ 0] = "8000011"; photonCutArray[ 0] = "007000092170008260400000"; mesonCutArray[ 0] = "01621035009000";    //RCut 35cm    
+       } else {
+               Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
+               return;
+       }
+       
+       TList *EventCutList = new TList();
+       TList *ConvCutList = new TList();
+       TList *MesonCutList = new TList();
+       
+       TList *HeaderList = new TList();
+       TObjString *Header1 = new TObjString("pi0_1");
+       HeaderList->Add(Header1);
+       TObjString *Header3 = new TObjString("eta_2");
+       HeaderList->Add(Header3);
+       
+       EventCutList->SetOwner(kTRUE);
+       AliConvEventCuts **analysisEventCuts = new AliConvEventCuts*[numberOfCuts];
+       ConvCutList->SetOwner(kTRUE);
+       AliConversionPhotonCuts **analysisCuts = new AliConversionPhotonCuts*[numberOfCuts];
+       MesonCutList->SetOwner(kTRUE);
+       AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];
+       
+       
+       for(Int_t i = 0; i<numberOfCuts; i++){  
+               analysisEventCuts[i] = new AliConvEventCuts();   
+               analysisEventCuts[i]->InitializeCutsFromCutString(eventCutArray[i].Data());
+               if (doEtaShiftIndCuts) {
+                       analysisEventCuts[i]->DoEtaShift(doEtaShiftIndCuts);
+                       analysisEventCuts[i]->SetEtaShift(stringShift);
+               }
+               EventCutList->Add(analysisEventCuts[i]);
+               analysisEventCuts[i]->SetFillCutHistograms("",kFALSE);
+               
+               analysisCuts[i] = new AliConversionPhotonCuts();
+               analysisCuts[i]->InitializeCutsFromCutString(photonCutArray[i].Data());
+               analysisCuts[i]->SetIsHeavyIon(isHeavyIon);
+               ConvCutList->Add(analysisCuts[i]);
+               analysisCuts[i]->SetFillCutHistograms("",kFALSE);
+
+               analysisMesonCuts[i] = new AliConversionMesonCuts();
+               analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
+               MesonCutList->Add(analysisMesonCuts[i]);
+               analysisMesonCuts[i]->SetFillCutHistograms("");
+               analysisEventCuts[i]->SetAcceptedHeader(HeaderList);
+       }
+       
+       task->SetEventCutList(numberOfCuts,EventCutList);
+       task->SetConversionCutList(numberOfCuts,ConvCutList);
+       task->SetMesonCutList(numberOfCuts,MesonCutList);
+       task->SetMoveParticleAccordingToVertex(kTRUE);
+       task->SetDoMesonAnalysis(kTRUE);
+       task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
+       task->SetDoPhotonQA(enableQAPhotonTask);  //Attention new switch small for Photon QA
+       
+       //connect containers
+       AliAnalysisDataContainer *coutput =
+               mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(),
+                                                       AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig));
+       
+       mgr->AddTask(task);
+       mgr->ConnectInput(task,0,cinput);
+       mgr->ConnectOutput(task,1,coutput);
+       
+       return;
  
 }