File size: 4,448 Bytes
0e7beee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 | import os
import re
import numpy as np
from datetime import datetime
from PIL import Image, ImageCms
import folder_paths
# Standard Apple ImageIO DQT Quantization Tables for JPEG (Luminance & Chrominance)
APPLE_DQT_LUMA = [
2, 1, 1, 2, 2, 4, 5, 6,
1, 1, 1, 2, 3, 6, 6, 6,
1, 1, 2, 2, 4, 6, 7, 6,
1, 2, 2, 3, 5, 9, 8, 7,
2, 2, 4, 6, 7, 11, 10, 8,
2, 3, 5, 6, 8, 10, 11, 9,
5, 6, 8, 9, 10, 12, 12, 10,
7, 9, 9, 10, 11, 10, 10, 10
]
APPLE_DQT_CHROMA = [
2, 2, 2, 4, 8, 8, 8, 8,
2, 2, 3, 5, 8, 8, 8, 8,
2, 3, 5, 8, 8, 8, 8, 8,
4, 5, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8,
8, 8, 8, 8, 8, 8, 8, 8
]
def process_filename_prefix(prefix):
def replace_date(match):
fmt_str = match.group(1)
replacements = [
("yyyy", "%Y"), ("YY", "%y"), ("yy", "%y"),
("MM", "%m"), ("dd", "%d"),
("HH", "%H"), ("hh", "%H"),
("mm", "%M"), ("ss", "%S")
]
for old, new in replacements:
fmt_str = fmt_str.replace(old, new)
return datetime.now().strftime(fmt_str)
prefix = re.sub(r"%date:([^%]+)%", replace_date, prefix)
return prefix
class SaveImageJPEG:
def __init__(self):
self.output_dir = folder_paths.get_output_directory()
self.type = "output"
self.prefix_append = ""
# Generate binary sRGB IEC61966-2.1 ICC profile
srgb_profile = ImageCms.createProfile("sRGB")
self.srgb_icc_bytes = ImageCms.ImageCmsProfile(srgb_profile).tobytes()
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"Images": ("IMAGE", {"tooltip": "Input images tensor to save."}),
"Filename Prefix": ("STRING", {"default": "iPhone_JPEG"}),
"Quality": ("INT", {"default": 88, "min": 1, "max": 100, "step": 1, "tooltip": "JPEG Quality factor (Apple default is 85-90%)."}),
"Use Apple JPEG Tables": ("BOOLEAN", {"default": True, "tooltip": "Use custom Apple DQT quantization tables."}),
"Subsampling": (["4:2:0", "4:2:2", "4:4:4"], {"default": "4:2:0", "tooltip": "Chroma subsampling mode (Apple default is 4:2:0)."}),
},
}
RETURN_TYPES = ()
FUNCTION = "save_images"
OUTPUT_NODE = True
CATEGORY = "📸 shots by Faded/Utils"
DESCRIPTION = "Saves JPEG images matching Apple's compression profile with dynamic date formatting."
def save_images(self, Images, **kwargs):
filename_prefix = kwargs.get("Filename Prefix", "iPhone_JPEG")
quality = kwargs.get("Quality", 88)
use_apple_jpeg_tables = kwargs.get("Use Apple JPEG Tables", True)
subsampling = kwargs.get("Subsampling", "4:2:0")
filename_prefix += self.prefix_append
filename_prefix = process_filename_prefix(filename_prefix)
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
filename_prefix, self.output_dir, Images[0].shape[2], Images[0].shape[1]
)
results = list()
qtables_param = [APPLE_DQT_LUMA, APPLE_DQT_CHROMA] if use_apple_jpeg_tables else None
for (batch_number, image) in enumerate(Images):
# Convert PyTorch tensor (float 0..1) to PIL Image (uint8 0..255 RGB)
i = 255. * image.cpu().numpy()
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
file = f"{filename}_{counter:05}_.jpg"
file_path = os.path.join(full_output_folder, file)
img.save(
file_path,
format="JPEG",
quality=quality,
subsampling=subsampling,
progressive=False,
qtables=qtables_param,
icc_profile=self.srgb_icc_bytes,
)
results.append({
"filename": file,
"subfolder": subfolder,
"type": self.type
})
counter += 1
return {"ui": {"images": results}}
NODE_CLASS_MAPPINGS = {
"SaveImageJPEG": SaveImageJPEG
}
NODE_DISPLAY_NAME_MAPPINGS = {
"SaveImageJPEG": "Save Image (JPEG)"
} |