✨ feat: add auto_simplify function

Author: SigureMo

README.md

@@ -17,7 +17,7 @@
 
 ### Quick start in CLI
 
-We recommend using the tool `uv` to run the without manually installing the package:
+We recommend using the tool `uv` to try it without manually installing the package:
 
 ```console
 $ uvx expr_simplifier cse "a * 4 + (a * 4)"
@@ -32,7 +32,23 @@ a and b
 
 ### As a library
 
-TODO...
+```console
+$ uv add expr-simplifier
+```
+
+```python
+from __future__ import annotations
+
+import ast
+
+from expr_simplifier import auto_simplify
+
+code = "a[1 + 1].b + a[3 - 1].b.c + (___x := 2) + (a[2].b and (___x == 2) and a[2].b)"
+tree = ast.parse(code, mode="eval")
+simplified_tree = auto_simplify(tree)
+print(ast.unparse(simplified_tree))
+# (___t_1 := a[2].b) + ___t_1.c + 2 + ___t_1
+```
 
 ## TODOs
 

examples/auto_simplify.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import ast
+
+from expr_simplifier import auto_simplify
+
+code = "a[1 + 1].b + a[3 - 1].b.c + (___x := 2) + (a[2].b and (___x == 2) and a[2].b)"
+tree = ast.parse(code, mode="eval")
+simplified_tree = auto_simplify(tree)
+print(ast.unparse(simplified_tree))
+# (___t_1 := a[2].b) + ___t_1.c + 2 + ___t_1

src/expr_simplifier/__init__.py

@@ -1,6 +1,8 @@
 # Meta information for the project.
 from __future__ import annotations
 
+from expr_simplifier.transforms.auto_simplify import auto_simplify as auto_simplify
+
 __version__ = "0.1.0"
 __author__ = "Nyakku Shigure"
 __year__ = "2023"

src/expr_simplifier/transforms/auto_simplify.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+import ast
+
+from expr_simplifier.transforms.constant_folding import apply_constant_folding
+from expr_simplifier.transforms.cse import apply_cse
+from expr_simplifier.transforms.logical_simplification import apply_logical_simplification
+from expr_simplifier.utils import loop_until_stable
+
+
+def auto_simplify(expr: ast.AST) -> ast.AST:
+    return loop_until_stable(expr, [apply_constant_folding, apply_logical_simplification, apply_cse], max_iter=100)