Switching to AsmJit generated chasing routines.

1 files changed, 101 insertions, 0 deletions

diff --git a/lib/AsmJit/CodeGenerator.cpp b/lib/AsmJit/CodeGenerator.cpp
new file mode 100644
index 0000000..180c918
--- /dev/null
+++ b/lib/AsmJit/CodeGenerator.cpp
@@ -0,0 +1,101 @@
+// AsmJit - Complete JIT Assembler for C++ Language.
+
+// Copyright (c) 2008-2010, Petr Kobalicek <kobalicek.petr@gmail.com>
+//
+// Permission is hereby granted, free of charge, to any person
+// obtaining a copy of this software and associated documentation
+// files (the "Software"), to deal in the Software without
+// restriction, including without limitation the rights to use,
+// copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the
+// Software is furnished to do so, subject to the following
+// conditions:
+//
+// The above copyright notice and this permission notice shall be
+// included in all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+// OTHER DEALINGS IN THE SOFTWARE.
+
+// [Dependencies]
+#include "Assembler.h"
+#include "CodeGenerator.h"
+#include "Defs.h"
+#include "MemoryManager.h"
+
+namespace AsmJit {
+
+// ============================================================================
+// [AsmJit::CodeGenerator]
+// ============================================================================
+
+CodeGenerator::CodeGenerator()
+{
+}
+
+CodeGenerator::~CodeGenerator()
+{
+}
+
+CodeGenerator* CodeGenerator::getGlobal()
+{
+  static JitCodeGenerator global;
+  return &global;
+}
+
+// ============================================================================
+// [AsmJit::JitCodeGenerator]
+// ============================================================================
+
+JitCodeGenerator::JitCodeGenerator() :
+  _memoryManager(NULL),
+  _allocType(MEMORY_ALLOC_FREEABLE)
+{
+}
+
+JitCodeGenerator::~JitCodeGenerator()
+{
+}
+
+uint32_t JitCodeGenerator::generate(void** dest, Assembler* assembler)
+{
+  // Disallow empty code generation.
+  sysuint_t codeSize = assembler->getCodeSize();
+  if (codeSize == 0)
+  {
+    *dest = NULL;
+    return AsmJit::ERROR_NO_FUNCTION;
+  }
+
+  // Switch to global memory manager if not provided.
+  MemoryManager* memmgr = getMemoryManager();
+  if (memmgr == NULL) memmgr = MemoryManager::getGlobal();
+
+  void* p = memmgr->alloc(codeSize, getAllocType());
+  if (p == NULL)
+  {
+    *dest = NULL;
+    return ERROR_NO_VIRTUAL_MEMORY;
+  }
+
+  // Relocate the code.
+  sysuint_t relocatedSize = assembler->relocCode(p);
+
+  // Return unused memory to mamory-manager.
+  if (relocatedSize < codeSize)
+  {
+    memmgr->shrink(p, relocatedSize);
+  }
+
+  // Return the code.
+  *dest = p;
+  return ERROR_NONE;
+}
+
+} // AsmJit namespace