Finishing Touches!

data/function.cpp

@@ -29,12 +29,4 @@ __Function<Ret(Args...)>::__Function(const char *const name_, __Function<Ret(Arg
     thunk(thunk_) {}
 
 // Thunks
-template <typename Ret, typename... Args>
-void __Function<Ret(Args...)>::enable_thunk(const thunk_enabler_t &thunk_enabler) {
-    if (enabled) {
-        ptr_type real_thunk = (ptr_type) thunk_enabler((void *) get(), (void *) thunk);
-        if (!is_virtual) {
-            func = real_thunk;
-        }
-    }
-}
+thunk_enabler_t thunk_enabler;

data/function.h

@@ -18,7 +18,7 @@ class __VirtualFunctionInfo {
 
 // Thunks
 typedef void *(*thunk_enabler_t)(void *target, void *thunk);
-void enable_all_thunks(const thunk_enabler_t &thunk_enabler);
+extern thunk_enabler_t thunk_enabler;
 
 // Function Information
 template <typename Ret, typename... Args>
@@ -40,27 +40,31 @@ public:
         if (!enabled) {
             return false;
         }
+        // Enable Thunk
+        enable_thunk();
         // Overwrite
         type original = get_thunk_target();
         thunk_target = [original, target](Args... args) {
-            return target(original, args...);
+            return target(original, std::forward<Args>(args)...);
         };
         return true;
     }
 
     // Getters
-    [[nodiscard]] ptr_type get_backup() const {
-        return backup;
-    }
-    [[nodiscard]] ptr_type get() const {
+    [[nodiscard]] ptr_type get(bool result_will_be_stored) {
         if (!enabled) {
             return nullptr;
-        } else if (is_virtual) {
+        } else {
+            if (result_will_be_stored) {
+                enable_thunk();
+            }
+            if (is_virtual) {
                 return *get_vtable_addr();
             } else {
                 return std::get<ptr_type>(func);
             }
         }
+    }
     [[nodiscard]] ptr_type *get_vtable_addr() const {
         if (is_virtual) {
             return std::get<__VirtualFunctionInfo<ptr_type>>(func).addr;
@@ -68,14 +72,11 @@ public:
             return nullptr;
         }
     }
-    [[nodiscard]] const char *get_name() const {
-        return name;
-    }
     [[nodiscard]] type get_thunk_target() const {
         if (thunk_target) {
             return thunk_target;
         } else {
-            return get_backup();
+            return backup;
         }
     }
 
@@ -84,6 +85,7 @@ private:
     const bool is_virtual;
     std::variant<ptr_type, __VirtualFunctionInfo<ptr_type>> func;
 
+public:
     // State
     const bool enabled;
     const char *const name;
@@ -91,9 +93,18 @@ private:
     // Backup Of Original Function Pointer
     const ptr_type backup;
 
+private:
     // Thunk
     const ptr_type thunk;
     type thunk_target;
-    void enable_thunk(const thunk_enabler_t &thunk_enabler);
-    friend void enable_all_thunks(const thunk_enabler_t &);
+    bool thunk_enabled = false;
+    void enable_thunk() {
+        if (!thunk_enabled && enabled) {
+            ptr_type real_thunk = (ptr_type) thunk_enabler((void *) backup, (void *) thunk);
+            if (!is_virtual) {
+                func = real_thunk;
+            }
+            thunk_enabled = true;
+        }
+    }
 };

data/out.cpp

@@ -13,8 +13,3 @@ struct __Thunk<Ret(Args...)> {
 };
 
 {{ main }}
-
-// Enable All Thunks
-void enable_all_thunks(const thunk_enabler_t &thunk_enabler) {
-{{ enableThunks }}
-}

src/index.ts

@@ -157,25 +157,22 @@ function makeCompiledCode(output: string) {
 
     // Generate
     let declarations = '';
-    let enableThunks = '';
     for (const structure of structureObjects) {
         const name = structure.getName();
         declarations += `// ${name}\n`;
         try {
             declarations += structure.generateCode();
-            enableThunks += structure.generateEnableThunks();
         } catch (e) {
             console.log(`Error Generating Code: ${name}: ${e instanceof Error ? e.stack : e}`);
             process.exit(1);
         }
         declarations += '\n';
     }
-    enableThunks = enableThunks.slice(0, -1); // Remove Last Newline
 
     // Write
     const headerPath = fs.realpathSync(headerOutput);
     const main = declarations.trim();
-    const result = formatFile('out.cpp', {headerPath, main, enableThunks, data: getDataDir()});
+    const result = formatFile('out.cpp', {headerPath, main, data: getDataDir()});
     fs.writeFileSync(output, result);
 }
 makeCompiledCode(sourceOutput);

src/struct.ts

@@ -144,7 +144,7 @@ export class Struct {
         if (isVirtual) {
             call += `this->vtable->${method.shortName}`;
         } else {
-            call += `${method.getName()}->get()`;
+            call += `${method.getName()}->get(false)`;
         }
         call += '(';
         if (!method.isStatic) {
@@ -269,27 +269,4 @@ export class Struct {
     setDirectParent(directParent: string) {
         this.#directParent = directParent;
     }
-
-    // Generate Part Of enable_all_thunks()
-    generateEnableThunks() {
-        // Get All Methods
-        const allMethods: Method[] = [];
-        for (const method of this.#methods) {
-            allMethods.push(method);
-        }
-        if (this.#vtable !== null && this.#vtable.canGenerateWrappers()) {
-            const virtualMethods = this.#vtable.getMethods();
-            for (const method of virtualMethods) {
-                if (method) {
-                    allMethods.push(method);
-                }
-            }
-        }
-        // Generate
-        let out = '';
-        for (const method of allMethods) {
-            out += `${INDENT}${method.getName()}->enable_thunk(thunk_enabler);\n`;
-        }
-        return out;
-    }
 }