Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-05-21 19:47:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

redirected subprocess outputs into Rich console

Browse Source
This commit is contained in:
k4yt3x
2022-02-28 01:54:52 +00:00
parent c0fe81bd2e
commit 4459f4d3be
4 changed files with 131 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
video2x/video2x.py
View File

@@ -248,10 +248,26 @@ class Video2X:
# wait for jobs in queue to deplete
while self.processed.value < total_frames - 1:
time.sleep(0.5)
# check processor health
for process in self.processor_processes:
if not process.is_alive():
raise Exception("process died unexpectedly")
# check decoder health
if (
not self.decoder.is_alive()
and self.decoder.exception is not None
):
raise Exception("decoder died unexpectedly")
# check encoder health
if (
not self.encoder.is_alive()
and self.encoder.exception is not None
):
raise Exception("encoder died unexpectedly")
# show progress bar when upscale starts
if progress.disable is True and self.processed.value > 0:
progress.disable = False
@@ -266,7 +282,8 @@ class Video2X:
# if SIGTERM is received or ^C is pressed
# TODO: pause and continue here
except (SystemExit, KeyboardInterrupt) as e:
logger.warning("Exit signal received, terminating")
logger.warning("Exit signal received, exiting gracefully")
logger.warning("Press ^C again to force terminate")
exception.append(e)
except Exception as e:
@@ -281,8 +298,8 @@ class Video2X:
# mark processing queue as closed
self.processing_queue.close()
# stop upscaler processes
logger.info("Stopping upscaler processes")
# stop processor processes
logger.info("Stopping processor processes")
for process in self.processor_processes:
process.terminate()
@@ -291,6 +308,7 @@ class Video2X:
process.join()
# ensure both the decoder and the encoder have exited
logger.info("Stopping decoder and encoder threads")
self.decoder.stop()
self.encoder.stop()
self.decoder.join()