diff options
Diffstat (limited to 'util.py')
| -rw-r--r-- | util.py | 61 |
1 files changed, 60 insertions, 1 deletions
@@ -4,6 +4,9 @@ import cv2 import platform import readline import os +import numpy as np +from queue import Queue +from threading import Thread, Lock def image_resize(image, width = None, height = None, inter = cv2.INTER_AREA): # initialize the dimensions of the image to be resized and @@ -36,6 +39,11 @@ def image_resize(image, width = None, height = None, inter = cv2.INTER_AREA): # return the resized image return resized +def image_embed(img, dimensions): + ret = np.zeros((dimensions[0], dimensions[1], 3), np.uint8) + ret[0:img.shape[0], 0:img.shape[1]] = img + return ret + ''' Fetch input prompt with prefilled text. @@ -76,4 +84,55 @@ def open_system(file): elif platform.system() == 'Windows': # Windows os.startfile(file) else: # linux variants - subprocess.call(('xdg-open', file))
\ No newline at end of file + subprocess.call(('xdg-open', file)) + +class Worker(Thread): + def __init__(self, tasks): + Thread.__init__(self) + self.tasks = tasks + self.daemon = True + self.lock = Lock() + self.start() + + def run(self): + while True: + func, args, kargs = self.tasks.get() + try: + if func.lower() == "terminate": + break + except: + try: + with self.lock: + func(*args, **kargs) + except Exception as exception: + print(exception) + self.tasks.task_done() + +class ThreadPool: + def __init__(self, num_threads, num_queue=None): + if num_queue is None or num_queue < num_threads: + num_queue = num_threads + self.tasks = Queue(num_queue) + self.threads = num_threads + for _ in range(num_threads): Worker(self.tasks) + + # This function can be called to terminate all the worker threads of the queue + def terminate(self): + self.wait_completion() + for _ in range(self.threads): self.add_task("terminate") + return None + + # This function can be called to add new work to the queue + def add_task(self, func, *args, **kargs): + self.tasks.put((func, args, kargs)) + + # This function can be called to wait till all the workers are done processing the pending works. If this function is called, the main will not process any new lines unless all the workers are done with the pending works. + def wait_completion(self): + self.tasks.join() + + # This function can be called to check if there are any pending/running works in the queue. If there are any works pending, the call will return Boolean True or else it will return Boolean False + def is_alive(self): + if self.tasks.unfinished_tasks == 0: + return False + else: + return True
\ No newline at end of file |