native_hash_set.hpp [plain text]
#ifndef PB_DS_NATIVE_HASH_SET_HPP
#define PB_DS_NATIVE_HASH_SET_HPP
#include <string>
#include <ext/pb_ds/detail/type_utils.hpp>
#include <ext/pb_ds/detail/standard_policies.hpp>
#include <native_type/assoc/native_hash_tag.hpp>
#include <io/xml.hpp>
#define PB_DS_USE_TR1 1
#ifdef PB_DS_USE_TR1
#include <tr1/unordered_set>
#else
#include <ext/hash_set>
#endif
namespace pb_ds
{
namespace test
{
#ifdef PB_DS_USE_TR1
#define PB_DS_BASE_C_DEC \
std::tr1::unordered_set<Key, Hash_Fn, Eq_Fn, \
typename Allocator::template rebind<Key>::other>
#else
#define PB_DS_BASE_C_DEC \
__gnu_cxx::hash_set<Key, Hash_Fn, Eq_Fn, \
typename Allocator::template rebind<Key>::other>
#endif
template<typename Key,
size_t Init_Size = 8,
typename Hash_Fn = typename pb_ds::detail::default_hash_fn<Key>::type,
typename Eq_Fn = std::equal_to<Key>,
typename Less_Fn = std::less<Key>,
typename Allocator = std::allocator<char>
#ifdef PB_DS_USE_TR1
, bool Cache_Hash = false
#endif
>
class native_hash_set : public PB_DS_BASE_C_DEC
{
private:
typedef PB_DS_BASE_C_DEC base_type;
public:
typedef native_hash_tag container_category;
native_hash_set() : base_type(Init_Size) { }
template<typename It>
native_hash_set(It f, It l) : base_type(f, l) { }
template<typename Pred>
inline size_t
erase_if(Pred pred)
{
size_t ersd = 0;
bool done = false;
while (!done)
{
typename base_type::iterator b_it = base_type::begin();
typename base_type::iterator e_it = base_type::end();
done = true;
while (b_it != e_it)
{
if (pred(*b_it))
{
++ersd;
done = false;
base_type::erase(*b_it);
b_it = e_it;
}
else
++b_it;
}
}
return ersd;
}
static std::string
name()
{
#ifdef PB_DS_USE_TR1
return std::string("n_hash_set_")
+ (Cache_Hash ? std::string("cah") : std::string("ncah"));
#else
return std::string("n_hash_set_ncah");
#endif
}
static std::string
desc()
{
#ifdef PB_DS_USE_TR1
const std::string cache_hash_desc =
make_xml_tag("cache_hash_code", "value",
Cache_Hash ? std::string("true") : std::string("false"));
return make_xml_tag("type", "value", "std_tr1_unordered_set",
cache_hash_desc);
#else
return make_xml_tag("type", "value", "__gnucxx_hash_set");
#endif
}
};
#undef PB_DS_BASE_C_DEC
} }
#endif