poxularandCursor
2f35b1c3f9
Route develop@1600 through inference createTensor resize, align reference JPEG?256?WebP net prep with the worker, and replace stale lut_hash 921fe3a5 with verified a48758ca in parity_pair. Co-authored-by: Cursor <cursoragent@cursor.com>
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
"""Apply merged LUT (Polarr probe style or legacy numpy)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from .lut import apply_merged_lut_preview
|
|
from .polarr_lut_rgb import apply_polarr_color_match_probe_style
|
|
|
|
RENDER_PROBE_EXPORT = "probe_export"
|
|
RENDER_POLARR_PROBE = "polarr_probe"
|
|
RENDER_NUMPY = "numpy_legacy"
|
|
|
|
|
|
def apply_merged_lut_output(
|
|
srgb_hwc: np.ndarray,
|
|
merged_lut: np.ndarray,
|
|
strength: float,
|
|
*,
|
|
render_mode: str = RENDER_PROBE_EXPORT,
|
|
lut_encoding: str = "srgb_srgb",
|
|
profile_stage: str = "current_profile_stages",
|
|
base_adjustments: dict | None = None,
|
|
) -> np.ndarray:
|
|
strength = float(np.clip(strength, 0.0, 1.0))
|
|
if render_mode == RENDER_NUMPY:
|
|
return apply_merged_lut_preview(srgb_hwc, merged_lut, strength)
|
|
if render_mode == RENDER_PROBE_EXPORT:
|
|
from .probe_parity.pipeline import apply_probe_export
|
|
|
|
return apply_probe_export(
|
|
srgb_hwc,
|
|
merged_lut,
|
|
strength,
|
|
base_adjustments=base_adjustments,
|
|
profile_stage=profile_stage,
|
|
lut_encoding=lut_encoding,
|
|
)
|
|
return apply_polarr_color_match_probe_style(
|
|
srgb_hwc,
|
|
merged_lut,
|
|
strength,
|
|
encoding=lut_encoding,
|
|
base_adjustments=base_adjustments,
|
|
profile_stage=profile_stage,
|
|
)
|