import ctypes import os import sys from ctypes import wintypes from PyQt6.QtWidgets import QWidget, QApplication from PyQt6.QtCore import Qt, QPoint, pyqtSignal, QPropertyAnimation, QEasingCurve, QRectF from PyQt6.QtGui import (QPainter, QPixmap, QBrush, QColor, QPainterPath, QRadialGradient, QLinearGradient, QIcon, QRegion) # 球体内容直径；外侧留一圈给投影 BALL_DIAM = 96 SHADOW_PAD = 5 # 窗口边长 = 圆直径（含投影环），main 里 show_near 用此尺寸 BALL_SIZE = BALL_DIAM + 1 * SHADOW_PAD DRAG_THRESHOLD = 6 def _resource_path(*parts: str) -> str: """开发与 PyInstaller 打包后均能定位项目根目录资源。""" base = sys._MEIPASS if hasattr(sys, "_MEIPASS") else os.path.dirname(os.path.dirname(os.path.abspath(__file__))) return os.path.join(base, *parts) def _resolve_logo_path(): for name in ("logo.png", "logo.ico"): p = _resource_path(name) if os.path.isfile(p): return p return None def _load_logo_pixmap(path): """ICO 多尺寸时用 QIcon 取较大图，避免位图过小、放大后发糊或带边线感。""" if not path: return QPixmap() if os.path.splitext(path)[1].lower() == ".ico": pm = QIcon(path).pixmap(512, 512) if not pm.isNull(): return pm return QPixmap(path) def _win32_set_ellipse_window_rgn(hwnd, w, h): """用系统区域裁切 HWND。""" if not hwnd or w < 2 or h < 2: return gdi32 = ctypes.windll.gdi32 user32 = ctypes.windll.user32 hrgn = gdi32.CreateEllipticRgn(0, 0, w, h) if not hrgn: return user32.SetWindowRgn(hwnd, hrgn, True) def _win32_disable_dwm_rounded_shell(hwnd): """减弱 Win11 给无边框窗套的圆角矩形描边/阴影（与自绘圆叠在一起会像方框）。""" if not hwnd: return try: dwm = ctypes.windll.dwmapi DWMWA_WINDOW_CORNER_PREFERENCE = 33 DWMWCP_DONOTROUND = 1 pref = ctypes.c_uint(DWMWCP_DONOTROUND) dwm.DwmSetWindowAttribute( wintypes.HWND(hwnd), wintypes.DWORD(DWMWA_WINDOW_CORNER_PREFERENCE), ctypes.byref(pref), ctypes.sizeof(pref), ) except Exception: pass class FloatBall(QWidget): clicked = pyqtSignal() right_clicked = pyqtSignal(QPoint) def __init__(self, parent=None): super().__init__(parent) self._stays_on_top = False self._apply_window_flags() self._logo_path = _resolve_logo_path() if self._logo_path: self.setWindowIcon(QIcon(self._logo_path)) self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground) self.setAttribute(Qt.WidgetAttribute.WA_NoSystemBackground) self.setAutoFillBackground(False) # 正方形窗口边长 = 圆直径，绘制也用满圆，避免出现「大圆环套小图」 self.setFixedSize(BALL_SIZE, BALL_SIZE) self._apply_circular_window_shape() self.setCursor(Qt.CursorShape.PointingHandCursor) self._logo = _load_logo_pixmap(self._logo_path) if self._logo_path else QPixmap() self._drag_start = QPoint() self._dragging = False self._hovered = False self._glow = 0.0 # 0.0~1.0 光晕强度 self._glow_anim = QPropertyAnimation(self, b"_glow_prop") self._glow_anim.setDuration(300) self._glow_anim.setEasingCurve(QEasingCurve.Type.OutCubic) self._place_default() def set_stays_on_top(self, on_top): """与主面板图钉联动：仅图钉开启时置顶，避免挡在其他程序前面。""" if on_top == self._stays_on_top: return self._stays_on_top = bool(on_top) was_visible = self.isVisible() geo = self.geometry() self._apply_window_flags() self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground, True) self.setAttribute(Qt.WidgetAttribute.WA_NoSystemBackground, True) if self._logo_path: self.setWindowIcon(QIcon(self._logo_path)) self.setCursor(Qt.CursorShape.PointingHandCursor) self._apply_circular_window_shape() if was_visible: self.setGeometry(geo) self.show() def _apply_circular_window_shape(self): w, h = self.width(), self.height() self.setMask(QRegion(0, 0, w, h, QRegion.RegionType.Ellipse)) if sys.platform == "win32": wid = int(self.winId()) if wid: _win32_set_ellipse_window_rgn(wid, w, h) _win32_disable_dwm_rounded_shell(wid) def showEvent(self, event): super().showEvent(event) if sys.platform == "win32": wid = int(self.winId()) if wid: _win32_set_ellipse_window_rgn(wid, self.width(), self.height()) _win32_disable_dwm_rounded_shell(wid) def _apply_window_flags(self): flags = ( Qt.WindowType.FramelessWindowHint | Qt.WindowType.Tool | Qt.WindowType.NoDropShadowWindowHint ) if self._stays_on_top: flags |= Qt.WindowType.WindowStaysOnTopHint self.setWindowFlags(flags) def _get_glow(self): return self._glow def _set_glow(self, v): self._glow = v; self.update() from PyQt6.QtCore import pyqtProperty _glow_prop = pyqtProperty(float, _get_glow, _set_glow) def _place_default(self): screen = QApplication.primaryScreen().availableGeometry() self.move( screen.right() - self.width() - 20, screen.top() + screen.height() // 2 - self.height() // 2, ) def _paint_radius_outer(self): w, h = self.width(), self.height() return min(w, h) * 0.5 - 0.5 def _paint_radius_ball(self): return self._paint_radius_outer() - SHADOW_PAD def paintEvent(self, event): p = QPainter(self) p.setRenderHint(QPainter.RenderHint.Antialiasing) w, h = self.width(), self.height() cx = w * 0.5 cy = h * 0.5 r_out = self._paint_radius_outer() r = self._paint_radius_ball() g = self._glow glow_strength = 0.28 + 0.72 * g disc_out = QRectF(cx - r_out, cy - r_out, r_out * 2, r_out * 2) disc = QRectF(cx - r, cy - r, r * 2, r * 2) # 0. 外环投影（内容圆与窗口边之间的环形：下重上轻，模拟落地阴影） p.setPen(Qt.PenStyle.NoPen) outer_path = QPainterPath() outer_path.addEllipse(disc_out) inner_hole = QPainterPath() inner_hole.addEllipse(disc) ring = outer_path.subtracted(inner_hole) sh = QLinearGradient(0, cy - r_out, 0, cy + r_out) sh.setColorAt(0.0, QColor(0, 0, 0, 0)) sh.setColorAt(0.45, QColor(0, 0, 0, 0)) sh.setColorAt(0.62, QColor(0, 0, 0, int(18 + 22 * glow_strength))) sh.setColorAt(0.82, QColor(0, 0, 0, int(38 + 35 * g))) sh.setColorAt(1.0, QColor(0, 0, 0, int(52 + 40 * g))) p.fillPath(ring, QBrush(sh)) clip = QPainterPath() clip.addEllipse(disc) p.setClipPath(clip) # 1. 毛玻璃底：半透明冷白渐变（磨砂基底） frost_bg = QLinearGradient(0, cy - r, 0, cy + r) frost_bg.setColorAt(0.0, QColor(255, 255, 255, int(175 + 35 * glow_strength))) frost_bg.setColorAt(0.42, QColor(244, 249, 255, int(130 + 40 * glow_strength))) frost_bg.setColorAt(1.0, QColor(228, 236, 248, int(155 + 45 * glow_strength))) p.setBrush(QBrush(frost_bg)) p.drawEllipse(disc) # 2. 斜向磨砂高光（模拟玻璃反光） sheen = QLinearGradient(cx - r, cy - r, cx + r * 0.55, cy + r * 0.85) sheen.setColorAt(0.0, QColor(255, 255, 255, 0)) sheen.setColorAt(0.38, QColor(255, 255, 255, int(38 + 45 * glow_strength))) sheen.setColorAt(0.52, QColor(255, 255, 255, int(14 + 18 * glow_strength))) sheen.setColorAt(1.0, QColor(255, 255, 255, 0)) p.setCompositionMode(QPainter.CompositionMode.CompositionMode_SoftLight) p.setBrush(QBrush(sheen)) p.drawEllipse(disc) p.setCompositionMode(QPainter.CompositionMode.CompositionMode_SourceOver) # 3. Logo（cover 铺满圆） if not self._logo.isNull(): p.setRenderHint(QPainter.RenderHint.SmoothPixmapTransform, True) cover = max(1, int(2 * r)) scaled = self._logo.scaled( cover, cover, Qt.AspectRatioMode.KeepAspectRatioByExpanding, Qt.TransformationMode.SmoothTransformation ) lx = int(cx - scaled.width() / 2) ly = int(cy - scaled.height() / 2) p.drawPixmap(lx, ly, scaled) # 4. 覆在图上的雾面（毛玻璃「蒙在内容上」） veil = QLinearGradient(cx, cy - r, cx, cy + r) veil.setColorAt(0.0, QColor(255, 255, 255, int(52 + 38 * g))) veil.setColorAt(0.35, QColor(255, 255, 255, int(28 + 22 * g))) veil.setColorAt(0.75, QColor(248, 252, 255, int(18 + 15 * g))) veil.setColorAt(1.0, QColor(240, 245, 252, int(22 + 18 * g))) p.setBrush(QBrush(veil)) p.drawEllipse(disc) # 5. 内缘光晕（边缘略亮，悬停时更柔） rim = QRadialGradient(cx, cy, r) rim.setColorAt(0.0, QColor(255, 255, 255, 0)) rim.setColorAt(0.62, QColor(255, 255, 255, 0)) rim.setColorAt(0.88, QColor(255, 255, 255, int(28 + 55 * glow_strength))) rim.setColorAt(0.97, QColor(255, 255, 255, int(45 + 85 * glow_strength))) rim.setColorAt(1.0, QColor(255, 255, 255, int(22 + 40 * glow_strength))) p.setBrush(QBrush(rim)) p.drawEllipse(disc) # 6. 顶区高光（玻璃上沿反光） hi = QRadialGradient(cx, cy - r * 0.22, r * 1.02) hi.setColorAt(0.0, QColor(255, 255, 255, int(42 + 55 * glow_strength))) hi.setColorAt(0.45, QColor(255, 255, 255, int(12 + 18 * glow_strength))) hi.setColorAt(1.0, QColor(255, 255, 255, 0)) p.setBrush(QBrush(hi)) p.drawEllipse(disc) p.setClipping(False) p.end() def mousePressEvent(self, event): if event.button() == Qt.MouseButton.LeftButton: self._drag_start = event.globalPosition().toPoint() self._dragging = False elif event.button() == Qt.MouseButton.RightButton: self.right_clicked.emit(event.globalPosition().toPoint()) def mouseMoveEvent(self, event): if not (event.buttons() & Qt.MouseButton.LeftButton): return delta = event.globalPosition().toPoint() - self._drag_start if not self._dragging and delta.manhattanLength() > DRAG_THRESHOLD: self._dragging = True if self._dragging: new_pos = self.pos() + delta self._drag_start = event.globalPosition().toPoint() screen = QApplication.primaryScreen().availableGeometry() new_pos.setX(max(screen.left(), min(new_pos.x(), screen.right() - self.width()))) new_pos.setY(max(screen.top(), min(new_pos.y(), screen.bottom() - self.height()))) self.move(new_pos) def mouseReleaseEvent(self, event): if event.button() == Qt.MouseButton.LeftButton: if not self._dragging: self.clicked.emit() else: self._snap_to_edge() def enterEvent(self, event): self._hovered = True self._glow_anim.stop() self._glow_anim.setStartValue(self._glow) self._glow_anim.setEndValue(1.0) self._glow_anim.start() def leaveEvent(self, event): self._hovered = False self._glow_anim.stop() self._glow_anim.setStartValue(self._glow) self._glow_anim.setEndValue(0.0) self._glow_anim.start() def _snap_to_edge(self): screen = QApplication.primaryScreen().availableGeometry() cx = self.x() + self.width() // 2 cy = self.y() + self.height() // 2 dists = { "l": cx - screen.left(), "r": screen.right() - cx, "t": cy - screen.top(), "b": screen.bottom() - cy, } side = min(dists, key=dists.get) ends = { "l": QPoint(screen.left(), self.y()), "r": QPoint(screen.right() - self.width(), self.y()), "t": QPoint(self.x(), screen.top()), "b": QPoint(self.x(), screen.bottom() - self.height()), } anim = QPropertyAnimation(self, b"pos") anim.setDuration(200) anim.setEasingCurve(QEasingCurve.Type.OutCubic) anim.setStartValue(self.pos()) anim.setEndValue(ends[side]) self._snap_anim = anim anim.start()