QuLab 2.11.16__py3-none-any.whl → 2.12.0__py3-none-any.whl

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.11.16"
+__version__ = "2.12.0"

qulab/visualization/plot_layout.py

@@ -54,6 +54,85 @@ def complete_layout(layout):
     return layout
 
 
+def read_xlsx_to_dict(filepath: str,
+                      key_col: int = 0,
+                      value_col: int = 4,
+                      sheet_name: str = 0) -> dict:
+    """
+    读取 .xlsx 文件，将第 key_col 列和第 value_col 列的值分别作为 key、value，返回一个字典。
+
+    :param filepath: Excel 文件路径
+    :param key_col: 用作字典键的列索引（0 表示第一列）
+    :param value_col: 用作字典值的列索引（0 表示第一列）
+    :param sheet_name: 要读取的 sheet 名称或索引，默认第一个 sheet
+    :return: { key: value, ... } 形式的字典
+    """
+    import pandas as pd
+
+    # 读取整个表格
+    # df = pd.read_excel(filepath, sheet_name=sheet_name, header=None)
+    df = pd.read_excel(filepath, sheet_name=sheet_name)
+
+    # 提取指定列，生成键值对并返回字典
+    keys = df.iloc[:, key_col]
+    values = df.iloc[:, value_col]
+    return dict(zip(keys, values))
+
+
+def load_layout_from_xlsx(qubit_info, coupler_info, pad_info):
+    """
+    从 .xlsx 文件中加载布局信息。
+
+    :param qubit_info: 量子比特信息文件路径
+    :param coupler_info: 耦合器信息文件路径
+    :param pad_info: 垫片信息文件路径
+    :return: 布局信息
+    """
+    import pandas as pd
+
+    _qubits = {}
+
+    qubit_pad = read_xlsx_to_dict(qubit_info, value_col=4)
+    coupler_pad = read_xlsx_to_dict(coupler_info, value_col=3)
+    pads = {
+        v: k
+        for k, v in qubit_pad.items()
+    } | {
+        v: k
+        for k, v in coupler_pad.items()
+    }
+    pad_info = read_xlsx_to_dict(pad_info, value_col=2)
+
+    lyt = {'qubits': {}, 'couplers': {}, 'feedlines': {}}
+
+    for pad, script in pad_info.items():
+        info = eval(script)
+        match info['style']:
+            case 'Q':
+                qubit = pads[pad]
+                lyt['qubits'][qubit] = {'pos': info['qb'], 'pad': pad}
+                _qubits[info['qb']] = qubit
+            case 'C':
+                coupler = pads[pad]
+                if 'cpl' in info:
+                    pos = tuple(info['cpl'])
+                else:
+                    pos = tuple(info['qb'])
+                lyt['couplers'][coupler] = {'qubits': pos, 'pad': pad}
+            case 'TL':
+                l = info['tl']
+                if f"T{l}" in lyt['feedlines']:
+                    lyt['feedlines'][f"T{l}"]['pads'].append(pad)
+                else:
+                    lyt['feedlines'][f"T{l}"] = {'pads': [pad]}
+
+    for coupler in lyt['couplers']:
+        qubits = lyt['couplers'][coupler]['qubits']
+        lyt['couplers'][coupler]['qubits'] = [_qubits[p] for p in qubits]
+
+    return complete_layout(lyt)
+
+
 def get_shared_coupler(layout, q1, q2):
     for c in layout['qubits'][q1]['couplers']:
         if q2 in layout['couplers'][c]['qubits']:

{qulab-2.11.16.dist-info → qulab-2.12.0.dist-info}/METADATA

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

{qulab-2.11.16.dist-info → qulab-2.12.0.dist-info}/RECORD

@@ -2,7 +2,7 @@ qulab/__init__.py,sha256=ijh3NWuSCU_KWRrPfddX4vfPcYPmvKs64vSTl9qd-SU,2178
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=B_8QdY9OMY7WU-F200v93BDJXJpQcKAHihnOXeEvv_w,3966
-qulab/version.py,sha256=z1ogNWpE3sfhj8z_vJdEtWVUcmSF3lOrnReD_efKPwM,23
+qulab/version.py,sha256=9qeMwbCMEAjH-vk_63Tak43katUwSOLexvq8N_IKi8c,22
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=ywKmwbGNBCVASx8OsgrRttaLz9ogwY38ZdKj7I-tdJ4,813
 qulab/cli/config.py,sha256=Z-lePcDMeeQmwFpdfQHqZuaXTPhWUHN5KrlSqrya73Y,6324
@@ -91,14 +91,14 @@ qulab/visualization/__init__.py,sha256=26cuHt3QIJXUb3VaMxlJx3IQTOUVJFKlYBZr7WMP5
 qulab/visualization/__main__.py,sha256=9zKK3yZFy0leU40ou6BpRC1Fsetfc1gjjFzIZYIwP6Y,1639
 qulab/visualization/_autoplot.py,sha256=2ECb2USU3_MSwo2d8qzMMO4DK9OU1bA6RYr7FCss1D0,14369
 qulab/visualization/plot_circ.py,sha256=Lxi-Nnik9MBtsNur0iyp9DTdy_IMALRsHBg2hZbPkGc,8769
-qulab/visualization/plot_layout.py,sha256=mRHinWX1Iq1qpcZ1OKrtcRoqJuv_a8d2qXrj23MeEjQ,17793
+qulab/visualization/plot_layout.py,sha256=Q8yOGYQ-oOle1W7cN4am9ABI8T-Xtk3W6BEX5bFSeWg,20464
 qulab/visualization/plot_seq.py,sha256=UWTS6p9nfX_7B8ehcYo6UnSTUCjkBsNU9jiOeW2calY,6751
 qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
 qulab/visualization/rot3d.py,sha256=CuLfG1jw1032HNNZxzC0OWtNzKsbj2e2WRYhMWDgvMQ,1321
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-qulab-2.11.16.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.11.16.dist-info/METADATA,sha256=XXJq2QfRxVsTyN5Szli0eQEIF-xEvr_4U0zsofxCMtc,3945
-qulab-2.11.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qulab-2.11.16.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.11.16.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.11.16.dist-info/RECORD,,
+qulab-2.12.0.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.12.0.dist-info/METADATA,sha256=ciOrjSwiDMrPsQizRcfSzDphr_TCmgrDFjx3Njx6qto,3944
+qulab-2.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qulab-2.12.0.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.12.0.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.12.0.dist-info/RECORD,,