Function hugin_refine

lib/pano_utils.py:78–116

(config, nice=19)

def hugin_refine(config, nice=19):
    pto_path = config.pto_path
    output_path =  os.path.join(config.scan_dir, f'{config.scan_id}.jpg')

    # Create Hugin project
    subprocess.run(['pto_gen', '--projection=2', '--fov=360', '-o', pto_path, *config.imglist], check=True)

    # Apply template
    subprocess.run(['pto_template', f'--output={pto_path}', f'--template={config.template_path}', pto_path], check=True)

    # Add control points
    subprocess.run(['cpfind', '-o', pto_path, '--multirow', pto_path], check=True)

    # Clean control points
    subprocess.run(['cpclean', '-o', pto_path, pto_path], check=True)

    # Find line features
    subprocess.run(['linefind', '-o', pto_path, pto_path], check=True)

    # Optimize the project
    subprocess.run(['pto_var', '-o', pto_path, '--opt', 'TrX,TrY,TrZ,r,Eev,Ra,Rb,Rc,Rd,Re,!TrX0,!TrY0,!TrZ0,!r0,!Eev0,!Ra1,!Rb1,!Rc1,!Rd1,!Re1', pto_path], check=True)
    subprocess.run(['autooptimiser', '-n', '-o', pto_path, pto_path], check=True)

    # Modify resolution of the temporary project file
    hugin_modify(pto_path, pto_path, width=config.get("PANO", "PANO_WIDTH"))

    # Start stitching as non-blocking thread with low priority
    if config.get("ENABLE_PANO"):
        cmd_string = ['nice', '-n', str(nice), 'hugin_executor', '--stitching', f'--prefix={output_path}', pto_path]
        process = subprocess.Popen(cmd_string)

        # Wait for the process to finish
        process.wait()

        # Check return code if stitching was successful
        if process.returncode != 0:
            raise Exception(f"Command failed with return code {process.returncode}")

    return output_path


def sample_color(img, uv, normalize_color=False):