[GR-23234] Implement gi_frame

PullRequest: graalpython/1046

Author: msimacek

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/generator/GeneratorBuiltins.java

@@ -44,6 +44,7 @@
 import com.oracle.graal.python.builtins.objects.PNone;
 import com.oracle.graal.python.builtins.objects.code.PCode;
 import com.oracle.graal.python.builtins.objects.common.SequenceNodes.GetObjectArrayNode;
+import com.oracle.graal.python.builtins.objects.dict.PDict;
 import com.oracle.graal.python.builtins.objects.exception.PBaseException;
 import com.oracle.graal.python.builtins.objects.frame.PFrame;
 import com.oracle.graal.python.builtins.objects.function.PArguments;
@@ -66,6 +67,7 @@
 import com.oracle.graal.python.nodes.function.builtins.PythonUnaryBuiltinNode;
 import com.oracle.graal.python.nodes.object.IsBuiltinClassProfile;
 import com.oracle.graal.python.runtime.exception.PException;
+import com.oracle.graal.python.runtime.object.PythonObjectFactory;
 import com.oracle.graal.python.util.PythonUtils;
 import com.oracle.truffle.api.CallTarget;
 import com.oracle.truffle.api.CompilerDirectives;
@@ -468,6 +470,30 @@ static Object getRunning(PGenerator self) {
         }
     }
 
+    @Builtin(name = "gi_frame", minNumOfPositionalArgs = 1, isGetter = true)
+    @GenerateNodeFactory
+    public abstract static class GetFrameNode extends PythonUnaryBuiltinNode {
+        @Specialization
+        static Object getFrame(PGenerator self,
+                        @Cached PythonObjectFactory factory) {
+            if (self.isFinished()) {
+                return PNone.NONE;
+            } else {
+                MaterializedFrame generatorFrame = PArguments.getGeneratorFrame(self.getArguments());
+                PDict locals = PArguments.getGeneratorFrameLocals(generatorFrame);
+                Object[] arguments = PArguments.create();
+                // TODO msimacek - this way, the generator will always have lineno == 1
+                Node location = self.getCurrentCallTarget().getRootNode();
+                PFrame frame = factory.createPFrame(PFrame.Reference.EMPTY, location, locals, false);
+                PArguments.setGlobals(arguments, PArguments.getGlobals(self.getArguments()));
+                PArguments.setClosure(arguments, PArguments.getClosure(self.getArguments()));
+                PArguments.setGeneratorFrame(arguments, generatorFrame);
+                frame.setArguments(arguments);
+                return frame;
+            }
+        }
+    }
+
     @Builtin(name = __REPR__, minNumOfPositionalArgs = 1)
     @GenerateNodeFactory
     abstract static class ReprNode extends PythonUnaryBuiltinNode {