AcceptInfo.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// Dear emacs, this is -*-c++-*-
 
#ifndef PATCORE_ACCEPT_INFO_H
#define PATCORE_ACCEPT_INFO_H
 
#include <string>
#include <map>
#include <bitset>
 
 
namespace asg {
  class AcceptInfo final
  {
 
  public: 
    static const unsigned int NBITS=32;
 
    AcceptInfo(const char* name="AcceptInfo")
      : m_name(name),
        m_cutMap()
    {}
  
  public:
 
    inline const char* getName() const { return m_name.c_str(); };
 
    inline unsigned int getNCuts() const { return m_cutMap.size(); };
 
    inline const std::bitset<NBITS>& getCutMask() const { return m_cutMask; };
 
 
    int addCut( const std::string& cutName, const std::string& cutDescription )
    {
      // Make sure that this new cuts doesn't exceed the number of bits available
      if ( m_cutMap.size() >= NBITS )
      {
        return -1;
      }
 
      // Add the cut to the map
      std::pair< std::string, unsigned int > cutPair = std::make_pair( cutDescription, m_cutMap.size() );
      m_cutMap.insert( std::make_pair( cutName, cutPair ) );
 
      // Return the position of the newly added cut in the bitmask
      int result = (m_cutMap.size() - 1);
      m_cutMask.set (result);
      return result;
    }
 
 
    inline unsigned int getCutPosition( const std::string& cutName ) const 
    {
      auto it = m_cutMap.find(cutName);
      return (it != m_cutMap.end()) ? (it->second).second : 999999;
    }
 
 
    const std::string& getCutName( unsigned int cutPosition ) const;
 
 
    const std::string& getCutDescription( const std::string& cutName ) const;
 
    const std::string& getCutDescription( unsigned int cutPosition ) const;
 
 
    inline void setCutDescription( const std::string& cutName, const std::string& cutDescription )
    {
      unsigned int cutPosition = getCutPosition(cutName);
      return setCutDescription( cutPosition, cutDescription );
    }
 
    void setCutDescription( unsigned int cutPosition, const std::string& cutDescription );
 
 
    // Private members
  private:
 
    std::string m_name;
 
    std::map< std::string, std::pair< std::string, unsigned int > > m_cutMap;
 
    std::bitset<NBITS> m_cutMask;
 
 
  }; // End: class definition
 
} // End: namespace asg
 
 
#endif