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"
 
// Columnar includes
#include "ColumnarCore/ColumnAccessor.h"
#include "ColumnarCore/ColumnarTool.h"
#include "ColumnarCore/ObjectRange.h"
 
// Local includes
#include "AssociationUtils/OverlapRemovalDefs.h"
 
namespace ADD_CMODE(ORUtils)
{

  template<columnar::ContainerIdConcept CI>
  class OverlapDecorationHelper : public columnar::ColumnarTool<>
  {
 
    public:

      OverlapDecorationHelper(const std::string& inputLabel,
                              const std::string& outputLabel,
                              bool outputPassValue = false);

      bool isInputObject(columnar::ObjectId<CI> obj) const;

      bool isRejectedObject(columnar::ObjectId<CI> obj) const;

      bool isSurvivingObject(columnar::ObjectId<CI> obj) const;

      char getObjectPriority(columnar::ObjectId<CI> obj) const;

      void setOverlapDecoration(columnar::ObjectId<CI> obj, bool result) const;

      void setObjectPass(columnar::ObjectId<CI> obj) const;

      void setObjectFail(columnar::ObjectId<CI> obj) const;

      bool isDecorated(columnar::ObjectRange<CI> container) const;

      void initializeDecorations(columnar::ObjectRange<CI> container) const;

      void resetDecorations(columnar::ObjectRange<CI> container) const;
 
    private:

      bool m_useInputLabel;

      columnar::ColumnAccessor<CI,char> m_inputAccessor;
      columnar::ColumnDecorator<CI,char> m_outputDecorator;

      bool m_outputPassValue;
 
  }; // class OverlapDecorationHelper
 
  //---------------------------------------------------------------------------
  // Constructor
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  OverlapDecorationHelper<CI>::
  OverlapDecorationHelper(const std::string& inputLabel,
                          const std::string& outputLabel,
                          bool outputPassValue)
    : m_useInputLabel(!inputLabel.empty()),
      m_outputDecorator(*this, outputLabel),
      m_outputPassValue(outputPassValue)
  {
    if(m_useInputLabel)
      resetAccessor (m_inputAccessor, *this, inputLabel);
  }
 
  //---------------------------------------------------------------------------
  // Determine if object is currently OK for input to OR
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  bool OverlapDecorationHelper<CI>::isInputObject
  (columnar::ObjectId<CI> obj) const
  {
    // Input label is turned off if empty string
    if(!m_useInputLabel) return true;
    return m_inputAccessor(obj);
  }
 
  //---------------------------------------------------------------------------
  // Determine if object is currently rejected by OR
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  bool OverlapDecorationHelper<CI>::isRejectedObject
  (columnar::ObjectId<CI> obj) const
  {
    // isRejected = isInput && (output == fail)
    return isInputObject(obj) && ( m_outputDecorator(obj) != m_outputPassValue );
  }
  //---------------------------------------------------------------------------
  // Determine if object is NOT currently rejected by OR (is surviving)
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  bool OverlapDecorationHelper<CI>::isSurvivingObject
  (columnar::ObjectId<CI> obj) const
  {
    // isSurviving = isInput && (output == pass)
    return isInputObject(obj) && ( m_outputDecorator(obj) == m_outputPassValue );
  }
  //---------------------------------------------------------------------------
  // Retrieve the user-defined object priority
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  char OverlapDecorationHelper<CI>::
  getObjectPriority(columnar::ObjectId<CI> obj) const
  {
    // We current reuse the input decoration as the priority score
    return m_inputAccessor(obj);
  }
 
  //---------------------------------------------------------------------------
  // Set the overlap (output) decoration
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  void OverlapDecorationHelper<CI>::setOverlapDecoration
  (columnar::ObjectId<CI> obj, bool result) const
  {
    m_outputDecorator(obj) = result;
  }
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  void OverlapDecorationHelper<CI>::setObjectPass(columnar::ObjectId<CI> obj) const
  {
    setOverlapDecoration(obj, m_outputPassValue);
  }
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  void OverlapDecorationHelper<CI>::setObjectFail(columnar::ObjectId<CI> obj) const
  {
    setOverlapDecoration(obj, !m_outputPassValue);
  }
  
  //---------------------------------------------------------------------------
  // Check if output decoration has been applied to a container
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  bool OverlapDecorationHelper<CI>::isDecorated
  (columnar::ObjectRange<CI> container) const
  {
    return container.size() > 0 &&
           m_outputDecorator.isAvailable(container[0]);
  }
  
  //---------------------------------------------------------------------------
  // Initialize output decoration
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  void OverlapDecorationHelper<CI>::initializeDecorations
  (columnar::ObjectRange<CI> container) const
  {
    if(!isDecorated(container))
      resetDecorations(container);
  }
  
  //---------------------------------------------------------------------------
  // Reset output decoration
  //---------------------------------------------------------------------------
  template<columnar::ContainerIdConcept CI>
  void OverlapDecorationHelper<CI>::resetDecorations
  (columnar::ObjectRange<CI> container) const
  {
    for(auto obj : container){
      // This isn't terrible intuitive, but in order to support both output
      // logic modes in a reasonable way, we initialize the output flag to the
      // logical AND of isInput and outputPassValue. This results in non-input
      // objects being initialized to 'false' regardless of output logic.
      bool result = ( isInputObject(obj) && m_outputPassValue );
      setOverlapDecoration(obj, result);
    }
    //for(auto obj : container) setObjectPass(obj);
  }
 
} // namespace ADD_CMODE(ORUtils)
 
#endif