qubasic 0.1.0__py3-none-any.whl

qbasic_core/state_display.py

@@ -0,0 +1,60 @@
+"""QBASIC state display mixin — STATE, BLOCH, HIST, PROBS commands."""
+
+from qbasic_core.engine import MAX_BLOCH_DISPLAY
+
+
+class StateDisplayMixin:
+    """State inspection commands for QBasicTerminal.
+
+    Requires: TerminalProtocol — uses self.locc_mode, self.last_sv,
+    self.last_counts, self.last_circuit, self.num_qubits,
+    self._locc_state(), self._locc_bloch(),
+    self._print_statevector(), self._print_probs(), self._print_bloch_single(),
+    self.print_histogram(), self.io.
+    """
+
+    def cmd_state(self, rest: str = '') -> None:
+        if self.locc_mode:
+            return self._locc_state(rest)
+        if self.last_sv is None:
+            self.io.writeln("?NO STATE — RUN first")
+            return
+        self._print_statevector(self.last_sv)
+
+    def cmd_hist(self) -> None:
+        if self.last_counts is None:
+            self.io.writeln("?NO RESULTS — RUN first")
+            return
+        self.print_histogram(self.last_counts)
+
+    def cmd_probs(self) -> None:
+        if self.last_sv is None:
+            self.io.writeln("?NO STATE — RUN first")
+            return
+        self._print_probs(self.last_sv)
+
+    def cmd_bloch(self, rest: str) -> None:
+        if self.locc_mode:
+            return self._locc_bloch(rest)
+        if self.last_sv is None:
+            self.io.writeln("?NO STATE — RUN first")
+            return
+        if rest:
+            q = int(rest)
+            self._print_bloch_single(self.last_sv, q)
+        else:
+            for q in range(min(self.num_qubits, MAX_BLOCH_DISPLAY)):
+                self._print_bloch_single(self.last_sv, q)
+                if q < min(self.num_qubits, MAX_BLOCH_DISPLAY) - 1:
+                    self.io.writeln('')
+
+    def cmd_circuit(self) -> None:
+        if self.last_circuit is None:
+            self.io.writeln("?NO CIRCUIT — RUN first")
+            return
+        try:
+            self.io.writeln(self.last_circuit.draw(output='text'))
+        except Exception:
+            self.io.writeln(f"Circuit: {self.last_circuit.num_qubits} qubits, "
+                           f"depth {self.last_circuit.depth()}, "
+                           f"{self.last_circuit.size()} gates")

qbasic_core/statements.py

@@ -0,0 +1,387 @@
+"""QBASIC typed statement AST — produced by the parser, consumed by the executor."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+
+@dataclass(frozen=True, slots=True)
+class Stmt:
+    """Base for all parsed statements."""
+    raw: str
+
+
+# ── Terminals ──────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class RemStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class MeasureStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class EndStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class ReturnStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class BarrierStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class WendStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class RestoreStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class EndSelectStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class EndSubStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class EndFunctionStmt(Stmt):
+    pass
+
+@dataclass(frozen=True, slots=True)
+class StopStmt(Stmt):
+    pass
+
+
+# ── Control flow ───────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class GotoStmt(Stmt):
+    target: int
+
+@dataclass(frozen=True, slots=True)
+class GosubStmt(Stmt):
+    target: int
+
+@dataclass(frozen=True, slots=True)
+class ForStmt(Stmt):
+    var: str
+    start_expr: str
+    end_expr: str
+    step_expr: str | None
+
+@dataclass(frozen=True, slots=True)
+class NextStmt(Stmt):
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class WhileStmt(Stmt):
+    condition: str
+
+@dataclass(frozen=True, slots=True)
+class IfThenStmt(Stmt):
+    condition: str
+    then_clause: str
+    else_clause: str | None
+
+@dataclass(frozen=True, slots=True)
+class DoStmt(Stmt):
+    kind: str | None   # WHILE or UNTIL or None
+    condition: str | None
+
+@dataclass(frozen=True, slots=True)
+class LoopStmt(Stmt):
+    kind: str | None
+    condition: str | None
+
+@dataclass(frozen=True, slots=True)
+class ExitStmt(Stmt):
+    target: str   # FOR, WHILE, DO, SUB, FUNCTION
+
+@dataclass(frozen=True, slots=True)
+class OnGotoStmt(Stmt):
+    expr: str
+    targets: tuple[int, ...]
+
+@dataclass(frozen=True, slots=True)
+class OnGosubStmt(Stmt):
+    expr: str
+    targets: tuple[int, ...]
+
+@dataclass(frozen=True, slots=True)
+class SelectCaseStmt(Stmt):
+    expr: str
+
+@dataclass(frozen=True, slots=True)
+class CaseStmt(Stmt):
+    value: str
+
+@dataclass(frozen=True, slots=True)
+class CallStmt(Stmt):
+    name: str
+    args: str
+
+@dataclass(frozen=True, slots=True)
+class SubStmt(Stmt):
+    name: str
+    params: str
+
+@dataclass(frozen=True, slots=True)
+class FunctionStmt(Stmt):
+    name: str
+    params: str
+
+@dataclass(frozen=True, slots=True)
+class OnErrorStmt(Stmt):
+    target: int
+
+@dataclass(frozen=True, slots=True)
+class ResumeStmt(Stmt):
+    arg: str | None
+
+@dataclass(frozen=True, slots=True)
+class ErrorStmt(Stmt):
+    code: int
+
+@dataclass(frozen=True, slots=True)
+class AssertStmt(Stmt):
+    condition: str
+
+@dataclass(frozen=True, slots=True)
+class SwapStmt(Stmt):
+    a: str
+    b: str
+
+@dataclass(frozen=True, slots=True)
+class DefFnStmt(Stmt):
+    name: str
+    params: str
+    body: str
+
+@dataclass(frozen=True, slots=True)
+class OptionBaseStmt(Stmt):
+    base: int
+
+@dataclass(frozen=True, slots=True)
+class OnMeasureStmt(Stmt):
+    target: int
+
+@dataclass(frozen=True, slots=True)
+class OnTimerStmt(Stmt):
+    interval: str
+    target: int
+
+@dataclass(frozen=True, slots=True)
+class DataStmt(Stmt):
+    values: str
+
+@dataclass(frozen=True, slots=True)
+class ReadStmt(Stmt):
+    var_list: str
+
+@dataclass(frozen=True, slots=True)
+class LocalStmt(Stmt):
+    var_list: str
+
+@dataclass(frozen=True, slots=True)
+class StaticStmt(Stmt):
+    var_list: str
+
+@dataclass(frozen=True, slots=True)
+class SharedStmt(Stmt):
+    var_list: str
+
+
+# ── Assignment ─────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class LetStmt(Stmt):
+    name: str
+    expr: str
+
+@dataclass(frozen=True, slots=True)
+class LetArrayStmt(Stmt):
+    name: str
+    index_expr: str
+    value_expr: str
+
+@dataclass(frozen=True, slots=True)
+class LetStrStmt(Stmt):
+    name: str
+    expr: str
+
+
+# ── I/O ────────────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class PrintStmt(Stmt):
+    expr: str
+
+@dataclass(frozen=True, slots=True)
+class PrintUsingStmt(Stmt):
+    fmt: str
+    values: str
+
+@dataclass(frozen=True, slots=True)
+class InputStmt(Stmt):
+    prompt: str | None
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class LineInputStmt(Stmt):
+    prompt: str | None
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class GetStmt(Stmt):
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class DimStmt(Stmt):
+    name: str
+    size: str
+
+@dataclass(frozen=True, slots=True)
+class RedimStmt(Stmt):
+    name: str
+    size: str
+
+@dataclass(frozen=True, slots=True)
+class EraseStmt(Stmt):
+    name: str
+
+@dataclass(frozen=True, slots=True)
+class PokeStmt(Stmt):
+    addr_expr: str
+    value_expr: str
+
+@dataclass(frozen=True, slots=True)
+class SysStmt(Stmt):
+    arg: str
+
+@dataclass(frozen=True, slots=True)
+class UnitaryStmt(Stmt):
+    name: str
+    matrix: str
+
+@dataclass(frozen=True, slots=True)
+class OpenStmt(Stmt):
+    path: str
+    mode: str
+    handle: int
+    encoding: str | None = None
+
+@dataclass(frozen=True, slots=True)
+class CloseStmt(Stmt):
+    handle: int
+
+@dataclass(frozen=True, slots=True)
+class PrintFileStmt(Stmt):
+    handle: int
+    data: str
+
+@dataclass(frozen=True, slots=True)
+class InputFileStmt(Stmt):
+    handle: int
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class LprintStmt(Stmt):
+    expr: str
+
+@dataclass(frozen=True, slots=True)
+class ScreenStmt(Stmt):
+    mode: str
+
+@dataclass(frozen=True, slots=True)
+class ColorStmt(Stmt):
+    fg: str
+    bg: str | None
+
+@dataclass(frozen=True, slots=True)
+class LocateStmt(Stmt):
+    row: str
+    col: str
+
+@dataclass(frozen=True, slots=True)
+class ImportStmt(Stmt):
+    path: str
+
+@dataclass(frozen=True, slots=True)
+class ChainStmt(Stmt):
+    path: str
+
+@dataclass(frozen=True, slots=True)
+class MergeStmt(Stmt):
+    path: str
+
+
+# ── Quantum ────────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class MeasStmt(Stmt):
+    qubit_expr: str
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class ResetStmt(Stmt):
+    qubit_expr: str
+
+@dataclass(frozen=True, slots=True)
+class MeasureBasisStmt(Stmt):
+    basis: str
+    qubit_expr: str
+
+@dataclass(frozen=True, slots=True)
+class SyndromeStmt(Stmt):
+    rest: str
+
+
+# ── LOCC ───────────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class SendStmt(Stmt):
+    reg: str
+    qubit_expr: str
+    var: str
+
+@dataclass(frozen=True, slots=True)
+class ShareStmt(Stmt):
+    reg1: str
+    q1: int
+    reg2: str
+    q2: int
+
+@dataclass(frozen=True, slots=True)
+class AtRegStmt(Stmt):
+    reg: str
+    inner: str
+
+
+# ── Compound ───────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class CompoundStmt(Stmt):
+    parts: tuple[str, ...]
+
+
+# ── Gate application ───────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class GateStmt(Stmt):
+    """Direct gate application: H 0, CX 0,1, RX PI/4, 0"""
+    name: str          # canonical gate name (after alias resolution)
+    args: tuple[str, ...]  # raw argument strings (params + qubits, unparsed)
+
+
+# ── Fallback ───────────────────────────────────────────────────────────
+
+@dataclass(frozen=True, slots=True)
+class RawStmt(Stmt):
+    """Unrecognized statement — falls through to legacy regex path."""
+    pass

qbasic_core/strings.py

@@ -0,0 +1,107 @@
+"""QBASIC string support — string variables, functions, and operations."""
+
+from __future__ import annotations
+
+import math
+from typing import Any
+
+
+def _left(s: str, n: float) -> str:
+    return s[:int(n)]
+
+def _right(s: str, n: float) -> str:
+    return s[-int(n):] if int(n) > 0 else ''
+
+def _mid(s: str, start: float, length: float = -1) -> str:
+    i = int(start) - 1  # 1-based in BASIC
+    if length < 0:
+        return s[i:]
+    return s[i:i + int(length)]
+
+def _instr(haystack: str, needle: str) -> float:
+    pos = haystack.find(needle)
+    return float(pos + 1) if pos >= 0 else 0.0  # 1-based, 0 = not found
+
+def _asc(s: str) -> float:
+    return float(ord(s[0])) if s else 0.0
+
+def _chr_fn(n: float) -> str:
+    return chr(int(n))
+
+def _str_fn(n: float) -> str:
+    v = float(n)
+    return str(int(v)) if v == int(v) else str(v)
+
+def _val_fn(s: str) -> float:
+    try:
+        return float(s)
+    except (ValueError, TypeError):
+        return 0.0
+
+def _hex_fn(n: float) -> str:
+    return format(int(n), 'X')
+
+def _bin_fn(n: float) -> str:
+    return format(int(n), 'b')
+
+def _len_fn(x: Any) -> float:
+    if isinstance(x, str):
+        return float(len(x))
+    if isinstance(x, (list, tuple)):
+        return float(len(x))
+    return float(len(str(x)))
+
+
+# Functions that return strings (names ending in $)
+STRING_FUNCS: dict[str, Any] = {
+    'LEFT$': _left, 'RIGHT$': _right, 'MID$': _mid,
+    'CHR$': _chr_fn, 'STR$': _str_fn, 'HEX$': _hex_fn, 'BIN$': _bin_fn,
+}
+
+# Functions that take/return mixed types
+MIXED_FUNCS: dict[str, Any] = {
+    'ASC': _asc, 'VAL': _val_fn, 'INSTR': _instr, 'LEN': _len_fn,
+}
+
+
+class StringMixin:
+    """String variable and function support for QBasicTerminal.
+
+    Requires: TerminalProtocol — uses self.variables, self._safe_eval().
+    """
+
+    def _is_string_var(self, name: str) -> bool:
+        return name.endswith('$')
+
+    def _get_string_ns(self) -> dict[str, Any]:
+        """Return namespace entries for string functions."""
+        ns: dict[str, Any] = {}
+        ns.update(STRING_FUNCS)
+        ns.update(MIXED_FUNCS)
+        # String variables
+        for k, v in self.variables.items():
+            if isinstance(v, str):
+                ns[k] = v
+        return ns
+
+    def _eval_string_expr(self, expr: str) -> str | float:
+        """Evaluate an expression that might return a string."""
+        expr = expr.strip()
+        # Quoted string literal
+        if (expr.startswith('"') and expr.endswith('"')) or \
+           (expr.startswith("'") and expr.endswith("'")):
+            return expr[1:-1]
+        # String variable
+        if expr in self.variables and isinstance(self.variables[expr], str):
+            return self.variables[expr]
+        # Try numeric
+        try:
+            return self.eval_expr(expr)
+        except Exception:
+            return expr
+
+    def cmd_let_str(self, name: str, expr: str) -> None:
+        """Assign a string value to a string variable."""
+        val = self._eval_string_expr(expr)
+        self.variables[name] = val
+        self.io.writeln(f"{name} = {val!r}" if isinstance(val, str) else f"{name} = {val}")