76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
class SmartBatchController:
|
|
"""
|
|
Craftopia — Smart Batch Controller 🎛️
|
|
Drives batch switching and prompt index from a single incrementing counter.
|
|
Replaces SELECT BATCH + WELKE REGEL + Math Expression + Math Int.
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"counter": ("INT", {
|
|
"default": 1,
|
|
"min": 1,
|
|
"max": 99999,
|
|
"step": 1,
|
|
"display": "number",
|
|
"control_after_generate": "increment",
|
|
}),
|
|
"batch_1_total": ("INT", {"default": 5, "min": 0, "max": 999, "step": 1, "display": "number"}),
|
|
"batch_2_total": ("INT", {"default": 8, "min": 0, "max": 999, "step": 1, "display": "number"}),
|
|
"batch_3_total": ("INT", {"default": 0, "min": 0, "max": 999, "step": 1, "display": "number"}),
|
|
"batch_4_total": ("INT", {"default": 0, "min": 0, "max": 999, "step": 1, "display": "number"}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", "INT", "INT")
|
|
RETURN_NAMES = ("select", "prompt_index", "total_all")
|
|
FUNCTION = "control"
|
|
CATEGORY = "Craftopia"
|
|
OUTPUT_NODE = True
|
|
|
|
def control(self, counter, batch_1_total, batch_2_total, batch_3_total, batch_4_total):
|
|
totals = [batch_1_total, batch_2_total, batch_3_total, batch_4_total]
|
|
|
|
# Filter out empty batches
|
|
active = [(i, t) for i, t in enumerate(totals) if t > 0]
|
|
total_all = sum(t for _, t in active)
|
|
|
|
if total_all == 0:
|
|
return {"ui": {"text": ["no prompts"]}, "result": (0, 0, 0)}
|
|
|
|
# Wrap counter within total_all (1-based counter → 0-based position)
|
|
position = (counter - 1) % total_all
|
|
|
|
# Find which batch and which prompt within that batch
|
|
cumulative = 0
|
|
batch_index = 0
|
|
prompt_index = 0
|
|
|
|
for i, t in active:
|
|
if position < cumulative + t:
|
|
batch_index = i # 0-based → Switch select
|
|
prompt_index = position - cumulative # 0-based within batch
|
|
break
|
|
cumulative += t
|
|
|
|
# UI label: "B2 — 3 / 8"
|
|
batch_number = batch_index + 1
|
|
batch_total = totals[batch_index]
|
|
prompt_display = prompt_index + 1 # 1-based for display
|
|
label = f"B{batch_number} — {prompt_display} / {batch_total}"
|
|
|
|
# prompt_index +1 because Pose node select is 0-based but we subtract 1 via Math Int
|
|
# Actually return 0-based so user can connect directly to Pose select
|
|
return {"ui": {"text": [label]}, "result": (batch_index, prompt_index, total_all)}
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SmartBatchController": SmartBatchController,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SmartBatchController": "Smart Batch Controller 🎛️",
|
|
}
|