Add: optimise random switches

OpenTTD · Oct 17, 2020 · aede745 · aede745
1 parent 7f62081
commit aede745
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nml/expression/spritegroup_ref.py b/nml/expression/spritegroup_ref.py
@@ -82,6 +82,9 @@ def reduce(self, id_dicts = [], unknown_id_fatal = True):
                     len(spritegroup.body.ranges) == 0:
                 generic.print_warning("Block '{}' returns a constant, optimising.".format(spritegroup.name.value), self.pos)
                 return spritegroup.body.default.value
+            elif isinstance(spritegroup, switch.RandomSwitch) and len(spritegroup.choices) == 1:
+                generic.print_warning("Block '{}' returns a constant, optimising.".format(spritegroup.name.value), self.pos)
+                return spritegroup.choices[0].result.value
         return self
 
     def supported_by_action2(self, raise_error):