21 lines
1.0 KiB
Python
21 lines
1.0 KiB
Python
import argparse, pygame, pathlib, os
|
|
from lib.pgfw.pgfw import extension
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("input_path", type=pathlib.Path)
|
|
parser.add_argument("output_path", type=pathlib.Path)
|
|
parser.add_argument("shift", type=int)
|
|
arguments = parser.parse_args()
|
|
pygame.display.set_mode((200, 100))
|
|
pygame.init()
|
|
for root, dirs, files in os.walk(arguments.input_path):
|
|
for name in files:
|
|
if name.endswith(".png"):
|
|
input_file_path = os.path.join(root, name)
|
|
shifted = extension.get_hue_shifted_surface(pygame.image.load(input_file_path).convert_alpha(), arguments.shift)
|
|
output_file_path = pathlib.Path(input_file_path.replace(str(arguments.input_path), str(arguments.output_path)))
|
|
os.makedirs(output_file_path.parent, exist_ok=True)
|
|
pygame.image.save(shifted, str(output_file_path))
|
|
print(f"saved hue shifted {input_file_path} to {output_file_path}")
|