From 06ad76b6ab86abdf3070fae8133c9d2d1661df98 Mon Sep 17 00:00:00 2001
From: Suthep Pomjaksilp <sp@laz0r.de>
Date: Fri, 22 Apr 2022 03:27:28 +0200
Subject: [PATCH] applets: add progress bar applet

Signed-off-by: Suthep Pomjaksilp <pomjaksi@physik.uni-kl.de>
---
 artiq/applets/progress_bar.py | 34 ++++++++++++++++++++++++++++++++++
 artiq/gui/applets.py          |  1 +
 2 files changed, 35 insertions(+)
 create mode 100644 artiq/applets/progress_bar.py

diff --git a/artiq/applets/progress_bar.py b/artiq/applets/progress_bar.py
new file mode 100644
index 000000000..bbded954c
--- /dev/null
+++ b/artiq/applets/progress_bar.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+
+from PyQt5 import QtWidgets
+
+from artiq.applets.simple import SimpleApplet
+
+
+class ProgressWidget(QtWidgets.QProgressBar):
+    def __init__(self, args):
+        QtWidgets.QProgressBar.__init__(self)
+        self.setMinimum(args.min)
+        self.setMaximum(args.max)
+        self.dataset_value = args.value
+
+    def data_changed(self, data, mods):
+        try:
+            value = round(data[self.dataset_value][1])
+        except (KeyError, ValueError, TypeError):
+            value = 0
+        self.setValue(value)
+
+
+
+def main():
+    applet = SimpleApplet(ProgressWidget)
+    applet.add_dataset("value", "counter")
+    applet.argparser.add_argument("--min", type=int, default=0,
+                                  help="minimum (left) value of the bar")
+    applet.argparser.add_argument("--max", type=int, default=100,
+                                  help="maximum (right) value of the bar")
+    applet.run()
+
+if __name__ == "__main__":
+    main()
diff --git a/artiq/gui/applets.py b/artiq/gui/applets.py
index 178593a07..940149631 100644
--- a/artiq/gui/applets.py
+++ b/artiq/gui/applets.py
@@ -225,6 +225,7 @@ _templates = [
                        "HIST_BIN_BOUNDARIES_DATASET "
                        "HISTS_COUNTS_DATASET"),
     ("Image", "${artiq_applet}image IMG_DATASET"),
+    ("Progress bar", "${artiq_applet}progress_bar VALUE"),
 ]