qubitclient 0.1.4__py3-none-any.whl

qubitclient/draw/spectrum2dpltplotter.py

@@ -0,0 +1,107 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from .pltplotter import QuantumDataPltPlotter
+
+class Spectrum2DDataPltPlotter(QuantumDataPltPlotter):
+    def __init__(self):
+        super().__init__("spectrum2d")
+
+    def plot_result_npy(self, **kwargs):
+        results = kwargs.get('results')
+        data_ndarray = kwargs.get('data_ndarray')
+
+        nums = len(results)*2
+        row = (nums // 2) + 1 if nums % 2 != 0 else nums // 2
+        col = min(nums, 2)
+
+        fig = plt.figure(figsize=(10 * col, 4 * row))
+        data_dict = data_ndarray.item() if isinstance(data_ndarray, np.ndarray) else data_ndarray
+        data_dict = data_dict['image']
+        dict_list = []
+        q_list = data_dict.keys()
+
+        for idx, q_name in enumerate(q_list):
+            npz_dict = {}
+            image_q = data_dict[q_name]
+            data = image_q[0]
+            if data.ndim != 2:
+                raise ValueError("数据格式无效，data不是二维数组")
+            data = np.array(data)
+            data = np.abs(data)
+
+            npz_dict['bias'] = image_q[1]
+            npz_dict['frequency'] = image_q[2]
+            npz_dict['iq_avg'] = data
+            npz_dict['name'] = q_name
+            dict_list.append(npz_dict)
+
+        for index in range(nums):
+            ax = fig.add_subplot(row, col, index + 1)
+            result = results[index//2]
+
+            points_list = []
+            for i in range(len(result["linepoints_list"])):
+                points_list.append(result["linepoints_list"][i])
+
+            plt.pcolormesh(dict_list[index//2]["bias"], dict_list[index//2]["frequency"], dict_list[index//2]["iq_avg"],
+                           shading='auto', cmap='viridis')
+            plt.colorbar(label='IQ Average')
+            colors = plt.cm.rainbow(np.linspace(0, 1, len(result["linepoints_list"])))
+
+            if (index % 2 != 0):
+                for i in range(len(points_list)):
+                    reflection_points = points_list[i]
+                    reflection_points = np.array(reflection_points)
+                    xy_x = reflection_points[:, 0]
+                    xy_y = reflection_points[:, 1]
+
+                    plt.scatter(xy_x, xy_y, color=colors[i],
+                                label=f'XY Points{i}-conf:{round(result["confidence_list"][i], 2)}', s=5,
+                                alpha=0.1)
+            file_name = dict_list[index//2]["name"]
+            plt.title(f"File: {file_name}")
+            plt.xlabel("Bias")
+            plt.ylabel("Frequency (GHz)")
+            plt.legend()
+        fig.tight_layout()
+        return fig
+
+    def plot_result_npz(self, **kwargs):
+        results = kwargs.get('results')
+        dict_list = kwargs.get('dict_list')
+        file_names = kwargs.get('file_names')
+
+        nums = len(results)*2
+        row = (nums // 2) + 1 if nums % 2 != 0 else nums // 2
+        col = min(nums, 2)
+
+        fig = plt.figure(figsize=(10 * col, 4 * row))
+
+        for index in range(nums):
+            ax = fig.add_subplot(row, col, index + 1)
+            result = results[index//2]
+            file_name = file_names[index//2]
+
+            points_list = []
+            for i in range(len(result["linepoints_list"])):
+                points_list.append(result["linepoints_list"][i])
+
+            plt.pcolormesh(dict_list[index//2]["bias"], dict_list[index//2]["frequency"], dict_list[index//2]["iq_avg"],
+                           shading='auto', cmap='viridis')
+            plt.colorbar(label='IQ Average')
+            colors = plt.cm.rainbow(np.linspace(0, 1, len(result["linepoints_list"])))
+            if(index%2!=0):
+                for i in range(len(points_list)):
+                    reflection_points = points_list[i]
+                    reflection_points = np.array(reflection_points)
+                    xy_x = reflection_points[:, 0]
+                    xy_y = reflection_points[:, 1]
+                    plt.scatter(xy_x, xy_y, color=colors[i],
+                                label=f'XY Points{i}-conf:{round(result["confidence_list"][i], 2)}', s=5,
+                                alpha=0.1)
+            plt.title(f"File: {file_name}")
+            plt.xlabel("Bias")
+            plt.ylabel("Frequency (GHz)")
+            plt.legend()
+        fig.tight_layout()
+        return fig

qubitclient/draw/spectrum2dplyplotter.py

@@ -0,0 +1,244 @@
+from .plyplotter import QuantumDataPlyPlotter
+import numpy as np
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+
+class Spectrum2DDataPlyPlotter(QuantumDataPlyPlotter):
+
+    def __init__(self):
+        super().__init__("spectrum2d")
+
+
+    def plot_result_npy(self, **kwargs):
+
+        results = kwargs.get('results')
+        data_ndarray = kwargs.get('data_ndarray')
+
+        # 参数验证
+        if results is None:
+            raise ValueError("缺少必需的 'results' 参数")
+        if data_ndarray is None:
+            raise ValueError("缺少必需的 'data_ndarray' 参数")
+        nums = len(results)
+        rows = (nums*2 // 2) + 1 if nums*2 % 2 != 0 else nums*2 // 2
+        cols = min(nums*2, 2)
+
+        # 处理数据字典
+        data_dict = data_ndarray.item() if isinstance(data_ndarray, np.ndarray) else data_ndarray
+        data_dict = data_dict['image']
+        data_dict = data_dict.item() if isinstance(data_dict, np.ndarray) else data_dict
+        dict_list = []
+        q_list = data_dict.keys()
+
+        # 准备数据列表
+        for idx, q_name in enumerate(q_list):
+            npz_dict = {}
+            image_q = data_dict[q_name]
+            data = image_q[0]
+            if data.ndim != 2:
+                raise ValueError("数据格式无效，data不是二维数组")
+            data = np.array(data)
+            data = np.abs(data)
+
+            npz_dict['bias'] = image_q[1]
+            npz_dict['frequency'] = image_q[2]
+            npz_dict['iq_avg'] = data
+            npz_dict['name'] = q_name
+            dict_list.append(npz_dict)
+        subplot_titles = []
+        for i in range(nums):
+            subplot_titles.append(f"File: {dict_list[i]['name']}")
+            subplot_titles.append(f"File: {dict_list[i]['name']}")
+
+        # 创建子图
+        fig = make_subplots(
+            rows=rows, cols=cols,
+            subplot_titles=subplot_titles,
+            vertical_spacing=0.01,
+            horizontal_spacing=0.1,
+            x_title="Bias",
+            y_title="Frequency (GHz)"
+        )
+
+        # 遍历每个结果绘制子图
+        for index in range(nums*2):
+            row = (index // cols) + 1
+            col = (index % cols) + 1
+
+            result = results[index//2]
+            data = dict_list[index//2]
+
+            # 准备点数据
+            points_list = []
+            for i in range(len(result["linepoints_list"])):
+                points_list.append(np.array(result["linepoints_list"][i]))
+
+            # 添加热力图
+            heatmap = go.Heatmap(
+                z=data["iq_avg"],
+                x=data["bias"],
+                y=data["frequency"],
+                colorscale="Viridis",
+                colorbar=dict(
+                    title="IQ Average",
+                    thickness=10,
+                    len=0.7,
+                    yanchor="middle",
+                    y=0.5
+                ),
+                showscale=(index == 0)  # 只在第一个子图显示颜色条
+            )
+            fig.add_trace(heatmap, row=row, col=col)
+            # 添加散点
+            colors = np.linspace(0, 1, len(points_list))
+            for i, points in enumerate(points_list):
+                if len(points) == 0:
+                    continue
+                xy_x = points[:, 0]
+                xy_y = points[:, 1]
+                scatter = go.Scatter(
+                    x=xy_x,
+                    y=xy_y,
+                    mode="markers",
+                    marker=dict(
+                        color=colors[i],
+                        colorscale="Rainbow",
+                        size=5,
+                        opacity=0.1,
+                        showscale=False
+                    ),
+                    name=f'XY Points{i}-conf:{round(result["confidence_list"][i], 2)}',
+                    legendgroup=f"group{index//2}",
+                    showlegend=(row == 1 and col == 1)  # 只在第一个子图显示图例
+                )
+                if(index%2!=0):
+                    fig.add_trace(scatter, row=row, col=col)
+
+        # 更新布局
+        fig.update_layout(
+            height=500 * rows,
+            width=900 * cols,
+            margin=dict(r=60, t=60, b=60, l=60),
+            legend=dict(
+                font=dict(family="Courier", size=12, color="black"),
+                borderwidth=1
+            )
+        )
+
+        # 更新坐标轴设置
+        fig.update_xaxes(
+            title_text="Bias",
+            title_font=dict(size=10),  # 缩小字体
+            title_standoff=8  # 增加标题与坐标轴的距离（单位：像素）
+        )
+        fig.update_yaxes(
+            title_text="Frequency (GHz)",
+            title_font=dict(size=10),
+            title_standoff=8
+        )
+        return fig
+        # 保存图片
+    def plot_result_npz(self, **kwargs):
+
+        results = kwargs.get('results')
+        dict_list = kwargs.get('dict_list')
+        file_names = kwargs.get('file_names')
+
+        nums = len(results)*2
+        rows = (nums // 2) + 1 if nums % 2 != 0 else nums // 2
+        cols = min(nums, 2)
+        subplot_titles = []
+        for name in file_names:
+            subplot_titles.append(f"File: {name}")
+            subplot_titles.append(f"File: {name}")
+
+        # 创建子图
+        fig = make_subplots(
+            rows=rows, cols=cols,
+            subplot_titles=subplot_titles,
+            vertical_spacing=0.015,
+            horizontal_spacing=0.1,
+            x_title="Bias",
+            y_title="Frequency (GHz)"
+        )
+
+        # 遍历每个结果绘制子图
+        for index in range(nums):
+            row = (index // cols) + 1
+            col = (index % cols) + 1
+
+            result = results[index//2]
+            data = dict_list[index//2]
+
+            # 准备点数据
+            points_list = []
+            for i in range(len(result["linepoints_list"])):
+                points_list.append(np.array(result["linepoints_list"][i]))
+
+            # 添加热力图
+            heatmap = go.Heatmap(
+                z=data["iq_avg"],
+                x=data["bias"],
+                y=data["frequency"],
+                colorscale="Viridis",
+                colorbar=dict(
+                    title="IQ Average",
+                    thickness=10,
+                    len=0.7,
+                    yanchor="middle",
+                    y=0.5
+                ),
+                showscale=(index == 0)  # 只在第一个子图显示颜色条
+            )
+            fig.add_trace(heatmap, row=row, col=col)
+
+            # 添加散点
+            colors = np.linspace(0, 1, len(points_list))
+            for i, points in enumerate(points_list):
+                if len(points) == 0:
+                    continue
+                xy_x = points[:, 0]
+                xy_y = points[:, 1]
+                scatter = go.Scatter(
+                    x=xy_x,
+                    y=xy_y,
+                    mode="markers",
+                    marker=dict(
+                        color=colors[i],
+                        colorscale="Rainbow",
+                        size=5,
+                        opacity=0.1,
+                        showscale=False
+                    ),
+                    name=f'XY Points{i}-conf:{round(result["confidence_list"][i], 2)}',
+                    legendgroup=f"group{index//2}",
+                    showlegend=(row == 1 and col == 1)  # 只在第一个子图显示图例
+                )
+                if (index % 2 != 0):
+                    fig.add_trace(scatter, row=row, col=col)
+
+        # 更新布局
+        fig.update_layout(
+            height=500 * rows,
+            width=900 * cols,
+            margin=dict(r=60, t=60, b=60, l=60),
+            legend=dict(
+                font=dict(family="Courier", size=12, color="black"),
+                borderwidth=1
+            )
+        )
+
+        # 更新坐标轴设置
+        fig.update_xaxes(
+            title_text="Bias",
+            title_font=dict(size=10),  # 缩小字体
+            title_standoff=8  # 增加标题与坐标轴的距离（单位：像素）
+        )
+        fig.update_yaxes(
+            title_text="Frequency (GHz)",
+            title_font=dict(size=10),
+            title_standoff=8
+        )
+
+        return fig
+        # 保存图片

qubitclient/draw/spectrum2dscopepltplotter.py

@@ -0,0 +1,72 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from .pltplotter import QuantumDataPltPlotter
+
+class Spectrum2DScopeDataPltPlotter(QuantumDataPltPlotter):
+    def __init__(self):
+        super().__init__("spectrum2dscope")
+
+    def plot_result_npy(self, **kwargs):
+        results = kwargs.get('result')
+        dict_param = kwargs.get('dict_param')
+
+        data = dict_param.item()
+        image = data["image"]
+        q_list = image.keys()
+        volt_list = []
+        freq_list = []
+        s_list = []
+        q_name_list=[]
+        for idx, q_name in enumerate(q_list):
+            image_q = image[q_name]
+
+            volt = image_q[1]
+            freq = image_q[2]
+            s = np.abs(image_q[0])
+            volt_list.append(volt)
+            freq_list.append(freq)
+            s_list.append(s)
+            q_name_list.append(q_name)
+        coslines_list= results['params']
+        cosconfs_list= results['confs']
+        coscompress_list= results['coscompress_list']
+        lines_list= results['lines_list']
+        lineconfs_list= results['lineconfs_list']
+
+        nums = len(volt_list)*2
+        row = (nums // 2) + 1 if nums % 2 != 0 else nums // 2
+        col = min(nums, 2)
+
+        fig = plt.figure(figsize=(5 * col, 4 * row))
+
+        for ii in range(nums):
+            ax = fig.add_subplot(row, col, ii + 1)
+
+            volt = volt_list[ii//2]
+            freq = freq_list[ii//2]
+            s = s_list[ii//2]
+            coslines = coslines_list[ii//2]
+            cosconfs = cosconfs_list[ii//2]
+            coscompress = coscompress_list[ii//2]
+            lines = lines_list[ii//2]
+            lineconfs = lineconfs_list[ii//2]
+            plt.pcolormesh(volt, freq, s, cmap='viridis')
+            if (ii % 2 != 0):
+                if (lines):
+                    for j, line in enumerate(lines):
+                        final_x_line = [item[0] for item in line]
+                        final_line_pred = [item[1] for item in line]
+                        plt.plot(final_x_line, final_line_pred, c='r')
+                        plt.text(volt[len(volt) // 2], freq[len(freq) // 2], f'confidence: {lineconfs[j]:.2f}', c='red',
+                                 size=15)
+
+                if (coslines):
+                    for j, cosline in enumerate(coslines):
+                        final_x_cos = [item[0] for item in cosline]
+                        final_cos_pred = [item[1] for item in cosline]
+                        plt.plot(final_x_cos, final_cos_pred, c='r')
+                        plt.text(volt[len(volt) // 2], freq[len(freq) // 2],
+                                 f'confidence: {cosconfs[j]:.2f}\ncompress: {coscompress[j]:.2f}', c='red', size=15)
+            ax.set_title(f"{q_name_list[ii//2]}")
+        fig.tight_layout()
+        return fig  # ✅ 返回 Figure 对象

qubitclient/draw/spectrum2dscopeplyplotter.py

@@ -0,0 +1,195 @@
+from .plyplotter import QuantumDataPlyPlotter
+import numpy as np
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+
+class Spectrum2DScopeDataPlyPlotter(QuantumDataPlyPlotter):
+
+    def __init__(self):
+        super().__init__("spectrum2dscope")
+
+
+    def plot_result_npy(self, **kwargs):
+
+        results = kwargs.get('result')
+        dict_param = kwargs.get('dict_param')
+
+        data = dict_param.item()
+        image = data["image"]
+        q_list = image.keys()
+
+        # 数据提取
+        volt_list = []
+        freq_list = []
+        s_list = []
+        q_name_list = []
+
+        for idx, q_name in enumerate(q_list):
+            image_q = image[q_name]
+            volt = image_q[1]
+            freq = image_q[2]
+            s = np.abs(image_q[0])
+
+            volt_list.append(volt)
+            freq_list.append(freq)
+            s_list.append(s)
+            q_name_list.append(q_name)
+
+        # 结果数据
+        coslines_list = results['params']
+        cosconfs_list = results['confs']
+        coscompress_list = results['coscompress_list']
+        lines_list = results['lines_list']
+        lineconfs_list = results['lineconfs_list']
+
+        # 计算子图布局
+        nums = len(volt_list) * 2
+        rows = (nums // 2) + 1 if nums % 2 != 0 else nums // 2
+        cols = min(nums, 2)
+
+        # 计算安全的垂直间距
+        max_vertical_spacing = 1 / (rows - 1) if rows > 1 else 0.1
+        safe_vertical_spacing = min(0.05, max_vertical_spacing - 0.01)
+
+        # 创建子图布局
+        fig = make_subplots(
+            rows=rows,
+            cols=cols,
+            vertical_spacing=0.01,
+            horizontal_spacing=0.1,
+            subplot_titles=[f"{q_name_list[ii // 2]}_Heatmap" if ii % 2 == 0
+                            else f"{q_name_list[ii // 2]}_WithCurves" for ii in range(nums)]
+        )
+
+        # 遍历所有子图位置
+        for ii in range(nums):
+            row_pos = (ii // cols) + 1
+            col_pos = (ii % cols) + 1
+
+            volt = volt_list[ii // 2]
+            freq = freq_list[ii // 2]
+            s = s_list[ii // 2]
+            coslines = coslines_list[ii // 2]
+            cosconfs = cosconfs_list[ii // 2]
+            coscompress = coscompress_list[ii // 2]
+            lines = lines_list[ii // 2]
+            lineconfs = lineconfs_list[ii // 2]
+
+            # 热力图数据
+            heatmap_trace = go.Heatmap(
+                x=volt,
+                y=freq,
+                z=s,
+                colorscale='Viridis',
+                showscale=True,
+                colorbar=dict(title='Intensity'),
+                hovertemplate=(
+                        'Volt: %{x}<br>' +
+                        'Freq: %{y}<br>' +
+                        'Intensity: %{z}<extra></extra>'
+                )
+            )
+
+            fig.add_trace(heatmap_trace, row=row_pos, col=col_pos)
+
+            # 在奇数编号的子图中添加曲线和线条
+            if (ii % 2 != 0):
+                # 添加直线
+                if lines:
+                    for j, line in enumerate(lines):
+                        if line:
+                            final_x_line = [item[0] for item in line]
+                            final_line_pred = [item[1] for item in line]
+
+                            line_trace = go.Scatter(
+                                x=final_x_line,
+                                y=final_line_pred,
+                                mode='lines',
+                                line=dict(color='red', width=3),
+                                name=f'Line {j + 1}',
+                                showlegend=False,
+                                hovertemplate=(
+                                        'Volt: %{x}<br>' +
+                                        'Freq: %{y}<br>' +
+                                        f'Confidence: {lineconfs[j]:.2f}<extra></extra>'
+                                )
+                            )
+                            fig.add_trace(line_trace, row=row_pos, col=col_pos)
+
+                            # 添加置信度文本
+                            mid_idx = len(volt) // 2
+                            if mid_idx < len(volt):
+                                fig.add_annotation(
+                                    x=volt[mid_idx],
+                                    y=freq[mid_idx],
+                                    text=f"conf: {lineconfs[j]:.2f}",
+                                    showarrow=False,
+                                    font=dict(color='red', size=12),
+                                    bgcolor='rgba(255,255,255,0.8)',
+                                    row=row_pos,
+                                    col=col_pos
+                                )
+
+                # 添加余弦曲线
+                if coslines:
+                    for j, cosline in enumerate(coslines):
+                        if cosline:
+                            final_x_cos = [item[0] for item in cosline]
+                            final_cos_pred = [item[1] for item in cosline]
+
+                            cos_trace = go.Scatter(
+                                x=final_x_cos,
+                                y=final_cos_pred,
+                                mode='lines',
+                                line=dict(color='red', width=3),
+                                name=f'Cosine {j + 1}',
+                                showlegend=False,
+                                hovertemplate=(
+                                        'Volt: %{x}<br>' +
+                                        'Freq: %{y}<br>' +
+                                        f'Confidence: {cosconfs[j]:.2f}<br>' +
+                                        f'Compress: {coscompress[j]:.2f}<extra></extra>'
+                                )
+                            )
+                            fig.add_trace(cos_trace, row=row_pos, col=col_pos)
+
+                            # 添加置信度和压缩比文本
+                            mid_idx = len(volt) // 2
+                            if mid_idx < len(volt):
+                                fig.add_annotation(
+                                    x=volt[mid_idx],
+                                    y=freq[mid_idx],
+                                    text=f"conf: {cosconfs[j]:.2f}<br>compress: {coscompress[j]:.2f}",
+                                    showarrow=False,
+                                    font=dict(color='red', size=12),
+                                    bgcolor='rgba(255,255,255,0.8)',
+                                    row=row_pos,
+                                    col=col_pos
+                                )
+
+
+        # 更新布局
+        fig.update_layout(
+            height=500 * rows,
+            width=900 * cols,
+            margin=dict(r=60, t=60, b=60, l=60),
+            legend=dict(
+                font=dict(family="Courier", size=12, color="black"),
+                borderwidth=1
+            )
+        )
+
+        # 更新坐标轴设置
+        fig.update_xaxes(
+            title_text="Bias",
+            title_font=dict(size=10),  # 缩小字体
+            title_standoff=8  # 增加标题与坐标轴的距离（单位：像素）
+        )
+        fig.update_yaxes(
+            title_text="Frequency (GHz)",
+            title_font=dict(size=10),
+            title_standoff=8
+        )
+
+        return fig
+        # 保存图片