PyPI - seqeyes-python - Versions diffs - 0.2.1__py3-none-any.whl → 0.2.6__py3-none-any.whl - Mend

seqeyes-python 0.2.1py3-none-any.whl → 0.2.6py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (9) hide show

seqeyes/__init__.py +1 -1
seqeyes/_plot.py +16 -3
seqeyes/_renderer.py +47 -18
seqeyes/resources/pulseq-bundle.js +944 -387
seqeyes/resources/viewer.html +43 -16
{seqeyes_python-0.2.1.dist-info → seqeyes_python-0.2.6.dist-info}/METADATA +13 -3
seqeyes_python-0.2.6.dist-info/RECORD +8 -0
seqeyes_python-0.2.1.dist-info/RECORD +0 -8
{seqeyes_python-0.2.1.dist-info → seqeyes_python-0.2.6.dist-info}/WHEEL +0 -0

seqeyes/__init__.py CHANGED Viewed

@@ -29,4 +29,4 @@ from seqeyes._plot import set, reset
 from seqeyes._renderer import SeqEyesViewer
 __all__ = ["set", "reset", "SeqEyesViewer"]
-__version__ = "0.2.1"
+__version__ = "0.2.6"

seqeyes/_plot.py CHANGED Viewed

@@ -20,6 +20,9 @@ from typing import Any, Optional, Tuple, Union
 _defaults: dict[str, Any] = {}
 _original_plot = None
+_original_repr_html = None
+_had_original_repr_html = False
+_patched_repr_html = None
 _patched = False
@@ -53,7 +56,7 @@ def reset() -> None:
     Call ``seqeyes.set()`` again to re‑enable at any time.
     """
-    global _patched
+    global _patched, _patched_repr_html
     _defaults.clear()
     _patched = False
@@ -64,6 +67,12 @@ def reset() -> None:
     if _original_plot is not None:
         _Seq.plot = _original_plot  # type: ignore[attr-defined]
+    if _patched_repr_html is not None and getattr(_Seq, "_repr_html_", None) is _patched_repr_html:
+        if _had_original_repr_html:
+            _Seq._repr_html_ = _original_repr_html  # type: ignore[attr-defined]
+        else:
+            delattr(_Seq, "_repr_html_")
+    _patched_repr_html = None
 # ── Internal helpers ──────────────────────────────────────────────────────
@@ -87,7 +96,7 @@ def _store_defaults(
 def _ensure_patched() -> None:
-    global _patched, _original_plot
+    global _patched, _original_plot, _original_repr_html, _had_original_repr_html, _patched_repr_html
     if _patched:
         return
@@ -100,6 +109,9 @@ def _ensure_patched() -> None:
     if _original_plot is None:
         _original_plot = getattr(_Seq, "plot", None)
+    if _patched_repr_html is None:
+        _had_original_repr_html = hasattr(_Seq, "_repr_html_")
+        _original_repr_html = getattr(_Seq, "_repr_html_", None)
     # ── Replacement .plot() ──────────────────────────────────────────
     def _seqeyes_plot(
@@ -108,7 +120,7 @@ def _ensure_patched() -> None:
         show_blocks: bool = False,
         time_range: Any = (0, float("inf")),
         time_disp: str = "s",
-        grad_disp: str = "kHz/m",
+        grad_disp: str = "Hz/m",
         **kwargs: Any,
     ) -> None:
         # Merge globals — per‑call args take precedence
@@ -147,6 +159,7 @@ def _ensure_patched() -> None:
         return SeqEyesViewer(_seq_to_text(self))._repr_html_()
     _Seq._repr_html_ = _seq_repr_html_  # type: ignore[attr-defined]
+    _patched_repr_html = _seq_repr_html_
 def _in_jupyter() -> bool:

seqeyes/_renderer.py CHANGED Viewed

@@ -12,7 +12,7 @@ import base64
 import json
 import os
 from pathlib import Path
-from typing import Optional
+from typing import Any, Optional, Union
 # ── Paths ─────────────────────────────────────────────────────────────────
@@ -26,6 +26,8 @@ _BUNDLE_CANDIDATES = [
     Path(__file__).resolve().parent.parent.parent.parent / "web" / "pulseq-bundle.js",
     Path.cwd() / "web" / "pulseq-bundle.js",
 ]
+_VALID_GRAD_UNITS = {"Hz/m", "mT/m", "G/cm"}
+SequenceSource = Union[str, bytes, bytearray, memoryview]
 def _find_bundle() -> str:
@@ -53,8 +55,13 @@ def _read_viewer_template() -> str:
     )
+def _normalize_grad_disp(unit: str) -> str:
+    """Return a viewer-supported gradient unit."""
+    return unit if unit in _VALID_GRAD_UNITS else "Hz/m"
 def _build_html(
-    seq_text: str,
+    sequence_source: SequenceSource,
     *,
     theme: Optional[str] = None,
     inject_bundle: bool = True,
@@ -62,14 +69,14 @@ def _build_html(
     show_blocks: bool = False,
     time_range: tuple = (0, float("inf")),
     time_disp: str = "s",
-    grad_disp: str = "kHz/m",
+    grad_disp: str = "Hz/m",
 ) -> str:
     """Assemble the complete viewer HTML.
     Parameters
     ----------
-    seq_text : str
-        Raw .seq file content.
+    sequence_source : str or bytes-like
+        Raw text ``.seq`` content or binary ``.bseq`` bytes.
     theme : str or None
         CSS theme class to apply to ``<body>``.
     inject_bundle : bool
@@ -83,11 +90,13 @@ def _build_html(
     time_disp : str
         Initial time display unit (``"s"``, ``"ms"``, ``"us"``).
     grad_disp : str
-        Initial gradient display unit (``"Hz/m"``, ``"kHz/m"``,
-        ``"mT/m"``, ``"G/cm"``).
+        Initial gradient display unit (``"Hz/m"``, ``"mT/m"``, ``"G/cm"``).
     """
     template = _read_viewer_template()
-    seq_b64 = base64.b64encode(seq_text.encode("utf-8")).decode("ascii")
+    grad_disp = _normalize_grad_disp(grad_disp)
+    source_is_text = isinstance(sequence_source, str)
+    source_bytes = sequence_source.encode("utf-8") if source_is_text else bytes(sequence_source)
+    source_b64 = base64.b64encode(source_bytes).decode("ascii")
     # 1. Inject the pulseq-bundle.js
     bundle_placeholder = "<!-- PULSEQ_BUNDLE_PLACEHOLDER -->"
@@ -102,7 +111,9 @@ def _build_html(
     # 2. Build options injection block (sequence data + display opts)
     opts = [
-        f"window.SEQEYES_RAW_B64 = {json.dumps(seq_b64)};",
+        f"window.SEQEYES_RAW_B64 = {json.dumps(source_b64)};",
+        f"window.SEQEYES_SOURCE_KIND = {json.dumps('text' if source_is_text else 'bytes')};",
+        f"window.SEQEYES_SOURCE_NAME = {json.dumps(label)};",
         f"window.SEQEYES_LABEL = {json.dumps(label)};",
         f"window.SEQEYES_SHOW_BLOCKS = {json.dumps(show_blocks)};",
         f"window.SEQEYES_TIME_RANGE = {json.dumps(list(time_range))};",
@@ -138,8 +149,8 @@ class SeqEyesViewer:
     Parameters
     ----------
-    seq_text : str
-        Raw .seq file content as a string.
+    sequence_source : str or bytes-like
+        Raw text ``.seq`` content or binary ``.bseq`` bytes.
     label : str
         Display label (not yet rendered on the viewer).
     show_blocks : bool
@@ -150,7 +161,7 @@ class SeqEyesViewer:
     time_disp : str
         Time axis unit — ``"s"``, ``"ms"``, or ``"us"``.
     grad_disp : str
-        Gradient axis unit — ``"Hz/m"``, ``"kHz/m"``, ``"mT/m"``, ``"G/cm"``.
+        Gradient axis unit — ``"Hz/m"``, ``"mT/m"``, or ``"G/cm"``.
     theme : str or None
         Colour theme.
     width : str
@@ -161,33 +172,51 @@ class SeqEyesViewer:
     def __init__(
         self,
-        seq_text: str,
+        sequence_source: SequenceSource,
         *,
         label: str = "",
         show_blocks: bool = False,
         time_range: tuple = (0, float("inf")),
         time_disp: str = "s",
-        grad_disp: str = "kHz/m",
+        grad_disp: str = "Hz/m",
         theme: Optional[str] = None,
         width: str = "100%",
         height: str = "550px",
     ) -> None:
-        self._seq_text = seq_text
+        if not isinstance(sequence_source, (str, bytes, bytearray, memoryview)):
+            raise TypeError("sequence_source must be .seq text or .bseq bytes")
+        self._sequence_source = sequence_source
         self._label = label
         self._show_blocks = show_blocks
         self._time_range = time_range
         self._time_disp = time_disp
-        self._grad_disp = grad_disp
+        self._grad_disp = _normalize_grad_disp(grad_disp)
         self._theme = theme
         self._width = width
         self._height = height
+    @classmethod
+    def from_file(cls, path: Union[str, os.PathLike[str]], **kwargs: Any) -> "SeqEyesViewer":
+        """Create a viewer from a local ``.seq`` or ``.bseq`` file."""
+        source_path = Path(path)
+        if source_path.suffix.lower() not in {".seq", ".bseq"}:
+            raise ValueError("SeqEyesViewer.from_file() expects a .seq or .bseq file")
+        if not source_path.is_file():
+            raise FileNotFoundError(source_path)
+        kwargs.setdefault("label", source_path.name)
+        source: SequenceSource = (
+            source_path.read_text(encoding="utf-8")
+            if source_path.suffix.lower() == ".seq"
+            else source_path.read_bytes()
+        )
+        return cls(source, **kwargs)
     # ── Jupyter integration ───────────────────────────────────────────
     def _repr_html_(self) -> str:
         """Return an HTML iframe that Jupyter renders inline."""
         html = _build_html(
-            self._seq_text,
+            self._sequence_source,
             theme=self._theme,
             label=self._label,
             show_blocks=self._show_blocks,
@@ -225,7 +254,7 @@ class SeqEyesViewer:
     def to_html(self, *, inject_bundle: bool = True) -> str:
         """Return the full standalone HTML string (for saving to a file)."""
         return _build_html(
-            self._seq_text,
+            self._sequence_source,
             theme=self._theme,
             inject_bundle=inject_bundle,
             label=self._label,

seqeyes-python 0.2.1__py3-none-any.whl → 0.2.6__py3-none-any.whl

seqeyes-python 0.2.1py3-none-any.whl → 0.2.6py3-none-any.whl