AliEmcalCorrectionClusterEnergyVariation.cxx

Go to the documentation of this file.

// AliEmcalCorrectionClusterEnergyVariation
//

#include "AliEmcalCorrectionClusterEnergyVariation.h"

#include "AliClusterContainer.h"

ClassImp(AliEmcalCorrectionClusterEnergyVariation);

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

AliEmcalCorrectionClusterEnergyVariation::AliEmcalCorrectionClusterEnergyVariation() :
  AliEmcalCorrectionComponent("AliEmcalCorrectionClusterEnergyVariation"),
  fEnergyScaleShift(0.),
  fSmearingWidth(0.),
  fRandom(0)
{
}

AliEmcalCorrectionClusterEnergyVariation::~AliEmcalCorrectionClusterEnergyVariation()
{
}

Bool_t AliEmcalCorrectionClusterEnergyVariation::Initialize()
{
  // Initialization
  AliEmcalCorrectionComponent::Initialize();
  
  GetProperty("energyScaleShift", fEnergyScaleShift);
  GetProperty("smearingWidth", fSmearingWidth);

  return kTRUE;
}

void AliEmcalCorrectionClusterEnergyVariation::UserCreateOutputObjects()
{   
  AliEmcalCorrectionComponent::UserCreateOutputObjects();
}

void AliEmcalCorrectionClusterEnergyVariation::ExecOnce()
{
  fRandom.SetSeed(0);
}

Bool_t AliEmcalCorrectionClusterEnergyVariation::Run()
{
  AliEmcalCorrectionComponent::Run();
  
  // Loop over all clusters
  AliTLorentzVector clusFourVec;
  AliVCluster *cluster = 0;
  AliClusterContainer * clusCont = 0;
  TIter nextClusCont(&fClusterCollArray);
  while ((clusCont = static_cast<AliClusterContainer*>(nextClusCont()))) {
    auto clusItCont = clusCont->all_momentum();
    for (AliClusterIterableMomentumContainer::iterator clusIterator = clusItCont.begin(); clusIterator != clusItCont.end(); ++clusIterator) {
      
      // Get the cluster energy (using the default energy type set in the cluster container)
      clusFourVec.Clear();
      clusFourVec = clusIterator->first;
      Double_t energyclus = clusFourVec.E();

      // Scale energy (if applicable)
      if (TMath::Abs(fEnergyScaleShift) > 1e-6) {
        energyclus *= (1 + fEnergyScaleShift);
      }
      
      // Smear energy (if applicable)
      if (fSmearingWidth > 1e-6) {
        energyclus *= (1 + fRandom.Gaus(0, fSmearingWidth) );
      }
      
      // Set the shifted/smeared energy to the user-defined energy field
      cluster = static_cast<AliVCluster *>(clusIterator->second);
      if (cluster->IsEMCAL() && energyclus > 0) {
        cluster->SetUserDefEnergy(AliVCluster::kUserDefEnergy1, energyclus);
      }
      
    }
  }

  return kTRUE;
}