HashMap.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 "libmesh/threads.h"

#include <unordered_map>

template <typename Key, typename T>
class HashMap : public std::unordered_map<Key, T>
{
public:
  inline T & operator[](const Key & k)
  {
    libMesh::Threads::spin_mutex::scoped_lock lock(spin_mutex);
    return std::unordered_map<Key, T>::operator[](k);
  }

  inline std::size_t erase(const Key & k)
  {
    libMesh::Threads::spin_mutex::scoped_lock lock(spin_mutex);
    return std::unordered_map<Key, T>::erase(k);
  }

  using typename std::unordered_map<Key, T>::const_iterator;
  using typename std::unordered_map<Key, T>::iterator;

  inline iterator find(const Key & k)
  {
    libMesh::Threads::spin_mutex::scoped_lock lock(spin_mutex);
    return std::unordered_map<Key, T>::find(k);
  }

  inline const_iterator find(const Key & k) const
  {
    libMesh::Threads::spin_mutex::scoped_lock lock(spin_mutex);
    return std::unordered_map<Key, T>::find(k);
  }

  inline bool contains(const Key & key) const { return this->find(key) != this->end(); }

private:
  mutable libMesh::Threads::spin_mutex spin_mutex;
};