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;
};