"""Generate depth data and preserve color for one complete image."""

from ..geometry.generate_depth import generate_depth
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)
        archive_path, _normal_path = generate_depth(
            context,
            parameters,
            images[0],
            output_depth=True,
        )
    context.check_cancelled()
    context.progress(1.0, "Depth Plane data is ready")
    return {"image_geometry": archive_path.name}