Qualtran-L1: Objectstrings

dev_tools/dump-bloq-manifest.py

@@ -0,0 +1,173 @@
+#  Copyright 2025 Google LLC
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+"""
+Script to dump a manifest of all bloq classes and bloq examples.
+
+This script finds all Bloq classes and examples in the library, serializes the examples
+to their objectstring representation, and writes the list of class names and
+example objectstrings to `qualtran/bloqs/manifest.py`.
+
+See `qualtran.l1.load_objectstring()` to load bloq objects from these strings.
+"""
+
+from functools import cached_property
+from typing import List, Tuple
+
+from attrs import frozen
+from qualtran_dev_tools.bloq_finder import get_bloq_classes, get_bloq_examples
+from qualtran_dev_tools.git_tools import get_git_root
+
+from qualtran import Bloq
+from qualtran.l1 import eval_cvalue_node, parse_objectstring, to_cobject_node
+from qualtran.l1.nodes import CArgNode, CObjectNode, LiteralNode
+
+MAXLEN = 300
+"""If the objectstring is too long, we make the executive decision to truncate it."""
+
+COPYRIGHT_NOTICE = """#  Copyright 2026 Google LLC
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License."""
+
+
+@frozen
+class BloqExampleListItem:
+    """Formatting a bloq example and its serialized form.
+
+    Attributes:
+        bloq: The bloq instance from the example.
+        bloq_example_name: The name of the bloq example.
+        cobject_node: The CObjectNode representing the serialized bloq.
+    """
+
+    bloq: Bloq
+    bloq_example_name: str
+    cobject_node: CObjectNode
+
+    @cached_property
+    def objectstring(self) -> str:
+        """Returns the canonical string representation of the CObjectNode."""
+        return self.cobject_node.canonical_str()
+
+    def maybe_commented_out(self, be_column_width: int = 30) -> Tuple[str, str, str]:
+        """Generates a string representation for the manifest, potentially commented out.
+
+        This method checks if the object string is too long, unparsable, unloadable, or
+        if the re-loaded bloq is unequal to the original. If any of these conditions are met,
+        it returns a commented-out string with a reason. Otherwise, it returns the
+        executable string representation.
+
+        Args:
+            be_column_width: The width of the column for the bloq example name for nicer formatting.
+
+        Returns:
+            manifest_entry_str: The string to be written to the manifest (possibly commented out).
+            reason: A reason string if it was commented out (empty otherwise).
+            reason_details: Detailed error message if applicable (empty otherwise).
+        """
+        quoted_be = f"'{self.bloq_example_name}',"
+        be = f'{quoted_be:{be_column_width}}'
+        be2 = f'{quoted_be:{be_column_width+2}}'
+
+        if len(self.objectstring) > MAXLEN:
+            trunc = repr(self.objectstring)[:120] + '...'
+            return f'# ({be}{trunc}', 'too long', ''
+
+        try:
+            node = parse_objectstring(self.objectstring)
+        except Exception as e:  # pylint: disable=broad-except
+            return f'# ({be}{self.objectstring!r}),', 'unparsable', str(e)
+
+        try:
+            bloq = eval_cvalue_node(node, safe=False)
+        except Exception as e:  # pylint: disable=broad-except
+            return f'# ({be}{self.objectstring!r}),', 'unloadable', str(e)
+
+        if bloq != self.bloq:
+            return f'# ({be}{self.objectstring!r}),', 'unequal', ''
+
+        return f'({be2}{self.objectstring!r}),', '', ''
+
+
+def main():
+    """Main entry point for the script.
+
+    Finds all bloq classes and examples, processes them, and writes the
+    `BLOQ_CLASS_NAMES` and `BLOQ_EXAMPLE_OBJECTSTRINGS` lists to
+    `qualtran/bloqs/manifest.py`.
+    """
+    bcs = get_bloq_classes()
+    names = sorted(bc._class_name_in_pkg_() for bc in bcs)
+
+    bes = get_bloq_examples()
+    items: List[BloqExampleListItem] = []
+    for be in bes:
+        bloq = be.make()
+        try:
+            cobject_node = to_cobject_node(bloq)
+            assert isinstance(cobject_node, CObjectNode)
+        except Exception as e:  # pylint: disable=broad-except
+            cobject_node = CObjectNode(
+                name=bloq._class_name_in_pkg_(), cargs=[CArgNode(None, LiteralNode(str(e)))]
+            )
+        items.append(
+            BloqExampleListItem(bloq=bloq, bloq_example_name=be.name, cobject_node=cobject_node)
+        )
+
+    items = sorted(items, key=lambda x: x.objectstring)
+    include_commented_out = True
+    be_objectstrings = []
+    for item in items:
+        serstr, reason, details = item.maybe_commented_out()
+
+        if (not reason) or include_commented_out:
+            be_objectstrings.append(serstr)
+
+        if reason:
+            reason = f'({reason})'
+            print(f"Skipping {reason:20s} {serstr}")
+
+        if details:
+            print(f'         {"":20s} ->', details)
+
+    reporoot = get_git_root()
+    with (reporoot / 'qualtran/bloqs/manifest.py').open('w') as f:
+        f.write(COPYRIGHT_NOTICE)
+        f.write('\n\n')
+        f.write('# This file is autogenerated\n')
+        f.write('# See dev_tools/dump-bloq-manifest.py\n')
+        f.write('# fmt: off\n\n')
+        f.write('BLOQ_CLASS_NAMES = [\n')
+        f.write('\n'.join([f'    "{name}",' for name in names]))
+        f.write('\n]\n')
+
+        f.write('\n')
+
+        f.write('BLOQ_EXAMPLE_OBJECTSTRINGS = [\n')
+        f.write('\n'.join([f'    {objstr}' for objstr in be_objectstrings]))
+        f.write('\n]\n')
+
+
+if __name__ == '__main__':
+    main()

docs/bloq_infra.rst

@@ -55,4 +55,5 @@ types (``Register``), and algorithms (``CompositeBloq``).
    drawing/musical_score.ipynb
    drawing/drawing_call_graph.ipynb
    simulation/xcheck_classical_quimb.ipynb
+   l1/L1-Objectstring.ipynb
    Autodoc.ipynb

qualtran/_infra/adjoint.py

@@ -174,6 +174,10 @@ def __str__(self) -> str:
         """Delegate to subbloq's `__str__` method."""
         return f'{str(self.subbloq)}†'
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        return 'qualtran'
+
     def wire_symbol(
         self, reg: Optional['Register'], idx: Tuple[int, ...] = tuple()
     ) -> 'WireSymbol':

qualtran/_infra/bloq.py

@@ -695,6 +695,10 @@ def wire_symbol(
     def __str__(self):
         return self.__class__.__name__
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        return '.'.join(cls.__module__.split('.')[:-1])
+
     @classmethod
     def _class_name_in_pkg_(cls) -> str:
         """The bloq class's name with its package.
@@ -703,5 +707,4 @@ def _class_name_in_pkg_(cls) -> str:
         `qualtran.bloqs.*`. Each bloq class is defined in a module (i.e. the
         "*.py" file) and re-exported one level up.
         """
-        pkg = '.'.join(cls.__module__.split('.')[:-1])
-        return f'{pkg}.{cls.__name__}'
+        return f'{cls._pkg_()}.{cls.__name__}'

qualtran/_infra/controlled.py

@@ -311,6 +311,10 @@ def get_single_ctrl_val(self) -> ControlBit:
 
         return int(control_bit)
 
+    @classmethod
+    def _pkg_(cls):
+        return 'qualtran'
+
 
 class AddControlledT(Protocol):
     """The signature for the `add_controlled` callback part of `ctrl_system`.
@@ -674,6 +678,10 @@ def build_call_graph(self, ssa: 'SympySymbolAllocator') -> 'BloqCountDictT':
     def adjoint(self) -> 'Bloq':
         return self.subbloq.adjoint().controlled(ctrl_spec=self.ctrl_spec)
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        return 'qualtran'
+
 
 def make_ctrl_system_with_correct_metabloq(
     bloq: 'Bloq', ctrl_spec: 'CtrlSpec'

qualtran/_infra/data_types.py

@@ -245,6 +245,10 @@ def iteration_length_or_zero(self) -> SymbolicInt:
         # TODO: remove https://github.com/quantumlib/Qualtran/issues/1716
         return getattr(self, 'iteration_length', 0)
 
+    @classmethod
+    def _pkg_(cls):
+        return 'qualtran'
+
     def __str__(self):
         return f'{self.__class__.__name__}({self.num_bits})'
 

qualtran/_infra/registers.py

@@ -49,6 +49,9 @@ class Side(enum.Flag):
     THRU = LEFT | RIGHT
     """The register is input/output."""
 
+    def __repr__(self):
+        return f'{self.__class__.__name__}.{self._name_}'
+
 
 @frozen
 class Register:
@@ -75,6 +78,13 @@ class Register:
     )
     side: Side = Side.THRU
 
+    @classmethod
+    def _pkg_(cls):
+        return 'qualtran'
+
+    def __repr__(self):
+        return f'Register({self.name!r}, dtype={self.dtype!r}, shape={self._shape!r}, side={self.side!r})'
+
     def __attrs_post_init__(self):
         if not isinstance(self.dtype, QCDType):
             raise ValueError(f'dtype must be a QCDType: found {type(self.dtype)}')

qualtran/bloqs/arithmetic/permutation.py

@@ -135,6 +135,11 @@ def build_call_graph(
 
         return super().build_call_graph(ssa)
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        # This isn't re-exported
+        return '.'.join(cls.__module__.split('.')[:])
+
 
 @bloq_example
 def _permutation_cycle() -> PermutationCycle:
@@ -281,6 +286,11 @@ def build_call_graph(
 
         return super().build_call_graph(ssa)
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        # This isn't re-exported
+        return '.'.join(cls.__module__.split('.')[:])
+
 
 @bloq_example
 def _permutation() -> Permutation:

qualtran/bloqs/arithmetic/sorting.py

@@ -282,6 +282,11 @@ def build_composite_bloq(
     def build_call_graph(self, ssa: 'SympySymbolAllocator') -> BloqCountDictT:
         return {Comparator(self.bitsize): self.num_comparisons}
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        # Local import
+        return '.'.join(cls.__module__.split('.')[:])
+
 
 @bloq_example
 def _bitonic_merge() -> BitonicMerge:
@@ -383,6 +388,11 @@ def build_composite_bloq(self, bb: 'BloqBuilder', xs: 'SoquetT') -> dict[str, 'S
     def build_call_graph(self, ssa: 'SympySymbolAllocator') -> BloqCountDictT:
         return {Comparator(self.bitsize): self.num_comparisons}
 
+    @classmethod
+    def _pkg_(cls) -> str:
+        # This isn't re-exported
+        return '.'.join(cls.__module__.split('.')[:])
+
 
 @bloq_example
 def _bitonic_sort() -> BitonicSort: