Store total element count in tuple-get instruction

3 files changed, 8 insertions, 2 deletions

diff --git a/src/main/java/org/shedlang/compiler/stackir/InstructionSerializer.java b/src/main/java/org/shedlang/compiler/stackir/InstructionSerializer.java
index fef6ce82..d8ded6d3 100644
--- a/src/main/java/org/shedlang/compiler/stackir/InstructionSerializer.java
+++ b/src/main/java/org/shedlang/compiler/stackir/InstructionSerializer.java
@@ -185,7 +185,8 @@ public class InstructionSerializer {
             case TupleGetInstruction tupleGetInstruction ->
                 treeList(
                     treeIdentifier("tuple-get"),
-                    treeInt(tupleGetInstruction.elementIndex())
+                    treeInt(tupleGetInstruction.elementIndex()),
+                    treeInt(tupleGetInstruction.elementCount())
                 );
 
             case VarLoadInstruction varLoadInstruction ->
diff --git a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
index 8c5bea8c..d036eebe 100644
--- a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
+++ b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
@@ -395,7 +395,11 @@ public class StackCompiler {
                     }
 
                     context = context
-                        .addInstruction(new TupleGetInstruction(elementIndex, tuple.source()));
+                        .addInstruction(new TupleGetInstruction(
+                            elementIndex,
+                            tuple.elements().size(),
+                            tuple.source()
+                        ));
 
                     context = compileTarget(tuple.elements().get(elementIndex), context);
                 }
diff --git a/src/main/java/org/shedlang/compiler/stackir/instructions/TupleGetInstruction.java b/src/main/java/org/shedlang/compiler/stackir/instructions/TupleGetInstruction.java
index b621a336..bf2e0456 100644
--- a/src/main/java/org/shedlang/compiler/stackir/instructions/TupleGetInstruction.java
+++ b/src/main/java/org/shedlang/compiler/stackir/instructions/TupleGetInstruction.java
@@ -8,6 +8,7 @@ import org.shedlang.compiler.sources.Source;
  */
 public record TupleGetInstruction(
     int elementIndex,
+    int elementCount,
     Source source
 ) implements Instruction {
 }