Fix PWG3->PWGHF

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

AliAnalysisTask *AddTaskTender(Bool_t useV0=kFALSE){
  //get the current analysis manager
  Bool_t checkEvtSelection = useV0;
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    Error("AddTask_tender_Tender", "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://");
  if (checkEvtSelection) {
    if (mgr->GetTasks()->First() != (TObject*)tender ) {
      TString firstName(mgr->GetTasks()->First()->GetName());
      Bool_t isSecond=(mgr->GetTasks()->At(1) == (TObject*)tender);
      if (! (firstName=="PIDResponseTask" && isSecond )){
        ::Error("When setting the tender to check the event selection, it has to be the first wagon, or the first after the Physics selection ! Aborting.");
        return NULL;
      }
    }
  }
  mgr->AddTask(tender);
  
  //========= Attach VZERO supply ======

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

 
  //========= Attach TPC supply ======
  AliTPCTenderSupply *tpcSupply=new AliTPCTenderSupply("TPCtender");
  tpcSupply->SetDebugLevel(2);
  //tpcSupply->SetMip(50.);
  tender->AddSupply(tpcSupply);
  //========= Attach TOF supply ======
  AliTOFTenderSupply *tofTender = new AliTOFTenderSupply("TOFtender");
  tender->AddSupply(tofTender);

  //========= Attach TRD supply ======
  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 PID supply ======
  tender->AddSupply(new AliPIDTenderSupply("PIDtender"));

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

  //================================================
  //              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;
}