Final sync of AMTL.

public/amtl/am-fixedarray.h

@@ -0,0 +1,99 @@
+// 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;
+  }
+
+ 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_

public/amtl/am-threadlocal.h

@@ -0,0 +1,175 @@
+// 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_