55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
"""
|
|
Split Primitives (misc)
|
|
|
|
Combine Primitives の combined 出力だけを受け取り、
|
|
結合時と同じ本数・型の Primitive 出力に戻します。末尾に length（INT）を付けます。
|
|
"""
|
|
|
|
from .utils import (
|
|
ByPassTypeTuple,
|
|
SplitReturnTypes,
|
|
combined_primitives_type,
|
|
get_category,
|
|
get_name,
|
|
unpack_combined_primitives,
|
|
)
|
|
|
|
class MiscSplitPrimitives:
|
|
"""combined を Primitive 出力列に分解するノード。"""
|
|
|
|
NAME = get_name("Split Primitives")
|
|
CATEGORY = get_category()
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls): # pylint: disable=invalid-name
|
|
return {
|
|
"required": {
|
|
"combined": (combined_primitives_type,),
|
|
},
|
|
}
|
|
|
|
# 全出力スロットを any_type として検証（2 番目以降に "INT" が誤適用されないよう専用クラスを使用）
|
|
RETURN_TYPES = SplitReturnTypes()
|
|
RETURN_NAMES = ByPassTypeTuple(("primitive_01", "length"))
|
|
FUNCTION = "split"
|
|
|
|
def split(self, combined):
|
|
"""
|
|
combined を Primitive 値の配列に戻し、末尾に length を付ける。
|
|
|
|
空データのときは length だけを返す（ComfyUI の可変出力と整合させるため）。
|
|
"""
|
|
values, _types, length = unpack_combined_primitives(combined)
|
|
if not values:
|
|
return (length,)
|
|
return tuple(list(values) + [length])
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
MiscSplitPrimitives.NAME: MiscSplitPrimitives,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
MiscSplitPrimitives.NAME: MiscSplitPrimitives.NAME,
|
|
}
|