extend qasm3 dumper with annotations in measurements

Author: 1ucian0

qiskit/qasm3/ast.py

@@ -388,11 +388,17 @@ class QuantumMeasurementAssignment(Statement):
         | indexIdentifier EQUALS quantumMeasurement  # eg: bits = measure qubits;
     """
 
-    __slots__ = ("identifier", "quantumMeasurement")
+    __slots__ = ("identifier", "quantumMeasurement", "annotations")
 
-    def __init__(self, identifier: Identifier, quantumMeasurement: QuantumMeasurement):
+    def __init__(
+        self,
+        identifier: Identifier,
+        quantumMeasurement: QuantumMeasurement,
+        annotations: Sequence[Annotation] = (),
+    ):
         self.identifier = identifier
         self.quantumMeasurement = quantumMeasurement
+        self.annotations = tuple(annotations)
 
 
 class Designator(ASTNode):

qiskit/qasm3/exporter.py

@@ -1003,7 +1003,22 @@ def build_current_scope(self) -> List[ast.Statement]:
                     [self._lookup_bit(operand) for operand in instruction.qubits]
                 )
                 clbit = self._lookup_bit(instruction.clbits[0])
-                nodes = [ast.QuantumMeasurementAssignment(clbit, measurement)]
+                nodes = [
+                    ast.QuantumMeasurementAssignment(
+                        clbit,
+                        measurement,
+                    )
+                ]
+                if hasattr(instruction.operation, "annotations"):
+                    annotations = [
+                        self.build_annotation(annotation)
+                        for annotation in instruction.operation.annotations
+                    ]
+                else:
+                    annotations = []
+                nodes = [
+                    ast.QuantumMeasurementAssignment(clbit, measurement, annotations=annotations)
+                ]
             elif isinstance(instruction.operation, Reset):
                 nodes = [
                     ast.QuantumReset(self._lookup_bit(operand)) for operand in instruction.qubits

qiskit/qasm3/printer.py

@@ -273,6 +273,8 @@ def _visit_QuantumMeasurement(self, node: ast.QuantumMeasurement) -> None:
         self._visit_sequence(node.identifierList, separator=", ")
 
     def _visit_QuantumMeasurementAssignment(self, node: ast.QuantumMeasurementAssignment) -> None:
+        for annotation in reversed(node.annotations):
+            self.visit(annotation)
         self._start_line()
         self.visit(node.identifier)
         self.stream.write(" = ")