OverlapDecorationHelper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ASSOCIATIONUTILS_OVERLAPDECORATIONHELPER_H
#define ASSOCIATIONUTILS_OVERLAPDECORATIONHELPER_H
 
// EDM includes
#include "xAODBase/IParticleContainer.h"
 
// Local includes
#include "AssociationUtils/OverlapRemovalDefs.h"
 
namespace ORUtils
{
 
  class OverlapDecorationHelper
  {
 
    public:
 
      OverlapDecorationHelper(const std::string& inputLabel,
                              const std::string& outputLabel,
                              bool outputPassValue = false);
 
      bool isInputObject(const xAOD::IParticle& obj) const;
 
      bool isRejectedObject(const xAOD::IParticle& obj) const;
 
      bool isSurvivingObject(const xAOD::IParticle& obj) const;
 
      char getObjectPriority(const xAOD::IParticle& obj) const;
 
      void setOverlapDecoration(const xAOD::IParticle& obj, bool result) const;
 
      void setObjectPass(const xAOD::IParticle& obj) const;
 
      void setObjectFail(const xAOD::IParticle& obj) const;
 
      bool isDecorated(const xAOD::IParticleContainer& container) const;
 
      void initializeDecorations(const xAOD::IParticleContainer& container) const;
 
      void resetDecorations(const xAOD::IParticleContainer& container) const;
 
    private:
 
      bool m_useInputLabel;
 
      ort::inputAccessor_t m_inputAccessor;
      ort::outputDecorator_t m_outputDecorator;
 
      bool m_outputPassValue;
 
  }; // class OverlapDecorationHelper
 
} // namespace ORUtils
 
#endif