exposed a bit more JIT

Josh Klontz
1 parent fe840eb4
Showing 2 changed files with 25 additions and 16 deletions
sdk/jitcv/jitcv.h
sdk/plugins/llvm.cpp
@@ -97,6 +97,9 @@ void jit_binary_apply(const jit_binary_kernel &amp;kernel, const jit_matrix &amp;src, ji
 jit_unary_kernel jit_square();
+typedef void (*jit_unary_core_t)(const jit_matrix *src, jit_matrix *dst, uint32_t size);
+typedef void (*jit_binary_core_t)(const jit_matrix *srcA, const jit_matrix *srcB, jit_matrix *dst, uint32_t size);
+
 #ifdef __cplusplus
 }
 #endif
@@ -250,8 +250,7 @@ class UnaryKernel : public UntrainableMetaTransform
 {
     Q_OBJECT
-    typedef void (*kernel_t)(const jit_matrix*, jit_matrix*, quint32);
-    kernel_t kernel;
+    jit_unary_core_t kernel;
     quint16 hash;
 public:
@@ -267,6 +266,19 @@ public:
         invoke(src, dst, size);
     }
+    jit_unary_core_t getKernel(const jit_matrix *src) const
+    {
+        const QString functionName = mangledName(*src);
+        Function *function = TheModule->getFunction(qPrintable(functionName));
+        if (function == NULL) {
+            function = compile(*src);
+            while (TheFunctionPassManager->run(*function));
+            TheExtraFunctionPassManager->run(*function);
+            function = TheModule->getFunction(qPrintable(functionName));
+        }
+        return (jit_unary_core_t)TheExecutionEngine->getPointerToFunction(function);
+    }
+
 private:
     QString mangledName() const
     {
@@ -362,17 +374,7 @@ private:
             QMutexLocker locker(&compilerLock);
             if (src.hash != hash) {
-                const QString functionName = mangledName(src);
-
-                Function *function = TheModule->getFunction(qPrintable(functionName));
-                if (function == NULL) {
-                    function = compile(src);
-                    while (TheFunctionPassManager->run(*function));
-                    TheExtraFunctionPassManager->run(*function);
-                    function = TheModule->getFunction(qPrintable(functionName));
-                }
-
-                const_cast<UnaryKernel*>(this)->kernel = (kernel_t)TheExecutionEngine->getPointerToFunction(function);
+                const_cast<UnaryKernel*>(this)->kernel = getKernel(&src);
                 const_cast<UnaryKernel*>(this)->hash = src.hash;
             }
         }
@@ -389,8 +391,7 @@ class BinaryKernel: public UntrainableMetaTransform
 {
     Q_OBJECT
-    typedef void (*kernel_t)(const jit_matrix*, const jit_matrix*, jit_matrix*, quint32);
-    kernel_t kernel;
+    jit_binary_core_t kernel;
     quint16 hashA, hashB;
 public:
@@ -467,7 +468,7 @@ private:
                     function = TheModule->getFunction(qPrintable(functionName));
                 }
-                const_cast<BinaryKernel*>(this)->kernel = (kernel_t)TheExecutionEngine->getPointerToFunction(function);
+                const_cast<BinaryKernel*>(this)->kernel = (jit_binary_core_t)TheExecutionEngine->getPointerToFunction(function);
                 const_cast<BinaryKernel*>(this)->hashA = srcA.hash;
                 const_cast<BinaryKernel*>(this)->hashB = srcB.hash;
             }
@@ -1003,4 +1004,9 @@ jit_unary_kernel jit_square()
     return &transform;
 }
+jit_unary_core_t jit_compile_unary_core(const void *kernel, const jit_matrix &m)
+{
+    return ((const UnaryKernel*)kernel)->getKernel(&m);
+}
+
 #include "llvm.moc"