add folder, file nodes and major usability improvements
This commit is contained in:
125
frontend/src/executionGraph.js
Normal file
125
frontend/src/executionGraph.js
Normal file
@@ -0,0 +1,125 @@
|
||||
const DATA_TYPES = new Set([
|
||||
'DATA_FIELD', 'IMAGE', 'LINE', 'MEASURE_TABLE', 'RECORD_TABLE', 'ANY_TABLE',
|
||||
'COORD', 'STATS_SOURCE', 'VALUE_SOURCE', 'COLORMAP', 'SAVE_LAYER', 'FONT', 'FILE_PATH', 'DIRECTORY',
|
||||
]);
|
||||
|
||||
function getInputName(handleId) {
|
||||
return handleId.split('::')[1];
|
||||
}
|
||||
|
||||
function getOutputSlot(handleId) {
|
||||
return parseInt(handleId.split('::')[1], 10);
|
||||
}
|
||||
|
||||
export function getConnectedNodeIds(edges) {
|
||||
const connectedNodeIds = new Set();
|
||||
for (const edge of edges) {
|
||||
connectedNodeIds.add(edge.source);
|
||||
connectedNodeIds.add(edge.target);
|
||||
}
|
||||
return connectedNodeIds;
|
||||
}
|
||||
|
||||
function isPreviewLoadNode(node) {
|
||||
return ['LoadFile', 'LoadDemo'].includes(node?.data?.className);
|
||||
}
|
||||
|
||||
function hasPreviewLoadSelection(node) {
|
||||
if (node?.data?.className === 'LoadFile') {
|
||||
return !!String(node.data?.widgetValues?.filename || '').trim();
|
||||
}
|
||||
if (node?.data?.className === 'LoadDemo') {
|
||||
return !!String(node.data?.widgetValues?.name || '').trim();
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
function getRunnableNodeIds(nodes, edges) {
|
||||
const connectedNodeIds = getConnectedNodeIds(edges);
|
||||
|
||||
const runnableNodeIds = new Set(connectedNodeIds);
|
||||
for (const node of nodes) {
|
||||
if (connectedNodeIds.has(node.id)) continue;
|
||||
if (isPreviewLoadNode(node) && hasPreviewLoadSelection(node)) {
|
||||
runnableNodeIds.add(node.id);
|
||||
}
|
||||
}
|
||||
|
||||
return runnableNodeIds;
|
||||
}
|
||||
|
||||
export function serializeExecutionGraph(nodes, edges, { excludeManualTrigger = false } = {}) {
|
||||
const runnableNodeIds = getRunnableNodeIds(nodes, edges);
|
||||
const prompt = {};
|
||||
|
||||
for (const node of nodes) {
|
||||
if (!runnableNodeIds.has(node.id)) continue;
|
||||
|
||||
const { className, definition, widgetValues } = node.data;
|
||||
if (!definition) continue;
|
||||
if (excludeManualTrigger && definition.manual_trigger) continue;
|
||||
|
||||
const inputs = {};
|
||||
|
||||
const allWidgets = {
|
||||
...(definition.input.required || {}),
|
||||
...(definition.input.optional || {}),
|
||||
};
|
||||
for (const [name, spec] of Object.entries(allWidgets)) {
|
||||
const [type] = Array.isArray(spec) ? spec : [spec];
|
||||
if (DATA_TYPES.has(type)) continue;
|
||||
if (type === 'BUTTON') continue;
|
||||
if (widgetValues[name] !== undefined) {
|
||||
inputs[name] = widgetValues[name];
|
||||
}
|
||||
}
|
||||
|
||||
const incoming = edges.filter((edge) => edge.target === node.id);
|
||||
for (const edge of incoming) {
|
||||
const inputName = getInputName(edge.targetHandle);
|
||||
const outputSlot = getOutputSlot(edge.sourceHandle);
|
||||
inputs[inputName] = [edge.source, outputSlot];
|
||||
}
|
||||
|
||||
prompt[node.id] = { class_type: className, inputs };
|
||||
}
|
||||
|
||||
return prompt;
|
||||
}
|
||||
|
||||
export function getAutoRunnableNodes(nodes, edges) {
|
||||
const runnableNodeIds = getRunnableNodeIds(nodes, edges);
|
||||
return nodes.filter((node) => runnableNodeIds.has(node.id));
|
||||
}
|
||||
|
||||
export function hasBlockingAutoRunInput(node, edges) {
|
||||
const def = node.data?.definition;
|
||||
if (!def || def.manual_trigger) return false;
|
||||
|
||||
const required = def.input.required || {};
|
||||
for (const [name, spec] of Object.entries(required)) {
|
||||
const [type, opts] = Array.isArray(spec) ? spec : [spec, {}];
|
||||
const hiddenByConnectedInput = (() => {
|
||||
const raw = opts?.hide_when_input_connected;
|
||||
if (!raw) return false;
|
||||
const inputs = Array.isArray(raw) ? raw : [raw];
|
||||
return inputs.some((inputName) => edges.some(
|
||||
(edge) => edge.target === node.id && getInputName(edge.targetHandle) === String(inputName)
|
||||
));
|
||||
})();
|
||||
|
||||
if (hiddenByConnectedInput) continue;
|
||||
|
||||
if (type === 'FILE_PICKER' || type === 'FOLDER_PICKER') {
|
||||
if (!node.data.widgetValues?.[name]) return true;
|
||||
continue;
|
||||
}
|
||||
if (!DATA_TYPES.has(type)) continue;
|
||||
const hasEdge = edges.some(
|
||||
(edge) => edge.target === node.id && getInputName(edge.targetHandle) === name
|
||||
);
|
||||
if (!hasEdge) return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
Reference in New Issue
Block a user