Update create_print_layover.py

create_print_layover.py

@@ -79,6 +79,33 @@ def create_hard_light_layover(img_in, img_layer, opacity, disable_type_checks: b
     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 create_soft_light_layover(img_in, img_layer, opacity, disable_type_checks: bool = False):
+    if not disable_type_checks:
+        _fcn_name = 'soft_light'
+        assert_image_format(img_in, _fcn_name, 'img_in')
+        assert_image_format(img_layer, _fcn_name, 'img_layer')
+        assert_opacity(opacity, _fcn_name)
+
+    img_in_norm = img_in / 255.0
+    img_layer_norm = img_layer / 255.0
+
+    ratio = _compose_alpha(img_in_norm, img_layer_norm, opacity)
+
+    # The following code does this:
+    #   multiply = img_in_norm[:, :, :3]*img_layer[:, :, :3]
+    #   screen = 1.0 - (1.0-img_in_norm[:, :, :3])*(1.0-img_layer[:, :, :3])
+    #   comp = (1.0 - img_in_norm[:, :, :3]) * multiply + img_in_norm[:, :, :3] * screen
+    #   ratio_rs = np.reshape(np.repeat(ratio,3),comp.shape)
+    #   img_out = comp*ratio_rs + img_in_norm[:, :, :3] * (1.0-ratio_rs)
+
+    comp = (1.0 - img_in_norm[:, :, :3]) * img_in_norm[:, :, :3] * img_layer_norm[:, :, :3] \
+           + img_in_norm[:, :, :3] * (1.0 - (1.0 - img_in_norm[:, :, :3]) * (1.0 - img_layer_norm[:, :, :3]))
+
+    ratio_rs = np.reshape(np.repeat(ratio, 3), [comp.shape[0], comp.shape[1], comp.shape[2]])
+    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])
@@ -177,3 +204,4 @@ def control_texture(texture_image, direction, overlap, width, height):
                                                     img2, overlap, direction)
     os.remove('tiled_image.png')
     cv2.imwrite('tiled_image_2.png', control_tile_image)
+