CreateDummyEl.cxx File Reference

#include "CreateDummyEl.h"
#include "xAODEgamma/Egamma.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEventInfo/EventInfo.h"
#include <xAODCaloEvent/CaloCluster.h>
#include <xAODCaloEvent/CaloClusterAuxContainer.h>
#include <xAODCaloEvent/CaloClusterContainer.h>
#include <xAODEgamma/ElectronAuxContainer.h>
#include "TruthUtils/ParticleConstants.h"
#include "AsgMessaging/MessageCheck.h"
#include "AsgMessaging/MsgStream.h"

Go to the source code of this file.

Namespaces
namespace	asg

Functions
StatusCode	getElectrons (const std::vector< std::pair< double, double > > &pt_eta, int runNumber, xAOD::TStore &store)

Function Documentation

getElectrons()

StatusCode getElectrons	(	const std::vector< std::pair< double, double > > &	pt_eta,
		int	runNumber,
		xAOD::TStore &	store )

Definition at line 74 of file CreateDummyEl.cxx.

{
  using namespace asg::CreateDummyEl;
  ANA_CHECK_SET_TYPE(int);
  setMsgLevel(MSG::INFO);
  // This is what we will return back
 
  // create the EventInfo
  std::unique_ptr<xAOD::EventInfo> eventInfo =
    std::make_unique<xAOD::EventInfo>();
  eventInfo->makePrivateStore();
  eventInfo->setEventNumber(363636);
  eventInfo->setRunNumber(runNumber);
  static SG::AuxElement::Decorator<unsigned int> randomrunnumber(
    "RandomRunNumber");
  randomrunnumber(*eventInfo) = runNumber;
  if (!store.record(std::move(eventInfo), "EventInfo").isSuccess()) {
    ANA_MSG_ERROR("Could not record EventInfo");
    return StatusCode::FAILURE;
  }
 
  size_t numel = pt_eta.size();
  // Create all clusters
  std::unique_ptr<xAOD::CaloClusterContainer> clusters =
    std::make_unique<xAOD::CaloClusterContainer>();
  std::unique_ptr<xAOD::CaloClusterAuxContainer> clAux =
    std::make_unique<xAOD::CaloClusterAuxContainer>();
  clusters->setStore(clAux.get());
  clusters->reserve(numel);
  for (const auto& i : pt_eta) {
    const double pt{ i.first };
    const double eta{ i.second };
    const double e{ pt * cosh(eta) };
    xAOD::CaloCluster* cluster = new xAOD::CaloCluster();
    clusters->push_back(cluster);
    fill_cluster(cluster, eta, 0.0, e);
  }
 
  if (!store.record(std::move(clusters), "MyClusters").isSuccess() ||
      !store.record(std::move(clAux), "MyClustersAux.").isSuccess()) {
    ANA_MSG_ERROR("Could not record clusters");
    return StatusCode::FAILURE;
  }
  // Create all electrons  one per cluster
  std::vector<ElementLink<xAOD::CaloClusterContainer>> links{};
  std::unique_ptr<xAOD::ElectronContainer> electrons =
    std::make_unique<xAOD::ElectronContainer>();
  std::unique_ptr<xAOD::ElectronAuxContainer> electronsAux =
    std::make_unique<xAOD::ElectronAuxContainer>();
  electrons->setStore(electronsAux.get());
  electrons->reserve(numel);
  for (const auto& i : pt_eta) {
    const double pt{ i.first };
    const double eta{ i.second };
    xAOD::Electron* el = new xAOD::Electron;
    electrons->push_back(el);
    links.clear();
    links.emplace_back("MyClusters", el->index());
    el->setCaloClusterLinks(links);
    el->setEta(eta);
    el->setPhi(0.0);
    el->setPt(pt);
  }
  if (!store.record(std::move(electrons), "MyElectrons").isSuccess() ||
      !store.record(std::move(electronsAux), "MyElectronAux.").isSuccess()) {
    ANA_MSG_ERROR("Could not record Electrons");
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}