def setData(self, data):
self.series.clear()
for i, j in zip(data, self.brushes):
barset = QBarSet("test")
barset.append(i)
barset.setBrush(j)
barset.setPen(self.pen)
self.series.append(barset)
def createBar(self):
min_num, max_num = 0, 100
linked_bag_list = []
try:
df = self.linked['Beam Diff'].dropna()
linked_bag_list = df.values.tolist()
min_num = int(min(linked_bag_list))
if min_num > 0: # check if greater than 0, set to 0
min_num = 0
max_num = int(max(linked_bag_list))
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = [0] * (max_num + 1) # choose the largest num as length of count
for num in linked_bag_list:
count[int(num)] += 1 # update every number's count
max_count = max(count)
setBar = QBarSet('Beam Difference Occurrence')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
chart.setTheme(QChart.ChartThemeBlueIcy)
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Linked Bins Histogram')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setTitleText("Attenuation Window")
axisX.setRange(min_num, max_num+20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setTitleText("Frequency")
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().hide()
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
return chartView
def build(self):
self.setWindowTitle("Histogram - " + self.name)
self.setMinimumSize(400, 300)
# Close when ctrl+w is triggered
QShortcut(QKeySequence("Ctrl+W"), self, self.close)
# 2D array -> 1D array
img = self.image.ravel()
# Process histogram
histogram, bin_edges = np.histogram(img, bins='auto')
bar_series = QBarSeries()
bar_ = QBarSet("")
# Append values
for val in histogram:
bar_.append(val)
pen = bar_.pen()
pen.setColor(Qt.black)
pen.setWidth(0.1)
bar_.setPen(pen)
# Append bar to the bar series
bar_series.append(bar_)
# Chart object
chart = QChart()
chart.addSeries(bar_series)
# Active animation
chart.setAnimationOptions(QChart.SeriesAnimations)
# Do not show title
chart.legend().setVisible(False)
# Draw Axes, with [min, max]
# Y axis
h_max = histogram.max()
axis_y = QValueAxis()
axis_y.setRange(0, h_max)
# X axis
be_max = bin_edges.max()
be_min = bin_edges.min()
axis_x = QValueAxis()
axis_x.setRange(be_min, be_max)
# Add axis to chart + rendering
chart.addAxis(axis_x, Qt.AlignBottom)
chart.addAxis(axis_y, Qt.AlignLeft)
view = QChartView(chart)
view.setRenderHint(QPainter.Antialiasing)
self.setCentralWidget(view)
def beforeDelay(self):
print("in before delay bar")
min_num, max_num = 0, 100
max_count = 0
total_stopped_time = []
try:
total_stopped_time = self.tm.total_stopped_time
max_num = max(total_stopped_time)
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = total_stopped_time
count = [0] * (int(max_num) + 1
) # choose the largest num as length of count
for num in total_stopped_time:
count[int(num)] += 1 # update every number's count
max_count = max(count)
print(len(total_stopped_time), max_count)
setBar = QBarSet('stop time')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Histogram (before)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num + 20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count + 20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def afterDelay(self):
# print("in after delay bar")
min_num, max_num = 0, 100
max_count = 0
total_stopped_after_delay = []
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
try:
total_stopped_after_delay = self.tm.total_stopped_after_delay
max_num = max(total_stopped_after_delay)
count = [0] * (int(max_num) + 1) # choose the largest num as length of count
for num in total_stopped_after_delay:
count[int(num)] += 1 # update every number's count
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('Stop Time Occurrence')
setBar.append(count)
brush = QBrush(QColor(0xA6E22E)) # Green
pen = QPen(QColor(0xA6E22E)) # Green
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Occurrence (after)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num+20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def beforeDelayDistribution(self):
min_num, max_num = 0, 100
max_count = 0
total_stopped_time = []
count = [0]
try:
total_stopped_time = self.tm.total_stopped_time
max_num = len(total_stopped_time) # change from max() to len() 2/12 11:11
count = total_stopped_time
max_count = max(count)
except (AttributeError, ValueError):
self.statusbar.showMessage('Data not ready')
setBar = QBarSet('stop time')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Stop time Distribution (before)')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count+20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def bars_stacked(name, table_df, x_bottom_cols, y_left_cols, y_right_cols,
legend_labels, tick_count):
"""
柱形堆叠图
:param name: 图表名称
:param table_df: 用于画图的pandas DataFrame对象
:param x_bottom_cols: 下轴列索引列表
:param y_left_cols: 左轴列索引列表
:param y_right_cols: 右轴列索引列表
:param legend_labels: 图例名称标签列表
:param tick_count: 横轴刻度标签数
:return: QChart实例
"""
""" 过滤轴 """
for y_left_col in y_left_cols:
if is_datetime64_any_dtype(table_df[y_left_col]): # 如果是时间轴
y_left_cols.remove(y_left_col)
for y_right_col in y_right_cols:
if is_datetime64_any_dtype(table_df[y_right_col]): # 如果是时间轴
y_right_cols.remove(y_right_col)
# 将x轴转为字符串
x_bottom = x_bottom_cols[0] # x轴列
if is_datetime64_any_dtype(table_df[x_bottom]): # 如果x轴是时间轴
table_df[x_bottom] = table_df[x_bottom].apply(
lambda x: x.strftime('%Y-%m-%d'))
else: # x轴非时间轴
table_df[x_bottom] = table_df[x_bottom].apply(lambda x: str(x))
font = QFont() # 轴文字风格
font.setPointSize(7) # 轴标签文字大小
chart = QChart() # 图表实例
chart.layout().setContentsMargins(0, 0, 0, 0) # chart的外边距
chart.setMargins(QMargins(15, 5, 15, 0)) # chart内边距
chart.setTitle(name) # chart名称
""" x轴 """
axis_x_bottom = QCategoryAxis(
chart, labelsPosition=QCategoryAxis.AxisLabelsPositionOnValue)
axis_x_bottom.setLabelsAngle(-90) # 逆时针旋转90度
axis_x_bottom.setLabelsFont(font) # 设置字体样式
axis_x_bottom.setGridLineVisible(False) # 竖向连接线不可见
# chart.addAxis(axis_x_bottom, Qt.AlignBottom) # 加入坐标x轴
has_x_labels = False # 收集x轴刻度的开关
chart.x_labels = list() # 绑定chart一个x轴的刻度列表
""" 左Y轴 """
axis_y_left = QValueAxis()
axis_y_left.setLabelsFont(font)
axis_y_left.setLabelFormat('%i')
chart.addAxis(axis_y_left, Qt.AlignLeft) # 图表加入左轴
""" 右Y轴 """
axis_y_right = QValueAxis()
axis_y_right.setLabelsFont(font)
axis_y_right.setLabelFormat('%.2f')
chart.addAxis(axis_y_right, Qt.AlignRight)
# 记录各轴的最值
x_bottom_min, x_bottom_max = 0, table_df.shape[0]
y_left_min, y_left_max = 0, 0
y_right_min, y_right_max = 0, 0
# 柱形图
left_bars = QBarSeries()
""" 根据左轴画柱形 """
for y_left_col in y_left_cols: # 根据左轴画折线
# 计算做轴的最值用于设置范围
table_df[y_left_col] = table_df[y_left_col].apply(
covert_float) # y轴列转为浮点数值
# 获取最值
y_min, y_max = table_df[y_left_col].min(), table_df[y_left_col].max()
if y_min < y_left_min:
y_left_min = y_min
if y_max > y_left_max:
y_left_max = y_max
# 取得图表的源数据作柱形
left_bar_data = table_df.iloc[:, [x_bottom, y_left_col]]
bar = QBarSet(legend_labels[y_left_col])
bar.setPen(QPen(Qt.transparent)) # 设置画笔轮廓线透明(数据量大会出现空白遮住柱形)
for index, point_item in enumerate(left_bar_data.values.tolist()):
bar.append(point_item[1])
if not has_x_labels:
chart.x_labels.append(point_item[0])
has_x_labels = True # 关闭添加轴标签
left_bars.append(bar) # 柱形加入系列
left_bars.attachAxis(axis_y_left)
# 左轴的范围
axis_y_left.setRange(y_left_min, y_left_max)
""" 根据右轴画柱形 """
right_bars = QBarSeries()
for y_right_col in y_right_cols: # 根据右轴画柱形
# 计算做轴的最值用于设置范围
table_df[y_right_col] = table_df[y_right_col].apply(
covert_float) # y轴列转为浮点数值
# 获取最值
y_min, y_max = table_df[y_right_col].min(), table_df[y_right_col].max()
if y_min < y_right_min:
y_right_min = y_min
if y_max > y_right_max:
y_right_max = y_max
# 取得图线的源数据作折线图
right_bar_data = table_df.iloc[:, [x_bottom, y_right_col]]
bar = QBarSet(legend_labels[y_right_col])
bar.setPen(QPen(Qt.transparent))
for position_index, point_item in enumerate(
right_bar_data.values.tolist()):
bar.append(point_item[1]) # 取出源数据后一条线就2列数据
right_bars.append(bar)
right_bars.attachAxis(axis_x_bottom)
right_bars.attachAxis(axis_y_right)
# 右轴范围
axis_y_right.setRange(y_right_min, y_right_max)
chart.addSeries(left_bars) # 左轴的柱形图加入图表
# print(right_bars.count())
if right_bars.count() != 0: # 为空时加入会导致空位
chart.addSeries(right_bars)
chart.setAxisX(axis_x_bottom, left_bars) # 关联设置x轴
# 横轴标签设置
x_bottom_interval = int(x_bottom_max / (tick_count - 1))
if x_bottom_interval == 0:
for i in range(0, x_bottom_max):
axis_x_bottom.append(chart.x_labels[i], i)
else:
for i in range(0, x_bottom_max, x_bottom_interval):
axis_x_bottom.append(chart.x_labels[i], i)
chart.legend().setAlignment(Qt.AlignBottom)
return chart
def createBar(self, MainWindow):
dp = DataProc()
# print(dp.linked)
linked_list = dp.linked['Beam Diff'].values.tolist()
# print(linked_list)
count = [0] * 200
# print(count)
for item in linked_list:
if item <= 160:
count[int(item)] += 1
set0 = QBarSet('Beam Difference')
# set0.setColor(QColor("blue"))
# add pen width to bar
pen = QPen(QColor(0xfdb157))
pen.setWidth(5)
set0.setPen(pen)
# set1 = QBarSet('X1')
set0.append(count)
# set0.append([random.randint(0, 10) for i in range(6)])
series = QBarSeries()
series.append(set0)
# series.barWidth()
# series.append(set1)
# Getting the Chart
chart = QChart() # Getting the Chart
# add font to title
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.addSeries(series) # Add data
chart.setTitle('Linked Bags Histogram') # Set title
chart.setAnimationOptions(QChart.SeriesAnimations) # Set animation
# months = ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun') # Set custom axis
times = [
'0-20', '20-40', '40-60', '60-80', '80-100', '100-120', '120-140',
'140-160'
]
# Set x-axis
axisX = QBarCategoryAxis()
# axisX.append(months)
axisX.append(times)
# Set y-axis
# axisY = QValueAxis()
# axisY.setRange(0, 200)
# Alignment for x,y axis
chart.addAxis(axisX, Qt.AlignBottom)
# chart.addAxis(axisY, Qt.AlignLeft)
# Enable legend and set custom axis alignment to bottom
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
# Create ChartView
chartView = QChartView(chart)
# add rendering
# chartView.setRenderHint(QPainter.Antialiasing)
# Set view at the center of MainWindw (show view)
MainWindow.setCentralWidget(chartView)
def createUnlinkedBar(self):
max_count = 0
unlinked_bag_list = []
try:
df = self.unlinked['Beam Diff'].dropna()
unlinked_bag_list = df.values.tolist()
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = [0] * 4
for num in unlinked_bag_list:
if -1000 <= num and num <= -51:
count[0] += 1
elif -50 <= num and num <= -1:
count[1] += 1
elif 151 <= num and num <= 200:
count[2] += 1
elif 201 <= num:
count[3] += 1
# print(count)
max_count = max(count)
setBar = QBarSet('Beam Difference Occurrence')
setBar.append(count)
series = QBarSeries()
series.append(setBar)
brush = QBrush(QColor(0xfdb157))
pen = QPen(QColor(0xfdb157))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Unlinked Bags Summary')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
labels = ['Not useful(-50 to -1000)', 'Pushed by operator(-1 to -50)', 'Slipping on belt(151 to 200)', 'Not useful 201+']
axisX = QBarCategoryAxis()
axisX.append(labels)
# chart.setAxisX(axisX, series)
chart.addAxis(axisX, Qt.AlignBottom)
# chart.ChartAreas[0].AxisX.LabelAutoFitStyle = LabelAutoFitStyle.WrodWrap
series.attachAxis(axisX)
axisY = QValueAxis()
axisY.setRange(0, max_count+1)
# chart.setAxisY(axisY, series)
chart.addAxis(axisY, Qt.AlignLeft)
series.attachAxis(axisY)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView
def createBar(self):
print("in create bar")
min_num, max_num = 0, 100
max_count = 0
linked_bag_list = []
try:
df = self.linked['Beam Diff'].dropna()
# df = self.dp.linked['Beam Diff'].dropna()
# print(df)
linked_bag_list = df.values.tolist()
min_num = int(min(linked_bag_list))
# print(min_num)
if min_num > 0: # check if greater than 0, set to 0
min_num = 0
max_num = int(max(linked_bag_list))
# print(max_count)
except AttributeError:
self.statusbar.showMessage('Data not ready')
count = linked_bag_list
count = [0] * (max_num + 1
) # choose the largest num as length of count
for num in linked_bag_list:
count[int(num)] += 1 # update every number's count
max_count = max(count)
setBar = QBarSet('Beam Difference')
setBar.append(count)
brush = QBrush(QColor(0x57B1FD))
pen = QPen(QColor(0x57B1FD))
pen.setWidth(2)
setBar.setPen(pen)
setBar.setBrush(brush)
series = QBarSeries()
series.append(setBar)
chart = QChart()
font = QFont()
font.setPixelSize(18)
chart.setTitleFont(font)
chart.setTitle('Linked Bags Histogram')
chart.addSeries(series)
chart.setAnimationOptions(QChart.SeriesAnimations)
axisX = QValueAxis()
axisX.setRange(min_num, max_num + 20)
chart.setAxisX(axisX, series)
axisY = QValueAxis()
axisY.setRange(0, max_count + 20)
chart.setAxisY(axisY, series)
chart.legend().setVisible(True)
chart.legend().setAlignment(Qt.AlignBottom)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
# MainWindow.setCentralWidget(chartView)
return chartView