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