basic miner loop

Author: ckrause

.github/workflows/main.yml

@@ -18,7 +18,5 @@ jobs:
           pip3 install pdoc3
       - name: Test
         run: nose2 tests -v
-      - name: Sample
-        run: python3 sample.py
       - name: Docs
         run: pdoc3 --html loda

loda/lang/program.py

@@ -2,7 +2,7 @@
 
 """Program model and serialization."""
 
-from .operation import Operation
+from .operation import Operation, Operand
 
 
 class Program:
@@ -51,3 +51,17 @@ def __str__(self) -> str:
             if op.type == Operation.Type.LPB:
                 indent += 1
         return result
+
+    def validate(self):
+        loop_depth = 0
+        for op in self.operations:
+            if op.type != Operation.Type.NOP and op.type != Operation.Type.LPE and op.target.type == Operand.Type.CONSTANT:
+                raise ValueError("target cannot be a constant")
+            if op.type == Operation.Type.LPB:
+                loop_depth += 1
+            elif op.type == Operation.Type.LPE:
+                if loop_depth == 0:
+                    raise ValueError("unexpected lpe")
+                loop_depth -= 1
+        if loop_depth != 0:
+            raise ValueError("missing lpe")

loda/mine/__init__.py

@@ -0,0 +1,3 @@
+"""Program mining for integer sequences."""
+
+from .miner import Miner

loda/mine/miner.py

@@ -0,0 +1,31 @@
+"""Miner class for finding integer sequence programs."""
+
+from loda.oeis import PrefixIndex, Sequence
+from loda.runtime import Evaluator, Interpreter
+
+
+class Miner:
+
+    def __init__(self, sequences: list, interpreter: Interpreter, generator):
+        self.__index = PrefixIndex(sequences)
+        self.__interpreter = interpreter
+        self.__generator = generator
+
+    def __call__(self):
+        program = self.__generator()
+        evaluator = Evaluator(program, self.__interpreter)
+        match = self.__index.global_match()
+        refine = True
+        try:
+            while refine:
+                term = evaluator()
+                # print(term)
+                refine = self.__index.refine_match(match, term)
+        except Exception as e:
+            print("evaluation error: {}".format(e))
+            return
+
+        ids = self.__index.get_match_ids(match)
+        for id in ids:
+            print("Found match for {}".format(Sequence(id)))
+            print(program)