DerivationFramework::IsolationTrackDecorator Class Reference

#include <IsolationTrackDecorator.h>

Inheritance diagram for DerivationFramework::IsolationTrackDecorator:

Collaboration diagram for DerivationFramework::IsolationTrackDecorator:

Public Member Functions
	IsolationTrackDecorator (const std::string &t, const std::string &n, const IInterface *p)
	Constructor with parameters. More...
virtual	~IsolationTrackDecorator ()=default
	Destructor. More...
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual StatusCode	addBranches () const override

Private Attributes
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_trackContainerKey
StringProperty	m_selectionString
StringProperty	m_prefix
StringArrayProperty	m_iso_suffix
StringProperty	m_selFlag
IntegerProperty	m_diff_ptvarcone
IntegerProperty	m_selFlagValue
ToolHandle< xAOD::ITrackIsolationTool >	m_trackIsolationTool {this, "TrackIsolationTool", ""}
	Athena configured tools. More...
ToolHandle< xAOD::ICaloTopoClusterIsolationTool >	m_caloIsolationTool {this, "CaloIsolationTool", ""}
std::vector< xAOD::Iso::IsolationType >	m_ptconeTypes
std::vector< xAOD::Iso::IsolationType >	m_ptvarconeTypes
std::vector< xAOD::Iso::IsolationType >	m_topoetconeTypes
IntegerArrayProperty	m_iso
xAOD::TrackCorrection	m_trkCorrList
xAOD::CaloCorrection	m_topoconeCorrList
xAOD::CaloCorrection	m_topoclusCorrList
SG::WriteDecorHandleKeyArray< xAOD::TrackParticleContainer >	m_ptconeDecoratorsKey
SG::WriteDecorHandleKeyArray< xAOD::TrackParticleContainer >	m_ptvarconeDecoratorsKey
SG::WriteDecorHandleKeyArray< xAOD::TrackParticleContainer >	m_topoetconeDecoratorsKey
SG::WriteDecorHandleKeyArray< xAOD::TrackParticleContainer >	m_topoetconeNonCoreConeDecoratorsKey
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_dec_trkFlagKey

Detailed Description

Author

daiya.nosp@m..aki.nosp@m.yama@.nosp@m.cern.nosp@m..ch

Definition at line 46 of file IsolationTrackDecorator.h.

Constructor & Destructor Documentation

IsolationTrackDecorator()

DerivationFramework::IsolationTrackDecorator::IsolationTrackDecorator	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Constructor with parameters.

Definition at line 24 of file IsolationTrackDecorator.cxx.

                                                          :
  base_class(t, n, p){}

~IsolationTrackDecorator()

virtual DerivationFramework::IsolationTrackDecorator::~IsolationTrackDecorator ( )

virtualdefault

Destructor.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::IsolationTrackDecorator::addBranches ( ) const

overridevirtual

Loop over tracks

Only decorate those passed selection

check flag

track isolation

Definition at line 103 of file IsolationTrackDecorator.cxx.

{
  const EventContext& ctx = Gaudi::Hive::currentContext();
  // retrieve track container
  SG::ReadHandle<xAOD::TrackParticleContainer> toDecorate(m_trackContainerKey,ctx);
  ATH_CHECK( toDecorate.isValid() );
 
  // Execute the text parser and update the mask
  std::vector<int> entries(toDecorate->size(), 1);
  if (m_parser) {
    entries = m_parser->evaluateAsVector();
    if (entries.size() != toDecorate->size()) {
      ATH_MSG_ERROR("Sizes incompatible! Are you sure your selection string used ID TrackParticles?");
      return StatusCode::FAILURE;
    }
  }
 
  // build the decorators
  std::vector<SG::WriteDecorHandle<xAOD::TrackParticleContainer, float>> ptconeDecorators;
  for(unsigned int i=0; i<m_ptconeTypes.size(); i++){
    ptconeDecorators.emplace_back(m_ptconeDecoratorsKey[i], ctx);
  }
  std::vector<SG::WriteDecorHandle<xAOD::TrackParticleContainer, float>> ptvarconeDecorators;
  for(unsigned int i=0; i<m_ptvarconeTypes.size(); i++){
    ptvarconeDecorators.emplace_back(m_ptconeDecoratorsKey[i], ctx);
  }
  std::vector<SG::WriteDecorHandle<xAOD::TrackParticleContainer, float>> topoetconeDecorators;
  for(unsigned int i=0; i<m_topoetconeTypes.size(); i++){
    topoetconeDecorators.emplace_back(m_topoetconeDecoratorsKey[i], ctx);
  }
  std::vector<SG::WriteDecorHandle<xAOD::TrackParticleContainer, float>> topoetconeNonCoreConeDecorators;
  for(unsigned int i=0; i<m_topoetconeTypes.size(); i++){
    topoetconeNonCoreConeDecorators.emplace_back(m_topoetconeNonCoreConeDecoratorsKey[i], ctx);
  }
  
  SG::WriteDecorHandle<xAOD::TrackParticleContainer, int> dec_trkFlag(m_dec_trkFlagKey, ctx);
  
  int ipar=0;
  for(auto particle : *toDecorate) {
    bool IsPassed=true;
 
    if (!entries[ipar++]) IsPassed=false;
 
    if(not m_selFlag.empty()){
      if(dec_trkFlag(*particle)!=m_selFlagValue) IsPassed=false;
    }
 
    if(m_iso.size()>0){
      if(IsPassed){
    xAOD::TrackIsolation resultTrack;
    xAOD::TrackIsolation resultTrackVar;
    xAOD::CaloIsolation resultCalo;
    xAOD::CaloIsolation resultCaloNonCoreCone; 
    if(m_ptconeTypes.size()) {
 
      if (m_trackIsolationTool->trackIsolation(resultTrack, *particle, m_ptconeTypes, m_trkCorrList)){
        for(unsigned int i=0; i<m_ptconeTypes.size(); i++){
          ptconeDecorators[i](*particle) = resultTrack.ptcones[i];
        }
      }else{
        ATH_MSG_WARNING("Failed to apply the track isolation for a particle");
      }
    }
 
    if(m_ptvarconeTypes.size()) {
 
      std::vector<xAOD::Iso::IsolationType> ptconeTypes = {};
      for(auto isoType : m_ptvarconeTypes) {
        int iso_tmp = static_cast<int>(isoType);
        iso_tmp -= m_diff_ptvarcone;
        ptconeTypes.push_back(static_cast<xAOD::Iso::IsolationType>(iso_tmp));
      }
 
      if (m_trackIsolationTool->trackIsolation(resultTrackVar, *particle, ptconeTypes, m_trkCorrList)){
        for(unsigned int i=0; i<m_ptvarconeTypes.size(); i++){
          ptvarconeDecorators[i](*particle) = resultTrackVar.ptvarcones_10GeVDivPt[i];
        }
      }else{
        ATH_MSG_WARNING("Failed to apply the track isolation for a particle");
      }
    }
 
    if(m_topoetconeTypes.size()) {
 
      if (m_caloIsolationTool->caloTopoClusterIsolation(resultCalo, *particle, m_topoetconeTypes, m_topoconeCorrList)){
        for(unsigned int i=0; i<m_topoetconeTypes.size(); i++){
          topoetconeDecorators[i](*particle) = resultCalo.etcones[i];
        }
      }else{
        ATH_MSG_WARNING("Failed to apply the track isolation for a particle");
      }
 
      if (m_caloIsolationTool->caloTopoClusterIsolation(resultCaloNonCoreCone, *particle, m_topoetconeTypes, m_topoclusCorrList)){
        for(unsigned int i=0; i<m_topoetconeTypes.size(); i++){
          topoetconeNonCoreConeDecorators[i](*particle) = resultCaloNonCoreCone.etcones[i];
        }
      }else {
        ATH_MSG_WARNING("Failed to apply the topo calo isolation for a particle ( non CoreCone )");
      }
 
    }
 
      }else{
    for(unsigned int i=0; i<m_ptconeTypes.size(); i++){
      ptconeDecorators[i](*particle) = -999.e3;
    }
    for(unsigned int i=0; i<m_ptvarconeTypes.size(); i++){
      ptvarconeDecorators[i](*particle) = -999.e3;
    }
    for(unsigned int i=0; i<m_topoetconeTypes.size(); i++){
      topoetconeDecorators[i](*particle) = -999.e3;
      topoetconeNonCoreConeDecorators[i](*particle) = -999.e3;
    }
      }
    }
  } 
 
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::IsolationTrackDecorator::finalize ( )

overridevirtual

Definition at line 95 of file IsolationTrackDecorator.cxx.

{
  ATH_MSG_DEBUG("finalize() ...");
  ATH_CHECK(finalizeParser());
 
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::IsolationTrackDecorator::initialize ( )

overridevirtual

create decorator list

Definition at line 30 of file IsolationTrackDecorator.cxx.

{
  ATH_MSG_DEBUG("initialize() ...");
 
  // load the matching tool
  if( ! m_caloIsolationTool.empty() ) {
    ATH_CHECK( m_caloIsolationTool.retrieve() );
    ATH_MSG_INFO( "Successfully retrived the CaloIsolationTool!" );
  }
 
  if( ! m_trackIsolationTool.empty() ) {
    ATH_CHECK( m_trackIsolationTool.retrieve() );
    ATH_MSG_INFO( "Successfully retrived the TrackIsolationTool!" );
  }
 
  ATH_CHECK(m_trackContainerKey.initialize());
 
  m_trkCorrList.trackbitset.set(static_cast<unsigned int>(xAOD::Iso::coreTrackPtr));
  m_topoconeCorrList.calobitset.set(static_cast<unsigned int>(xAOD::Iso::coreCone));
  m_topoconeCorrList.calobitset.set(static_cast<unsigned int>(xAOD::Iso::pileupCorrection));
  m_topoclusCorrList.calobitset.set(static_cast<unsigned int>(xAOD::Iso::pileupCorrection));
 
  m_ptconeTypes.clear();
  m_ptvarconeTypes.clear();
  m_topoetconeTypes.clear();
 
  for(unsigned int i=0; i<m_iso.size(); i++){
    xAOD::Iso::IsolationType isoType = static_cast<xAOD::Iso::IsolationType>(m_iso[i]);
    xAOD::Iso::IsolationFlavour flavour = static_cast<xAOD::Iso::IsolationFlavour>(isoType/10.);
 
    if(flavour == xAOD::Iso::ptcone) {
      m_ptconeTypes.push_back(isoType);
      m_ptconeDecoratorsKey.emplace_back(m_trackContainerKey.key() + "." + m_prefix + m_iso_suffix[i]);
    }
    if(flavour == xAOD::Iso::ptvarcone) {
      m_ptvarconeTypes.push_back(isoType);
      m_ptvarconeDecoratorsKey.emplace_back(m_trackContainerKey.key() + "." + m_prefix + m_iso_suffix[i]);
    }
    if(flavour == xAOD::Iso::topoetcone) {
      m_topoetconeTypes.push_back(isoType);
      m_topoetconeDecoratorsKey.emplace_back(m_trackContainerKey.key() + "." + m_prefix + m_iso_suffix[i]);
      m_topoetconeNonCoreConeDecoratorsKey.emplace_back(m_trackContainerKey.key() + "." + m_prefix + m_iso_suffix[i] + "NonCoreCone");
    }
  }
  ATH_CHECK(m_ptconeDecoratorsKey.initialize());
  ATH_CHECK(m_ptvarconeDecoratorsKey.initialize());
  ATH_CHECK(m_topoetconeDecoratorsKey.initialize());
  ATH_CHECK(m_topoetconeNonCoreConeDecoratorsKey.initialize());
 
  // Set up the text-parsing machinery for thinning the tracks directly according to user cuts
  if (not m_selectionString.empty()) {
    ATH_CHECK(initializeParser(m_selectionString));
  }
 
  if(not m_selFlag.empty()){
    m_dec_trkFlagKey = m_trackContainerKey.key() + "." + m_selFlag;
  } else {
    m_dec_trkFlagKey = m_trackContainerKey.key() + ".no_key";
  }
  ATH_CHECK(m_dec_trkFlagKey.initialize());
 
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_caloIsolationTool

ToolHandle<xAOD::ICaloTopoClusterIsolationTool> DerivationFramework::IsolationTrackDecorator::m_caloIsolationTool {this, "CaloIsolationTool", ""}

private

Definition at line 79 of file IsolationTrackDecorator.h.

m_dec_trkFlagKey

SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_dec_trkFlagKey

private

Initial value:

{
      this, "dec_trkFlagKey", ""}

Definition at line 98 of file IsolationTrackDecorator.h.

m_diff_ptvarcone

IntegerProperty DerivationFramework::IsolationTrackDecorator::m_diff_ptvarcone

private

Initial value:

{
      this, "DiffPtvarcone", xAOD::Iso::ptvarcone20 - xAOD::Iso::ptcone20, "difference between Iso::ptvarcone20 and Iso::ptcone20"}

Definition at line 72 of file IsolationTrackDecorator.h.

m_iso

IntegerArrayProperty DerivationFramework::IsolationTrackDecorator::m_iso

private

Initial value:

{
      this, "iso", {}, "isolation types vector<int>"}

Definition at line 84 of file IsolationTrackDecorator.h.

m_iso_suffix

StringArrayProperty DerivationFramework::IsolationTrackDecorator::m_iso_suffix

private

Initial value:

{
      this, "isoSuffix", {""}, "suffix for the isolations"}

Definition at line 68 of file IsolationTrackDecorator.h.

m_prefix

StringProperty DerivationFramework::IsolationTrackDecorator::m_prefix

private

Initial value:

{
      this, "Prefix", "", "prefix"}

Definition at line 66 of file IsolationTrackDecorator.h.

m_ptconeDecoratorsKey

SG::WriteDecorHandleKeyArray<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_ptconeDecoratorsKey

private

Initial value:

{
      this, "ptconeDecoratorsKey", {}}

Definition at line 90 of file IsolationTrackDecorator.h.

m_ptconeTypes

std::vector<xAOD::Iso::IsolationType> DerivationFramework::IsolationTrackDecorator::m_ptconeTypes

private

Definition at line 81 of file IsolationTrackDecorator.h.

m_ptvarconeDecoratorsKey

SG::WriteDecorHandleKeyArray<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_ptvarconeDecoratorsKey

private

Initial value:

{
      this, "ptvarconeDecoratorsKey", {}}

Definition at line 92 of file IsolationTrackDecorator.h.

m_ptvarconeTypes

std::vector<xAOD::Iso::IsolationType> DerivationFramework::IsolationTrackDecorator::m_ptvarconeTypes

private

Definition at line 82 of file IsolationTrackDecorator.h.

m_selectionString

StringProperty DerivationFramework::IsolationTrackDecorator::m_selectionString

private

Initial value:

{
      this, "SelectionString", "", "selection string"}

Definition at line 64 of file IsolationTrackDecorator.h.

m_selFlag

StringProperty DerivationFramework::IsolationTrackDecorator::m_selFlag

private

Initial value:

{
      this, "SelectionFlag", "", "selection flag"}

Definition at line 70 of file IsolationTrackDecorator.h.

m_selFlagValue

IntegerProperty DerivationFramework::IsolationTrackDecorator::m_selFlagValue

private

Initial value:

{
      this, "SelectionFlagValue", 1, "selection flag value"}

Definition at line 74 of file IsolationTrackDecorator.h.

m_topoclusCorrList

xAOD::CaloCorrection DerivationFramework::IsolationTrackDecorator::m_topoclusCorrList

private

Definition at line 88 of file IsolationTrackDecorator.h.

m_topoconeCorrList

xAOD::CaloCorrection DerivationFramework::IsolationTrackDecorator::m_topoconeCorrList

private

Definition at line 87 of file IsolationTrackDecorator.h.

m_topoetconeDecoratorsKey

SG::WriteDecorHandleKeyArray<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_topoetconeDecoratorsKey

private

Initial value:

{
      this, "topoetconeDecoratorsKey", {}}

Definition at line 94 of file IsolationTrackDecorator.h.

m_topoetconeNonCoreConeDecoratorsKey

SG::WriteDecorHandleKeyArray<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_topoetconeNonCoreConeDecoratorsKey

private

Initial value:

{
      this, "topoetconeNonCoreConeDecoratorsKey", {}}

Definition at line 96 of file IsolationTrackDecorator.h.

m_topoetconeTypes

std::vector<xAOD::Iso::IsolationType> DerivationFramework::IsolationTrackDecorator::m_topoetconeTypes

private

Definition at line 83 of file IsolationTrackDecorator.h.

m_trackContainerKey

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::IsolationTrackDecorator::m_trackContainerKey

private

Initial value:

{
      this, "TargetContainer", "InDetTrackParticles", "track particle container name"}

Definition at line 62 of file IsolationTrackDecorator.h.

m_trackIsolationTool

ToolHandle<xAOD::ITrackIsolationTool> DerivationFramework::IsolationTrackDecorator::m_trackIsolationTool {this, "TrackIsolationTool", ""}

private

Athena configured tools.

Definition at line 78 of file IsolationTrackDecorator.h.

m_trkCorrList

xAOD::TrackCorrection DerivationFramework::IsolationTrackDecorator::m_trkCorrList

private

Definition at line 86 of file IsolationTrackDecorator.h.

---

The documentation for this class was generated from the following files:

• IsolationTrackDecorator.h
• IsolationTrackDecorator.cxx