ONE仿真平台的reports画图

ONE仿真平台的reports画图

1. 代码画的是 MessageStatsReport.txt 报告下的**[“delivery_prob”,“overhead_ratio”,“latency_avg”,“hopcount_avg”,]**这几个字段

2. 配置文件输出报告的格式要求：

   • Scenario.name = %%Group.router%%%%Group.bufferSize%%

     （如果是在不同的缓存状态下，如果是别的变量，在配置文件进行修改）

   • 这样的格式是为了方便代码的正则表达式提取

     （提取出以XXX开头，以MessageStatsReport.txt结尾的文件，并将文件名中的数值变量作为绘图时的横坐标值）

代码部分

import os
import re
import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from scipy.interpolate import make_interp_spline
from sympy.physics.control.control_plots import matplotlib

# 设置默认字体
plt.rcParams['font.family'] = 'Times New Roman'
plt.rcParams['font.size'] = 4
matplotlib.rc("font", family='Microsoft YaHei')
# 设置线条宽度
plt.rcParams['lines.linewidth'] = 0.5
# 设置坐标轴标签和标题的字体大小
plt.rcParams['axes.labelsize'] = 7
plt.rcParams['axes.titlesize'] = 7
# 设置图例字体大小
plt.rcParams['legend.fontsize'] = 3

def get_data(file_pattern,laber_name):
    # 存储数字和delivery_prob值的列表
    x_values = []
    y_values = []
    # 列出目标文件夹中的所有文件
    for filename in os.listdir(folder_path):
        # 如果文件名符合正则表达式规则，则打开文件并查找delivery_prob的值
        if re.match(file_pattern, filename) and os.path.isfile(os.path.join(folder_path, filename)):
            with open(os.path.join(folder_path, filename), 'r') as file:
                for line in file:
                    # 假设每行的格式为“delivery_prob: 0.12345”
                    match = re.search(laber_name+':\s+(\d+\.\d+)', line)
                    if match:
                        delivery_prob = float(match.group(1))
                        # 提取文件名中的数字，并将其与delivery_prob值相关联
                        digit_match = re.search('\d+', filename)
                        if digit_match:
                            x_value = int(digit_match.group(0))
                            x_values.append(x_value)
                            y_values.append(delivery_prob)
    return x_values,y_values


def draw(x,Y,laberY,laber,Pic_i):
    plt.subplot(2, 2, Pic_i + 1)
    color=["b","g","r","c","m","y","k"]
    Oshape=[".","v","+","x","1"]
    c = 0
    # 定义每条线的x和y坐标
    print(Y)
    result = []
    for i in range(len(Y)): #0 1
        for j in range(len(x)):
            result.append([x[j], Y[i][j]])
        result = sorted(result, key=lambda x: x[0])

        x1 = [item[0] for item in result]
        y1 = [item[1] for item in result]
        plt.plot(x1, y1, color[c], label=laber[c], marker=Oshape[c], markersize=2)
        c=c+1
        # 添加图例
        plt.legend()
        plt.xlabel('BufferSize  M')
        plt.ylabel(laberY)

        print(result)
        result = []

    # plt.show()

if __name__ == '__main__':
    # 文件夹路径
    folder_path = 'C:/Users/the-one-1.6.0/report1'
    # 正则表达式匹配规则，用于筛选文件名
    laber_name=["delivery_prob","overhead_ratio","latency_avg","hopcount_avg",]      #绘制的数据——修改部分2
    file_name=[r'^Epidemic.*MessageStatsReport.txt$',                # 要修改的部分1——添加文件名——改正则表达式
              r'^ProphetRouter.*MessageStatsReport.txt$',
              r'^MDCE.*MessageStatsReport.txt$',
               r'^FirstContactRouter.*MessageStatsReport.txt$',
               r'^SprayAndWaitRouter.*MessageStatsReport.txt$'
              ]

    laber = []
    for name in file_name:
        pattern = re.compile(r'^(?:\^)?(.*?)\.\*')
        match = pattern.search(name)
        if match:
            laber.append(match.group(1))
    # print(laber)   #['Epidemic', 'ProphetRouter', 'MDCE', 'FirstContactRouter', 'SprayAndWaitRouter']

    for i in range(4):
        print(i)
        Y = []
        labername = laber_name[i]
        for filename in file_name:
            x, y = get_data(filename, labername)
            Y.append(y)  # [[x],[[y1],[y2],[y3]]]
        draw(x,Y,labername,laber,i)
        # laber_name 为纵坐标标签，laber为 图例说明,i 为第几个图

# 调节每个子图之间的间隔
plt.subplots_adjust(top=0.97,
                    bottom=0.13,
                    left=0.075,
                    right=0.985,
                    hspace=0.285,
                    wspace=0.45)

plt.savefig('result.pdf')

# 显示图像，并启用滚动
plt.show(block=False)
plt.tight_layout()

代码说明：将图片写入到pdf，方便查看。最终结果在result.pdf 内

要修改的部分：

laber_name=["delivery_prob","overhead_ratio","latency_avg","hopcount_avg",]      #绘制的数据——修改部分2
    file_name=[r'^Epidemic.*MessageStatsReport.txt$',                # 要修改的部分1——添加文件名——改正则表达式
              r'^ProphetRouter.*MessageStatsReport.txt$',
              r'^MDCE.*MessageStatsReport.txt$',
               r'^FirstContactRouter.*MessageStatsReport.txt$',
               r'^SprayAndWaitRouter.*MessageStatsReport.txt$'
              ]

结果样例：

ps：输出的报告格式样例：

备注：

• 代码的结果图合在一起展示，是为了方便查看。可以将for i in range(4): 循环去掉 得到单图

• **for i in range(4)**这部分的4写死了，是四种图

• 图例可能有的会被隐藏，问题不大

• 不建议拿Matplotlib画的图放到论文，还是用MATLAB画吧。这个只是方便日常快速查看结果。

• 代码测试不多，有错误自己改。