Update create_print_layover.py

create_print_layover.py

@@ -78,3 +78,100 @@ def create_hard_light_layover(img_in, img_layer, opacity, disable_type_checks: b
     img_out = comp * ratio_rs + img_in_norm[:, :, :3] * (1.0 - ratio_rs)
     img_out = np.nan_to_num(np.dstack((img_out, img_in_norm[:, :, 3])))  # add alpha channel and replace nans
     return img_out * 255.0
+
+def stitch_images(image1, image2, overlap_width):
+    """Stitch two images side by side with overlapping edges."""
+    height = min(image1.shape[0], image2.shape[0])
+    image1 = cv2.resize(image1, (image1.shape[1], height))
+    image2 = cv2.resize(image2, (image2.shape[1], height))
+
+    mask = np.zeros((height, overlap_width), dtype=np.float32)
+    mask[:, :overlap_width] = np.linspace(1, 0, overlap_width)
+    mask = np.dstack([mask] * 3)
+
+    overlap1 = image1[:, -overlap_width:]
+    overlap2 = image2[:, :overlap_width]
+    blended_overlap = overlap1 * mask + overlap2 * (1 - mask)
+    blended_overlap = blended_overlap.astype(np.uint8)
+
+    stitched_image = np.hstack((image1[:, :-overlap_width], blended_overlap, image2[:, overlap_width:]))
+    return stitched_image
+
+
+def tile_image_to_dimensions(image, target_width, target_height, overlap_width):
+    times_width = target_width // (image.shape[1] - overlap_width) + 1
+    times_height = target_height // (image.shape[0] - overlap_width) + 1
+    row_image = image
+    for _ in range(times_width - 1):
+        row_image = stitch_images(row_image, image, overlap_width)
+    final_image = row_image
+    for _ in range(times_height - 1):
+        final_image = np.vstack((final_image, row_image))
+    final_image = final_image[:target_height, :target_width]
+    return final_image
+
+
+def create_image_with_feather_tile(tile_image, width, height, overlap_width):
+    from PIL import Image
+    tile_image = Image.open(tile_image)
+    tile_image.save("tiled_image_pil_converted.png")
+    tile_cv2 = cv2.imread("tiled_image_pil_converted.png")
+    tiled_image = tile_image_to_dimensions(tile_cv2, width, height, overlap_width)
+    cv2.imwrite('tiled_image.png', tiled_image)
+
+
+def stitch_images_with_control(image1, image2, overlap_width, direction='horizontal'):
+    """Stitch two images side by side or top and bottom with overlapping edges."""
+    if direction == 'horizontal':
+        height = min(image1.shape[0], image2.shape[0])
+        image1 = cv2.resize(image1, (image1.shape[1], height))
+        image2 = cv2.resize(image2, (image2.shape[1], height))
+        mask = np.zeros((height, overlap_width), dtype=np.float32)
+        mask[:, :overlap_width] = np.linspace(1, 0, overlap_width)
+        mask = np.dstack([mask] * 3)
+        overlap1 = image1[:, -overlap_width:]
+        overlap2 = image2[:, :overlap_width]
+        blended_overlap = overlap1 * mask + overlap2 * (1 - mask)
+        blended_overlap = blended_overlap.astype(np.uint8)
+        stitched_image = np.hstack((image1[:, :-overlap_width], blended_overlap, image2[:, overlap_width:]))
+    elif direction == 'vertical':
+        width = min(image1.shape[1], image2.shape[1])
+        image1 = cv2.resize(image1, (width, image1.shape[0]))
+        image2 = cv2.resize(image2, (width, image2.shape[0]))
+        mask = np.zeros((overlap_width, width), dtype=np.float32)
+        mask[:overlap_width, :] = np.linspace(1, 0, overlap_width).reshape(-1, 1)
+        mask = np.dstack([mask] * 3)
+        overlap1 = image1[-overlap_width:, :]
+        overlap2 = image2[:overlap_width, :]
+        blended_overlap = overlap1 * mask + overlap2 * (1 - mask)
+        blended_overlap = blended_overlap.astype(np.uint8)
+        stitched_image = np.vstack((image1[:-overlap_width, :], blended_overlap, image2[overlap_width:, :]))
+    else:
+        raise ValueError("Direction must be 'horizontal' or 'vertical'")
+    return stitched_image
+
+
+def color_extract(image_path, new_width, new_height):
+    from PIL import Image
+    format_image = Image.open(image_path)
+    format_image.save("pil_image.png")
+    image = cv2.imread("pil_image.png")
+    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    height, width, _ = image_rgb.shape
+    center_pixel = image_rgb[height // 2, width // 2]
+    color = center_pixel
+    new_image = np.full((new_height, new_width, 3), color, dtype=np.uint8)
+    output_path = 'color_image.jpg'
+    cv2.imwrite(output_path, cv2.cvtColor(new_image, cv2.COLOR_RGB2BGR))
+    print(f'Color image saved at {output_path}')
+
+
+def control_texture(texture_image, direction, overlap, width, height):
+    import os
+    import cv2
+    color_extract(texture_image, width, height)
+    create_image_with_feather_tile(texture_image, width, height, overlap)
+    control_tile_image = stitch_images_with_control('color_image.jpg',
+                                                    'tiled_image.png', overlap, direction)
+    os.remove('tiled_image.png')
+    cv2.imwrite('tiled_image.png', control_tile_image)