Add optimization for generator bodies that don't introduce new members (#1013)

If a generator object literal doesn't add any object members, we
can simply use the parent in its place.

Co-authored-by: Kushal Pisavadia <kushi.p@gmail.com>

pkl-core/src/main/java/org/pkl/core/ast/expression/generator/GeneratorObjectLiteralNode.java

@@ -73,6 +73,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = "checkObjectCannotHaveParameters()")
   protected VmDynamic evalDynamic(VirtualFrame frame, VmDynamic parent) {
     var data = executeChildren(frame, parent, parent.getLength());
+    if (data.hasNoMembers()) {
+      return parent;
+    }
     var result = new VmDynamic(frame.materialize(), parent, data.members(), data.length());
     return data.storeGeneratorFrames(result);
   }
@@ -81,6 +84,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   protected VmTyped evalTyped(VirtualFrame frame, VmTyped parent) {
     VmUtils.checkIsInstantiable(parent.getVmClass(), getParentNode());
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return parent;
+    }
     assert data.hasNoGeneratorFrames();
     return new VmTyped(frame.materialize(), parent, parent.getVmClass(), data.members());
   }
@@ -88,6 +94,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = "checkListingCannotHaveParameters()")
   protected VmListing evalListing(VirtualFrame frame, VmListing parent) {
     var data = executeChildren(frame, parent, parent.getLength());
+    if (data.hasNoMembers()) {
+      return parent;
+    }
     var result = new VmListing(frame.materialize(), parent, data.members(), data.length());
     return data.storeGeneratorFrames(result);
   }
@@ -95,6 +104,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = "checkMappingCannotHaveParameters()")
   protected VmMapping evalMapping(VirtualFrame frame, VmMapping parent) {
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return parent;
+    }
     var result = new VmMapping(frame.materialize(), parent, data.members());
     return data.storeGeneratorFrames(result);
   }
@@ -118,6 +130,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = {"parent == getDynamicClass()", "checkObjectCannotHaveParameters()"})
   protected VmDynamic evalDynamicClass(VirtualFrame frame, VmClass parent) {
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return VmDynamic.empty();
+    }
     var result =
         new VmDynamic(frame.materialize(), parent.getPrototype(), data.members(), data.length());
     return data.storeGeneratorFrames(result);
@@ -126,6 +141,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = {"parent == getMappingClass()", "checkMappingCannotHaveParameters()"})
   protected VmMapping evalMappingClass(VirtualFrame frame, VmClass parent) {
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return VmMapping.empty();
+    }
     var result = new VmMapping(frame.materialize(), parent.getPrototype(), data.members());
     return data.storeGeneratorFrames(result);
   }
@@ -133,6 +151,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   @Specialization(guards = {"parent == getListingClass()", "checkListingCannotHaveParameters()"})
   protected VmListing evalListingClass(VirtualFrame frame, VmClass parent) {
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return VmListing.empty();
+    }
     var result =
         new VmListing(frame.materialize(), parent.getPrototype(), data.members(), data.length());
     return data.storeGeneratorFrames(result);
@@ -142,6 +163,9 @@ public abstract class GeneratorObjectLiteralNode extends ObjectLiteralNode {
   protected VmTyped evalTypedObjectClass(VirtualFrame frame, VmClass parent) {
     VmUtils.checkIsInstantiable(parent, getParentNode());
     var data = executeChildren(frame, parent, 0);
+    if (data.hasNoMembers()) {
+      return parent.getPrototype();
+    }
     assert data.hasNoGeneratorFrames();
     return new VmTyped(frame.materialize(), parent.getPrototype(), parent, data.members());
   }

pkl-core/src/main/java/org/pkl/core/ast/expression/generator/ObjectData.java

@@ -55,6 +55,10 @@ public final class ObjectData {
     return length;
   }
 
+  boolean hasNoMembers() {
+    return members.isEmpty();
+  }
+
   boolean hasNoGeneratorFrames() {
     return generatorFrames.isEmpty();
   }