""" novelWriter – GUI Main Window Status Bar ======================================== File History: Created: 2019-04-20 [0.0.1] GuiMainStatus This file is a part of novelWriter Copyright (C) 2019 Veronica Berglyd Olsen and novelWriter contributors This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . """ from __future__ import annotations import logging from datetime import datetime from time import time from PyQt6.QtCore import QLocale, pyqtSlot from PyQt6.QtWidgets import QApplication, QLabel, QStatusBar, QWidget from novelwriter import CONFIG, SHARED from novelwriter.common import formatTime from novelwriter.constants import nwConst from novelwriter.enum import nwTrinary from novelwriter.extensions.modified import NClickableLabel from novelwriter.extensions.statusled import StatusLED logger = logging.getLogger(__name__) class GuiMainStatus(QStatusBar): def __init__(self, parent: QWidget) -> None: super().__init__(parent=parent) logger.debug("Create: GuiMainStatus") self._refTime = -1.0 self._userIdle = False self._debugInfo = False iPx = SHARED.theme.baseIconHeight # Permanent Widgets # ================= # The Spell Checker Language self.langIcon = QLabel("", self) self.langText = QLabel(self.tr("None"), self) self.langIcon.setContentsMargins(0, 0, 0, 0) self.langText.setContentsMargins(0, 0, 8, 0) self.addPermanentWidget(self.langIcon) self.addPermanentWidget(self.langText) # The Editor Status self.docIcon = StatusLED(iPx, iPx, self) self.docText = QLabel(self.tr("Editor"), self) self.docIcon.setContentsMargins(0, 0, 0, 0) self.docText.setContentsMargins(0, 0, 8, 0) self.addPermanentWidget(self.docIcon) self.addPermanentWidget(self.docText) # The Project Status self.projIcon = StatusLED(iPx, iPx, self) self.projText = QLabel(self.tr("Project"), self) self.projIcon.setContentsMargins(0, 0, 0, 0) self.projText.setContentsMargins(0, 0, 8, 0) self.addPermanentWidget(self.projIcon) self.addPermanentWidget(self.projText) # The Project and Session Stats self.statsIcon = QLabel(self) self.statsText = QLabel("", self) self.statsIcon.setContentsMargins(0, 0, 0, 0) self.statsText.setContentsMargins(0, 0, 8, 0) self.addPermanentWidget(self.statsIcon) self.addPermanentWidget(self.statsText) # The Session Clock # Set the minimum width so the label doesn't rescale every second self.timeIcon = NClickableLabel(self) self.timeIcon.mouseClicked.connect(self._onClickTimerLabel) self.timeText = NClickableLabel("", self) self.timeText.setToolTip(self.tr("Session Time")) self.timeText.setMinimumWidth(SHARED.theme.getTextWidth("00:00:00:")) self.timeIcon.setContentsMargins(0, 0, 0, 0) self.timeText.setContentsMargins(0, 0, 0, 0) self.timeText.setVisible(CONFIG.showSessionTime) self.timeText.mouseClicked.connect(self._onClickTimerLabel) self.addPermanentWidget(self.timeIcon) self.addPermanentWidget(self.timeText) # Other Settings self.setSizeGripEnabled(True) logger.debug("Ready: GuiMainStatus") self.updateTheme() self.clearStatus() return def clearStatus(self) -> None: """Reset all widgets on the status bar to default values.""" self.setRefTime(-1.0) self.setLanguage(*SHARED.spelling.describeDict()) self.setProjectStats(0, 0) self.setProjectStatus(nwTrinary.NEUTRAL) self.setDocumentStatus(nwTrinary.NEUTRAL) self.updateTime() return def updateTheme(self) -> None: """Update theme elements.""" iPx = SHARED.theme.baseIconHeight self.langIcon.setPixmap(SHARED.theme.getPixmap("language", (iPx, iPx))) self.statsIcon.setPixmap(SHARED.theme.getPixmap("stats", (iPx, iPx))) self.timePixmap = SHARED.theme.getPixmap("timer", (iPx, iPx)) self.idlePixmap = SHARED.theme.getPixmap("timer_off", (iPx, iPx)) self.timeIcon.setPixmap(self.timePixmap) colNone = SHARED.theme.getIconColor("default").darker(150) colSaved = SHARED.theme.getIconColor("green").darker(150) colUnsaved = SHARED.theme.getIconColor("red").darker(150) self.docIcon.setColors(colNone, colSaved, colUnsaved) self.projIcon.setColors(colNone, colSaved, colUnsaved) return ## # Setters ## def setRefTime(self, refTime: float) -> None: """Set the reference time for the status bar clock.""" self._refTime = refTime return def setProjectStatus(self, state: nwTrinary) -> None: """Set the project status colour icon.""" self.projIcon.setState(state) return def setDocumentStatus(self, state: nwTrinary) -> None: """Set the document status colour icon.""" self.docIcon.setState(state) return def setUserIdle(self, idle: bool) -> None: """Change the idle status icon.""" if not CONFIG.stopWhenIdle: idle = False if self._userIdle != idle: if idle: self.timeIcon.setPixmap(self.idlePixmap) else: self.timeIcon.setPixmap(self.timePixmap) self._userIdle = idle return def setProjectStats(self, pWC: int, sWC: int) -> None: """Update the current project statistics.""" self.statsText.setText(self.tr("Words: {0} ({1})").format(f"{pWC:n}", f"{sWC:+n}")) if CONFIG.incNotesWCount: self.statsText.setToolTip(self.tr("Project word count (session change)")) else: self.statsText.setToolTip(self.tr("Novel word count (session change)")) return def updateTime(self, idleTime: float = 0.0) -> None: """Update the session clock.""" if self._refTime < 0.0: self.timeText.setText("00:00:00") else: if CONFIG.stopWhenIdle: sessTime = round(time() - self._refTime - idleTime) else: sessTime = round(time() - self._refTime) self.timeText.setText(formatTime(sessTime)) return ## # Public Slots ## @pyqtSlot(str) def setStatusMessage(self, message: str) -> None: """Set the status bar message to display.""" self.showMessage(message, nwConst.STATUS_MSG_TIMEOUT) QApplication.processEvents() return @pyqtSlot(str, str) def setLanguage(self, language: str, provider: str) -> None: """Set the language code for the spell checker.""" if language == "None": self.langText.setText(self.tr("None")) self.langText.setToolTip("") else: self.langText.setText(QLocale(language).nativeLanguageName().title()) self.langText.setToolTip(f"{language} ({provider})" if provider else language) return @pyqtSlot(bool) def updateProjectStatus(self, status: bool) -> None: """Update the project status.""" self.setProjectStatus(nwTrinary.NEGATIVE if status else nwTrinary.POSITIVE) return @pyqtSlot(bool) def updateDocumentStatus(self, status: bool) -> None: """Update the document status.""" self.setDocumentStatus(nwTrinary.NEGATIVE if status else nwTrinary.POSITIVE) return ## # Private Slots ## @pyqtSlot() def _onClickTimerLabel(self) -> None: """Process mouse click on timer label.""" state = not CONFIG.showSessionTime self.timeText.setVisible(state) CONFIG.showSessionTime = state return ## # Debug ## def memInfo(self) -> None: # pragma: no cover """Display memory info on the status bar. This is used to investigate memory usage and Qt widgets that get left in memory. Enabled by the --meminfo command line flag. By default, this tracks memory usage diff after launch. To track full memory usage, set environment variable PYTHONTRACEMALLOC=1 before starting novelWriter. """ import tracemalloc count = len(QApplication.allWidgets()) if not self._debugInfo: if tracemalloc.is_tracing(): self._traceMallocRef = "Total" else: self._traceMallocRef = "Relative" tracemalloc.start() self._debugInfo = True current, peak = tracemalloc.get_traced_memory() stamp = datetime.now().strftime("%H:%M:%S") message = ( f"Widgets: {count} \u2013 " f"{self._traceMallocRef} Memory: {current/1024:,.2f} kiB \u2013 " f"Peak: {peak/1024:,.2f} kiB" ) self.showMessage(f"Debug [{stamp}] {message}", 6000) logger.debug("[MEMINFO] %s", message) return