ROOT logo
AliAnalysisTask *AddTask_mwinn_JPsiMCcontrol(){
  //get the current analysis manager
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    Error("AddTask_mwinn_JPsiMCcontrol", "No analysis manager found.");
    return 0;
  }


  //Do we have an MC handler?
  Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
  
  //Get the current train configuration
  TString trainConfig=gSystem->Getenv("CONFIG_FILE");
  
  //set config file name
  TString configFile("$ALICE_ROOT/PWGDQ/dielectron/macrosJPSI/ConfigJpsi_mw_pPb_MC.C");
  //TString configFile("$TRAIN_ROOT/jpsi_JPSI/ConfigJpsiStandard.C");
  if ( trainConfig.Contains("PbPb") ) configFile="$TRAIN_ROOT/jpsi_JPSI/ConfigJpsi2eePbPb.C";
  
  //if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
  //  ::Info("AddTaskJPSI", "Using AOD configuration");
  //  configFile="$TRAIN_ROOT/util/dielectron/dielectron/macros/ConfigJpsi2eeDataAOD.C";
  // }

  TString list=gSystem->Getenv("LIST");
  //create task and add it to the manager
  AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDieData");
  if (!hasMC&&(!list.Contains("LHC12h")) ) task->UsePhysicsSelection();//taking out for testing
  if (list.Contains("LHC13b"))task->SetTriggerMask(AliVEvent::kINT7); //kINT7?, MB-trigger for MB pPb
  if (list.Contains("LHC13c"))task->SetTriggerMask(AliVEvent::kINT7); //kINT7?, MB-trigger for MB pPb
  if (list.Contains("LHC13d"))task->SetTriggerMask(AliVEvent::kTRD); //TRD trigger
  if (list.Contains("LHC13e"))task->SetTriggerMask(AliVEvent::kTRD); //TRD trigger
  if (list.Contains("LHC11d")) task->SetTriggerMask(AliVEvent::kEMCEJE+AliVEvent::kEMC7+AliVEvent::kEMCEGA);
  //if (list.Contains("LHC12h")) task->SetTRDtrigger(1+2);
  mgr->AddTask(task);

  
  //load dielectron configuration file
  gROOT->LoadMacro(configFile.Data());
  
  //add dielectron analysis with different cuts to the task
  cout << "!!!!!!!!!!!! nDie = " << nDie << endl;
  for (Int_t i=0; i<nDie; ++i){ //nDie defined in config file
    AliDielectron *jpsi=ConfigJpsi_mw_pPb_MC(i);
    if (!jpsi) continue;
    task->AddDielectron(jpsi);
//    printf("add: %s\n",jpsi->GetName());
  }
  
  //Add event filter
  AliDielectronEventCuts *eventCuts=new AliDielectronEventCuts("eventCuts","Vertex Track && |vtxZ|<10 && ncontrib>0");
  eventCuts->SetRequireVertex();//NOTE: all of these cuts can for some reasons not be applied to self-filtered AODs by mwinn in 
  //(list/hera/alice/mwinn/mwinn/train/lists/...)
  eventCuts->SetRequire2013vertexandevent();
  eventCuts->SetMinVtxContributors(1);
  eventCuts->SetVertexZ(-10.,10.);
  //  eventCuts->Print();
//   eventCuts->SetVertexType(AliDielectronEventCuts::kVtxTracksOrSPD);
//   eventCuts->SetRequireV0and();
//   if ( trainConfig=="PbPb" ){
//     eventCuts->SetCutOnMultipicityITSTPC();
//   }
   task->SetEventFilter(eventCuts);

//   task->SetTriggerOnV0AND();
  if ( trainConfig=="pp" ) task->SetRejectPileup();
  
  //create output container
  TString containerName= "JPSI.root";
  AliAnalysisDataContainer *coutput1 =
    mgr->CreateContainer("mwinnMCcontrol_tree",
                         TTree::Class(),
                         AliAnalysisManager::kExchangeContainer,
                         containerName.Data());
  
  AliAnalysisDataContainer *cOutputHist1 =
    mgr->CreateContainer("mwinnMCcontrol_QA",
                         TList::Class(),
                         AliAnalysisManager::kOutputContainer,
                         containerName.Data());

  AliAnalysisDataContainer *cOutputHist2 =
    mgr->CreateContainer("mwinnMCcontrol_CF",
                         TList::Class(),
                         AliAnalysisManager::kOutputContainer,
                         containerName.Data());
  
  AliAnalysisDataContainer *cOutputHist3 =
    mgr->CreateContainer("mwinnMCcontrol_EventStat",
                         TH1D::Class(),
                         AliAnalysisManager::kOutputContainer,
                         containerName.Data());
  
  mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
  mgr->ConnectOutput(task, 0, coutput1 );
  mgr->ConnectOutput(task, 1, cOutputHist1);
  mgr->ConnectOutput(task, 2, cOutputHist2);
  mgr->ConnectOutput(task, 3, cOutputHist3);
  
  return task;
}
 AddTask_mwinn_JPsiMCcontrol.C:1
 AddTask_mwinn_JPsiMCcontrol.C:2
 AddTask_mwinn_JPsiMCcontrol.C:3
 AddTask_mwinn_JPsiMCcontrol.C:4
 AddTask_mwinn_JPsiMCcontrol.C:5
 AddTask_mwinn_JPsiMCcontrol.C:6
 AddTask_mwinn_JPsiMCcontrol.C:7
 AddTask_mwinn_JPsiMCcontrol.C:8
 AddTask_mwinn_JPsiMCcontrol.C:9
 AddTask_mwinn_JPsiMCcontrol.C:10
 AddTask_mwinn_JPsiMCcontrol.C:11
 AddTask_mwinn_JPsiMCcontrol.C:12
 AddTask_mwinn_JPsiMCcontrol.C:13
 AddTask_mwinn_JPsiMCcontrol.C:14
 AddTask_mwinn_JPsiMCcontrol.C:15
 AddTask_mwinn_JPsiMCcontrol.C:16
 AddTask_mwinn_JPsiMCcontrol.C:17
 AddTask_mwinn_JPsiMCcontrol.C:18
 AddTask_mwinn_JPsiMCcontrol.C:19
 AddTask_mwinn_JPsiMCcontrol.C:20
 AddTask_mwinn_JPsiMCcontrol.C:21
 AddTask_mwinn_JPsiMCcontrol.C:22
 AddTask_mwinn_JPsiMCcontrol.C:23
 AddTask_mwinn_JPsiMCcontrol.C:24
 AddTask_mwinn_JPsiMCcontrol.C:25
 AddTask_mwinn_JPsiMCcontrol.C:26
 AddTask_mwinn_JPsiMCcontrol.C:27
 AddTask_mwinn_JPsiMCcontrol.C:28
 AddTask_mwinn_JPsiMCcontrol.C:29
 AddTask_mwinn_JPsiMCcontrol.C:30
 AddTask_mwinn_JPsiMCcontrol.C:31
 AddTask_mwinn_JPsiMCcontrol.C:32
 AddTask_mwinn_JPsiMCcontrol.C:33
 AddTask_mwinn_JPsiMCcontrol.C:34
 AddTask_mwinn_JPsiMCcontrol.C:35
 AddTask_mwinn_JPsiMCcontrol.C:36
 AddTask_mwinn_JPsiMCcontrol.C:37
 AddTask_mwinn_JPsiMCcontrol.C:38
 AddTask_mwinn_JPsiMCcontrol.C:39
 AddTask_mwinn_JPsiMCcontrol.C:40
 AddTask_mwinn_JPsiMCcontrol.C:41
 AddTask_mwinn_JPsiMCcontrol.C:42
 AddTask_mwinn_JPsiMCcontrol.C:43
 AddTask_mwinn_JPsiMCcontrol.C:44
 AddTask_mwinn_JPsiMCcontrol.C:45
 AddTask_mwinn_JPsiMCcontrol.C:46
 AddTask_mwinn_JPsiMCcontrol.C:47
 AddTask_mwinn_JPsiMCcontrol.C:48
 AddTask_mwinn_JPsiMCcontrol.C:49
 AddTask_mwinn_JPsiMCcontrol.C:50
 AddTask_mwinn_JPsiMCcontrol.C:51
 AddTask_mwinn_JPsiMCcontrol.C:52
 AddTask_mwinn_JPsiMCcontrol.C:53
 AddTask_mwinn_JPsiMCcontrol.C:54
 AddTask_mwinn_JPsiMCcontrol.C:55
 AddTask_mwinn_JPsiMCcontrol.C:56
 AddTask_mwinn_JPsiMCcontrol.C:57
 AddTask_mwinn_JPsiMCcontrol.C:58
 AddTask_mwinn_JPsiMCcontrol.C:59
 AddTask_mwinn_JPsiMCcontrol.C:60
 AddTask_mwinn_JPsiMCcontrol.C:61
 AddTask_mwinn_JPsiMCcontrol.C:62
 AddTask_mwinn_JPsiMCcontrol.C:63
 AddTask_mwinn_JPsiMCcontrol.C:64
 AddTask_mwinn_JPsiMCcontrol.C:65
 AddTask_mwinn_JPsiMCcontrol.C:66
 AddTask_mwinn_JPsiMCcontrol.C:67
 AddTask_mwinn_JPsiMCcontrol.C:68
 AddTask_mwinn_JPsiMCcontrol.C:69
 AddTask_mwinn_JPsiMCcontrol.C:70
 AddTask_mwinn_JPsiMCcontrol.C:71
 AddTask_mwinn_JPsiMCcontrol.C:72
 AddTask_mwinn_JPsiMCcontrol.C:73
 AddTask_mwinn_JPsiMCcontrol.C:74
 AddTask_mwinn_JPsiMCcontrol.C:75
 AddTask_mwinn_JPsiMCcontrol.C:76
 AddTask_mwinn_JPsiMCcontrol.C:77
 AddTask_mwinn_JPsiMCcontrol.C:78
 AddTask_mwinn_JPsiMCcontrol.C:79
 AddTask_mwinn_JPsiMCcontrol.C:80
 AddTask_mwinn_JPsiMCcontrol.C:81
 AddTask_mwinn_JPsiMCcontrol.C:82
 AddTask_mwinn_JPsiMCcontrol.C:83
 AddTask_mwinn_JPsiMCcontrol.C:84
 AddTask_mwinn_JPsiMCcontrol.C:85
 AddTask_mwinn_JPsiMCcontrol.C:86
 AddTask_mwinn_JPsiMCcontrol.C:87
 AddTask_mwinn_JPsiMCcontrol.C:88
 AddTask_mwinn_JPsiMCcontrol.C:89
 AddTask_mwinn_JPsiMCcontrol.C:90
 AddTask_mwinn_JPsiMCcontrol.C:91
 AddTask_mwinn_JPsiMCcontrol.C:92
 AddTask_mwinn_JPsiMCcontrol.C:93
 AddTask_mwinn_JPsiMCcontrol.C:94
 AddTask_mwinn_JPsiMCcontrol.C:95
 AddTask_mwinn_JPsiMCcontrol.C:96
 AddTask_mwinn_JPsiMCcontrol.C:97
 AddTask_mwinn_JPsiMCcontrol.C:98
 AddTask_mwinn_JPsiMCcontrol.C:99
 AddTask_mwinn_JPsiMCcontrol.C:100
 AddTask_mwinn_JPsiMCcontrol.C:101
 AddTask_mwinn_JPsiMCcontrol.C:102
 AddTask_mwinn_JPsiMCcontrol.C:103