gp_ht_map_.hpp

Go to the documentation of this file.

// -*- C++ -*-

// Copyright (C) 2005-2013 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the terms
// of the GNU General Public License as published by the Free Software
// Foundation; either version 3, or (at your option) any later
// version.

// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// General Public License for more details.

// Under Section 7 of GPL version 3, you are granted additional
// permissions described in the GCC Runtime Library Exception, version
// 3.1, as published by the Free Software Foundation.

// You should have received a copy of the GNU General Public License and
// a copy of the GCC Runtime Library Exception along with this program;
// see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
// <http://www.gnu.org/licenses/>.

// Copyright (C) 2004 Ami Tavory and Vladimir Dreizin, IBM-HRL.

// Permission to use, copy, modify, sell, and distribute this software
// is hereby granted without fee, provided that the above copyright
// notice appears in all copies, and that both that copyright notice
// and this permission notice appear in supporting documentation. None
// of the above authors, nor IBM Haifa Research Laboratories, make any
// representation about the suitability of this software for any
// purpose. It is provided "as is" without express or implied
// warranty.

#include <ext/pb_ds/tag_and_trait.hpp>
#include <ext/pb_ds/detail/hash_fn/ranged_probe_fn.hpp>
#include <ext/pb_ds/detail/types_traits.hpp>
#include <ext/pb_ds/exception.hpp>
#include <ext/pb_ds/detail/eq_fn/hash_eq_fn.hpp>
#include <utility>
#ifdef PB_DS_HT_MAP_TRACE_
#include <iostream>
#endif
#ifdef _GLIBCXX_DEBUG
#include <ext/pb_ds/detail/debug_map_base.hpp>
#endif
#include <debug/debug.h>

namespace __gnu_pbds
{
  namespace detail
  {
#ifdef PB_DS_DATA_TRUE_INDICATOR
#define PB_DS_GP_HASH_NAME gp_ht_map
#endif

#ifdef PB_DS_DATA_FALSE_INDICATOR
#define PB_DS_GP_HASH_NAME gp_ht_set
#endif

#define PB_DS_CLASS_T_DEC \
   template<typename Key, typename Mapped, typename Hash_Fn, typename Eq_Fn, \
        typename _Alloc, bool Store_Hash, typename Comb_Probe_Fn, \
        typename Probe_Fn,  typename Resize_Policy>

#define PB_DS_CLASS_C_DEC \
   PB_DS_GP_HASH_NAME<Key, Mapped, Hash_Fn, Eq_Fn, _Alloc, \
            Store_Hash, Comb_Probe_Fn, Probe_Fn, Resize_Policy>

#define PB_DS_HASH_EQ_FN_C_DEC \
    hash_eq_fn<Key, Eq_Fn, _Alloc, Store_Hash>

#define PB_DS_RANGED_PROBE_FN_C_DEC \
   ranged_probe_fn<Key, Hash_Fn, _Alloc, Comb_Probe_Fn, Probe_Fn, Store_Hash>

#define PB_DS_GP_HASH_TRAITS_BASE \
   types_traits<Key, Mapped, _Alloc, Store_Hash>

#ifdef _GLIBCXX_DEBUG
#define PB_DS_DEBUG_MAP_BASE_C_DEC \
   debug_map_base<Key, Eq_Fn, \
          typename _Alloc::template rebind<Key>::other::const_reference>
#endif


    template<typename Key,
         typename Mapped,
         typename Hash_Fn,
         typename Eq_Fn,
         typename _Alloc,
         bool Store_Hash,
         typename Comb_Probe_Fn,
         typename Probe_Fn,
         typename Resize_Policy>
    class PB_DS_GP_HASH_NAME :
#ifdef _GLIBCXX_DEBUG
      protected PB_DS_DEBUG_MAP_BASE_C_DEC,
#endif
      public PB_DS_HASH_EQ_FN_C_DEC,
      public Resize_Policy,
      public PB_DS_RANGED_PROBE_FN_C_DEC,
      public PB_DS_GP_HASH_TRAITS_BASE
    {
    private:
      typedef PB_DS_GP_HASH_TRAITS_BASE         traits_base;
      typedef typename traits_base::value_type  value_type_;
      typedef typename traits_base::pointer     pointer_;
      typedef typename traits_base::const_pointer const_pointer_;
      typedef typename traits_base::reference   reference_;
      typedef typename traits_base::const_reference const_reference_;
      typedef typename traits_base::comp_hash   comp_hash;

      enum entry_status
    {
      empty_entry_status,
      valid_entry_status,
      erased_entry_status
    } __attribute__ ((packed));

      struct entry : public traits_base::stored_data_type
      {
    entry_status m_stat;
      };

      typedef typename _Alloc::template rebind<entry>::other entry_allocator;
      typedef typename entry_allocator::pointer entry_pointer;
      typedef typename entry_allocator::const_pointer const_entry_pointer;
      typedef typename entry_allocator::reference entry_reference;
      typedef typename entry_allocator::const_reference const_entry_reference;
      typedef typename entry_allocator::pointer entry_array;

      typedef PB_DS_RANGED_PROBE_FN_C_DEC   ranged_probe_fn_base;

#ifdef _GLIBCXX_DEBUG
      typedef PB_DS_DEBUG_MAP_BASE_C_DEC    debug_base;
#endif

      typedef PB_DS_HASH_EQ_FN_C_DEC        hash_eq_fn_base;
      typedef Resize_Policy             resize_base;

#define PB_DS_GEN_POS typename _Alloc::size_type

#include <ext/pb_ds/detail/unordered_iterator/point_const_iterator.hpp>
#include <ext/pb_ds/detail/unordered_iterator/point_iterator.hpp>
#include <ext/pb_ds/detail/unordered_iterator/const_iterator.hpp>
#include <ext/pb_ds/detail/unordered_iterator/iterator.hpp>

#undef PB_DS_GEN_POS

    public:
      typedef _Alloc                allocator_type;
      typedef typename _Alloc::size_type    size_type;
      typedef typename _Alloc::difference_type  difference_type;
      typedef Hash_Fn               hash_fn;
      typedef Eq_Fn                 eq_fn;
      typedef Probe_Fn              probe_fn;
      typedef Comb_Probe_Fn             comb_probe_fn;
      typedef Resize_Policy             resize_policy;

      enum
    {
      store_hash = Store_Hash
    };

      typedef typename traits_base::key_type    key_type;
      typedef typename traits_base::key_pointer key_pointer;
      typedef typename traits_base::key_const_pointer key_const_pointer;
      typedef typename traits_base::key_reference key_reference;
      typedef typename traits_base::key_const_reference key_const_reference;
      typedef typename traits_base::mapped_type mapped_type;
      typedef typename traits_base::mapped_pointer mapped_pointer;
      typedef typename traits_base::mapped_const_pointer mapped_const_pointer;
      typedef typename traits_base::mapped_reference mapped_reference;
      typedef typename traits_base::mapped_const_reference mapped_const_reference;
      typedef typename traits_base::value_type value_type;
      typedef typename traits_base::pointer pointer;
      typedef typename traits_base::const_pointer const_pointer;
      typedef typename traits_base::reference reference;
      typedef typename traits_base::const_reference const_reference;

#ifdef PB_DS_DATA_TRUE_INDICATOR
      typedef point_iterator_           point_iterator;
#endif

#ifdef PB_DS_DATA_FALSE_INDICATOR
      typedef point_const_iterator_         point_iterator;
#endif

      typedef point_const_iterator_         point_const_iterator;

#ifdef PB_DS_DATA_TRUE_INDICATOR
      typedef iterator_             iterator;
#endif

#ifdef PB_DS_DATA_FALSE_INDICATOR
      typedef const_iterator_           iterator;
#endif

      typedef const_iterator_           const_iterator;

      PB_DS_GP_HASH_NAME();

      PB_DS_GP_HASH_NAME(const PB_DS_CLASS_C_DEC&);

      PB_DS_GP_HASH_NAME(const Hash_Fn&);

      PB_DS_GP_HASH_NAME(const Hash_Fn&, const Eq_Fn&);

      PB_DS_GP_HASH_NAME(const Hash_Fn&, const Eq_Fn&, const Comb_Probe_Fn&);

      PB_DS_GP_HASH_NAME(const Hash_Fn&, const Eq_Fn&, const Comb_Probe_Fn&,
             const Probe_Fn&);

      PB_DS_GP_HASH_NAME(const Hash_Fn&, const Eq_Fn&, const Comb_Probe_Fn&,
             const Probe_Fn&, const Resize_Policy&);

      template<typename It>
      void
      copy_from_range(It, It);

      virtual
      ~PB_DS_GP_HASH_NAME();

      void
      swap(PB_DS_CLASS_C_DEC&);

      inline size_type
      size() const;

      inline size_type
      max_size() const;

      inline bool
      empty() const;

      Hash_Fn&
      get_hash_fn();

      const Hash_Fn&
      get_hash_fn() const;

      Eq_Fn&
      get_eq_fn();

      const Eq_Fn&
      get_eq_fn() const;

      Probe_Fn&
      get_probe_fn();

      const Probe_Fn&
      get_probe_fn() const;

      Comb_Probe_Fn&
      get_comb_probe_fn();

      const Comb_Probe_Fn&
      get_comb_probe_fn() const;

      Resize_Policy&
      get_resize_policy();

      const Resize_Policy&
      get_resize_policy() const;

      inline std::pair<point_iterator, bool>
      insert(const_reference r_val)
      {
       _GLIBCXX_DEBUG_ONLY(PB_DS_CLASS_C_DEC::assert_valid(__FILE__, __LINE__);)
    return insert_imp(r_val, traits_base::m_store_extra_indicator);
      }

      inline mapped_reference
      operator[](key_const_reference r_key)
      {
#ifdef PB_DS_DATA_TRUE_INDICATOR
    return subscript_imp(r_key, traits_base::m_store_extra_indicator);
#else
    insert(r_key);
    return traits_base::s_null_type;
#endif
      }

      inline point_iterator
      find(key_const_reference);

      inline point_const_iterator
      find(key_const_reference) const;

      inline point_iterator
      find_end();

      inline point_const_iterator
      find_end() const;

      inline bool
      erase(key_const_reference);

      template<typename Pred>
        inline size_type
        erase_if(Pred);

      void
      clear();

      inline iterator
      begin();

      inline const_iterator
      begin() const;

      inline iterator
      end();
      
      inline const_iterator
      end() const;

#ifdef _GLIBCXX_DEBUG
      void
      assert_valid(const char*, int) const;
#endif

#ifdef PB_DS_HT_MAP_TRACE_
      void
      trace() const;
#endif

    private:
#ifdef PB_DS_DATA_TRUE_INDICATOR
      friend class iterator_;
#endif

      friend class const_iterator_;

      void
      deallocate_all();

      void
      initialize();

      void
      erase_all_valid_entries(entry_array, size_type);

      inline bool
      do_resize_if_needed();

      inline void
      do_resize_if_needed_no_throw();

      void
      resize_imp(size_type);

      virtual void
      do_resize(size_type);

      void
      resize_imp(entry_array, size_type);

      inline void
      resize_imp_reassign(entry_pointer, entry_array, false_type);

      inline void
      resize_imp_reassign(entry_pointer, entry_array, true_type);

      inline size_type
      find_ins_pos(key_const_reference, false_type);

      inline comp_hash
      find_ins_pos(key_const_reference, true_type);

      inline std::pair<point_iterator, bool>
      insert_imp(const_reference, false_type);

      inline std::pair<point_iterator, bool>
      insert_imp(const_reference, true_type);

      inline pointer
      insert_new_imp(const_reference r_val, size_type pos)
      {
    _GLIBCXX_DEBUG_ASSERT(m_entries[pos].m_stat != valid_entry_status);

    if (do_resize_if_needed())
      pos = find_ins_pos(PB_DS_V2F(r_val),
                 traits_base::m_store_extra_indicator);

    _GLIBCXX_DEBUG_ASSERT(m_entries[pos].m_stat != valid_entry_status);
    entry* const p_e = m_entries + pos;
    new (&p_e->m_value) value_type(r_val);
    p_e->m_stat = valid_entry_status;
    resize_base::notify_inserted(++m_num_used_e);

    _GLIBCXX_DEBUG_ONLY(debug_base::insert_new(PB_DS_V2F(p_e->m_value));)
    _GLIBCXX_DEBUG_ONLY(assert_valid(__FILE__, __LINE__);)
    return &p_e->m_value;
      }

      inline pointer
      insert_new_imp(const_reference r_val, comp_hash& r_pos_hash_pair)
      {
    _GLIBCXX_DEBUG_ASSERT(m_entries[r_pos_hash_pair.first].m_stat !=
                  valid_entry_status);

    if (do_resize_if_needed())
      r_pos_hash_pair = find_ins_pos(PB_DS_V2F(r_val),
                     traits_base::m_store_extra_indicator);

    _GLIBCXX_DEBUG_ASSERT(m_entries[r_pos_hash_pair.first].m_stat !=
                  valid_entry_status);

    entry* const p_e = m_entries + r_pos_hash_pair.first;
    new (&p_e->m_value) value_type(r_val);
    p_e->m_hash = r_pos_hash_pair.second;
    p_e->m_stat = valid_entry_status;

    resize_base::notify_inserted(++m_num_used_e);

    _GLIBCXX_DEBUG_ONLY(debug_base::insert_new(PB_DS_V2F(p_e->m_value));)
    _GLIBCXX_DEBUG_ONLY(assert_valid(__FILE__, __LINE__);)
    return &p_e->m_value;
      }

#ifdef PB_DS_DATA_TRUE_INDICATOR
      inline mapped_reference
      subscript_imp(key_const_reference key, false_type)
      {
    _GLIBCXX_DEBUG_ONLY(assert_valid(__FILE__, __LINE__);)

    const size_type pos = find_ins_pos(key,
                     traits_base::m_store_extra_indicator);

    entry_pointer p_e = &m_entries[pos];
    if (p_e->m_stat != valid_entry_status)
      return insert_new_imp(value_type(key, mapped_type()), pos)->second;

    PB_DS_CHECK_KEY_EXISTS(key)
    return p_e->m_value.second;
      }

      inline mapped_reference
      subscript_imp(key_const_reference key, true_type)
      {
    _GLIBCXX_DEBUG_ONLY(assert_valid(__FILE__, __LINE__);)

    comp_hash pos_hash_pair = find_ins_pos(key,
                     traits_base::m_store_extra_indicator);

    if (m_entries[pos_hash_pair.first].m_stat != valid_entry_status)
      return insert_new_imp(value_type(key, mapped_type()),
                 pos_hash_pair)->second;

    PB_DS_CHECK_KEY_EXISTS(key)
    return (m_entries + pos_hash_pair.first)->m_value.second;
      }
#endif

      inline pointer
      find_key_pointer(key_const_reference key, false_type)
      {
    const size_type hash = ranged_probe_fn_base::operator()(key);
    resize_base::notify_find_search_start();

    // Loop until entry is found or until all possible entries accessed.
    for (size_type i = 0; i < m_num_e; ++i)
      {
        const size_type pos = ranged_probe_fn_base::operator()(key,
                                   hash, i);

        entry* const p_e = m_entries + pos;
        switch (p_e->m_stat)
          {
          case empty_entry_status:
        {
          resize_base::notify_find_search_end();
          PB_DS_CHECK_KEY_DOES_NOT_EXIST(key)
          return 0;
        }
        break;
          case valid_entry_status:
        if (hash_eq_fn_base::operator()(PB_DS_V2F(p_e->m_value), key))
          {
            resize_base::notify_find_search_end();
            PB_DS_CHECK_KEY_EXISTS(key)
            return pointer(&p_e->m_value);
          }
        break;
          case erased_entry_status:
        break;
          default:
        _GLIBCXX_DEBUG_ASSERT(0);
          };

        resize_base::notify_find_search_collision();
      }

    PB_DS_CHECK_KEY_DOES_NOT_EXIST(key)
    resize_base::notify_find_search_end();
    return 0;
      }

      inline pointer
      find_key_pointer(key_const_reference key, true_type)
      {
    comp_hash pos_hash_pair = ranged_probe_fn_base::operator()(key);
    resize_base::notify_find_search_start();

    // Loop until entry is found or until all possible entries accessed.
    for (size_type i = 0; i < m_num_e; ++i)
      {
        const size_type pos =
          ranged_probe_fn_base::operator()(key, pos_hash_pair.second, i);

        entry* const p_e = m_entries + pos;

        switch(p_e->m_stat)
          {
          case empty_entry_status:
        {
          resize_base::notify_find_search_end();
          PB_DS_CHECK_KEY_DOES_NOT_EXIST(key)
          return 0;
        }
        break;
          case valid_entry_status:
        if (hash_eq_fn_base::operator()(PB_DS_V2F(p_e->m_value),
                        p_e->m_hash,
                        key, pos_hash_pair.second))
          {
            resize_base::notify_find_search_end();
            PB_DS_CHECK_KEY_EXISTS(key)
            return pointer(&p_e->m_value);
          }
        break;
          case erased_entry_status:
        break;
          default:
        _GLIBCXX_DEBUG_ASSERT(0);
          };

        resize_base::notify_find_search_collision();
      }

    PB_DS_CHECK_KEY_DOES_NOT_EXIST(key)
    resize_base::notify_find_search_end();
    return 0;
      }

      inline bool
      erase_imp(key_const_reference, true_type);

      inline bool
      erase_imp(key_const_reference, false_type);

      inline void
      erase_entry(entry_pointer);

#ifdef PB_DS_DATA_TRUE_INDICATOR
      void
      inc_it_state(pointer& r_p_value, size_type& r_pos) const
      { inc_it_state((mapped_const_pointer& )r_p_value, r_pos); }
#endif

      void
      inc_it_state(const_pointer& r_p_value, size_type& r_pos) const
      {
    _GLIBCXX_DEBUG_ASSERT(r_p_value != 0);
    for (++r_pos; r_pos < m_num_e; ++r_pos)
      {
        const_entry_pointer p_e =& m_entries[r_pos];
        if (p_e->m_stat == valid_entry_status)
          {
        r_p_value =& p_e->m_value;
        return;
          }
      }
    r_p_value = 0;
      }

      void
      get_start_it_state(const_pointer& r_p_value, size_type& r_pos) const
      {
    for (r_pos = 0; r_pos < m_num_e; ++r_pos)
      {
        const_entry_pointer p_e = &m_entries[r_pos];
        if (p_e->m_stat == valid_entry_status)
          {
        r_p_value = &p_e->m_value;
        return;
          }
      }
    r_p_value = 0;
      }

      void
      get_start_it_state(pointer& r_p_value, size_type& r_pos)
      {
    for (r_pos = 0; r_pos < m_num_e; ++r_pos)
      {
        entry_pointer p_e = &m_entries[r_pos];
        if (p_e->m_stat == valid_entry_status)
          {
        r_p_value = &p_e->m_value;
        return;
          }
      }
    r_p_value = 0;
      }

#ifdef _GLIBCXX_DEBUG
      void
      assert_entry_array_valid(const entry_array, false_type,
                   const char*, int) const;

      void
      assert_entry_array_valid(const entry_array, true_type,
                   const char*, int) const;
#endif

      static entry_allocator    s_entry_allocator;
      static iterator       s_end_it;
      static const_iterator     s_const_end_it;

      size_type         m_num_e;
      size_type         m_num_used_e;
      entry_pointer         m_entries;

      enum
    {
      store_hash_ok = !Store_Hash
              || !is_same<Hash_Fn, __gnu_pbds::null_type>::value
    };

      PB_DS_STATIC_ASSERT(sth, store_hash_ok);
    };

#include <ext/pb_ds/detail/gp_hash_table_map_/constructor_destructor_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/find_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/resize_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/debug_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/info_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/policy_access_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/erase_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/iterator_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/insert_fn_imps.hpp>
#include <ext/pb_ds/detail/gp_hash_table_map_/trace_fn_imps.hpp>

#undef PB_DS_CLASS_T_DEC
#undef PB_DS_CLASS_C_DEC
#undef PB_DS_HASH_EQ_FN_C_DEC
#undef PB_DS_RANGED_PROBE_FN_C_DEC
#undef PB_DS_GP_HASH_TRAITS_BASE
#undef PB_DS_DEBUG_MAP_BASE_C_DEC
#undef PB_DS_GP_HASH_NAME
  } // namespace detail
} // namespace __gnu_pbds