aswi2021vochomurka-gitlab/aswi2021vochomurka/view/main_view.py @ 21e4cc2f

import logging
import math
import random

from PyQt5.QtCore import QSize, QThread, QObject, pyqtSignal
from PyQt5.QtWidgets import QMainWindow, QPlainTextEdit, QDialog, QHBoxLayout
from numpy import pi, sin, cos, tan, exp
from matplotlib.pyplot import subplot

from aswi2021vochomurka.model.Message import Message
from aswi2021vochomurka.service.mqtt.mqtt_subscriber import MQTTSubscriber
from aswi2021vochomurka.service.subscriber import Subscriber
from aswi2021vochomurka.service.subscriber_callback import SubscriberCallback
from aswi2021vochomurka.service.subscriber_params import SubscriberParams, ConnectionParams

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
import matplotlib.pyplot as plt

import sys
from PyQt5.QtWidgets import QDialog, QApplication, QPushButton, QVBoxLayout
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
import matplotlib.pyplot as plt
import random

from aswi2021vochomurka.view.logger_view import LoggerView


class Worker(QObject, SubscriberCallback):
    connected = pyqtSignal()
    disconnected = pyqtSignal()
    error = pyqtSignal(Exception)
    newMessage = pyqtSignal(str)
    subscriber: Subscriber = None

    params = SubscriberParams(
        ["/home/1", "/home/2"],
        10,
        ConnectionParams("localhost", 1883, 60),
        True
    )

    def start(self):
        self.subscriber = MQTTSubscriber(self, self.params)
        self.subscriber.start()

    def onConnected(self):
        self.connected.emit()

    def onDisconnected(self):
        self.disconnected.emit()

    def onError(self):
        self.error.emit()

    def onMessage(self, message: Message):
        self.newMessage.emit(message.topic)
        self.window.plot(message)

    def onCloseTopic(self, topic: str):
        pass


class MainView(QDialog):
    worker: Worker = None
    workerThread: QThread = None

    def __init__(self):
        super(MainView, self).__init__()

        self.chartsNum = 0
        self.arrayData = []

        self.dataIndex = 0
        self.dataDict = {}

        self.figure = plt.figure(figsize=([500,500]))

        self.canvas = FigureCanvas(self.figure)
        self.toolbar = NavigationToolbar(self.canvas, self)

        self.setMinimumSize(QSize(440, 240))
        self.setWindowTitle("MQTT demo")

        # Add logger text field
        logger = LoggerView(self)
        formatter = logging.Formatter('%(asctime)s %(message)s', '%H:%M')
        logger.setFormatter(formatter)
        logger.setLevel(logging.INFO)
        logging.getLogger('').addHandler(logger)

        layout = QVBoxLayout()
        layout.addWidget(logger.widget)
        layout.addWidget(self.toolbar)
        layout.addWidget(self.canvas)

        self.setLayout(layout)

        self.initSubscriber()

    def plot(self, message: Message):
        self.figure.clear()

        if message.topic in self.dataDict:
            self.dataDict[message.topic].append(message.value)
        else:
            self.dataDict[message.topic] = [message.value]
            self.chartsNum += 1

        rows = math.ceil(self.chartsNum / 2)

        b = 0
        for a in self.dataDict.values():
            self.figure.add_subplot(rows, 2, b + 1)
            b += 1
            plt.plot(a)

        self.canvas.draw()

    def initSubscriber(self):
        self.workerThread = QThread()
        self.worker = Worker()
        self.worker.moveToThread(self.workerThread)
        self.workerThread.started.connect(self.worker.start)
        # self.worker.newMessage.connect(
        #    lambda message: self.b.insertPlainText(message + "\n")
        # )
        self.worker.window = self
        self.workerThread.start()