Added uses_variables method

Author: Theomat

synth/syntax/program.py

@@ -25,6 +25,12 @@ def __hash__(self) -> int:
     def __repr__(self) -> str:
         return self.__str__()
 
+    def uses_variables(self) -> bool:
+        """
+        Returns true if a variable is used.
+        """
+        return False
+
     def used_variables(self) -> Set[int]:
         """
         Returns the set of used variables numbers in this program.
@@ -139,6 +145,9 @@ def __init__(self, variable: int, type: Type = UnknownType()):
     def is_invariant(self, constant_types: Set[PrimitiveType]) -> bool:
         return False
 
+    def uses_variables(self) -> bool:
+        return True
+
     def clone(self) -> "Program":
         return Variable(self.variable)
 
@@ -297,6 +306,11 @@ def is_constant(self) -> bool:
             arg.is_constant() for arg in self.arguments
         )
 
+    def uses_variables(self) -> bool:
+        return self.function.uses_variables() or any(
+            arg.uses_variables() for arg in self.arguments
+        )
+
     def constants(self) -> Generator[Optional["Constant"], None, None]:
         g = [self.function.constants()] + [arg.constants() for arg in self.arguments]
         for gen in g:
@@ -382,6 +396,9 @@ def all_constants_instantiation(
     def __eq__(self, other: Any) -> bool:
         return isinstance(other, Lambda) and self.body == other.body
 
+    def uses_variables(self) -> bool:
+        return self.body.uses_variables()
+
     def __add_used_variables__(self, vars: Set[int]) -> None:
         return self.body.__add_used_variables__(vars)