FlagSet.h

Go to the documentation of this file.

// Copyright (c) 2002-present, OpenMS Inc. -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Chris Bielow $
// $Authors: Chris Bielow $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/CONCEPT/Macros.h> // for OPENMS_PRECONDITION
 
namespace OpenMS
{
  template<class ENUM>
  class FlagSet
  {
  public:
    FlagSet() 
      : value_(0)
    {}
 
    explicit FlagSet(const ENUM& en)
      : value_(getPow_(en))
    {
    }
 
    FlagSet(const FlagSet& stat) = default;
 
    FlagSet& operator=(const FlagSet& stat) = default;
 
    FlagSet& operator=(const ENUM& en) = delete;
 
 
    ~FlagSet() = default;
 
    bool operator==(const FlagSet& stat) const
    {
      return (value_ == stat.value_);
    }
 
    FlagSet operator&(const ENUM& en) const
    {
      FlagSet s(*this) ;
      s &= en;
      return s;
    }
 
    FlagSet operator&(const FlagSet& rhs) const
    {
      FlagSet s(*this);
      s &= rhs;
      return s;
    }
 
    FlagSet& operator&=(const ENUM& en)
    {
      value_ &= getPow_(en);
      return *this;
    }
  
    FlagSet& operator&=(const FlagSet& rhs)
    {
      value_ &= rhs.value_;
      return *this;
    }
 
    FlagSet operator|(const ENUM& en) const
    {
      FlagSet s(*this);
      s.value_ |= getPow_(en);
      return s;
    }
    FlagSet operator|(const FlagSet& rhs) const
    {
      FlagSet s(*this);
      s.value_ |= rhs.value_;
      return s;
    }
 
    FlagSet& operator|=(const ENUM& en)
    {
      value_ |= getPow_(en);
      return *this;
    }
 
    FlagSet& operator|=(const FlagSet& rhs)
    {
      value_ |= rhs.value_;
      return *this;
    }
 
    FlagSet operator+(const ENUM& en) const
    {
      return *this | en;
    }
 
    FlagSet operator+(const FlagSet& en) const
    {
      return *this | en;
    }
    FlagSet& operator+=(const ENUM& rhs)
    {
      return *this |= rhs;
    }
 
    FlagSet& operator+=(const FlagSet& rhs)
    {
      return *this |= rhs;
    }
 
    FlagSet operator-(const FlagSet& rhs)
    {
      FlagSet r(*this);
      r -= rhs;
      return r;
    }
 
    FlagSet& operator-=(const FlagSet& rhs)
    {
      auto overlap = value_ & rhs.value_;
      value_ ^= overlap; // disable bits which overlap with rhs using XOR
      return *this;
    }
 
    FlagSet operator-(const ENUM& rhs)
    {
      FlagSet r(*this);
      r -= rhs;
      return r;
    }
 
    FlagSet& operator-=(const ENUM& rhs)
    {
      auto overlap = value_ & FlagSet(rhs).value_;
      value_ ^= overlap; // disable bits which overlap with rhs using XOR
      return *this;
    }
 
    bool isSuperSetOf(const FlagSet& required) const
    {
      return ((*this | required) == *this);
    }
 
    bool isSuperSetOf(const ENUM& required) const
    {
      return ((*this | required) == *this);
    }
 
    bool empty() const
    {
      return value_ == 0;
    }
 
    UInt64 value() const
    {
      return value_;
    }
 
  private:
    UInt64 getPow_(const ENUM& en) const
    {
      OPENMS_PRECONDITION((int)en >= 0, "Enum value is too small!")
      OPENMS_PRECONDITION((int)en <= 63, "Enum value is too large!")
      return UInt64(1) << UInt64(en);
    }
    UInt64 value_;
  };
}