nodes/kijai-ComfyUI-IC-Light
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
kijai
2024-05-10 19:18:16 +03:00
parent bc544816f9
commit e113fb7dfb
4 changed files with 53 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/image.py
+38 -32
View File
@@ -12,51 +12,57 @@ class LightPosition(Enum):
BOTTOM_LEFT = "Bottom Left Light"
BOTTOM_RIGHT = "Bottom Right Light"
def generate_gradient_image(width:int, height:int, lightPosition:LightPosition):
def generate_gradient_image(width:int, height:int, start_color: tuple, end_color: tuple, multiplier: float, lightPosition:LightPosition):
"""
Generate a gradient image with a light source effect.
Parameters:
width (int): Width of the image.
height (int): Height of the image.
lightPosition (str): Position of the light source.
It can be 'Left Light', 'Right Light', 'Top Light', 'Bottom Light',
'Top Left Light', 'Top Right Light', 'Bottom Left Light', 'Bottom Right Light'.
start_color: Starting color RGB of the gradient.
end_color: Ending color RGB of the gradient.
multiplier: Weight of light.
lightPosition (LightPosition): Position of the light source.
Returns:
np.array: 2D gradient image array.
"""
# Create a gradient from 0 to 1 and apply multiplier
if lightPosition == LightPosition.LEFT:
gradient = np.tile(np.linspace(255, 0, width), (height, 1))
gradient = np.tile(np.linspace(0, 1, width)**multiplier, (height, 1))
elif lightPosition == LightPosition.RIGHT:
gradient = np.tile(np.linspace(0, 255, width), (height, 1))
gradient = np.tile(np.linspace(1, 0, width)**multiplier, (height, 1))
elif lightPosition == LightPosition.TOP:
gradient = np.tile(np.linspace(255, 0, height), (width, 1)).T
gradient = np.tile(np.linspace(0, 1, height)**multiplier, (width, 1)).T
elif lightPosition == LightPosition.BOTTOM:
gradient = np.tile(np.linspace(0, 255, height), (width, 1)).T
elif lightPosition == LightPosition.TOP_LEFT:
x = np.linspace(255, 0, width)
y = np.linspace(255, 0, height)
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = (x_mesh + y_mesh) / 2
elif lightPosition == LightPosition.TOP_RIGHT:
x = np.linspace(0, 255, width)
y = np.linspace(255, 0, height)
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = (x_mesh + y_mesh) / 2
elif lightPosition == LightPosition.BOTTOM_LEFT:
x = np.linspace(255, 0, width)
y = np.linspace(0, 255, height)
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = (x_mesh + y_mesh) / 2
gradient = np.tile(np.linspace(1, 0, height)**multiplier, (width, 1)).T
elif lightPosition == LightPosition.BOTTOM_RIGHT:
x = np.linspace(0, 255, width)
y = np.linspace(0, 255, height)
x = np.linspace(1, 0, width)**multiplier
y = np.linspace(1, 0, height)**multiplier
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = (x_mesh + y_mesh) / 2
gradient = np.sqrt(x_mesh**2 + y_mesh**2) / np.sqrt(2.0)
elif lightPosition == LightPosition.BOTTOM_LEFT:
x = np.linspace(0, 1, width)**multiplier
y = np.linspace(1, 0, height)**multiplier
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = np.sqrt(x_mesh**2 + y_mesh**2) / np.sqrt(2.0)
elif lightPosition == LightPosition.TOP_RIGHT:
x = np.linspace(1, 0, width)**multiplier
y = np.linspace(0, 1, height)**multiplier
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = np.sqrt(x_mesh**2 + y_mesh**2) / np.sqrt(2.0)
elif lightPosition == LightPosition.TOP_LEFT:
x = np.linspace(0, 1, width)**multiplier
y = np.linspace(0, 1, height)**multiplier
x_mesh, y_mesh = np.meshgrid(x, y)
gradient = np.sqrt(x_mesh**2 + y_mesh**2) / np.sqrt(2.0)
else:
raise ValueError("Unsupported position. Choose from 'Left Light', 'Right Light', 'Top Light', 'Bottom Light','Top Left Light', 'Top Right Light', 'Bottom Left Light', 'Bottom Right Light'.")
gradient = np.stack((gradient,) * 3, axis=-1).astype(np.uint8)
raise ValueError(f"Unsupported position. Choose from {', '.join([member.value for member in LightPosition])}.")
return gradient
# Interpolate between start_color and end_color based on the gradient
gradient_img = np.zeros((height, width, 3), dtype=np.float32)
for i in range(3):
gradient_img[..., i] = start_color[i] + (end_color[i] - start_color[i]) * gradient
gradient_img = np.clip(gradient_img, 0, 255).astype(np.uint8)
return gradient_img