QuLab 2.11.5__py3-none-any.whl → 2.11.7__py3-none-any.whl

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.11.5"
+__version__ = "2.11.7"

qulab/visualization/plot_layout.py

@@ -6,6 +6,8 @@ import numpy as np
 from matplotlib import cm
 from matplotlib.colors import Normalize
 
+from .rot3d import projection, rot_round
+
 layout_example = {
     'qubits': {
         'Q0': {
@@ -118,6 +120,7 @@ def plot_range(ax,
                color=None,
                text_color='k',
                bounder_color='k',
+               rotation=0,
                lw=0.5,
                fontsize=9):
     x, y = path
@@ -138,6 +141,7 @@ def plot_range(ax,
                 text,
                 ha='center',
                 va='center',
+                rotation=rotation,
                 color=text_color,
                 fontsize=fontsize)
 
@@ -150,7 +154,7 @@ def circle_path(pos, r, n=40):
     return xx, yy
 
 
-def circle_link_path(pos1, pos2, r1, r2, width, n=20):
+def circle_link_path(pos1, pos2, r1, r2, width, n=2):
     width = min(2 * max(r1, r2), width)
 
     x1, y1 = pos1
@@ -202,54 +206,185 @@ def circle_half_directed_link_path(pos1, pos2, r1, r2, width, n=20):
                       xx2[:-1]]), np.hstack([yy2[-1], yy1, a.imag, yy2[:-1]])
 
 
-def draw(layout, ax=None, qubit_cbar=True, coupler_cbar=True, origin='upper'):
+def project(pos, camera=(0, 0, 100), rotation=None):
+    if rotation is None:
+        rotation = {
+            'axis': (0, 0, 1),
+            'angle': 0,
+            'center': (0, 0, 0),
+        }
+    x0, y0, z0 = camera
+    if len(pos) == 2:
+        x, y, z = [*pos, 0]
+    else:
+        x, y, z = pos
+    x, y, z = rot_round(x, y, z, rotation['axis'], rotation['angle'],
+                        rotation['center'])
+    x, y, d = projection(x - x0, y - y0, z, d0=z0)
+    return x + x0, y + y0
+
+
+def _draw_qubit(ax, qubit, pos):
+    path = circle_path(pos, qubit.get('radius', 0.5))
+    plot_range(ax,
+               path,
+               qubit.get('text', ''),
+               qubit.get('color', None),
+               lw=qubit.get('lw', 0.5),
+               fontsize=qubit.get('fontsize', 9),
+               rotation=qubit.get('text_rotation', 0),
+               text_color=qubit.get('text_color', 'k'),
+               bounder_color=qubit.get('bounder_color', 'k'))
+
+
+def _draw_coupler(ax, coupler, layout, q1, q2, pos1, pos2):
+    r1 = layout['qubits'][q1].get('radius', 0.5)
+    r2 = layout['qubits'][q2].get('radius', 0.5)
+    width = coupler.get('width', 0.5)
+    lw = coupler.get('lw', 0.5)
+
+    text_rotation = 180 * np.arctan2(pos2[1] - pos1[1],
+                                     pos2[0] - pos1[0]) / np.pi
+
+    path = circle_link_path(pos1, pos2, r1, r2, width)
+    plot_range(ax,
+               path,
+               coupler.get('text', ''),
+               color=coupler.get('color', None),
+               lw=0,
+               fontsize=coupler.get('fontsize', 9),
+               rotation=coupler.get('text_rotation', text_rotation),
+               text_color=coupler.get('text_color', 'k'))
+    if lw > 0:
+        x, y = circle_link_path(pos1, pos2, r1, r2, width, n=2)
+        ax.plot(x[:2], y[:2], lw=lw, color=coupler.get('bounder_color', 'k'))
+        ax.plot(x[2:], y[2:], lw=lw, color=coupler.get('bounder_color', 'k'))
+
+
+def get_range(layout):
+    coods = []
+    for q in layout['qubits']:
+        pos = layout['qubits'][q]['pos']
+        if len(pos) == 2:
+            x, y, z = [*pos, 0]
+        else:
+            x, y, z = pos
+        coods.append([x, y, z])
+    coods = np.array(coods)
+
+    center = coods.mean(axis=0)
+    radius = np.max(np.sqrt(np.sum((coods - center)**2, axis=1)))
+
+    return center, radius
+
+
+def sorted_by_distance(layout, camera, rotation):
+    qubits = []
+    camera = np.array(camera)
+    for q in layout['qubits']:
+        pos = layout['qubits'][q]['pos']
+        if len(pos) == 2:
+            x, y, z = [*pos, 0]
+        else:
+            x, y, z = pos
+        x, y, z = rot_round(x, y, z, rotation['axis'], rotation['angle'],
+                            rotation['center'])
+        d = np.sqrt(np.sum((np.array([x, y, z]) - camera)**2))
+        qubits.append((d, q))
+    qubits = sorted(qubits, key=lambda x: x[0])
+
+    plot_order = []
+    for _, q in reversed(qubits):
+        for c in layout['qubits'][q]['couplers']:
+            if ('C', c) not in plot_order:
+                plot_order.append(('C', c))
+        plot_order.append(('Q', q))
+    return plot_order
+
+
+def draw(layout,
+         ax=None,
+         qubit_cbar=True,
+         coupler_cbar=True,
+         origin='upper',
+         camera=None,
+         rotation=None):
+    """
+    Draw a layout with qubits and couplers.
+
+    Parameters
+    ----------
+    layout: dict
+        A dictionary containing the layout of the qubits and couplers.
+        The dictionary should have the following structure:
+        {
+            'qubits': {
+                'Q0': {'pos': (x, y), 'radius': r, 'text': 'Q0'},
+                ...
+            },
+            'couplers': {
+                'C0': {'qubits': ['Q0', 'Q1'], 'width': w, 'text': 'C0'},
+                ...
+            }
+        }
+    ax: matplotlib.axes.Axes
+        The axes on which to draw the layout. If None, the current axes will be used.
+    qubit_cbar: bool
+        Whether to draw a colorbar for the qubits.
+    coupler_cbar: bool
+        Whether to draw a colorbar for the couplers.
+    origin: str
+        The origin of the layout. If 'upper', the y-coordinates will be inverted.
+    camera: tuple
+        The position of the camera in 3D space. Default is (0, 0, 1).
+    rotation: dict
+        The rotation of the layout. The dictionary should have the following structure:
+        {
+            'axis': (x, y, z),
+            'angle': angle,
+            'center': (x, y, z)
+        }
+        If None, no rotation will be applied.
+    """
     if ax is None:
         ax = plt.gca()
 
-    for qubit in layout['qubits'].values():
-        pos = qubit['pos']
-        if origin == 'upper':
-            pos = pos[0], -pos[1]
-        path = circle_path(pos, qubit.get('radius', 0.5))
-        plot_range(ax,
-                   path,
-                   qubit.get('text', ''),
-                   qubit.get('color', None),
-                   lw=qubit.get('lw', 0.5),
-                   fontsize=qubit.get('fontsize', 9),
-                   text_color=qubit.get('text_color', 'k'),
-                   bounder_color=qubit.get('bounder_color', 'k'))
-
-    for coupler in layout['couplers'].values():
-        q1, q2 = coupler['qubits']
-        pos1 = layout['qubits'][q1]['pos']
-        pos2 = layout['qubits'][q2]['pos']
-        if origin == 'upper':
-            pos1 = pos1[0], -pos1[1]
-            pos2 = pos2[0], -pos2[1]
-        r1 = layout['qubits'][q1].get('radius', 0.5)
-        r2 = layout['qubits'][q2].get('radius', 0.5)
-        width = coupler.get('width', 0.5)
-        lw = coupler.get('lw', 0.5)
-
-        path = circle_link_path(pos1, pos2, r1, r2, width)
-        plot_range(ax,
-                   path,
-                   coupler.get('text', ''),
-                   color=coupler.get('color', None),
-                   lw=0,
-                   fontsize=coupler.get('fontsize', 9),
-                   text_color=coupler.get('text_color', 'k'))
-        if lw > 0:
-            x, y = circle_link_path(pos1, pos2, r1, r2, width, n=2)
-            ax.plot(x[:2],
-                    y[:2],
-                    lw=lw,
-                    color=coupler.get('bounder_color', 'k'))
-            ax.plot(x[2:],
-                    y[2:],
-                    lw=lw,
-                    color=coupler.get('bounder_color', 'k'))
+    center, radius = get_range(layout)
+
+    if rotation is None:
+        rotation = {
+            'axis': (0, 0, 1),
+            'angle': 0,
+        }
+    if 'center' not in rotation:
+        rotation['center'] = center
+    if camera is None:
+        camera = (center[0], center[1], center[2] + radius * 5)
+
+    plot_order = sorted_by_distance(layout, camera, rotation)
+
+    if origin == 'upper':
+        camera = [camera[0], -camera[1], camera[2]]
+    for kind, name in plot_order:
+        if kind == 'Q':
+            pos = layout['qubits'][name]['pos']
+            if origin == 'upper':
+                pos = [pos[0], -pos[1], *pos[2:]]
+            pos = project(pos, camera, rotation)
+            _draw_qubit(ax, layout['qubits'][name], pos)
+        elif kind == 'C':
+            coupler = layout['couplers'][name]
+            q1, q2 = coupler['qubits']
+            pos1 = layout['qubits'][q1]['pos']
+            pos2 = layout['qubits'][q2]['pos']
+            if origin == 'upper':
+                pos1 = [pos1[0], -pos1[1], *pos1[2:]]
+                pos2 = [pos2[0], -pos2[1], *pos2[2:]]
+            pos1 = project(pos1, camera, rotation)
+            pos2 = project(pos2, camera, rotation)
+            _draw_coupler(ax, coupler, layout, q1, q2, pos1, pos2)
+        else:
+            pass
 
     ax.axis('equal')
     ax.set_axis_off()

qulab/visualization/rot3d.py

@@ -14,10 +14,36 @@ def _rot(theta, v):
 
 
 def rot_round(x, y, z, v=[0, 0, 1], theta=0, c=[0, 0, 0]):
+    """
+    Rotate a point (x, y, z) around a vector v by an angle theta
+
+    Parameters
+    ----------
+    x, y, z: coordinates of the point to be rotated
+    v: vector to rotate around
+    theta: angle to rotate by (in radians)
+    c: center of rotation
+
+    Returns
+    -------
+    ret: rotated coordinates
+    """
     ret = (np.array([x, y, z]).T - np.array(c)) @ _rot(theta, v) + np.array(c)
     return ret.T
 
 
-def projection(x, y, z, y0=1):
+def projection(x, y, z, d0=1):
+    """
+    Project a point (x, y, z) onto a plane at z = 0
+
+    Parameters
+    ----------
+    x, y, z: coordinates of the point to be projected
+    d0: distance from the camera to the plane
+
+    Returns
+    -------
+    ret: projected coordinates
+    """
     d = np.sqrt(x**2 + y**2 + z**2)
-    return x * y0 / y, z * y0 / y, d
+    return x * d0 / (d0 - z), y * d0 / (d0 - z), d

{qulab-2.11.5.dist-info → qulab-2.11.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.11.5
+Version: 2.11.7
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>

{qulab-2.11.5.dist-info → qulab-2.11.7.dist-info}/RECORD

@@ -2,7 +2,7 @@ qulab/__init__.py,sha256=hmf6R3jiM5NtJY1mSptogYnH5DMTR2dTzlXykqLxQzg,2027
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=w5oSw9Ypux6l0oB-MzlQvT40iX_rzQ5wXcwzr7CIVf8,4107
-qulab/version.py,sha256=fNipWoN2EmgidscGxW-j_-B8q7t8BEt85UV6DPo_yRA,22
+qulab/version.py,sha256=H2A8gHyN0Jg2EL7bDXRWVKNMgM1isfEU_bBVahUV62U,22
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=ywKmwbGNBCVASx8OsgrRttaLz9ogwY38ZdKj7I-tdJ4,813
 qulab/cli/config.py,sha256=tZPSBLbf2x_Brb2UBuA1bsni8nC8WOPPGyWIi8m7j1I,5459
@@ -91,14 +91,14 @@ qulab/visualization/__init__.py,sha256=26cuHt3QIJXUb3VaMxlJx3IQTOUVJFKlYBZr7WMP5
 qulab/visualization/__main__.py,sha256=9zKK3yZFy0leU40ou6BpRC1Fsetfc1gjjFzIZYIwP6Y,1639
 qulab/visualization/_autoplot.py,sha256=31B7pn1pK19abpQDGYBU9a_27cDL87LBpx9vKqlcYAo,14165
 qulab/visualization/plot_circ.py,sha256=qZS3auNG4qIyUvC-ijTce17xJwGBvpRvPUqPEx64KUY,8759
-qulab/visualization/plot_layout.py,sha256=mfdKJUyjTNB3YyD6GcqM38GEwiae4g6Us9HJyAwen6Q,13535
+qulab/visualization/plot_layout.py,sha256=n9yEmCOl7RJw2l0hXXRVsnLfDOMlK7A52kQZG_Y_NgM,17630
 qulab/visualization/plot_seq.py,sha256=UWTS6p9nfX_7B8ehcYo6UnSTUCjkBsNU9jiOeW2calY,6751
 qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
-qulab/visualization/rot3d.py,sha256=lMrEJlRLwYe6NMBlGkKYpp_V9CTipOAuDy6QW_cQK00,734
+qulab/visualization/rot3d.py,sha256=CuLfG1jw1032HNNZxzC0OWtNzKsbj2e2WRYhMWDgvMQ,1321
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-qulab-2.11.5.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.11.5.dist-info/METADATA,sha256=HUf9_g4vh9bEiGfmG0uO8WcL10bnkUvw5DfveaLro4w,3896
-qulab-2.11.5.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-qulab-2.11.5.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.11.5.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.11.5.dist-info/RECORD,,
+qulab-2.11.7.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.11.7.dist-info/METADATA,sha256=XSG-tpZgI0O7k5DAGckiZSgu9QjgIvRU4h1cVO4oSqc,3896
+qulab-2.11.7.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+qulab-2.11.7.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.11.7.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.11.7.dist-info/RECORD,,

{qulab-2.11.5.dist-info → qulab-2.11.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any