|
|
import numpy as np
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
# 设置中文字体
|
|
|
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
|
|
|
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
|
|
|
|
|
|
mylist = [] # 记录时间安排，两个一组表示开头结尾
|
|
|
labels = [] # 记录调度方案，用于产生不同的颜色
|
|
|
|
|
|
# 改进颜色生成函数
|
|
|
def get_color(m_str):
|
|
|
"对于不同的工件产生不同的RGB颜色"
|
|
|
num = int(m_str)
|
|
|
# 使用 matplotlib 的 colormap 生成更均匀的颜色
|
|
|
from matplotlib.cm import get_cmap
|
|
|
cmap = get_cmap('tab20') # 选择一个 colormap，例如 'tab20'
|
|
|
return cmap(num % 20) # 确保 num 在 colormap 的范围内
|
|
|
|
|
|
try:
|
|
|
inputfile = open("output/time.txt", 'r', encoding='utf-8')
|
|
|
while True:
|
|
|
templine = inputfile.readline()
|
|
|
if templine == '':
|
|
|
break
|
|
|
templine = templine.strip('\t\n')
|
|
|
templist = templine.split('\t')
|
|
|
templist1 = []
|
|
|
for item in templist:
|
|
|
templist1.append(int(item))
|
|
|
mylist.append(templist1)
|
|
|
inputfile.close()
|
|
|
except FileNotFoundError:
|
|
|
print("Error: 文件 time.txt 不存在")
|
|
|
exit()
|
|
|
|
|
|
try:
|
|
|
inputfile = open("output/management.txt", 'r', encoding='utf-8')
|
|
|
while True:
|
|
|
templine = inputfile.readline()
|
|
|
if templine == '':
|
|
|
break
|
|
|
templine = templine.strip('\t\n')
|
|
|
templist = templine.split('\t')
|
|
|
labels.append(templist)
|
|
|
inputfile.close()
|
|
|
except FileNotFoundError:
|
|
|
print("Error: 文件 management.txt 不存在")
|
|
|
exit()
|
|
|
|
|
|
y = np.array(mylist)
|
|
|
for i in range(len(mylist)):
|
|
|
for j in range(int(len(mylist[i])/2)):
|
|
|
left = y[i][2 * j]
|
|
|
width = y[i][2 * j + 1] - left
|
|
|
if width > 0: # 确保宽度为正值
|
|
|
plt.barh(i + 1, width, left=left, color=get_color(labels[i][j]))
|
|
|
|
|
|
plt.xlabel("time")
|
|
|
plt.ylabel("machine")
|
|
|
plt.title("甘特图") # 设置甘特图的标题
|
|
|
|
|
|
# 添加图例
|
|
|
unique_labels = list(set([item for sublist in labels for item in sublist]))
|
|
|
colors = [get_color(label) for label in unique_labels]
|
|
|
plt.legend(handles=[plt.Rectangle((0, 0), 1, 1, color=get_color(label)) for label in unique_labels], labels=unique_labels)
|
|
|
|
|
|
plt.show()
|
|
|
|
|
|
mylist = []
|
|
|
try:
|
|
|
inputfile = open("data/trace.txt", 'r', encoding='utf-8')
|
|
|
while True:
|
|
|
templine = inputfile.readline()
|
|
|
if templine == '':
|
|
|
break
|
|
|
templine = templine.strip('\t\n')
|
|
|
templist = templine.split('\t')
|
|
|
for item in templist:
|
|
|
mylist.append(int(item))
|
|
|
inputfile.close()
|
|
|
except FileNotFoundError:
|
|
|
print("Error: 文件 trace.txt 不存在")
|
|
|
exit()
|
|
|
|
|
|
plt.plot(mylist, linewidth=1)
|
|
|
plt.title("目标函数变化图", fontsize=20)
|
|
|
plt.xlabel("x", fontsize=10)
|
|
|
plt.ylabel("f", fontsize=14)
|
|
|
plt.tick_params(axis='both', labelsize=10)
|
|
|
|
|
|
plt.show() |
