AliEmcalCorrectionComponent.cxx

Go to the documentation of this file.

// AliEmcalCorrectionComponent
//

#include "AliEmcalCorrectionComponent.h"

#include <yaml-cpp/yaml.h>

#include "AliEmcalList.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(0),
  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(0),
  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()
{
}

std::string AliEmcalCorrectionComponent::DetermineUseDefaultName(inputObjectType objType, Bool_t esdMode)
{
  std::string inputObjectBranch = "";
  if (objType == kCluster) {
    if (esdMode == true) {
      inputObjectBranch = "CaloClusters";
    }
    else {
      inputObjectBranch = "caloClusters";
    }
  }
  else if (objType == kTrack) {
    if (esdMode == true) {
      inputObjectBranch = "Tracks";
    }
    else {
      inputObjectBranch = "tracks";
    }
  }
  else if (objType == kCaloCells) {
    if (esdMode == true) {
      inputObjectBranch = "EMCALCells";
    }
    else {
      inputObjectBranch = "emcalCells";
    }
  }
  else {
    // Default to empty if we are given an unrecognized type with "usedefault"
    inputObjectBranch = "";
  }

  return inputObjectBranch;
}

void AliEmcalCorrectionComponent::AddContainer(inputObjectType contType)
{
  // Determine the type of branch to request
  std::string containerBranch = "";
  std::string containerBranchInConfiguration = "BranchName";
  if (contType == kCluster) {
    containerBranchInConfiguration = "cluster" + containerBranchInConfiguration;
  }
  else if (contType == kTrack) {
    containerBranchInConfiguration = "track" + containerBranchInConfiguration;
  } else {
    AliFatal(TString::Format("%s: Must specify type of container when requesting branch.", GetName()));
  }
  
  // Retrieve branch name
  GetProperty(containerBranchInConfiguration, containerBranch);
  // Should be unnecessary, since the user can only do this if done explicitly.
  /*if (containerBranch == "")
  {
    AliFatal(TString::Format("%s: Request %i container, but the container branch is empty!", GetName(), contType));
  }*/

  // Determine proper name if using "usedefault" pattern
  if (containerBranch == "usedefault") {
    containerBranch = DetermineUseDefaultName(contType, fEsdMode);
  }

  // Create containers and set them to the name of the component
  if (contType == kCluster)
  {
    fClusCont = new AliClusterContainer(containerBranch.c_str());
    fClusCont->SetName(GetName());
  }
  else if (contType == kTrack)
  {
    if (containerBranch == "mcparticles") {
      fPartCont = new AliMCParticleContainer(containerBranch.c_str());
    }
    else {
      fPartCont = new AliTrackContainer(containerBranch.c_str());
    }
    fPartCont->SetName(GetName());
  }
}

Bool_t AliEmcalCorrectionComponent::Initialize()
{
  AliDebug(3, Form("%s", __PRETTY_FUNCTION__));
  
  // Dynamic cast to see if event if ESD or AOD
  fEsdMode = dynamic_cast<AliESDEvent*>(fEvent);
  
  // 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;
  
  // Output
  fOutput = new AliEmcalList();
  fOutput->SetOwner();

  return kTRUE;
}

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
  {
    AliError(Form("Pass number string not found: %s", 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);
    }
    
  }
  
}