boosthm.h

   1 /*
   2 Project: J-Ben
   3 Author:  Paul Goins
   4 Website: http://www.vultaire.net/software/jben/
   5 License: GNU General Public License (GPL) version 2
   6          (http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt)
   7
   8 File: boosthm.h
   9
  10 This program is free software; you can redistribute it and/or modify
  11 it under the terms of the GNU General Public License as published by
  12 the Free Software Foundation; either version 2 of the License, or
  13 (at your option) any later version.
  14
  15 This program is distributed in the hope that it will be useful,
  16 but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 GNU General Public License for more details.
  19
  20 You should have received a copy of the GNU General Public License
  21 along with this program.  If not, see <http://www.gnu.org/licenses/>
  22 */
  23
  24 #ifndef boosthm_h
  25 #define boosthm_h
  26
  27 #define BOOST_MULTI_INDEX_DISABLE_SERIALIZATION
  28
  29 #include <boost/multi_index_container.hpp>
  30 #include <boost/multi_index/hashed_index.hpp>
  31 #include <boost/multi_index/member.hpp>
  32
  33 using namespace std;
  34 using namespace boost;
  35 using namespace boost::multi_index;
  36
  37 template<typename key, typename value>
  38 class BoostHM : public
  39         multi_index_container<
  40                 pair<key, value>,
  41                 indexed_by<
  42                         hashed_unique<
  43                                 member<
  44                                         pair<key, value>,
  45                                         key,
  46                                         &pair<key, value>::first
  47                                 >
  48                         >
  49                 >
  50         > {
  51 public:
  52         const value& operator[](const key& k);
  53         const value& operator[](const key& k) const;
  54         bool assign(const key& k, const value& v);
  55 private:
  56 };
  57
  58 /* NOTE: These two operator[] overloads do not handle non-existant keys!
  59    Use with caution! */
  60 template<typename key, typename value>
  61 const value& BoostHM<key,value>::operator[](const key& k) {
  62         return this->find(k)->second;
  63 }
  64
  65 template<typename key, typename value>
  66 const value& BoostHM<key,value>::operator[](const key& k) const {
  67         return this->find(k)->second;
  68 }
  69
  70 template<typename key, typename value>
  71 bool BoostHM<key,value>::assign(const key& k, const value& v) {
  72         pair<key,value> newEntry(k,v);
  73
  74         pair<typename BoostHM<key,value>::iterator, bool> p = this->insert(newEntry);
  75         if(!p.second) {
  76                 if(!this->replace(p.first, newEntry)) return false;
  77         }
  78         return true;
  79 }
  80
  81 #endif