CoCoandClaude Opus 4.6
62b62d2a96
- Add ruff configuration to pyproject.toml (py310, 120 char lines) - Fix trailing whitespace (W291/W293), modernize type annotations (UP006/UP045) - Remove unused imports (F401), simplify dict iteration (SIM118) - Use f-string conversion flags (RUF010), modernize exception aliases (UP024) - Replace try/except pass with contextlib.suppress (SIM105) - Prefix unused variables with underscore (F841/RUF059) - Convert lambdas to def statements (E731) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
import os
|
|
import logging
|
|
|
|
# Import centralized logging setup
|
|
try:
|
|
from ..utils.debug_utils import setup_logging
|
|
setup_logging()
|
|
except ImportError:
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Import EXR utilities
|
|
try:
|
|
from ..utils.exr_utils import ExrProcessor
|
|
except ImportError as exc:
|
|
raise ImportError("EXR utilities are required but not available. Please ensure utils are properly installed.") from exc
|
|
|
|
class LoadExr:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image_path": ("STRING", {
|
|
"default": "path/to/image.exr",
|
|
"description": "Full path to the EXR file"
|
|
}),
|
|
"normalize": ("BOOLEAN", {
|
|
"default": False,
|
|
"description": "Normalize image values to the 0-1 range"
|
|
})
|
|
},
|
|
"hidden": {
|
|
"node_id": "UNIQUE_ID",
|
|
"layer_data": "DICT"
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE", "MASK", "CRYPTOMATTE", "LAYERS", "STRING", "STRING", "STRING")
|
|
RETURN_NAMES = ("image", "alpha", "cryptomatte", "layers", "layer names", "raw layer info", "metadata")
|
|
|
|
FUNCTION = "load_image"
|
|
CATEGORY = "Image/EXR"
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, image_path, normalize=False, **kwargs):
|
|
"""
|
|
Smart caching based on file modification time and size.
|
|
Only reload if file actually changed or parameters changed.
|
|
"""
|
|
try:
|
|
if not os.path.isfile(image_path):
|
|
return float("NaN") # File doesn't exist, always try to load
|
|
|
|
stat = os.stat(image_path)
|
|
# Create hash from file path, modification time, size, and normalize parameter
|
|
return f"{image_path}_{stat.st_mtime}_{stat.st_size}_{normalize}"
|
|
except Exception:
|
|
# If we can't access file info, always try to load
|
|
return float("NaN")
|
|
|
|
def load_image(self, image_path: str, normalize: bool = False,
|
|
node_id: str = None, layer_data: dict = None, **kwargs) -> list:
|
|
"""
|
|
Load a single EXR image with support for multiple layers/channel groups.
|
|
Returns:
|
|
- Base RGB image tensor (image)
|
|
- Alpha channel tensor (alpha)
|
|
- Dictionary of all cryptomatte layers as tensors (cryptomatte)
|
|
- Dictionary of all non-cryptomatte layers as tensors (layers)
|
|
- List of processed layer names matching keys in the returned dictionaries (layer names)
|
|
- List of raw channel names from the file (raw layer info)
|
|
- Metadata as JSON string (metadata)
|
|
"""
|
|
|
|
# Check for OIIO availability
|
|
ExrProcessor.check_oiio_availability()
|
|
|
|
try:
|
|
# Validate single image path
|
|
if not os.path.isfile(image_path):
|
|
raise FileNotFoundError(f"Image not found: {image_path}")
|
|
|
|
# Use shared EXR processing functionality
|
|
return ExrProcessor.process_exr_data(image_path, normalize, node_id, layer_data)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error loading EXR file {image_path}: {e!s}")
|
|
raise
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|