stimflow 0.1.0__tar.gz → 1.16.dev0__tar.gz

{stimflow-0.1.0 → stimflow-1.16.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stimflow
-Version: 0.1.0
+Version: 1.16.dev0
 Summary: A library for creating quantum error correction circuits.
 Home-page: https://github.com/quantumlib/stim
 Author: Craig Gidney
@@ -38,7 +38,7 @@ stimflow also includes functionality for:
 
 - Transpiling (`stimflow.transpile_to_z_basis_interaction_circuit(...)`)
 - Adding Noise (`stimflow.NoiseModel.uniform_depolarizing(p).noisy_circuit(...)`)
-- Visualizing (`stimflow.make_3d_model`, `stimflow.stim_circuit_html_viewer`)
+- Visualizing (`stimflow.make_3d_model`, `stimflow.html_viewer`)
 
 # Documentation
 
@@ -153,7 +153,7 @@ def main():
     noisy_circuit = sf.NoiseModel.uniform_depolarizing(1e-3).noisy_circuit(circuit)
     distance = len(noisy_circuit.shortest_graphlike_error())
     assert distance == 7
-    sf.stim_circuit_html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
+    sf.html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
 
 
 if __name__ == "__main__":

{stimflow-0.1.0 → stimflow-1.16.dev0}/README.md

@@ -16,7 +16,7 @@ stimflow also includes functionality for:
 
 - Transpiling (`stimflow.transpile_to_z_basis_interaction_circuit(...)`)
 - Adding Noise (`stimflow.NoiseModel.uniform_depolarizing(p).noisy_circuit(...)`)
-- Visualizing (`stimflow.make_3d_model`, `stimflow.stim_circuit_html_viewer`)
+- Visualizing (`stimflow.make_3d_model`, `stimflow.html_viewer`)
 
 # Documentation
 
@@ -131,7 +131,7 @@ def main():
     noisy_circuit = sf.NoiseModel.uniform_depolarizing(1e-3).noisy_circuit(circuit)
     distance = len(noisy_circuit.shortest_graphlike_error())
     assert distance == 7
-    sf.stim_circuit_html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
+    sf.html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
 
 
 if __name__ == "__main__":

{stimflow-0.1.0 → stimflow-1.16.dev0}/setup.py

@@ -19,7 +19,7 @@ with open('README.md', encoding='UTF-8') as f:
 with open('requirements.txt', encoding='UTF-8') as f:
     requirements = f.read().splitlines()
 
-__version__ = '0.1.0'
+__version__ = '1.16.dev0'
 
 setup(
     name='stimflow',

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/__init__.py

@@ -46,7 +46,7 @@ from stimflow._viz import (
     Viewable3dModelGLTF,
     html_viewer_for_gltf_model,
     make_3d_model,
-    stim_circuit_html_viewer,
-    svg,
+    html_viewer,
+    svg_viewer,
     TextDataFor3DModel,
 )

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_chunk.py

@@ -509,20 +509,9 @@ class Chunk:
         tile_color_func: Callable[[Tile], tuple[float, float, float, float]] | None = None,
         known_error: Iterable[stim.ExplainedError] | None = None,
     ) -> str_html:
-        from stimflow._viz import stim_circuit_html_viewer
-
-        circuit = self.to_closed_circuit()
-        if background is None:
-            start = self.start_patch()
-            end = self.end_patch()
-            if len(start.tiles) == 0:
-                background = end
-            elif len(end.tiles) == 0:
-                background = start
-            else:
-                background = {0: start, circuit.num_ticks: end}
-        return stim_circuit_html_viewer(
-            circuit, background=background, tile_color_func=tile_color_func, known_error=known_error
+        from stimflow._viz import html_viewer
+        return html_viewer(
+            self, background=background, tile_color_func=tile_color_func, known_error=known_error
         )
 
     def __mul__(self, other: int) -> ChunkLoop:

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_chunk_compiler.py

@@ -89,9 +89,9 @@ class ChunkCompiler:
         copy = self.copy()
         if copy.open_flows:
             copy.append_magic_end_chunk()
-        from stimflow._viz import stim_circuit_html_viewer
+        from stimflow._viz import html_viewer
 
-        return stim_circuit_html_viewer(
+        return html_viewer(
             circuit=copy.finish_circuit(), background=self.cur_end_interface()
         )
 

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_chunk_interface.py

@@ -64,9 +64,9 @@ class ChunkInterface:
         elif other is not None:
             flat.extend(other)
 
-        from stimflow._viz import svg
+        from stimflow._viz import svg_viewer
 
-        return svg(
+        return svg_viewer(
             objects=flat,
             show_obs=show_obs,
             show_measure_qubits=show_measure_qubits,

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_chunk_loop.py

@@ -161,13 +161,13 @@ class ChunkLoop:
         known_error: Iterable[stim.ExplainedError] | None = None,
     ) -> str_html:
         """Returns an HTML document containing a viewer for the chunk loop's circuit."""
-        from stimflow._viz import stim_circuit_html_viewer
+        from stimflow._viz import html_viewer
 
         if patch is None:
             patch = self.start_patch()
             if len(patch.tiles) == 0:
                 patch = self.end_patch()
-        return stim_circuit_html_viewer(
+        return html_viewer(
             self.to_closed_circuit(),
             background=patch,
             tile_color_func=tile_color_func,

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_patch.py

@@ -5,7 +5,7 @@ import functools
 from collections.abc import Callable, Iterable, Iterator
 from typing import Literal, overload, TYPE_CHECKING
 
-from stimflow._core import PauliMap, str_svg, Tile
+from stimflow._core import PauliMap, str_svg, Tile, min_max_complex
 
 if TYPE_CHECKING:
     from stimflow._chunk._stabilizer_code import StabilizerCode
@@ -22,7 +22,7 @@ class Patch:
             elif isinstance(tile, PauliMap):
                 kept_tiles.append(tile.to_tile())
             else:
-                raise ValueError(f"Don't know how to interpret this as a stimflow.Tile: {tile=}")
+                raise ValueError(f"Don't know how to convert a {type(tile)} into a stimflow.Tile: {tile=}")
         if not do_not_sort:
             kept_tiles = sorted(kept_tiles)
 
@@ -39,6 +39,9 @@ class Patch:
     def __getitem__(self, item: slice) -> Patch:
         pass
 
+    def _min_max_complex_(self) -> tuple[complex, complex]:
+        return min_max_complex(self.used_set, default=0)
+
     def __getitem__(self, item: int | slice) -> Patch | Tile:
         if isinstance(item, slice):
             return Patch(self.tiles[item])
@@ -113,12 +116,12 @@ class Patch:
         tile_color_func: Callable[[Tile], str] | None = None,
     ) -> str_svg:
         from stimflow._chunk._stabilizer_code import StabilizerCode
-        from stimflow._viz import svg
+        from stimflow._viz import svg_viewer
 
         patches = [self] + ([other] if isinstance(other, (Patch, StabilizerCode)) else list(other))
 
-        return svg(
-            objects=patches,
+        return svg_viewer(
+            patches,
             show_measure_qubits=show_measure_qubits,
             show_data_qubits=show_data_qubits,
             show_order=show_order,

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_chunk/_stabilizer_code.py

@@ -355,19 +355,33 @@ class StabilizerCode:
                         f"\n    {loc2.gate_target.pauli_type} at {loc2.coords}"
                     )
 
+    def _min_max_complex_(self) -> tuple[complex, complex]:
+        return min_max_complex(self.used_set, default=0)
+
     def find_distance(self, *, max_search_weight: int) -> int:
         return len(self.find_logical_error(max_search_weight=max_search_weight))
 
-    def find_logical_error(self, *, max_search_weight: int) -> list[stim.ExplainedError]:
+    def find_logical_error(self, *, max_search_weight: int, return_stim_explained_error: bool = False) -> PauliMap | list[stim.ExplainedError]:
         circuit = self.make_code_capacity_circuit(noise=1e-3)
         if max_search_weight == 2:
-            return circuit.shortest_graphlike_error(canonicalize_circuit_errors=True)
-        return circuit.search_for_undetectable_logical_errors(
-            dont_explore_edges_with_degree_above=max_search_weight,
-            dont_explore_detection_event_sets_with_size_above=max_search_weight,
-            dont_explore_edges_increasing_symptom_degree=False,
-            canonicalize_circuit_errors=True,
-        )
+            result = circuit.shortest_graphlike_error(canonicalize_circuit_errors=True)
+        else:
+            result = circuit.search_for_undetectable_logical_errors(
+                dont_explore_edges_with_degree_above=max_search_weight,
+                dont_explore_detection_event_sets_with_size_above=max_search_weight,
+                dont_explore_edges_increasing_symptom_degree=False,
+                canonicalize_circuit_errors=True,
+            )
+        if return_stim_explained_error:
+            return result
+
+        pauli_map = {}
+        for err in result:
+            for loc in err.circuit_error_locations:
+                for term in loc.flipped_pauli_product:
+                    real, imag = term.coords
+                    pauli_map[real + 1j*imag] = term.gate_target.pauli_type
+        return PauliMap(pauli_map)
 
     def with_observables_from_basis(self, basis: Literal["X", "Y", "Z"]) -> StabilizerCode:
         if basis == "X":
@@ -459,7 +473,7 @@ class StabilizerCode:
                 t2 = tile2.to_pauli_map()
                 if not t1.commutes(t2):
                     raise ValueError(
-                        f"Tile stabilizer {t1=} anticommutes with tile stabilizer {t2=}."
+                        f"The following two stabilizers anticommute:\n    {t1}\n    {t2}\n\nMore details:\n\n{tile1!r}\n\n{tile2!r}"
                     )
 
         for tile in self.stabilizers.tiles:
@@ -538,10 +552,10 @@ class StabilizerCode:
         elif other is not None:
             flat.extend(other)
 
-        from stimflow._viz import svg
+        from stimflow._viz import svg_viewer
 
-        return svg(
-            objects=flat,
+        return svg_viewer(
+            flat,
             title=title,
             show_obs=show_obs,
             canvas_height=canvas_height,

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_core/_flow.py

@@ -281,7 +281,6 @@ class Flow:
             start=new_start,
             end=new_end,
             measurement_indices=xor_sorted(self.measurement_indices + other.measurement_indices),
-            obs_name=self.obs_name,
             flags=self.flags | other.flags,
             center=new_center,
             sign=(None if self.sign is None else self.sign ^ other.sign),

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_core/_pauli_map.py

@@ -5,7 +5,7 @@ from typing import Any, cast, Literal, TYPE_CHECKING
 
 import stim
 
-from stimflow._core._complex_util import sorted_complex
+from stimflow._core._complex_util import sorted_complex, min_max_complex
 
 if TYPE_CHECKING:
     from stimflow._core._tile import Tile
@@ -122,6 +122,15 @@ class PauliMap:
             self._dict = {}
         self._hash = hash((self.obs_name, tuple(self._dict.items())))
 
+    def _min_max_complex_(self) -> tuple[complex, complex]:
+        return min_max_complex(self.keys(), default=0)
+
+    def _inline_svg_(self, *, q2p: Callable[[complex], complex], out_lines: list[str]):
+        scale = abs(q2p(1) - q2p(0))
+        for q, p in self.items():
+            pt = q2p(q)
+            out_lines.append(f'''<text x="{pt.real}" y="{pt.imag}" dominant-baseline="central" text-anchor="middle" font-size="{scale}">{p}</text>''')
+
     @staticmethod
     def from_xs(xs: Iterable[complex], *, name: Any = None) -> PauliMap:
         """Returns a PauliMap mapping the given qubits to the X basis."""
@@ -203,7 +212,7 @@ class PauliMap:
             c = "IXZY"[cx + cz * 2]
             if c != "I":
                 result[q] = cast(Literal["X", "Y", "Z"], c)
-        return PauliMap(result)
+        return PauliMap(result, obs_name=self.obs_name if self.obs_name == other.obs_name else None)
 
     def __repr__(self) -> str:
         if self.obs_name is None:

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_layers/_layer_circuit.py

@@ -5,6 +5,7 @@ from typing import Any, cast, Literal, TypeVar
 
 import stim
 
+from stimflow._core import min_max_complex
 from stimflow._layers._layer_det_obs_annotation import DetObsAnnotationLayer
 from stimflow._layers._layer_empty import LayerEmpty
 from stimflow._layers._layer_feedback import LayerFeedback
@@ -37,6 +38,12 @@ class LayerCircuit:
 
     layers: list[Layer] = dataclasses.field(default_factory=list)
 
+    def _min_max_complex_(self) -> tuple[complex, complex]:
+        return min_max_complex(
+            v[0] + v[1] * 1j
+            for v in obj.to_stim_circuit().get_final_qubit_coordinates().values()
+        )
+
     def touched(self) -> set[int]:
         result = set()
         for layer in self.layers:

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/__init__.py

@@ -2,11 +2,11 @@ from stimflow._viz._3d_model_viewer import (
     Viewable3dModelGLTF,
     html_viewer_for_gltf_model,
 )
-from stimflow._viz._viz_circuit_html import stim_circuit_html_viewer
+from stimflow._viz._viz_html import html_viewer
 from stimflow._viz._3d_model import (
     LineDataFor3DModel,
     TriangleDataFor3DModel,
     TextDataFor3DModel,
     make_3d_model,
 )
-from stimflow._viz._viz_svg import svg
+from stimflow._viz._viz_svg import svg_viewer

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/_viz_circuit_html.py

@@ -7,7 +7,7 @@ import math
 import random
 import sys
 from collections.abc import Callable, Iterable
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 import stim
 
@@ -750,7 +750,7 @@ def append_patch_polygons(
         out.append(line)
 
 
-def stim_circuit_html_viewer(
+def _stim_circuit_html_viewer(
     circuit: stim.Circuit,
     *,
     background: (

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/_viz_circuit_html_test.py

@@ -25,5 +25,5 @@ def test_viewer_works_with_all_gates():
         else:
             targets = [0, 1]
         circuit.append(name, targets, args)
-    viewer = stimflow.stim_circuit_html_viewer(circuit)
+    viewer = stimflow.html_viewer(circuit)
     assert viewer is not None

stimflow-1.16.dev0/src/stimflow/_viz/_viz_html.py

@@ -0,0 +1,84 @@
+from __future__ import annotations
+
+from collections.abc import Callable, Iterable
+from typing import TYPE_CHECKING, Any
+
+import stim
+
+from stimflow._viz._viz_circuit_html import _stim_circuit_html_viewer
+
+if TYPE_CHECKING:
+    import stimflow
+    from stimflow._core import str_html
+
+
+def html_viewer(
+    obj: stim.Circuit | Any,
+    *,
+    background: (
+        stimflow.Patch
+        | stimflow.StabilizerCode
+        | stimflow.ChunkInterface
+        | dict[int, stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface]
+        | None
+    ) = None,
+    tile_color_func: (
+        Callable[[stimflow.Tile], tuple[float, float, float, float] | tuple[float, float, float] | str]
+        | None
+    ) = None,
+    width: int = 500,
+    height: int = 500,
+    known_error: Iterable[stim.ExplainedError] | None = None,
+) -> str_html:
+    """Creates an HTML page for viewing the given object.
+
+    Args:
+        obj: The object to be visualized.
+        background: Something to draw in the background of the viewer (e.g. the
+            stimflow.StabilizerCode implemented by the circuit being viewed).
+        tile_color_func: Customizes how stabilizers and other operators are drawn.
+        width: The width of the viewer.
+        height: The height of the viewer.
+        known_error: An error (e.g. returned from stim.Circuit.shortest_graphlike_error)
+            to show as part of the object.
+
+    Returns:
+        The HTML string (as a stimflow.str_html, which inherits from python's `str`).
+
+        (The result is of type stimflow.str_html so that its viewer is shown automatically
+        in Jupyter notebooks, and also for convenience methods like `write_to`.)
+    """
+
+    from stimflow._chunk import Chunk
+    if isinstance(obj, stim.Circuit):
+        return _stim_circuit_html_viewer(
+            obj,
+            background=background,
+            tile_color_func=tile_color_func,
+            width=width,
+            height=height,
+            known_error=known_error,
+        )
+
+    elif isinstance(obj, Chunk):
+        circuit = obj.to_closed_circuit()
+        if background is None:
+            start = obj.start_patch()
+            end = obj.end_patch()
+            if len(start.tiles) == 0:
+                background = end
+            elif len(end.tiles) == 0:
+                background = start
+            else:
+                background = {0: start, circuit.num_ticks: end}
+        return _stim_circuit_html_viewer(
+            circuit,
+            background=background,
+            tile_color_func=tile_color_func,
+            known_error=known_error,
+            width=width,
+            height=height,
+        )
+
+    else:
+        raise NotImplementedError(f"Don't know how to make an html viewer for {type(obj)=}.")

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/_viz_patch_svg.py

@@ -245,6 +245,15 @@ def _draw_patch(
     stabilizer_style: Literal["polygon", "circles"] | None,
     observable_style: Literal["label", "polygon", "circles"],
 ) -> None:
+    from stimflow._chunk import Patch, StabilizerCode, ChunkInterface
+    if hasattr(obj, "_inline_svg_"):
+        obj._inline_svg_(
+            out_lines=out_lines,
+            q2p=q2p,
+        )
+        return
+    if not isinstance(obj, (Patch, StabilizerCode, ChunkInterface, stim.Circuit)):
+        raise NotImplementedError(f'{type(obj)=}')
     if isinstance(obj, stim.Circuit):
         from stimflow._viz._viz_circuit_layer_svg import append_circuit_layer_to_svg
 
@@ -275,25 +284,20 @@ def _draw_patch(
                     f"WARNING: No logical error will be drawn.\n    Reason: {ex}",
                     file=sys.stderr,
                 )
-                err = []
-            for e in err:
-                for loc in e.circuit_error_locations:
-                    for loc2 in loc.flipped_pauli_product:
-                        real, imag = loc2.coords
-                        q = real + 1j * imag
-                        p = loc2.gate_target.pauli_type
-                        labels.append(
-                            (
-                                q,
-                                p + "!",
-                                {
-                                    "text-anchor": "middle",
-                                    "dominant-baseline": "central",
-                                    "font-size": scale_factor * 1.1,
-                                    "fill": BASE_COLORS_DARK[p],
-                                },
-                            )
-                        )
+                err = {}
+            for q, p in err.items():
+                labels.append(
+                    (
+                        q,
+                        p + "!",
+                        {
+                            "text-anchor": "middle",
+                            "dominant-baseline": "central",
+                            "font-size": scale_factor * 1.1,
+                            "fill": BASE_COLORS_DARK[p],
+                        },
+                    )
+                )
 
     if isinstance(obj, StabilizerCode) and show_obs:
         _draw_obs(
@@ -314,9 +318,9 @@ def _draw_patch(
 
     all_points = set(system_qubits)
     if show_data_qubits:
-        all_points |= obj.data_set
+        all_points |= getattr(obj, 'data_set', set())
     if show_measure_qubits:
-        all_points |= obj.measure_set
+        all_points |= getattr(obj, 'measure_set', set())
     if show_coords and all_points:
         all_x = sorted({q.real for q in all_points})
         all_y = sorted({q.imag for q in all_points})

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/_viz_patch_svg_test.py

@@ -30,5 +30,5 @@ def test_patch_svg_runs():
             ),
         ]
     )
-    svg_content = stimflow.svg([patch])
+    svg_content = stimflow.svg_viewer([patch])
     assert svg_content is not None

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow/_viz/_viz_svg.py

@@ -2,9 +2,10 @@ from __future__ import annotations
 
 import math
 from collections.abc import Callable, Iterable
-from typing import Literal, TYPE_CHECKING
+from typing import Literal, TYPE_CHECKING, Any
 
 import stim
+from stimflow import PauliMap
 
 from stimflow._viz._viz_patch_svg import _draw_patch
 
@@ -12,10 +13,10 @@ if TYPE_CHECKING:
     import stimflow
 
 
-def svg(
-    objects: Iterable[stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface | stim.Circuit],
+def svg_viewer(
+    obj: stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface | stim.Circuit | PauliMap | Any | Iterable[stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface | stim.Circuit | PauliMap | Any],
     *,
-    background: stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface | stim.Circuit | None = None,
+    background: stimflow.Patch | stimflow.StabilizerCode | stimflow.ChunkInterface | stim.Circuit | PauliMap | Any | None = None,
     title: str | list[str] | None = None,
     canvas_height: int | None = None,
     show_order: bool = False,
@@ -41,32 +42,39 @@ def svg(
     show_frames: bool = True,
     pad: float | None = None,
 ) -> stimflow.str_svg:
-    """Returns an SVG image of the given objects."""
+    """Returns an SVG image of the given objects.
+    """
+    from stimflow._chunk import Patch, StabilizerCode, ChunkInterface
+    if hasattr(obj, '_inline_svg_') or isinstance(obj, (Patch, StabilizerCode, ChunkInterface, stim.Circuit, PauliMap)):
+        objects = [obj]
+    else:
+        objects = obj
+
     system_qubits = frozenset(system_qubits)
     if canvas_height is None:
         canvas_height = 500
 
     extra_used_coords = frozenset(extra_used_coords)
-    from stimflow._layers import LayerCircuit
-
-    patches = tuple(
-        patch.to_stim_circuit() if isinstance(patch, LayerCircuit) else patch for patch in objects
-    )
-    all_points: set[complex] = set()
-    all_points.update(system_qubits)
-    all_points.update(extra_used_coords)
-    for patch in patches:
-        if isinstance(patch, stim.Circuit):
-            all_points.update(
-                v[0] + v[1] * 1j for v in patch.get_final_qubit_coordinates().values()
+
+    min_max_points: set[complex] = set()
+    min_max_points.update(system_qubits)
+    min_max_points.update(extra_used_coords)
+    for e in [*objects, background]:
+        if e is None:
+            continue
+        elif hasattr(e, '_min_max_complex_'):
+            min_max_points.update(e._min_max_complex_())
+        elif isinstance(e, stim.Circuit):
+            min_max_points.update(
+                v[0] + v[1] * 1j for v in e.get_final_qubit_coordinates().values()
             )
         else:
-            all_points.update(patch.used_set)
+            raise NotImplementedError(f"Don't know how to determine qubits used by {type(e)=}")
     if show_all_qubits:
-        system_qubits = frozenset(all_points)
+        system_qubits = frozenset(min_max_points)
     from stimflow._core import min_max_complex
 
-    min_c, max_c = min_max_complex(all_points, default=0)
+    min_c, max_c = min_max_complex(min_max_points, default=0)
     min_c -= 0.5 + 0.5j
     max_c += 0.5 + 0.5j
     offset: complex = 0
@@ -82,14 +90,14 @@ def svg(
     box_x_pitch = box_width + pad
     box_y_pitch = box_height + pad
     if cols is None and rows is None:
-        cols = min(len(patches), math.ceil(math.sqrt(len(patches) * 2)))
-        rows = math.ceil(len(patches) / max(1, cols))
+        cols = min(len(objects), math.ceil(math.sqrt(len(objects) * 2)))
+        rows = math.ceil(len(objects) / max(1, cols))
     elif cols is None:
-        cols = math.ceil(len(patches) / max(1, rows))
+        cols = math.ceil(len(objects) / max(1, rows))
     elif rows is None:
-        rows = math.ceil(len(patches) / max(1, cols))
+        rows = math.ceil(len(objects) / max(1, cols))
     else:
-        assert cols * rows >= len(patches)
+        assert cols * rows >= len(objects)
     total_height = max(1.0, box_y_pitch * rows - pad + offset.imag)
     total_width = max(1.0, box_x_pitch * cols - pad + offset.real)
     scale_factor = canvas_height / max(total_height, 1)
@@ -132,7 +140,7 @@ def svg(
             )
 
     clip_path_id_ptr = [0]
-    for plan_i, plan in enumerate(patches):
+    for plan_i, plan in enumerate(objects):
         layers = [plan]
         if background is not None:
             if isinstance(background, (tuple, list)):
@@ -160,7 +168,7 @@ def svg(
 
     # Draw frame outlines
     if show_frames:
-        for outline_index in range(len(patches)):
+        for outline_index in range(len(objects)):
             a = patch_q2p(outline_index, min_c)
             a += offset
             b = patch_q2p(outline_index, max_c)

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stimflow
-Version: 0.1.0
+Version: 1.16.dev0
 Summary: A library for creating quantum error correction circuits.
 Home-page: https://github.com/quantumlib/stim
 Author: Craig Gidney
@@ -38,7 +38,7 @@ stimflow also includes functionality for:
 
 - Transpiling (`stimflow.transpile_to_z_basis_interaction_circuit(...)`)
 - Adding Noise (`stimflow.NoiseModel.uniform_depolarizing(p).noisy_circuit(...)`)
-- Visualizing (`stimflow.make_3d_model`, `stimflow.stim_circuit_html_viewer`)
+- Visualizing (`stimflow.make_3d_model`, `stimflow.html_viewer`)
 
 # Documentation
 
@@ -153,7 +153,7 @@ def main():
     noisy_circuit = sf.NoiseModel.uniform_depolarizing(1e-3).noisy_circuit(circuit)
     distance = len(noisy_circuit.shortest_graphlike_error())
     assert distance == 7
-    sf.stim_circuit_html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
+    sf.html_viewer(noisy_circuit, background=code).write_to('tmp2.html')
 
 
 if __name__ == "__main__":

{stimflow-0.1.0 → stimflow-1.16.dev0}/src/stimflow.egg-info/SOURCES.txt

@@ -82,6 +82,7 @@ src/stimflow/_viz/__init__.py
 src/stimflow/_viz/_viz_circuit_html.py
 src/stimflow/_viz/_viz_circuit_html_test.py
 src/stimflow/_viz/_viz_circuit_layer_svg.py
+src/stimflow/_viz/_viz_html.py
 src/stimflow/_viz/_viz_patch_svg.py
 src/stimflow/_viz/_viz_patch_svg_test.py
 src/stimflow/_viz/_viz_svg.py