- Reshuffling of the particle codes in AliPID. Now the light nuclei are between the

[u/mrichter/AliRoot.git] / ANALYSIS / TenderSupplies / AddTaskTender.C

AliAnalysisTask *AddTaskTender(Bool_t useV0=kFALSE, 
                               Bool_t useTPC=kTRUE,
                               Bool_t useTOF=kTRUE,
                               Bool_t useTRD=kTRUE,
                               Bool_t usePID=kTRUE,
                               Bool_t useVTX=kTRUE,
                               Bool_t useT0=kTRUE,
                               Bool_t useEmc=kFALSE,
                               Bool_t usePtFix=kFALSE)
{
  if (!(useV0 | useTPC | useTOF | useTRD | usePID | useVTX | | useT0 | useEmc | usePtFix)) {
     ::Error("AddTaskTender", "No supply added to tender, so tender not created");
     return 0;
  }   
  //get the current analysis manager
  Bool_t checkEvtSelection = useV0;
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    ::Error("AddTaskTender", "No analysis manager found.");
    return 0;
  }
  // currently don't accept AOD input
  if (!mgr->GetInputEventHandler()->InheritsFrom(AliESDInputHandler::Class())) {
    ::Error("AddTask_tender_Tender","The analysis tender only works with ESD input!");
    return 0;
  }
  
  //========= Add tender to the ANALYSIS manager and set default storage =====
  AliTender *tender=new AliTender("AnalysisTender");
  tender->SetCheckEventSelection(checkEvtSelection);
  tender->SetDefaultCDBStorage("raw://");
  mgr->AddTask(tender);
  
  Bool_t cachePID=kFALSE;

  //check that that tender is the first task after the pid response
  TString firstName(mgr->GetTasks()->First()->GetName());
  Bool_t isSecond=(mgr->GetTasks()->At(1) == (TObject*)tender);

  if (! (firstName=="PIDResponseTask" && isSecond )){
    Fatal("AddTaskTender","When using the tender the first task needs to be the PIDResponse and the tender the second task!!!");
    return NULL;
  }
  
  AliAnalysisTaskPIDResponse *pidResp=(AliAnalysisTaskPIDResponse*)mgr->GetTasks()->First();
  if (pidResp->GetCachePID()){
    usePID=kTRUE;
    pidResp->SetCachePID(kFALSE);
    cachePID=kTRUE;
  }
  
  //========= Attach VZERO supply ======

  if (useV0) {
     AliVZEROTenderSupply *vzeroSupply=new AliVZEROTenderSupply("VZEROtender");
     vzeroSupply->SetDebug(kFALSE);
     tender->AddSupply(vzeroSupply);
  }   

 
  //========= Attach TPC supply ======
  if (useTPC) {
     AliTPCTenderSupply *tpcSupply=new AliTPCTenderSupply("TPCtender");
     tpcSupply->SetDebugLevel(2);
     //tpcSupply->SetMip(50.);
     tender->AddSupply(tpcSupply);
  }   

  //========= Attach track 1/pt correction supply ======
  if (usePtFix) {
     AliTrackFixTenderSupply *trfixSupply=new AliTrackFixTenderSupply("PTInvFix");
     //trfixSupply->SetDebugLevel(2);
     tender->AddSupply(trfixSupply);
  }   

  //========= Attach T0 supply ======
  if (useT0) {
    AliT0TenderSupply *t0Tender = new AliT0TenderSupply("T0tender");
    t0Tender ->SetPass4LHC11aCorrection(kTRUE);
    tender->AddSupply(t0Tender);
  }   

  //========= Attach TOF supply ======
  if (useTOF) {
    AliTOFTenderSupply *tofTender = new AliTOFTenderSupply("TOFtender");
    tender->AddSupply(tofTender);
 }   

  //========= Attach TRD supply ======
  if (useTRD) {
    AliTRDTenderSupply *trdSupply=new AliTRDTenderSupply("TRDtender");
    trdSupply->SetLoadReferencesFromCDB();
    // Mask Bad chambers
    trdSupply->AddBadChamber(265);      // low drift
    trdSupply->AddBadChamber(50);
    trdSupply->AddBadChamber(524);      // low drift
    trdSupply->AddBadChamber(32);       // intermediate gain
    trdSupply->AddBadChamber(15);
    trdSupply->AddBadChamber(231);      // low gain
    trdSupply->AddBadChamber(273);      // intermediate gain
    trdSupply->AddBadChamber(532);
    trdSupply->AddBadChamber(5);        // low drift
    trdSupply->AddBadChamber(227);
    trdSupply->AddBadChamber(287);      // low drift
    trdSupply->AddBadChamber(212);      // intermediate gain
    trdSupply->AddBadChamber(228);      // low gain
    trdSupply->AddBadChamber(52);       // low gain
    trdSupply->AddBadChamber(169);      // low drift
    trdSupply->AddBadChamber(236);      // low drift

    trdSupply->SetPIDmethod(AliTRDTenderSupply::k1DLQpid);
    trdSupply->SetNormalizationFactor(1./7.603);
    tender->AddSupply(trdSupply);
  }  

  //========= Attach Primary Vertex supply ======
  if (useVTX) {
    tender->AddSupply(new AliVtxTenderSupply("PriVtxtender"));
  }  

  //========= Attach EMCAL supply ======
  if (useEmc) {
    AliEMCALTenderSupply *emcSupply = new AliEMCALTenderSupply("EmcTender");
    emcSupply->SetDefaults();
    tender->AddSupply(emcSupply);
  }  

  //========= Attach PID supply ======
  if (usePID) {
    AliPIDTenderSupply *pidSupply=new AliPIDTenderSupply("PIDtender");
    pidSupply->SetCachePID(cachePID);
    tender->AddSupply(pidSupply);
  }

  //================================================
  //              data containers
  //================================================

    //            define output containers, please use 'username'_'somename'
  AliAnalysisDataContainer *coutput1 =
      mgr->CreateContainer("tender_event", AliESDEvent::Class(),
                           AliAnalysisManager::kExchangeContainer,"default_tender");
 
  //           connect containers
  mgr->ConnectInput  (tender,  0, mgr->GetCommonInputContainer() );
  mgr->ConnectOutput (tender,  1, coutput1);
 
  return tender;
}