1 files changed, 44 insertions, 0 deletions
diff --git a/encoder.py b/encoder.py
new file mode 100644
index 0000000..c5d76e1
--- /dev/null
+++ b/encoder.py
@@ -0,0 +1,44 @@
+import sys
+import numpy as np
+from PyQt6.QtWidgets import QApplication, QWidget, QLabel, QVBoxLayout
+from PyQt6.QtGui import QPixmap
+from PyQt6.QtCore import QTimer
+from PIL import Image, ImageQt
+
+fps = 30
+h = 200
+w = 200
+c = 0
+
+
+class EncoderWidget(QWidget):
+    def __init__(self):
+        super().__init__()
+        self.timer = QTimer(self)
+        self.timer.timeout.connect(self.update)
+        self.timer.start(1000 // fps)
+        self.label = QLabel(self)
+        layout = QVBoxLayout(self)
+        layout.addWidget(self.label)
+        layout.setContentsMargins(0, 0, 0, 0)
+        self.setLayout(layout)
+        self.showFullScreen()
+
+    def update(self):
+        global c
+        if c == 0:
+            array = np.random.randint(0, 256, (h, w, 3))
+            c = 1
+        else:
+            array = np.zeros((h, w, 3))
+            c = 0
+        img = Image.fromarray(array.astype(np.uint8), mode="RGB")
+        qt_img = ImageQt.ImageQt(img)
+        pixmap = QPixmap.fromImage(qt_img).scaled(self.size())
+        self.label.setPixmap(pixmap)
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    widget = EncoderWidget()
+    sys.exit(app.exec())