118 lines
2.9 KiB
Python
118 lines
2.9 KiB
Python
import logging
|
|
import random
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DarkFaceIndexGenerator(object):
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"pad_to": (
|
|
"INT",
|
|
{"forceInput": True},
|
|
),
|
|
},
|
|
"required": {
|
|
"number_of_faces": (
|
|
"INT",
|
|
{"forceInput": True},
|
|
),
|
|
"randomize": (
|
|
"BOOLEAN",
|
|
{"default": True},
|
|
),
|
|
"randomize_after_padding": (
|
|
"BOOLEAN",
|
|
{"default": True},
|
|
),
|
|
"seed": (
|
|
"INT",
|
|
{
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 0xFFFFFFFFFFFFFFFF,
|
|
"forceInput": True,
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
FUNCTION = "face_index_generator"
|
|
|
|
CATEGORY = "DarkPrompt"
|
|
|
|
def face_index_generator(
|
|
self,
|
|
number_of_faces,
|
|
randomize=True,
|
|
randomize_after_padding=True,
|
|
seed=1,
|
|
pad_to=None,
|
|
**kwargs
|
|
):
|
|
faces = []
|
|
for i in range(0, number_of_faces):
|
|
faces.append(str(i))
|
|
|
|
if randomize and not randomize_after_padding:
|
|
random.seed(seed)
|
|
random.shuffle(faces)
|
|
|
|
# ReActor is pretty dumb with faces, so we need to ensure every source
|
|
# face also has an input face even if they don't exist, otherwise it
|
|
# won't swap any faces
|
|
|
|
if pad_to and number_of_faces < pad_to:
|
|
for i in range(number_of_faces, pad_to):
|
|
faces.append(str(i))
|
|
|
|
if randomize and randomize_after_padding:
|
|
random.seed(seed)
|
|
random.shuffle(faces)
|
|
|
|
return (",".join(faces),)
|
|
|
|
|
|
class DarkFaceIndexShuffle(object):
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {},
|
|
"required": {
|
|
"faces_index_csv": (
|
|
"STRING",
|
|
{},
|
|
),
|
|
"seed": (
|
|
"INT",
|
|
{
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 0xFFFFFFFFFFFFFFFF,
|
|
"forceInput": True,
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING",)
|
|
FUNCTION = "face_index_shuffle"
|
|
|
|
CATEGORY = "DarkPrompt"
|
|
|
|
def face_index_shuffle(self, faces_index_csv, seed=1, **kwargs):
|
|
face_index = faces_index_csv.split(",")
|
|
|
|
random.seed(seed)
|
|
random.shuffle(face_index)
|
|
return (",".join(face_index),)
|