Add sympy Serialization

dev_tools/qualtran_dev_tools/bloq_report_card.py

@@ -105,6 +105,7 @@ def get_bloq_report_card(
     bexamples: Optional[Iterable[BloqExample]] = None,
     package_prefix: str = 'qualtran.bloqs.',
 ) -> pd.DataFrame:
+
     if bclasses is None:
         bclasses = get_bloq_classes()
     if bexamples is None:

qualtran/protos/__init__.py

@@ -11,4 +11,3 @@
 #  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.
-

qualtran/protos/bloq.proto

@@ -21,6 +21,7 @@ import "qualtran/protos/args.proto";
 import "qualtran/protos/registers.proto";
 import "qualtran/protos/data_types.proto";
 import "qualtran/protos/ctrl_spec.proto";
+import "qualtran/protos/sympy.proto";
 
 package qualtran;
 
@@ -32,7 +33,7 @@ message BloqArg {
     double float_val = 3;
     string string_val = 4;
     // Sympy expression generated using str(expr).
-    string sympy_expr = 5;
+    Term sympy_expr = 5;
     // N-dimensional numpy array stored as bytes.
     NDArray ndarray = 6;
     // Integer reference of a subbloq. Assumes access to a BloqLibrary.

qualtran/protos/bloq_pb2.pyi

@@ -26,6 +26,7 @@ import qualtran.protos.args_pb2
 import qualtran.protos.ctrl_spec_pb2
 import qualtran.protos.data_types_pb2
 import qualtran.protos.registers_pb2
+import qualtran.protos.sympy_pb2
 import sys
 
 if sys.version_info >= (3, 8):
@@ -56,8 +57,9 @@ class BloqArg(google.protobuf.message.Message):
     int_val: builtins.int
     float_val: builtins.float
     string_val: builtins.str
-    sympy_expr: builtins.str
-    """Sympy expression generated using str(expr)."""
+    @property
+    def sympy_expr(self) -> qualtran.protos.sympy_pb2.Term:
+        """Sympy expression generated using str(expr)."""
     @property
     def ndarray(self) -> qualtran.protos.args_pb2.NDArray:
         """N-dimensional numpy array stored as bytes."""
@@ -86,7 +88,7 @@ class BloqArg(google.protobuf.message.Message):
         int_val: builtins.int = ...,
         float_val: builtins.float = ...,
         string_val: builtins.str = ...,
-        sympy_expr: builtins.str = ...,
+        sympy_expr: qualtran.protos.sympy_pb2.Term | None = ...,
         ndarray: qualtran.protos.args_pb2.NDArray | None = ...,
         subbloq: builtins.int = ...,
         cirq_json_gzip: builtins.bytes = ...,

qualtran/protos/sympy.proto

@@ -0,0 +1,75 @@
+/*
+  Copyright 2023 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.
+*/
+
+syntax = "proto3";
+package qualtran;
+
+// A function sympy expression.
+enum Function {
+    // Each Term has an associated function. A "NONE" function means that the term
+    // is made up of a single parameter and can not be decomposed further.
+    NONE = 0;
+    Add = 1;
+    Mul = 2;
+    Pow = 3;
+    Mod = 4;
+    Log = 5;
+    Floor = 6;
+    Ceiling = 7;
+    Max = 8;
+    Min = 9;
+    Sin = 10;
+    Cos = 11;
+    Tan = 12;
+}
+
+// Represents a constant, rational number.
+message Rational {
+    Parameter numerator = 1;
+    Parameter denominator = 2;
+}
+
+enum ConstSymbol {
+    Pi = 0;
+    E = 1;
+    EulerGamma = 2;
+    Infinity = 3;
+    ImaginaryUnit = 4;
+}
+
+// A single parameter of a sympy expression.
+message Parameter {
+    oneof parameter {
+        int32 const_int = 1;
+        string symbol = 2;
+        Rational const_rat = 3;
+        float const_float = 4;
+        ConstSymbol const_symbol = 5;
+
+    }
+}
+
+message Operand {
+    oneof operand {
+        Term term = 1;
+        Parameter parameter = 2;
+    }
+}
+
+message Term {
+    Function function = 1;
+    repeated Operand operands = 2;
+}