param parser

Author: plutoniumm

setup.py

@@ -5,7 +5,7 @@
 
 setup(
   name='abrax',
-  version='0.2.0',
+  version='0.2.1',
   description='A Quantum Circuit DSL',
   long_description=long_description,
   long_description_content_type='text/markdown',

src/abrax/parser.py

@@ -1,6 +1,6 @@
-import re
+from re import compile
 
-no_gate=re.compile("-{1,}")
+no_gate = compile("-{1,}")
 
 def parse_circuit(string):
   lines = map(lambda e: e.strip(), string.strip().split("\n"))
@@ -25,6 +25,14 @@ def parse_circuit(string):
 
   return list(map(list, zip(*by_rows)))
 
+def parse_param(p):
+  # if float or int continue
+  if isinstance(p, (float, int)):
+    return p
+  # if not, it's a string, if includes . it's a float
+  p = p.strip()
+  return float(p) if "." in p else int(p)
+
 def resolve_circuit(circuit, qc, config):
   for layerNo, layer in enumerate(circuit):
     if all([e.lower() == "-" for e in layer]):
@@ -40,10 +48,14 @@ def resolve_circuit(circuit, qc, config):
         gate_name = gate[:gate.index("(")]
         param = gate[gate.index("(") + 1:gate.index(")")]
         op = getattr(qc, gate_name)
+
         if "," in param:
-          param = list(map(float, param.split(","))) + [wireNo]
+          param = param \
+            .split(",") \
+            .map(lambda e: parse_param(e))
+          param = param + [wireNo]
         else:
-          param = [float(param), wireNo]
+          param = [parse_param(param), wireNo]
       else:
         param = [wireNo]
         op = getattr(qc, gate)

test/index.py

@@ -1,15 +1,16 @@
 from qiskit import QuantumCircuit
 from abrax import A
+from numpy import pi
 
 qc = QuantumCircuit(5)
-circuit = A("""
+circuit = A(f"""
   // I AM JESUS
   // SPACES ARE SIGNIFICANT, DONOT add them inside a gate
-  - H CX(4) RX(10)
-  - H -    RY(55)
-  - H -    CX(4)
-  - H X    H
-  - H -    -
+  - H CX(4)  RX({pi})
+  - H RZ(45) RY({5.5})
+  - H ---    CX(4)
+  - H X      H
+  - H ---    -------
   """, qc
 )