cli.py

import cv2
import sys

import palette
import color
from image import Image
from pipeline import PipelineConfig

def main():
    if len(sys.argv) == 1 or len(sys.argv) > 6:
        print(f"usage: {sys.argv[0]} <filename> <palette> <pixelsize> <space> <noisesize>")
        sys.exit(1)
    filename = sys.argv[1]
    image = cv2.imread(filename)
    pixel_size = 1
    config = PipelineConfig()
    config.use_noise_size(0)
    all_palettes = palette.load("palettes.yaml")
    if len(sys.argv) >= 6:
        if sys.argv[5].replace(".", "").isdigit() and (noise_size := float(sys.argv[5])) >= 0:
            config.use_noise_size(noise_size)
    if len(sys.argv) >= 5:
        if (color_space := sys.argv[4].lower()) in color.ColorSpace:
            config.use_color_space(color.ColorSpace(color_space))
    if len(sys.argv) >= 4:
        if sys.argv[3].isdigit() and (pixel_size := int(sys.argv[3])) != 1:
            pixel_size = pixel_size
    if len(sys.argv) >= 3:
        arg = sys.argv[2].lower()
        if arg in all_palettes:
            config.use_palette(all_palettes[arg])
        elif arg.isdigit():
            config.auto_generate_palette(int(arg))
    pipeline = config.create_pipeline()
    new_image = pipeline.run(Image.from_cv2(image, pixel_size))
    cv2.imshow("img", new_image.cv2())
    cv2.waitKey(0)
    cv2.destroyAllWindows()

if __name__ == "__main__":
    main()