Add Moiré transform

Tests/test_moire_synthesis.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from PIL import Image, ImageOps
+
+
+def test_moire_output_size() -> None:
+    img = Image.new("RGB", (100, 100), (128, 128, 128))
+    result = ImageOps.moire(img)
+    assert result.size == img.size
+
+
+def test_moire_output_mode() -> None:
+    img = Image.new("RGB", (100, 100), (128, 128, 128))
+    result = ImageOps.moire(img)
+    assert result.mode == "RGB"
+
+
+def test_moire_accepts_non_rgb() -> None:
+    img = Image.new("L", (100, 100), 128)
+    result = ImageOps.moire(img)
+    assert result.mode == "RGB"

src/PIL/ImageOps.py

@@ -25,6 +25,17 @@
 from typing import Literal, Protocol, cast, overload
 
 from . import ExifTags, Image, ImagePalette
+from ._moire import (
+    _add_noise,
+    _bayer_resampling,
+    _demosaic_bilinear,
+    _denoise,
+    _flat_top_filtering,
+    _jpeg_compression,
+    _lcd_resampling,
+    _projective_transformation,
+    _radial_distortion,
+)
 
 #
 # helpers
@@ -647,6 +658,28 @@ def mirror(image: Image.Image) -> Image.Image:
     return image.transpose(Image.Transpose.FLIP_LEFT_RIGHT)
 
 
+def moire(image: Image.Image) -> Image.Image:
+    """
+    Generate a synthetic Moire image.
+    :param image:
+    :return: An image.
+    """
+    if len(image.getbands()) == 1:
+        image = image.convert("RGB")
+
+    resampled_img = _lcd_resampling(image)
+    projective_transform = _projective_transformation(resampled_img)
+    distorted_img = _radial_distortion(projective_transform)
+    filtered_img = _flat_top_filtering(distorted_img)
+    Bayer = _bayer_resampling(filtered_img)
+    Bayer_noise = _add_noise(Bayer)
+    rgb_img = _demosaic_bilinear(Bayer_noise)
+    rgb_denoised = _denoise(rgb_img)
+    compressed_img = _jpeg_compression(rgb_denoised)
+
+    return compressed_img
+
+
 def posterize(image: Image.Image, bits: int) -> Image.Image:
     """
     Reduce the number of bits for each color channel.

src/PIL/_moire.py

@@ -0,0 +1,296 @@
+from __future__ import annotations
+
+import io
+import math
+import random
+from typing import cast
+
+from . import Image, ImageFilter
+
+
+def _lcd_resampling(img: Image.Image) -> Image.Image:
+    """
+    Simulate an LCD display by mapping each pixel to a single RGB subpixel
+    in the repeating R-G-B stripe layout.
+
+    :param img:
+    :return: An image.
+    """
+    w, h = img.size
+    resampled_img = Image.new("RGB", (w, h))
+
+    for y in range(h):
+        num = 1
+        for x in range(w):
+            r, g, b = cast(tuple[int, int, int], img.getpixel((x, y)))
+            if num % 3 == 0:
+                resampled_img.putpixel((x, y), (0, 0, b))
+            elif num % 3 == 1:
+                resampled_img.putpixel((x, y), (r, 0, 0))
+            else:
+                resampled_img.putpixel((x, y), (0, g, 0))
+            num += 1
+
+    return resampled_img
+
+
+def _projective_transformation(img: Image.Image) -> Image.Image:
+    """
+    Apply a random projective transformation to simulate varying camera
+    position and orientation relative to the display.
+
+    :param img:
+    :return: An image.
+    """
+    w, h = img.size
+    theta = math.radians(random.uniform(-1, 1))
+
+    # rotation
+    a = math.cos(theta)
+    b = -math.sin(theta)
+    d = math.sin(theta)
+    e = math.cos(theta)
+
+    # Translation
+    c = random.uniform(-0.01 * w, 0.01 * w)
+    f = random.uniform(-0.01 * h, 0.01 * h)
+
+    # Perspective distortion
+    g = random.uniform(-1e-5, 1e-5)
+    h_p = random.uniform(-1e-5, 1e-5)
+
+    # H
+    coeffs = (a, b, c, d, e, f, g, h_p)
+
+    return img.transform(
+        (w, h), Image.Transform.PERSPECTIVE, coeffs, resample=Image.Resampling.BICUBIC
+    )
+
+
+def _radial_distortion(img: Image.Image, k: float = -1e-7) -> Image.Image:
+    """
+    Use radial distortion function to simulate lens distortion
+
+    :param img:
+    :param k:
+    :return: An image
+    """
+    w, h = img.size
+    radial_distort = Image.new("RGB", (w, h))
+
+    cx = w / 2
+    cy = h / 2
+
+    for y in range(h):
+        for x in range(w):
+            r, g, b = cast(tuple[int, int, int], img.getpixel((x, y)))
+            xc = x - cx
+            yc = y - cy
+            radius2 = xc**2 + yc**2
+
+            factor = 1 + k * radius2
+            radial_x = int(xc * factor + cx)
+            radial_y = int(yc * factor + cy)
+
+            # Boundary check
+            if 0 <= radial_x < w and 0 <= radial_y < h:
+                radial_distort.putpixel((radial_x, radial_y), (r, g, b))
+
+    return radial_distort
+
+
+def _flat_top_kernel(
+    size: int = 5, sigma: float = 1.0, n: int = 2
+) -> list[list[float]]:
+    """
+    Generate a flat-top Gaussian kernel.
+
+    :param size: the size of the kernel to be produced
+    :param sigma: controls the broadness of the Gaussian kernel
+    :param n: controls the flatness of the kernel peak
+    :return: An Array
+    """
+    kernel = []
+    center = size // 2
+    total = 0.0
+
+    for y in range(size):
+        row = []
+        for x in range(size):
+            dx = x - center
+            dy = y - center
+            r2 = dx * dx + dy * dy
+            value = math.exp(-((r2 / (2 * sigma * sigma)) ** n))
+            row.append(value)
+            total += value
+        kernel.append(row)
+
+    for y in range(size):
+        for x in range(size):
+            kernel[y][x] /= total
+
+    return kernel
+
+
+def _flat_top_filtering(
+    img: Image.Image, size: int = 5, sigma: float = 1.0, n: int = 2
+) -> Image.Image:
+    """
+    Applying the flat top gaussian kernel on the image to simulate anti-aliasing fiter
+
+    :param img:
+    :param size:
+    :param sigma:
+    :param n:
+    :return: An image
+    """
+    kernel = _flat_top_kernel(size=size, sigma=sigma, n=n)
+    flat_kernel = []
+    for row in kernel:
+        flat_kernel.extend(row)
+
+    return img.filter(ImageFilter.Kernel((5, 5), flat_kernel, scale=1))
+
+
+def _bayer_resampling(img: Image.Image) -> Image.Image:
+    """
+    Simulate a Bayer CFA (GRBG) where each pixel only captures one color channel
+
+    :param img:
+    :return: An image
+    """
+    w, h = img.size
+    resample = Image.new("RGB", (w, h))
+
+    for y in range(h):
+        for x in range(w):
+            r, g, b = cast(tuple[int, int, int], img.getpixel((x, y)))
+            if y % 2 == 0:
+                if x % 2 == 0:
+                    resample.putpixel((x, y), (0, g, 0))
+                else:
+                    resample.putpixel((x, y), (r, 0, 0))
+            else:
+                if x % 2 == 0:
+                    resample.putpixel((x, y), (0, 0, b))
+                else:
+                    resample.putpixel((x, y), (0, g, 0))
+
+    return resample
+
+
+def _add_noise(img: Image.Image) -> Image.Image:
+    """
+    Add standard normal noise to the image to simulate sensor noise
+
+    :param img:
+    :return: An image
+    """
+    w, h = img.size
+    noisy = Image.new("RGB", (w, h))
+    for y in range(h):
+        for x in range(w):
+            r, g, b = cast(tuple[int, int, int], img.getpixel((x, y)))
+            nr = int(r + random.gauss(0, 1))
+            ng = int(g + random.gauss(0, 1))
+            nb = int(b + random.gauss(0, 1))
+            noisy.putpixel((x, y), (nr, ng, nb))
+
+    return noisy
+
+
+def _clamp(v: int, lo: int, hi: int) -> int:
+    return lo if v < lo else (hi if v > hi else v)
+
+
+def _get_channel(img: Image.Image, x: int, y: int, ch: int, w: int, h: int) -> int:
+    x = _clamp(x, 0, w - 1)
+    y = _clamp(y, 0, h - 1)
+    return cast(tuple[int, int, int], img.getpixel((x, y)))[ch]
+
+
+def _demosaic_bilinear(img: Image.Image) -> Image.Image:
+    """
+    Reconstruct the full RGB image from the Bayer CFA image using bilinear interpolation
+    of the other 2 remaining channels from nearby pixels at each pixel
+
+    :param img:
+    :return: An image
+    """
+    w, h = img.size
+    out = Image.new("RGB", (w, h))
+
+    for y in range(h):
+        for x in range(w):
+            pixel = cast(tuple[int, int, int], img.getpixel((x, y)))
+
+            if y % 2 == 0 and x % 2 == 0:
+                new_r = (
+                    _get_channel(img, x - 1, y, 0, w, h)
+                    + _get_channel(img, x + 1, y, 0, w, h)
+                ) >> 1
+                new_g = pixel[1]
+                new_b = (
+                    _get_channel(img, x, y - 1, 2, w, h)
+                    + _get_channel(img, x, y + 1, 2, w, h)
+                ) >> 1
+
+            elif y % 2 == 0 and x % 2 == 1:
+                new_r = pixel[0]
+                new_g = (
+                    _get_channel(img, x - 1, y, 1, w, h)
+                    + _get_channel(img, x + 1, y, 1, w, h)
+                    + _get_channel(img, x, y - 1, 1, w, h)
+                    + _get_channel(img, x, y + 1, 1, w, h)
+                ) >> 2
+                new_b = (
+                    _get_channel(img, x - 1, y - 1, 2, w, h)
+                    + _get_channel(img, x + 1, y - 1, 2, w, h)
+                    + _get_channel(img, x - 1, y + 1, 2, w, h)
+                    + _get_channel(img, x + 1, y + 1, 2, w, h)
+                ) >> 2
+
+            elif y % 2 == 1 and x % 2 == 0:
+                new_r = (
+                    _get_channel(img, x - 1, y - 1, 0, w, h)
+                    + _get_channel(img, x + 1, y - 1, 0, w, h)
+                    + _get_channel(img, x - 1, y + 1, 0, w, h)
+                    + _get_channel(img, x + 1, y + 1, 0, w, h)
+                ) >> 2
+                new_g = (
+                    _get_channel(img, x - 1, y, 1, w, h)
+                    + _get_channel(img, x + 1, y, 1, w, h)
+                    + _get_channel(img, x, y - 1, 1, w, h)
+                    + _get_channel(img, x, y + 1, 1, w, h)
+                ) >> 2
+                new_b = pixel[2]
+
+            else:
+                new_r = (
+                    _get_channel(img, x, y - 1, 0, w, h)
+                    + _get_channel(img, x, y + 1, 0, w, h)
+                ) >> 1
+                new_g = pixel[1]
+                new_b = (
+                    _get_channel(img, x - 1, y, 2, w, h)
+                    + _get_channel(img, x + 1, y, 2, w, h)
+                ) >> 1
+
+            out.putpixel(
+                (x, y),
+                (_clamp(new_r, 0, 255), _clamp(new_g, 0, 255), _clamp(new_b, 0, 255)),
+            )
+
+    return out
+
+
+def _denoise(img: Image.Image) -> Image.Image:
+    return img.filter(ImageFilter.GaussianBlur(radius=1))
+
+
+def _jpeg_compression(img: Image.Image) -> Image.Image:
+    buffer = io.BytesIO()
+    img.save(buffer, format="JPEG")
+    buffer.seek(0)
+
+    return Image.open(buffer).convert("RGB")