MapOfSets.h

Go to the documentation of this file.

// Copyright (c) 2009-2014 Alan Wright. All rights reserved.
// Distributable under the terms of either the Apache License (Version 2.0)
// or the GNU Lesser General Public License.
 
#ifndef MAPOFSETS_H
#define MAPOFSETS_H
 
#include "Lucene.h"
 
namespace Lucene {

template <class MAPKEY, class MAPHASH, class MAPEQUAL, class SETVALUE, class SETHASH, class SETEQUAL>
class MapOfSets {
public:
    typedef HashSet<SETVALUE, SETHASH, SETEQUAL> set_type;
    typedef HashMap<MAPKEY, set_type, MAPHASH, MAPEQUAL> map_type;
 
    MapOfSets(map_type m) {
        theMap = m;
    }
 
protected:
    map_type theMap;
 
public:
    map_type getMap() {
        return theMap;
    }

    int32_t put(MAPKEY key, SETVALUE val) {
        typename map_type::iterator entry = theMap.find(key);
        if (entry != theMap.end()) {
            entry->second.add(val);
            return entry->second.size();
        } else {
            set_type theSet(set_type::newInstance());
            theSet.add(val);
            theMap.put(key, theSet);
            return 1;
        }
    }

    int32_t putAll(MAPKEY key, set_type vals) {
        typename map_type::iterator entry = theMap.find(key);
        if (entry != theMap.end()) {
            entry->second.addAll(vals.begin(), vals.end());
            return entry->second.size();
        } else {
            set_type theSet(set_type::newInstance(vals.begin(), vals.end()));
            theMap.put(key, theSet);
            return theSet.size();
        }
    }
};
 
}
 
#endif