"""Generate model-space Z Depth and preserve color for one complete image."""

from ..geometry.prediction_artifacts import generate_moge2_artifacts
from ..media import copy_color_images, input_frames


def run(context, parameters):
    context.progress(0.05, "Preparing Depth Plane")
    context.directory.mkdir(parents=True, exist_ok=True)
    with input_frames(context, parameters) as images:
        if len(images) != 1:
            raise ValueError("Depth Plane currently supports one image")
        copy_color_images(images, context.directory)
        artifacts = generate_moge2_artifacts(
            context,
            parameters,
            images[0],
            depth_mode="Z",
            normal_mode="TANGENT",
        )
    context.check_cancelled()
    context.progress(1.0, "Depth Plane data is ready")
    return {key: path.name for key, path in artifacts.items()}