vnyash
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from typing import List, Optional
|
|
import hashlib
|
|
import os
|
|
|
|
import deepfuze.globals
|
|
from deepfuze.filesystem import is_directory
|
|
from deepfuze.typing import Padding, Fps
|
|
|
|
|
|
def normalize_output_path(target_path : Optional[str], output_path : Optional[str]) -> Optional[str]:
|
|
if target_path and output_path:
|
|
target_name, target_extension = os.path.splitext(os.path.basename(target_path))
|
|
if is_directory(output_path):
|
|
output_hash = hashlib.sha1(str(deepfuze.globals.__dict__).encode('utf-8')).hexdigest()[:8]
|
|
output_name = target_name + '-' + output_hash
|
|
return os.path.join(output_path, output_name + target_extension)
|
|
output_name, output_extension = os.path.splitext(os.path.basename(output_path))
|
|
output_directory_path = os.path.dirname(output_path)
|
|
if is_directory(output_directory_path) and output_extension:
|
|
return os.path.join(output_directory_path, output_name + target_extension)
|
|
return None
|
|
|
|
|
|
def normalize_padding(padding : Optional[List[int]]) -> Optional[Padding]:
|
|
if padding and len(padding) == 1:
|
|
return tuple([ padding[0] ] * 4) #type:ignore[return-value]
|
|
if padding and len(padding) == 2:
|
|
return tuple([ padding[0], padding[1], padding[0], padding[1] ]) #type:ignore[return-value]
|
|
if padding and len(padding) == 3:
|
|
return tuple([ padding[0], padding[1], padding[2], padding[1] ]) #type:ignore[return-value]
|
|
if padding and len(padding) == 4:
|
|
return tuple(padding) #type:ignore[return-value]
|
|
return None
|
|
|
|
|
|
def normalize_fps(fps : Optional[float]) -> Optional[Fps]:
|
|
if fps is not None:
|
|
return max(1.0, min(fps, 60.0))
|
|
return None
|