QuLab 2.11.16__tar.gz → 2.12.1__tar.gz

{qulab-2.11.16 → qulab-2.12.1}/PKG-INFO

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

{qulab-2.11.16 → qulab-2.12.1}/QuLab.egg-info/PKG-INFO

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

{qulab-2.11.16 → qulab-2.12.1}/qulab/cli/commands.py

@@ -1,7 +1,7 @@
 import click
 
-from ..executor.cli import (boot, create, export, get, load, maintain, delete,
-                            reproduce, run, set)
+from ..executor.cli import (boot, create, delete, export, get, load, maintain,
+                            reboot, reproduce, run, set)
 from ..monitor.__main__ import main as monitor
 from ..scan.server import server
 from ..sys.net.cli import dht
@@ -39,3 +39,4 @@ reg.add_command(delete)
 reg.add_command(load)
 reg.add_command(export)
 cli.add_command(boot)
+cli.add_command(reboot)

{qulab-2.11.16 → qulab-2.12.1}/qulab/executor/cli.py

@@ -24,7 +24,7 @@ from .utils import workflow_template
 
 
 @logger.catch(reraise=True)
-def run_script(script_path, extra_paths=None):
+def run_script(script_path, args=(), extra_paths=None):
     """Run a script in a new process, inheriting current PYTHONPATH plus any extra paths.
     
     Args:
@@ -35,7 +35,7 @@ def run_script(script_path, extra_paths=None):
     import sys
 
     # Launch the new process with the modified environment
-    proc = subprocess.Popen([sys.executable, script_path],
+    proc = subprocess.Popen([sys.executable, script_path, *args],
                             env=combined_env(extra_paths))
     proc.communicate()
 
@@ -370,6 +370,23 @@ def boot(bootstrap):
         run_script(bootstrap)
 
 
+@click.command()
+@click.option('--bootstrap',
+              '-b',
+              default=lambda: get_config_value("bootstrap", Path),
+              help='The path of the bootstrap.')
+def reboot(bootstrap):
+    """Reboot the executor.
+    
+    Reboots the executor to reset the state and start fresh.
+    
+    Args:
+        bootstrap: Path to the bootstrap script
+    """
+    if bootstrap is not None:
+        run_script(bootstrap, args=('--reboot',))
+
+
 def parse_dynamic_option_value(value):
     """解析动态参数值"""
     try:

{qulab-2.11.16 → qulab-2.12.1}/qulab/utils.py

@@ -24,33 +24,36 @@ def combined_env(extra_paths=None):
     return env
 
 
-def run_detached(script, env=None):
+def run_detached(script, env=None, cwd=None):
     """
     启动可执行文件并完全分离（优先用 tmux/screen），无需额外终端窗口
     支持 Windows、Linux 和 macOS
     """
     if env is None:
         env = combined_env()
+    if cwd is None:
+        cwd = os.getcwd()
     try:
         if sys.platform == 'win32' or not _unix_detach_with_tmux_or_screen(
-                script, env):
+                script, env, cwd):
             # 回退到带终端窗口的方案
-            run_detached_with_terminal(script, env)
+            run_detached_with_terminal(script, env, cwd)
 
     except Exception as e:
         click.echo(f"启动失败: {e}")
         sys.exit(1)
 
 
-def _windows_start(script, env):
+def _windows_start(script, env, cwd):
     """Windows 弹窗启动方案"""
     subprocess.Popen(f'start cmd /k "{script}"',
                      shell=True,
                      env=env,
+                     cwd=cwd,
                      creationflags=subprocess.CREATE_NEW_PROCESS_GROUP)
 
 
-def _unix_detach_with_tmux_or_screen(script, env):
+def _unix_detach_with_tmux_or_screen(script, env, cwd):
     """Unix 后台分离方案（无窗口）"""
     safe_path = shlex.quote(script)
     session_name = f"qulab_{int(time.time())}"
@@ -63,7 +66,7 @@ def _unix_detach_with_tmux_or_screen(script, env):
             "-d",
             "-s",
             session_name,
-            script + " ; tmux wait-for -S finished",  # 等待命令结束
+            f"cd {shlex.quote(cwd)} && {script} ; tmux wait-for -S finished",  # 等待命令结束
             ";",
             "tmux",
             "wait-for",
@@ -72,6 +75,7 @@ def _unix_detach_with_tmux_or_screen(script, env):
         subprocess.Popen(" ".join(command),
                          shell=True,
                          env=env,
+                         cwd=cwd,
                          start_new_session=True)
         click.echo(f"已启动 tmux 会话: {session_name}")
         click.echo(f"你可以使用 `tmux attach -t {session_name}` 来查看输出")
@@ -79,8 +83,11 @@ def _unix_detach_with_tmux_or_screen(script, env):
 
     # 尝试 screen
     elif _check_command_exists("screen", env):
-        command = ["screen", "-dmS", session_name, script]
-        subprocess.Popen(command, env=env, start_new_session=True)
+        command = [
+            "screen", "-dmS", session_name, "sh", "-c",
+            f"cd {shlex.quote(cwd)} && {script}"
+        ]
+        subprocess.Popen(command, env=env, cwd=cwd, start_new_session=True)
         click.echo(f"已启动 screen 会话: {session_name}")
         click.echo(f"你可以使用 `screen -r {session_name}` 来查看输出")
         return True
@@ -88,29 +95,37 @@ def _unix_detach_with_tmux_or_screen(script, env):
     return False
 
 
-def run_detached_with_terminal(script, env=None):
+def run_detached_with_terminal(script, env=None, cwd=None):
     """回退到带终端窗口的方案"""
     if env is None:
         env = combined_env()
+    if cwd is None:
+        cwd = os.getcwd()
 
     if sys.platform == 'win32':
-        _windows_start(script, env)
+        _windows_start(script, env, cwd)
     elif sys.platform == 'darwin':
         # script=shlex.quote(script)
-        script = f'tell app "Terminal" to do script "{script}"'
-        subprocess.Popen(["osascript", "-e", script],
+        terminal_script = f'tell app "Terminal" to do script "cd {shlex.quote(cwd)} && {script}"'
+        subprocess.Popen(["osascript", "-e", terminal_script],
                          env=env,
+                         cwd=cwd,
                          start_new_session=True)
     else:
         try:
+            subprocess.Popen([
+                "gnome-terminal", "--", "sh", "-c",
+                f"cd {shlex.quote(cwd)} && {script}"
+            ],
+                             env=env,
+                             cwd=cwd,
+                             start_new_session=True)
+        except FileNotFoundError:
             subprocess.Popen(
-                ["gnome-terminal", "--", "sh", "-c", script],
+                ["xterm", "-e", f"sh -c 'cd {shlex.quote(cwd)} && {script}'"],
                 env=env,
+                cwd=cwd,
                 start_new_session=True)
-        except FileNotFoundError:
-            subprocess.Popen(["xterm", "-e", script],
-                             env=env,
-                             start_new_session=True)
 
 
 def _check_command_exists(cmd, env):

qulab-2.12.1/qulab/version.py

@@ -0,0 +1 @@
+__version__ = "2.12.1"

{qulab-2.11.16 → qulab-2.12.1}/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/qulab/version.py

@@ -1 +0,0 @@
-__version__ = "2.11.16"