vector.h

Go to the documentation of this file.

// Author(s): Jan Friso Groote, Maurice Laveaux
// Copyright: see the accompanying file COPYING or copy at
// https://github.com/mCRL2org/mCRL2/blob/master/COPYING
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//
//
 
#ifndef MCRL2_ATERMPP_STANDARD_CONTAINER_VECTOR_H
#define MCRL2_ATERMPP_STANDARD_CONTAINER_VECTOR_H
 
#include <vector>
 
#include "mcrl2/atermpp/detail/aterm_container.h"
#include "mcrl2/atermpp/detail/thread_aterm_pool.h"
#include "mcrl2/utilities/shared_mutex.h"
 
namespace atermpp
{
 
template < class T, class Alloc = std::allocator<detail::reference_aterm<T> >, bool ThreadSafe = false > 
class vector : public std::vector< detail::reference_aterm<T>, Alloc >
{
protected:
  typedef std::vector< detail::reference_aterm<T>, Alloc > super;
 
  detail::generic_aterm_container<std::vector<detail::reference_aterm<T>, Alloc>> container_wrapper;
 
public: 
  typedef typename super::allocator_type allocator_type;
  typedef typename super::value_type value_type;
  typedef typename super::size_type size_type;
  typedef typename super::reference reference;
  typedef typename super::iterator iterator;
  typedef typename super::const_iterator const_iterator;
  
  vector()
   : super(),
     container_wrapper(*this)
  {}
 
  explicit vector (const allocator_type& alloc)
   : super::vector(alloc),
     container_wrapper(*this)     
  {}
 
  explicit vector (size_type n, const allocator_type& alloc = allocator_type())
   : super::vector(n, alloc),
     container_wrapper(*this)     
  {}
 
  vector (size_type n, const value_type& val, const allocator_type& alloc = allocator_type())
   : super::vector(n, detail::reference_aterm(val), alloc),
     container_wrapper(*this)
  {}
 
  template <class InputIterator>
  vector (InputIterator first, InputIterator last, const allocator_type& alloc = allocator_type())
   : super::vector(first, last, alloc),
     container_wrapper(*this)     
  {}
 
  vector (const vector& x)
   : super::vector(x),
     container_wrapper(*this)     
  {}
 
  vector (const vector& x, const allocator_type& alloc)
   : super::vector(x, alloc),
     container_wrapper(*this)     
  {}
  
  vector (vector&& x)
   : super::vector(std::move(x)),
     container_wrapper(*this)     
  {}
 
  vector (vector&& x, const allocator_type& alloc)
   : super::vector(std::move(x), alloc),
     container_wrapper(*this)
  {}
 
  vector (std::initializer_list<value_type> il, const allocator_type& alloc = allocator_type())
    : super::vector(il, alloc),
      container_wrapper(*this)      
  {}
 
  vector& operator=(const vector& x) = default;
  
  vector& operator=(vector&& x) = default;
 
  ~vector() {}
  
  void shrink_to_fit()
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::shrink_to_fit();
    } else {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::shrink_to_fit();
    }
  }
 
  void clear() noexcept
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::clear();
    } else {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::clear();
    }
  }
 
  iterator insert( const_iterator pos, const T& value )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
 
      // This is not thread safe otherwise since the length or end iterator is updated during this producedure.
      return super::insert(pos, value);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::insert(pos, value);
  }
 
  iterator insert( const_iterator pos, T&& value )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::insert(pos, value);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::insert(pos, value);
  }
  
  iterator insert( const_iterator pos, size_type count, const T& value )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::insert(pos, count, value);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::insert(pos, count, value);
  }
    
  template< class InputIt >
  iterator insert( const_iterator pos,
                  InputIt first, InputIt last )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::insert(pos, first, last); 
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::insert(pos, first, last);  
  }
    
  iterator insert( const_iterator pos, std::initializer_list<T> ilist )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::insert(pos, ilist);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::insert(pos, ilist);
  }
  
  template< class... Args >
  iterator emplace( const_iterator pos, Args&&... args )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::emplace(pos, args...);   
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::emplace(pos, args...);   
  }
 
  iterator erase( const_iterator pos )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::erase(pos);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::erase(pos);
  }
 
  iterator erase( const_iterator first, const_iterator last )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::erase(first, last);
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::erase(first, last);
  }
 
  void push_back( const T& value )
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::push_back(value);
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::push_back(value);
    }
  }
 
  void push_back( T&& value )
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::push_back(value);
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::push_back(value);
    }
  }
 
  template< class... Args >
  reference emplace_back( Args&&... args )
  {
    if constexpr (ThreadSafe) {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      return super::emplace_back(args...);  
    }
 
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::emplace_back(args...);  
  }
 
  void pop_back()
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::pop_back();
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::pop_back();
    }
  }
 
  void resize( size_type count )
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::resize(count);
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::resize(count);
    }
  }
 
  void resize( size_type count, const value_type& value )
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::resize(count, value);
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::resize(count, value);
    }
  }
 
  void swap( vector& other ) noexcept
  {
    if constexpr (ThreadSafe) 
    {
      mcrl2::utilities::lock_guard guard = detail::g_thread_term_pool().lock();
      super::swap(other); // Invalidates end() so must be protected.
    } 
    else 
    {
      mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
      super::swap(other); // Invalidates end() so must be protected.
    }
  }
 
  std::size_t size() const
  {
    // Concurrent read/write on the size.
    mcrl2::utilities::shared_guard guard = detail::g_thread_term_pool().lock_shared();
    return super::size();
  }
};
 
} // namespace atermpp
#endif // MCRL2_ATERMPP_STANDARD_CONTAINER_VECTOR_H