Improve QASM parser to handle multi-character register names

pyzx/circuit/qasmparser.py

@@ -140,7 +140,18 @@ def extract_command_parts(self, c: str) -> Tuple[str,List[Fraction],List[str]]:
     def parse_command(self, c: str, registers: Dict[str,Tuple[int,int]]) -> List[Gate]:
         gates: List[Gate] = []
         name, phases, args = self.extract_command_parts(c)
-        if name in ("barrier","creg","measure", "id"): return gates
+        if name in ("barrier","creg", "id"): return gates
+        if name == "measure":
+            target, result_bit = args[0].split(' -> ')
+            # Extract the register name and index separately for both target and result
+            target_reg, target_idx = target.split('[')
+            result_reg, result_idx = result_bit.split('[')
+            # Remove the trailing ']' and convert to int
+            target_qbit = int(target_idx[:-1])
+            result_bit = int(result_idx[:-1])
+            gate = Measurement(target_qbit, result_bit)
+            gates.append(gate)
+            return gates
         if name in ("opaque", "if"):
             raise TypeError("Unsupported operation {}".format(c))
         if name == "qreg":
@@ -154,9 +165,12 @@ def parse_command(self, c: str, registers: Dict[str,Tuple[int,int]]) -> List[Gat
         dim = 1
         for a in args:
             if "[" in a:
+                # Split at the first '[' to handle multi-character register names
                 regname, valp = a.split("[",1)
+                # Remove the trailing ']' before converting to int
                 val = int(valp[:-1])
-                if regname not in registers: raise TypeError("Invalid register {}".format(regname))
+                if regname not in registers: 
+                    raise TypeError("Invalid register {}".format(regname))
                 qubit_values.append([registers[regname][0]+val])
             else:
                 if is_range: