AliEmcalCorrectionCellBadChannel.cxx

Go to the documentation of this file.

// AliEmcalCorrectionCellBadChannel
//

#include <TObjArray.h>
#include <TFile.h>
#include "AliEMCALGeometry.h"
#include "AliEMCALRecoUtils.h"
#include "AliAODEvent.h"

#include "AliEmcalCorrectionCellBadChannel.h"

ClassImp(AliEmcalCorrectionCellBadChannel);

// Actually registers the class with the base class
RegisterCorrectionComponent<AliEmcalCorrectionCellBadChannel> AliEmcalCorrectionCellBadChannel::reg("AliEmcalCorrectionCellBadChannel");

AliEmcalCorrectionCellBadChannel::AliEmcalCorrectionCellBadChannel() :
  AliEmcalCorrectionComponent("AliEmcalCorrectionCellBadChannel")
  ,fCellEnergyDistBefore(0)
  ,fCellEnergyDistAfter(0)
{
}

AliEmcalCorrectionCellBadChannel::~AliEmcalCorrectionCellBadChannel()
{
}

Bool_t AliEmcalCorrectionCellBadChannel::Initialize()
{
  // Initialization
  AliEmcalCorrectionComponent::Initialize();
  
  AliWarning("Init EMCAL cell bad channel removal");
  
  // init reco utils
  if (!fRecoUtils)
    fRecoUtils  = new AliEMCALRecoUtils;

  fRecoUtils->SetPositionAlgorithm(AliEMCALRecoUtils::kPosTowerGlobal);

  TString customBCmapPath = "";
  GetProperty("customBadChannelFilePath", customBCmapPath);
  if (customBCmapPath!="")
    AliEmcalCorrectionComponent::SetCustomBadChannels(customBCmapPath);

  Bool_t dead = kFALSE;
  GetProperty("acceptDead", dead);
  if ( dead ) fRecoUtils->SetDeadChannelAsGood();

  Bool_t hot = kFALSE;
  GetProperty("acceptHot", hot);
  if ( hot ) fRecoUtils->SetHotChannelAsGood();  
  
  Bool_t warm = kFALSE;
  GetProperty("acceptWarm", warm);
  if ( warm ) fRecoUtils->SetWarmChannelAsGood();
  
  return kTRUE;
}

void AliEmcalCorrectionCellBadChannel::UserCreateOutputObjects()
{
  AliEmcalCorrectionComponent::UserCreateOutputObjects();

  if (fCreateHisto){
    fCellEnergyDistBefore = new TH1F("hCellEnergyDistBefore","hCellEnergyDistBefore;E_{cell} (GeV)",1000,0,10);
    fOutput->Add(fCellEnergyDistBefore);
    fCellEnergyDistAfter = new TH1F("hCellEnergyDistAfter","hCellEnergyDistAfter;E_{cell} (GeV)",1000,0,10);
    fOutput->Add(fCellEnergyDistAfter);
  }
}

Bool_t AliEmcalCorrectionCellBadChannel::Run()
{
  AliEmcalCorrectionComponent::Run();
  
  if (!fEventManager.InputEvent()) {
    AliError("Event ptr = 0, returning");
    return kFALSE;
  }
  
  CheckIfRunChanged();
  
  // CONFIGURE THE RECO UTILS -------------------------------------------------

  fRecoUtils->SwitchOnBadChannelsRemoval();
  
  // START PROCESSING ---------------------------------------------------------
  // Test if cells present
  if (fCaloCells->GetNumberOfCells()<=0)
  {
    AliWarning(Form("Number of EMCAL cells = %d, returning", fCaloCells->GetNumberOfCells()));
    return kFALSE;
  }
  
  // mark the cells not recalibrated
  fRecoUtils->ResetCellsCalibrated();

  if(fCreateHisto)
    FillCellQA(fCellEnergyDistBefore); // "before" QA
  
  // CELL RECALIBRATION -------------------------------------------------------
  // update cell objects
  UpdateCells();
  
  if(fCreateHisto)
    FillCellQA(fCellEnergyDistAfter); // "after" QA

  return kTRUE;
}

Bool_t AliEmcalCorrectionCellBadChannel::CheckIfRunChanged()
{
  Bool_t runChanged = AliEmcalCorrectionComponent::CheckIfRunChanged();
  
  if (runChanged) {
    // init bad channels
    Int_t fInitBC = InitBadChannels();
    if (fInitBC==0) {
      AliError("InitBadChannels returned false, returning");
    }
    if (fInitBC==1) {
      AliWarning("InitBadChannels OK");
    }
    if (fInitBC>1) {
      AliWarning(Form("No external hot channel set: %d - %s", fEventManager.InputEvent()->GetRunNumber(), fFilepass.Data()));
    }
  }
  return runChanged;
}