Function plot_benchmark_multi

benchmark/bench.py:257–407

Draw multi-subplot, multi-group, multi-metric bar charts for time/cpu_time/different payload sizes. Generate separate plots for sync/async and session/non-session combinations.

(df, filename)

def plot_benchmark_multi(df, filename):
    """
    Draw multi-subplot, multi-group, multi-metric bar charts for time/cpu_time/different payload sizes.
    Generate separate plots for sync/async and session/non-session combinations.
    """
    # Keep only necessary columns
    df = df[["name", "session", "threads", "size", "time", "cpu_time"]].copy()
    df["threads"] = df["threads"].fillna(1).astype(int)

    # Get unique session types
    existing_session_types = df["session"].unique()

    sizes = sorted(df["size"].unique(), key=lambda x: int(x.replace("k", "")))
    stat_types = ["time", "cpu_time"]

    # Separate main sessions (non-threaded) and threaded sessions
    main_sessions = [s for s in existing_session_types if not s.startswith("Threaded")]
    threaded_sessions = [s for s in existing_session_types if s.startswith("Threaded")]

    # Plot main sessions (sync and async)
    if main_sessions:
        num_sessions = len(main_sessions)
        # Allocate more height for each subplot to ensure sufficient spacing
        subplot_height = 8  # Fixed height for each subplot
        total_height = subplot_height * num_sessions + 2  # Extra 2 inches for spacing

        fig, axes = plt.subplots(
            num_sessions,
            1,
            figsize=(20, total_height),
            constrained_layout=False,  # Disable constrained_layout, use manual layout
        )

        if num_sessions == 1:
            axes = [axes]

        for idx, session in enumerate(main_sessions):
            ax = axes[idx]
            subdf = df[df["session"] == session]
            names = subdf["name"].unique()
            x = np.arange(len(names))
            width = 0.12

            max_height = 0

            for i, size in enumerate(sizes):
                for j, stat in enumerate(stat_types):
                    vals = []
                    for name in names:
                        v = subdf[(subdf["name"] == name) & (subdf["size"] == size)][
                            stat
                        ]
                        vals.append(v.values[0] if not v.empty else 0)
                    offset = (i * len(stat_types) + j) * width
                    rects = ax.bar(x + offset, vals, width, label=f"{stat} {size}")
                    ax.bar_label(rects, padding=2, fontsize=7, rotation=90)
                    if vals:
                        max_height = max(max_height, max(vals))