pack.py

from pathlib import Path
from patch import create_patch, create_patched
from difference import compare_image
from cli import RESET, RED, GREEN, ORANGE, BLUE, MAGENTA, CYAN, BOLD
from traverse import check_out_path, print_indented


SUFFIXES = [".png"]


def create_texture_patch_pack(original_path: Path, modified_path: Path, patch_path: Path, filter_names: list[str] = [], print_full_path: bool = False, overwrite: bool = False) -> None:
    error_paths = []
    def callback_dir(path: Path, level: int):
        text = path.name
        if images := [p for p in path.iterdir() if p.is_file and p.suffix.lower() in SUFFIXES]:
            text += f" ({len(images)})" # TODO: /total
        print_indented(f"{BOLD}{MAGENTA}{text}{RESET}", level)
    def callback_file(image_modified_path: Path, level: int):
        if image_modified_path.suffix in SUFFIXES: # and image_modified_path.name == "bch-bench-wood.png":
            relative_replacements_path = image_modified_path.relative_to(modified_path)
            image_patch_path = patch_path.joinpath(relative_replacements_path)
            image_original_path = original_path.joinpath(relative_replacements_path)
            text = (image_original_path if print_full_path else relative_replacements_path).as_posix()
            print_indented("… " + text, level, end=(None if modified_path == None else "\r"))
            image_patch_path.parent.mkdir(parents=True, exist_ok=True)
            try:
                if not image_original_path.exists():
                    raise FileNotFoundError("Original file does not exist")
                elif not overwrite and image_patch_path.exists():
                    raise FileExistsError("Not allowed to overwrite")
                create_patch(image_original_path, image_modified_path, image_patch_path, filter_names)
            except FileExistsError as e:
                print_indented(f"{GREEN}✖{RESET} {text} SKIPPED: not allowed to overwrite", level, end="\n", flush=True)
            except FileNotFoundError as e:
                print_indented(f"{ORANGE}✖{RESET} {text}", level, end="\t", flush=True)
                print("warning:", str(e))
            except Exception as e:
                error_paths.append(image_original_path)
                print_indented(f"{RED}✖{RESET} {text}", level, end="\t", flush=True)
                print("error:", str(e))
            else:
                print_indented(f"{GREEN}✔{RESET}", level, flush=True) # https://symbolsdb.com/check-mark-symbol
    check_out_path(modified_path, callback_dir, callback_file)
    if error_paths:
        print(f"Encountered {len(error_paths)} errors:")
        for path in error_paths:
            print("  " + str(path))


def create_texture_pack(original_path: Path, patch_path: Path, pack_path: Path, modified_path: Path|None = None, filter_names: list[str] = [], print_full_path: bool = False, overwrite: bool = False) -> None:
    error_paths = []
    def callback_dir(path: Path, level: int):
        text = path.name
        if images := [p for p in path.iterdir() if p.is_file and p.suffix.lower() in SUFFIXES]:
            text += f" ({len(images)})"
        print_indented(f"{BOLD}{CYAN}{text}{RESET}", level)
    def callback_file(image_patch_path: Path, level: int):
        relative_replacements_path = image_patch_path.relative_to(patch_path)
        image_pack_path = pack_path.joinpath(relative_replacements_path)
        image_original_path = original_path.joinpath(relative_replacements_path)
        text = (image_pack_path if print_full_path else relative_replacements_path).as_posix()
        print_indented("… " + text, level, end="\r")
        image_pack_path.parent.mkdir(parents=True, exist_ok=True)
        try:
            if not overwrite and image_pack_path.exists():
                raise FileExistsError("Not allowed to overwrite")
            create_patched(image_original_path, image_patch_path, image_pack_path, filter_names)
            if modified_path:
                image_modified_path = modified_path.joinpath(relative_replacements_path)
                if (difference := compare_image(image_modified_path, image_pack_path)) == (0, 0):
                    print_indented(f"{GREEN}✔{RESET}", level, flush=True) # https://symbolsdb.com/check-mark-symbol
                else:
                    print_indented(f"{RED}✖{RESET} {text}", level, end="\t", flush=True)
                    print(f"({BLUE}{difference[0]}{RESET}, {RED}{difference[1]}{RESET})")
            else:
                print_indented(f"{GREEN}✔{RESET}", level, flush=True) # https://symbolsdb.com/check-mark-symbol
        except FileExistsError as e:
            print_indented(f"{GREEN}✖{RESET} {text} SKIPPED: {e}", level, end="\n", flush=True)
        except AssertionError as e:
            print_indented(f"{RED}✖{RESET} {text}", level, end="\t", flush=True)
            print("assertion error:", str(e))
        except Exception as e:
            error_paths.append(image_original_path)
            print_indented(f"{RED}✖{RESET} {text}", level, end="\t", flush=True)
            print("error:", str(e))
    check_out_path(patch_path, callback_dir, callback_file)
    if error_paths:
        print(f"Encountered {len(error_paths)} errors:")
        for path in error_paths:
            print("  " + str(path))


if __name__ == "__main__":
    # create_texture_patch_pack(Path("./test/modified"), Path("./test/original"), Path("./test/patch"))
    # create_texture_pack(Path("./test/patch"), Path("./test/original"), Path("./test/pack"))
    create_texture_pack(Path("./test/patch"), Path("./test/original"), Path("./test/pack"), Path("./test/modified"))