StaticallyAllocatedSet.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "MooseError.h"
 
#include <array>
 
namespace MooseUtils
{
 
template <typename T, std::size_t N>
class StaticallyAllocatedSet
{
public:
  typedef typename std::array<T, N>::iterator iterator;
  typedef typename std::array<T, N>::const_iterator const_iterator;
 
  StaticallyAllocatedSet() : _end_pos(0) {}
 
  std::size_t size() const { return _end_pos; }
 
  bool empty() const { return !_end_pos; }
 
  void insert(const T & value)
  {
    if (contains(value))
      return;
 
    if (_end_pos == N)
      mooseError("Out of space in StaticallyAllocatedSet (size = ", N, ")");
 
    _data[_end_pos] = value;
    ++_end_pos;
  }
 
  void clear() { _end_pos = 0; }
 
  iterator begin() { return _data.begin(); }
  const_iterator begin() const { return _data.begin(); }
 
  iterator end() { return _data.begin() + _end_pos; }
  const_iterator end() const { return _data.begin() + _end_pos; }
 
  bool contains(const T & value) const
  {
    for (std::size_t i = 0; i < _end_pos; i++)
      if (_data[i] == value)
        return true;
 
    return false;
  }
 
  void swap(StaticallyAllocatedSet<T, N> & other)
  {
    _data.swap(other._data);
    std::swap(_end_pos, other._end_pos);
  }
 
  std::size_t dataEndPos() const { return _end_pos; }
 
private:
  std::array<T, N> _data;
 
  std::size_t _end_pos;
};
 
} // namespace MooseUtils