new test for mp

2018-05-17 21:44:45 +02:00
parent 32828d3a65
commit f78ddbc005
8 changed files with 324 additions and 25 deletions
@@ -0,0 +1,124 @@
+#import gobject
+#import pygtk
+#pyGtk.require('2.0')
+#import gtk
+#import multiprocessing
+#import threading
+#import time
+import gi
+gi.require_version('Gtk', '3.0')
+from gi.repository import Gtk, Gio, GObject as gobject
+import time
+from multiprocessing import Process, Value, Lock, Pool, Queue, cpu_count
+from threading import Thread
+
+gobject.threads_init()
+
+class Listener(gobject.GObject):
+    __gsignals__ = {
+        'updated' : (gobject.SIGNAL_RUN_LAST,
+                     gobject.TYPE_NONE,
+                     (gobject.TYPE_FLOAT, gobject.TYPE_STRING)),
+        'finished': (gobject.SIGNAL_RUN_LAST,
+                     gobject.TYPE_NONE,
+                     ())
+    }
+
+    def __init__(self, queue):
+        gobject.GObject.__init__(self)
+        self.queue = queue
+
+    def go(self):
+        print("Listener has started")
+        while True:
+            # Listen for results on the queue and process them accordingly                            
+            data = self.queue.get()
+            # Check if finished                                                                       
+            if data[1]=="finished":
+                print("Listener is finishing.")
+                self.emit("finished")
+                return
+            else:
+                self.emit('updated', data[0], data[1])
+
+gobject.type_register(Listener)
+
+class Worker():
+    def __init__(self, queue):
+        self.queue = queue
+
+    def go(self):
+        print("The worker has started doing some work (counting from 0 to 9)")
+        for i in range(10):
+            proportion = (float(i+1))/10
+            self.queue.put((proportion, "working..."))
+            time.sleep(0.5)
+        self.queue.put((1.0, "finished"))
+        print("The worker has finished.")
+
+
+class Interface:
+    def __init__(self):
+        self.process = None
+        self.progress = Gtk.ProgressBar()
+        button = Gtk.Button("Go!")
+        button.connect("clicked", self.go)
+        vbox = Gtk.VBox(spacing=5)
+        vbox.pack_start(self.progress)
+        vbox.pack_start(button)
+        vbox.show_all()
+        self.frame = vbox
+
+    def main(self):
+        window = Gtk.Window(Gtk.WINDOW_TOPLEVEL)
+        window.set_border_width(10)
+        window.add(self.frame)
+        window.show()
+        window.connect("destroy", self.destroy)
+
+        Gtk.main()
+
+    def destroy(self, widget, data=None):
+        Gtk.main_quit()
+
+
+    def callbackDisplay(self, obj, fraction, text, data=None):
+        self.progress.set_fraction(fraction)
+        self.progress.set_text(text)
+
+    def callbackFinished(self, obj, data=None):
+        if self.process==None:
+            raise RuntimeError("No worker process started")
+        print("all done; joining worker process")
+        self.process.join()
+        self.process = None
+
+        self.progress.set_fraction(1.0)
+        self.progress.set_text("done")
+
+    def go(self, widget, data=None):
+        if self.process!=None:
+            return
+
+        print("Creating shared Queue")
+        queue = Queue()
+
+        print("Creating Worker")
+        worker = Worker(queue)
+
+        print("Creating Listener")
+        listener = Listener(queue)
+        listener.connect("updated",self.callbackDisplay)
+        listener.connect("finished",self.callbackFinished)
+
+        print("Starting Worker")
+        self.process = Process(target=worker.go, args=())
+        self.process.start()
+
+        print("Starting Listener")
+        thread = Thread(target=listener.go, args=())
+        thread.start()
+
+if __name__ == '__main__':
+    gui = Interface()
+    gui.main()