mirror of
https://github.com/k4yt3x/video2x.git
synced 2025-03-12 19:22:13 +00:00
70 lines
2.2 KiB
Python
Executable File
70 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Copyright (C) 2018-2022 K4YT3X and contributors.
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
Name: Processor Abstract Class
|
|
Author: K4YT3X
|
|
Date Created: April 9, 2022
|
|
Last Modified: April 9, 2022
|
|
"""
|
|
|
|
from abc import ABC, abstractmethod
|
|
from multiprocessing import Queue
|
|
from multiprocessing.managers import DictProxy
|
|
from multiprocessing.sharedctypes import Synchronized
|
|
|
|
from PIL import Image, ImageChops, ImageStat
|
|
|
|
|
|
class Processor(ABC):
|
|
def __init__(
|
|
self, tasks_queue: Queue, processed_frames: DictProxy, pause_flag: Synchronized
|
|
) -> None:
|
|
self.tasks_queue = tasks_queue
|
|
self.processed_frames = processed_frames
|
|
self.pause_flag = pause_flag
|
|
|
|
@abstractmethod
|
|
def process(self):
|
|
raise NotImplementedError
|
|
|
|
@staticmethod
|
|
def get_image_diff(image0: Image.Image, image1: Image.Image) -> float:
|
|
"""
|
|
get the percentage difference between two images
|
|
|
|
:param image0 Image.Image: the image to compare
|
|
:param image1 Image.Image: the image to compare against
|
|
:rtype float: precentage difference between two frames
|
|
"""
|
|
difference_stat = ImageStat.Stat(ImageChops.difference(image0, image1))
|
|
return sum(difference_stat.mean) / (len(difference_stat.mean) * 255) * 100
|
|
|
|
"""
|
|
def run(
|
|
self,
|
|
) -> None:
|
|
self.running = True
|
|
while self.running is True:
|
|
self.process()
|
|
self.running = False
|
|
return super().run()
|
|
|
|
def stop(self, _signal_number, _frame) -> None:
|
|
self.running = False
|
|
"""
|