TRD/Macros/AddTaskTRDCalib.C

   1 //=============================================================================
   2 //
   3 // *** AddTaskTRDCalib
   4 //
   5 // This macros setup the TRD calibration task
   6 //
   7 //=============================================================================
   8
   9 const AliTRDCalDet *GetCalDet(Int_t runNumber);
  10
  11 AliAnalysisTask  *AddTaskTRDCalib(Int_t runNumber)
  12 {
  13   gSystem->Load("libTRDcalib");
  14   // pointer to the analysis manager
  15   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  16   if (!mgr) {
  17     Error("AddTaskTRDCalib", "No analysis manager to connect to.");
  18     return NULL;
  19   }
  20
  21   // check the input handler
  22   if (!mgr->GetInputEventHandler()) {
  23     ::Error("AddTask", "This task requires an input event handler");
  24     return NULL;
  25   }
  26
  27   /////////////////////////
  28   // The TRD calib Task
  29   /////////////////////////
  30   AliTRDCalibTask *calibTask = new AliTRDCalibTask();
  31   calibTask->SetHisto2d(kTRUE);
  32   calibTask->SetVector2d(kTRUE);
  33   calibTask->SetVdriftLinear(kTRUE);
  34   calibTask->SetNz(0,0);
  35   calibTask->SetNrphi(0,0);
  36   calibTask->SetNz(0,1);
  37   calibTask->SetNrphi(0,1);
  38   calibTask->SetNz(0,2);
  39   calibTask->SetNrphi(0,2);
  40   calibTask->SetLow(0);
  41   calibTask->SetHigh(30);
  42   calibTask->SetFillZero(kFALSE);
  43   //calibTask->SetDebug(3);
  44   calibTask->SetNbTimeBins(30);
  45   //calibTask->SetMaxEvent(20);
  46   calibTask->SetRequirePrimaryVertex(kTRUE);
  47   calibTask->SetMinNbOfContributors(1);
  48   calibTask->SetMaxCluster(100.0);
  49   calibTask->SetNbMaxCluster(2);
  50   calibTask->SetCalDetGain(GetCalDet(runNumber));
  51
  52   /////////////////////////////
  53   // Track cuts
  54   /////////////////////////////
  55   AliESDtrackCuts *trackCuts = new AliESDtrackCuts("trackcuts","trackcuts");
  56   trackCuts->SetMinNClustersTPC(50);
  57   trackCuts->SetMaxChi2PerClusterTPC(3.5);
  58   //trackCuts->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);
  59   trackCuts->SetRequireTPCRefit(kTRUE);
  60   //trackCuts->SetRequireITSRefit(kTRUE);
  61   //trackCuts->SetMinNsigmaToVertex(10);
  62   trackCuts->SetRequireSigmaToVertex(kFALSE);
  63   trackCuts->SetAcceptKinkDaughters(kFALSE);
  64   trackCuts->SetMaxDCAToVertexZ(30.0);
  65   trackCuts->SetMaxDCAToVertexXY(3.0);
  66   trackCuts->SetDCAToVertex2D(kFALSE);
  67
  68   calibTask->SetESDtrackCuts(trackCuts);
  69
  70   mgr->AddTask(calibTask);
  71
  72   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
  73
  74   if (!cinput) cinput = mgr->CreateContainer("cchain",TChain::Class(),
  75                                       AliAnalysisManager::kInputContainer);
  76
  77   AliAnalysisDataContainer *coutput =mgr->CreateContainer("TRDCalib",TList::Class(), AliAnalysisManager::kOutputContainer, "AliESDfriends_v1.root");
  78
  79
  80   mgr->ConnectInput(calibTask,0,cinput);
  81   mgr->ConnectOutput(calibTask,0,coutput);
  82   return calibTask;
  83
  84 }
  85
  86 const AliTRDCalDet *GetCalDet(Int_t runNumber){
  87   //
  88   // Get Cal Det for gas gain
  89   //
  90
  91   AliCDBEntry *entry = AliCDBManager::Instance()->Get("TRD/Calib/ChamberGainFactor",runNumber);
  92   if(!entry) {
  93     printf("Found no entry\n");
  94     return 0x0;
  95   }
  96   const AliTRDCalDet* calDet = (AliTRDCalDet *)entry->GetObject();
  97
  98   return calDet;
  99
 100 }