DerivationFramework::IsolationTrackDecorator Class Reference

#include <IsolationTrackDecorator.h>

Inheritance diagram for DerivationFramework::IsolationTrackDecorator:

Collaboration diagram for DerivationFramework::IsolationTrackDecorator:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	finalize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

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.
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.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::IsolationTrackDecorator::addBranches ( const EventContext & ctx ) const

finaloverridevirtual

Loop over tracks

Only decorate those passed selection

check flag

track isolation

Definition at line 97 of file IsolationTrackDecorator.cxx.

{
  // 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 ( )

finaloverridevirtual

Definition at line 89 of file IsolationTrackDecorator.cxx.

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

initialize()

StatusCode DerivationFramework::IsolationTrackDecorator::initialize ( )

finaloverridevirtual

create decorator list

Definition at line 24 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 retrieved the CaloIsolationTool!" );
  }
 
  if( ! m_trackIsolationTool.empty() ) {
    ATH_CHECK( m_trackIsolationTool.retrieve() );
    ATH_MSG_INFO( "Successfully retrieved 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 76 of file IsolationTrackDecorator.h.

{this, "CaloIsolationTool", ""};

m_dec_trkFlagKey

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

private

Initial value:

{
      this, "dec_trkFlagKey", ""}

Definition at line 95 of file IsolationTrackDecorator.h.

                                                                       {
      this, "dec_trkFlagKey", ""};

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 69 of file IsolationTrackDecorator.h.

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

m_iso

IntegerArrayProperty DerivationFramework::IsolationTrackDecorator::m_iso

private

Initial value:

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

Definition at line 81 of file IsolationTrackDecorator.h.

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

m_iso_suffix

StringArrayProperty DerivationFramework::IsolationTrackDecorator::m_iso_suffix

private

Initial value:

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

Definition at line 65 of file IsolationTrackDecorator.h.

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

m_prefix

StringProperty DerivationFramework::IsolationTrackDecorator::m_prefix

private

Initial value:

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

Definition at line 63 of file IsolationTrackDecorator.h.

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

m_ptconeDecoratorsKey

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

private

Initial value:

{
      this, "ptconeDecoratorsKey", {}}

Definition at line 87 of file IsolationTrackDecorator.h.

                                                                                 {
      this, "ptconeDecoratorsKey", {}};

m_ptconeTypes

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

private

Definition at line 78 of file IsolationTrackDecorator.h.

m_ptvarconeDecoratorsKey

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

private

Initial value:

{
      this, "ptvarconeDecoratorsKey", {}}

Definition at line 89 of file IsolationTrackDecorator.h.

                                                                                    {
      this, "ptvarconeDecoratorsKey", {}};

m_ptvarconeTypes

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

private

Definition at line 79 of file IsolationTrackDecorator.h.

m_selectionString

StringProperty DerivationFramework::IsolationTrackDecorator::m_selectionString

private

Initial value:

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

Definition at line 61 of file IsolationTrackDecorator.h.

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

m_selFlag

StringProperty DerivationFramework::IsolationTrackDecorator::m_selFlag

private

Initial value:

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

Definition at line 67 of file IsolationTrackDecorator.h.

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

m_selFlagValue

IntegerProperty DerivationFramework::IsolationTrackDecorator::m_selFlagValue

private

Initial value:

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

Definition at line 71 of file IsolationTrackDecorator.h.

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

m_topoclusCorrList

xAOD::CaloCorrection DerivationFramework::IsolationTrackDecorator::m_topoclusCorrList

private

Definition at line 85 of file IsolationTrackDecorator.h.

m_topoconeCorrList

xAOD::CaloCorrection DerivationFramework::IsolationTrackDecorator::m_topoconeCorrList

private

Definition at line 84 of file IsolationTrackDecorator.h.

m_topoetconeDecoratorsKey

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

private

Initial value:

{
      this, "topoetconeDecoratorsKey", {}}

Definition at line 91 of file IsolationTrackDecorator.h.

                                                                                     {
      this, "topoetconeDecoratorsKey", {}};

m_topoetconeNonCoreConeDecoratorsKey

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

private

Initial value:

{
      this, "topoetconeNonCoreConeDecoratorsKey", {}}

Definition at line 93 of file IsolationTrackDecorator.h.

                                                                                                {
      this, "topoetconeNonCoreConeDecoratorsKey", {}};

m_topoetconeTypes

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

private

Definition at line 80 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 59 of file IsolationTrackDecorator.h.

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

m_trackIsolationTool

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

private

Athena configured tools.

Definition at line 75 of file IsolationTrackDecorator.h.

{this, "TrackIsolationTool", ""};

m_trkCorrList

xAOD::TrackCorrection DerivationFramework::IsolationTrackDecorator::m_trkCorrList

private

Definition at line 83 of file IsolationTrackDecorator.h.

---

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

• IsolationTrackDecorator.h
• IsolationTrackDecorator.cxx