311 lines
16 KiB
Python
311 lines
16 KiB
Python
from PyQt6.QtWidgets import QWidget, QSizePolicy
|
||
from PyQt6.QtGui import QPainter, QColor, QPen, QPainterPath, QLinearGradient
|
||
from PyQt6.QtCore import Qt, QRectF, QPointF
|
||
import math
|
||
|
||
# 统一底色与圆角
|
||
BACKGROUND_COLOR = QColor(245, 246, 250)
|
||
PANEL_BORDER_COLOR = QColor(225, 228, 236)
|
||
CORNER_RADIUS = 10
|
||
|
||
def map_ratio_to_color(r: float) -> QColor:
|
||
hue_stops = [(0.0,0.58),(0.25,0.64),(0.50,0.70),(0.75,0.78),(1.0,0.85)]
|
||
def lerp(a,b,t): return a + (b-a)*t
|
||
if r <= 0: h = hue_stops[0][1]
|
||
elif r >= 1: h = hue_stops[-1][1]
|
||
else:
|
||
for i in range(1,len(hue_stops)):
|
||
if r <= hue_stops[i][0]:
|
||
lpos, lh = hue_stops[i-1]; rpos, rh = hue_stops[i]
|
||
local_t = 0 if rpos==lpos else (r-lpos)/(rpos-lpos)
|
||
h = lerp(lh,rh,local_t); break
|
||
sat = 0.35 + 0.20 * r
|
||
valb = 0.75 + 0.15 * (1 - abs(r-0.5)*2)
|
||
return QColor.fromHsvF(h, sat, valb)
|
||
|
||
class BarChartWidget(QWidget):
|
||
def __init__(self, data, title='', unit='分钟', labels=None, value_format=None, label_rotation: int = 0, parent=None):
|
||
super().__init__(parent)
|
||
self.data = data or []
|
||
self.title = title
|
||
self.unit = unit
|
||
self.labels = labels or [str(i) for i in range(len(self.data))]
|
||
self.value_format = value_format
|
||
self.label_rotation = label_rotation # 0=不旋转, 45=倾斜
|
||
self.setMinimumHeight(180)
|
||
self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding)
|
||
|
||
def sizeHint(self):
|
||
from PyQt6.QtCore import QSize
|
||
return QSize(500, 260)
|
||
|
||
def setData(self, data, labels=None):
|
||
self.data = data or []
|
||
self.labels = labels or [str(i) for i in range(len(self.data))]
|
||
self.update()
|
||
|
||
def paintEvent(self, event):
|
||
p = QPainter(self)
|
||
p.setRenderHint(QPainter.RenderHint.Antialiasing)
|
||
rect = self.rect()
|
||
p.setPen(QPen(PANEL_BORDER_COLOR,1))
|
||
p.setBrush(BACKGROUND_COLOR)
|
||
# 背景圆角
|
||
p.drawRoundedRect(QRectF(rect).adjusted(0.5,0.5,-0.5,-0.5), CORNER_RADIUS, CORNER_RADIUS)
|
||
if not self.data:
|
||
p.setPen(QColor('#666666'))
|
||
p.drawText(rect, Qt.AlignmentFlag.AlignCenter, '暂无数据')
|
||
p.end(); return
|
||
max_val = max(self.data) or 1
|
||
title_left, title_top = 12, 6
|
||
margin_top = 28
|
||
margin_bottom = 28 if self.label_rotation == 0 else 50
|
||
margin_left, margin_right = 40, 10
|
||
title_height = 18
|
||
w, h = rect.width(), rect.height()
|
||
chart_rect = QRectF(margin_left, margin_top, w - margin_left - margin_right, h - margin_top - margin_bottom)
|
||
bar_count = len(self.data)
|
||
bar_space = chart_rect.width() / bar_count
|
||
bar_width = bar_space * 0.55
|
||
# 标题
|
||
p.setPen(Qt.GlobalColor.black)
|
||
font = p.font(); font.setPointSize(10); font.setBold(True); p.setFont(font)
|
||
p.drawText(QRectF(title_left, title_top, chart_rect.width(), title_height), Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter, self.title)
|
||
# 网格
|
||
font.setPointSize(8); font.setBold(False); p.setFont(font)
|
||
pen_grid = QPen(QColor('#dddddd')); pen_grid.setStyle(Qt.PenStyle.DotLine)
|
||
p.setPen(pen_grid)
|
||
grid_lines = 4
|
||
for i in range(grid_lines+1):
|
||
y = chart_rect.top() + chart_rect.height() * (1 - i / grid_lines)
|
||
p.drawLine(QPointF(margin_left, y), QPointF(margin_left + chart_rect.width(), y))
|
||
val = max_val * i / grid_lines
|
||
p.setPen(QColor('#555555'))
|
||
label = f'{val:.0f}' if max_val >= 10 else f'{val:.1f}'
|
||
p.drawText(QRectF(2, y-8, margin_left-4, 14), Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter, label)
|
||
p.setPen(pen_grid)
|
||
# 单位
|
||
p.setPen(QColor('#333333'))
|
||
p.drawText(2, margin_top-4, margin_left-4, 14, Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignBottom, self.unit)
|
||
# 柱子
|
||
for idx, v in enumerate(self.data):
|
||
x_center = chart_rect.left() + idx * bar_space + bar_space/2
|
||
bar_h = 0 if max_val == 0 else chart_rect.height() * (v / max_val)
|
||
bar_rect = QRectF(x_center - bar_width/2, chart_rect.bottom()-bar_h, bar_width, bar_h)
|
||
ratio = 0.0 if max_val == 0 else max(0.0, min(1.0, v / max_val))
|
||
color = map_ratio_to_color(ratio)
|
||
p.setPen(Qt.PenStyle.NoPen); p.setBrush(color)
|
||
p.drawRoundedRect(bar_rect, 2, 2)
|
||
if v > 0:
|
||
p.setPen(QColor('#222222'))
|
||
sf = p.font(); sf.setPointSize(7); p.setFont(sf)
|
||
val_str = self.value_format(v) if self.value_format else (f'{v:.0f}' if v >= 10 else f'{v:.1f}')
|
||
p.drawText(QRectF(bar_rect.left()-2, bar_rect.top()-14, bar_rect.width()+4, 12), Qt.AlignmentFlag.AlignCenter, val_str)
|
||
# x 标签
|
||
p.setPen(QColor('#333333'))
|
||
lf = p.font(); lf.setPointSize(8); p.setFont(lf)
|
||
for idx, text in enumerate(self.labels):
|
||
x_center = chart_rect.left() + idx * bar_space + bar_space/2
|
||
if self.label_rotation:
|
||
p.save()
|
||
p.translate(x_center, chart_rect.bottom()+4)
|
||
p.rotate(-45)
|
||
# 旋转后使用固定宽度避免重叠 (40px)
|
||
p.drawText(QRectF(-20, 0, 40, 14), Qt.AlignmentFlag.AlignCenter, text)
|
||
p.restore()
|
||
else:
|
||
p.drawText(QRectF(x_center - bar_space/2, chart_rect.bottom()+2, bar_space, 14), Qt.AlignmentFlag.AlignHCenter | Qt.AlignmentFlag.AlignTop, text)
|
||
p.end()
|
||
|
||
class BubbleMetricsWidget(QWidget):
|
||
def __init__(self, metrics, parent=None):
|
||
super().__init__(parent)
|
||
self.metrics = metrics
|
||
self.setMinimumHeight(240)
|
||
self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding)
|
||
|
||
def sizeHint(self):
|
||
from PyQt6.QtCore import QSize
|
||
return QSize(500, 300)
|
||
|
||
def setMetrics(self, metrics):
|
||
self.metrics = metrics; self.update()
|
||
|
||
def paintEvent(self, event):
|
||
p = QPainter(self)
|
||
p.setRenderHint(QPainter.RenderHint.Antialiasing)
|
||
rect = self.rect()
|
||
p.setPen(QPen(PANEL_BORDER_COLOR,1)); p.setBrush(BACKGROUND_COLOR)
|
||
p.drawRoundedRect(QRectF(rect).adjusted(0.5,0.5,-0.5,-0.5), CORNER_RADIUS, CORNER_RADIUS)
|
||
title_left, title_top, title_height = 12, 6, 20
|
||
margin = 2
|
||
content_rect = QRectF(rect.left()+margin, rect.top()+title_height+margin, rect.width()-2*margin, rect.height()-title_height-2*margin)
|
||
if not self.metrics:
|
||
p.drawText(rect, Qt.AlignmentFlag.AlignCenter, '暂无数据'); p.end(); return
|
||
# 归一化：统一把不同单位数值映射到“分钟等价”尺度，以便半径比较
|
||
# 规则：小时(h) -> v*60；分钟(m) -> v；书本数量(book) -> v*60 (一书折算 1 小时，避免过小)
|
||
minute_values = []
|
||
for (lbl, v, u, c) in self.metrics:
|
||
if u == 'h':
|
||
minute_values.append(v * 60)
|
||
elif u == 'm':
|
||
minute_values.append(v)
|
||
elif u == 'book': # 新增：书籍数量指标
|
||
minute_values.append(v * 60) # 折算：1 本 ≈ 60 分钟，使气泡视觉权重合理
|
||
else: # 未知单位直接使用原值
|
||
minute_values.append(v)
|
||
max_minutes = max(minute_values) if minute_values else 1
|
||
radii_norm = [0.3 + 0.7 * math.sqrt((mv/max_minutes) if max_minutes>0 else 0) for mv in minute_values]
|
||
# 默认布局：4 指标与 5 指标（包含“已读”）使用不同的预设坐标，尽量避免重叠
|
||
labels = [m[0] for m in self.metrics]
|
||
if '已读' in labels and len(labels) >= 5:
|
||
default_pos = {
|
||
'全年': (0.18, 0.55),
|
||
'月均': (0.38, 0.20),
|
||
'近7天': (0.82, 0.55),
|
||
'日均': (0.38, 0.85),
|
||
'已读': (0.60, 0.55)
|
||
}
|
||
else:
|
||
default_pos = { '全年':(0.20,0.50), '月均':(0.50,0.18), '近7天':(0.80,0.50), '日均':(0.50,0.82), '已读':(0.50,0.50) }
|
||
if any(lbl not in default_pos for (lbl,*_) in self.metrics):
|
||
step = 1.0/(len(self.metrics)+1)
|
||
default_pos = {m[0]:(step*(i+1),0.5) for i,m in enumerate(self.metrics)}
|
||
W,H = content_rect.width(), content_rect.height(); base_len = min(W,H)
|
||
centers = [(content_rect.left()+default_pos.get(lbl,(0.5,0.5))[0]*W,
|
||
content_rect.top()+default_pos.get(lbl,(0.5,0.5))[1]*H) for (lbl,*_) in self.metrics]
|
||
constraints=[]
|
||
for (cx,cy),r in zip(centers,radii_norm):
|
||
dist=min(cx-content_rect.left(), content_rect.right()-cx, cy-content_rect.top(), content_rect.bottom()-cy)
|
||
if r>0: constraints.append(dist/(r*base_len))
|
||
for i in range(len(centers)):
|
||
for j in range(i+1,len(centers)):
|
||
dx = centers[i][0]-centers[j][0]; dy = centers[i][1]-centers[j][1]
|
||
d = math.hypot(dx,dy)
|
||
if d>0:
|
||
gap=6; constraints.append((d-gap)/((radii_norm[i]+radii_norm[j])*base_len))
|
||
S = max(0.05, min(constraints) if constraints else 0.5)
|
||
for (label,val,unit,color), r_norm, (cx,cy) in zip(self.metrics, radii_norm, centers):
|
||
R = r_norm * S * base_len
|
||
circle_rect = QRectF(cx-R, cy-R, 2*R, 2*R)
|
||
p.setPen(QPen(QColor('#ffffff'),2)); p.setBrush(QColor(color)); p.drawEllipse(circle_rect)
|
||
p.setPen(QColor('#ffffff'))
|
||
f = p.font(); f.setBold(True); f.setPointSize(max(9,int(R/5))); p.setFont(f)
|
||
if unit == 'h':
|
||
txt_core = f"{val:.0f} 小时" if val >= 10 else f"{val:.1f} 小时"
|
||
elif unit == 'm':
|
||
txt_core = f"{val:.0f} 分钟"
|
||
elif unit == 'book':
|
||
txt_core = f"{val:.0f} 本书"
|
||
else:
|
||
txt_core = f"{val:.0f}"
|
||
p.drawText(circle_rect, Qt.AlignmentFlag.AlignCenter, f"{txt_core}\n{label}")
|
||
# 标题（为空保留对齐占位）
|
||
p.setPen(QColor('#222222'))
|
||
f = p.font(); f.setBold(True); f.setPointSize(10); p.setFont(f)
|
||
p.drawText(QRectF(title_left, title_top, rect.width()-title_left-8, title_height), Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter, '')
|
||
p.end()
|
||
|
||
class ScatterChartWidget(QWidget):
|
||
def __init__(self, data, title='', unit='小时', labels=None, parent=None):
|
||
super().__init__(parent)
|
||
self.data = data or []
|
||
self.labels = labels or [str(i) for i in range(len(self.data))]
|
||
self.title = title; self.unit = unit
|
||
self.setMinimumHeight(180)
|
||
self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding)
|
||
|
||
def sizeHint(self):
|
||
from PyQt6.QtCore import QSize
|
||
return QSize(500, 260)
|
||
|
||
def setData(self, data, labels=None):
|
||
self.data = data or []
|
||
self.labels = labels or [str(i) for i in range(len(self.data))]
|
||
self.update()
|
||
|
||
def paintEvent(self, event):
|
||
p = QPainter(self)
|
||
p.setRenderHint(QPainter.RenderHint.Antialiasing)
|
||
rect = self.rect()
|
||
p.setPen(QPen(PANEL_BORDER_COLOR,1)); p.setBrush(BACKGROUND_COLOR)
|
||
p.drawRoundedRect(QRectF(rect).adjusted(0.5,0.5,-0.5,-0.5), CORNER_RADIUS, CORNER_RADIUS)
|
||
if not self.data:
|
||
p.drawText(rect, Qt.AlignmentFlag.AlignCenter, '暂无数据'); p.end(); return
|
||
title_left,title_top = 12,6
|
||
margin_top,margin_bottom = 28,32
|
||
margin_left,margin_right = 40,10
|
||
title_height = 18
|
||
w,h = rect.width(), rect.height()
|
||
chart_rect = QRectF(margin_left, margin_top, w - margin_left - margin_right, h - margin_top - margin_bottom)
|
||
# 标题
|
||
p.setPen(Qt.GlobalColor.black)
|
||
font = p.font(); font.setPointSize(10); font.setBold(True); p.setFont(font)
|
||
p.drawText(QRectF(title_left, title_top, chart_rect.width(), title_height), Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter, self.title)
|
||
# 网格
|
||
max_val = max(self.data) if self.data else 1
|
||
if max_val <= 0: max_val = 1
|
||
grid_lines=4
|
||
font.setPointSize(8); font.setBold(False); p.setFont(font)
|
||
pen_grid = QPen(QColor('#dddddd')); pen_grid.setStyle(Qt.PenStyle.DotLine)
|
||
for i in range(grid_lines+1):
|
||
y = chart_rect.top() + chart_rect.height()*(1 - i/grid_lines)
|
||
p.setPen(pen_grid); p.drawLine(QPointF(chart_rect.left(), y), QPointF(chart_rect.right(), y))
|
||
val = max_val * i / grid_lines
|
||
p.setPen(QColor('#555555'))
|
||
label = f'{val:.0f}' if max_val>=10 else f'{val:.1f}'
|
||
p.drawText(QRectF(2, y-8, margin_left-4, 14), Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter, label)
|
||
# 单位
|
||
p.setPen(QColor('#333333'))
|
||
p.drawText(2, margin_top-4, margin_left-4, 14, Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignBottom, self.unit)
|
||
# 点
|
||
count = len(self.data)
|
||
x_step = chart_rect.width()/(count-1) if count>1 else chart_rect.width()/2
|
||
points=[]
|
||
for i,v in enumerate(self.data):
|
||
x = chart_rect.left()+i*x_step
|
||
y = chart_rect.bottom() - (v/max_val)*chart_rect.height()
|
||
ratio = max(0.0,min(1.0,v/max_val))
|
||
color = map_ratio_to_color(ratio)
|
||
points.append((x,y,v,color,ratio))
|
||
# 均值线
|
||
avg_val = sum(self.data)/len(self.data) if self.data else 0
|
||
avg_y = chart_rect.bottom() - (avg_val/max_val)*chart_rect.height()
|
||
pen_avg = QPen(QColor('#aaaadd')); pen_avg.setStyle(Qt.PenStyle.DashLine); pen_avg.setWidth(1)
|
||
p.setPen(pen_avg); p.drawLine(QPointF(chart_rect.left(), avg_y), QPointF(chart_rect.right(), avg_y))
|
||
p.setPen(QColor('#666688'))
|
||
f2 = p.font(); f2.setPointSize(7); p.setFont(f2)
|
||
p.drawText(QRectF(chart_rect.right()-60, avg_y-10, 58, 12), Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter, f'均值 {avg_val:.1f}')
|
||
# 折线+面积
|
||
if len(points) >=2:
|
||
pen_line = QPen(QColor('#8890c0')); pen_line.setWidth(1); p.setPen(pen_line)
|
||
for i in range(len(points)-1):
|
||
p.drawLine(QPointF(points[i][0], points[i][1]), QPointF(points[i+1][0], points[i+1][1]))
|
||
grad = QLinearGradient(chart_rect.left(), chart_rect.top(), chart_rect.left(), chart_rect.bottom())
|
||
grad.setColorAt(0.0, QColor(200,205,235,90)); grad.setColorAt(1.0, QColor(200,205,235,10))
|
||
path = QPainterPath(QPointF(points[0][0], chart_rect.bottom()))
|
||
for (x,y, *_ ) in points: path.lineTo(x,y)
|
||
path.lineTo(points[-1][0], chart_rect.bottom()); path.closeSubpath(); p.fillPath(path, grad)
|
||
# 极值
|
||
max_point = max(points, key=lambda t:t[2]) if points else None
|
||
min_point = min(points, key=lambda t:t[2]) if points else None
|
||
for (x,y,v,color,ratio) in points:
|
||
r = 5 + 7*ratio
|
||
pen = QPen(QColor('#ffffff'),1.2)
|
||
if max_point and (x,y,v)==max_point[:3]: pen = QPen(QColor('#ff8888'),1.5)
|
||
elif min_point and (x,y,v)==min_point[:3]: pen = QPen(QColor('#66aac6'),1.5)
|
||
p.setPen(pen); p.setBrush(color); p.drawEllipse(QPointF(x,y), r, r)
|
||
label_y_top = y - r - 10; place_above = label_y_top > chart_rect.top()+4
|
||
txt_rect = QRectF(x-24, (label_y_top if place_above else y + r + 2), 48, 12)
|
||
p.setPen(QColor('#222222'))
|
||
sf = p.font(); sf.setPointSize(7); p.setFont(sf)
|
||
val_str = f'{v:.1f}' if v < 10 else f'{v:.0f}'
|
||
p.drawText(txt_rect, Qt.AlignmentFlag.AlignCenter, val_str)
|
||
# x 标签
|
||
p.setPen(QColor('#333333'))
|
||
lf = p.font(); lf.setPointSize(8); p.setFont(lf)
|
||
for i, lab in enumerate(self.labels):
|
||
x = chart_rect.left() + (i * x_step if count>1 else chart_rect.width()/2)
|
||
p.drawText(QRectF(x-20, chart_rect.bottom()+2, 40, 14), Qt.AlignmentFlag.AlignHCenter | Qt.AlignmentFlag.AlignTop, lab)
|
||
p.end()
|