From f71ae2c3c37a564bb9ced0e378ed08d805657fc2 Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 15 Jul 2024 22:04:06 -0400
Subject: [PATCH 1/7] Minor changes to .g4 file, removes need for
 StatementStatement in ASTBuilder

---
 src/main/antlr/SFML.g4                        | 18 ++--
 .../java/ca/teamdman/sfml/ast/ASTBuilder.java | 92 +++++++------------
 2 files changed, 41 insertions(+), 69 deletions(-)

diff --git a/src/main/antlr/SFML.g4 b/src/main/antlr/SFML.g4
index af741d452..984f570ed 100644
--- a/src/main/antlr/SFML.g4
+++ b/src/main/antlr/SFML.g4
@@ -25,18 +25,18 @@ interval: TICK             #Tick
 //
 
 block           : statement* ;
-statement       : inputstatement    #InputStatementStatement
-                | outputstatement   #OutputStatementStatement
-                | ifstatement       #IfStatementStatement
-                | forgetstatement   #ForgetStatementStatement
+statement       : inputStatement
+                | outputStatement
+                | ifStatement
+                | forgetStatement
                 ;
 
 // IO STATEMENT
-forgetstatement : FORGET label? (COMMA label)* COMMA?;
-inputstatement  : INPUT inputmatchers? resourceexclusion? FROM EACH? labelaccess
+forgetStatement : FORGET label? (COMMA label)* COMMA?;
+inputStatement  : INPUT inputmatchers? resourceexclusion? FROM EACH? labelaccess
                 | FROM EACH? labelaccess INPUT inputmatchers? resourceexclusion?
                 ;
-outputstatement : OUTPUT outputmatchers? resourceexclusion? TO EACH? labelaccess
+outputStatement : OUTPUT outputmatchers? resourceexclusion? TO EACH? labelaccess
                 | TO EACH? labelaccess OUTPUT outputmatchers? resourceexclusion?
                 ;
 inputmatchers   : movement; // separate for different defaults
@@ -74,7 +74,7 @@ rangeset        : range (COMMA range)*;
 range           : number (DASH number)? ;
 
 
-ifstatement     : IF boolexpr THEN block (ELSE IF boolexpr THEN block)* (ELSE block)? END;
+ifStatement     : IF boolexpr THEN block (ELSE IF boolexpr THEN block)* (ELSE block)? END;
 boolexpr        : TRUE                                  #BooleanTrue
                 | FALSE                                 #BooleanFalse
                 | LPAREN boolexpr RPAREN                #BooleanParen
@@ -204,7 +204,7 @@ DO              : D O ;
 WORLD           : W O R L D ;
 PROGRAM         : P R O G R A M ;
 END             : E N D ;
-NAME            : N A M E ;
+NAME            : N A M E;
 
 // GENERAL SYMBOLS
 // used by triggers and as a set operator
diff --git a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
index 677d0b05c..6f9710369 100644
--- a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
+++ b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
@@ -23,7 +23,7 @@ public List<Pair<ASTNode, ParserRuleContext>> getNodesUnderCursor(int cursorPos)
                 .stream()
                 .filter(pair -> pair.getSecond() != null)
                 .filter(pair -> pair.getSecond().start.getStartIndex() <= cursorPos
-                                && pair.getSecond().stop.getStopIndex() >= cursorPos)
+                        && pair.getSecond().stop.getStopIndex() >= cursorPos)
                 .collect(Collectors.toList());
     }
 
@@ -106,8 +106,8 @@ public Label visitRawLabel(SFMLParser.RawLabelContext ctx) {
         var label = new Label(ctx.getText());
         if (label.name().length() > Program.MAX_LABEL_LENGTH) {
             throw new IllegalArgumentException("Label name cannot be longer than "
-                                               + Program.MAX_LABEL_LENGTH
-                                               + " characters.");
+                    + Program.MAX_LABEL_LENGTH
+                    + " characters.");
         }
         USED_LABELS.add(label);
         AST_NODE_CONTEXTS.add(new Pair<>(label, ctx));
@@ -119,8 +119,8 @@ public Label visitStringLabel(SFMLParser.StringLabelContext ctx) {
         var label = new Label(visitString(ctx.string()).value());
         if (label.name().length() > Program.MAX_LABEL_LENGTH) {
             throw new IllegalArgumentException("Label name cannot be longer than "
-                                               + Program.MAX_LABEL_LENGTH
-                                               + " characters.");
+                    + Program.MAX_LABEL_LENGTH
+                    + " characters.");
         }
         USED_LABELS.add(label);
         AST_NODE_CONTEXTS.add(new Pair<>(label, ctx));
@@ -154,8 +154,8 @@ public ASTNode visitTimerTrigger(SFMLParser.TimerTriggerContext ctx) {
 
         // get default min interval
         int minInterval = timerTrigger.usesOnlyForgeEnergyResourceIO()
-                          ? SFMConfig.getOrDefault(SFMConfig.COMMON.timerTriggerMinimumIntervalInTicksWhenOnlyForgeEnergyIO)
-                          : SFMConfig.getOrDefault(SFMConfig.COMMON.timerTriggerMinimumIntervalInTicks);
+                ? SFMConfig.getOrDefault(SFMConfig.COMMON.timerTriggerMinimumIntervalInTicksWhenOnlyForgeEnergyIO)
+                : SFMConfig.getOrDefault(SFMConfig.COMMON.timerTriggerMinimumIntervalInTicks);
 
         // validate interval
         if (time.getTicks() < minInterval) {
@@ -188,8 +188,8 @@ public ASTNode visitBooleanRedstone(SFMLParser.BooleanRedstoneContext ctx) {
                                 .getManager()
                                 .getLevel()
                                 .getBestNeighborSignal(programContext
-                                                               .getManager()
-                                                               .getBlockPos()),
+                                        .getManager()
+                                        .getBlockPos()),
                         (long) finalNum
                 ),
                 ctx.getText()
@@ -243,21 +243,7 @@ public Interval visitSeconds(SFMLParser.SecondsContext ctx) {
     }
 
     @Override
-    public InputStatement visitInputStatementStatement(SFMLParser.InputStatementStatementContext ctx) {
-        InputStatement input = (InputStatement) visit(ctx.inputstatement());
-        AST_NODE_CONTEXTS.add(new Pair<>(input, ctx));
-        return input;
-    }
-
-    @Override
-    public OutputStatement visitOutputStatementStatement(SFMLParser.OutputStatementStatementContext ctx) {
-        OutputStatement output = (OutputStatement) visit(ctx.outputstatement());
-        AST_NODE_CONTEXTS.add(new Pair<>(output, ctx));
-        return output;
-    }
-
-    @Override
-    public InputStatement visitInputstatement(SFMLParser.InputstatementContext ctx) {
+    public InputStatement visitInputStatement(SFMLParser.InputStatementContext ctx) {
         var labelAccess = visitLabelaccess(ctx.labelaccess());
         var matchers = visitInputmatchers(ctx.inputmatchers());
         var exclusions = visitResourceexclusion(ctx.resourceexclusion());
@@ -268,7 +254,7 @@ public InputStatement visitInputstatement(SFMLParser.InputstatementContext ctx)
     }
 
     @Override
-    public OutputStatement visitOutputstatement(SFMLParser.OutputstatementContext ctx) {
+    public OutputStatement visitOutputStatement(SFMLParser.OutputStatementContext ctx) {
         var labelAccess = visitLabelaccess(ctx.labelaccess());
         var matchers = visitOutputmatchers(ctx.outputmatchers());
         var exclusions = visitResourceexclusion(ctx.resourceexclusion());
@@ -301,12 +287,12 @@ public LabelAccess visitLabelaccess(SFMLParser.LabelaccessContext ctx) {
     public RoundRobin visitRoundrobin(@Nullable SFMLParser.RoundrobinContext ctx) {
         if (ctx == null) return RoundRobin.disabled();
         return ctx.BLOCK() != null
-               ? new RoundRobin(RoundRobin.Behaviour.BY_BLOCK)
-               : new RoundRobin(RoundRobin.Behaviour.BY_LABEL);
+                ? new RoundRobin(RoundRobin.Behaviour.BY_BLOCK)
+                : new RoundRobin(RoundRobin.Behaviour.BY_LABEL);
     }
 
     @Override
-    public IfStatement visitIfstatement(SFMLParser.IfstatementContext ctx) {
+    public IfStatement visitIfStatement(SFMLParser.IfStatementContext ctx) {
         var conditions = ctx
                 .boolexpr()
                 .stream()
@@ -348,13 +334,6 @@ public IfStatement visitIfstatement(SFMLParser.IfstatementContext ctx) {
         return nestedStatement;
     }
 
-    @Override
-    public IfStatement visitIfStatementStatement(SFMLParser.IfStatementStatementContext ctx) {
-        IfStatement ifStatement = visitIfstatement(ctx.ifstatement());
-        AST_NODE_CONTEXTS.add(new Pair<>(ifStatement, ctx));
-        return ifStatement;
-    }
-
     @Override
     public BoolExpr visitBooleanTrue(SFMLParser.BooleanTrueContext ctx) {
         BoolExpr boolExpr = new BoolExpr(__ -> true, "TRUE");
@@ -462,11 +441,11 @@ public Limit visitQuantityRetentionLimit(SFMLParser.QuantityRetentionLimitContex
     public ResourceIdSet visitResourceexclusion(@Nullable SFMLParser.ResourceexclusionContext ctx) {
         if (ctx == null) return ResourceIdSet.EMPTY;
         ResourceIdSet resourceIdSet = new ResourceIdSet(ctx
-                                                                .resourceid()
-                                                                .stream()
-                                                                .map(this::visit)
-                                                                .map(ResourceIdentifier.class::cast)
-                                                                .collect(HashSet::new, HashSet::add, HashSet::addAll));
+                .resourceid()
+                .stream()
+                .map(this::visit)
+                .map(ResourceIdentifier.class::cast)
+                .collect(HashSet::new, HashSet::add, HashSet::addAll));
         AST_NODE_CONTEXTS.add(new Pair<>(resourceIdSet, ctx));
         return resourceIdSet;
     }
@@ -539,14 +518,7 @@ public NumberRangeSet visitSlotqualifier(@Nullable SFMLParser.SlotqualifierConte
     }
 
     @Override
-    public ASTNode visitForgetStatementStatement(SFMLParser.ForgetStatementStatementContext ctx) {
-        ForgetStatement statement = (ForgetStatement) visit(ctx.forgetstatement());
-        AST_NODE_CONTEXTS.add(new Pair<>(statement, ctx));
-        return statement;
-    }
-
-    @Override
-    public ForgetStatement visitForgetstatement(SFMLParser.ForgetstatementContext ctx) {
+    public ForgetStatement visitForgetStatement(SFMLParser.ForgetStatementContext ctx) {
         Set<Label> labels = ctx
                 .label()
                 .stream()
@@ -563,10 +535,10 @@ public ForgetStatement visitForgetstatement(SFMLParser.ForgetstatementContext ct
     public NumberRangeSet visitRangeset(@Nullable SFMLParser.RangesetContext ctx) {
         if (ctx == null) return NumberRangeSet.MAX_RANGE;
         NumberRangeSet numberRangeSet = new NumberRangeSet(ctx
-                                                                   .range()
-                                                                   .stream()
-                                                                   .map(this::visitRange)
-                                                                   .toArray(NumberRange[]::new));
+                .range()
+                .stream()
+                .map(this::visitRange)
+                .toArray(NumberRange[]::new));
         AST_NODE_CONTEXTS.add(new Pair<>(numberRangeSet, ctx));
         return numberRangeSet;
     }
@@ -610,8 +582,8 @@ public ResourceQuantity visitRetention(@Nullable SFMLParser.RetentionContext ctx
         ResourceQuantity quantity = new ResourceQuantity(
                 visitNumber(ctx.number()),
                 ctx.EACH() != null
-                ? ResourceQuantity.IdExpansionBehaviour.EXPAND
-                : ResourceQuantity.IdExpansionBehaviour.NO_EXPAND
+                        ? ResourceQuantity.IdExpansionBehaviour.EXPAND
+                        : ResourceQuantity.IdExpansionBehaviour.NO_EXPAND
         );
         AST_NODE_CONTEXTS.add(new Pair<>(quantity, ctx));
         return quantity;
@@ -623,8 +595,8 @@ public ResourceQuantity visitQuantity(@Nullable SFMLParser.QuantityContext ctx)
         ResourceQuantity quantity = new ResourceQuantity(
                 visitNumber(ctx.number()),
                 ctx.EACH() != null
-                ? ResourceQuantity.IdExpansionBehaviour.EXPAND
-                : ResourceQuantity.IdExpansionBehaviour.NO_EXPAND
+                        ? ResourceQuantity.IdExpansionBehaviour.EXPAND
+                        : ResourceQuantity.IdExpansionBehaviour.NO_EXPAND
         );
         AST_NODE_CONTEXTS.add(new Pair<>(quantity, ctx));
         return quantity;
@@ -641,9 +613,9 @@ public DirectionQualifier visitEachSide(SFMLParser.EachSideContext ctx) {
     public DirectionQualifier visitListedSides(SFMLParser.ListedSidesContext ctx) {
         DirectionQualifier directionQualifier = new DirectionQualifier(
                 EnumSet.copyOf(ctx.side().stream()
-                                       .map(this::visitSide)
-                                       .map(DirectionQualifier::lookup)
-                                       .toList())
+                        .map(this::visitSide)
+                        .map(DirectionQualifier::lookup)
+                        .toList())
         );
         AST_NODE_CONTEXTS.add(new Pair<>(directionQualifier, ctx));
         return directionQualifier;
@@ -678,7 +650,7 @@ private void assertResourceLimitDoesntExpandHuge(ResourceLimits limits) {
         if (limits.createInputTrackers().size() > 512) {
             throw new IllegalArgumentException(
                     "Resource limit expands to more than 512 trackers, this is likely a mistake where the \"EACH\" keyword is being used. The code: "
-                    + limits);
+                            + limits);
         }
     }
 }

From 38952203609bd382536d6d9974e3edc90d2f6b5c Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 16:02:10 -0400
Subject: [PATCH 2/7] Implement WHERE statements

---
 src/main/antlr/SFML.g4                        |  12 ++-
 .../ProgramSyntaxHighlightingHelper.java      |   1 +
 ...ntainerExportsInspectionRequestPacket.java |   3 +-
 .../sfm/common/resourcetype/ResourceType.java |  40 +++++++
 .../ca/teamdman/sfm/common/util/SFMUtils.java |   6 +-
 .../java/ca/teamdman/sfml/ast/ASTBuilder.java |  53 ++++++++-
 .../ca/teamdman/sfml/ast/ForgetStatement.java |   3 +-
 .../ca/teamdman/sfml/ast/InputStatement.java  |  93 ++++++++--------
 .../ca/teamdman/sfml/ast/LabelAccess.java     |   4 +-
 .../ca/teamdman/sfml/ast/OutputStatement.java | 102 +++++++++---------
 .../teamdman/sfml/ast/ResourceComparer.java   |  17 +++
 .../ca/teamdman/sfml/ast/WhereStatement.java  |  26 +++++
 12 files changed, 257 insertions(+), 103 deletions(-)
 create mode 100644 src/main/java/ca/teamdman/sfml/ast/WhereStatement.java

diff --git a/src/main/antlr/SFML.g4 b/src/main/antlr/SFML.g4
index 984f570ed..b560814c4 100644
--- a/src/main/antlr/SFML.g4
+++ b/src/main/antlr/SFML.g4
@@ -112,12 +112,20 @@ setOp           : OVERALL
 //
 // IO HELPERS
 //
-labelaccess     : label (COMMA label)* roundrobin? sidequalifier? slotqualifier?;
-roundrobin: ROUND ROBIN BY (LABEL | BLOCK);
+labelaccess     : label (COMMA label)* roundrobin? sidequalifier? slotqualifier? (WHERE where)?;
+roundrobin      : ROUND ROBIN BY (LABEL | BLOCK);
 label           : (IDENTIFIER|REDSTONE)   #RawLabel
                 | string                  #StringLabel
                 ;
 
+where           : LPAREN where RPAREN             # WhereParen
+                | NOT where                       # WhereNegation
+                | where AND where                 # WhereConjunction
+                | where OR where                  # WhereDisjunction
+                | resourcecomparison              # WhereComparison
+                ;
+
+
 resourceid      : (IDENTIFIER|REDSTONE) (COLON (IDENTIFIER|REDSTONE)? (COLON (IDENTIFIER|REDSTONE)? (COLON (IDENTIFIER|REDSTONE)?)?)?)? # Resource
                 | string                             # StringResource
                 ;
diff --git a/src/main/java/ca/teamdman/sfm/client/ProgramSyntaxHighlightingHelper.java b/src/main/java/ca/teamdman/sfm/client/ProgramSyntaxHighlightingHelper.java
index 7ce84495e..01c81c625 100644
--- a/src/main/java/ca/teamdman/sfm/client/ProgramSyntaxHighlightingHelper.java
+++ b/src/main/java/ca/teamdman/sfm/client/ProgramSyntaxHighlightingHelper.java
@@ -100,6 +100,7 @@ private static ChatFormatting getColour(Token token) {
             case SFMLLexer.TRUE:
             case SFMLLexer.FALSE:
             case SFMLLexer.FORGET:
+            case SFMLLexer.WHERE:
                 return ChatFormatting.BLUE;
             case SFMLLexer.IDENTIFIER:
             case SFMLLexer.STRING:
diff --git a/src/main/java/ca/teamdman/sfm/common/net/ServerboundContainerExportsInspectionRequestPacket.java b/src/main/java/ca/teamdman/sfm/common/net/ServerboundContainerExportsInspectionRequestPacket.java
index d499f02fd..99de6db48 100644
--- a/src/main/java/ca/teamdman/sfm/common/net/ServerboundContainerExportsInspectionRequestPacket.java
+++ b/src/main/java/ca/teamdman/sfm/common/net/ServerboundContainerExportsInspectionRequestPacket.java
@@ -167,7 +167,8 @@ public static <STACK, ITEM, CAP> String buildInspectionResults(
                                                                ? EnumSet.noneOf(Direction.class)
                                                                : EnumSet.of(direction)),
                                         NumberRangeSet.MAX_RANGE,
-                                        RoundRobin.disabled()
+                                        RoundRobin.disabled(),
+                                        WhereStatement.ALWAYS_TRUE
                                 ),
                                 new ResourceLimits(
                                         resourceLimitList.stream().distinct().toList(),
diff --git a/src/main/java/ca/teamdman/sfm/common/resourcetype/ResourceType.java b/src/main/java/ca/teamdman/sfm/common/resourcetype/ResourceType.java
index c9e846c91..22c38622c 100644
--- a/src/main/java/ca/teamdman/sfm/common/resourcetype/ResourceType.java
+++ b/src/main/java/ca/teamdman/sfm/common/resourcetype/ResourceType.java
@@ -154,6 +154,46 @@ public void forEachCapability(
         }
     }
 
+    public void forCapabilityOfBlock(
+            ProgramContext programContext,
+            DirectionQualifier directionQualifier,
+            Pair<Label, BlockPos> labelPosPair,
+            CapabilityConsumer<CAP> consumer
+    ) {
+        CableNetwork network = programContext.getNetwork();
+
+        var label = labelPosPair.getFirst();
+        var blockPos =  labelPosPair.getSecond();
+
+        for (Direction dir : (Iterable<? extends Direction>) directionQualifier.stream()::iterator) {
+            // Get capability from the network
+            Optional<CAP> maybeCap = network
+                    .getCapability(CAPABILITY_KIND, blockPos, dir, programContext.getLogger())
+                    .resolve();
+            if (maybeCap.isPresent()) {
+                programContext
+                        .getLogger()
+                        .debug(x -> x.accept(Constants.LocalizationKeys.LOG_RESOURCE_TYPE_GET_CAPABILITIES_CAP_PRESENT.get(
+                                displayAsCapabilityClass(),
+                                blockPos,
+                                dir
+                        )));
+                CAP cap = maybeCap.get();
+                consumer.accept(label, blockPos, dir, cap);
+
+            } else {
+                // Log error
+                programContext
+                        .getLogger()
+                        .error(x -> x.accept(Constants.LocalizationKeys.LOG_RESOURCE_TYPE_GET_CAPABILITIES_CAP_NOT_PRESENT.get(
+                                displayAsCapabilityClass(),
+                                blockPos,
+                                dir
+                        )));
+            }
+        }
+    }
+
     public Stream<STACK> getStacksInSlots(
             CAP cap,
             NumberRangeSet slots
diff --git a/src/main/java/ca/teamdman/sfm/common/util/SFMUtils.java b/src/main/java/ca/teamdman/sfm/common/util/SFMUtils.java
index 490570ed9..1e5edcd4f 100644
--- a/src/main/java/ca/teamdman/sfm/common/util/SFMUtils.java
+++ b/src/main/java/ca/teamdman/sfm/common/util/SFMUtils.java
@@ -155,7 +155,8 @@ public static <STACK, ITEM, CAP> Optional<InputStatement> getInputStatementForSl
                         labelAccess.directions(),
                         inputStatement.labelAccess()
                                 .slots(),
-                        RoundRobin.disabled()
+                        RoundRobin.disabled(),
+                        labelAccess.where()
                 ), inputStatement.resourceLimits(), inputStatement.each()));
     }
 
@@ -178,7 +179,8 @@ public static <STACK, ITEM, CAP> InputStatement getInputStatementForStack(
                 new NumberRangeSet(
                         new NumberRange[]{new NumberRange(slot, slot)}
                 ),
-                RoundRobin.disabled()
+                RoundRobin.disabled(),
+                WhereStatement.ALWAYS_TRUE
         );
         Limit limit = new Limit(
                 new ResourceQuantity(
diff --git a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
index 6f9710369..88fdd0aa0 100644
--- a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
+++ b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
@@ -273,11 +273,21 @@ public LabelAccess visitLabelaccess(SFMLParser.LabelaccessContext ctx) {
         } else {
             directionQualifier = (DirectionQualifier) visit(directionQualifierCtx);
         }
+
+        var whereCtx = ctx.where();
+        WhereStatement where;
+        if (whereCtx == null) {
+            where = WhereStatement.ALWAYS_TRUE;
+        } else {
+            where = (WhereStatement) visit(whereCtx);
+        }
+
         LabelAccess labelAccess = new LabelAccess(
                 ctx.label().stream().map(this::visit).map(Label.class::cast).collect(Collectors.toList()),
                 directionQualifier,
                 visitSlotqualifier(ctx.slotqualifier()),
-                visitRoundrobin(ctx.roundrobin())
+                visitRoundrobin(ctx.roundrobin()),
+                where
         );
         AST_NODE_CONTEXTS.add(new Pair<>(labelAccess, ctx));
         return labelAccess;
@@ -628,6 +638,47 @@ public Side visitSide(SFMLParser.SideContext ctx) {
         return side;
     }
 
+    @Override
+    public WhereStatement visitWhereConjunction(SFMLParser.WhereConjunctionContext ctx) {
+        var left = (WhereStatement) visit(ctx.where(0));
+        var right = (WhereStatement) visit(ctx.where(1));
+        WhereStatement where = new WhereStatement(left.and(right), left.sourceCode() + " AND " + right.sourceCode());
+        AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
+        return where;
+    }
+
+    @Override
+    public WhereStatement visitWhereParen(SFMLParser.WhereParenContext ctx) {
+        WhereStatement where = (WhereStatement) visit(ctx.where());
+        AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
+        return where;
+    }
+
+    @Override
+    public WhereStatement visitWhereDisjunction(SFMLParser.WhereDisjunctionContext ctx) {
+        var left = (WhereStatement) visit(ctx.where(0));
+        var right = (WhereStatement) visit(ctx.where(1));
+        WhereStatement where = new WhereStatement(left.or(right), left.sourceCode() + " OR " + right.sourceCode());
+        AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
+        return where;
+    }
+
+    @Override
+    public WhereStatement visitWhereNegation(SFMLParser.WhereNegationContext ctx) {
+        var x = (WhereStatement) visit(ctx.where());
+        WhereStatement where = new WhereStatement(x.negate(), "NOT " + x.sourceCode());
+        AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
+        return where;
+    }
+
+    @Override
+    public WhereStatement visitWhereComparison(SFMLParser.WhereComparisonContext ctx) {
+        var resourceComparison = visitResourcecomparison(ctx.resourcecomparison());
+        WhereStatement where = resourceComparison.toWhereExpression("WHERE " + resourceComparison.toString());
+        AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
+        return where;
+    }
+
     @Override
     public Block visitBlock(@Nullable SFMLParser.BlockContext ctx) {
         if (ctx == null) return new Block(Collections.emptyList());
diff --git a/src/main/java/ca/teamdman/sfml/ast/ForgetStatement.java b/src/main/java/ca/teamdman/sfml/ast/ForgetStatement.java
index 656ea242b..835eedacc 100644
--- a/src/main/java/ca/teamdman/sfml/ast/ForgetStatement.java
+++ b/src/main/java/ca/teamdman/sfml/ast/ForgetStatement.java
@@ -29,7 +29,8 @@ public void tick(ProgramContext context) {
                             newLabels,
                             oldInputStatement.labelAccess().directions(),
                             oldInputStatement.labelAccess().slots(),
-                            oldInputStatement.labelAccess().roundRobin()
+                            oldInputStatement.labelAccess().roundRobin(),
+                            oldInputStatement.labelAccess().where()
                     ),
                     oldInputStatement.resourceLimits(),
                     oldInputStatement.each()
diff --git a/src/main/java/ca/teamdman/sfml/ast/InputStatement.java b/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
index 62842f2c3..f51b93d97 100644
--- a/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
+++ b/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
@@ -1,8 +1,8 @@
 package ca.teamdman.sfml.ast;
 
-import ca.teamdman.sfm.common.Constants;
 import ca.teamdman.sfm.common.program.*;
 import ca.teamdman.sfm.common.resourcetype.ResourceType;
+import com.mojang.datafixers.util.Pair;
 import net.minecraft.core.BlockPos;
 import net.minecraft.core.Direction;
 import org.jetbrains.annotations.Nullable;
@@ -80,56 +80,41 @@ public void gatherSlots(
                 .getReferencedResourceTypes()
                 .collect(Collectors.toSet());
 
+        Consumer<LimitedInputSlot<?, ?, ?>> finalSlotConsumer = slotConsumer;
+
+        List<InputResourceTracker<?, ?, ?>> inputTrackers = null;
         if (!each) {
-            // log not each
             context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_NOT_EACH.get()));
+            inputTrackers = resourceLimits.createInputTrackers();
+        } else {
+            context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_EACH.get()));
+        }
 
-            // create a single matcher to be shared by all capabilities
-            List<InputResourceTracker<?, ?, ?>> inputTrackers = resourceLimits.createInputTrackers();
+        for (var labelPosPair : getBlocksFromLabels(context)) {
+            // Tracker for block
+            if (each) {
+                inputTrackers = resourceLimits.createInputTrackers();
+            }
+            List<LimitedInputSlot<?, ?, ?>> limitedInputSlotList = new ArrayList<>();
+            HashMap<Object, Long> resourceTable = new HashMap<>();
+            // Loop over capability
             for (var type : referencedResourceTypes) {
-                // log gather for resource type
-                context
-                        .getLogger()
-                        .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_FOR_RESOURCE_TYPE.get(
-                                type.displayAsCapabilityClass(),
-                                type.displayAsCapabilityClass()
-                        )));
-
-                // gather slots for each capability found for positions tagged by a provided label
-                Consumer<LimitedInputSlot<?, ?, ?>> finalSlotConsumer = slotConsumer;
-                type.forEachCapability(context, labelAccess, (label, pos, direction, cap) -> gatherSlotsForCap(
+                // Get slots to loop over
+                List<InputResourceTracker<?, ?, ?>> finalInputTrackers = inputTrackers;
+                type.forCapabilityOfBlock(context, labelAccess.directions(), labelPosPair, (label, pos, direction, cap) -> gatherSlotsForCap(
                         context,
                         (ResourceType<Object, Object, Object>) type,
                         label, pos, direction, cap,
-                        inputTrackers,
-                        finalSlotConsumer
+                        finalInputTrackers,
+                        limitedInputSlotList,
+                        resourceTable
                 ));
             }
-        } else {
-            // log yes each
-            context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_EACH.get()));
 
-            for (ResourceType type : referencedResourceTypes) {
-                // log gather for resource type
-                context
-                        .getLogger()
-                        .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_FOR_RESOURCE_TYPE.get(
-                                type.displayAsCapabilityClass(),
-                                type.displayAsCapabilityClass()
-                        )));
-
-                // gather slots for each capability found for positions tagged by a provided label
-                Consumer<LimitedInputSlot<?, ?, ?>> finalSlotConsumer = slotConsumer;
-                type.forEachCapability(context, labelAccess, (label, pos, direction, cap) -> {
-                    List<InputResourceTracker<?, ?, ?>> inputTrackers = resourceLimits.createInputTrackers();
-                    gatherSlotsForCap(
-                            context,
-                            (ResourceType<Object, Object, Object>) type,
-                            label, pos, direction, cap,
-                            inputTrackers,
-                            finalSlotConsumer
-                    );
-                });
+            if (labelAccess().where().test(context, resourceTable)) {
+                for (var slotList : limitedInputSlotList) {
+                    finalSlotConsumer.accept(slotList);
+                }
             }
         }
     }
@@ -205,7 +190,7 @@ public void freeSlots() {
         }
     }
 
-    public void freeSlotsIf(Predicate<LimitedInputSlot<?,?,?>> condition) {
+    public void freeSlotsIf(Predicate<LimitedInputSlot<?, ?, ?>> condition) {
         if (limitedInputSlotsCache != null) {
             Iterator<LimitedInputSlot<?, ?, ?>> iterator = limitedInputSlotsCache.iterator();
             while (iterator.hasNext()) {
@@ -239,28 +224,33 @@ private <STACK, ITEM, CAP> void gatherSlotsForCap(
             Direction direction,
             CAP capability,
             List<InputResourceTracker<?, ?, ?>> trackers,
-            Consumer<LimitedInputSlot<?, ?, ?>> acceptor
+            List<LimitedInputSlot<?, ?, ?>> limitedInputSlotList,
+            HashMap<STACK, Long> resourceTable
     ) {
         context
                 .getLogger()
-                .debug(x -> x.accept(Constants.LocalizationKeys.LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_RANGE.get(
+                .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_RANGE.get(
                         labelAccess.slots())));
         for (int slot = 0; slot < type.getSlots(capability); slot++) {
             int finalSlot = slot;
             if (labelAccess.slots().contains(slot)) {
                 STACK stack = type.getStackInSlot(capability, slot);
                 if (shouldCreateSlot(type, stack)) {
+                    if (type.matchesCapabilityType(capability)) {
+                        // Add items to resourceTable, this is done before checking if it's a resource we want to move
+                        resourceTable.put(stack, resourceTable.getOrDefault(stack, 0L) + type.getAmount(stack));
+                    }
                     for (InputResourceTracker<?, ?, ?> tracker : trackers) {
                         if (tracker.matchesCapabilityType(capability) && tracker.test(stack)) {
                             context
                                     .getLogger()
-                                    .debug(x -> x.accept(Constants.LocalizationKeys.LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_SLOT_CREATED.get(
+                                    .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_SLOT_CREATED.get(
                                             finalSlot,
                                             stack,
                                             tracker.toString()
                                     )));
                             //noinspection unchecked
-                            acceptor.accept(LimitedInputSlotObjectPool.acquire(
+                            limitedInputSlotList.add(LimitedInputSlotObjectPool.acquire(
                                     label, pos, direction, slot, capability,
                                     (InputResourceTracker<STACK, ITEM, CAP>) tracker,
                                     stack
@@ -291,4 +281,15 @@ private <STACK, ITEM, CAP> boolean shouldCreateSlot(
         return !type.isEmpty(stack);
     }
 
+    private List<Pair<Label, BlockPos>> getBlocksFromLabels(ProgramContext context) {
+        LabelPositionHolder labelPositionHolder = context.getLabelPositionHolder();
+        ArrayList<Pair<Label, BlockPos>> blocks = new ArrayList<>();
+        for (Label label : labelAccess.labels()) {
+            for (BlockPos pos : labelPositionHolder.getPositions(label.name())) {
+                blocks.add(Pair.of(label, pos));
+            }
+        }
+
+        return blocks;
+    }
 }
diff --git a/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java b/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
index 1791b383c..dd534b373 100644
--- a/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
+++ b/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
@@ -8,7 +8,8 @@ public record LabelAccess(
         List<Label> labels,
         DirectionQualifier directions,
         NumberRangeSet slots,
-        RoundRobin roundRobin
+        RoundRobin roundRobin,
+        WhereStatement where
 ) implements ASTNode {
     @Override
     public String toString() {
@@ -34,6 +35,7 @@ public String toString() {
                 }
             }
         }
+        builder.append(" ").append(where);
         return builder.toString();
     }
 }
diff --git a/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java b/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
index 97beb79aa..83fc6b6d6 100644
--- a/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
+++ b/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
@@ -5,15 +5,13 @@
 import ca.teamdman.sfm.common.program.*;
 import ca.teamdman.sfm.common.registry.SFMResourceTypes;
 import ca.teamdman.sfm.common.resourcetype.ResourceType;
+import com.mojang.datafixers.util.Pair;
 import net.minecraft.core.BlockPos;
 import net.minecraft.core.Direction;
 
-import java.util.ArrayDeque;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
 import java.util.function.Consumer;
 import java.util.stream.Collectors;
-import java.util.stream.Stream;
 
 import static ca.teamdman.sfm.common.Constants.LocalizationKeys.*;
 
@@ -332,54 +330,43 @@ public void gatherSlots(
     ) {
         context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS.get(toStringPretty())));
 
-        Stream<ResourceType> types = resourceLimits.getReferencedResourceTypes();
+        Set<ResourceType> referencedResourceTypes = resourceLimits
+                .getReferencedResourceTypes()
+                .collect(Collectors.toSet());
 
+        List<OutputResourceTracker<?, ?, ?>> outputTracker = null;
         if (!each) {
             context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_NOT_EACH.get()));
-            // create a single matcher to be shared by all capabilities
-            List<OutputResourceTracker<?, ?, ?>> outputTracker = resourceLimits.createOutputTrackers();
-            for (var type : (Iterable<ResourceType>) types::iterator) {
-                context
-                        .getLogger()
-                        .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_FOR_RESOURCE_TYPE.get(
-                                type.displayAsCapabilityClass(),
-                                type.displayAsCapabilityClass()
-                        )));
-                type.forEachCapability(context, labelAccess, (
-                        (label, pos, direction, cap) -> gatherSlotsForCap(
-                                context,
-                                (ResourceType<Object, Object, Object>) type,
-                                label,
-                                pos,
-                                direction,
-                                cap,
-                                outputTracker,
-                                slotConsumer
-                        )
-                ));
-            }
+            outputTracker = resourceLimits.createOutputTrackers();
         } else {
             context.getLogger().debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_EACH.get()));
-            for (var type : (Iterable<ResourceType>) types::iterator) {
-                context
-                        .getLogger()
-                        .debug(x -> x.accept(LOG_PROGRAM_TICK_IO_STATEMENT_GATHER_SLOTS_FOR_RESOURCE_TYPE.get(
-                                type.displayAsCapabilityClass(),
-                                type.displayAsCapabilityClass()
-                        )));
-                type.forEachCapability(context, labelAccess, (label, pos, direction, cap) -> {
-                    List<OutputResourceTracker<?, ?, ?>> outputTracker = resourceLimits.createOutputTrackers();
-                    gatherSlotsForCap(
-                            context,
-                            (ResourceType<Object, Object, Object>) type,
-                            label,
-                            pos,
-                            direction,
-                            cap,
-                            outputTracker,
-                            slotConsumer
-                    );
-                });
+        }
+
+        for (var labelPosPair : getBlocksFromLabels(context)) {
+            // Tracker for block
+            if (each) {
+                outputTracker = resourceLimits.createOutputTrackers();
+            }
+            List<LimitedOutputSlot<?, ?, ?>> limitedOutputSlotList = new ArrayList<>();
+            HashMap<Object, Long> resourceTable = new HashMap<>();
+            // Loop over capability
+            for (var type : referencedResourceTypes) {
+                // Get slots to loop over
+                List<OutputResourceTracker<?, ?, ?>> finalOutputTracker = outputTracker;
+                type.forCapabilityOfBlock(context, labelAccess.directions(), labelPosPair, (label, pos, direction, cap) -> gatherSlotsForCap(
+                        context,
+                        (ResourceType<Object, Object, Object>) type,
+                        label, pos, direction, cap,
+                        finalOutputTracker,
+                        limitedOutputSlotList,
+                        resourceTable
+                ));
+            }
+
+            if (labelAccess().where().test(context, resourceTable)) {
+                for (var slotList : limitedOutputSlotList) {
+                    slotConsumer.accept(slotList);
+                }
             }
         }
     }
@@ -447,7 +434,8 @@ private <STACK, ITEM, CAP> void gatherSlotsForCap(
             Direction direction,
             CAP capability,
             List<OutputResourceTracker<?, ?, ?>> trackers,
-            Consumer<LimitedOutputSlot<?, ?, ?>> acceptor
+            List<LimitedOutputSlot<?, ?, ?>> limitedOutputSlotList,
+            HashMap<Object, Long> resourceTable
     ) {
         context
                 .getLogger()
@@ -457,6 +445,10 @@ private <STACK, ITEM, CAP> void gatherSlotsForCap(
             if (labelAccess.slots().contains(slot)) {
                 STACK stack = type.getStackInSlot(capability, slot);
                 boolean shouldCreateSlot = shouldCreateSlot(type, capability, stack, slot);
+                if (type.matchesCapabilityType(capability)) {
+                    // Add items to resourceTable
+                    resourceTable.put(stack, resourceTable.getOrDefault(stack, 0L) + type.getAmount(stack));
+                }
                 //noinspection rawtypes
                 for (OutputResourceTracker tracker : trackers) {
                     // we don't also test the tracker because we can deposit into empty slots
@@ -474,7 +466,7 @@ private <STACK, ITEM, CAP> void gatherSlotsForCap(
                                             tracker.toString()
                                     )));
                             //noinspection unchecked
-                            acceptor.accept(LimitedOutputSlotObjectPool.acquire(
+                            limitedOutputSlotList.add(LimitedOutputSlotObjectPool.acquire(
                                     label,
                                     pos,
                                     direction,
@@ -517,4 +509,16 @@ private <STACK, ITEM, CAP> boolean shouldCreateSlot(
         // those blocks hold many more items than normal in a single stack
         return type.getAmount(stack) < type.getMaxStackSize(cap, slot);
     }
+
+    private List<Pair<Label, BlockPos>> getBlocksFromLabels(ProgramContext context) {
+        LabelPositionHolder labelPositionHolder = context.getLabelPositionHolder();
+        ArrayList<Pair<Label, BlockPos>> blocks = new ArrayList<>();
+        for (Label label : labelAccess.labels()) {
+            for (BlockPos pos : labelPositionHolder.getPositions(label.name())) {
+                blocks.add(Pair.of(label, pos));
+            }
+        }
+
+        return blocks;
+    }
 }
diff --git a/src/main/java/ca/teamdman/sfml/ast/ResourceComparer.java b/src/main/java/ca/teamdman/sfml/ast/ResourceComparer.java
index 118ad1e3c..1b3b83157 100644
--- a/src/main/java/ca/teamdman/sfml/ast/ResourceComparer.java
+++ b/src/main/java/ca/teamdman/sfml/ast/ResourceComparer.java
@@ -39,6 +39,23 @@ public BoolExpr toBooleanExpression(SetOperator setOp, LabelAccess labelAccess,
         );
     }
 
+    public WhereStatement toWhereExpression(String sourceCode) {
+        return new WhereStatement(
+                (context, resourceTable) -> {
+                    AtomicLong count = new AtomicLong(0);
+
+                    resourceTable.forEach((key, value) -> {
+                        if (res.test(key)) {
+                            count.addAndGet(value);
+                        }
+                    });
+
+                    return op.test(count.longValue(), quantity.number().value());
+                },
+                sourceCode
+        );
+    }
+
     @Override
     public String toString() {
         return op().getSourceCode() + " " + quantity() + " " + res().toStringCondensed();
diff --git a/src/main/java/ca/teamdman/sfml/ast/WhereStatement.java b/src/main/java/ca/teamdman/sfml/ast/WhereStatement.java
new file mode 100644
index 000000000..0149990f9
--- /dev/null
+++ b/src/main/java/ca/teamdman/sfml/ast/WhereStatement.java
@@ -0,0 +1,26 @@
+package ca.teamdman.sfml.ast;
+
+import ca.teamdman.sfm.common.program.ProgramContext;
+
+import java.util.HashMap;
+import java.util.function.BiPredicate;
+
+public record WhereStatement(
+        BiPredicate<ProgramContext, HashMap<Object, Long>> pred,
+        String sourceCode
+) implements BiPredicate<ProgramContext, HashMap<Object, Long>>, ASTNode {
+    public static final WhereStatement ALWAYS_TRUE = new WhereStatement((__, ___) -> true, "");
+
+    public boolean test(ProgramContext context, HashMap<Object, Long> resourceTable) {
+        return pred.test(context, resourceTable);
+    }
+
+    public WhereStatement negate() {
+        return new WhereStatement(pred.negate(), "NOT " + sourceCode);
+    }
+
+    @Override
+    public String toString() {
+        return sourceCode;
+    }
+}

From b7b3adba3725d971d23ab8055bf623bfd5f7726f Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 16:04:33 -0400
Subject: [PATCH 3/7] Fix tests

---
 src/test/java/ca/teamdman/sfm/SFMTests.java      | 3 ++-
 src/test/java/ca/teamdman/sfml/CodegenTests.java | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/test/java/ca/teamdman/sfm/SFMTests.java b/src/test/java/ca/teamdman/sfm/SFMTests.java
index a1b5d0b75..dd06e8e31 100644
--- a/src/test/java/ca/teamdman/sfm/SFMTests.java
+++ b/src/test/java/ca/teamdman/sfm/SFMTests.java
@@ -46,7 +46,8 @@ public void roundRobinByBlockDistinct() {
                 Stream.of("a", "b", "c").map(Label::new).toList(),
                 new DirectionQualifier(EnumSet.of(Direction.DOWN)),
                 NumberRangeSet.MAX_RANGE,
-                new RoundRobin(RoundRobin.Behaviour.BY_BLOCK)
+                new RoundRobin(RoundRobin.Behaviour.BY_BLOCK),
+                WhereStatement.ALWAYS_TRUE
         );
         LabelPositionHolder labelPositions = LabelPositionHolder.empty();
         labelPositions.add("a", new BlockPos(0, 0, 0));
diff --git a/src/test/java/ca/teamdman/sfml/CodegenTests.java b/src/test/java/ca/teamdman/sfml/CodegenTests.java
index a6d7c2818..26eec472c 100644
--- a/src/test/java/ca/teamdman/sfml/CodegenTests.java
+++ b/src/test/java/ca/teamdman/sfml/CodegenTests.java
@@ -75,7 +75,8 @@ public void codegen() {
                 List.of(new Label("a")),
                 DirectionQualifier.NULL_DIRECTION,
                 NumberRangeSet.MAX_RANGE,
-                RoundRobin.disabled()
+                RoundRobin.disabled(),
+                WhereStatement.ALWAYS_TRUE
         );
         var program = new Program(
                 new ASTBuilder(),

From aca6a491f900e7d7e46685553fcf93925b11dcd1 Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 20:01:50 -0400
Subject: [PATCH 4/7] Add WHERE tests for inputs and outputs

---
 .../teamdman/sfm/SFMCorrectnessGameTests.java |  95 +++++++++++++++++-
 .../resources/data/sfm/structures/3x3x1.nbt   | Bin 0 -> 167 bytes
 2 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 src/gametest/resources/data/sfm/structures/3x3x1.nbt

diff --git a/src/gametest/java/ca/teamdman/sfm/SFMCorrectnessGameTests.java b/src/gametest/java/ca/teamdman/sfm/SFMCorrectnessGameTests.java
index 7f8f9e219..155476d49 100644
--- a/src/gametest/java/ca/teamdman/sfm/SFMCorrectnessGameTests.java
+++ b/src/gametest/java/ca/teamdman/sfm/SFMCorrectnessGameTests.java
@@ -3436,4 +3436,97 @@ public static void conditional_output_inspection(GameTestHelper helper) {
             assertTrue(rightChest.getStackInSlot(0).getCount() == 32, "Dirt did not arrive");
         });
     }
-}
+
+    @GameTest(template = "3x3x1")
+    public static void input_where(GameTestHelper helper) {
+        // Only the right chest should move items
+        BlockPos managerPos = new BlockPos(1, 2, 0);
+        helper.setBlock(managerPos, SFMBlocks.MANAGER_BLOCK.get());
+        // Create positions
+        BlockPos leftPos = new BlockPos(2, 2, 0);
+        BlockPos rightPos = new BlockPos(0, 2, 0);
+        BlockPos abovePos = new BlockPos(1, 3, 0);
+        // Set blocks
+        helper.setBlock(leftPos, SFMBlocks.TEST_BARREL_BLOCK.get());
+        helper.setBlock(rightPos, SFMBlocks.TEST_BARREL_BLOCK.get());
+        helper.setBlock(abovePos, SFMBlocks.TEST_BARREL_BLOCK.get());
+
+        var leftChest = getItemHandler(helper, leftPos);
+        var rightChest = getItemHandler(helper, rightPos);
+        var topChest = getItemHandler(helper, abovePos);
+
+        leftChest.insertItem(0, new ItemStack(Blocks.DIRT, 1), false);
+
+        rightChest.insertItem(0, new ItemStack(Blocks.DIRT, 1), false);
+        rightChest.insertItem(1, new ItemStack(Blocks.STONE, 1), false);
+
+        ManagerBlockEntity manager = (ManagerBlockEntity) helper.getBlockEntity(managerPos);
+        manager.setItem(0, new ItemStack(SFMItems.DISK_ITEM.get()));
+        manager.setProgram("""
+                                   EVERY 20 TICKS DO
+                                       INPUT FROM a WHERE > 0 stone
+                                       OUTPUT dirt TO b
+                                   END
+                                   """.stripTrailing().stripIndent());
+        // set the labels
+        LabelPositionHolder.empty()
+                .add("a", helper.absolutePos(leftPos))
+                .add("a", helper.absolutePos(rightPos))
+                .add("b", helper.absolutePos(abovePos))
+                .save(manager.getDisk().get());
+
+        succeedIfManagerDidThingWithoutLagging(helper, manager, () -> {
+            assertTrue(leftChest.getStackInSlot(0).getCount() == 1, "Dirt moved");
+            assertTrue(rightChest.getStackInSlot(0).getCount() == 0, "Dirt did not move");
+            assertTrue(topChest.getStackInSlot(0).getCount() == 1, "Dirt did not move");
+
+            helper.succeed();
+        });
+    }
+
+    @GameTest(template = "3x3x1")
+    public static void output_where(GameTestHelper helper) {
+        // Only the right chest should get items
+        BlockPos managerPos = new BlockPos(1, 2, 0);
+        helper.setBlock(managerPos, SFMBlocks.MANAGER_BLOCK.get());
+        // Create positions
+        BlockPos leftPos = new BlockPos(2, 2, 0);
+        BlockPos rightPos = new BlockPos(0, 2, 0);
+        BlockPos abovePos = new BlockPos(1, 3, 0);
+        // Set blocks
+        helper.setBlock(leftPos, SFMBlocks.TEST_BARREL_BLOCK.get());
+        helper.setBlock(rightPos, SFMBlocks.TEST_BARREL_BLOCK.get());
+        helper.setBlock(abovePos, SFMBlocks.TEST_BARREL_BLOCK.get());
+
+        var leftChest = getItemHandler(helper, leftPos);
+        var rightChest = getItemHandler(helper, rightPos);
+        var topChest = getItemHandler(helper, abovePos);
+
+        rightChest.insertItem(1, new ItemStack(Blocks.STONE, 1), false);
+
+        topChest.insertItem(0, new ItemStack(Blocks.DIRT, 2), false);
+
+        ManagerBlockEntity manager = (ManagerBlockEntity) helper.getBlockEntity(managerPos);
+        manager.setItem(0, new ItemStack(SFMItems.DISK_ITEM.get()));
+        manager.setProgram("""
+                                   EVERY 20 TICKS DO
+                                       INPUT FROM a
+                                       OUTPUT RETAIN 1 dirt TO EACH b WHERE > 0 stone
+                                   END
+                                   """.stripTrailing().stripIndent());
+        // set the labels
+        LabelPositionHolder.empty()
+                .add("b", helper.absolutePos(leftPos))
+                .add("b", helper.absolutePos(rightPos))
+                .add("a", helper.absolutePos(abovePos))
+                .save(manager.getDisk().get());
+
+        succeedIfManagerDidThingWithoutLagging(helper, manager, () -> {
+            assertTrue(leftChest.getStackInSlot(0).getCount() == 0, "Dirt moved");
+            assertTrue(rightChest.getStackInSlot(0).getCount() == 1, "Dirt did not move");
+            assertTrue(topChest.getStackInSlot(0).getCount() == 1, "Dirt did not move");
+
+            helper.succeed();
+        });
+    }
+}
\ No newline at end of file
diff --git a/src/gametest/resources/data/sfm/structures/3x3x1.nbt b/src/gametest/resources/data/sfm/structures/3x3x1.nbt
new file mode 100644
index 0000000000000000000000000000000000000000..55845be4c0a8b30eff70541d87f6e8a1f2bb27a5
GIT binary patch
literal 167
zcmV;Y09gMYiwFP!00000|BcVR4uVh&1>nPv$VGhuU&7hh+0oThsId|5CA5=IFC?N2
z8fns@$#>ec031@42PXig9XaIeO-xuZltXr@LwVPtA2Nre4Lag?C0MAD@hdSk(F_yM
zFc^1DO!Vc}KNJ5j%a&?3<KfLXM|z}&v2Db2+0oVZZKy?WT%E~RtkUO3wD};CuHU;l
VkkN@dmEeK(4o_8O=(KME003V6PW=D?

literal 0
HcmV?d00001


From e0e496417507d2b563a9e6d9dd998f310833b973 Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 20:10:44 -0400
Subject: [PATCH 5/7] Fix inspection test

---
 src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java  | 2 +-
 src/main/java/ca/teamdman/sfml/ast/LabelAccess.java | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
index 88fdd0aa0..2dde77672 100644
--- a/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
+++ b/src/main/java/ca/teamdman/sfml/ast/ASTBuilder.java
@@ -674,7 +674,7 @@ public WhereStatement visitWhereNegation(SFMLParser.WhereNegationContext ctx) {
     @Override
     public WhereStatement visitWhereComparison(SFMLParser.WhereComparisonContext ctx) {
         var resourceComparison = visitResourcecomparison(ctx.resourcecomparison());
-        WhereStatement where = resourceComparison.toWhereExpression("WHERE " + resourceComparison.toString());
+        WhereStatement where = resourceComparison.toWhereExpression(" WHERE " + resourceComparison.toString());
         AST_NODE_CONTEXTS.add(new Pair<>(where, ctx));
         return where;
     }
diff --git a/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java b/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
index dd534b373..dbb2c2f1e 100644
--- a/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
+++ b/src/main/java/ca/teamdman/sfml/ast/LabelAccess.java
@@ -35,7 +35,7 @@ public String toString() {
                 }
             }
         }
-        builder.append(" ").append(where);
+        builder.append(where);
         return builder.toString();
     }
 }

From c753340f7a7f1ab6de6c7df036bd21bab33df7c7 Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 20:14:18 -0400
Subject: [PATCH 6/7] Fix roundrobin

---
 .../java/ca/teamdman/sfml/ast/InputStatement.java   | 13 +++++--------
 .../java/ca/teamdman/sfml/ast/OutputStatement.java  | 13 +++++--------
 2 files changed, 10 insertions(+), 16 deletions(-)

diff --git a/src/main/java/ca/teamdman/sfml/ast/InputStatement.java b/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
index f51b93d97..1a203f2cd 100644
--- a/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
+++ b/src/main/java/ca/teamdman/sfml/ast/InputStatement.java
@@ -282,14 +282,11 @@ private <STACK, ITEM, CAP> boolean shouldCreateSlot(
     }
 
     private List<Pair<Label, BlockPos>> getBlocksFromLabels(ProgramContext context) {
+        RoundRobin roundRobin = labelAccess.roundRobin();
         LabelPositionHolder labelPositionHolder = context.getLabelPositionHolder();
-        ArrayList<Pair<Label, BlockPos>> blocks = new ArrayList<>();
-        for (Label label : labelAccess.labels()) {
-            for (BlockPos pos : labelPositionHolder.getPositions(label.name())) {
-                blocks.add(Pair.of(label, pos));
-            }
-        }
-
-        return blocks;
+        return roundRobin.getPositionsForLabels(
+                labelAccess,
+                labelPositionHolder
+        );
     }
 }
diff --git a/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java b/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
index 83fc6b6d6..8aea17afa 100644
--- a/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
+++ b/src/main/java/ca/teamdman/sfml/ast/OutputStatement.java
@@ -511,14 +511,11 @@ private <STACK, ITEM, CAP> boolean shouldCreateSlot(
     }
 
     private List<Pair<Label, BlockPos>> getBlocksFromLabels(ProgramContext context) {
+        RoundRobin roundRobin = labelAccess.roundRobin();
         LabelPositionHolder labelPositionHolder = context.getLabelPositionHolder();
-        ArrayList<Pair<Label, BlockPos>> blocks = new ArrayList<>();
-        for (Label label : labelAccess.labels()) {
-            for (BlockPos pos : labelPositionHolder.getPositions(label.name())) {
-                blocks.add(Pair.of(label, pos));
-            }
-        }
-
-        return blocks;
+        return roundRobin.getPositionsForLabels(
+                labelAccess,
+                labelPositionHolder
+        );
     }
 }

From 05ee1a74845caebce361a734ab4bc43632414923 Mon Sep 17 00:00:00 2001
From: Burkino <24849979+Burkino@users.noreply.github.com>
Date: Mon, 29 Jul 2024 22:13:30 -0400
Subject: [PATCH 7/7] Add WHERE example

---
 .../template_programs/where_statement.sfml    | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 src/main/resources/assets/sfm/template_programs/where_statement.sfml

diff --git a/src/main/resources/assets/sfm/template_programs/where_statement.sfml b/src/main/resources/assets/sfm/template_programs/where_statement.sfml
new file mode 100644
index 000000000..8b5e4a777
--- /dev/null
+++ b/src/main/resources/assets/sfm/template_programs/where_statement.sfml
@@ -0,0 +1,43 @@
+NAME "WHERE Statements"
+
+-- WHERE statements allow you to filter blocks in
+-- a label by the contents of the block
+-- They support most of the same functionality as
+-- IF statements though they can't compare 2
+-- inventories (or slots) against each other
+
+EVERY 20 TICKS DO
+    INPUT FROM source WHERE > 5 dirt AND < 5 stone
+    OUTPUT TO EACH dest WHERE (= 0 dirt)
+END
+
+-- An issue with resource types
+EVERY 20 TICKS DO
+    -- If you are moving resource type X
+    -- but WHERE is checking resource type Y
+    -- you need to specify that you want to move
+    -- 0 of resource type Y on the input or output
+    -- statement with the WHERE condition
+    -- This is only necessary if you are not planning
+    -- on moving resource type Y in this statement
+
+    INPUT fluid:: FROM source
+    -- The following statement will not work
+    OUTPUT fluid:: TO dest WHERE > 0 dirt
+
+    -- To fix this, do
+    OUTPUT 0 item::, fluid:: TO dest WHERE > 0 dirt
+END
+
+-- Slots
+EVERY 20 TICKS DO
+    -- WHERE statements will use the same slots as
+    -- the SLOTS statement. This can be useful for
+    -- machines with distinct groups of slots, such as
+    -- extended inscribers
+    INPUT FROM source
+    OUTPUT gold_ingot, silicon
+        TO EACH exinscriber SLOTS 0-3 WHERE > 0 universal_press
+    OUTPUT gold_ingot, silicon
+        TO EACH exinscriber SLOTS 4-7 WHERE > 0 universal_press
+END
\ No newline at end of file