fix native image compilation

At this commit, the internal testsuite has been shown to run as fast as
the current master.

Author: Killian Perlin

lkql_jit/language/src/main/java/com/adacore/lkql_jit/langkit_translator/passes/Hierarchy.java

@@ -7,6 +7,7 @@
 
 import com.adacore.libadalang.Libadalang;
 import com.adacore.libadalang.Libadalang.AdaNode;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import java.util.*;
 
 /**
@@ -74,6 +75,7 @@ private void become(int size, boolean[] matrix, HashMap<String, Integer> index)
         this.classNamesToIndex = index;
     }
 
+    @TruffleBoundary
     public static Hierarchy initial() {
         // Collect all inital classes
         final Class<? extends AdaNode>[] initialClasses = Libadalang.NODE_DESCRIPTION_MAP.values()
@@ -123,10 +125,12 @@ public boolean isInstance(String classX, String classY) {
             classNamesToIndex.get(classY)];
     }
 
+    @TruffleBoundary
     public void add(String classX) {
         addAll(Collections.singleton(classX));
     }
 
+    @TruffleBoundary
     public void addAll(Collection<String> classes) {
         final int size = this.size() + classes.size();
         final boolean[] matrix = new boolean[size * size];
@@ -172,10 +176,12 @@ public void addInstanceOfRelation(String classX, String classY) {
         }
     }
 
+    @TruffleBoundary
     public void remove(String classX) {
         removeAll(Collections.singleton(classX));
     }
 
+    @TruffleBoundary
     public void removeAll(Collection<String> classes) {
         // Compute all the subtypes of X to mask wich row/column to keep
         final boolean[] deleteMask = new boolean[size()];
@@ -230,6 +236,7 @@ public void removeAll(Collection<String> classes) {
     }
 
     // Primarily used in testing
+    @TruffleBoundary
     public boolean equals(Object otherObject) {
         if (otherObject == null) return false;
         if (otherObject instanceof Hierarchy other) {

lkql_jit/language/src/main/java/com/adacore/lkql_jit/nodes/pass/ClassDecl.java

@@ -5,6 +5,7 @@
 
 package com.adacore.lkql_jit.nodes.pass;
 
+import com.adacore.lkql_jit.exception.LKQLRuntimeException;
 import com.adacore.lkql_jit.nodes.LKQLNode;
 import com.oracle.truffle.api.frame.VirtualFrame;
 import com.oracle.truffle.api.source.SourceSection;
@@ -28,8 +29,7 @@ public ClassDecl(SourceSection location, String name, ArrayList<String> fields)
 
     @Override
     public Object executeGeneric(VirtualFrame frame) {
-        // TODO Auto-generated method stub
-        throw new UnsupportedOperationException("Unimplemented method 'executeGeneric'");
+        throw LKQLRuntimeException.shouldNotExecute(this);
     }
 
     @Override

lkql_jit/language/src/main/java/com/adacore/lkql_jit/nodes/pass/PassExpr.java

@@ -10,9 +10,11 @@
 import com.adacore.lkql_jit.nodes.expressions.Expr;
 import com.adacore.lkql_jit.nodes.expressions.value_read.ReadArgument;
 import com.adacore.lkql_jit.runtime.values.DynamicAdaNode;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import com.oracle.truffle.api.dsl.Fallback;
 import com.oracle.truffle.api.dsl.NodeChild;
 import com.oracle.truffle.api.dsl.Specialization;
+import com.oracle.truffle.api.frame.MaterializedFrame;
 import com.oracle.truffle.api.frame.VirtualFrame;
 import com.oracle.truffle.api.source.SourceSection;
 import java.util.*;
@@ -59,8 +61,10 @@ protected PassExpr(
     @Specialization
     public Object onDynamicAdaNode(VirtualFrame frame, DynamicAdaNode input) {
         final var typingContext = LKQLLanguage.getContext(this).getTypingContext();
-        typingContext.addAll(add.classes.stream().map(cd -> cd.name).toList());
-        final var updatedTree = getUpdatedTree(input, frame);
+        for (var c : add.classes) {
+            typingContext.add(c.name);
+        }
+        final var updatedTree = getUpdatedTree(input, frame.materialize());
         typingContext.removeAll(del.classes);
         return updatedTree;
     }
@@ -100,7 +104,8 @@ public Optional<Integer> getPreviousSlot() {
     // core logic
 
     // bottom up rewriting of the tree
-    private DynamicAdaNode getUpdatedTree(DynamicAdaNode tree, VirtualFrame frame) {
+    @TruffleBoundary
+    private DynamicAdaNode getUpdatedTree(DynamicAdaNode tree, MaterializedFrame frame) {
         // recurse on children first (bottom up)
         for (var child : List.copyOf(tree.children.entrySet())) {
             tree.children.put(child.getKey(), getUpdatedTree(child.getValue(), frame));

lkql_jit/language/src/main/java/com/adacore/lkql_jit/nodes/pass/RunPass.java

@@ -6,11 +6,14 @@
 package com.adacore.lkql_jit.nodes.pass;
 
 import com.adacore.lkql_jit.LKQLLanguage;
+import com.adacore.lkql_jit.exception.LKQLRuntimeException;
 import com.adacore.lkql_jit.langkit_translator.passes.Hierarchy;
 import com.adacore.lkql_jit.nodes.LKQLNode;
 import com.adacore.lkql_jit.runtime.values.AdaNodeProxy;
 import com.adacore.lkql_jit.runtime.values.LKQLFunction;
 import com.adacore.lkql_jit.utils.functions.FrameUtils;
+import com.adacore.lkql_jit.utils.functions.ObjectUtils;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import com.oracle.truffle.api.frame.VirtualFrame;
 import com.oracle.truffle.api.interop.ArityException;
 import com.oracle.truffle.api.interop.InteropLibrary;
@@ -36,35 +39,38 @@ public RunPass(SourceSection location, int slot) {
 
     @Override
     public Object executeGeneric(VirtualFrame frame) {
+        final var ctx = LKQLLanguage.getContext(this);
+
         // Build pass call-chain
         final Deque<LKQLFunction> callChain = new ArrayDeque<>();
         callChain.push((LKQLFunction) FrameUtils.readLocal(frame, slot));
         while (((PassExpr) callChain.peek().rootNode.getBody()).getPreviousSlot().isPresent()) {
             callChain.push(
                 (LKQLFunction) FrameUtils.readLocal(
                     frame,
-                    ((PassExpr) callChain.peek().rootNode.getBody()).getPreviousSlot().get()
+                    // orElse is necessary to garantee no exception for native image
+                    ((PassExpr) callChain.peek().rootNode.getBody()).getPreviousSlot().orElse(0)
                 )
             );
         }
 
         // Setup all units roots
-        final var roots = LKQLLanguage.getContext(this).getAllUnitsRoots();
+        final var roots = ctx.getAllUnitsRoots();
         final var units = new Object[roots.length];
         for (int i = 0; i < roots.length; i++) {
-            if (LKQLLanguage.getContext(this).isVerbose()) {
-                System.out.println(i + ")\n" + roots[i].dumpTree());
+            if (ctx.isVerbose()) {
+                debugAST(i, roots[i].dumpTree());
             }
             units[i] = AdaNodeProxy.convertAST(roots[i]);
         }
 
-        LKQLLanguage.getContext(this).setTypingContext(Hierarchy.initial());
+        ctx.setTypingContext(Hierarchy.initial());
 
         do {
-            final var pass = callChain.pop();
+            final var pass = callChain.poll();
 
-            if (LKQLLanguage.getContext(this).isVerbose()) {
-                System.out.println("running pass:" + pass);
+            if (ctx.isVerbose()) {
+                ctx.println(ObjectUtils.toString(pass));
             }
 
             for (int i = 0; i < units.length; i++) {
@@ -73,11 +79,11 @@ public Object executeGeneric(VirtualFrame frame) {
                 } catch (
                     UnsupportedTypeException | ArityException | UnsupportedMessageException e
                 ) {
-                    e.printStackTrace();
+                    LKQLRuntimeException.fromJavaException(e, this);
                 }
 
-                if (LKQLLanguage.getContext(this).isVerbose()) {
-                    System.out.println(i + ")\n" + units[i]);
+                if (ctx.isVerbose()) {
+                    debugAST(i, units[i]);
                 }
             }
         } while (!callChain.isEmpty());
@@ -89,6 +95,11 @@ public int getSlot() {
         return slot;
     }
 
+    @TruffleBoundary
+    private void debugAST(int num, Object ast) {
+        System.out.println(num + ")\n" + ast);
+    }
+
     @Override
     public String toString(int indentLevel) {
         // TODO Auto-generated method stub

lkql_jit/language/src/main/java/com/adacore/lkql_jit/runtime/values/AdaNodeProxy.java

@@ -7,6 +7,7 @@
 
 import com.adacore.langkit_support.LangkitSupport.NodeInterface;
 import com.adacore.libadalang.Libadalang;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import java.util.HashMap;
 
 /**
@@ -21,10 +22,12 @@ public class AdaNodeProxy extends DynamicAdaNode {
     /** The native libadalang node this class proxies to */
     public final NodeInterface base;
 
+    @TruffleBoundary
     public static AdaNodeProxy convertAST(NodeInterface root) {
         return new AdaNodeProxy(root);
     }
 
+    @TruffleBoundary
     private AdaNodeProxy(NodeInterface root) {
         super(
             ((Libadalang.AdaNode) root).getClassName(),

lkql_jit/language/src/main/java/com/adacore/lkql_jit/runtime/values/DynamicAdaNode.java

@@ -6,6 +6,7 @@
 package com.adacore.lkql_jit.runtime.values;
 
 import com.adacore.lkql_jit.runtime.values.bases.BasicLKQLValue;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import java.util.HashMap;
 
 /**
@@ -24,6 +25,7 @@ public class DynamicAdaNode extends BasicLKQLValue {
      * This constructor is the one used during lowering
      * and should be the default one
      */
+    @TruffleBoundary
     public DynamicAdaNode(
         String kind,
         HashMap<String, DynamicAdaNode> children,
@@ -40,6 +42,7 @@ public DynamicAdaNode(
      * This constructor is used only in the context of a nanopass
      * by a constructor call
      */
+    @TruffleBoundary
     public DynamicAdaNode(String kind, Object[] args, String[] argnames) {
         this.kind = kind;
         this.children = new HashMap<>();
@@ -56,6 +59,7 @@ public DynamicAdaNode(String kind, Object[] args, String[] argnames) {
         this.isList = false;
     }
 
+    @TruffleBoundary
     public Object getField(String name) {
         var res = fields.get(name);
         return res != null ? res : children.get(name);