FilesExpand file tree

master

/

nodes_color.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

52 lines (42 loc) · 1.6 KB

master

/

nodes_color.py

Copy path

Top

File metadata and controls

• Code
• Blame

52 lines (42 loc) · 1.6 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

from typing_extensions import override

from comfy_api.latest import ComfyExtension, io

from comfy_extras.color_util import hex_to_rgb

class ColorToRGBInt(io.ComfyNode):

@classmethod

def define_schema(cls) -> io.Schema:

return io.Schema(

node_id="ColorToRGBInt",

display_name="Color Picker",

category="utilities",

description="Return a color RGB integer value and hexadecimal representation.",

inputs=[

io.Color.Input("color"),

],

outputs=[

io.Int.Output(display_name="rgb_int"),

io.Color.Output(display_name="hex"),

io.Float.Output(display_name="alpha"),

],

)

@classmethod

def execute(cls, color: str) -> io.NodeOutput:

# expect format #RRGGBB or #RRGGBBAA

if len(color) not in (7, 9) or color[0] != "#":

raise ValueError("Color must be in format #RRGGBB or #RRGGBBAA")

try:

int(color[1:], 16)

except ValueError:

raise ValueError("Color must be in format #RRGGBB or #RRGGBBAA") from None

alpha = 1.0

if len(color) == 9:

alpha = int(color[7:9], 16) / 255.0

color = color[:7]

r, g, b = hex_to_rgb(color)

rgb_int = r * 256 * 256 + g * 256 + b

return io.NodeOutput(rgb_int, color, alpha)

class ColorExtension(ComfyExtension):

@override

async def get_node_list(self) -> list[type[io.ComfyNode]]:

return [ColorToRGBInt]

async def comfy_entrypoint() -> ColorExtension:

return ColorExtension()