AliEmcalCorrectionComponent.cxx

Go to the documentation of this file.

// AliEmcalCorrectionComponent
//

#include "AliEmcalCorrectionComponent.h"

#include <TFile.h>
#include <TH1.h>

#include "AliEmcalList.h"
#include "AliEMCALRecoUtils.h"
#include "AliAnalysisManager.h"
#include "AliVEvent.h"
#include "AliClusterContainer.h"
#include "AliTrackContainer.h"
#include "AliParticleContainer.h"
#include "AliMCParticleContainer.h"

ClassImp(AliEmcalCorrectionComponent);

// Must create an instance of the map, since it is static
AliEmcalCorrectionComponentFactory::map_type * AliEmcalCorrectionComponentFactory::componentMap = new AliEmcalCorrectionComponentFactory::map_type;

AliEmcalCorrectionComponent::AliEmcalCorrectionComponent() :
  TNamed("AliEmcalCorrectionComponent", "AliEmcalCorrectionComponent"),
  fUserConfiguration(),
  fDefaultConfiguration(),
  fCreateHisto(kTRUE),
  fRun(-1),
  fFilepass(""),
  fGetPassFromFileName(kTRUE),
  fEvent(0),
  fEsdMode(0),
  fMCEvent(0),
  fCent(0),
  fNcentBins(4),
  fCentBin(0),
  fNbins(250),
  fMinBinPt(0),
  fMaxBinPt(250),
  fGeom(0),
  fIsEmbedded(kFALSE),
  fMinMCLabel(0),
  fClusCont(0),
  fPartCont(0),
  fCaloCells(0),
  fRecoUtils(0),
  fOutput(0),
  fBasePath("")

{
  // Default constructor
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  fVertex[0] = 0;
  fVertex[1] = 0;
  fVertex[2] = 0;
}

AliEmcalCorrectionComponent::AliEmcalCorrectionComponent(const char * name) :
  TNamed(name, name),
  fUserConfiguration(),
  fDefaultConfiguration(),
  fCreateHisto(kTRUE),
  fRun(-1),
  fFilepass(""),
  fGetPassFromFileName(kTRUE),
  fEvent(0),
  fEsdMode(0),
  fMCEvent(0),
  fCent(0),
  fNcentBins(4),
  fCentBin(0),
  fNbins(250),
  fMinBinPt(0),
  fMaxBinPt(250),
  fGeom(0),
  fIsEmbedded(kFALSE),
  fMinMCLabel(0),
  fClusCont(0),
  fPartCont(0),
  fCaloCells(0),
  fRecoUtils(0),
  fOutput(0),
  fBasePath("")

{
  // Standard constructor
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  fVertex[0] = 0;
  fVertex[1] = 0;
  fVertex[2] = 0;
}

AliEmcalCorrectionComponent::~AliEmcalCorrectionComponent()
{
}

Bool_t AliEmcalCorrectionComponent::Initialize()
{
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  // Read in pass. If it is empty, set flag to automatically find the pass from the filename.
  std::string tempString = "";
  GetProperty("pass", tempString);
  fFilepass = tempString.c_str();
  if(fFilepass != "")
    fGetPassFromFileName = kFALSE;

  return kTRUE;
}

void AliEmcalCorrectionComponent::UserCreateOutputObjects()
{
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  // Setup Output
  fOutput = new AliEmcalList();
  fOutput->SetOwner();
}

void AliEmcalCorrectionComponent::ExecOnce()
{
  // Initialize using ExecOnce()
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
}

Bool_t AliEmcalCorrectionComponent::Run()
{
  // Run
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  if(fGetPassFromFileName)
    GetPass();
  
  return kTRUE;
}

Bool_t AliEmcalCorrectionComponent::UserNotify()
{
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  return kTRUE;
}

void AliEmcalCorrectionComponent::GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff)
{
  phidiff = 999;
  etadiff = 999;
  
  if (!t||!v) return;
  
  Double_t veta = t->GetTrackEtaOnEMCal();
  Double_t vphi = t->GetTrackPhiOnEMCal();
  
  Float_t pos[3] = {0};
  v->GetPosition(pos);
  TVector3 cpos(pos);
  Double_t ceta     = cpos.Eta();
  Double_t cphi     = cpos.Phi();
  etadiff=veta-ceta;
  phidiff=TVector2::Phi_mpi_pi(vphi-cphi);
}

void AliEmcalCorrectionComponent::UpdateCells()
{
  
  if (!fEvent) return ;
  
  Int_t bunchCrossNo = fEvent->GetBunchCrossNumber();
  
  if (fRecoUtils)
    fRecoUtils->RecalibrateCells(fCaloCells, bunchCrossNo);
  
  fCaloCells->Sort();
}

Bool_t AliEmcalCorrectionComponent::RunChanged()
{
  // Get run number.
  return fRun != fEvent->GetRunNumber();
}


bool AliEmcalCorrectionComponent::IsSharedValue(std::string & value)
{
  std::size_t sharedParameterLocation = value.find("sharedParameters:");
  if (sharedParameterLocation != std::string::npos)
  {
    // "sharedParameters:" is 17 characters long
    value.erase(sharedParameterLocation, sharedParameterLocation + 17);
    return true;
  }
  // Return false otherwise
  return false;
}

void AliEmcalCorrectionComponent::GetPass()
{
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  TTree *inputTree = mgr->GetTree();
  
  if (!inputTree)
  {
    AliError("Pointer to tree = 0, returning");
    return;
  }
  
  TFile *inputFile = inputTree->GetCurrentFile();
  if (!inputFile) {
    AliError("Null pointer input file, returning");
    return;
  }
  
  TString fname(inputFile->GetName());
  if      (fname.Contains("pass1")) fFilepass = TString("pass1");
  else if (fname.Contains("pass2")) fFilepass = TString("pass2");
  else if (fname.Contains("pass3")) fFilepass = TString("pass3");
  else if (fname.Contains("pass4")) fFilepass = TString("pass4");
  else if (fname.Contains("pass5")) fFilepass = TString("pass5");
  else if (fname.Contains("LHC11c") &&
           fname.Contains("spc_calo")) fFilepass = TString("spc_calo");
  else if (fname.Contains("calo") || fname.Contains("high_lumi"))
    
  {
    //printf("AliEMCALTenderSupply::GetPass() - Path contains <calo> or <high-lumi>, set as <pass1>\n");
    fFilepass = TString("pass1");
  }
  else if (fname.Contains("LHC14a1a"))
  {
    AliInfo("Energy calibration activated for this MC production!");
    fFilepass = TString("LHC14a1a");
  }
  else
  {
    AliFatal(Form("Pass number string not found: %s. Please set the pass number in the configuration!", fname.Data()));
    return;
  }
}

void AliEmcalCorrectionComponent::FillCellQA(TH1F* h){
  TString name = h->GetName();
  
  Short_t  absId  =-1;
  Double_t ecell = 0;
  Double_t tcell = 0;
  Double_t efrac = 0;
  Int_t  mclabel = -1;
  
  for (Int_t iCell = 0; iCell < fCaloCells->GetNumberOfCells(); iCell++){
    
    fCaloCells->GetCell(iCell, absId, ecell, tcell, mclabel, efrac);
    if(name.Contains("Energy")){
      h->Fill(ecell);
    }
    else if(name.Contains("Time")){
      h->Fill(tcell);
    }
    
  }
  
}