🟧 CR Starburst Colors¶
Documentation¶
- Class name:
CR Starburst Colors - Category:
🧩 Comfyroll Studio/👾 Graphics/🌈 Pattern - Output node:
False
The CR Starburst Colors node is designed to generate starburst patterns with alternating colors. It utilizes matplotlib to set up a plot and draw triangles in a starburst formation, allowing for customization of the number of triangles, bounding box size, and color scheme. This node is ideal for creating visually striking backgrounds or elements in graphics projects.
Input types¶
Required¶
width- Sets the width of the plot area for the starburst pattern, influencing the overall size of the generated graphic.
- Comfy dtype:
INT - Python dtype:
int
height- Determines the height of the plot area, working alongside 'width' to define the dimensions of the starburst pattern.
- Comfy dtype:
INT - Python dtype:
int
num_triangles- Specifies the number of triangles to draw in the starburst pattern, allowing for control over the pattern's complexity and density.
- Comfy dtype:
INT - Python dtype:
int
color_1- Specifies the first color in the alternating color scheme of the starburst pattern. This can be a predefined color or a custom hex value if 'custom' is selected.
- Comfy dtype:
COMBO[STRING] - Python dtype:
str
color_2- Defines the second color in the alternating color scheme, complementing the first color. Like 'color_1', it supports predefined colors or custom hex values.
- Comfy dtype:
COMBO[STRING] - Python dtype:
str
center_x- Sets the x-coordinate of the center point for the starburst pattern.
- Comfy dtype:
INT - Python dtype:
int
center_y- Sets the y-coordinate of the center point for the starburst pattern.
- Comfy dtype:
INT - Python dtype:
int
rotation- Determines the rotation angle of the starburst pattern, allowing for dynamic visual effects.
- Comfy dtype:
FLOAT - Python dtype:
float
bbox_factor- Controls the size of the bounding box for the starburst pattern, affecting how tightly the triangles are packed together.
- Comfy dtype:
FLOAT - Python dtype:
float
Optional¶
color1_hex- Specifies the hex value for the first color, used when 'color_1' is set to 'custom'.
- Comfy dtype:
STRING - Python dtype:
str
color2_hex- Specifies the hex value for the second color, used when 'color_2' is set to 'custom'.
- Comfy dtype:
STRING - Python dtype:
str
Output types¶
IMAGE- Comfy dtype:
IMAGE - Generates an image of the starburst pattern with the specified parameters.
- Python dtype:
torch.Tensor
- Comfy dtype:
show_help- Comfy dtype:
STRING - Provides a URL to documentation or help related to the starburst pattern generation.
- Python dtype:
str
- Comfy dtype:
Usage tips¶
- Infra type:
CPU - Common nodes: unknown
Source code¶
class CR_StarburstColors:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"width": ("INT", {"default": 512, "min": 64, "max": 4096}),
"height": ("INT", {"default": 512, "min": 64, "max": 4096}),
"num_triangles": ("INT", {"default": 6, "min": 1, "max": 512}),
"color_1": (COLORS,),
"color_2": (COLORS,),
"center_x": ("INT", {"default": 0, "min": 0, "max": 512}),
"center_y": ("INT", {"default": 0, "min": 0, "max": 512}),
"rotation": ("FLOAT", {"default": 0, "min": 0, "max": 720}),
"bbox_factor": ("FLOAT", {"default": 2, "min": 0, "max": 2, "step": .01}),
},
"optional": {
"color1_hex": ("STRING", {"multiline": False, "default": "#000000"}),
"color2_hex": ("STRING", {"multiline": False, "default": "#000000"}),
}
}
RETURN_TYPES = ("IMAGE", "STRING", )
RETURN_NAMES = ("IMAGE", "show_help", )
FUNCTION = "draw"
CATEGORY = icons.get("Comfyroll/Graphics/Pattern")
def draw(self, width, height, num_triangles, color_1, color_2,
center_x, center_y, bbox_factor, rotation=0,
color1_hex='#000000', color2_hex='#000000'):
# Get RGB values for the text color
if color_1 == "custom":
color_1 = color1_hex
else:
color_1 = color_1
if color_2 == "custom":
color_2 = color2_hex
else:
color_2 = color_2
# Set up the plot
fig, ax = plt.subplots()
x = width/100
y = height/100
fig, ax = plt.subplots(figsize=(x,y))
plt.xlim(-x/2, x/2)
plt.ylim(-y/2, y/2)
plt.axis('off')
plt.tight_layout(pad=0, w_pad=0, h_pad=0)
plt.autoscale(False)
# Set the size of the starburst bounding box in x and y dimensions
box_width = bbox_factor * x
box_height = bbox_factor * y
# Initialize a color list for alternating colors
colors = [color_1, color_2]
tri = num_triangles
# Draw the starburst triangles with alternating colors and square pattern
for i in range(tri):
# Calculate the endpoints of the triangle with varying length
x1 = center_x/100
y1 = center_y/100
x2_unrotated = (box_width / 2) * np.cos(np.radians(i * 360 / tri))
y2_unrotated = (box_height / 2) * np.sin(np.radians(i * 360 / tri))
x3_unrotated = (box_width / 2) * np.cos(np.radians((i + 1) * 360 / tri))
y3_unrotated = (box_height / 2) * np.sin(np.radians((i + 1) * 360 / tri))
#apply rotation transform
x2 = x2_unrotated * np.cos(np.radians(rotation)) - y2_unrotated * np.sin(np.radians(rotation))
y2 = x2_unrotated * np.sin(np.radians(rotation)) + y2_unrotated * np.cos(np.radians(rotation))
x3 = x3_unrotated * np.cos(np.radians(rotation)) - y3_unrotated * np.sin(np.radians(rotation))
y3 = x3_unrotated * np.sin(np.radians(rotation)) + y3_unrotated * np.cos(np.radians(rotation))
# Plot the triangle with alternating colors
ax.fill([x1, x2, x3, x1], [y1, y2, y3, y1], color=colors[i % 2])
img_buf = io.BytesIO()
plt.savefig(img_buf, format='png')
img = Image.open(img_buf)
image_out = pil2tensor(img.convert("RGB"))
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/Pattern-Nodes#cr-starburst-colors"
return (image_out, show_help, )