d

Pipfile.lock

@@ -1,7 +1,7 @@
 {
     "_meta": {
         "hash": {
-            "sha256": "3c94591e63af431c312ac30f7f7e19d0f406f796ce447837cc5edd363ff2764d"
+            "sha256": "1a84fb4e4ea0b66ca791e226c4da40bd974bc5eceec5868bb5752aad478c55e8"
         },
         "pipfile-spec": 6,
         "requires": {
@@ -259,7 +259,7 @@
                 "sha256:94078db9184491e15bce0a56d9186e0aec95f16ac20b12d00e06d4e36f1058a6",
                 "sha256:98a535c62a4fcfcc362528592f69b26f7caec587d32cd55688db580be0287ae0"
             ],
-            "markers": "python_version >= '3.6'",
+            "index": "pypi",
             "version": "==4.2.0"
         },
         "sphinxcontrib-applehelp": {

agtest/gen.py

@@ -34,11 +34,12 @@ class ParseEquation:
 
 
 class GenResult:
-    def __init__(self, pd: str = "", ex: str = ""):
+    def __init__(self, program: List[Decl]):
+        self.program = program
         self.literals: Dict[str, str] = dict()
         self.parse_rules: defaultdict[str, List[ParseEquation]] = defaultdict(list)
         self.starts: Set[str] = set()
-        self.extra = ex
+        self.extra: str = ""
         self.nonterminal_map: Dict[str, str] = dict()
 
     @property
@@ -77,9 +78,7 @@ class GenResult:
         return "\n".join(s)
 
 
-def gen(program: List[Decl]) -> GenResult:
-    res = GenResult()
-
+    def gen(self) -> None:
         def v(name: str) -> str:
             return f"__ag_{name}"
 
@@ -92,14 +91,14 @@ def gen(program: List[Decl]) -> GenResult:
         ifaces: Dict[str, Iface] = dict(
             map(
                 lambda c: (c.name, cast(Iface, c)),
-            filter(lambda c: isinstance(c, Iface), program),
+                filter(lambda c: isinstance(c, Iface), self.program),
             )
         )
 
         # list of node -> iface mappings
         what_ifaces: Dict[str, Set[str]] = dict()
         what_fields: Dict[str, Dict[str, str]] = dict()
-    for node in filter(lambda c: isinstance(c, Node), program):
+        for node in filter(lambda c: isinstance(c, Node), self.program):
             node = cast(Node, node)
             # all_fields = dict()
             what_ifaces[node.name] = set(node.ifaces)
@@ -148,14 +147,14 @@ def gen(program: List[Decl]) -> GenResult:
 
         node_map: Dict[str, NodeDesc] = dict()
 
-    for _node in filter(lambda c: isinstance(c, Node), program):
+        for _node in filter(lambda c: isinstance(c, Node), self.program):
             nd = NodeDesc(cast(Node, _node))
             node_map[_node.name] = nd
-        res.nonterminal_map[nd.name] = nd.nonterminal
+            self.nonterminal_map[nd.name] = nd.nonterminal
 
         for node_desc in node_map.values():
             assert isinstance(node_desc, NodeDesc)
-        res.starts.add(node_desc.nonterminal)
+            self.starts.add(node_desc.nonterminal)
 
             class_fields = []
             for field_name, field_ty in what_fields[node_desc.name].items():
@@ -169,7 +168,7 @@ def gen(program: List[Decl]) -> GenResult:
                     pass
                 """
             )
-        res.extra += class_decl
+            self.extra += class_decl
 
             print(node_desc.name, node_desc.node.ifaces)
 
@@ -180,7 +179,7 @@ def gen(program: List[Decl]) -> GenResult:
                     class {v_class_name}({node_desc.nonterminal}): pass
                     """
                 )
-            res.extra += class_decl
+                self.extra += class_decl
 
                 prod_name = gensym(node_desc.nonterminal + "_")
                 print("PRODUCTION", prod_name, variant.prod)
@@ -199,12 +198,12 @@ def gen(program: List[Decl]) -> GenResult:
                                 )
                         elif isinstance(sym.ty, NodeRegex):
                             sym_name = gensym("sym")
-                        res.literals[sym_name] = f"/{sym.ty.pat.pattern}/"
+                            self.literals[sym_name] = f"/{sym.ty.pat.pattern}/"
                             return sym_name
                     elif isinstance(sym, SymLit):
                         sym_name = gensym("lit")
                         # hack to make repr have double quotes
-                    res.literals[sym_name] = json.dumps(sym.lit)
+                        self.literals[sym_name] = json.dumps(sym.lit)
                         return sym_name
                     raise Exception(f"unhandled {sym.__class__}")
 
@@ -212,7 +211,7 @@ def gen(program: List[Decl]) -> GenResult:
                 for sym in variant.prod:
                     n = resolve_production(sym)
                     seq.append(n)
-            res.parse_rules[node_desc.nonterminal].append(
+                self.parse_rules[node_desc.nonterminal].append(
                     ParseEquation(prod_name, seq, v_class_name)
                 )
 
@@ -233,4 +232,3 @@ def gen(program: List[Decl]) -> GenResult:
                     print("RHS", eq.rhs, eq.rhs.id)
                     collect_required_thunks(copy.deepcopy(env), eq.rhs)
 
-    return res

agtest/main.py

@@ -4,6 +4,7 @@ from os import path
 import json
 import importlib
 from lark import Lark
+import sys
 
 from agtest.ast import *
 from agtest.gen import *

docs/agtest.rst

@@ -4,14 +4,6 @@ agtest package
 Submodules
 ----------
 
-agtest.ast module
------------------
-
-.. automodule:: agtest.ast
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
 agtest.gen module
 -----------------
 
@@ -20,14 +12,6 @@ agtest.gen module
    :undoc-members:
    :show-inheritance:
 
-agtest.main module
-------------------
-
-.. automodule:: agtest.main
-   :members:
-   :undoc-members:
-   :show-inheritance:
-
 Module contents
 ---------------
 

docs/index.rst

@@ -10,6 +10,8 @@ Welcome to agtest's documentation!
    :maxdepth: 2
    :caption: Contents:
 
+   agtest
+
 
 
 Indices and tables