ROOT logo
AliAnalysisTask *AddTask_jbook_JPsi(TString config="1",
						TString cfg="ConfigJpsi_jb_PbPb.C",
				    Bool_t gridconf=kFALSE,
				    Bool_t hasMC=kFALSE,
				    ULong64_t triggers=AliVEvent::kCentral | AliVEvent::kSemiCentral | AliVEvent::kMB,
						Bool_t bMultiToSingle=kTRUE){

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

  //Do we have an MC handler?
  TString list = gSystem->Getenv("LIST");
  if(!list.IsNull()) {
    if( list.Contains("LHC10h")   || list.Contains("LHC11h")   ) hasMC=kFALSE;
    if( list.Contains("LHC11a10") || list.Contains("LHC12a17") ) hasMC=kTRUE;
  }

  //Do we have an AOD handler?
  Bool_t isAOD=(mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class() ? kTRUE : kFALSE);

  // set AOD debug levels
  if(isAOD) {
    mgr->AddClassDebug("AliAODTrack", AliLog::kFatal);
    mgr->AddClassDebug("AliAODpidUtil", AliLog::kInfo);
  }

  //set config file name
  TString configFile("");
  printf("%s \n",gSystem->pwd());
  TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
  if(cfg.IsNull()) cfg="ConfigJpsi_jb_PbPb.C";

  // the different paths
  TString gsiPath("$TRAIN_ROOT/jbook_jpsi/");
  TString alienPath("alien:///alice/cern.ch/user/j/jbook/PWGDQ/dielectron/macrosJPSI/");
  TString alirootPath("$ALICE_ROOT/PWGDQ/dielectron/macrosJPSI/");

  ////////// >>>>>>>>>> gsi config
  if (!trainRoot.IsNull())  configFile=gsiPath.Data();
  ////////// >>>>>>>>>> alien config
  else if(!gSystem->Exec(Form("alien_cp %s/%s .",alienPath.Data(),cfg.Data()))) {
    gSystem->Exec(Form("ls -l %s",gSystem->pwd()));
    configFile=gSystem->pwd();
  }
  else {
    printf("ERROR: couldn't copy file %s/%s from grid \n", alienPath.Data(),cfg.Data() );
    return;
  }
  ///////// >>>>>>>>> aliroot config
  if(!gridconf && trainRoot.IsNull()) configFile=alirootPath.Data();
  ///////// add config to path
  configFile+="/";
  configFile+=cfg.Data();

  // trigger selection
  ULong64_t triggerSets[]={AliVEvent::kCentral , AliVEvent::kSemiCentral , AliVEvent::kMB,
			   AliVEvent::kCentral | AliVEvent::kSemiCentral | AliVEvent::kMB};
  const char* triggerNames[]={"Central","SemiCentral","MB","ALL"};
  const char* onlineRejection[]={"","CCENT","",""};

  // find out the configured triggers
  Int_t j=0;
  for(j=0; j<4; j++) {
    if(triggers!=triggerSets[j]) continue;
    else break;
  }

  // print overall configuration
  printf("production: %s MC: %d \n",  list.Data(),hasMC);
  printf("triggers:   %s \n",         triggerNames[j]  );
  printf("config:     %s Grid: %d \n",configFile.Data(),gridconf);

  //create task(s)
  AliAnalysisTaskMultiDielectron *task;
  if(!bMultiToSingle) {
    // create one multi task
    task = new AliAnalysisTaskMultiDielectron(Form("MultiDieJB"));
    task->SetBeamEnergy(1380.);
    task->SetTriggerMask(triggers);
    if(strlen(onlineRejection[j])) task->SetFiredTriggerName(onlineRejection[j],kTRUE);
    if(!hasMC) task->UsePhysicsSelection();
  }

  // event filter
  AliDielectronEventCuts *eventCuts=new AliDielectronEventCuts("vertex","vertex");
  if(isAOD) eventCuts->SetVertexType(AliDielectronEventCuts::kVtxAny);
  eventCuts->SetRequireVertex();
  eventCuts->SetMinVtxContributors(1);
  if(hasMC) eventCuts->SetVertexZ(-10.,+10.); //for data this is done by in the config
  eventCuts->SetCentralityRange(0,90.);
  eventCuts->Print();
  if(!bMultiToSingle)       task->SetEventFilter(eventCuts);

  //load dielectron configuration file (only once)
  TString checkconfig="ConfigJpsi_jb_PbPb";
  if (!gROOT->GetListOfGlobalFunctions()->FindObject(checkconfig.Data()))
    gROOT->LoadMacro(configFile.Data());

  //define default output container
  TString containerName = "JPSI.root";

  //add dielectron analysis with different cuts to the task
  for (Int_t i=0; i<nDie; ++i) { //nDie defined in config file

    //only configs switched ON will pass
    if(config.Length()<=i || config(i,1)!="1") { printf(" %d switched OFF \n",i); continue; }

    // load configuration
    AliDielectron *jpsi=ConfigJpsi_jb_PbPb(i,hasMC,triggers);
    if(!jpsi) continue;

    // create unique title
    TString unitit = Form("%s_%s",triggerNames[j],jpsi->GetName());

    // create single tasks instead of one multi task (decreasing size of CF container)
    if(bMultiToSingle) {
      task = new AliAnalysisTaskMultiDielectron(Form("MultiDieJB_%s",unitit.Data()));
      task->SetBeamEnergy(1380.);
      task->SetTriggerMask(triggers);
      if(strlen(onlineRejection[j])) task->SetFiredTriggerName(onlineRejection[j],kTRUE);
      if(!hasMC) task->UsePhysicsSelection();
    }

    // add dielectron to the task and manager
    task->AddDielectron(jpsi);

    // multiple output connection
    if(bMultiToSingle) {
      task->SetEventFilter(eventCuts);
      mgr->AddTask(task);

      //create output sub containers
      unitit.Prepend("jbook_QA_");
      AliAnalysisDataContainer *cOutputHist1 =
	mgr->CreateContainer(unitit.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,containerName.Data());
      unitit.ReplaceAll("_QA_","_CF_");
      AliAnalysisDataContainer *cOutputHist2 =
	mgr->CreateContainer(unitit.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,containerName.Data());
      unitit.ReplaceAll("_CF_","_EventStat_");
      AliAnalysisDataContainer *cOutputHist3 =
	mgr->CreateContainer(unitit.Data(), 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);
    }

    printf(" %s added\n",jpsi->GetName());

  } //end : loop over configs

  
  // multiple output connection
  if(!bMultiToSingle) {
    mgr->AddTask(task);
    
    //create output sub containers
    AliAnalysisDataContainer *cOutputHist1 =
      mgr->CreateContainer("jbook_QA", TList::Class(), AliAnalysisManager::kOutputContainer, containerName.Data());
    AliAnalysisDataContainer *cOutputHist2 =
      mgr->CreateContainer("jbook_CF", TList::Class(), AliAnalysisManager::kOutputContainer, containerName.Data());
    AliAnalysisDataContainer *cOutputHist3 =
      mgr->CreateContainer("jbook_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_jbook_JPsi.C:1
 AddTask_jbook_JPsi.C:2
 AddTask_jbook_JPsi.C:3
 AddTask_jbook_JPsi.C:4
 AddTask_jbook_JPsi.C:5
 AddTask_jbook_JPsi.C:6
 AddTask_jbook_JPsi.C:7
 AddTask_jbook_JPsi.C:8
 AddTask_jbook_JPsi.C:9
 AddTask_jbook_JPsi.C:10
 AddTask_jbook_JPsi.C:11
 AddTask_jbook_JPsi.C:12
 AddTask_jbook_JPsi.C:13
 AddTask_jbook_JPsi.C:14
 AddTask_jbook_JPsi.C:15
 AddTask_jbook_JPsi.C:16
 AddTask_jbook_JPsi.C:17
 AddTask_jbook_JPsi.C:18
 AddTask_jbook_JPsi.C:19
 AddTask_jbook_JPsi.C:20
 AddTask_jbook_JPsi.C:21
 AddTask_jbook_JPsi.C:22
 AddTask_jbook_JPsi.C:23
 AddTask_jbook_JPsi.C:24
 AddTask_jbook_JPsi.C:25
 AddTask_jbook_JPsi.C:26
 AddTask_jbook_JPsi.C:27
 AddTask_jbook_JPsi.C:28
 AddTask_jbook_JPsi.C:29
 AddTask_jbook_JPsi.C:30
 AddTask_jbook_JPsi.C:31
 AddTask_jbook_JPsi.C:32
 AddTask_jbook_JPsi.C:33
 AddTask_jbook_JPsi.C:34
 AddTask_jbook_JPsi.C:35
 AddTask_jbook_JPsi.C:36
 AddTask_jbook_JPsi.C:37
 AddTask_jbook_JPsi.C:38
 AddTask_jbook_JPsi.C:39
 AddTask_jbook_JPsi.C:40
 AddTask_jbook_JPsi.C:41
 AddTask_jbook_JPsi.C:42
 AddTask_jbook_JPsi.C:43
 AddTask_jbook_JPsi.C:44
 AddTask_jbook_JPsi.C:45
 AddTask_jbook_JPsi.C:46
 AddTask_jbook_JPsi.C:47
 AddTask_jbook_JPsi.C:48
 AddTask_jbook_JPsi.C:49
 AddTask_jbook_JPsi.C:50
 AddTask_jbook_JPsi.C:51
 AddTask_jbook_JPsi.C:52
 AddTask_jbook_JPsi.C:53
 AddTask_jbook_JPsi.C:54
 AddTask_jbook_JPsi.C:55
 AddTask_jbook_JPsi.C:56
 AddTask_jbook_JPsi.C:57
 AddTask_jbook_JPsi.C:58
 AddTask_jbook_JPsi.C:59
 AddTask_jbook_JPsi.C:60
 AddTask_jbook_JPsi.C:61
 AddTask_jbook_JPsi.C:62
 AddTask_jbook_JPsi.C:63
 AddTask_jbook_JPsi.C:64
 AddTask_jbook_JPsi.C:65
 AddTask_jbook_JPsi.C:66
 AddTask_jbook_JPsi.C:67
 AddTask_jbook_JPsi.C:68
 AddTask_jbook_JPsi.C:69
 AddTask_jbook_JPsi.C:70
 AddTask_jbook_JPsi.C:71
 AddTask_jbook_JPsi.C:72
 AddTask_jbook_JPsi.C:73
 AddTask_jbook_JPsi.C:74
 AddTask_jbook_JPsi.C:75
 AddTask_jbook_JPsi.C:76
 AddTask_jbook_JPsi.C:77
 AddTask_jbook_JPsi.C:78
 AddTask_jbook_JPsi.C:79
 AddTask_jbook_JPsi.C:80
 AddTask_jbook_JPsi.C:81
 AddTask_jbook_JPsi.C:82
 AddTask_jbook_JPsi.C:83
 AddTask_jbook_JPsi.C:84
 AddTask_jbook_JPsi.C:85
 AddTask_jbook_JPsi.C:86
 AddTask_jbook_JPsi.C:87
 AddTask_jbook_JPsi.C:88
 AddTask_jbook_JPsi.C:89
 AddTask_jbook_JPsi.C:90
 AddTask_jbook_JPsi.C:91
 AddTask_jbook_JPsi.C:92
 AddTask_jbook_JPsi.C:93
 AddTask_jbook_JPsi.C:94
 AddTask_jbook_JPsi.C:95
 AddTask_jbook_JPsi.C:96
 AddTask_jbook_JPsi.C:97
 AddTask_jbook_JPsi.C:98
 AddTask_jbook_JPsi.C:99
 AddTask_jbook_JPsi.C:100
 AddTask_jbook_JPsi.C:101
 AddTask_jbook_JPsi.C:102
 AddTask_jbook_JPsi.C:103
 AddTask_jbook_JPsi.C:104
 AddTask_jbook_JPsi.C:105
 AddTask_jbook_JPsi.C:106
 AddTask_jbook_JPsi.C:107
 AddTask_jbook_JPsi.C:108
 AddTask_jbook_JPsi.C:109
 AddTask_jbook_JPsi.C:110
 AddTask_jbook_JPsi.C:111
 AddTask_jbook_JPsi.C:112
 AddTask_jbook_JPsi.C:113
 AddTask_jbook_JPsi.C:114
 AddTask_jbook_JPsi.C:115
 AddTask_jbook_JPsi.C:116
 AddTask_jbook_JPsi.C:117
 AddTask_jbook_JPsi.C:118
 AddTask_jbook_JPsi.C:119
 AddTask_jbook_JPsi.C:120
 AddTask_jbook_JPsi.C:121
 AddTask_jbook_JPsi.C:122
 AddTask_jbook_JPsi.C:123
 AddTask_jbook_JPsi.C:124
 AddTask_jbook_JPsi.C:125
 AddTask_jbook_JPsi.C:126
 AddTask_jbook_JPsi.C:127
 AddTask_jbook_JPsi.C:128
 AddTask_jbook_JPsi.C:129
 AddTask_jbook_JPsi.C:130
 AddTask_jbook_JPsi.C:131
 AddTask_jbook_JPsi.C:132
 AddTask_jbook_JPsi.C:133
 AddTask_jbook_JPsi.C:134
 AddTask_jbook_JPsi.C:135
 AddTask_jbook_JPsi.C:136
 AddTask_jbook_JPsi.C:137
 AddTask_jbook_JPsi.C:138
 AddTask_jbook_JPsi.C:139
 AddTask_jbook_JPsi.C:140
 AddTask_jbook_JPsi.C:141
 AddTask_jbook_JPsi.C:142
 AddTask_jbook_JPsi.C:143
 AddTask_jbook_JPsi.C:144
 AddTask_jbook_JPsi.C:145
 AddTask_jbook_JPsi.C:146
 AddTask_jbook_JPsi.C:147
 AddTask_jbook_JPsi.C:148
 AddTask_jbook_JPsi.C:149
 AddTask_jbook_JPsi.C:150
 AddTask_jbook_JPsi.C:151
 AddTask_jbook_JPsi.C:152
 AddTask_jbook_JPsi.C:153
 AddTask_jbook_JPsi.C:154
 AddTask_jbook_JPsi.C:155
 AddTask_jbook_JPsi.C:156
 AddTask_jbook_JPsi.C:157
 AddTask_jbook_JPsi.C:158
 AddTask_jbook_JPsi.C:159
 AddTask_jbook_JPsi.C:160
 AddTask_jbook_JPsi.C:161
 AddTask_jbook_JPsi.C:162
 AddTask_jbook_JPsi.C:163
 AddTask_jbook_JPsi.C:164
 AddTask_jbook_JPsi.C:165
 AddTask_jbook_JPsi.C:166
 AddTask_jbook_JPsi.C:167
 AddTask_jbook_JPsi.C:168
 AddTask_jbook_JPsi.C:169
 AddTask_jbook_JPsi.C:170
 AddTask_jbook_JPsi.C:171
 AddTask_jbook_JPsi.C:172
 AddTask_jbook_JPsi.C:173
 AddTask_jbook_JPsi.C:174
 AddTask_jbook_JPsi.C:175
 AddTask_jbook_JPsi.C:176
 AddTask_jbook_JPsi.C:177
 AddTask_jbook_JPsi.C:178
 AddTask_jbook_JPsi.C:179
 AddTask_jbook_JPsi.C:180
 AddTask_jbook_JPsi.C:181