AliAnalysisTaskEmcalJetLight.cxx

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
#include "AliAnalysisTaskEmcalJetLight.h"

#include <TClonesArray.h>
#include <TList.h>

#include "AliAnalysisManager.h"
#include "AliCentrality.h"
#include "AliEMCALGeometry.h"
#include "AliESDEvent.h"
#include "AliEmcalJet.h"
#include "AliLog.h"
#include "AliRhoParameter.h"
#include "AliLocalRhoParameter.h"
#include "AliVCluster.h"
#include "AliVEventHandler.h"
#include "AliVParticle.h"

ClassImp(AliAnalysisTaskEmcalJetLight);

AliAnalysisTaskEmcalJetLight::AliAnalysisTaskEmcalJetLight() :
  AliAnalysisTaskEmcalLight(),
  fJetCollArray()
{
}

AliAnalysisTaskEmcalJetLight::AliAnalysisTaskEmcalJetLight(const char *name, Bool_t histo) :
  AliAnalysisTaskEmcalLight(name, histo),
  fJetCollArray()
{
}

AliAnalysisTaskEmcalJetLight::~AliAnalysisTaskEmcalJetLight()
{
  for (auto cont_it : fJetCollArray) delete cont_it.second;
}

void AliAnalysisTaskEmcalJetLight::ExecOnce()
{
  AliAnalysisTaskEmcalLight::ExecOnce();

  //Load all requested jet branches - each container knows name already
  if (fJetCollArray.size() == 0) {
    AliWarning("There are no jet collections");
    return;
  }

  for (auto cont_it : fJetCollArray) {
    AliJetContainer *cont = cont_it.second;
    cont->SetRunNumber(InputEvent()->GetRunNumber());
    cont->SetArray(InputEvent());
    cont->LoadRho(InputEvent());
  }
}

Bool_t AliAnalysisTaskEmcalJetLight::RetrieveEventObjects()
{
  if (!AliAnalysisTaskEmcalLight::RetrieveEventObjects()) return kFALSE;

  for (auto cont_it : fJetCollArray) cont_it.second->NextEvent(InputEvent());

  return kTRUE;
}

AliJetContainer* AliAnalysisTaskEmcalJetLight::AddJetContainer(EJetType_t jetType, EJetAlgo_t jetAlgo, ERecoScheme_t recoScheme, Double_t radius,
    UInt_t accType, std::string partContName, std::string clusContName, TString tag)
{
  AliParticleContainer* partCont = nullptr;
  AliClusterContainer* clusCont =  nullptr;
  auto partContSearch = fParticleCollArray.find(partContName);
  if (partContSearch != fParticleCollArray.end()) partCont = partContSearch->second;
  auto clusContSearch = fClusterCollArray.find(clusContName);
  if (clusContSearch != fClusterCollArray.end()) clusCont = clusContSearch->second;
  if (!partCont && !clusCont) {
    AliError(Form("Could not find neither particle nor cluster container with names '%s' and '%s'", partContName.c_str(), clusContName.c_str()));
    return nullptr;
  }

  return AddJetContainer(jetType, jetAlgo, recoScheme, radius, accType, partCont, clusCont, tag);
}

AliJetContainer* AliAnalysisTaskEmcalJetLight::AddJetContainer(EJetType_t jetType, EJetAlgo_t jetAlgo, ERecoScheme_t recoScheme, Double_t radius, UInt_t accType,
    AliParticleContainer* partCont, AliClusterContainer* clusCont, TString tag)
{
  AliJetContainer *cont = new AliJetContainer(jetType, jetAlgo, recoScheme, radius, partCont, clusCont, tag);
  cont->SetJetAcceptanceType(accType);
  AdoptJetContainer(cont);

  return cont;
}

AliJetContainer* AliAnalysisTaskEmcalJetLight::GetJetContainer(std::string name) const
{
  std::map<std::string, AliJetContainer*>::const_iterator cont_it = fJetCollArray.find(name);
  if (cont_it != fJetCollArray.end()) return cont_it->second;
  else return nullptr;
}