cds/container/striped_map/std_map.h

   1 //$$CDS-header$$
   2
   3 #ifndef __CDS_CONTAINER_STRIPED_MAP_STD_MAP_ADAPTER_H
   4 #define __CDS_CONTAINER_STRIPED_MAP_STD_MAP_ADAPTER_H
   5
   6 #include <cds/container/striped_set/adapter.h>
   7 #include <map>
   8
   9 //@cond
  10 namespace cds { namespace container {
  11     namespace striped_set {
  12
  13         // Copy policy for map
  14         template <typename Key, typename T, typename Traits, typename Alloc>
  15         struct copy_item_policy< std::map< Key, T, Traits, Alloc > >
  16         {
  17             typedef std::map< Key, T, Traits, Alloc > map_type;
  18             typedef typename map_type::value_type pair_type;
  19             typedef typename map_type::iterator    iterator;
  20
  21             void operator()( map_type& map, iterator itWhat )
  22             {
  23                 map.insert( *itWhat );
  24             }
  25         };
  26
  27         // Swap item policy
  28         template <typename Key, typename T, typename Traits, typename Alloc>
  29         struct swap_item_policy< std::map< Key, T, Traits, Alloc > >
  30         {
  31             typedef std::map< Key, T, Traits, Alloc > map_type;
  32             typedef typename map_type::value_type pair_type;
  33             typedef typename map_type::iterator    iterator;
  34
  35             void operator()( map_type& map, iterator itWhat )
  36             {
  37                 std::pair< typename map_type::iterator, bool > ret = map.insert( pair_type( itWhat->first, typename pair_type::second_type() ));
  38                 assert( ret.second )    ;   // successful insertion
  39                 std::swap( ret.first->second, itWhat->second );
  40             }
  41         };
  42
  43         // Move policy for map
  44         template <typename Key, typename T, typename Traits, typename Alloc>
  45         struct move_item_policy< std::map< Key, T, Traits, Alloc > >
  46         {
  47             typedef std::map< Key, T, Traits, Alloc > map_type;
  48             typedef typename map_type::value_type pair_type;
  49             typedef typename map_type::iterator    iterator;
  50
  51             void operator()( map_type& map, iterator itWhat  )
  52             {
  53                 map.insert( std::move( *itWhat ) );
  54             }
  55         };
  56     }   // namespace striped_set
  57 }} // namespace cds::container
  58
  59 namespace cds { namespace intrusive { namespace striped_set {
  60
  61     /// std::set adapter for hash set bucket
  62     template <typename Key, typename T, class Traits, class Alloc, CDS_SPEC_OPTIONS>
  63     class adapt< std::map< Key, T, Traits, Alloc>, CDS_OPTIONS >
  64     {
  65     public:
  66         typedef std::map< Key, T, Traits, Alloc>     container_type          ;   ///< underlying container type
  67
  68     private:
  69         /// Adapted container type
  70         class adapted_container: public cds::container::striped_set::adapted_container
  71         {
  72         public:
  73             typedef typename container_type::value_type     value_type  ;   ///< value type stored in the container
  74             typedef typename container_type::key_type       key_type;
  75             typedef typename container_type::mapped_type    mapped_type;
  76             typedef typename container_type::iterator       iterator ;   ///< container iterator
  77             typedef typename container_type::const_iterator const_iterator ;    ///< container const iterator
  78
  79             static bool const has_find_with = false;
  80             static bool const has_erase_with = false;
  81
  82         private:
  83             //@cond
  84             typedef typename cds::opt::select<
  85                 typename cds::opt::value<
  86                     typename cds::opt::find_option<
  87                         cds::opt::copy_policy< cds::container::striped_set::move_item >
  88                         , CDS_OPTIONS
  89                     >::type
  90                 >::copy_policy
  91                 , cds::container::striped_set::copy_item, cds::container::striped_set::copy_item_policy<container_type>
  92                 , cds::container::striped_set::swap_item, cds::container::striped_set::swap_item_policy<container_type>
  93                 , cds::container::striped_set::move_item, cds::container::striped_set::move_item_policy<container_type>
  94             >::type copy_item;
  95             //@endcond
  96
  97         private:
  98             //@cond
  99             container_type  m_Map;
 100             //@endcond
 101
 102         public:
 103
 104             template <typename Q, typename Func>
 105             bool insert( const Q& key, Func f )
 106             {
 107                 std::pair<iterator, bool> res = m_Map.insert( value_type( key, mapped_type() ) );
 108                 if ( res.second )
 109                     ::cds::unref(f)( *res.first );
 110                 return res.second;
 111             }
 112
 113             template <typename Q, typename... Args>
 114             bool emplace( Q&& key, Args&&... args )
 115             {
 116                 std::pair<iterator, bool> res = m_Map.emplace( std::forward<Q>(key), std::move(mapped_type( std::forward<Args>(args)...)));
 117                 return res.second;
 118             }
 119
 120             template <typename Q, typename Func>
 121             std::pair<bool, bool> ensure( const Q& key, Func func )
 122             {
 123                 std::pair<iterator, bool> res = m_Map.insert( value_type( key, mapped_type() ));
 124                 cds::unref(func)( res.second, *res.first );
 125                 return std::make_pair( true, res.second );
 126             }
 127
 128             template <typename Q, typename Func>
 129             bool erase( const Q& key, Func f )
 130             {
 131                 iterator it = m_Map.find( key_type(key) );
 132                 if ( it == m_Map.end() )
 133                     return false;
 134                 cds::unref(f)( *it );
 135                 m_Map.erase( it );
 136                 return true;
 137             }
 138
 139             template <typename Q, typename Func>
 140             bool find( Q& val, Func f )
 141             {
 142                 iterator it = m_Map.find( key_type(val) );
 143                 if ( it == m_Map.end() )
 144                     return false;
 145                 cds::unref(f)( *it, val );
 146                 return true;
 147             }
 148
 149             /// Clears the container
 150             void clear()
 151             {
 152                 m_Map.clear();
 153             }
 154
 155             iterator begin()                { return m_Map.begin(); }
 156             const_iterator begin() const    { return m_Map.begin(); }
 157             iterator end()                  { return m_Map.end(); }
 158             const_iterator end() const      { return m_Map.end(); }
 159
 160             void move_item( adapted_container& /*from*/, iterator itWhat )
 161             {
 162                 assert( m_Map.find( itWhat->first ) == m_Map.end() );
 163                 copy_item()( m_Map, itWhat );
 164             }
 165
 166             size_t size() const
 167             {
 168                 return m_Map.size();
 169             }
 170         };
 171
 172     public:
 173         typedef adapted_container type ; ///< Result of \p adapt metafunction
 174     };
 175 }}} // namespace cds::intrusive::striped_set
 176
 177 //@endcond
 178
 179 #endif // #ifndef __CDS_CONTAINER_STRIPED_MAP_STD_MAP_ADAPTER_H