/* vim: set ts=4 sw=4 tw=99 et:

  *

  * Copyright (C) 2012 David Anderson

  *

  * This file is part of SourcePawn.

  *

  * SourcePawn is free software: you can redistribute it and/or modify it under

  * the terms of the GNU General Public License as published by the Free

  * Software Foundation, either version 3 of the License, or (at your option)

  * any later version.

  * 

  * SourcePawn is distributed in the hope that it will be useful, but WITHOUT ANY

  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

  *

  * You should have received a copy of the GNU General Public License along with

  * SourcePawn. If not, see http://www.gnu.org/licenses/.

  */

 #ifndef _include_jitcraft_thread_h_

 #define _include_jitcraft_thread_h_

 #include <stdarg.h>

 #include "PoolAllocator.h"

 #include "Stack.h"

 #include "Heap.h"

 #include "Barriers.h"

 #include "Handles.h"

 #include "Messages.h"

 #include "RootScopes.h"

 #include "VirtualObjectVisitor.h"

 namespace ke {

 class TypeManager;

 class String;

 class StringPool;

 class GlobalRoots;

 class ArrayMap;

 class Exception;

 class NativeTable;

 class PackageCache;

 class Zone

 {

     PoolAllocator pool_;

     Stack stack_;

     Heap heap_;

     // Map cache.

     Unbarriered<Map> maps_[MapKinds_Total];

     Unbarriered<ArrayMap> externalGenericArrayMap_;

     Unbarriered<ArrayMap> externalCharArrayMap_;

     TypeManager *types_;

     StringPool *symbols_;

     GlobalRoots *globalRoots_;

     NativeTable *nativeTable_;

     PackageCache *packages_;

     // Root list.

     Local<Object> *lastLocalRoot_;

     RootScopeManager rootScopes_;

     AutoRooted *specialRoots_;

     // Exception stuff.

     bool hasPendingException_;

     Ke::IMessageReporter *reporter_;

     Ke::IPublicFunctionListener *publicFunctionListener_;

   private:

     Zone *thisFromCtor() {

         return this;

     }

   public:

     Zone();

     ~Zone();

     bool initialize();

     PoolAllocator &pool() {

         return pool_;

     }

     Stack &stack() {

         return stack_;

     }

     Heap &heap() {

         return heap_;

     }

     Object *allocate(MapKind kind, size_t bytes, Heap::Tenure tenure = Heap::Tenure_Default) {

         assert(maps_[kind]);

         Local<Map> map(this, maps_[kind]);

         return allocate(map, bytes, tenure);

     }

     Object *allocate(Handle<Map> map, size_t bytes, Heap::Tenure tenure = Heap::Tenure_Default) {

         return heap_.allocate(map, bytes, tenure);

     }

     Map *getMap(MapKind kind) {

         return maps_[kind];

     }

     Map *allocateMap(MapKind kind, size_t bytes, MapKind parent = MapKind_Meta);

     ArrayMap *externalGenericArrayMap() {

         return externalGenericArrayMap_;

     }

     ArrayMap *externalCharArrayMap() {

         return externalCharArrayMap_;

     }

     TypeManager *types() const {

         return types_;

     }

     AutoRooted **lastSpecialRoot() {

         return &specialRoots_;

     }

     Local<Object> **lastLocalRoot() {

         return &lastLocalRoot_;

     }

     Object **makeScopedRoot(Object *ptr) {

         return rootScopes_.make(ptr);

     }

     RootScopeManager &rootScopes() {

         return rootScopes_;

     }

     StringPool *symbols() {

         return symbols_;

     }

     GlobalRoots *globalRoots() {

         return globalRoots_;

     }

     NativeTable *nativeTable() {

         return nativeTable_;

     }

     PackageCache *packages() {

         return packages_;

     }

     String *makeSymbol(const char *string, size_t length);

     String *makeSymbol(const char *string);

     void reportOutOfMemory();

     void reportAllocationOverflow();

     void reportError(Message message, ...);

     void reportErrorVA(Message message, va_list ap);

     bool hasPendingException();

     void clearPendingException();

     void installErrorReporter(Ke::IMessageReporter *reporter);

     Ke::IMessageReporter *errorReporter() {

         return reporter_;

     }

     void accept(VirtualObjectVisitor *visitor);

     void sweepWeakRoots();

     void setPublicFunctionListener(Ke::IPublicFunctionListener *listener);

     void registerPublic(Handle<Function> fun);

     void unregisterPublic(Handle<Function> fun);

     // Helpers that report OOM.

     void *malloc(size_t size);

     void *realloc(void *ptr, size_t size);

 };

 // In a multi-threaded system, this would be a TLS variable.

 // In the interest of simplicity, we are single-threaded for now.

 extern Zone *SingleThreadedZone;

 // Exposed for the compiler.

 char *BuildErrorMessage(const char *format, va_list ap);

 void FreeErrorMessage(const char *message);

 static inline Zone *ZONE()

 {

     return SingleThreadedZone;

 }

 template <typename T>

 Local<T>::Local(Zone *zone, T *thing)

   : zone_(zone),

     thing_(thing),

     prev_(*zone->lastLocalRoot())

 {

     *zone->lastLocalRoot() = toRaw();

 }

 template <typename T>

 Local<T>::~Local()

 {

     assert(*zone_->lastLocalRoot() == toRaw());

     *zone_->lastLocalRoot() = prev_;

 }

 template <typename T>

 ScopedRoot<T>::ScopedRoot()

   : thingp_(make(NULL))

 {

 }

 template <typename T>

 ScopedRoot<T>::ScopedRoot(T *thing)

   : thingp_(make(thing))

 {

     // Type check that T is an Object.

     (void)static_cast<Object *>(thing);

 }

 template <typename T> T **

 ScopedRoot<T>::make(T *other)

 {

     return reinterpret_cast<T **>(ZONE()->makeScopedRoot(other));

 }

 RootScope::RootScope(Zone *zone)

   : Ke::HandleScope(zone, zone->rootScopes().position(), *zone->rootScopes().scope())

 {

 }

 }

 #endif // _include_jitcraft_thread_h_