Merge pull request #299 from alliedmodders/amtl-upstream
Use upstream AMTL as a submodule.
This commit is contained in:
commit
7f24f137d6
3
.gitmodules
vendored
Normal file
3
.gitmodules
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
[submodule "public/amtl"]
|
||||
path = public/amtl
|
||||
url = https://github.com/alliedmodders/amtl
|
@ -326,7 +326,7 @@ class SMConfig(object):
|
||||
os.path.join(builder.sourcePath, 'public'),
|
||||
os.path.join(builder.sourcePath, 'public', 'extensions'),
|
||||
os.path.join(builder.sourcePath, 'sourcepawn', 'include'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl', 'include'),
|
||||
]
|
||||
return compiler
|
||||
|
||||
|
@ -48,19 +48,19 @@ public:
|
||||
style = 0;
|
||||
access = 0;
|
||||
}
|
||||
CItem(ke::Moveable<CItem> other)
|
||||
: info(ke::Move(other->info)),
|
||||
display(ke::Move(other->display))
|
||||
CItem(CItem &&other)
|
||||
: info(ke::Move(other.info)),
|
||||
display(ke::Move(other.display))
|
||||
{
|
||||
style = other->style;
|
||||
access = other->access;
|
||||
style = other.style;
|
||||
access = other.access;
|
||||
}
|
||||
CItem & operator =(ke::Moveable<CItem> other)
|
||||
CItem & operator =(CItem &&other)
|
||||
{
|
||||
info = ke::Move(other->info);
|
||||
display = ke::Move(other->display);
|
||||
style = other->style;
|
||||
access = other->access;
|
||||
info = ke::Move(other.info);
|
||||
display = ke::Move(other.display);
|
||||
style = other.style;
|
||||
access = other.access;
|
||||
return *this;
|
||||
}
|
||||
|
||||
|
@ -6,7 +6,7 @@ binary.compiler.cxxincludes += [
|
||||
os.path.join(builder.sourcePath, 'core', 'logic'),
|
||||
os.path.join(builder.sourcePath, 'public'),
|
||||
os.path.join(builder.sourcePath, 'sourcepawn', 'include'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl', 'include'),
|
||||
os.path.join(SM.mms_root, 'core', 'sourcehook')
|
||||
]
|
||||
binary.compiler.defines += [
|
||||
|
@ -63,11 +63,11 @@ public:
|
||||
: control_(0)
|
||||
{
|
||||
}
|
||||
Entry(ke::Moveable<Entry> other)
|
||||
Entry(Entry &&other)
|
||||
{
|
||||
control_ = other->control_;
|
||||
data_ = other->data_;
|
||||
other->control_ = 0;
|
||||
control_ = other.control_;
|
||||
data_ = other.data_;
|
||||
other.control_ = 0;
|
||||
}
|
||||
|
||||
~Entry()
|
||||
|
@ -6,7 +6,7 @@
|
||||
#include <IBinTools.h>
|
||||
#include <convar.h>
|
||||
#include <sh_list.h>
|
||||
#include <amtl/am-vector.h>
|
||||
#include <am-vector.h>
|
||||
#include <vtable_hook_helper.h>
|
||||
|
||||
#include <iplayerinfo.h>
|
||||
|
@ -35,7 +35,7 @@
|
||||
class CUserCmd;
|
||||
|
||||
#include "extension.h"
|
||||
#include "amtl/am-vector.h"
|
||||
#include <am-vector.h>
|
||||
#include "vtable_hook_helper.h"
|
||||
|
||||
class CHookManager : IPluginsListener, IFeatureProvider
|
||||
|
1
public/amtl
Submodule
1
public/amtl
Submodule
@ -0,0 +1 @@
|
||||
Subproject commit 1bb5196ce39ff3ed59445e51f9223734d4bcf886
|
@ -1,65 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_allocatorpolicies_h_
|
||||
#define _include_amtl_allocatorpolicies_h_
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// The default system allocator policy will crash on out-of-memory.
|
||||
class SystemAllocatorPolicy
|
||||
{
|
||||
public:
|
||||
void reportOutOfMemory() {
|
||||
fprintf(stderr, "OUT OF MEMORY\n");
|
||||
abort();
|
||||
}
|
||||
void reportAllocationOverflow() {
|
||||
fprintf(stderr, "OUT OF MEMORY\n");
|
||||
abort();
|
||||
}
|
||||
|
||||
public:
|
||||
void free(void *memory) {
|
||||
::free(memory);
|
||||
}
|
||||
void *malloc(size_t bytes) {
|
||||
void *ptr = ::malloc(bytes);
|
||||
if (!ptr)
|
||||
reportOutOfMemory();
|
||||
return ptr;
|
||||
}
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_allocatorpolicies_h_
|
@ -1,102 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_atomics_h_
|
||||
#define _include_amtl_atomics_h_
|
||||
|
||||
#include <am-utility.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
#if defined(_MSC_VER)
|
||||
extern "C" {
|
||||
long __cdecl _InterlockedIncrement(long volatile *dest);
|
||||
long __cdecl _InterlockedDecrement(long volatile *dest);
|
||||
}
|
||||
# pragma intrinsic(_InterlockedIncrement)
|
||||
# pragma intrinsic(_InterlockedDecrement)
|
||||
#endif
|
||||
|
||||
template <size_t Width>
|
||||
struct AtomicOps;
|
||||
|
||||
template <>
|
||||
struct AtomicOps<4>
|
||||
{
|
||||
#if defined(_MSC_VER)
|
||||
typedef long Type;
|
||||
|
||||
static Type Increment(Type *ptr) {
|
||||
return _InterlockedIncrement(ptr);
|
||||
}
|
||||
static Type Decrement(Type *ptr) {
|
||||
return _InterlockedDecrement(ptr);
|
||||
};
|
||||
#elif defined(__GNUC__)
|
||||
typedef int Type;
|
||||
|
||||
// x86/x64 notes: When using GCC < 4.8, this will compile to a spinlock.
|
||||
// On 4.8+, or when using Clang, we'll get the more optimal "lock addl"
|
||||
// variant.
|
||||
static Type Increment(Type *ptr) {
|
||||
return __sync_add_and_fetch(ptr, 1);
|
||||
}
|
||||
static Type Decrement(Type *ptr) {
|
||||
return __sync_sub_and_fetch(ptr, 1);
|
||||
}
|
||||
#endif
|
||||
};
|
||||
|
||||
class AtomicRefCount
|
||||
{
|
||||
typedef AtomicOps<sizeof(uintptr_t)> Ops;
|
||||
|
||||
public:
|
||||
AtomicRefCount(uintptr_t value)
|
||||
: value_(value)
|
||||
{
|
||||
}
|
||||
|
||||
void increment() {
|
||||
Ops::Increment(&value_);
|
||||
}
|
||||
|
||||
// Return false if all references are gone.
|
||||
bool decrement() {
|
||||
return Ops::Decrement(&value_) != 0;
|
||||
}
|
||||
|
||||
private:
|
||||
Ops::Type value_;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_atomics_h_
|
||||
|
@ -1,103 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013-2014, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
#ifndef _include_amtl_fixedarray_h_
|
||||
#define _include_amtl_fixedarray_h_
|
||||
|
||||
#include <am-utility.h>
|
||||
#include <am-allocator-policies.h>
|
||||
#include <am-moveable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
template <typename T, typename AllocPolicy = SystemAllocatorPolicy>
|
||||
class FixedArray : public AllocPolicy
|
||||
{
|
||||
public:
|
||||
FixedArray(size_t length, AllocPolicy = AllocPolicy()) {
|
||||
length_ = length;
|
||||
data_ = (T *)this->malloc(sizeof(T) * length_);
|
||||
if (!data_)
|
||||
return;
|
||||
|
||||
for (size_t i = 0; i < length_; i++)
|
||||
new (&data_[i]) T();
|
||||
}
|
||||
~FixedArray() {
|
||||
for (size_t i = 0; i < length_; i++)
|
||||
data_[i].~T();
|
||||
this->free(data_);
|
||||
}
|
||||
|
||||
// This call may be skipped if the allocator policy is infallible.
|
||||
bool initialize() {
|
||||
return length_ == 0 || !!data_;
|
||||
}
|
||||
|
||||
size_t length() const {
|
||||
return length_;
|
||||
}
|
||||
T &operator [](size_t index) {
|
||||
return at(index);
|
||||
}
|
||||
const T &operator [](size_t index) const {
|
||||
return at(index);
|
||||
}
|
||||
T &at(size_t index) {
|
||||
assert(index < length());
|
||||
return data_[index];
|
||||
}
|
||||
const T &at(size_t index) const {
|
||||
assert(index < length());
|
||||
return data_[index];
|
||||
}
|
||||
void set(size_t index, const T &t) {
|
||||
assert(index < length());
|
||||
data_[index] = t;
|
||||
}
|
||||
void set(size_t index, ke::Moveable<T> t) {
|
||||
assert(index < length());
|
||||
data_[index] = t;
|
||||
}
|
||||
|
||||
T *buffer() const {
|
||||
return data_;
|
||||
}
|
||||
|
||||
private:
|
||||
FixedArray(const FixedArray &other) KE_DELETE;
|
||||
FixedArray &operator =(const FixedArray &other) KE_DELETE;
|
||||
|
||||
private:
|
||||
size_t length_;
|
||||
T *data_;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_fixedarray_h_
|
@ -1,50 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_float_h_
|
||||
#define _include_amtl_float_h_
|
||||
|
||||
#include <math.h>
|
||||
#include <float.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
static inline bool
|
||||
IsNaN(double v)
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
return !!_isnan(v);
|
||||
#else
|
||||
return isnan(v);
|
||||
#endif
|
||||
}
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_float_h_
|
@ -1,201 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_hashmap_h_
|
||||
#define _include_amtl_hashmap_h_
|
||||
|
||||
#include <am-hashtable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// Template parameters:
|
||||
//
|
||||
// K - Key type.
|
||||
// V - Value type.
|
||||
// HashPolicy - A struct with a hash and comparator function for each lookup type:
|
||||
// static uint32_t hash(const Type &value);
|
||||
// static bool matches(const Type &value, const K &key);
|
||||
//
|
||||
// All types that match a given key, must compute the same hash.
|
||||
//
|
||||
// Note that like HashTable, a HashMap is not usable until init() has been called.
|
||||
template <typename K,
|
||||
typename V,
|
||||
typename HashPolicy,
|
||||
typename AllocPolicy = SystemAllocatorPolicy>
|
||||
class HashMap : public AllocPolicy
|
||||
{
|
||||
private:
|
||||
struct Entry
|
||||
{
|
||||
K key;
|
||||
V value;
|
||||
|
||||
Entry()
|
||||
{
|
||||
}
|
||||
Entry(Moveable<Entry> other)
|
||||
: key(Moveable<K>(other->key)),
|
||||
value(Moveable<V>(other->value))
|
||||
{
|
||||
}
|
||||
|
||||
Entry(const K &aKey, const V &aValue)
|
||||
: key(aKey),
|
||||
value(aValue)
|
||||
{ }
|
||||
Entry(const K &aKey, Moveable<V> aValue)
|
||||
: key(aKey),
|
||||
value(aValue)
|
||||
{ }
|
||||
Entry(Moveable<K> aKey, const V &aValue)
|
||||
: key(aKey),
|
||||
value(aValue)
|
||||
{ }
|
||||
Entry(Moveable<K> aKey, Moveable<V> aValue)
|
||||
: key(aKey),
|
||||
value(aValue)
|
||||
{ }
|
||||
};
|
||||
|
||||
struct Policy
|
||||
{
|
||||
typedef Entry Payload;
|
||||
|
||||
template <typename Lookup>
|
||||
static uint32_t hash(const Lookup &key) {
|
||||
return HashPolicy::hash(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
static bool matches(const Lookup &key, const Payload &payload) {
|
||||
return HashPolicy::matches(key, payload.key);
|
||||
}
|
||||
};
|
||||
|
||||
typedef HashTable<Policy, AllocPolicy> Internal;
|
||||
|
||||
public:
|
||||
HashMap(AllocPolicy ap = AllocPolicy())
|
||||
: table_(ap)
|
||||
{
|
||||
}
|
||||
|
||||
// capacity must be a power of two.
|
||||
bool init(size_t capacity = 16) {
|
||||
return table_.init(capacity);
|
||||
}
|
||||
|
||||
typedef typename Internal::Result Result;
|
||||
typedef typename Internal::Insert Insert;
|
||||
typedef typename Internal::iterator iterator;
|
||||
|
||||
template <typename Lookup>
|
||||
Result find(const Lookup &key) {
|
||||
return table_.find(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
Insert findForAdd(const Lookup &key) {
|
||||
return table_.findForAdd(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
void removeIfExists(const Lookup &key) {
|
||||
return table_.remove(key);
|
||||
}
|
||||
|
||||
void remove(Result &r) {
|
||||
table_.remove(r);
|
||||
}
|
||||
|
||||
// The map must not have been mutated in between findForAdd() and add().
|
||||
// The Insert object is still valid after add() returns, however.
|
||||
bool add(Insert &i, const K &key, const V &value) {
|
||||
Entry entry(key, value);
|
||||
return table_.add(i, ke::Move(entry));
|
||||
}
|
||||
bool add(Insert &i, Moveable<K> key, const V &value) {
|
||||
Entry entry(key, value);
|
||||
return table_.add(i, ke::Move(entry));
|
||||
}
|
||||
bool add(Insert &i, const K &key, Moveable<V> value) {
|
||||
Entry entry(key, value);
|
||||
return table_.add(i, ke::Move(entry));
|
||||
}
|
||||
bool add(Insert &i, Moveable<K> key, Moveable<V> value) {
|
||||
Entry entry(key, value);
|
||||
return table_.add(i, ke::Move(entry));
|
||||
}
|
||||
bool add(Insert &i, Moveable<K> key) {
|
||||
Entry entry(key, V());
|
||||
return table_.add(i, ke::Move(entry));
|
||||
}
|
||||
|
||||
// This can be used to avoid compiler constructed temporaries, since AMTL
|
||||
// does not yet support move semantics. If you use this, the key and value
|
||||
// must be set after.
|
||||
bool add(Insert &i) {
|
||||
return table_.add(i);
|
||||
}
|
||||
|
||||
iterator iter() {
|
||||
return iterator(&table_);
|
||||
}
|
||||
|
||||
void clear() {
|
||||
table_.clear();
|
||||
}
|
||||
|
||||
size_t elements() const {
|
||||
return table_.elements();
|
||||
}
|
||||
|
||||
size_t estimateMemoryUse() const {
|
||||
return table_.estimateMemoryUse();
|
||||
}
|
||||
|
||||
private:
|
||||
Internal table_;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
struct PointerPolicy
|
||||
{
|
||||
static inline uint32_t hash(T *p) {
|
||||
return HashPointer(p);
|
||||
}
|
||||
static inline bool matches(T *p1, T *p2) {
|
||||
return p1 == p2;
|
||||
}
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_hashmap_h_
|
@ -1,129 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_hashmap_h_
|
||||
#define _include_amtl_hashmap_h_
|
||||
|
||||
#include <am-hashtable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// Template parameters:
|
||||
//
|
||||
// K - Key type.
|
||||
// HashPolicy - A struct with a hash and comparator function for each lookup type:
|
||||
// static uint32_t hash(const Type &value);
|
||||
// static bool matches(const Type &value, const K &key);
|
||||
//
|
||||
// Like HashMap and HashTable, init() must be called to construct the set.
|
||||
template <typename K,
|
||||
typename HashPolicy,
|
||||
typename AllocPolicy = SystemAllocatorPolicy>
|
||||
class HashSet : public AllocPolicy
|
||||
{
|
||||
struct Policy {
|
||||
typedef K Payload;
|
||||
|
||||
template <typename Lookup>
|
||||
static uint32_t hash(const Lookup &key) {
|
||||
return HashPolicy::hash(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
static bool matches(const Lookup &key, const Payload &payload) {
|
||||
return HashPolicy::matches(key, payload);
|
||||
}
|
||||
};
|
||||
|
||||
typedef HashTable<Policy, AllocPolicy> Internal;
|
||||
|
||||
public:
|
||||
HashSet(AllocPolicy ap = AllocPolicy())
|
||||
: table_(ap)
|
||||
{
|
||||
}
|
||||
|
||||
// capacity must be a power of two.
|
||||
bool init(size_t capacity = 16) {
|
||||
return table_.init(capacity);
|
||||
}
|
||||
|
||||
typedef typename Internal::Result Result;
|
||||
typedef typename Internal::Insert Insert;
|
||||
|
||||
template <typename Lookup>
|
||||
Result find(const Lookup &key) {
|
||||
return table_.find(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
Insert findForAdd(const Lookup &key) {
|
||||
return table_.findForAdd(key);
|
||||
}
|
||||
|
||||
template <typename Lookup>
|
||||
void removeIfExists(const Lookup &key) {
|
||||
return table_.remove(key);
|
||||
}
|
||||
|
||||
void remove(Result &r) {
|
||||
table_.remove(r);
|
||||
}
|
||||
|
||||
// The map must not have been mutated in between findForAdd() and add().
|
||||
// The Insert object is still valid after add() returns, however.
|
||||
bool add(Insert &i, const K &key) {
|
||||
return table_.add(i, key);
|
||||
}
|
||||
bool add(Insert &i, Moveable<K> key) {
|
||||
return table_.add(i, key);
|
||||
}
|
||||
|
||||
// This can be used to avoid compiler constructed temporaries, since AMTL
|
||||
// does not yet support move semantics. If you use this, the key and value
|
||||
// must be set after.
|
||||
bool add(Insert &i) {
|
||||
return table_.add(i);
|
||||
}
|
||||
|
||||
void clear() {
|
||||
table_.clear();
|
||||
}
|
||||
|
||||
size_t estimateMemoryUse() const {
|
||||
return table_.estimateMemoryUse();
|
||||
}
|
||||
|
||||
private:
|
||||
Internal table_;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_hashset_h_
|
@ -1,631 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _INCLUDE_KEIMA_HASHTABLE_H_
|
||||
#define _INCLUDE_KEIMA_HASHTABLE_H_
|
||||
|
||||
#include <new>
|
||||
#include <limits.h>
|
||||
#include <stdlib.h>
|
||||
#include "am-allocator-policies.h"
|
||||
#include "am-utility.h"
|
||||
#include "am-moveable.h"
|
||||
|
||||
namespace ke {
|
||||
|
||||
namespace detail {
|
||||
template <typename T>
|
||||
class HashTableEntry
|
||||
{
|
||||
uint32_t hash_;
|
||||
T t_;
|
||||
|
||||
public:
|
||||
static const uint32_t kFreeHash = 0;
|
||||
static const uint32_t kRemovedHash = 1;
|
||||
|
||||
public:
|
||||
void setHash(uint32_t hash) {
|
||||
hash_ = hash;
|
||||
}
|
||||
void construct() {
|
||||
new (&t_) T();
|
||||
}
|
||||
void construct(const T &t) {
|
||||
new (&t_) T(t);
|
||||
}
|
||||
void construct(Moveable<T> t) {
|
||||
new (&t_) T(t);
|
||||
}
|
||||
uint32_t hash() const {
|
||||
return hash_;
|
||||
}
|
||||
void setRemoved() {
|
||||
destruct();
|
||||
hash_ = kRemovedHash;
|
||||
}
|
||||
void setFree() {
|
||||
destruct();
|
||||
hash_ = kFreeHash;
|
||||
}
|
||||
void initialize() {
|
||||
hash_ = kFreeHash;
|
||||
}
|
||||
void destruct() {
|
||||
if (isLive())
|
||||
t_.~T();
|
||||
}
|
||||
bool removed() const {
|
||||
return hash_ == kRemovedHash;
|
||||
}
|
||||
bool free() const {
|
||||
return hash_ == kFreeHash;
|
||||
}
|
||||
bool isLive() const {
|
||||
return hash_ > kRemovedHash;
|
||||
}
|
||||
T &payload() {
|
||||
assert(isLive());
|
||||
return t_;
|
||||
}
|
||||
bool sameHash(uint32_t hash) const {
|
||||
return hash_ == hash;
|
||||
}
|
||||
|
||||
private:
|
||||
HashTableEntry(const HashTableEntry &other) KE_DELETE;
|
||||
HashTableEntry &operator =(const HashTableEntry &other) KE_DELETE;
|
||||
};
|
||||
}
|
||||
|
||||
// The HashPolicy for the table must have the following members:
|
||||
//
|
||||
// Payload
|
||||
// static uint32_t hash(const LookupType &key);
|
||||
// static bool matches(const LookupType &key, const Payload &other);
|
||||
//
|
||||
// Payload must be a type, and LookupType is any type that lookups will be
|
||||
// performed with (these functions can be overloaded). Example:
|
||||
//
|
||||
// struct Policy {
|
||||
// typedef KeyValuePair Payload;
|
||||
// static uint32 hash(const Key &key) {
|
||||
// ...
|
||||
// }
|
||||
// static bool matches(const Key &key, const KeyValuePair &pair) {
|
||||
// ...
|
||||
// }
|
||||
// };
|
||||
//
|
||||
// Note that the table is not usable until init() has been called.
|
||||
//
|
||||
template <typename HashPolicy, typename AllocPolicy = SystemAllocatorPolicy>
|
||||
class HashTable : public AllocPolicy
|
||||
{
|
||||
friend class iterator;
|
||||
|
||||
typedef typename HashPolicy::Payload Payload;
|
||||
typedef detail::HashTableEntry<Payload> Entry;
|
||||
|
||||
private:
|
||||
static const uint32_t kMinCapacity = 16;
|
||||
static const uint32_t kMaxCapacity = INT_MAX / sizeof(Entry);
|
||||
|
||||
template <typename Key>
|
||||
uint32_t computeHash(const Key &key) {
|
||||
// Multiply by golden ratio.
|
||||
uint32_t hash = HashPolicy::hash(key) * 0x9E3779B9;
|
||||
if (hash == Entry::kFreeHash || hash == Entry::kRemovedHash)
|
||||
hash += 2;
|
||||
return hash;
|
||||
}
|
||||
|
||||
Entry *createTable(uint32_t capacity) {
|
||||
assert(capacity <= kMaxCapacity);
|
||||
|
||||
Entry *table = (Entry *)this->malloc(capacity * sizeof(Entry));
|
||||
if (!table)
|
||||
return NULL;
|
||||
|
||||
for (size_t i = 0; i < capacity; i++)
|
||||
table[i].initialize();
|
||||
|
||||
return table;
|
||||
}
|
||||
|
||||
public:
|
||||
class Result
|
||||
{
|
||||
friend class HashTable;
|
||||
|
||||
Entry *entry_;
|
||||
|
||||
Entry &entry() {
|
||||
return *entry_;
|
||||
}
|
||||
|
||||
public:
|
||||
Result(Entry *entry)
|
||||
: entry_(entry)
|
||||
{ }
|
||||
|
||||
Payload * operator ->() {
|
||||
return &entry_->payload();
|
||||
}
|
||||
Payload & operator *() {
|
||||
return entry_->payload();
|
||||
}
|
||||
|
||||
bool found() const {
|
||||
return entry_->isLive();
|
||||
}
|
||||
};
|
||||
|
||||
class Insert : public Result
|
||||
{
|
||||
uint32_t hash_;
|
||||
|
||||
public:
|
||||
Insert(Entry *entry, uint32_t hash)
|
||||
: Result(entry),
|
||||
hash_(hash)
|
||||
{
|
||||
}
|
||||
|
||||
uint32_t hash() const {
|
||||
return hash_;
|
||||
}
|
||||
};
|
||||
|
||||
private:
|
||||
class Probulator {
|
||||
uint32_t hash_;
|
||||
uint32_t capacity_;
|
||||
|
||||
public:
|
||||
Probulator(uint32_t hash, uint32_t capacity)
|
||||
: hash_(hash),
|
||||
capacity_(capacity)
|
||||
{
|
||||
assert(IsPowerOfTwo(capacity_));
|
||||
}
|
||||
|
||||
uint32_t entry() const {
|
||||
return hash_ & (capacity_ - 1);
|
||||
}
|
||||
uint32_t next() {
|
||||
hash_++;
|
||||
return entry();
|
||||
}
|
||||
};
|
||||
|
||||
bool underloaded() const {
|
||||
// Check if the table is underloaded: < 25% entries used.
|
||||
return (capacity_ > kMinCapacity) && (nelements_ + ndeleted_ < capacity_ / 4);
|
||||
}
|
||||
bool overloaded() const {
|
||||
// Grow if the table is overloaded: > 75% entries used.
|
||||
return (nelements_ + ndeleted_) > ((capacity_ / 2) + (capacity_ / 4));
|
||||
}
|
||||
|
||||
bool shrink() {
|
||||
if ((capacity_ >> 1) < minCapacity_)
|
||||
return true;
|
||||
return changeCapacity(capacity_ >> 1);
|
||||
}
|
||||
|
||||
bool grow() {
|
||||
if (capacity_ >= kMaxCapacity) {
|
||||
this->reportAllocationOverflow();
|
||||
return false;
|
||||
}
|
||||
return changeCapacity(capacity_ << 1);
|
||||
}
|
||||
|
||||
bool changeCapacity(uint32_t newCapacity) {
|
||||
assert(newCapacity <= kMaxCapacity);
|
||||
|
||||
Entry *newTable = createTable(newCapacity);
|
||||
if (!newTable)
|
||||
return false;
|
||||
|
||||
Entry *oldTable = table_;
|
||||
uint32_t oldCapacity = capacity_;
|
||||
|
||||
table_ = newTable;
|
||||
capacity_ = newCapacity;
|
||||
ndeleted_ = 0;
|
||||
|
||||
for (uint32_t i = 0; i < oldCapacity; i++) {
|
||||
Entry &oldEntry = oldTable[i];
|
||||
if (oldEntry.isLive()) {
|
||||
Insert p = insertUnique(oldEntry.hash());
|
||||
p.entry().setHash(p.hash());
|
||||
p.entry().construct(Moveable<Payload>(oldEntry.payload()));
|
||||
}
|
||||
oldEntry.destruct();
|
||||
}
|
||||
this->free(oldTable);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// For use when the key is known to be unique.
|
||||
Insert insertUnique(uint32_t hash) {
|
||||
Probulator probulator(hash, capacity_);
|
||||
|
||||
Entry *e = &table_[probulator.entry()];
|
||||
for (;;) {
|
||||
if (e->free() || e->removed())
|
||||
break;
|
||||
e = &table_[probulator.next()];
|
||||
}
|
||||
|
||||
return Insert(e, hash);
|
||||
}
|
||||
|
||||
template <typename Key>
|
||||
Result lookup(const Key &key) {
|
||||
uint32_t hash = computeHash(key);
|
||||
Probulator probulator(hash, capacity_);
|
||||
|
||||
Entry *e = &table_[probulator.entry()];
|
||||
for (;;) {
|
||||
if (e->free())
|
||||
break;
|
||||
if (e->isLive() &&
|
||||
e->sameHash(hash) &&
|
||||
HashPolicy::matches(key, e->payload()))
|
||||
{
|
||||
return Result(e);
|
||||
}
|
||||
e = &table_[probulator.next()];
|
||||
}
|
||||
|
||||
return Result(e);
|
||||
}
|
||||
|
||||
template <typename Key>
|
||||
Insert lookupForAdd(const Key &key) {
|
||||
uint32_t hash = computeHash(key);
|
||||
Probulator probulator(hash, capacity_);
|
||||
|
||||
Entry *e = &table_[probulator.entry()];
|
||||
for (;;) {
|
||||
if (!e->isLive())
|
||||
break;
|
||||
if (e->sameHash(hash) && HashPolicy::matches(key, e->payload()))
|
||||
break;
|
||||
e = &table_[probulator.next()];
|
||||
}
|
||||
|
||||
return Insert(e, hash);
|
||||
}
|
||||
|
||||
bool internalAdd(Insert &i) {
|
||||
assert(!i.found());
|
||||
|
||||
// If the entry is deleted, just re-use the slot.
|
||||
if (i.entry().removed()) {
|
||||
ndeleted_--;
|
||||
} else {
|
||||
// Otherwise, see if we're at max capacity.
|
||||
if (nelements_ == kMaxCapacity) {
|
||||
this->reportAllocationOverflow();
|
||||
return false;
|
||||
}
|
||||
|
||||
// Check if the table is over or underloaded. The table is always at
|
||||
// least 25% free, so this check is enough to guarantee one free slot.
|
||||
// (Without one free slot, insertion search could infinite loop.)
|
||||
uint32_t oldCapacity = capacity_;
|
||||
if (!checkDensity())
|
||||
return false;
|
||||
|
||||
// If the table changed size, we need to find a new insertion point.
|
||||
// Note that a removed entry is impossible: either we caught it above,
|
||||
// or we just resized and no entries are removed.
|
||||
if (capacity_ != oldCapacity)
|
||||
i = insertUnique(i.hash());
|
||||
}
|
||||
|
||||
nelements_++;
|
||||
i.entry().setHash(i.hash());
|
||||
return true;
|
||||
}
|
||||
|
||||
void removeEntry(Entry &e) {
|
||||
assert(e.isLive());
|
||||
e.setRemoved();
|
||||
ndeleted_++;
|
||||
nelements_--;
|
||||
}
|
||||
|
||||
public:
|
||||
HashTable(AllocPolicy ap = AllocPolicy())
|
||||
: AllocPolicy(ap),
|
||||
capacity_(0),
|
||||
nelements_(0),
|
||||
ndeleted_(0),
|
||||
table_(NULL),
|
||||
minCapacity_(kMinCapacity)
|
||||
{
|
||||
}
|
||||
|
||||
~HashTable()
|
||||
{
|
||||
for (uint32_t i = 0; i < capacity_; i++)
|
||||
table_[i].destruct();
|
||||
this->free(table_);
|
||||
}
|
||||
|
||||
bool init(uint32_t capacity = 0) {
|
||||
if (capacity < kMinCapacity) {
|
||||
capacity = kMinCapacity;
|
||||
} else if (capacity > kMaxCapacity) {
|
||||
this->reportAllocationOverflow();
|
||||
return false;
|
||||
}
|
||||
|
||||
minCapacity_ = capacity;
|
||||
|
||||
assert(IsPowerOfTwo(capacity));
|
||||
capacity_ = capacity;
|
||||
|
||||
table_ = createTable(capacity_);
|
||||
if (!table_)
|
||||
return false;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// The Result object must not be used past mutating table operations.
|
||||
template <typename Key>
|
||||
Result find(const Key &key) {
|
||||
return lookup(key);
|
||||
}
|
||||
|
||||
// The Insert object must not be used past mutating table operations.
|
||||
template <typename Key>
|
||||
Insert findForAdd(const Key &key) {
|
||||
return lookupForAdd(key);
|
||||
}
|
||||
|
||||
template <typename Key>
|
||||
void removeIfExists(const Key &key) {
|
||||
Result r = find(key);
|
||||
if (!r.found())
|
||||
return;
|
||||
remove(r);
|
||||
}
|
||||
|
||||
void remove(Result &r) {
|
||||
assert(r.found());
|
||||
removeEntry(r.entry());
|
||||
}
|
||||
|
||||
// The table must not have been mutated in between findForAdd() and add().
|
||||
// The Insert object is still valid after add() returns, however.
|
||||
bool add(Insert &i, const Payload &payload) {
|
||||
if (!internalAdd(i))
|
||||
return false;
|
||||
i.entry().construct(payload);
|
||||
return true;
|
||||
}
|
||||
bool add(Insert &i, Moveable<Payload> payload) {
|
||||
if (!internalAdd(i))
|
||||
return false;
|
||||
i.entry().construct(payload);
|
||||
return true;
|
||||
}
|
||||
bool add(Insert &i) {
|
||||
if (!internalAdd(i))
|
||||
return false;
|
||||
i.entry().construct();
|
||||
return true;
|
||||
}
|
||||
|
||||
bool checkDensity() {
|
||||
if (underloaded())
|
||||
return shrink();
|
||||
if (overloaded())
|
||||
return grow();
|
||||
return true;
|
||||
}
|
||||
|
||||
void clear() {
|
||||
for (size_t i = 0; i < capacity_; i++) {
|
||||
table_[i].setFree();
|
||||
}
|
||||
ndeleted_ = 0;
|
||||
nelements_ = 0;
|
||||
}
|
||||
|
||||
size_t elements() const {
|
||||
return nelements_;
|
||||
}
|
||||
|
||||
size_t estimateMemoryUse() const {
|
||||
return sizeof(Entry) * capacity_;
|
||||
}
|
||||
|
||||
public:
|
||||
// It is illegal to mutate a HashTable during iteration.
|
||||
class iterator
|
||||
{
|
||||
public:
|
||||
iterator(HashTable *table)
|
||||
: table_(table),
|
||||
i_(table->table_),
|
||||
end_(table->table_ + table->capacity_)
|
||||
{
|
||||
while (i_ < end_ && !i_->isLive())
|
||||
i_++;
|
||||
}
|
||||
|
||||
bool empty() const {
|
||||
return i_ == end_;
|
||||
}
|
||||
|
||||
void erase() {
|
||||
assert(!empty());
|
||||
table_->removeEntry(*i_);
|
||||
}
|
||||
|
||||
Payload *operator ->() const {
|
||||
return &i_->payload();
|
||||
}
|
||||
Payload &operator *() const {
|
||||
return i_->payload();
|
||||
}
|
||||
|
||||
void next() {
|
||||
do {
|
||||
i_++;
|
||||
} while (i_ < end_ && !i_->isLive());
|
||||
}
|
||||
|
||||
private:
|
||||
HashTable *table_;
|
||||
Entry *i_;
|
||||
Entry *end_;
|
||||
};
|
||||
|
||||
private:
|
||||
HashTable(const HashTable &other) KE_DELETE;
|
||||
HashTable &operator =(const HashTable &other) KE_DELETE;
|
||||
|
||||
private:
|
||||
uint32_t capacity_;
|
||||
uint32_t nelements_;
|
||||
uint32_t ndeleted_;
|
||||
Entry *table_;
|
||||
uint32_t minCapacity_;
|
||||
};
|
||||
|
||||
// Bob Jenkin's one-at-a-time hash function[1].
|
||||
//
|
||||
// [1] http://burtleburtle.net/bob/hash/doobs.html
|
||||
class CharacterStreamHasher
|
||||
{
|
||||
uint32_t hash;
|
||||
|
||||
public:
|
||||
CharacterStreamHasher()
|
||||
: hash(0)
|
||||
{ }
|
||||
|
||||
void add(char c) {
|
||||
hash += c;
|
||||
hash += (hash << 10);
|
||||
hash ^= (hash >> 6);
|
||||
}
|
||||
|
||||
void add(const char *s, size_t length) {
|
||||
for (size_t i = 0; i < length; i++)
|
||||
add(s[i]);
|
||||
}
|
||||
|
||||
uint32_t result() {
|
||||
hash += (hash << 3);
|
||||
hash ^= (hash >> 11);
|
||||
hash += (hash << 15);
|
||||
return hash;
|
||||
}
|
||||
};
|
||||
|
||||
static inline uint32_t
|
||||
HashCharSequence(const char *s, size_t length)
|
||||
{
|
||||
CharacterStreamHasher hasher;
|
||||
hasher.add(s, length);
|
||||
return hasher.result();
|
||||
}
|
||||
|
||||
static inline uint32_t
|
||||
FastHashCharSequence(const char *s, size_t length)
|
||||
{
|
||||
uint32_t hash = 0;
|
||||
for (size_t i = 0; i < length; i++)
|
||||
hash = s[i] + (hash << 6) + (hash << 16) - hash;
|
||||
return hash;
|
||||
}
|
||||
|
||||
// From http://burtleburtle.net/bob/hash/integer.html
|
||||
static inline uint32_t
|
||||
HashInt32(int32_t a)
|
||||
{
|
||||
a = (a ^ 61) ^ (a >> 16);
|
||||
a = a + (a << 3);
|
||||
a = a ^ (a >> 4);
|
||||
a = a * 0x27d4eb2d;
|
||||
a = a ^ (a >> 15);
|
||||
return a;
|
||||
}
|
||||
|
||||
// From http://www.cris.com/~Ttwang/tech/inthash.htm
|
||||
static inline uint32_t
|
||||
HashInt64(int64_t key)
|
||||
{
|
||||
key = (~key) + (key << 18); // key = (key << 18) - key - 1;
|
||||
key = key ^ (uint64_t(key) >> 31);
|
||||
key = key * 21; // key = (key + (key << 2)) + (key << 4);
|
||||
key = key ^ (uint64_t(key) >> 11);
|
||||
key = key + (key << 6);
|
||||
key = key ^ (uint64_t(key) >> 22);
|
||||
return uint32_t(key);
|
||||
}
|
||||
|
||||
template <size_t Size>
|
||||
static inline uint32_t
|
||||
HashInteger(uintptr_t value);
|
||||
|
||||
template <>
|
||||
inline uint32_t
|
||||
HashInteger<4>(uintptr_t value)
|
||||
{
|
||||
return HashInt32(value);
|
||||
}
|
||||
|
||||
template <>
|
||||
inline uint32_t
|
||||
HashInteger<8>(uintptr_t value)
|
||||
{
|
||||
return HashInt64(value);
|
||||
}
|
||||
|
||||
static inline uint32_t
|
||||
HashPointer(void *ptr)
|
||||
{
|
||||
return HashInteger<sizeof(ptr)>(reinterpret_cast<uintptr_t>(ptr));
|
||||
}
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _INCLUDE_KEIMA_HASHTABLE_H_
|
@ -1,183 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_inline_list_h_
|
||||
#define _include_amtl_inline_list_h_
|
||||
|
||||
#include <stddef.h>
|
||||
#include <assert.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
template <typename T> class InlineList;
|
||||
|
||||
// Objects can recursively inherit from InlineListNode in order to have
|
||||
// membership in an InlineList<T>.
|
||||
template <typename T>
|
||||
class InlineListNode
|
||||
{
|
||||
friend class InlineList<T>;
|
||||
|
||||
public:
|
||||
InlineListNode()
|
||||
: next_(NULL),
|
||||
prev_(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
InlineListNode(InlineListNode *next, InlineListNode *prev)
|
||||
: next_(next),
|
||||
prev_(prev)
|
||||
{
|
||||
}
|
||||
|
||||
protected:
|
||||
InlineListNode *next_;
|
||||
InlineListNode *prev_;
|
||||
};
|
||||
|
||||
// An InlineList is a linked list that threads link pointers through objects,
|
||||
// rather than allocating node memory. A node can be in at most one list at
|
||||
// any time.
|
||||
//
|
||||
// Since InlineLists are designed to be very cheap, there is no requirement
|
||||
// that elements be removed from a list once the list is destructed. However,
|
||||
// for as long as the list is alive, all of its contained nodes must also
|
||||
// be alive.
|
||||
template <typename T>
|
||||
class InlineList
|
||||
{
|
||||
typedef InlineListNode<T> Node;
|
||||
|
||||
Node head_;
|
||||
|
||||
// Work around a clang bug where we can't initialize with &head_ in the ctor.
|
||||
inline Node *head() {
|
||||
return &head_;
|
||||
}
|
||||
|
||||
public:
|
||||
InlineList()
|
||||
: head_(head(), head())
|
||||
{
|
||||
}
|
||||
|
||||
~InlineList()
|
||||
{
|
||||
#if !defined(NDEBUG)
|
||||
// Remove all items to clear their next/prev fields.
|
||||
while (begin() != end())
|
||||
remove(*begin());
|
||||
#endif
|
||||
}
|
||||
|
||||
public:
|
||||
class iterator
|
||||
{
|
||||
friend class InlineList;
|
||||
Node *iter_;
|
||||
|
||||
public:
|
||||
iterator(Node *iter)
|
||||
: iter_(iter)
|
||||
{
|
||||
}
|
||||
|
||||
iterator & operator ++() {
|
||||
iter_ = iter_->next;
|
||||
return *this;
|
||||
}
|
||||
iterator operator ++(int) {
|
||||
iterator old(*this);
|
||||
iter_ = iter_->next_;
|
||||
return old;
|
||||
}
|
||||
T * operator *() {
|
||||
return static_cast<T *>(iter_);
|
||||
}
|
||||
T * operator ->() {
|
||||
return static_cast<T *>(iter_);
|
||||
}
|
||||
bool operator !=(const iterator &where) const {
|
||||
return iter_ != where.iter_;
|
||||
}
|
||||
bool operator ==(const iterator &where) const {
|
||||
return iter_ == where.iter_;
|
||||
}
|
||||
};
|
||||
|
||||
iterator begin() {
|
||||
return iterator(head_.next_);
|
||||
}
|
||||
|
||||
iterator end() {
|
||||
return iterator(&head_);
|
||||
}
|
||||
|
||||
iterator erase(iterator &at) {
|
||||
iterator next = at;
|
||||
next++;
|
||||
|
||||
remove(at.iter_);
|
||||
|
||||
// Iterator is no longer valid.
|
||||
at.iter_ = NULL;
|
||||
|
||||
return next;
|
||||
}
|
||||
|
||||
bool empty() const {
|
||||
return head_.next_ == &head_;
|
||||
}
|
||||
|
||||
void remove(Node *t) {
|
||||
t->prev_->next_ = t->next_;
|
||||
t->next_->prev_ = t->prev_;
|
||||
|
||||
#if !defined(NDEBUG)
|
||||
t->next_ = NULL;
|
||||
t->prev_ = NULL;
|
||||
#endif
|
||||
}
|
||||
|
||||
void append(Node *t) {
|
||||
assert(!t->next_);
|
||||
assert(!t->prev_);
|
||||
|
||||
t->prev_ = head_.prev_;
|
||||
t->next_ = &head_;
|
||||
head_.prev_->next_ = t;
|
||||
head_.prev_ = t;
|
||||
}
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_inline_list_h_
|
||||
|
@ -1,309 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_linkedlist_h_
|
||||
#define _include_amtl_linkedlist_h_
|
||||
|
||||
#include <new>
|
||||
#include <stdlib.h>
|
||||
#include <am-allocator-policies.h>
|
||||
#include <am-utility.h>
|
||||
#include <am-moveable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// LinkedList, analagous to std::list or SourceHook::List. Since it performs a
|
||||
// malloc() and free() on every contained node, it should be avoided unless
|
||||
// absolutely necessary, or for when allocation performance is not a factor. It
|
||||
// is provided here to safely port old AlliedModders code to AMTL.
|
||||
//
|
||||
// In order to use a circular chain, LinkedList's allocation size includes
|
||||
// exactly one T. If T is very large, LinkedList should be allocated on the
|
||||
// heap, to avoid using the stack.
|
||||
template <class T, class AllocPolicy = SystemAllocatorPolicy>
|
||||
class LinkedList : public AllocPolicy
|
||||
{
|
||||
public:
|
||||
friend class iterator;
|
||||
|
||||
class Node
|
||||
{
|
||||
public:
|
||||
Node(const T &o)
|
||||
: obj(o)
|
||||
{
|
||||
}
|
||||
Node(Moveable<T> o)
|
||||
: obj(o)
|
||||
{
|
||||
}
|
||||
|
||||
T obj;
|
||||
Node *next;
|
||||
Node *prev;
|
||||
};
|
||||
|
||||
public:
|
||||
LinkedList(AllocPolicy = AllocPolicy())
|
||||
: length_(0)
|
||||
{
|
||||
head()->prev = head();
|
||||
head()->next = head();
|
||||
}
|
||||
~LinkedList() {
|
||||
clear();
|
||||
}
|
||||
|
||||
bool append(const T &obj) {
|
||||
return insertBefore(end(), obj) != end();
|
||||
}
|
||||
bool append(Moveable<T> obj) {
|
||||
return insertBefore(end(), obj) != end();
|
||||
}
|
||||
|
||||
bool prepend(const T &obj) {
|
||||
return insertBefore(begin(), obj) != begin();
|
||||
}
|
||||
bool prepend(Moveable<T> obj) {
|
||||
return insertBefore(begin(), obj) != begin();
|
||||
}
|
||||
|
||||
size_t length() const {
|
||||
return length_;
|
||||
}
|
||||
|
||||
void clear() {
|
||||
Node *node = head()->next;
|
||||
Node *temp;
|
||||
head()->next = head();
|
||||
head()->prev = head();
|
||||
|
||||
// Iterate through the nodes until we find the sentinel again.
|
||||
while (node != head()) {
|
||||
temp = node->next;
|
||||
freeNode(node);
|
||||
node = temp;
|
||||
}
|
||||
length_ = 0;
|
||||
}
|
||||
|
||||
bool empty() const {
|
||||
return (length_ == 0);
|
||||
}
|
||||
|
||||
T &front() {
|
||||
assert(!empty());
|
||||
return head()->next->obj;
|
||||
}
|
||||
T &back() {
|
||||
assert(!empty());
|
||||
return head()->prev->obj;
|
||||
}
|
||||
|
||||
private:
|
||||
const Node *head() const {
|
||||
return sentinel_.address();
|
||||
}
|
||||
Node *head() {
|
||||
return sentinel_.address();
|
||||
}
|
||||
|
||||
Node *allocNode(const T &obj) {
|
||||
Node *node = (Node *)this->malloc(sizeof(Node));
|
||||
if (!node)
|
||||
return NULL;
|
||||
new (node) Node(obj);
|
||||
return node;
|
||||
}
|
||||
Node *allocNode(Moveable<T> obj) {
|
||||
Node *node = (Node *)this->malloc(sizeof(Node));
|
||||
if (!node)
|
||||
return NULL;
|
||||
new (node) Node(obj);
|
||||
return node;
|
||||
}
|
||||
|
||||
void freeNode(Node *node) {
|
||||
node->obj.~T();
|
||||
this->free(node);
|
||||
}
|
||||
|
||||
private:
|
||||
StorageBuffer<Node> sentinel_;
|
||||
size_t length_;
|
||||
|
||||
public:
|
||||
class iterator
|
||||
{
|
||||
friend class LinkedList;
|
||||
|
||||
public:
|
||||
iterator()
|
||||
: this_(NULL)
|
||||
{
|
||||
}
|
||||
iterator(const LinkedList &src)
|
||||
: this_(src.head())
|
||||
{
|
||||
}
|
||||
iterator(Node *n)
|
||||
: this_(n)
|
||||
{
|
||||
}
|
||||
iterator(const iterator &where)
|
||||
: this_(where.this_)
|
||||
{
|
||||
}
|
||||
|
||||
iterator &operator --() {
|
||||
if (this_)
|
||||
this_ = this_->prev;
|
||||
return *this;
|
||||
}
|
||||
iterator operator --(int) {
|
||||
iterator old(*this);
|
||||
if (this_)
|
||||
this_ = this_->prev;
|
||||
return old;
|
||||
}
|
||||
iterator &operator ++() {
|
||||
if (this_)
|
||||
this_ = this_->next;
|
||||
return *this;
|
||||
}
|
||||
iterator operator ++(int) {
|
||||
iterator old(*this);
|
||||
if (this_)
|
||||
this_ = this_->next;
|
||||
return old;
|
||||
}
|
||||
|
||||
const T &operator * () const {
|
||||
return this_->obj;
|
||||
}
|
||||
T &operator * () {
|
||||
return this_->obj;
|
||||
}
|
||||
T *operator ->() {
|
||||
return &this_->obj;
|
||||
}
|
||||
const T *operator ->() const {
|
||||
return &(this_->obj);
|
||||
}
|
||||
|
||||
bool operator !=(const iterator &where) const {
|
||||
return (this_ != where.this_);
|
||||
}
|
||||
bool operator ==(const iterator &where) const {
|
||||
return (this_ == where.this_);
|
||||
}
|
||||
|
||||
operator bool() {
|
||||
return !!this_;
|
||||
}
|
||||
|
||||
private:
|
||||
Node *this_;
|
||||
};
|
||||
|
||||
private:
|
||||
// Insert obj right before where.
|
||||
iterator insert(iterator where, Node *node) {
|
||||
if (!node)
|
||||
return where;
|
||||
|
||||
Node *pWhereNode = where.this_;
|
||||
|
||||
pWhereNode->prev->next = node;
|
||||
node->prev = pWhereNode->prev;
|
||||
pWhereNode->prev = node;
|
||||
node->next = pWhereNode;
|
||||
|
||||
length_++;
|
||||
return iterator(node);
|
||||
}
|
||||
|
||||
public:
|
||||
iterator begin() {
|
||||
return iterator(head()->next);
|
||||
}
|
||||
iterator end() {
|
||||
return iterator(head());
|
||||
}
|
||||
iterator erase(iterator where) {
|
||||
Node *pNode = where.this_;
|
||||
iterator iter(where);
|
||||
iter++;
|
||||
|
||||
pNode->prev->next = pNode->next;
|
||||
pNode->next->prev = pNode->prev;
|
||||
|
||||
freeNode(pNode);
|
||||
length_--;
|
||||
|
||||
return iter;
|
||||
}
|
||||
iterator insertBefore(iterator where, const T &obj) {
|
||||
return insert(where, allocNode(obj));
|
||||
}
|
||||
iterator insertBefore(iterator where, Moveable<T> obj) {
|
||||
return insert(where, allocNode(obj));
|
||||
}
|
||||
|
||||
public:
|
||||
// Removes one instance of |obj| from the list, if found.
|
||||
void remove(const T &obj) {
|
||||
for (iterator b = begin(); b != end(); b++) {
|
||||
if (*b == obj) {
|
||||
erase(b);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <typename U>
|
||||
iterator find(const U &equ) {
|
||||
for (iterator iter = begin(); iter != end(); iter++) {
|
||||
if (*iter == equ)
|
||||
return iter;
|
||||
}
|
||||
return end();
|
||||
}
|
||||
|
||||
|
||||
private:
|
||||
// These are disallowed because they basically violate the failure handling
|
||||
// model for AllocPolicies and are also likely to have abysmal performance.
|
||||
LinkedList &operator =(const LinkedList<T> &other) KE_DELETE;
|
||||
LinkedList(const LinkedList<T> &other) KE_DELETE;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif //_INCLUDE_CSDM_LIST_H
|
@ -1,73 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_moveable_h_
|
||||
#define _include_amtl_moveable_h_
|
||||
|
||||
namespace ke {
|
||||
|
||||
// This is a feature in C++11, but since AM projects do not have access to
|
||||
// C++11 yet, we provide templates to implement move semantics. A class can
|
||||
// provide a constructor for (ke::Moveable<T> t) which containers will try
|
||||
// to use.
|
||||
//
|
||||
// When implementing a constructor that takes a Moveable, the object being
|
||||
// moved should be left in a state that is safe, since its destructor will
|
||||
// be called even though it has been moved.
|
||||
|
||||
template <typename T>
|
||||
struct Moveable
|
||||
{
|
||||
public:
|
||||
explicit Moveable(T &t)
|
||||
: t_(t)
|
||||
{
|
||||
}
|
||||
|
||||
T *operator ->() {
|
||||
return &t_;
|
||||
}
|
||||
operator T &() {
|
||||
return t_;
|
||||
}
|
||||
|
||||
private:
|
||||
T &t_;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
static inline Moveable<T>
|
||||
Move(T &t)
|
||||
{
|
||||
return Moveable<T>(t);
|
||||
}
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_moveable_h_
|
@ -1,71 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_ts_refcounting_h_
|
||||
#define _include_amtl_ts_refcounting_h_
|
||||
|
||||
#include <am-refcounting.h>
|
||||
#include <am-atomics.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// See the comment above Refcounted<T> for more information. This class is
|
||||
// identical, except changing the reference count is guaranteed to be atomic
|
||||
// with respect to other threads changing the reference count.
|
||||
template <typename T>
|
||||
class RefcountedThreadsafe
|
||||
{
|
||||
public:
|
||||
RefcountedThreadsafe()
|
||||
: refcount_(0)
|
||||
{
|
||||
}
|
||||
|
||||
void AddRef() {
|
||||
refcount_.increment();
|
||||
}
|
||||
bool Release() {
|
||||
if (!refcount_.decrement()) {
|
||||
delete static_cast<T *>(this);
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
protected:
|
||||
~RefcountedThreadsafe() {
|
||||
}
|
||||
|
||||
private:
|
||||
AtomicRefCount refcount_;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_ts_refcounting_h_
|
@ -1,336 +0,0 @@
|
||||
// vim: set sts=8 ts=4 sw=4 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_refcounting_h_
|
||||
#define _include_amtl_refcounting_h_
|
||||
|
||||
#include <am-utility.h>
|
||||
#include <am-moveable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
template <typename T> class Ref;
|
||||
|
||||
// Objects in AMTL inheriting from Refcounted will have an initial refcount
|
||||
// of 0. However, in some systems (such as COM), the initial refcount is 1,
|
||||
// or functions may return raw pointers that have been AddRef'd. In these
|
||||
// cases it would be a mistake to use Ref<> or PassRef<>, since the object
|
||||
// would leak an extra reference.
|
||||
//
|
||||
// This container holds a refcounted object without addrefing it. This is
|
||||
// intended only for interacting with functions which return an object that
|
||||
// has been manually AddRef'd. Note that this will perform a Release(), so
|
||||
// so it is necessary to assign it to retain the object.
|
||||
template <typename T>
|
||||
class AlreadyRefed
|
||||
{
|
||||
public:
|
||||
AlreadyRefed(T *t)
|
||||
: thing_(t)
|
||||
{
|
||||
}
|
||||
AlreadyRefed(const AlreadyRefed<T> &other)
|
||||
: thing_(other.thing_)
|
||||
{
|
||||
// If copy elision for some reason doesn't happen (for example, when
|
||||
// returning from AdoptRef), just null out the source ref.
|
||||
other.thing_ = NULL;
|
||||
}
|
||||
~AlreadyRefed() {
|
||||
if (thing_)
|
||||
thing_->Release();
|
||||
}
|
||||
|
||||
T *release() const {
|
||||
return ReturnAndVoid(thing_);
|
||||
}
|
||||
|
||||
private:
|
||||
mutable T *thing_;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
static inline AlreadyRefed<T>
|
||||
AdoptRef(T *t)
|
||||
{
|
||||
return AlreadyRefed<T>(t);
|
||||
}
|
||||
|
||||
// When returning a value, we'd rather not be needlessly changing the refcount,
|
||||
// so we have a special type to use for returns.
|
||||
template <typename T>
|
||||
class PassRef
|
||||
{
|
||||
public:
|
||||
PassRef(T *thing)
|
||||
: thing_(thing)
|
||||
{
|
||||
AddRef();
|
||||
}
|
||||
PassRef()
|
||||
: thing_(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
PassRef(const AlreadyRefed<T *> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
// Don't addref, newborn means already addref'd.
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
PassRef(const AlreadyRefed<S *> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
// Don't addref, newborn means already addref'd.
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
inline PassRef(const Ref<S> &other);
|
||||
|
||||
PassRef(const PassRef &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
template <typename S>
|
||||
PassRef(const PassRef<S> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
~PassRef()
|
||||
{
|
||||
Release();
|
||||
}
|
||||
|
||||
operator T &() {
|
||||
return *thing_;
|
||||
}
|
||||
operator T *() const {
|
||||
return thing_;
|
||||
}
|
||||
T *operator ->() const {
|
||||
return operator *();
|
||||
}
|
||||
T *operator *() const {
|
||||
return thing_;
|
||||
}
|
||||
bool operator !() const {
|
||||
return !thing_;
|
||||
}
|
||||
|
||||
T *release() const {
|
||||
return ReturnAndVoid(thing_);
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
PassRef &operator =(const PassRef<S> &other) {
|
||||
Release();
|
||||
thing_ = other.release();
|
||||
return *this;
|
||||
}
|
||||
|
||||
private:
|
||||
// Disallowed operators.
|
||||
PassRef &operator =(T *other);
|
||||
PassRef &operator =(AlreadyRefed<T> &other);
|
||||
|
||||
void AddRef() {
|
||||
if (thing_)
|
||||
thing_->AddRef();
|
||||
}
|
||||
void Release() {
|
||||
if (thing_)
|
||||
thing_->Release();
|
||||
}
|
||||
|
||||
private:
|
||||
mutable T *thing_;
|
||||
};
|
||||
|
||||
// Classes which are refcounted should inherit from this. Note that reference
|
||||
// counts start at 0 in AMTL, rather than 1. This avoids the complexity of
|
||||
// having to adopt the initial ref upon allocation. However, this also means
|
||||
// invoking Release() on a newly allocated object is illegal. Newborn objects
|
||||
// must either be assigned to a Ref or PassRef (NOT an AdoptRef/AlreadyRefed),
|
||||
// or must be deleted using |delete|.
|
||||
template <typename T>
|
||||
class Refcounted
|
||||
{
|
||||
public:
|
||||
Refcounted()
|
||||
: refcount_(0)
|
||||
{
|
||||
}
|
||||
|
||||
void AddRef() {
|
||||
refcount_++;
|
||||
}
|
||||
void Release() {
|
||||
assert(refcount_ > 0);
|
||||
if (--refcount_ == 0)
|
||||
delete static_cast<T *>(this);
|
||||
}
|
||||
|
||||
protected:
|
||||
~Refcounted() {
|
||||
}
|
||||
|
||||
private:
|
||||
uintptr_t refcount_;
|
||||
};
|
||||
|
||||
// Simple class for automatic refcounting.
|
||||
template <typename T>
|
||||
class Ref
|
||||
{
|
||||
public:
|
||||
Ref(T *thing)
|
||||
: thing_(thing)
|
||||
{
|
||||
AddRef();
|
||||
}
|
||||
|
||||
Ref()
|
||||
: thing_(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
Ref(const Ref &other)
|
||||
: thing_(other.thing_)
|
||||
{
|
||||
AddRef();
|
||||
}
|
||||
Ref(Moveable<Ref> other)
|
||||
: thing_(other->thing_)
|
||||
{
|
||||
other->thing_ = NULL;
|
||||
}
|
||||
template <typename S>
|
||||
Ref(const Ref<S> &other)
|
||||
: thing_(*other)
|
||||
{
|
||||
AddRef();
|
||||
}
|
||||
Ref(const PassRef<T> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
template <typename S>
|
||||
Ref(const PassRef<S> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
Ref(const AlreadyRefed<T> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
template <typename S>
|
||||
Ref(const AlreadyRefed<S> &other)
|
||||
: thing_(other.release())
|
||||
{
|
||||
}
|
||||
~Ref()
|
||||
{
|
||||
Release();
|
||||
}
|
||||
|
||||
T *operator ->() const {
|
||||
return operator *();
|
||||
}
|
||||
T *operator *() const {
|
||||
return thing_;
|
||||
}
|
||||
operator T *() {
|
||||
return thing_;
|
||||
}
|
||||
bool operator !() const {
|
||||
return !thing_;
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
Ref &operator =(S *thing) {
|
||||
Release();
|
||||
thing_ = thing;
|
||||
AddRef();
|
||||
return *this;
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
Ref &operator =(const PassRef<S> &other) {
|
||||
Release();
|
||||
thing_ = other.release();
|
||||
return *this;
|
||||
}
|
||||
|
||||
template <typename S>
|
||||
Ref &operator =(const AlreadyRefed<S> &other) {
|
||||
Release();
|
||||
thing_ = other.release();
|
||||
return *this;
|
||||
}
|
||||
|
||||
Ref &operator =(const Ref &other) {
|
||||
Release();
|
||||
thing_ = other.thing_;
|
||||
AddRef();
|
||||
return *this;
|
||||
}
|
||||
|
||||
Ref &operator =(Moveable<Ref> other) {
|
||||
Release();
|
||||
thing_ = other->thing_;
|
||||
other->thing_ = NULL;
|
||||
return *this;
|
||||
}
|
||||
|
||||
private:
|
||||
void AddRef() {
|
||||
if (thing_)
|
||||
thing_->AddRef();
|
||||
}
|
||||
void Release() {
|
||||
if (thing_)
|
||||
thing_->Release();
|
||||
}
|
||||
|
||||
protected:
|
||||
T *thing_;
|
||||
};
|
||||
|
||||
template <typename T> template <typename S>
|
||||
PassRef<T>::PassRef(const Ref<S> &other)
|
||||
: thing_(*other)
|
||||
{
|
||||
AddRef();
|
||||
}
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_refcounting_h_
|
||||
|
@ -1,136 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_string_h_
|
||||
#define _include_amtl_string_h_
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <am-utility.h>
|
||||
#include <am-moveable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// ASCII string.
|
||||
class AString
|
||||
{
|
||||
public:
|
||||
AString()
|
||||
: length_(0)
|
||||
{
|
||||
}
|
||||
|
||||
explicit AString(const char *str) {
|
||||
set(str, strlen(str));
|
||||
}
|
||||
AString(const char *str, size_t length) {
|
||||
set(str, length);
|
||||
}
|
||||
AString(const AString &other) {
|
||||
if (other.length_)
|
||||
set(other.chars_, other.length_);
|
||||
else
|
||||
length_ = 0;
|
||||
}
|
||||
AString(Moveable<AString> other)
|
||||
: chars_(other->chars_.take()),
|
||||
length_(other->length_)
|
||||
{
|
||||
other->length_ = 0;
|
||||
}
|
||||
|
||||
AString &operator =(const char *str) {
|
||||
if (str && str[0]) {
|
||||
set(str, strlen(str));
|
||||
} else {
|
||||
chars_ = NULL;
|
||||
length_ = 0;
|
||||
}
|
||||
return *this;
|
||||
}
|
||||
AString &operator =(const AString &other) {
|
||||
if (other.length_) {
|
||||
set(other.chars_, other.length_);
|
||||
} else {
|
||||
chars_ = NULL;
|
||||
length_ = 0;
|
||||
}
|
||||
return *this;
|
||||
}
|
||||
AString &operator =(Moveable<AString> other) {
|
||||
chars_ = other->chars_.take();
|
||||
length_ = other->length_;
|
||||
other->length_ = 0;
|
||||
return *this;
|
||||
}
|
||||
|
||||
int compare(const char *str) const {
|
||||
return strcmp(chars(), str);
|
||||
}
|
||||
int compare(const AString &other) const {
|
||||
return strcmp(chars(), other.chars());
|
||||
}
|
||||
bool operator ==(const AString &other) const {
|
||||
return other.length() == length() &&
|
||||
memcmp(other.chars(), chars(), length()) == 0;
|
||||
}
|
||||
|
||||
char operator [](size_t index) const {
|
||||
assert(index < length());
|
||||
return chars()[index];
|
||||
}
|
||||
|
||||
size_t length() const {
|
||||
return length_;
|
||||
}
|
||||
|
||||
const char *chars() const {
|
||||
if (!chars_)
|
||||
return "";
|
||||
return chars_;
|
||||
}
|
||||
|
||||
private:
|
||||
static const size_t kInvalidLength = (size_t)-1;
|
||||
|
||||
void set(const char *str, size_t length) {
|
||||
chars_ = new char[length + 1];
|
||||
length_ = length;
|
||||
memcpy(chars_, str, length);
|
||||
chars_[length] = '\0';
|
||||
}
|
||||
|
||||
private:
|
||||
AutoArray<char> chars_;
|
||||
size_t length_;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_string_h_
|
@ -1,219 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_thread_posix_h_
|
||||
#define _include_amtl_thread_posix_h_
|
||||
|
||||
#include <pthread.h>
|
||||
#include <sys/time.h>
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#if defined(__linux__)
|
||||
# include <sys/prctl.h>
|
||||
#endif
|
||||
#if defined(__APPLE__)
|
||||
# include <dlfcn.h>
|
||||
#endif
|
||||
|
||||
namespace ke {
|
||||
|
||||
class Mutex : public Lockable
|
||||
{
|
||||
public:
|
||||
Mutex() {
|
||||
#if !defined(NDEBUG)
|
||||
int rv =
|
||||
#endif
|
||||
pthread_mutex_init(&mutex_, NULL);
|
||||
assert(rv == 0);
|
||||
}
|
||||
~Mutex() {
|
||||
pthread_mutex_destroy(&mutex_);
|
||||
}
|
||||
|
||||
bool DoTryLock() KE_OVERRIDE {
|
||||
return pthread_mutex_trylock(&mutex_) == 0;
|
||||
}
|
||||
|
||||
void DoLock() KE_OVERRIDE {
|
||||
pthread_mutex_lock(&mutex_);
|
||||
}
|
||||
|
||||
void DoUnlock() KE_OVERRIDE {
|
||||
pthread_mutex_unlock(&mutex_);
|
||||
}
|
||||
|
||||
pthread_mutex_t *raw() {
|
||||
return &mutex_;
|
||||
}
|
||||
|
||||
private:
|
||||
pthread_mutex_t mutex_;
|
||||
};
|
||||
|
||||
// Currently, this class only supports single-listener CVs.
|
||||
class ConditionVariable : public Lockable
|
||||
{
|
||||
public:
|
||||
ConditionVariable() {
|
||||
#if !defined(NDEBUG)
|
||||
int rv =
|
||||
#endif
|
||||
pthread_cond_init(&cv_, NULL);
|
||||
assert(rv == 0);
|
||||
}
|
||||
~ConditionVariable() {
|
||||
pthread_cond_destroy(&cv_);
|
||||
}
|
||||
|
||||
bool DoTryLock() KE_OVERRIDE {
|
||||
return mutex_.DoTryLock();
|
||||
}
|
||||
void DoLock() KE_OVERRIDE {
|
||||
mutex_.DoLock();
|
||||
}
|
||||
void DoUnlock() KE_OVERRIDE {
|
||||
mutex_.DoUnlock();
|
||||
}
|
||||
|
||||
void Notify() {
|
||||
AssertCurrentThreadOwns();
|
||||
pthread_cond_signal(&cv_);
|
||||
}
|
||||
|
||||
WaitResult Wait(size_t timeout_ms) {
|
||||
AssertCurrentThreadOwns();
|
||||
|
||||
#if defined(__linux__)
|
||||
struct timespec ts;
|
||||
if (clock_gettime(CLOCK_REALTIME, &ts) == -1)
|
||||
return Wait_Error;
|
||||
#else
|
||||
struct timeval tv;
|
||||
gettimeofday(&tv, NULL);
|
||||
|
||||
struct timespec ts;
|
||||
ts.tv_sec = tv.tv_sec;
|
||||
ts.tv_nsec = tv.tv_usec * 1000;
|
||||
#endif
|
||||
|
||||
ts.tv_sec += timeout_ms / 1000;
|
||||
ts.tv_nsec += (timeout_ms % 1000) * 1000000;
|
||||
if (ts.tv_nsec >= 1000000000) {
|
||||
ts.tv_sec++;
|
||||
ts.tv_nsec -= 1000000000;
|
||||
}
|
||||
|
||||
DebugSetUnlocked();
|
||||
int rv = pthread_cond_timedwait(&cv_, mutex_.raw(), &ts);
|
||||
DebugSetLocked();
|
||||
|
||||
if (rv == ETIMEDOUT)
|
||||
return Wait_Timeout;
|
||||
if (rv == 0)
|
||||
return Wait_Signaled;
|
||||
return Wait_Error;
|
||||
}
|
||||
|
||||
WaitResult Wait() {
|
||||
AssertCurrentThreadOwns();
|
||||
|
||||
DebugSetUnlocked();
|
||||
int rv = pthread_cond_wait(&cv_, mutex_.raw());
|
||||
DebugSetLocked();
|
||||
|
||||
if (rv == 0)
|
||||
return Wait_Signaled;
|
||||
return Wait_Error;
|
||||
}
|
||||
|
||||
private:
|
||||
Mutex mutex_;
|
||||
pthread_cond_t cv_;
|
||||
};
|
||||
|
||||
class Thread
|
||||
{
|
||||
struct ThreadData {
|
||||
IRunnable *run;
|
||||
char name[17];
|
||||
};
|
||||
public:
|
||||
Thread(IRunnable *run, const char *name = NULL) {
|
||||
ThreadData *data = new ThreadData;
|
||||
data->run = run;
|
||||
snprintf(data->name, sizeof(data->name), "%s", name ? name : "");
|
||||
|
||||
initialized_ = (pthread_create(&thread_, NULL, Main, data) == 0);
|
||||
if (!initialized_)
|
||||
delete data;
|
||||
}
|
||||
~Thread() {
|
||||
if (!Succeeded())
|
||||
return;
|
||||
pthread_detach(thread_);
|
||||
}
|
||||
|
||||
bool Succeeded() const {
|
||||
return initialized_;
|
||||
}
|
||||
|
||||
void Join() {
|
||||
if (!Succeeded())
|
||||
return;
|
||||
pthread_join(thread_, NULL);
|
||||
initialized_ = false;
|
||||
}
|
||||
|
||||
private:
|
||||
static void *Main(void *arg) {
|
||||
AutoPtr<ThreadData> data((ThreadData *)arg);
|
||||
|
||||
if (data->name[0]) {
|
||||
#if defined(__linux__)
|
||||
prctl(PR_SET_NAME, (unsigned long)data->name);
|
||||
#elif defined(__APPLE__)
|
||||
int (*fn)(const char *) = (int (*)(const char *))dlsym(RTLD_DEFAULT, "pthread_setname_np");
|
||||
if (fn)
|
||||
fn(data->name);
|
||||
#endif
|
||||
}
|
||||
data->run->Run();
|
||||
return NULL;
|
||||
}
|
||||
|
||||
private:
|
||||
bool initialized_;
|
||||
pthread_t thread_;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_thread_posix_h_
|
||||
|
@ -1,265 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_threads_
|
||||
#define _include_amtl_threads_
|
||||
|
||||
#include <assert.h>
|
||||
#if defined(_MSC_VER)
|
||||
# include <windows.h>
|
||||
# include <WinBase.h>
|
||||
#else
|
||||
# include <pthread.h>
|
||||
#endif
|
||||
#include <am-utility.h>
|
||||
|
||||
// Thread primitives for SourcePawn.
|
||||
//
|
||||
// Linking Requirements:
|
||||
//
|
||||
// OS X: None (-lpthread and -ldl are optional)
|
||||
// Windows: None
|
||||
// Linux: -lpthread -lrt required
|
||||
//
|
||||
// -- Mutexes --
|
||||
//
|
||||
// A Lockable is a mutual exclusion primitive. It can be owned by at most one
|
||||
// thread at a time, and ownership blocks any other thread from taking taking
|
||||
// ownership. Ownership must be acquired and released on the same thread.
|
||||
// Lockables are not re-entrant.
|
||||
//
|
||||
// While a few classes support the Lockable interface, the simplest Lockable
|
||||
// object that can be instantiated is a Mutex.
|
||||
//
|
||||
// -- Condition Variables --
|
||||
//
|
||||
// A ConditionVariable provides mutually exclusive access based on a
|
||||
// condition ocurring. CVs provide two capabilities: Wait(), which will block
|
||||
// until the condition is triggered, and Notify(), which signals any blocking
|
||||
// thread that the condition has occurred.
|
||||
//
|
||||
// Condition variables have an underlying mutex lock. This lock must be
|
||||
// acquired before calling Wait() or Notify(). It is automatically released
|
||||
// once Wait begins blocking. This operation is atomic with respect to other
|
||||
// threads and the mutex. For example, it is not possible for the lock to be
|
||||
// acquired by another thread in between unlocking and blocking. Since Notify
|
||||
// also requires the lock to be acquired, there is no risk of an event
|
||||
// accidentally dissipating into thin air because it was sent before the other
|
||||
// thread began blocking.
|
||||
//
|
||||
// When Wait() returns, the lock is automatically re-acquired. This operation
|
||||
// is NOT atomic. In between waking up and re-acquiring the lock, another
|
||||
// thread may steal the lock and issue another event. Applications must
|
||||
// account for this. For example, a message pump should check that there are
|
||||
// no messages left to process before blocking again.
|
||||
//
|
||||
// Likewise, it is also not defined whether a Signal() will have any effect
|
||||
// while a thread is not waiting on the monitor. This is yet another reason
|
||||
// the above paragraph is so important - applications should, under a lock of
|
||||
// the condition variable - check for state changes before waiting.
|
||||
//
|
||||
// -- Threads --
|
||||
//
|
||||
// A Thread object, when created, spawns a new thread with the given callback
|
||||
// (the callbacks must implement IRunnable). Threads have one method of
|
||||
// interest, Join(), which will block until the thread's execution finishes.
|
||||
// Deleting a thread object will free any operating system resources associated
|
||||
// with that thread, if the thread has finished executing.
|
||||
//
|
||||
// Threads can fail to spawn; make sure to check Succeeded().
|
||||
//
|
||||
|
||||
namespace ke {
|
||||
|
||||
// Abstraction for getting a unique thread identifier. Debug-only.
|
||||
#if defined(_MSC_VER)
|
||||
typedef DWORD ThreadId;
|
||||
|
||||
static inline ThreadId GetCurrentThreadId()
|
||||
{
|
||||
return ::GetCurrentThreadId();
|
||||
}
|
||||
#else
|
||||
typedef pthread_t ThreadId;
|
||||
|
||||
static inline ThreadId GetCurrentThreadId()
|
||||
{
|
||||
return pthread_self();
|
||||
}
|
||||
#endif
|
||||
|
||||
// Classes which use non-reentrant, same-thread lock/unlock semantics should
|
||||
// inherit from this and implement DoLock/DoUnlock.
|
||||
class Lockable
|
||||
{
|
||||
public:
|
||||
Lockable()
|
||||
{
|
||||
#if !defined(NDEBUG)
|
||||
owner_ = 0;
|
||||
#endif
|
||||
}
|
||||
virtual ~Lockable() {
|
||||
}
|
||||
|
||||
bool TryLock() {
|
||||
if (DoTryLock()) {
|
||||
DebugSetLocked();
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void Lock() {
|
||||
assert(Owner() != GetCurrentThreadId());
|
||||
DoLock();
|
||||
DebugSetLocked();
|
||||
}
|
||||
|
||||
void Unlock() {
|
||||
assert(Owner() == GetCurrentThreadId());
|
||||
DebugSetUnlocked();
|
||||
DoUnlock();
|
||||
}
|
||||
|
||||
void AssertCurrentThreadOwns() const {
|
||||
assert(Owner() == GetCurrentThreadId());
|
||||
}
|
||||
#if !defined(NDEBUG)
|
||||
bool Locked() const {
|
||||
return owner_ != 0;
|
||||
}
|
||||
ThreadId Owner() const {
|
||||
return owner_;
|
||||
}
|
||||
#endif
|
||||
|
||||
virtual bool DoTryLock() = 0;
|
||||
virtual void DoLock() = 0;
|
||||
virtual void DoUnlock() = 0;
|
||||
|
||||
protected:
|
||||
void DebugSetUnlocked() {
|
||||
#if !defined(NDEBUG)
|
||||
owner_ = 0;
|
||||
#endif
|
||||
}
|
||||
void DebugSetLocked() {
|
||||
#if !defined(NDEBUG)
|
||||
owner_ = GetCurrentThreadId();
|
||||
#endif
|
||||
}
|
||||
|
||||
protected:
|
||||
#if !defined(NDEBUG)
|
||||
ThreadId owner_;
|
||||
#endif
|
||||
};
|
||||
|
||||
// RAII for automatically locking and unlocking an object.
|
||||
class AutoLock
|
||||
{
|
||||
public:
|
||||
AutoLock(Lockable *lock)
|
||||
: lock_(lock)
|
||||
{
|
||||
lock_->Lock();
|
||||
}
|
||||
~AutoLock() {
|
||||
lock_->Unlock();
|
||||
}
|
||||
|
||||
private:
|
||||
Lockable *lock_;
|
||||
};
|
||||
|
||||
class AutoTryLock
|
||||
{
|
||||
public:
|
||||
AutoTryLock(Lockable *lock)
|
||||
{
|
||||
lock_ = lock->TryLock() ? lock : NULL;
|
||||
}
|
||||
~AutoTryLock() {
|
||||
if (lock_)
|
||||
lock_->Unlock();
|
||||
}
|
||||
|
||||
private:
|
||||
Lockable *lock_;
|
||||
};
|
||||
|
||||
// RAII for automatically unlocking and relocking an object.
|
||||
class AutoUnlock
|
||||
{
|
||||
public:
|
||||
AutoUnlock(Lockable *lock)
|
||||
: lock_(lock)
|
||||
{
|
||||
lock_->Unlock();
|
||||
}
|
||||
~AutoUnlock() {
|
||||
lock_->Lock();
|
||||
}
|
||||
|
||||
private:
|
||||
Lockable *lock_;
|
||||
};
|
||||
|
||||
enum WaitResult {
|
||||
// Woke up because something happened.
|
||||
Wait_Signaled,
|
||||
|
||||
// Woke up because nothing happened and a timeout was specified.
|
||||
Wait_Timeout,
|
||||
|
||||
// Woke up, but because of an error.
|
||||
Wait_Error
|
||||
};
|
||||
|
||||
// This must be implemented in order to spawn a new thread.
|
||||
class IRunnable
|
||||
{
|
||||
public:
|
||||
virtual ~IRunnable() {
|
||||
}
|
||||
|
||||
virtual void Run() = 0;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
// Include the actual thread implementations.
|
||||
#if defined(_MSC_VER)
|
||||
# include "am-thread-windows.h"
|
||||
#else
|
||||
# include "am-thread-posix.h"
|
||||
#endif
|
||||
|
||||
#endif // _include_amtl_threads_
|
@ -1,161 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_thread_windows_h_
|
||||
#define _include_amtl_thread_windows_h_
|
||||
|
||||
#include <windows.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
class CriticalSection : public Lockable
|
||||
{
|
||||
public:
|
||||
CriticalSection() {
|
||||
InitializeCriticalSection(&cs_);
|
||||
}
|
||||
~CriticalSection() {
|
||||
DeleteCriticalSection(&cs_);
|
||||
}
|
||||
|
||||
bool DoTryLock() KE_OVERRIDE {
|
||||
return !!TryEnterCriticalSection(&cs_);
|
||||
}
|
||||
void DoLock() KE_OVERRIDE {
|
||||
EnterCriticalSection(&cs_);
|
||||
}
|
||||
|
||||
void DoUnlock() KE_OVERRIDE {
|
||||
LeaveCriticalSection(&cs_);
|
||||
}
|
||||
|
||||
private:
|
||||
CRITICAL_SECTION cs_;
|
||||
};
|
||||
|
||||
typedef CriticalSection Mutex;
|
||||
|
||||
// Currently, this class only supports single-listener CVs.
|
||||
class ConditionVariable : public Lockable
|
||||
{
|
||||
public:
|
||||
ConditionVariable() {
|
||||
event_ = CreateEvent(NULL, FALSE, FALSE, NULL);
|
||||
}
|
||||
~ConditionVariable() {
|
||||
CloseHandle(event_);
|
||||
}
|
||||
|
||||
bool DoTryLock() KE_OVERRIDE {
|
||||
return cs_.DoTryLock();
|
||||
}
|
||||
void DoLock() KE_OVERRIDE {
|
||||
cs_.DoLock();
|
||||
}
|
||||
void DoUnlock() KE_OVERRIDE {
|
||||
cs_.DoUnlock();
|
||||
}
|
||||
|
||||
void Notify() {
|
||||
AssertCurrentThreadOwns();
|
||||
SetEvent(event_);
|
||||
}
|
||||
|
||||
WaitResult Wait(size_t timeout_ms) {
|
||||
// This will assert if the lock has not been acquired. We don't need to be
|
||||
// atomic here, like pthread_cond_wait, because the event bit will stick
|
||||
// until reset by a wait function.
|
||||
Unlock();
|
||||
DWORD rv = WaitForSingleObject(event_, timeout_ms);
|
||||
Lock();
|
||||
|
||||
if (rv == WAIT_TIMEOUT)
|
||||
return Wait_Timeout;
|
||||
if (rv == WAIT_FAILED)
|
||||
return Wait_Error;
|
||||
return Wait_Signaled;
|
||||
}
|
||||
|
||||
WaitResult Wait() {
|
||||
return Wait(INFINITE);
|
||||
}
|
||||
|
||||
private:
|
||||
CriticalSection cs_;
|
||||
HANDLE event_;
|
||||
};
|
||||
|
||||
class Thread
|
||||
{
|
||||
public:
|
||||
Thread(IRunnable *run, const char *name = NULL) {
|
||||
thread_ = CreateThread(NULL, 0, Main, run, 0, NULL);
|
||||
}
|
||||
~Thread() {
|
||||
if (!thread_)
|
||||
return;
|
||||
CloseHandle(thread_);
|
||||
}
|
||||
|
||||
bool Succeeded() const {
|
||||
return !!thread_;
|
||||
}
|
||||
|
||||
void Join() {
|
||||
if (!Succeeded())
|
||||
return;
|
||||
WaitForSingleObject(thread_, INFINITE);
|
||||
}
|
||||
|
||||
HANDLE handle() const {
|
||||
return thread_;
|
||||
}
|
||||
|
||||
private:
|
||||
static DWORD WINAPI Main(LPVOID arg) {
|
||||
((IRunnable *)arg)->Run();
|
||||
return 0;
|
||||
}
|
||||
|
||||
#pragma pack(push, 8)
|
||||
struct ThreadNameInfo {
|
||||
DWORD dwType;
|
||||
LPCSTR szName;
|
||||
DWORD dwThreadID;
|
||||
DWORD dwFlags;
|
||||
};
|
||||
#pragma pack(pop)
|
||||
|
||||
private:
|
||||
HANDLE thread_;
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_thread_windows_h_
|
@ -1,175 +0,0 @@
|
||||
// vim: set sts=2 ts=8 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_thread_local_h_
|
||||
#define _include_amtl_thread_local_h_
|
||||
|
||||
#include <am-thread-utils.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
// Stores a per-thread value. In single-threaded mode (KE_SINGLE_THREADED),
|
||||
// this is a no-op container wrapper.
|
||||
//
|
||||
// T must be castable to uintptr_t.
|
||||
//
|
||||
// When assigning to a ThreadLocal<T>, the assigment will automatically attempt
|
||||
// to allocate thread-local storage from the operating system. If it fails, it
|
||||
// will abort the program. If this is undesirable, you may call allocate()
|
||||
// up-front and handle the error case manually.
|
||||
//
|
||||
// The number of thread local slots available to processes is limited (on
|
||||
// Linux, it is generally 1024). It is best to use ThreadLocal sparingly to
|
||||
// play nicely with other libraries.
|
||||
//
|
||||
// ThreadLocal will free the underlying thread-local storage slot in its
|
||||
// destructor, but it is not an AutoPtr. It does not delete pointers. Since
|
||||
// one thread's value is only observable from that thread, make sure to free
|
||||
// the contained resource (if necessary) before the thread exits.
|
||||
template <typename T>
|
||||
class ThreadLocal
|
||||
{
|
||||
public:
|
||||
void operator =(const T &other) {
|
||||
set(other);
|
||||
}
|
||||
|
||||
T operator *() const {
|
||||
return get();
|
||||
}
|
||||
T operator ->() const {
|
||||
return get();
|
||||
}
|
||||
bool operator !() const {
|
||||
return !get();
|
||||
}
|
||||
bool operator ==(const T &other) const {
|
||||
return get() == other;
|
||||
}
|
||||
bool operator !=(const T &other) const {
|
||||
return get() != other;
|
||||
}
|
||||
|
||||
private:
|
||||
ThreadLocal(const ThreadLocal &other) KE_DELETE;
|
||||
ThreadLocal &operator =(const ThreadLocal &other) KE_DELETE;
|
||||
|
||||
#if !defined(KE_SINGLE_THREADED)
|
||||
private:
|
||||
int allocated_;
|
||||
|
||||
public:
|
||||
ThreadLocal() {
|
||||
allocated_ = 0;
|
||||
}
|
||||
|
||||
T get() const {
|
||||
if (!allocated_)
|
||||
return T();
|
||||
return internalGet();
|
||||
}
|
||||
void set(const T &t) {
|
||||
if (!allocated_ && !allocate()) {
|
||||
fprintf(stderr, "could not allocate thread-local storage\n");
|
||||
abort();
|
||||
}
|
||||
internalSet(t);
|
||||
}
|
||||
|
||||
# if defined(_MSC_VER)
|
||||
~ThreadLocal() {
|
||||
if (allocated_)
|
||||
TlsFree(key_);
|
||||
}
|
||||
|
||||
private:
|
||||
T internalGet() const {
|
||||
return reinterpret_cast<T>(TlsGetValue(key_));
|
||||
}
|
||||
void internalSet(const T &t) {
|
||||
TlsSetValue(key_, reinterpret_cast<LPVOID>(t));
|
||||
}
|
||||
bool allocate() {
|
||||
if (InterlockedCompareExchange(&allocated_, 1, 0) == 1)
|
||||
return true;
|
||||
key_ = TlsAlloc();
|
||||
return key_ != TLS_OUT_OF_INDEXES;
|
||||
}
|
||||
|
||||
DWORD key_;
|
||||
|
||||
# else
|
||||
public:
|
||||
~ThreadLocal() {
|
||||
if (allocated_)
|
||||
pthread_key_delete(key_);
|
||||
}
|
||||
|
||||
bool allocate() {
|
||||
if (!__sync_bool_compare_and_swap(&allocated_, 0, 1))
|
||||
return true;
|
||||
return pthread_key_create(&key_, NULL) == 0;
|
||||
}
|
||||
|
||||
private:
|
||||
T internalGet() const {
|
||||
return (T)reinterpret_cast<uintptr_t>(pthread_getspecific(key_));
|
||||
}
|
||||
void internalSet(const T &t) {
|
||||
pthread_setspecific(key_, reinterpret_cast<void *>(t));
|
||||
}
|
||||
|
||||
pthread_key_t key_;
|
||||
# endif // !_MSC_VER
|
||||
|
||||
#else // KE_SINGLE_THREADED
|
||||
public:
|
||||
ThreadLocal() {
|
||||
t_ = T();
|
||||
}
|
||||
|
||||
bool allocate() {
|
||||
return true;
|
||||
}
|
||||
|
||||
T get() const {
|
||||
return t_;
|
||||
}
|
||||
void set(const T &t) {
|
||||
t_ = t;
|
||||
}
|
||||
|
||||
private:
|
||||
T t_;
|
||||
#endif
|
||||
};
|
||||
|
||||
} // namespace ke
|
||||
|
||||
#endif // _include_amtl_thread_local_h_
|
@ -1,405 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _include_amtl_utility_h_
|
||||
#define _include_amtl_utility_h_
|
||||
|
||||
#include <assert.h>
|
||||
#include <stddef.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdint.h>
|
||||
#if defined(_MSC_VER)
|
||||
# include <intrin.h>
|
||||
#endif
|
||||
#include <am-moveable.h>
|
||||
|
||||
#define KE_32BIT
|
||||
|
||||
#if defined(_MSC_VER)
|
||||
# pragma warning(disable:4355)
|
||||
#endif
|
||||
|
||||
namespace ke {
|
||||
|
||||
static const size_t kMallocAlignment = sizeof(void *) * 2;
|
||||
|
||||
static const size_t kKB = 1024;
|
||||
static const size_t kMB = 1024 * kKB;
|
||||
static const size_t kGB = 1024 * kMB;
|
||||
|
||||
typedef uint8_t * Address;
|
||||
|
||||
template <typename T> T
|
||||
ReturnAndVoid(T &t)
|
||||
{
|
||||
T saved = t;
|
||||
t = T();
|
||||
return saved;
|
||||
}
|
||||
|
||||
#if __cplusplus >= 201103L
|
||||
# define KE_CXX11
|
||||
#endif
|
||||
|
||||
#if defined(KE_CXX11)
|
||||
# define KE_DELETE = delete
|
||||
# define KE_OVERRIDE override
|
||||
#else
|
||||
# define KE_DELETE
|
||||
# define KE_OVERRIDE
|
||||
#endif
|
||||
|
||||
// Wrapper that automatically deletes its contents. The pointer can be taken
|
||||
// to avoid destruction.
|
||||
template <typename T>
|
||||
class AutoPtr
|
||||
{
|
||||
T *t_;
|
||||
|
||||
public:
|
||||
AutoPtr()
|
||||
: t_(NULL)
|
||||
{
|
||||
}
|
||||
explicit AutoPtr(T *t)
|
||||
: t_(t)
|
||||
{
|
||||
}
|
||||
AutoPtr(Moveable<AutoPtr<T> > other)
|
||||
{
|
||||
t_ = other->t_;
|
||||
other->t_ = NULL;
|
||||
}
|
||||
~AutoPtr() {
|
||||
delete t_;
|
||||
}
|
||||
T *take() {
|
||||
return ReturnAndVoid(t_);
|
||||
}
|
||||
T *operator *() const {
|
||||
return t_;
|
||||
}
|
||||
T *operator ->() const {
|
||||
return t_;
|
||||
}
|
||||
operator T *() const {
|
||||
return t_;
|
||||
}
|
||||
T *operator =(T *t) {
|
||||
delete t_;
|
||||
t_ = t;
|
||||
return t_;
|
||||
}
|
||||
T *operator =(Moveable<AutoPtr<T> > other) {
|
||||
delete t_;
|
||||
t_ = other->t_;
|
||||
other->t_ = NULL;
|
||||
return t_;
|
||||
}
|
||||
bool operator !() const {
|
||||
return !t_;
|
||||
}
|
||||
|
||||
private:
|
||||
AutoPtr(const AutoPtr &other) KE_DELETE;
|
||||
AutoPtr &operator =(const AutoPtr &other) KE_DELETE;
|
||||
};
|
||||
|
||||
// Wrapper that automatically deletes its contents. The pointer can be taken
|
||||
// to avoid destruction.
|
||||
template <typename T>
|
||||
class AutoArray
|
||||
{
|
||||
T *t_;
|
||||
|
||||
public:
|
||||
AutoArray()
|
||||
: t_(NULL)
|
||||
{
|
||||
}
|
||||
explicit AutoArray(T *t)
|
||||
: t_(t)
|
||||
{
|
||||
}
|
||||
~AutoArray() {
|
||||
delete [] t_;
|
||||
}
|
||||
T *take() {
|
||||
return ReturnAndVoid(t_);
|
||||
}
|
||||
T *operator *() const {
|
||||
return t_;
|
||||
}
|
||||
T &operator [](size_t index) {
|
||||
return t_[index];
|
||||
}
|
||||
const T &operator [](size_t index) const {
|
||||
return t_[index];
|
||||
}
|
||||
operator T *() const {
|
||||
return t_;
|
||||
}
|
||||
void operator =(T *t) {
|
||||
delete [] t_;
|
||||
t_ = t;
|
||||
}
|
||||
bool operator !() const {
|
||||
return !t_;
|
||||
}
|
||||
};
|
||||
|
||||
static inline size_t
|
||||
Log2(size_t number)
|
||||
{
|
||||
assert(number != 0);
|
||||
|
||||
#if defined _MSC_VER
|
||||
unsigned long rval;
|
||||
# ifdef _M_IX86
|
||||
_BitScanReverse(&rval, number);
|
||||
# elif _M_X64
|
||||
_BitScanReverse64(&rval, number);
|
||||
# endif
|
||||
return rval;
|
||||
#elif defined __GNUC__
|
||||
return 31 - __builtin_clz(number);
|
||||
#else
|
||||
size_t bit;
|
||||
asm("bsr %1, %0\n"
|
||||
: "=r" (bit)
|
||||
: "rm" (number));
|
||||
return bit;
|
||||
#endif
|
||||
}
|
||||
|
||||
static inline size_t
|
||||
FindRightmostBit(size_t number)
|
||||
{
|
||||
assert(number != 0);
|
||||
|
||||
#if defined _MSC_VER
|
||||
unsigned long rval;
|
||||
# ifdef _M_IX86
|
||||
_BitScanForward(&rval, number);
|
||||
# elif _M_X64
|
||||
_BitScanForward64(&rval, number);
|
||||
# endif
|
||||
return rval;
|
||||
#elif defined __GNUC__
|
||||
return __builtin_ctz(number);
|
||||
#else
|
||||
size_t bit;
|
||||
asm("bsf %1, %0\n"
|
||||
: "=r" (bit)
|
||||
: "rm" (number));
|
||||
return bit;
|
||||
#endif
|
||||
}
|
||||
|
||||
static inline bool
|
||||
IsPowerOfTwo(size_t value)
|
||||
{
|
||||
if (value == 0)
|
||||
return false;
|
||||
return !(value & (value - 1));
|
||||
}
|
||||
|
||||
static inline size_t
|
||||
Align(size_t count, size_t alignment)
|
||||
{
|
||||
assert(IsPowerOfTwo(alignment));
|
||||
return count + (alignment - (count % alignment)) % alignment;
|
||||
}
|
||||
|
||||
static inline bool
|
||||
IsUint32AddSafe(unsigned a, unsigned b)
|
||||
{
|
||||
if (!a || !b)
|
||||
return true;
|
||||
size_t log2_a = Log2(a);
|
||||
size_t log2_b = Log2(b);
|
||||
return (log2_a < sizeof(unsigned) * 8) &&
|
||||
(log2_b < sizeof(unsigned) * 8);
|
||||
}
|
||||
|
||||
static inline bool
|
||||
IsUintPtrAddSafe(size_t a, size_t b)
|
||||
{
|
||||
if (!a || !b)
|
||||
return true;
|
||||
size_t log2_a = Log2(a);
|
||||
size_t log2_b = Log2(b);
|
||||
return (log2_a < sizeof(size_t) * 8) &&
|
||||
(log2_b < sizeof(size_t) * 8);
|
||||
}
|
||||
|
||||
static inline bool
|
||||
IsUint32MultiplySafe(unsigned a, unsigned b)
|
||||
{
|
||||
if (a <= 1 || b <= 1)
|
||||
return true;
|
||||
|
||||
size_t log2_a = Log2(a);
|
||||
size_t log2_b = Log2(b);
|
||||
return log2_a + log2_b <= sizeof(unsigned) * 8;
|
||||
}
|
||||
|
||||
static inline bool
|
||||
IsUintPtrMultiplySafe(size_t a, size_t b)
|
||||
{
|
||||
if (a <= 1 || b <= 1)
|
||||
return true;
|
||||
|
||||
size_t log2_a = Log2(a);
|
||||
size_t log2_b = Log2(b);
|
||||
return log2_a + log2_b <= sizeof(size_t) * 8;
|
||||
}
|
||||
|
||||
#define ARRAY_LENGTH(array) (sizeof(array) / sizeof(array[0]))
|
||||
#define STATIC_ASSERT(cond) extern int static_assert_f(int a[(cond) ? 1 : -1])
|
||||
|
||||
#define IS_ALIGNED(addr, alignment) (!(uintptr_t(addr) & ((alignment) - 1)))
|
||||
|
||||
template <typename T>
|
||||
static inline bool
|
||||
IsAligned(T addr, size_t alignment)
|
||||
{
|
||||
assert(IsPowerOfTwo(alignment));
|
||||
return !(uintptr_t(addr) & (alignment - 1));
|
||||
}
|
||||
|
||||
static inline Address
|
||||
AlignedBase(Address addr, size_t alignment)
|
||||
{
|
||||
assert(IsPowerOfTwo(alignment));
|
||||
return Address(uintptr_t(addr) & ~(alignment - 1));
|
||||
}
|
||||
|
||||
template <typename T> static inline T
|
||||
Min(const T &t1, const T &t2)
|
||||
{
|
||||
return t1 < t2 ? t1 : t2;
|
||||
}
|
||||
|
||||
template <typename T> static inline T
|
||||
Max(const T &t1, const T &t2)
|
||||
{
|
||||
return t1 > t2 ? t1 : t2;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
class StorageBuffer
|
||||
{
|
||||
public:
|
||||
T *address() {
|
||||
return reinterpret_cast<T *>(buffer_);
|
||||
}
|
||||
const T *address() const {
|
||||
return reinterpret_cast<const T *>(buffer_);
|
||||
}
|
||||
|
||||
private:
|
||||
union {
|
||||
char buffer_[sizeof(T)];
|
||||
uint64_t aligned_;
|
||||
};
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
class SaveAndSet
|
||||
{
|
||||
public:
|
||||
SaveAndSet(T *location, const T &value)
|
||||
: location_(location),
|
||||
old_(*location)
|
||||
{
|
||||
*location_ = value;
|
||||
}
|
||||
~SaveAndSet() {
|
||||
*location_ = old_;
|
||||
}
|
||||
|
||||
private:
|
||||
T *location_;
|
||||
T old_;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
class StackLinked
|
||||
{
|
||||
public:
|
||||
StackLinked<T>(T **prevp)
|
||||
: prevp_(prevp),
|
||||
prev_(*prevp)
|
||||
{
|
||||
*prevp_ = static_cast<T *>(this);
|
||||
}
|
||||
virtual ~StackLinked() {
|
||||
assert(*prevp_ == this);
|
||||
*prevp_ = prev_;
|
||||
}
|
||||
|
||||
private:
|
||||
T **prevp_;
|
||||
T *prev_;
|
||||
};
|
||||
|
||||
#if __cplusplus >= 201103L
|
||||
# define KE_CXX11
|
||||
#endif
|
||||
|
||||
#if defined(KE_CXX11)
|
||||
# define KE_DELETE = delete
|
||||
# define KE_OVERRIDE override
|
||||
#else
|
||||
# define KE_DELETE
|
||||
# define KE_OVERRIDE
|
||||
#endif
|
||||
|
||||
#if defined(_MSC_VER)
|
||||
# define KE_SIZET_FMT "%Iu"
|
||||
# define KE_I64_FMT "%I64d"
|
||||
# define KE_U64_FMT "%I64u"
|
||||
#elif defined(__GNUC__)
|
||||
# define KE_SIZET_FMT "%zu"
|
||||
# define KE_I64_FMT "%lld"
|
||||
# define KE_U64_FMT "%llu"
|
||||
#else
|
||||
# error "Implement format specifier string"
|
||||
#endif
|
||||
|
||||
#if defined(__GNUC__)
|
||||
# define KE_CRITICAL_LIKELY(x) __builtin_expect(!!(x), 1)
|
||||
#else
|
||||
# define KE_CRITICAL_LIKELY(x) x
|
||||
#endif
|
||||
|
||||
}
|
||||
|
||||
#endif // _include_amtl_utility_h_
|
@ -1,252 +0,0 @@
|
||||
// vim: set sts=8 ts=2 sw=2 tw=99 et:
|
||||
//
|
||||
// Copyright (C) 2013, David Anderson and AlliedModders LLC
|
||||
// All rights reserved.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright notice, this
|
||||
// list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
// * Neither the name of AlliedModders LLC nor the names of its contributors
|
||||
// may be used to endorse or promote products derived from this software
|
||||
// without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
// POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef _INCLUDE_KEIMA_TPL_CPP_VECTOR_H_
|
||||
#define _INCLUDE_KEIMA_TPL_CPP_VECTOR_H_
|
||||
|
||||
#include <new>
|
||||
#include <stdlib.h>
|
||||
#include <am-allocator-policies.h>
|
||||
#include <am-utility.h>
|
||||
#include <am-moveable.h>
|
||||
|
||||
namespace ke {
|
||||
|
||||
template <typename T, typename AllocPolicy = SystemAllocatorPolicy>
|
||||
class Vector : public AllocPolicy
|
||||
{
|
||||
public:
|
||||
Vector(AllocPolicy = AllocPolicy())
|
||||
: data_(NULL),
|
||||
nitems_(0),
|
||||
maxsize_(0)
|
||||
{
|
||||
}
|
||||
|
||||
Vector(Moveable<Vector<T, AllocPolicy> > other) {
|
||||
data_ = other->data_;
|
||||
nitems_ = other->nitems_;
|
||||
maxsize_ = other->maxsize_;
|
||||
other->reset();
|
||||
}
|
||||
|
||||
~Vector() {
|
||||
zap();
|
||||
}
|
||||
|
||||
bool append(const T &item) {
|
||||
if (!growIfNeeded(1))
|
||||
return false;
|
||||
new (&data_[nitems_]) T(item);
|
||||
nitems_++;
|
||||
return true;
|
||||
}
|
||||
bool append(Moveable<T> item) {
|
||||
if (!growIfNeeded(1))
|
||||
return false;
|
||||
new (&data_[nitems_]) T(item);
|
||||
nitems_++;
|
||||
return true;
|
||||
}
|
||||
void infallibleAppend(const T &item) {
|
||||
assert(growIfNeeded(1));
|
||||
new (&data_[nitems_]) T(item);
|
||||
nitems_++;
|
||||
}
|
||||
void infallibleAppend(Moveable<T> item) {
|
||||
assert(growIfNeeded(1));
|
||||
new (&data_[nitems_]) T(item);
|
||||
nitems_++;
|
||||
}
|
||||
|
||||
// Shift all elements including |at| up by one, and insert |item| at the
|
||||
// given position. If |at| is one greater than the last usable index,
|
||||
// i.e. |at == length()|, then this is the same as append(). No other
|
||||
// invalid indexes are allowed.
|
||||
//
|
||||
// This is a linear-time operation.
|
||||
bool insert(size_t at, const T &item) {
|
||||
if (at == length())
|
||||
return append(item);
|
||||
if (!moveUp(at))
|
||||
return false;
|
||||
new (&data_[at]) T(item);
|
||||
return true;
|
||||
}
|
||||
bool insert(size_t at, Moveable<T> item) {
|
||||
if (at == length())
|
||||
return append(item);
|
||||
if (!moveUp(at))
|
||||
return false;
|
||||
new (&data_[at]) T(item);
|
||||
return true;
|
||||
}
|
||||
|
||||
// Shift all elements at the given position down, removing the given
|
||||
// element. This is a linear-time operation.
|
||||
void remove(size_t at) {
|
||||
for (size_t i = at; i < length() - 1; i++)
|
||||
data_[i] = Moveable<T>(data_[i + 1]);
|
||||
pop();
|
||||
}
|
||||
|
||||
T popCopy() {
|
||||
T t = at(length() - 1);
|
||||
pop();
|
||||
return t;
|
||||
}
|
||||
void pop() {
|
||||
assert(nitems_);
|
||||
data_[nitems_ - 1].~T();
|
||||
nitems_--;
|
||||
}
|
||||
bool empty() const {
|
||||
return length() == 0;
|
||||
}
|
||||
size_t length() const {
|
||||
return nitems_;
|
||||
}
|
||||
T& at(size_t i) {
|
||||
assert(i < length());
|
||||
return data_[i];
|
||||
}
|
||||
const T& at(size_t i) const {
|
||||
assert(i < length());
|
||||
return data_[i];
|
||||
}
|
||||
T& operator [](size_t i) {
|
||||
return at(i);
|
||||
}
|
||||
const T& operator [](size_t i) const {
|
||||
return at(i);
|
||||
}
|
||||
void clear() {
|
||||
nitems_ = 0;
|
||||
}
|
||||
const T &back() const {
|
||||
return at(length() - 1);
|
||||
}
|
||||
T &back() {
|
||||
return at(length() - 1);
|
||||
}
|
||||
|
||||
T *buffer() const {
|
||||
return data_;
|
||||
}
|
||||
|
||||
bool ensure(size_t desired) {
|
||||
if (desired <= length())
|
||||
return true;
|
||||
|
||||
return growIfNeeded(desired - length());
|
||||
}
|
||||
|
||||
Vector &operator =(Moveable<Vector<T, AllocPolicy> > other) {
|
||||
data_ = other->data_;
|
||||
nitems_ = other->nitems_;
|
||||
maxsize_ = other->maxsize_;
|
||||
other->reset();
|
||||
return *this;
|
||||
}
|
||||
|
||||
private:
|
||||
// These are disallowed because they basically violate the failure handling
|
||||
// model for AllocPolicies and are also likely to have abysmal performance.
|
||||
Vector(const Vector<T> &other) KE_DELETE;
|
||||
Vector &operator =(const Vector<T> &other) KE_DELETE;
|
||||
|
||||
private:
|
||||
void zap() {
|
||||
for (size_t i = 0; i < nitems_; i++)
|
||||
data_[i].~T();
|
||||
this->free(data_);
|
||||
}
|
||||
void reset() {
|
||||
data_ = NULL;
|
||||
nitems_ = 0;
|
||||
maxsize_ = 0;
|
||||
}
|
||||
|
||||
bool moveUp(size_t at) {
|
||||
// Note: we don't use append() here. Passing an element as a Moveable into
|
||||
// insert() or append() can break, since the underlying storage could be
|
||||
// reallocated, invalidating the Moveable reference. Instead, we inline
|
||||
// the logic to append() to ensure growIfNeeded occurs before any
|
||||
// references are taken.
|
||||
if (!growIfNeeded(1))
|
||||
return false;
|
||||
new (&data_[nitems_]) T(Moveable<T>(data_[nitems_ - 1]));
|
||||
nitems_++;
|
||||
for (size_t i = nitems_ - 2; i > at; i--)
|
||||
data_[i] = Moveable<T>(data_[i - 1]);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool growIfNeeded(size_t needed)
|
||||
{
|
||||
if (!IsUintPtrAddSafe(nitems_, needed)) {
|
||||
this->reportAllocationOverflow();
|
||||
return false;
|
||||
}
|
||||
if (nitems_ + needed < maxsize_)
|
||||
return true;
|
||||
|
||||
size_t new_maxsize = maxsize_ ? maxsize_ : 8;
|
||||
while (nitems_ + needed > new_maxsize) {
|
||||
if (!IsUintPtrMultiplySafe(new_maxsize, 2)) {
|
||||
this->reportAllocationOverflow();
|
||||
return false;
|
||||
}
|
||||
new_maxsize *= 2;
|
||||
}
|
||||
|
||||
T* newdata = (T*)this->malloc(sizeof(T) * new_maxsize);
|
||||
if (newdata == NULL)
|
||||
return false;
|
||||
for (size_t i = 0; i < nitems_; i++) {
|
||||
new (&newdata[i]) T(Moveable<T>(data_[i]));
|
||||
data_[i].~T();
|
||||
}
|
||||
this->free(data_);
|
||||
|
||||
data_ = newdata;
|
||||
maxsize_ = new_maxsize;
|
||||
return true;
|
||||
}
|
||||
|
||||
private:
|
||||
T* data_;
|
||||
size_t nitems_;
|
||||
size_t maxsize_;
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
#endif /* _INCLUDE_KEIMA_TPL_CPP_VECTOR_H_ */
|
||||
|
@ -29,6 +29,8 @@
|
||||
* Version: $Id$
|
||||
*/
|
||||
|
||||
#include <am-moveable.h>
|
||||
|
||||
#ifndef _include_sourcemod_namehashset_h_
|
||||
#define _include_sourcemod_namehashset_h_
|
||||
|
||||
@ -117,14 +119,10 @@ public:
|
||||
return table_.findForAdd(aKey);
|
||||
}
|
||||
|
||||
void add(Insert &i, const T &value)
|
||||
template <typename U>
|
||||
void add(Insert &i, U &&value)
|
||||
{
|
||||
return table_.add(i, value);
|
||||
}
|
||||
|
||||
void add(Insert &i, ke::Moveable<T> value)
|
||||
{
|
||||
return table_.add(i, value);
|
||||
return table_.add(i, ke::Forward<U>(value));
|
||||
}
|
||||
|
||||
bool retrieve(const char *aKey, T *value)
|
||||
@ -137,22 +135,14 @@ public:
|
||||
return true;
|
||||
}
|
||||
|
||||
bool insert(const char *aKey, const T &value)
|
||||
template <typename U>
|
||||
bool insert(const char *aKey, U &&value)
|
||||
{
|
||||
CharsAndLength key(aKey);
|
||||
Insert i = table_.findForAdd(key);
|
||||
if (i.found())
|
||||
return false;
|
||||
return table_.add(i, value);
|
||||
}
|
||||
|
||||
bool insert(const char *aKey, ke::Moveable<T> value)
|
||||
{
|
||||
CharsAndLength key(aKey);
|
||||
Insert i = table_.findForAdd(key);
|
||||
if (i.found())
|
||||
return false;
|
||||
return table_.add(i, value);
|
||||
return table_.add(i, ke::Forward<U>(value));
|
||||
}
|
||||
|
||||
bool contains(const char *aKey)
|
||||
|
@ -5,7 +5,7 @@ binary = SM.Program(builder, 'spcomp')
|
||||
compiler = binary.compiler
|
||||
compiler.includes += [
|
||||
os.path.join(builder.sourcePath, 'public'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl', 'include'),
|
||||
os.path.join(builder.sourcePath, 'public', 'sourcepawn'),
|
||||
os.path.join(builder.currentSourcePath, '..', 'include'),
|
||||
os.path.join(builder.currentSourcePath, '..', 'third_party'),
|
||||
|
@ -7,7 +7,7 @@ Includes = [
|
||||
os.path.join(builder.currentSourcePath, '..', 'third_party'),
|
||||
os.path.join(builder.sourcePath, 'public'),
|
||||
os.path.join(builder.sourcePath, 'public', 'sourcepawn'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl'),
|
||||
os.path.join(builder.sourcePath, 'public', 'amtl', 'include'),
|
||||
os.path.join(builder.sourcePath, 'public', 'jit'),
|
||||
os.path.join(builder.sourcePath, 'public', 'jit', 'x86'),
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user