Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
01f58377c2
|
|||
|
6ef5dc28ed
|
|||
|
3450d70047
|
|||
|
731b9e9b1e
|
|||
|
72f07d0a50
|
|||
|
a56e8f445e
|
@@ -5,6 +5,7 @@ ENV RUSTUP_HOME=/usr/local/rustup \
|
|||||||
PATH=/usr/local/cargo/bin:$PATH
|
PATH=/usr/local/cargo/bin:$PATH
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y \
|
RUN apt-get update && apt-get install -y \
|
||||||
|
openssh-client \
|
||||||
ca-certificates=20230311+deb12u1 \
|
ca-certificates=20230311+deb12u1 \
|
||||||
gpg=2.2.40-1.1+deb12u2 \
|
gpg=2.2.40-1.1+deb12u2 \
|
||||||
gpg-agent=2.2.40-1.1+deb12u2 \
|
gpg-agent=2.2.40-1.1+deb12u2 \
|
||||||
|
|||||||
@@ -183,7 +183,7 @@
|
|||||||
activeNodeId = node.id;
|
activeNodeId = node.id;
|
||||||
}}
|
}}
|
||||||
>
|
>
|
||||||
{node.id.split('/').at(-1)}
|
{node.meta?.title ?? node.id.split('/').at(-1)}
|
||||||
</div>
|
</div>
|
||||||
{/each}
|
{/each}
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
import { describe, expect, it } from 'vitest';
|
import { assert, describe, expect, it } from 'vitest';
|
||||||
import { GraphManager } from './graph-manager.svelte';
|
import { GraphManager } from './graph-manager.svelte';
|
||||||
import {
|
import {
|
||||||
createMockNodeRegistry,
|
createMockNodeRegistry,
|
||||||
@@ -9,257 +9,399 @@ import {
|
|||||||
mockVec3OutputNode
|
mockVec3OutputNode
|
||||||
} from './test-utils';
|
} from './test-utils';
|
||||||
|
|
||||||
describe('GraphManager', () => {
|
describe('groupNodes', () => {
|
||||||
describe('getPossibleSockets', () => {
|
it('should not do anything if no nodes are selected', () => {
|
||||||
describe('when dragging an output socket', () => {
|
const registry = createMockNodeRegistry([
|
||||||
it('should return compatible input sockets based on type', () => {
|
mockFloatOutputNode,
|
||||||
const registry = createMockNodeRegistry([
|
mockFloatInputNode,
|
||||||
mockFloatOutputNode,
|
mockGeometryOutputNode,
|
||||||
mockFloatInputNode,
|
mockPathInputNode
|
||||||
mockGeometryOutputNode,
|
]);
|
||||||
mockPathInputNode
|
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
const floatInputNode = manager.createNode({
|
const floatInputNode = manager.createNode({
|
||||||
type: 'test/node/input',
|
type: 'test/node/input',
|
||||||
position: [100, 100],
|
position: [100, 100],
|
||||||
props: {}
|
props: {}
|
||||||
});
|
});
|
||||||
|
|
||||||
const floatOutputNode = manager.createNode({
|
assert.isDefined(floatInputNode);
|
||||||
type: 'test/node/output',
|
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(floatInputNode).toBeDefined();
|
const floatOutputNode = manager.createNode({
|
||||||
expect(floatOutputNode).toBeDefined();
|
type: 'test/node/output',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
assert.isDefined(floatOutputNode);
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
const edge = manager.createEdge(floatInputNode, 0, floatOutputNode, 'input');
|
||||||
node: floatOutputNode!,
|
assert.isDefined(edge);
|
||||||
index: 0,
|
manager.save();
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(possibleSockets.length).toBe(1);
|
manager.groupNodes([]);
|
||||||
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
|
||||||
expect(socketNodeIds).toContain(floatInputNode!.id);
|
const graph = manager.serialize();
|
||||||
|
expect(graph.nodes.length).toBe(2);
|
||||||
|
expect(graph.edges.length).toBe(1);
|
||||||
|
expect(graph.groups.length).toBe(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should group selected nodes and create a group node', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode,
|
||||||
|
mockGeometryOutputNode,
|
||||||
|
mockPathInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const floatInputNode = manager.createNode({
|
||||||
|
type: 'test/node/input',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
assert.isDefined(floatInputNode);
|
||||||
|
|
||||||
|
const floatOutputNode = manager.createNode({
|
||||||
|
type: 'test/node/output',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
assert.isDefined(floatOutputNode);
|
||||||
|
|
||||||
|
const edge = manager.createEdge(floatInputNode, 0, floatOutputNode, 'input');
|
||||||
|
assert.isDefined(edge);
|
||||||
|
manager.save();
|
||||||
|
|
||||||
|
const groupNode = manager.groupNodes([floatInputNode.id]);
|
||||||
|
assert.isDefined(groupNode);
|
||||||
|
|
||||||
|
const graph = manager.serialize();
|
||||||
|
|
||||||
|
expect(graph.nodes.map(n => n.id), 'graph to contain group node').to.contain(groupNode.id);
|
||||||
|
expect(graph.groups[0].nodes.map(n => n.id), 'group graph to contain float node').to.contain(
|
||||||
|
floatInputNode.id
|
||||||
|
);
|
||||||
|
expect(graph.nodes.map(n => n.id)).not.to.contain(floatInputNode.id);
|
||||||
|
|
||||||
|
expect(graph.nodes.length).toBe(2);
|
||||||
|
expect(graph.edges.length).toBe(1);
|
||||||
|
expect(graph.groups.length).toBe(1);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should rewire external edges when grouping a middle node in a chain', () => {
|
||||||
|
const registry = createMockNodeRegistry([mockFloatOutputNode, mockFloatInputNode]);
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
// A → B → C (float chain: output → middle → input)
|
||||||
|
const nodeA = manager.createNode({ type: 'test/node/output', position: [0, 0], props: {} });
|
||||||
|
const nodeB = manager.createNode({ type: 'test/node/output', position: [100, 0], props: {} });
|
||||||
|
const nodeC = manager.createNode({ type: 'test/node/input', position: [200, 0], props: {} });
|
||||||
|
|
||||||
|
assert.isDefined(nodeA);
|
||||||
|
assert.isDefined(nodeB);
|
||||||
|
assert.isDefined(nodeC);
|
||||||
|
|
||||||
|
manager.createEdge(nodeA, 0, nodeB, 'input');
|
||||||
|
manager.createEdge(nodeB, 0, nodeC, 'value');
|
||||||
|
|
||||||
|
const groupNode = manager.groupNodes([nodeB.id]);
|
||||||
|
assert.isDefined(groupNode);
|
||||||
|
|
||||||
|
const graph = manager.serialize();
|
||||||
|
|
||||||
|
// Top-level: A, C, groupNode — B is gone
|
||||||
|
expect(graph.nodes.length, 'top-level node count').toBe(3);
|
||||||
|
const topLevelIds = graph.nodes.map(n => n.id);
|
||||||
|
expect(topLevelIds).toContain(nodeA.id);
|
||||||
|
expect(topLevelIds).toContain(nodeC.id);
|
||||||
|
expect(topLevelIds).toContain(groupNode.id);
|
||||||
|
expect(topLevelIds).not.toContain(nodeB.id);
|
||||||
|
|
||||||
|
// Both original edges survive, now routing through the group node
|
||||||
|
expect(graph.edges.length, 'edge count unchanged').toBe(2);
|
||||||
|
const edgeSources = graph.edges.map(e => e[0]);
|
||||||
|
const edgeTargets = graph.edges.map(e => e[2]);
|
||||||
|
expect(edgeTargets).toContain(groupNode.id); // A → groupNode
|
||||||
|
expect(edgeSources).toContain(groupNode.id); // groupNode → C
|
||||||
|
|
||||||
|
// One group definition was created
|
||||||
|
expect(graph.groups.length).toBe(1);
|
||||||
|
const group = graph.groups[0];
|
||||||
|
|
||||||
|
// Group contains B plus the two boundary nodes
|
||||||
|
const groupNodeIds = group.nodes.map(n => n.id);
|
||||||
|
expect(groupNodeIds).toContain(nodeB.id);
|
||||||
|
const inputBoundary = group.nodes.find(n => n.type === '__internal/group/input');
|
||||||
|
const outputBoundary = group.nodes.find(n => n.type === '__internal/group/output');
|
||||||
|
expect(inputBoundary, 'group input boundary node').toBeDefined();
|
||||||
|
expect(outputBoundary, 'group output boundary node').toBeDefined();
|
||||||
|
|
||||||
|
// Group declares one input slot and one output slot
|
||||||
|
expect(Object.keys(group.inputs ?? {}).length, 'group input count').toBe(1);
|
||||||
|
expect(group.outputs?.length, 'group output count').toBe(1);
|
||||||
|
|
||||||
|
// Internal edges wire: inputBoundary → B → outputBoundary
|
||||||
|
expect(group.edges.length, 'internal edge count').toBe(2);
|
||||||
|
const internalSources = group.edges.map(e => e[0]);
|
||||||
|
const internalTargets = group.edges.map(e => e[2]);
|
||||||
|
expect(internalTargets).toContain(nodeB.id);
|
||||||
|
expect(internalSources).toContain(nodeB.id);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('getPossibleSockets', () => {
|
||||||
|
describe('when dragging an output socket', () => {
|
||||||
|
it('should return compatible input sockets based on type', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode,
|
||||||
|
mockGeometryOutputNode,
|
||||||
|
mockPathInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const floatInputNode = manager.createNode({
|
||||||
|
type: 'test/node/input',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should exclude self node from possible sockets', () => {
|
const floatOutputNode = manager.createNode({
|
||||||
const registry = createMockNodeRegistry([
|
type: 'test/node/output',
|
||||||
mockFloatOutputNode,
|
position: [0, 0],
|
||||||
mockFloatInputNode
|
props: {}
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
|
||||||
|
|
||||||
const floatInputNode = manager.createNode({
|
|
||||||
type: 'test/node/input',
|
|
||||||
position: [100, 100],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(floatInputNode).toBeDefined();
|
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
|
||||||
node: floatInputNode!,
|
|
||||||
index: 'value',
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
|
||||||
expect(socketNodeIds).not.toContain(floatInputNode!.id);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should exclude parent nodes from possible sockets when dragging output', () => {
|
expect(floatInputNode).toBeDefined();
|
||||||
const registry = createMockNodeRegistry([
|
expect(floatOutputNode).toBeDefined();
|
||||||
mockFloatOutputNode,
|
|
||||||
mockFloatInputNode
|
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: floatOutputNode!,
|
||||||
const parentNode = manager.createNode({
|
index: 0,
|
||||||
type: 'test/node/output',
|
position: [0, 0]
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const childNode = manager.createNode({
|
|
||||||
type: 'test/node/input',
|
|
||||||
position: [100, 100],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(parentNode).toBeDefined();
|
|
||||||
expect(childNode).toBeDefined();
|
|
||||||
|
|
||||||
if (parentNode && childNode) {
|
|
||||||
manager.createEdge(parentNode, 0, childNode, 'value');
|
|
||||||
}
|
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
|
||||||
node: parentNode!,
|
|
||||||
index: 0,
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
|
||||||
expect(socketNodeIds).not.toContain(childNode!.id);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return sockets compatible with accepts property', () => {
|
expect(possibleSockets.length).toBe(1);
|
||||||
const registry = createMockNodeRegistry([
|
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
||||||
mockGeometryOutputNode,
|
expect(socketNodeIds).toContain(floatInputNode!.id);
|
||||||
mockPathInputNode
|
});
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
it('should exclude self node from possible sockets', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
const geometryOutputNode = manager.createNode({
|
const manager = new GraphManager(registry);
|
||||||
type: 'test/node/geometry',
|
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const pathInputNode = manager.createNode({
|
const floatInputNode = manager.createNode({
|
||||||
type: 'test/node/path',
|
type: 'test/node/input',
|
||||||
position: [100, 100],
|
position: [100, 100],
|
||||||
props: {}
|
props: {}
|
||||||
});
|
|
||||||
|
|
||||||
expect(geometryOutputNode).toBeDefined();
|
|
||||||
expect(pathInputNode).toBeDefined();
|
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
|
||||||
node: geometryOutputNode!,
|
|
||||||
index: 0,
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
|
||||||
expect(socketNodeIds).toContain(pathInputNode!.id);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return empty array when no compatible sockets exist', () => {
|
expect(floatInputNode).toBeDefined();
|
||||||
const registry = createMockNodeRegistry([
|
|
||||||
mockVec3OutputNode,
|
|
||||||
mockFloatInputNode
|
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: floatInputNode!,
|
||||||
const vec3OutputNode = manager.createNode({
|
index: 'value',
|
||||||
type: 'test/node/vec3',
|
position: [0, 0]
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const floatInputNode = manager.createNode({
|
|
||||||
type: 'test/node/input',
|
|
||||||
position: [100, 100],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(vec3OutputNode).toBeDefined();
|
|
||||||
expect(floatInputNode).toBeDefined();
|
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
|
||||||
node: vec3OutputNode!,
|
|
||||||
index: 0,
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
|
||||||
expect(socketNodeIds).not.toContain(floatInputNode!.id);
|
|
||||||
expect(possibleSockets.length).toBe(0);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return socket info with correct socket key for inputs', () => {
|
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
||||||
const registry = createMockNodeRegistry([
|
expect(socketNodeIds).not.toContain(floatInputNode!.id);
|
||||||
mockFloatOutputNode,
|
});
|
||||||
mockFloatInputNode
|
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
it('should exclude parent nodes from possible sockets when dragging output', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
const floatOutputNode = manager.createNode({
|
const manager = new GraphManager(registry);
|
||||||
type: 'test/node/output',
|
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const floatInputNode = manager.createNode({
|
const parentNode = manager.createNode({
|
||||||
type: 'test/node/input',
|
type: 'test/node/output',
|
||||||
position: [100, 100],
|
position: [0, 0],
|
||||||
props: {}
|
props: {}
|
||||||
});
|
|
||||||
|
|
||||||
expect(floatOutputNode).toBeDefined();
|
|
||||||
expect(floatInputNode).toBeDefined();
|
|
||||||
|
|
||||||
const possibleSockets = manager.getPossibleSockets({
|
|
||||||
node: floatOutputNode!,
|
|
||||||
index: 0,
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
const matchingSocket = possibleSockets.find(([node]) => node.id === floatInputNode!.id);
|
|
||||||
expect(matchingSocket).toBeDefined();
|
|
||||||
expect(matchingSocket![1]).toBe('value');
|
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return multiple compatible sockets', () => {
|
const childNode = manager.createNode({
|
||||||
const registry = createMockNodeRegistry([
|
type: 'test/node/input',
|
||||||
mockFloatOutputNode,
|
position: [100, 100],
|
||||||
mockFloatInputNode,
|
props: {}
|
||||||
mockGeometryOutputNode,
|
|
||||||
mockPathInputNode
|
|
||||||
]);
|
|
||||||
|
|
||||||
const manager = new GraphManager(registry);
|
|
||||||
|
|
||||||
const floatOutputNode = manager.createNode({
|
|
||||||
type: 'test/node/output',
|
|
||||||
position: [0, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const geometryOutputNode = manager.createNode({
|
|
||||||
type: 'test/node/geometry',
|
|
||||||
position: [200, 0],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const floatInputNode = manager.createNode({
|
|
||||||
type: 'test/node/input',
|
|
||||||
position: [100, 100],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
const pathInputNode = manager.createNode({
|
|
||||||
type: 'test/node/path',
|
|
||||||
position: [300, 100],
|
|
||||||
props: {}
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(floatOutputNode).toBeDefined();
|
|
||||||
expect(geometryOutputNode).toBeDefined();
|
|
||||||
expect(floatInputNode).toBeDefined();
|
|
||||||
expect(pathInputNode).toBeDefined();
|
|
||||||
|
|
||||||
const possibleSocketsForFloat = manager.getPossibleSockets({
|
|
||||||
node: floatOutputNode!,
|
|
||||||
index: 0,
|
|
||||||
position: [0, 0]
|
|
||||||
});
|
|
||||||
|
|
||||||
expect(possibleSocketsForFloat.length).toBe(1);
|
|
||||||
expect(possibleSocketsForFloat.map(([n]) => n.id)).toContain(floatInputNode!.id);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
|
expect(parentNode).toBeDefined();
|
||||||
|
expect(childNode).toBeDefined();
|
||||||
|
|
||||||
|
if (parentNode && childNode) {
|
||||||
|
manager.createEdge(parentNode, 0, childNode, 'value');
|
||||||
|
}
|
||||||
|
|
||||||
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: parentNode!,
|
||||||
|
index: 0,
|
||||||
|
position: [0, 0]
|
||||||
|
});
|
||||||
|
|
||||||
|
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
||||||
|
expect(socketNodeIds).not.toContain(childNode!.id);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return sockets compatible with accepts property', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockGeometryOutputNode,
|
||||||
|
mockPathInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const geometryOutputNode = manager.createNode({
|
||||||
|
type: 'test/node/geometry',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const pathInputNode = manager.createNode({
|
||||||
|
type: 'test/node/path',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(geometryOutputNode).toBeDefined();
|
||||||
|
expect(pathInputNode).toBeDefined();
|
||||||
|
|
||||||
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: geometryOutputNode!,
|
||||||
|
index: 0,
|
||||||
|
position: [0, 0]
|
||||||
|
});
|
||||||
|
|
||||||
|
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
||||||
|
expect(socketNodeIds).toContain(pathInputNode!.id);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return empty array when no compatible sockets exist', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockVec3OutputNode,
|
||||||
|
mockFloatInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const vec3OutputNode = manager.createNode({
|
||||||
|
type: 'test/node/vec3',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const floatInputNode = manager.createNode({
|
||||||
|
type: 'test/node/input',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(vec3OutputNode).toBeDefined();
|
||||||
|
expect(floatInputNode).toBeDefined();
|
||||||
|
|
||||||
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: vec3OutputNode!,
|
||||||
|
index: 0,
|
||||||
|
position: [0, 0]
|
||||||
|
});
|
||||||
|
|
||||||
|
const socketNodeIds = possibleSockets.map(([node]) => node.id);
|
||||||
|
expect(socketNodeIds).not.toContain(floatInputNode!.id);
|
||||||
|
expect(possibleSockets.length).toBe(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return socket info with correct socket key for inputs', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const floatOutputNode = manager.createNode({
|
||||||
|
type: 'test/node/output',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const floatInputNode = manager.createNode({
|
||||||
|
type: 'test/node/input',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(floatOutputNode).toBeDefined();
|
||||||
|
expect(floatInputNode).toBeDefined();
|
||||||
|
|
||||||
|
const possibleSockets = manager.getPossibleSockets({
|
||||||
|
node: floatOutputNode!,
|
||||||
|
index: 0,
|
||||||
|
position: [0, 0]
|
||||||
|
});
|
||||||
|
|
||||||
|
const matchingSocket = possibleSockets.find(([node]) => node.id === floatInputNode!.id);
|
||||||
|
expect(matchingSocket).toBeDefined();
|
||||||
|
expect(matchingSocket![1]).toBe('value');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return multiple compatible sockets', () => {
|
||||||
|
const registry = createMockNodeRegistry([
|
||||||
|
mockFloatOutputNode,
|
||||||
|
mockFloatInputNode,
|
||||||
|
mockGeometryOutputNode,
|
||||||
|
mockPathInputNode
|
||||||
|
]);
|
||||||
|
|
||||||
|
const manager = new GraphManager(registry);
|
||||||
|
|
||||||
|
const floatOutputNode = manager.createNode({
|
||||||
|
type: 'test/node/output',
|
||||||
|
position: [0, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const geometryOutputNode = manager.createNode({
|
||||||
|
type: 'test/node/geometry',
|
||||||
|
position: [200, 0],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const floatInputNode = manager.createNode({
|
||||||
|
type: 'test/node/input',
|
||||||
|
position: [100, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
const pathInputNode = manager.createNode({
|
||||||
|
type: 'test/node/path',
|
||||||
|
position: [300, 100],
|
||||||
|
props: {}
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(floatOutputNode).toBeDefined();
|
||||||
|
expect(geometryOutputNode).toBeDefined();
|
||||||
|
expect(floatInputNode).toBeDefined();
|
||||||
|
expect(pathInputNode).toBeDefined();
|
||||||
|
|
||||||
|
const possibleSocketsForFloat = manager.getPossibleSockets({
|
||||||
|
node: floatOutputNode!,
|
||||||
|
index: 0,
|
||||||
|
position: [0, 0]
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(possibleSocketsForFloat.length).toBe(1);
|
||||||
|
expect(possibleSocketsForFloat.map(([n]) => n.id)).toContain(floatInputNode!.id);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
import throttle from '$lib/helpers/throttle';
|
import throttle from '$lib/helpers/throttle';
|
||||||
import { RemoteNodeRegistry } from '$lib/node-registry/index';
|
import { RemoteNodeRegistry } from '$lib/node-registry/index';
|
||||||
import type {
|
import type {
|
||||||
|
Box,
|
||||||
Edge,
|
Edge,
|
||||||
Graph,
|
Graph,
|
||||||
|
GroupDefinition,
|
||||||
NodeDefinition,
|
NodeDefinition,
|
||||||
NodeId,
|
NodeId,
|
||||||
NodeInput,
|
NodeInput,
|
||||||
@@ -67,7 +69,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
status = $state<'loading' | 'idle' | 'error'>();
|
status = $state<'loading' | 'idle' | 'error'>();
|
||||||
loaded = false;
|
loaded = false;
|
||||||
|
|
||||||
graph: Graph = { id: 0, nodes: [], edges: [] };
|
graph: Graph = { id: 0, nodes: [], edges: [], groups: [] };
|
||||||
id = $state(0);
|
id = $state(0);
|
||||||
|
|
||||||
nodes = new SvelteMap<number, NodeInstance>();
|
nodes = new SvelteMap<number, NodeInstance>();
|
||||||
@@ -110,10 +112,29 @@ export class GraphManager extends EventEmitter<{
|
|||||||
edge[2].id,
|
edge[2].id,
|
||||||
edge[3]
|
edge[3]
|
||||||
]) as Graph['edges'];
|
]) as Graph['edges'];
|
||||||
|
|
||||||
|
const groups = this.graph.groups?.map((group) => {
|
||||||
|
const groupNodes = group.nodes.map((node) => ({
|
||||||
|
id: node.id,
|
||||||
|
position: [...node.position],
|
||||||
|
type: node.type,
|
||||||
|
props: node.props
|
||||||
|
})) as NodeInstance[];
|
||||||
|
|
||||||
|
return {
|
||||||
|
id: group.id,
|
||||||
|
inputs: group.inputs,
|
||||||
|
outputs: group.outputs,
|
||||||
|
nodes: groupNodes,
|
||||||
|
edges: group.edges
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
const serialized = {
|
const serialized = {
|
||||||
id: this.graph.id,
|
id: this.graph.id,
|
||||||
settings: $state.snapshot(this.settings),
|
settings: $state.snapshot(this.settings),
|
||||||
meta: $state.snapshot(this.graph.meta),
|
meta: $state.snapshot(this.graph.meta),
|
||||||
|
groups,
|
||||||
nodes,
|
nodes,
|
||||||
edges
|
edges
|
||||||
};
|
};
|
||||||
@@ -305,19 +326,36 @@ export class GraphManager extends EventEmitter<{
|
|||||||
const a = performance.now();
|
const a = performance.now();
|
||||||
|
|
||||||
this.loaded = false;
|
this.loaded = false;
|
||||||
|
graph.groups ??= [];
|
||||||
this.graph = graph;
|
this.graph = graph;
|
||||||
this.status = 'loading';
|
this.status = 'loading';
|
||||||
this.id = graph.id;
|
this.id = graph.id;
|
||||||
|
|
||||||
logger.info('loading graph', { nodes: graph.nodes, edges: graph.edges, id: graph.id });
|
logger.info(
|
||||||
|
'loading graph',
|
||||||
|
{ nodes: graph.nodes, edges: graph.edges, id: graph.id }
|
||||||
|
);
|
||||||
|
|
||||||
|
const nodeIds = Array
|
||||||
|
.from(
|
||||||
|
new SvelteSet(
|
||||||
|
[
|
||||||
|
...graph.nodes,
|
||||||
|
graph?.groups?.map(g => g.nodes).flat()
|
||||||
|
]
|
||||||
|
.filter(n => n && 'type' in n)
|
||||||
|
.map((n) => n.type)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
.filter(n => !n.startsWith('__internal/'));
|
||||||
|
|
||||||
const nodeIds = Array.from(new SvelteSet([...graph.nodes.map((n) => n.type)]));
|
|
||||||
await this.registry.load(nodeIds);
|
await this.registry.load(nodeIds);
|
||||||
|
|
||||||
// Fetch all nodes from all collections of the loaded nodes
|
// Fetch all nodes from all collections of the loaded nodes
|
||||||
const allCollections = new SvelteSet<`${string}/${string}`>();
|
const allCollections = new SvelteSet<`${string}/${string}`>();
|
||||||
for (const id of nodeIds) {
|
for (const id of nodeIds) {
|
||||||
const [user, collection] = id.split('/');
|
const [user, collection] = id.split('/');
|
||||||
|
if (user === '__internal') continue;
|
||||||
allCollections.add(`${user}/${collection}`);
|
allCollections.add(`${user}/${collection}`);
|
||||||
}
|
}
|
||||||
for (const collection of allCollections) {
|
for (const collection of allCollections) {
|
||||||
@@ -333,7 +371,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
|
|
||||||
for (const node of this.graph.nodes) {
|
for (const node of this.graph.nodes) {
|
||||||
const nodeType = this.registry.getNode(node.type);
|
const nodeType = this.registry.getNode(node.type);
|
||||||
if (!nodeType) {
|
if (!nodeType && !node.type.startsWith('__internal/')) {
|
||||||
logger.error(`Node type not found: ${node.type}`);
|
logger.error(`Node type not found: ${node.type}`);
|
||||||
this.status = 'error';
|
this.status = 'error';
|
||||||
return;
|
return;
|
||||||
@@ -389,15 +427,46 @@ export class GraphManager extends EventEmitter<{
|
|||||||
}
|
}
|
||||||
|
|
||||||
getAllNodes() {
|
getAllNodes() {
|
||||||
return Array.from(this.nodes.values());
|
return Array
|
||||||
|
.from(this.nodes.values());
|
||||||
}
|
}
|
||||||
|
|
||||||
getNode(id: number) {
|
getNode(id: number) {
|
||||||
return this.nodes.get(id);
|
return this.nodes.get(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
getNodeType(id: string) {
|
getNodeType(node: NodeInstance) {
|
||||||
return this.registry.getNode(id);
|
// Construct the group inputs on the fly
|
||||||
|
if (node.type === '__internal/group/instance') {
|
||||||
|
const groupDefinition = this.getGroup(node.props?.groupId as number);
|
||||||
|
|
||||||
|
if (!groupDefinition) {
|
||||||
|
logger.error(`Group not found: ${node.props?.groupId}`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const inputs = {
|
||||||
|
'groupId': {
|
||||||
|
type: 'select',
|
||||||
|
label: '',
|
||||||
|
value: node.props?.groupId,
|
||||||
|
internal: true,
|
||||||
|
options: this.graph.groups.map(g => g.id)
|
||||||
|
},
|
||||||
|
...(node.state.type?.inputs || {}),
|
||||||
|
...groupDefinition?.inputs
|
||||||
|
};
|
||||||
|
|
||||||
|
const groupType = {
|
||||||
|
...node.state.type,
|
||||||
|
inputs,
|
||||||
|
outputs: groupDefinition?.outputs?.map(o => o.type)
|
||||||
|
} as NodeDefinition;
|
||||||
|
|
||||||
|
return groupType;
|
||||||
|
}
|
||||||
|
|
||||||
|
return node.state.type;
|
||||||
}
|
}
|
||||||
|
|
||||||
async loadNodeType(id: NodeId) {
|
async loadNodeType(id: NodeId) {
|
||||||
@@ -459,6 +528,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
}
|
}
|
||||||
|
|
||||||
removeNode(node: NodeInstance, { restoreEdges = false } = {}) {
|
removeNode(node: NodeInstance, { restoreEdges = false } = {}) {
|
||||||
|
console.log('REMOVING NODE', $state.snapshot({ node }));
|
||||||
const edgesToNode = this.edges.filter((edge) => edge[2].id === node.id);
|
const edgesToNode = this.edges.filter((edge) => edge[2].id === node.id);
|
||||||
const edgesFromNode = this.edges.filter((edge) => edge[0].id === node.id);
|
const edgesFromNode = this.edges.filter((edge) => edge[0].id === node.id);
|
||||||
for (const edge of [...edgesToNode, ...edgesFromNode]) {
|
for (const edge of [...edgesToNode, ...edgesFromNode]) {
|
||||||
@@ -502,8 +572,22 @@ export class GraphManager extends EventEmitter<{
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
getGroup(id: number) {
|
||||||
|
return this.graph.groups.find(g => g.id === id);
|
||||||
|
}
|
||||||
|
|
||||||
createNodeId() {
|
createNodeId() {
|
||||||
return Math.max(0, ...this.nodes.keys()) + 1;
|
const ids = [
|
||||||
|
...this.nodes.keys(),
|
||||||
|
...this.graph.groups.map(g => g.id),
|
||||||
|
...this.graph.groups.flatMap(g => g.nodes.map(n => n.id))
|
||||||
|
];
|
||||||
|
|
||||||
|
let id = 0;
|
||||||
|
while (ids.includes(id)) {
|
||||||
|
id++;
|
||||||
|
}
|
||||||
|
return id;
|
||||||
}
|
}
|
||||||
|
|
||||||
createGraph(nodes: NodeInstance[], edges: [number, number, number, string][]) {
|
createGraph(nodes: NodeInstance[], edges: [number, number, number, string][]) {
|
||||||
@@ -516,7 +600,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
const id = startId++;
|
const id = startId++;
|
||||||
idMap.set(node.id, id);
|
idMap.set(node.id, id);
|
||||||
const type = this.registry.getNode(node.type);
|
const type = this.registry.getNode(node.type);
|
||||||
if (!type) {
|
if (!type && !node.type.startsWith('__internal/')) {
|
||||||
throw new Error(`Node type not found: ${node.type}`);
|
throw new Error(`Node type not found: ${node.type}`);
|
||||||
}
|
}
|
||||||
return { ...node, id, tmp: { type } };
|
return { ...node, id, tmp: { type } };
|
||||||
@@ -549,6 +633,148 @@ export class GraphManager extends EventEmitter<{
|
|||||||
return nodes;
|
return nodes;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
removeUnusedGroups() {
|
||||||
|
const usedGroups = new Set(this.getAllNodes().map(n => n.props?.groupId));
|
||||||
|
const unusedGroupAmount = this.graph.groups.length - usedGroups.size;
|
||||||
|
this.graph.groups = this.graph.groups.filter(g => usedGroups.has(g.id));
|
||||||
|
this.save();
|
||||||
|
return unusedGroupAmount;
|
||||||
|
}
|
||||||
|
|
||||||
|
groupNodes(nodeIds: number[]) {
|
||||||
|
this.startUndoGroup();
|
||||||
|
this.removeUnusedGroups();
|
||||||
|
|
||||||
|
const nodes = [
|
||||||
|
...new Set(nodeIds).values().map(id => this.getNode(id)).filter(Boolean)
|
||||||
|
] as NodeInstance[];
|
||||||
|
|
||||||
|
if (!nodes.length) return;
|
||||||
|
|
||||||
|
logger.log(`Grouping ${nodes.length} nodes`, { nodes });
|
||||||
|
|
||||||
|
const ids = new Set(nodes.map(n => n.id));
|
||||||
|
|
||||||
|
// We use the map to dedupe when one external node is connected to multiple internal nodes
|
||||||
|
// ┌──internal_a
|
||||||
|
// external──┤
|
||||||
|
// └──internal_b
|
||||||
|
// This should only result in one group input not two
|
||||||
|
const incomingEdges = this.edges.filter((edge) => ids.has(edge[2].id) && !ids.has(edge[0].id));
|
||||||
|
const groupInputs = new Map<string, Edge>();
|
||||||
|
for (const edge of incomingEdges) {
|
||||||
|
groupInputs.set(`${edge[0].id}-${edge[1]}`, edge);
|
||||||
|
}
|
||||||
|
|
||||||
|
// And the same for the outputs
|
||||||
|
const outgoingEdges = this.edges.filter((edge) => ids.has(edge[0].id) && !ids.has(edge[2].id));
|
||||||
|
const groupOutputs = new Map<string, Edge>();
|
||||||
|
for (const edge of outgoingEdges) {
|
||||||
|
groupOutputs.set(`${edge[2].id}-${edge[3]}`, edge);
|
||||||
|
}
|
||||||
|
|
||||||
|
const inputs: Record<string, NodeInput> = {};
|
||||||
|
[...groupInputs.values()].forEach((edge, i) => {
|
||||||
|
const input = {
|
||||||
|
label: `Input ${i}`,
|
||||||
|
type: edge[0].state.type?.outputs?.[edge[1]] || '*'
|
||||||
|
};
|
||||||
|
inputs[`input_${i}`] = input as NodeInput;
|
||||||
|
});
|
||||||
|
|
||||||
|
const outputs = [...groupOutputs.values()].map((edge, i) => ({
|
||||||
|
label: `Output ${i}`,
|
||||||
|
type: edge[2].state.type?.inputs?.[edge[3]].type || '*'
|
||||||
|
}));
|
||||||
|
|
||||||
|
const groupPosition = [0, 0] as [number, number];
|
||||||
|
const bounds: Box = { minX: Infinity, maxX: -Infinity, minY: Infinity, maxY: -Infinity };
|
||||||
|
for (const node of nodes) {
|
||||||
|
groupPosition[0] += node.position[0];
|
||||||
|
groupPosition[1] += node.position[1];
|
||||||
|
bounds.minX = Math.min(bounds.minX, node.position[0]);
|
||||||
|
bounds.maxX = Math.max(bounds.maxX, node.position[0]);
|
||||||
|
bounds.minY = Math.min(bounds.minY, node.position[1]);
|
||||||
|
bounds.maxY = Math.max(bounds.maxY, node.position[1]);
|
||||||
|
}
|
||||||
|
groupPosition[0] /= nodes.length;
|
||||||
|
groupPosition[1] /= nodes.length;
|
||||||
|
|
||||||
|
const groupInputNode: NodeInstance = {
|
||||||
|
id: this.createNodeId(),
|
||||||
|
type: '__internal/group/input',
|
||||||
|
position: [bounds.minX - 50, (bounds.minY + bounds.maxY) / 2],
|
||||||
|
state: {}
|
||||||
|
};
|
||||||
|
|
||||||
|
const groupOutputNode: NodeInstance = {
|
||||||
|
id: this.createNodeId(),
|
||||||
|
type: '__internal/group/output',
|
||||||
|
position: [bounds.maxX + 25, (bounds.minY + bounds.maxY) / 2],
|
||||||
|
state: {}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Edges that are inside the group
|
||||||
|
const internalEdges = this.edges.filter((edge) => {
|
||||||
|
return ids.has(edge[0].id) || ids.has(edge[2].id);
|
||||||
|
}).map((edge) => {
|
||||||
|
// Going in from the group
|
||||||
|
if (!ids.has(edge[0].id)) {
|
||||||
|
return [groupInputNode.id, 0, edge[2].id, edge[3]];
|
||||||
|
// Going out to the group
|
||||||
|
} else if (!ids.has(edge[2].id)) {
|
||||||
|
return [edge[0].id, edge[1], groupOutputNode.id, 'Out'];
|
||||||
|
}
|
||||||
|
return [edge[0].id, edge[1], edge[2].id, edge[3]];
|
||||||
|
}) as [number, number, number, string][];
|
||||||
|
|
||||||
|
const groupId = this.createNodeId();
|
||||||
|
const groupDefinition: GroupDefinition = {
|
||||||
|
id: groupId,
|
||||||
|
inputs: inputs,
|
||||||
|
outputs: outputs,
|
||||||
|
edges: internalEdges,
|
||||||
|
nodes: [groupInputNode, ...nodes, groupOutputNode]
|
||||||
|
};
|
||||||
|
|
||||||
|
const groupNode = this.createNode({
|
||||||
|
type: '__internal/group/instance',
|
||||||
|
position: [groupPosition[0], groupPosition[1]],
|
||||||
|
props: {
|
||||||
|
groupId: groupId
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!groupNode) throw new Error('Failed to create group node');
|
||||||
|
|
||||||
|
// Update the edges that are now inside
|
||||||
|
// the group to be connected to that group node
|
||||||
|
const externalEdges = this.edges.map((edge) => {
|
||||||
|
if (ids.has(edge[2].id)) {
|
||||||
|
// Edge going into the group
|
||||||
|
return [edge[0], edge[1], groupNode, 'input_0'] as Edge;
|
||||||
|
} else if (ids.has(edge[0].id)) {
|
||||||
|
// Edge going out of the group
|
||||||
|
return [groupNode, 0, edge[2], edge[3]] as Edge;
|
||||||
|
}
|
||||||
|
return edge;
|
||||||
|
});
|
||||||
|
|
||||||
|
this.graph.groups.push(groupDefinition);
|
||||||
|
this.nodes.set(groupNode.id, groupNode);
|
||||||
|
this.edges = externalEdges;
|
||||||
|
|
||||||
|
// Remove nodes from graph which are not part of the group
|
||||||
|
for (const node of nodes) {
|
||||||
|
this.removeNode(node);
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('FINISHED', this.serialize());
|
||||||
|
this.saveUndoGroup();
|
||||||
|
|
||||||
|
return groupNode;
|
||||||
|
}
|
||||||
|
|
||||||
createNode({
|
createNode({
|
||||||
type,
|
type,
|
||||||
position,
|
position,
|
||||||
@@ -559,7 +785,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
props: NodeInstance['props'];
|
props: NodeInstance['props'];
|
||||||
}) {
|
}) {
|
||||||
const nodeType = this.registry.getNode(type);
|
const nodeType = this.registry.getNode(type);
|
||||||
if (!nodeType) {
|
if (!nodeType && !type.startsWith('__internal/')) {
|
||||||
logger.error(`Node type not found: ${type}`);
|
logger.error(`Node type not found: ${type}`);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -597,11 +823,14 @@ export class GraphManager extends EventEmitter<{
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const fromType = this.getNodeType(from);
|
||||||
|
const toType = this.getNodeType(to);
|
||||||
|
|
||||||
// check if socket types match
|
// check if socket types match
|
||||||
const fromSocketType = from.state?.type?.outputs?.[fromSocket];
|
const fromSocketType = fromType?.outputs?.[fromSocket];
|
||||||
const toSocketType = [to.state?.type?.inputs?.[toSocket]?.type];
|
const toSocketType = [toType?.inputs?.[toSocket]?.type];
|
||||||
if (to.state?.type?.inputs?.[toSocket]?.accepts) {
|
if (toType?.inputs?.[toSocket]?.accepts) {
|
||||||
toSocketType.push(...(to?.state?.type?.inputs?.[toSocket]?.accepts || []));
|
toSocketType.push(...(toType?.inputs?.[toSocket]?.accepts || []));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!areSocketsCompatible(fromSocketType, toSocketType)) {
|
if (!areSocketsCompatible(fromSocketType, toSocketType)) {
|
||||||
@@ -724,7 +953,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
}
|
}
|
||||||
|
|
||||||
getPossibleSockets({ node, index }: Socket): [NodeInstance, string | number][] {
|
getPossibleSockets({ node, index }: Socket): [NodeInstance, string | number][] {
|
||||||
const nodeType = node?.state?.type;
|
const nodeType = this.getNodeType(node);
|
||||||
if (!nodeType) return [];
|
if (!nodeType) return [];
|
||||||
|
|
||||||
const sockets: [NodeInstance, string | number][] = [];
|
const sockets: [NodeInstance, string | number][] = [];
|
||||||
@@ -740,7 +969,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
const ownType = nodeType?.inputs?.[index].type;
|
const ownType = nodeType?.inputs?.[index].type;
|
||||||
|
|
||||||
for (const node of nodes) {
|
for (const node of nodes) {
|
||||||
const nodeType = node?.state?.type;
|
const nodeType = this.getNodeType(node);
|
||||||
const inputs = nodeType?.outputs;
|
const inputs = nodeType?.outputs;
|
||||||
if (!inputs) continue;
|
if (!inputs) continue;
|
||||||
for (let index = 0; index < inputs.length; index++) {
|
for (let index = 0; index < inputs.length; index++) {
|
||||||
@@ -772,7 +1001,7 @@ export class GraphManager extends EventEmitter<{
|
|||||||
const ownType = nodeType.outputs?.[index];
|
const ownType = nodeType.outputs?.[index];
|
||||||
|
|
||||||
for (const node of nodes) {
|
for (const node of nodes) {
|
||||||
const inputs = node?.state?.type?.inputs;
|
const inputs = this.getNodeType(node)?.inputs;
|
||||||
if (!inputs) continue;
|
if (!inputs) continue;
|
||||||
for (const key in inputs) {
|
for (const key in inputs) {
|
||||||
const otherType = [inputs[key].type];
|
const otherType = [inputs[key].type];
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import { SvelteMap, SvelteSet } from 'svelte/reactivity';
|
|||||||
import type { OrthographicCamera, Vector3 } from 'three';
|
import type { OrthographicCamera, Vector3 } from 'three';
|
||||||
import type { GraphManager } from './graph-manager.svelte';
|
import type { GraphManager } from './graph-manager.svelte';
|
||||||
import { ColorGenerator } from './graph/colors';
|
import { ColorGenerator } from './graph/colors';
|
||||||
import { getNodeHeight, getSocketPosition } from './helpers/nodeHelpers';
|
import { getNodeHeight, getParameterHeight } from './helpers/nodeHelpers';
|
||||||
|
|
||||||
const graphStateKey = Symbol('graph-state');
|
const graphStateKey = Symbol('graph-state');
|
||||||
export function getGraphState() {
|
export function getGraphState() {
|
||||||
@@ -203,7 +203,7 @@ export class GraphState {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const debugNode = this.graph.createNode({
|
const debugNode = this.graph.createNode({
|
||||||
type: 'max/plantarium/debug',
|
type: '__internal/node/debug',
|
||||||
position: [node.position[0] + 30, node.position[1]],
|
position: [node.position[0] + 30, node.position[1]],
|
||||||
props: {}
|
props: {}
|
||||||
});
|
});
|
||||||
@@ -240,6 +240,10 @@ export class GraphState {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
groupSelectedNodes() {
|
||||||
|
return this.graph.groupNodes([...this.selectedNodes.keys(), this.activeNodeId]);
|
||||||
|
}
|
||||||
|
|
||||||
centerNode(node?: NodeInstance) {
|
centerNode(node?: NodeInstance) {
|
||||||
const average = [0, 0, 4];
|
const average = [0, 0, 4];
|
||||||
if (node) {
|
if (node) {
|
||||||
@@ -301,7 +305,7 @@ export class GraphState {
|
|||||||
if (edge[3] === index) {
|
if (edge[3] === index) {
|
||||||
node = edge[0];
|
node = edge[0];
|
||||||
index = edge[1];
|
index = edge[1];
|
||||||
position = getSocketPosition(node, index);
|
position = this.getSocketPosition(node, index);
|
||||||
this.graph.removeEdge(edge);
|
this.graph.removeEdge(edge);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@@ -321,7 +325,7 @@ export class GraphState {
|
|||||||
return {
|
return {
|
||||||
node,
|
node,
|
||||||
index,
|
index,
|
||||||
position: getSocketPosition(node, index)
|
position: this.getSocketPosition(node, index)
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -358,7 +362,7 @@ export class GraphState {
|
|||||||
for (const node of this.graph.nodes.values()) {
|
for (const node of this.graph.nodes.values()) {
|
||||||
const x = node.position[0];
|
const x = node.position[0];
|
||||||
const y = node.position[1];
|
const y = node.position[1];
|
||||||
const height = getNodeHeight(node.state.type!);
|
const height = getNodeHeight(this.graph.getNodeType(node)!);
|
||||||
if (downX > x && downX < x + 20 && downY > y && downY < y + height) {
|
if (downX > x && downX < x + 20 && downY > y && downY < y + height) {
|
||||||
clickedNodeId = node.id;
|
clickedNodeId = node.id;
|
||||||
break;
|
break;
|
||||||
@@ -370,7 +374,7 @@ export class GraphState {
|
|||||||
}
|
}
|
||||||
|
|
||||||
isNodeInView(node: NodeInstance) {
|
isNodeInView(node: NodeInstance) {
|
||||||
const height = getNodeHeight(node.state.type!);
|
const height = getNodeHeight(this.graph.getNodeType(node)!);
|
||||||
const width = 20;
|
const width = 20;
|
||||||
return node.position[0] > this.cameraBounds[0] - width
|
return node.position[0] > this.cameraBounds[0] - width
|
||||||
&& node.position[0] < this.cameraBounds[1]
|
&& node.position[0] < this.cameraBounds[1]
|
||||||
@@ -381,4 +385,38 @@ export class GraphState {
|
|||||||
openNodePalette() {
|
openNodePalette() {
|
||||||
this.addMenuPosition = [this.mousePosition[0], this.mousePosition[1]];
|
this.addMenuPosition = [this.mousePosition[0], this.mousePosition[1]];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
enterGroupNode() {
|
||||||
|
if (this.activeNodeId === -1) return;
|
||||||
|
const selectedNode = this.graph.getNode(this.activeNodeId);
|
||||||
|
if (!selectedNode || selectedNode.type.startsWith('__internal/group/instance')) return;
|
||||||
|
}
|
||||||
|
|
||||||
|
getSocketPosition(
|
||||||
|
node: NodeInstance,
|
||||||
|
index: string | number
|
||||||
|
): [number, number] {
|
||||||
|
if (typeof index === 'number') {
|
||||||
|
return [
|
||||||
|
(node?.state?.x ?? node.position[0]) + 20,
|
||||||
|
(node?.state?.y ?? node.position[1]) + 2.5 + 10 * index
|
||||||
|
];
|
||||||
|
} else {
|
||||||
|
let height = 5;
|
||||||
|
const nodeType = this.graph.getNodeType(node)!;
|
||||||
|
const inputs = nodeType.inputs || {};
|
||||||
|
for (const inputKey in inputs) {
|
||||||
|
const h = getParameterHeight(nodeType, inputKey) / 10;
|
||||||
|
if (inputKey === index) {
|
||||||
|
height += h / 2;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
height += h;
|
||||||
|
}
|
||||||
|
return [
|
||||||
|
node?.state?.x ?? node.position[0],
|
||||||
|
(node?.state?.y ?? node.position[1]) + height
|
||||||
|
];
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,7 +11,6 @@
|
|||||||
import Debug from '../debug/Debug.svelte';
|
import Debug from '../debug/Debug.svelte';
|
||||||
import EdgeEl from '../edges/Edge.svelte';
|
import EdgeEl from '../edges/Edge.svelte';
|
||||||
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
||||||
import { getSocketPosition } from '../helpers/nodeHelpers';
|
|
||||||
import NodeEl from '../node/Node.svelte';
|
import NodeEl from '../node/Node.svelte';
|
||||||
import { maxZoom, minZoom } from './constants';
|
import { maxZoom, minZoom } from './constants';
|
||||||
import { FileDropEventManager } from './drop.events';
|
import { FileDropEventManager } from './drop.events';
|
||||||
@@ -39,8 +38,8 @@
|
|||||||
return [0, 0, 0, 0];
|
return [0, 0, 0, 0];
|
||||||
}
|
}
|
||||||
|
|
||||||
const pos1 = getSocketPosition(fromNode, edge[1]);
|
const pos1 = graphState.getSocketPosition(fromNode, edge[1]);
|
||||||
const pos2 = getSocketPosition(toNode, edge[3]);
|
const pos2 = graphState.getSocketPosition(toNode, edge[3]);
|
||||||
return [pos1[0], pos1[1], pos2[0], pos2[1]];
|
return [pos1[0], pos1[1], pos2[0], pos2[1]];
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -96,11 +95,12 @@
|
|||||||
graphState.addMenuPosition = null;
|
graphState.addMenuPosition = null;
|
||||||
}
|
}
|
||||||
|
|
||||||
function getSocketType(node: NodeInstance, index: number | string): string {
|
function getSocketType(node: NodeInstance, index: number | string, e: unknown): string {
|
||||||
|
const nodeType = graph.getNodeType(node);
|
||||||
if (typeof index === 'string') {
|
if (typeof index === 'string') {
|
||||||
return node.state.type?.inputs?.[index].type || 'unknown';
|
return nodeType?.inputs?.[index].type || 'unknown';
|
||||||
}
|
}
|
||||||
return node.state.type?.outputs?.[index] || 'unknown';
|
return nodeType?.outputs?.[index] || 'unknown';
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
@@ -182,8 +182,8 @@
|
|||||||
{#if graphState.activeSocket}
|
{#if graphState.activeSocket}
|
||||||
<EdgeEl
|
<EdgeEl
|
||||||
z={graphState.cameraPosition[2]}
|
z={graphState.cameraPosition[2]}
|
||||||
inputType={getSocketType(graphState.activeSocket.node, graphState.activeSocket.index)}
|
inputType={getSocketType(graphState.activeSocket.node, graphState.activeSocket.index, 'c')}
|
||||||
outputType={getSocketType(graphState.activeSocket.node, graphState.activeSocket.index)}
|
outputType={getSocketType(graphState.activeSocket.node, graphState.activeSocket.index, 'd')}
|
||||||
x1={graphState.activeSocket.position[0]}
|
x1={graphState.activeSocket.position[0]}
|
||||||
y1={graphState.activeSocket.position[1]}
|
y1={graphState.activeSocket.position[1]}
|
||||||
x2={graphState.edgeEndPosition?.[0] ?? graphState.mousePosition[0]}
|
x2={graphState.edgeEndPosition?.[0] ?? graphState.mousePosition[0]}
|
||||||
@@ -196,8 +196,8 @@
|
|||||||
<EdgeEl
|
<EdgeEl
|
||||||
id={graph.getEdgeId(edge)}
|
id={graph.getEdgeId(edge)}
|
||||||
z={graphState.cameraPosition[2]}
|
z={graphState.cameraPosition[2]}
|
||||||
inputType={getSocketType(edge[0], edge[1])}
|
inputType={getSocketType(edge[0], edge[1], 'a')}
|
||||||
outputType={getSocketType(edge[2], edge[3])}
|
outputType={getSocketType(edge[2], edge[3], 'b')}
|
||||||
{x1}
|
{x1}
|
||||||
{y1}
|
{y1}
|
||||||
{x2}
|
{x2}
|
||||||
@@ -216,7 +216,7 @@
|
|||||||
style:transform={`scale(${graphState.cameraPosition[2] * 0.1})`}
|
style:transform={`scale(${graphState.cameraPosition[2] * 0.1})`}
|
||||||
class:hovering-sockets={graphState.activeSocket}
|
class:hovering-sockets={graphState.activeSocket}
|
||||||
>
|
>
|
||||||
{#each graph.nodes.values() as node (node.id)}
|
{#each graph.getAllNodes() as node (node.id)}
|
||||||
<NodeEl
|
<NodeEl
|
||||||
{node}
|
{node}
|
||||||
inView={graphState.isNodeInView(node)}
|
inView={graphState.isNodeInView(node)}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import { createKeyMap } from '$lib/helpers/createKeyMap';
|
import { createKeyMap } from '$lib/helpers/createKeyMap';
|
||||||
import type { Graph, NodeInstance, NodeRegistry } from '@nodarium/types';
|
import type { Graph, NodeInstance, NodeRegistry } from '@nodarium/types';
|
||||||
|
import { onMount } from 'svelte';
|
||||||
import { GraphManager } from '../graph-manager.svelte';
|
import { GraphManager } from '../graph-manager.svelte';
|
||||||
import { GraphState, setGraphManager, setGraphState } from '../graph-state.svelte';
|
import { GraphState, setGraphManager, setGraphState } from '../graph-state.svelte';
|
||||||
import { setupKeymaps } from '../keymaps';
|
import { setupKeymaps } from '../keymaps';
|
||||||
@@ -83,7 +84,7 @@
|
|||||||
manager.on('save', (save) => onsave?.(save));
|
manager.on('save', (save) => onsave?.(save));
|
||||||
|
|
||||||
$effect(() => {
|
$effect(() => {
|
||||||
if (graph) {
|
if (graph && (manager.status !== 'idle' || manager.graph.id !== graph.id)) {
|
||||||
manager.load(graph);
|
manager.load(graph);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -23,36 +23,11 @@ export function getParameterHeight(node: NodeDefinition, inputKey: string) {
|
|||||||
return 50;
|
return 50;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function getSocketPosition(
|
|
||||||
node: NodeInstance,
|
|
||||||
index: string | number
|
|
||||||
): [number, number] {
|
|
||||||
if (typeof index === 'number') {
|
|
||||||
return [
|
|
||||||
(node?.state?.x ?? node.position[0]) + 20,
|
|
||||||
(node?.state?.y ?? node.position[1]) + 2.5 + 10 * index
|
|
||||||
];
|
|
||||||
} else {
|
|
||||||
let height = 5;
|
|
||||||
const nodeType = node.state.type!;
|
|
||||||
const inputs = nodeType.inputs || {};
|
|
||||||
for (const inputKey in inputs) {
|
|
||||||
const h = getParameterHeight(nodeType, inputKey) / 10;
|
|
||||||
if (inputKey === index) {
|
|
||||||
height += h / 2;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
height += h;
|
|
||||||
}
|
|
||||||
return [
|
|
||||||
node?.state?.x ?? node.position[0],
|
|
||||||
(node?.state?.y ?? node.position[1]) + height
|
|
||||||
];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const nodeHeightCache: Record<string, number> = {};
|
const nodeHeightCache: Record<string, number> = {};
|
||||||
export function getNodeHeight(node: NodeDefinition) {
|
export function getNodeHeight(node: NodeDefinition) {
|
||||||
|
if (!node) {
|
||||||
|
console.trace('Node is undefined', node);
|
||||||
|
}
|
||||||
if (node.id in nodeHeightCache) {
|
if (node.id in nodeHeightCache) {
|
||||||
return nodeHeightCache[node.id];
|
return nodeHeightCache[node.id];
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -54,6 +54,20 @@ export function setupKeymaps(keymap: Keymap, graph: GraphManager, graphState: Gr
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
keymap.addShortcut({
|
||||||
|
key: 'g',
|
||||||
|
ctrl: true,
|
||||||
|
preventDefault: true,
|
||||||
|
description: 'Group selected nodes',
|
||||||
|
callback: () => graphState.groupSelectedNodes()
|
||||||
|
});
|
||||||
|
|
||||||
|
keymap.addShortcut({
|
||||||
|
key: 'Tab',
|
||||||
|
description: 'Enter selected node group',
|
||||||
|
callback: () => graphState.enterGroupNode()
|
||||||
|
});
|
||||||
|
|
||||||
keymap.addShortcut({
|
keymap.addShortcut({
|
||||||
key: 'A',
|
key: 'A',
|
||||||
shift: true,
|
shift: true,
|
||||||
|
|||||||
@@ -3,13 +3,14 @@
|
|||||||
import type { NodeInstance } from '@nodarium/types';
|
import type { NodeInstance } from '@nodarium/types';
|
||||||
import { T } from '@threlte/core';
|
import { T } from '@threlte/core';
|
||||||
import { type Mesh } from 'three';
|
import { type Mesh } from 'three';
|
||||||
import { getGraphState } from '../graph-state.svelte';
|
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
||||||
import { colors } from '../graph/colors.svelte';
|
import { colors } from '../graph/colors.svelte';
|
||||||
import { getNodeHeight, getParameterHeight } from '../helpers/nodeHelpers';
|
import { getNodeHeight, getParameterHeight } from '../helpers/nodeHelpers';
|
||||||
import NodeFrag from './Node.frag';
|
import NodeFrag from './Node.frag';
|
||||||
import NodeVert from './Node.vert';
|
import NodeVert from './Node.vert';
|
||||||
import NodeHtml from './NodeHTML.svelte';
|
import NodeHtml from './NodeHTML.svelte';
|
||||||
|
|
||||||
|
const graph = getGraphManager();
|
||||||
const graphState = getGraphState();
|
const graphState = getGraphState();
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
@@ -18,7 +19,7 @@
|
|||||||
};
|
};
|
||||||
let { node = $bindable(), inView }: Props = $props();
|
let { node = $bindable(), inView }: Props = $props();
|
||||||
|
|
||||||
const nodeType = $derived(node.state.type!);
|
const nodeType = $derived(graph.getNodeType(node)!);
|
||||||
|
|
||||||
const isActive = $derived(graphState.activeNodeId === node.id);
|
const isActive = $derived(graphState.activeNodeId === node.id);
|
||||||
const isSelected = $derived(graphState.selectedNodes.has(node.id));
|
const isSelected = $derived(graphState.selectedNodes.has(node.id));
|
||||||
@@ -40,7 +41,7 @@
|
|||||||
|
|
||||||
let meshRef: Mesh | undefined = $state();
|
let meshRef: Mesh | undefined = $state();
|
||||||
|
|
||||||
const height = getNodeHeight(node.state.type!);
|
const height = $derived(getNodeHeight(nodeType));
|
||||||
|
|
||||||
const zoom = $derived(graphState.cameraPosition[2]);
|
const zoom = $derived(graphState.cameraPosition[2]);
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,12 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import type { NodeInstance } from '@nodarium/types';
|
import type { NodeInstance } from '@nodarium/types';
|
||||||
import { getGraphState } from '../graph-state.svelte';
|
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
||||||
import NodeHeader from './NodeHeader.svelte';
|
import NodeHeader from './NodeHeader.svelte';
|
||||||
import NodeParameter from './NodeParameter.svelte';
|
import NodeParameter from './NodeParameter.svelte';
|
||||||
|
|
||||||
let ref: HTMLDivElement;
|
let ref: HTMLDivElement;
|
||||||
|
|
||||||
|
const graph = getGraphManager();
|
||||||
const graphState = getGraphState();
|
const graphState = getGraphState();
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
@@ -30,8 +31,12 @@
|
|||||||
const zOffset = Math.random() - 0.5;
|
const zOffset = Math.random() - 0.5;
|
||||||
const zLimit = 2 - zOffset;
|
const zLimit = 2 - zOffset;
|
||||||
|
|
||||||
const parameters = Object.entries(node.state?.type?.inputs || {}).filter(
|
const nodeType = $derived(graph.getNodeType(node));
|
||||||
(p) => p[1].type !== 'seed' && !('setting' in p[1]) && p[1]?.hidden !== true
|
|
||||||
|
const parameters = $derived(
|
||||||
|
Object.entries(nodeType?.inputs || {}).filter(
|
||||||
|
(p) => p[1].type !== 'seed' && !('setting' in p[1]) && p[1]?.hidden !== true
|
||||||
|
) || {}
|
||||||
);
|
);
|
||||||
|
|
||||||
$effect(() => {
|
$effect(() => {
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import { appSettings } from '$lib/settings/app-settings.svelte';
|
import { appSettings } from '$lib/settings/app-settings.svelte';
|
||||||
import type { NodeInstance, Socket } from '@nodarium/types';
|
import type { NodeInstance, Socket } from '@nodarium/types';
|
||||||
import { getGraphState } from '../graph-state.svelte';
|
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
||||||
import { createNodePath } from '../helpers/index.js';
|
import { createNodePath } from '../helpers/index.js';
|
||||||
import { getSocketPosition } from '../helpers/nodeHelpers';
|
|
||||||
|
|
||||||
const graphState = getGraphState();
|
const graphState = getGraphState();
|
||||||
|
const graph = getGraphManager();
|
||||||
|
|
||||||
const { node }: { node: NodeInstance } = $props();
|
const { node }: { node: NodeInstance } = $props();
|
||||||
|
|
||||||
@@ -16,13 +16,14 @@
|
|||||||
graphState.setDownSocket?.({
|
graphState.setDownSocket?.({
|
||||||
node,
|
node,
|
||||||
index: 0,
|
index: 0,
|
||||||
position: getSocketPosition?.(node, 0)
|
position: graphState.getSocketPosition?.(node, 0)
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const cornerTop = 10;
|
const cornerTop = 10;
|
||||||
const rightBump = $derived(!!node?.state?.type?.outputs?.length);
|
const nodeType = $derived(graph.getNodeType(node));
|
||||||
|
const rightBump = $derived(!!nodeType?.outputs?.length);
|
||||||
const aspectRatio = 0.25;
|
const aspectRatio = 0.25;
|
||||||
|
|
||||||
const path = $derived(
|
const path = $derived(
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
import type { NodeInput, NodeInstance, Socket } from '@nodarium/types';
|
import type { NodeInput, NodeInstance, Socket } from '@nodarium/types';
|
||||||
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
import { getGraphManager, getGraphState } from '../graph-state.svelte';
|
||||||
import { createNodePath } from '../helpers';
|
import { createNodePath } from '../helpers';
|
||||||
import { getParameterHeight, getSocketPosition } from '../helpers/nodeHelpers';
|
import { getParameterHeight } from '../helpers/nodeHelpers';
|
||||||
import NodeInputEl from './NodeInput.svelte';
|
import NodeInputEl from './NodeInput.svelte';
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
@@ -19,7 +19,7 @@
|
|||||||
|
|
||||||
let { node = $bindable(), input, id, isLast }: Props = $props();
|
let { node = $bindable(), input, id, isLast }: Props = $props();
|
||||||
|
|
||||||
const nodeType = $derived(node.state.type!);
|
const nodeType = $derived(graph.getNodeType(node)!);
|
||||||
|
|
||||||
const inputType = $derived(nodeType.inputs?.[id]);
|
const inputType = $derived(nodeType.inputs?.[id]);
|
||||||
|
|
||||||
@@ -32,7 +32,7 @@
|
|||||||
graphState.setDownSocket({
|
graphState.setDownSocket({
|
||||||
node,
|
node,
|
||||||
index: id,
|
index: id,
|
||||||
position: getSocketPosition(node, id)
|
position: graphState.getSocketPosition(node, id)
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -23,7 +23,11 @@ export function createMockNodeRegistry(nodes: NodeDefinition[]): NodeRegistry {
|
|||||||
|
|
||||||
export const mockFloatOutputNode: NodeDefinition = {
|
export const mockFloatOutputNode: NodeDefinition = {
|
||||||
id: 'test/node/output',
|
id: 'test/node/output',
|
||||||
inputs: {},
|
inputs: {
|
||||||
|
'input': {
|
||||||
|
type: 'float'
|
||||||
|
}
|
||||||
|
},
|
||||||
outputs: ['float'],
|
outputs: ['float'],
|
||||||
meta: { title: 'Float Output' },
|
meta: { title: 'Float Output' },
|
||||||
execute: () => new Int32Array()
|
execute: () => new Int32Array()
|
||||||
@@ -32,7 +36,7 @@ export const mockFloatOutputNode: NodeDefinition = {
|
|||||||
export const mockFloatInputNode: NodeDefinition = {
|
export const mockFloatInputNode: NodeDefinition = {
|
||||||
id: 'test/node/input',
|
id: 'test/node/input',
|
||||||
inputs: { value: { type: 'float' } },
|
inputs: { value: { type: 'float' } },
|
||||||
outputs: [],
|
outputs: ['float'],
|
||||||
meta: { title: 'Float Input' },
|
meta: { title: 'Float Input' },
|
||||||
execute: () => new Int32Array()
|
execute: () => new Int32Array()
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
export const debugNode = {
|
export const debugNode = {
|
||||||
id: 'max/plantarium/debug',
|
id: '__internal/debug/instance',
|
||||||
inputs: {
|
inputs: {
|
||||||
input: {
|
input: {
|
||||||
type: '*'
|
type: '*'
|
||||||
|
|||||||
@@ -0,0 +1,13 @@
|
|||||||
|
export const groupNode = {
|
||||||
|
id: '__internal/group/instance',
|
||||||
|
meta: { title: 'Group' },
|
||||||
|
inputs: {
|
||||||
|
input: {
|
||||||
|
type: 'select',
|
||||||
|
values: []
|
||||||
|
}
|
||||||
|
},
|
||||||
|
execute(_data: Int32Array): Int32Array {
|
||||||
|
return _data;
|
||||||
|
}
|
||||||
|
} as const;
|
||||||
@@ -88,6 +88,7 @@ export class RemoteNodeRegistry implements NodeRegistry {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fetchNodeDefinition(nodeId: `${string}/${string}/${string}`) {
|
async fetchNodeDefinition(nodeId: `${string}/${string}/${string}`) {
|
||||||
|
if (nodeId.startsWith('__internal/')) return;
|
||||||
return this.fetchJson(`nodes/${nodeId}.json`);
|
return this.fetchJson(`nodes/${nodeId}.json`);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -109,6 +110,8 @@ export class RemoteNodeRegistry implements NodeRegistry {
|
|||||||
return this.nodes.get(id)!;
|
return this.nodes.get(id)!;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (id.startsWith('__internal/')) return;
|
||||||
|
|
||||||
const wasmBuffer = await this.fetchNodeWasm(id);
|
const wasmBuffer = await this.fetchNodeWasm(id);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|||||||
@@ -6,6 +6,145 @@ import type {
|
|||||||
RuntimeExecutor,
|
RuntimeExecutor,
|
||||||
SyncCache
|
SyncCache
|
||||||
} from '@nodarium/types';
|
} from '@nodarium/types';
|
||||||
|
|
||||||
|
function isGroupInstanceType(type: string): boolean {
|
||||||
|
return type === '__virtual/group/instance';
|
||||||
|
}
|
||||||
|
|
||||||
|
export function expandGroups(graph: Graph): Graph {
|
||||||
|
if (!graph.groups || Object.keys(graph.groups).length === 0) {
|
||||||
|
return graph;
|
||||||
|
}
|
||||||
|
|
||||||
|
let nodes = [...graph.nodes];
|
||||||
|
let edges = [...graph.edges];
|
||||||
|
const groups = graph.groups;
|
||||||
|
|
||||||
|
let changed = true;
|
||||||
|
while (changed) {
|
||||||
|
changed = false;
|
||||||
|
|
||||||
|
for (let i = 0; i < nodes.length; i++) {
|
||||||
|
const node = nodes[i];
|
||||||
|
if (!isGroupInstanceType(node.type)) continue;
|
||||||
|
|
||||||
|
const groupId = (node.props as Record<string, unknown> | undefined)?.groupId as
|
||||||
|
| string
|
||||||
|
| undefined;
|
||||||
|
if (!groupId) continue;
|
||||||
|
const group = groups[groupId];
|
||||||
|
if (!group) continue;
|
||||||
|
|
||||||
|
changed = true;
|
||||||
|
|
||||||
|
// Recursively expand nested groups inside this group's internal graph
|
||||||
|
const expandedInternal = expandGroups({
|
||||||
|
id: 0,
|
||||||
|
nodes: group.graph.nodes,
|
||||||
|
edges: group.graph.edges,
|
||||||
|
groups
|
||||||
|
});
|
||||||
|
|
||||||
|
const ID_PREFIX = node.id * 1000000;
|
||||||
|
const idMap = new Map<number, number>();
|
||||||
|
|
||||||
|
const inputVirtualNode = expandedInternal.nodes.find(
|
||||||
|
n => n.type === '__virtual/group/input'
|
||||||
|
);
|
||||||
|
const outputVirtualNode = expandedInternal.nodes.find(
|
||||||
|
n => n.type === '__virtual/group/output'
|
||||||
|
);
|
||||||
|
|
||||||
|
const realInternalNodes = expandedInternal.nodes.filter(
|
||||||
|
n => n.type !== '__virtual/group/input' && n.type !== '__virtual/group/output'
|
||||||
|
);
|
||||||
|
|
||||||
|
for (const n of realInternalNodes) {
|
||||||
|
idMap.set(n.id, ID_PREFIX + n.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
const parentIncomingEdges = edges.filter(e => e[2] === node.id);
|
||||||
|
const parentOutgoingEdges = edges.filter(e => e[0] === node.id);
|
||||||
|
|
||||||
|
// Edges from/to virtual nodes in the expanded internal graph
|
||||||
|
const edgesFromInput = expandedInternal.edges.filter(
|
||||||
|
e => e[0] === inputVirtualNode?.id
|
||||||
|
);
|
||||||
|
const edgesToOutput = expandedInternal.edges.filter(
|
||||||
|
e => e[2] === outputVirtualNode?.id
|
||||||
|
);
|
||||||
|
|
||||||
|
const newEdges: Graph['edges'] = [];
|
||||||
|
|
||||||
|
// Short-circuit: parent source → internal target (via group input)
|
||||||
|
for (const parentEdge of parentIncomingEdges) {
|
||||||
|
const socketName = parentEdge[3];
|
||||||
|
const socketIdx = group.inputs.findIndex(s => s.name === socketName);
|
||||||
|
if (socketIdx === -1) continue;
|
||||||
|
|
||||||
|
for (const internalEdge of edgesFromInput.filter(e => e[1] === socketIdx)) {
|
||||||
|
const remappedId = idMap.get(internalEdge[2]);
|
||||||
|
if (remappedId !== undefined) {
|
||||||
|
newEdges.push([parentEdge[0], parentEdge[1], remappedId, internalEdge[3]]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Short-circuit: internal source → parent target (via group output)
|
||||||
|
for (const parentEdge of parentOutgoingEdges) {
|
||||||
|
const outputIdx = parentEdge[1];
|
||||||
|
const outputSocketName = group.outputs[outputIdx]?.name;
|
||||||
|
if (!outputSocketName) continue;
|
||||||
|
|
||||||
|
for (const internalEdge of edgesToOutput.filter(e => e[3] === outputSocketName)) {
|
||||||
|
const remappedId = idMap.get(internalEdge[0]);
|
||||||
|
if (remappedId !== undefined) {
|
||||||
|
newEdges.push([remappedId, internalEdge[1], parentEdge[2], parentEdge[3]]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remap internal-to-internal edges
|
||||||
|
const internalEdges = expandedInternal.edges.filter(
|
||||||
|
e =>
|
||||||
|
e[0] !== inputVirtualNode?.id
|
||||||
|
&& e[0] !== outputVirtualNode?.id
|
||||||
|
&& e[2] !== inputVirtualNode?.id
|
||||||
|
&& e[2] !== outputVirtualNode?.id
|
||||||
|
);
|
||||||
|
|
||||||
|
for (const e of internalEdges) {
|
||||||
|
const fromId = idMap.get(e[0]);
|
||||||
|
const toId = idMap.get(e[2]);
|
||||||
|
if (fromId !== undefined && toId !== undefined) {
|
||||||
|
newEdges.push([fromId, e[1], toId, e[3]]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove the group node
|
||||||
|
nodes.splice(i, 1);
|
||||||
|
|
||||||
|
// Add remapped internal nodes
|
||||||
|
for (const n of realInternalNodes) {
|
||||||
|
nodes.push({ ...n, id: idMap.get(n.id)! });
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove group node's edges and add short-circuit edges
|
||||||
|
const groupEdgeKeys = new Set([
|
||||||
|
...parentIncomingEdges.map(e => `${e[0]}-${e[1]}-${e[2]}-${e[3]}`),
|
||||||
|
...parentOutgoingEdges.map(e => `${e[0]}-${e[1]}-${e[2]}-${e[3]}`)
|
||||||
|
]);
|
||||||
|
edges = edges.filter(
|
||||||
|
e => !groupEdgeKeys.has(`${e[0]}-${e[1]}-${e[2]}-${e[3]}`)
|
||||||
|
);
|
||||||
|
edges.push(...newEdges);
|
||||||
|
|
||||||
|
break; // Restart loop with updated nodes array
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return { ...graph, nodes, edges };
|
||||||
|
}
|
||||||
import {
|
import {
|
||||||
concatEncodedArrays,
|
concatEncodedArrays,
|
||||||
createLogger,
|
createLogger,
|
||||||
@@ -75,7 +214,11 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
|
|||||||
throw new Error('Node registry is not ready');
|
throw new Error('Node registry is not ready');
|
||||||
}
|
}
|
||||||
|
|
||||||
await this.registry.load(graph.nodes.map((node) => node.type));
|
// Only load non-virtual types (virtual nodes are resolved locally)
|
||||||
|
const nonVirtualTypes = graph.nodes
|
||||||
|
.map(node => node.type)
|
||||||
|
.filter(t => !t.startsWith('__virtual/'));
|
||||||
|
await this.registry.load(nonVirtualTypes as any);
|
||||||
|
|
||||||
const typeMap = new Map<string, NodeDefinition>();
|
const typeMap = new Map<string, NodeDefinition>();
|
||||||
for (const node of graph.nodes) {
|
for (const node of graph.nodes) {
|
||||||
@@ -163,6 +306,9 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
|
|||||||
let a = performance.now();
|
let a = performance.now();
|
||||||
this.debugData = {};
|
this.debugData = {};
|
||||||
|
|
||||||
|
// Expand group nodes into a flat graph before execution
|
||||||
|
graph = expandGroups(graph);
|
||||||
|
|
||||||
// Then we add some metadata to the graph
|
// Then we add some metadata to the graph
|
||||||
const [outputNode, nodes] = await this.addMetaData(graph);
|
const [outputNode, nodes] = await this.addMetaData(graph);
|
||||||
let b = performance.now();
|
let b = performance.now();
|
||||||
@@ -219,7 +365,7 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
|
|||||||
if (inputNode) {
|
if (inputNode) {
|
||||||
if (results[inputNode.id] === undefined) {
|
if (results[inputNode.id] === undefined) {
|
||||||
throw new Error(
|
throw new Error(
|
||||||
`Node ${node.type} is missing input from node ${inputNode.type}`
|
`Node ${node.type} is missing input from node ${inputNode.type}#${inputNode.id}`
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
return results[inputNode.id];
|
return results[inputNode.id];
|
||||||
|
|||||||
@@ -24,3 +24,9 @@
|
|||||||
{:else}
|
{:else}
|
||||||
<p class="mx-4 mt-4">No node selected</p>
|
<p class="mx-4 mt-4">No node selected</p>
|
||||||
{/if}
|
{/if}
|
||||||
|
|
||||||
|
{#if manager?.graph.groups.length}
|
||||||
|
<button onclick={() => manager.removeUnusedGroups()}>
|
||||||
|
remove unused groups
|
||||||
|
</button>
|
||||||
|
{/if}
|
||||||
|
|||||||
@@ -1,20 +1,23 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import type { Graph } from '$lib/types';
|
import type { Graph } from '$lib/types';
|
||||||
|
import { JsonViewer } from '@nodarium/ui';
|
||||||
|
|
||||||
const { graph }: { graph?: Graph } = $props();
|
const { graph }: { graph?: Graph } = $props();
|
||||||
|
|
||||||
function convert(g: Graph): string {
|
const data = $derived(
|
||||||
return JSON.stringify(
|
graph
|
||||||
{
|
? {
|
||||||
...g,
|
...graph,
|
||||||
nodes: g.nodes.map((n: object) => ({ ...n, tmp: undefined, state: undefined }))
|
nodes: graph.nodes.map((n: object) => ({ ...n, tmp: undefined, state: undefined }))
|
||||||
},
|
}
|
||||||
null,
|
: null
|
||||||
2
|
);
|
||||||
);
|
|
||||||
}
|
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<pre>
|
<div class="overflow-auto p-2">
|
||||||
{graph ? convert(graph) : "No graph loaded"}
|
{#if data}
|
||||||
</pre>
|
<JsonViewer value={data} path="graph" />
|
||||||
|
{:else}
|
||||||
|
<span class="font-mono text-xs text-neutral-500">No graph loaded</span>
|
||||||
|
{/if}
|
||||||
|
</div>
|
||||||
|
|||||||
@@ -4,7 +4,7 @@
|
|||||||
import Grid from '$lib/grid';
|
import Grid from '$lib/grid';
|
||||||
import { debounceAsyncFunction } from '$lib/helpers';
|
import { debounceAsyncFunction } from '$lib/helpers';
|
||||||
import { createKeyMap } from '$lib/helpers/createKeyMap';
|
import { createKeyMap } from '$lib/helpers/createKeyMap';
|
||||||
import { debugNode } from '$lib/node-registry/debugNode.js';
|
import { debugNode } from '$lib/node-registry/debugNode';
|
||||||
import { IndexDBCache, RemoteNodeRegistry } from '$lib/node-registry/index';
|
import { IndexDBCache, RemoteNodeRegistry } from '$lib/node-registry/index';
|
||||||
import NodeStore from '$lib/node-store/NodeStore.svelte';
|
import NodeStore from '$lib/node-store/NodeStore.svelte';
|
||||||
import PerformanceViewer from '$lib/performance/PerformanceViewer.svelte';
|
import PerformanceViewer from '$lib/performance/PerformanceViewer.svelte';
|
||||||
@@ -321,7 +321,7 @@
|
|||||||
hidden={!appSettings.value.debug.advancedMode}
|
hidden={!appSettings.value.debug.advancedMode}
|
||||||
icon="i-[tabler--code]"
|
icon="i-[tabler--code]"
|
||||||
>
|
>
|
||||||
<GraphSource graph={pm.graph ?? manager?.serialize()} />
|
<GraphSource graph={manager?.serialize()} />
|
||||||
</Panel>
|
</Panel>
|
||||||
<Panel
|
<Panel
|
||||||
id="benchmark"
|
id="benchmark"
|
||||||
|
|||||||
+204
@@ -0,0 +1,204 @@
|
|||||||
|
# Nodarium — LLM Reference
|
||||||
|
|
||||||
|
## What It Is
|
||||||
|
|
||||||
|
Nodarium is a **node-based visual programming editor**. Users wire together nodes on a 2D canvas; each node is a WebAssembly module that receives typed inputs and produces typed outputs. A live Three.js viewer renders the resulting 3D geometry/paths/instances.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Repository Layout
|
||||||
|
|
||||||
|
```
|
||||||
|
/
|
||||||
|
├── app/ # SvelteKit web app
|
||||||
|
│ └── src/
|
||||||
|
│ ├── routes/+page.svelte # App entry point
|
||||||
|
│ └── lib/
|
||||||
|
│ ├── graph-interface/ # Canvas editor (UI + state)
|
||||||
|
│ ├── runtime/ # WASM execution engine
|
||||||
|
│ ├── node-registry/ # Fetch & cache node definitions
|
||||||
|
│ ├── project-manager/ # IndexDB persistence
|
||||||
|
│ ├── result-viewer/ # Three.js 3D output
|
||||||
|
│ ├── sidebar/ # UI panels
|
||||||
|
│ └── settings/ # App + graph settings
|
||||||
|
├── packages/
|
||||||
|
│ ├── types/ # Shared TypeScript types + Zod schemas
|
||||||
|
│ ├── utils/ # Logging, hashing, WASM wrapping, perf
|
||||||
|
│ ├── ui/ # Reusable Svelte UI components
|
||||||
|
│ ├── planty/ # Tutorial system
|
||||||
|
│ └── macros/ # Build-time macros
|
||||||
|
└── docs/
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Core Architecture
|
||||||
|
|
||||||
|
```
|
||||||
|
User Interaction
|
||||||
|
└── GraphInterface
|
||||||
|
├── GraphState ← UI: selection, camera, mouse, clipboard
|
||||||
|
└── GraphManager ← Logic: nodes, edges, history, serialization
|
||||||
|
├── NodeRegistry ← fetches WASM definitions (remote API + IndexDB cache)
|
||||||
|
├── HistoryManager ← undo/redo (jsondiffpatch deltas)
|
||||||
|
└── emit('result') → RuntimeExecutor
|
||||||
|
└── node.execute(Int32Array) per node
|
||||||
|
└── ResultViewer (Three.js/Threlte)
|
||||||
|
```
|
||||||
|
|
||||||
|
**Event flow:**
|
||||||
|
1. User edits graph → GraphManager mutates state
|
||||||
|
2. GraphManager emits `save` → ProjectManager persists to IndexDB
|
||||||
|
3. GraphManager emits `result` → Runtime executes graph → Viewer updates
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Critical Files
|
||||||
|
|
||||||
|
| File | Role |
|
||||||
|
|------|------|
|
||||||
|
| `app/src/routes/+page.svelte` | Wires all systems; creates GraphManager, runtime, registry |
|
||||||
|
| `app/src/lib/graph-interface/graph-manager.svelte.ts` | Central graph logic: createNode, createEdge, serialize, load, history |
|
||||||
|
| `app/src/lib/graph-interface/graph-state.svelte.ts` | UI state: camera, selection, mouse, clipboard, groupSelectedNodes |
|
||||||
|
| `app/src/lib/graph-interface/graph/Graph.svelte` | Canvas renderer |
|
||||||
|
| `app/src/lib/graph-interface/node/Node.svelte` | 3D mesh node (Three.js shader) |
|
||||||
|
| `app/src/lib/graph-interface/node/NodeHTML.svelte` | HTML overlay: labels + parameters |
|
||||||
|
| `app/src/lib/graph-interface/node/NodeHeader.svelte` | Node title bar |
|
||||||
|
| `app/src/lib/graph-interface/keymaps.ts` | Keyboard shortcuts |
|
||||||
|
| `app/src/lib/graph-interface/helpers/nodeHelpers.ts` | Node height calculations |
|
||||||
|
| `app/src/lib/graph-interface/graph/colors.svelte.ts` | Socket type → color mapping |
|
||||||
|
| `app/src/lib/runtime/runtime-executor.ts` | Executes nodes in DAG order; expandGroups() |
|
||||||
|
| `app/src/lib/node-registry/index.ts` | RemoteNodeRegistry entry |
|
||||||
|
| `app/src/lib/node-registry/groupNode.ts` | Built-in group node definition |
|
||||||
|
| `app/src/lib/node-registry/debugNode.ts` | Built-in debug node |
|
||||||
|
| `app/src/lib/sidebar/panels/ActiveNodeSettings.svelte` | Per-node settings panel |
|
||||||
|
| `packages/types/src/types.ts` | Graph, NodeInstance, NodeDefinition, Edge, GroupDefinition |
|
||||||
|
| `packages/types/src/inputs.ts` | NodeInput union types (float, vec3, geometry, path, …) |
|
||||||
|
| `packages/utils/src/wasm.ts` | createWasmWrapper() — wraps WASM bytes into a NodeDefinition |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Key Types
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
// packages/types/src/types.ts
|
||||||
|
|
||||||
|
type NodeId = `${string}/${string}/${string}` // e.g. "max/plantarium/stem"
|
||||||
|
|
||||||
|
type NodeInstance = {
|
||||||
|
id: number
|
||||||
|
type: NodeId
|
||||||
|
position: [number, number]
|
||||||
|
props?: Record<string, number | number[]> // current parameter values
|
||||||
|
meta?: { title?: string; lastModified?: string }
|
||||||
|
state: NodeRuntimeState // runtime-only, NOT serialized
|
||||||
|
}
|
||||||
|
|
||||||
|
type NodeRuntimeState = {
|
||||||
|
type?: NodeDefinition // resolved definition
|
||||||
|
parents?: NodeInstance[]
|
||||||
|
children?: NodeInstance[]
|
||||||
|
x?: number; y?: number // interpolated position
|
||||||
|
mesh?: Mesh // Three.js mesh reference
|
||||||
|
ref?: HTMLElement
|
||||||
|
}
|
||||||
|
|
||||||
|
type NodeDefinition = {
|
||||||
|
id: NodeId
|
||||||
|
inputs?: Record<string, NodeInput>
|
||||||
|
outputs?: string[] // output type names
|
||||||
|
meta?: { title?: string; description?: string }
|
||||||
|
execute(input: Int32Array): Int32Array // WASM function
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edge: [fromNode, outputIndex, toNode, inputSocketName]
|
||||||
|
type Edge = [NodeInstance, number, NodeInstance, string]
|
||||||
|
|
||||||
|
type Graph = {
|
||||||
|
nodes: NodeInstance[]
|
||||||
|
edges: [number, number, number, string][] // serialized (IDs, not refs)
|
||||||
|
settings: Record<string, unknown>
|
||||||
|
groups: GroupDefinition[]
|
||||||
|
}
|
||||||
|
|
||||||
|
type GroupDefinition = {
|
||||||
|
id: number
|
||||||
|
nodes: NodeInstance[]
|
||||||
|
edges: Edge[]
|
||||||
|
inputs?: Record<string, NodeInput>
|
||||||
|
outputs?: string[]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### NodeInput socket types
|
||||||
|
`float` | `integer` | `boolean` | `select` | `seed` | `vec3` | `geometry` | `path` | `shape` | `color` | `*` (wildcard)
|
||||||
|
|
||||||
|
Each input can have: `value` (default), `label`, `hidden`, `external`, `setting` (link to graph setting), `accepts` (extra compatible types).
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Patterns & Conventions
|
||||||
|
|
||||||
|
### Svelte 5 reactivity
|
||||||
|
The codebase uses Svelte 5 runes throughout — `$state`, `$derived`, `$effect`. Collections use `SvelteMap<K,V>` and `SvelteSet<T>` (from `svelte/reactivity`) instead of plain Map/Set so that mutations trigger reactive updates.
|
||||||
|
|
||||||
|
### Context API
|
||||||
|
`GraphManager` and `GraphState` are provided via Svelte context (`setContext` / `getContext`) inside `GraphInterface`. All child components (Node, Edge, etc.) consume them via context rather than props.
|
||||||
|
|
||||||
|
### Edge representation
|
||||||
|
In memory, edges are `[NodeInstance, outputIndex, NodeInstance, inputSocketName]` — direct object references for fast traversal. On serialization (`Graph.edges`), they become `[nodeId, outputIndex, nodeId, inputSocketName]` (plain IDs).
|
||||||
|
|
||||||
|
### Socket compatibility
|
||||||
|
```typescript
|
||||||
|
areSocketsCompatible(outputType: string, inputType: string | string[]): boolean
|
||||||
|
// '*' wildcard matches any type; 'geometry' accepts ['geometry', 'instances']
|
||||||
|
```
|
||||||
|
|
||||||
|
### WASM execution interface
|
||||||
|
Every node exposes a single function: `execute(input: Int32Array): Int32Array`.
|
||||||
|
Data encoding (Plantarium):
|
||||||
|
- `[0, stemDepth, ...x,y,z,thickness]` — path
|
||||||
|
- `[1, vertexCount, faceCount, ...faces, ...vertices, ...normals]` — geometry
|
||||||
|
- `[2, vertexCount, faceCount, instanceCount, stemDepth, ...]` — instances
|
||||||
|
|
||||||
|
### Event emitter
|
||||||
|
`GraphManager extends EventEmitter<{ save, result, settings }>`. Subscribe with `manager.on('result', cb)`. Used to decouple the editor UI from runtime execution and persistence.
|
||||||
|
|
||||||
|
### History
|
||||||
|
Every mutation goes through `HistoryManager`. Call `this.history.save(this.serialize())` before mutations; undo/redo replays jsondiffpatch deltas.
|
||||||
|
|
||||||
|
### Internal node IDs
|
||||||
|
Built-in nodes use the `__internal/` namespace: `__internal/group/instance`, `__internal/node/debug`. Virtual boundary nodes use `__virtual/`: `__virtual/group/input`, `__virtual/group/output`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## In-Progress: Node Groups (`feat/group-node-own`)
|
||||||
|
|
||||||
|
Group selected nodes with **Ctrl+G**. A `GroupDefinition` is stored in `Graph.groups[]`; a group instance node (`__internal/group/instance`) referencing it by `props.groupId` replaces the selected nodes.
|
||||||
|
|
||||||
|
**Known gaps as of 2026-05-03:**
|
||||||
|
|
||||||
|
| Issue | Location |
|
||||||
|
|-------|----------|
|
||||||
|
| `createGroupNode()` called but not defined | `graph-state.svelte.ts:334` → missing in `graph-manager.svelte.ts` |
|
||||||
|
| Runtime expects `group.graph.nodes/edges`; schema has flat `nodes/edges` | `runtime-executor.ts` vs `types.ts` |
|
||||||
|
| Runtime expects `group.inputs` as array; schema defines it as `Record<string, NodeInput>` | Same mismatch |
|
||||||
|
| `enterGroupNode()` is a stub — no group navigation | `graph-state.svelte.ts` |
|
||||||
|
| `serialize()` writes parent-graph edges into group instead of group-internal edges | `graph-manager.svelte.ts` |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Dev Commands
|
||||||
|
|
||||||
|
Run from `app/`:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
npm run dev # start dev server (Vite)
|
||||||
|
npm run build # production build
|
||||||
|
npm run check # svelte-check + tsc
|
||||||
|
npm run lint # eslint
|
||||||
|
npm run test # unit (vitest) + e2e (playwright)
|
||||||
|
npm run test:unit # vitest only
|
||||||
|
npm run test:e2e # playwright only
|
||||||
|
npm run bench # benchmark runner
|
||||||
|
```
|
||||||
@@ -4,11 +4,12 @@ export type {
|
|||||||
Box,
|
Box,
|
||||||
Edge,
|
Edge,
|
||||||
Graph,
|
Graph,
|
||||||
|
GroupDefinition,
|
||||||
NodeDefinition,
|
NodeDefinition,
|
||||||
NodeId,
|
NodeId,
|
||||||
NodeInstance,
|
NodeInstance,
|
||||||
SerializedNode,
|
SerializedNode,
|
||||||
Socket
|
Socket
|
||||||
} from './types';
|
} from './types';
|
||||||
export { GraphSchema, NodeSchema } from './types';
|
export { GraphSchema, GroupSchema, NodeSchema } from './types';
|
||||||
export { NodeDefinitionSchema } from './types';
|
export { NodeDefinitionSchema } from './types';
|
||||||
|
|||||||
@@ -76,6 +76,19 @@ export type Socket = {
|
|||||||
|
|
||||||
export type Edge = [NodeInstance, number, NodeInstance, string];
|
export type Edge = [NodeInstance, number, NodeInstance, string];
|
||||||
|
|
||||||
|
export const GroupSchema = z.object({
|
||||||
|
id: z.number(),
|
||||||
|
nodes: z.array(NodeSchema),
|
||||||
|
edges: z.array(z.tuple([z.number(), z.number(), z.number(), z.string()])),
|
||||||
|
inputs: z.record(z.string(), NodeInputSchema).optional(),
|
||||||
|
outputs: z.array(z.object({
|
||||||
|
type: z.string(),
|
||||||
|
label: z.string().optional()
|
||||||
|
})).optional()
|
||||||
|
});
|
||||||
|
|
||||||
|
export type GroupDefinition = z.infer<typeof GroupSchema>;
|
||||||
|
|
||||||
export const GraphSchema = z.object({
|
export const GraphSchema = z.object({
|
||||||
id: z.number(),
|
id: z.number(),
|
||||||
meta: z
|
meta: z
|
||||||
@@ -86,7 +99,8 @@ export const GraphSchema = z.object({
|
|||||||
.optional(),
|
.optional(),
|
||||||
settings: z.record(z.string(), z.any()).optional(),
|
settings: z.record(z.string(), z.any()).optional(),
|
||||||
nodes: z.array(NodeSchema),
|
nodes: z.array(NodeSchema),
|
||||||
edges: z.array(z.tuple([z.number(), z.number(), z.number(), z.string()]))
|
edges: z.array(z.tuple([z.number(), z.number(), z.number(), z.string()])),
|
||||||
|
groups: z.array(GroupSchema)
|
||||||
});
|
});
|
||||||
|
|
||||||
export type Graph = z.infer<typeof GraphSchema>;
|
export type Graph = z.infer<typeof GraphSchema>;
|
||||||
|
|||||||
@@ -0,0 +1,137 @@
|
|||||||
|
<script module>
|
||||||
|
const cache = new Map<string, Record<string, boolean>>();
|
||||||
|
|
||||||
|
function getStore(root: string): Record<string, boolean> {
|
||||||
|
if (!cache.has(root)) {
|
||||||
|
try {
|
||||||
|
const raw = localStorage.getItem(`json_viewer:${root}`);
|
||||||
|
cache.set(root, raw ? JSON.parse(raw) : {});
|
||||||
|
} catch {
|
||||||
|
cache.set(root, {});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cache.get(root)!;
|
||||||
|
}
|
||||||
|
|
||||||
|
function readOpen(path: string, fallback: boolean): boolean {
|
||||||
|
const root = path.split('/')[0];
|
||||||
|
const store = getStore(root);
|
||||||
|
return path in store ? store[path] : fallback;
|
||||||
|
}
|
||||||
|
|
||||||
|
function writeOpen(path: string, value: boolean) {
|
||||||
|
const root = path.split('/')[0];
|
||||||
|
const store = getStore(root);
|
||||||
|
store[path] = value;
|
||||||
|
try {
|
||||||
|
localStorage.setItem(`json_viewer:${root}`, JSON.stringify(store));
|
||||||
|
} catch { /* quota exceeded etc */ }
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<script lang="ts">
|
||||||
|
import { browser } from '$app/environment';
|
||||||
|
import JsonViewer from './JsonViewer.svelte';
|
||||||
|
|
||||||
|
let {
|
||||||
|
value,
|
||||||
|
key,
|
||||||
|
depth = 0,
|
||||||
|
path = ''
|
||||||
|
}: { value: unknown; key?: string; depth?: number; path?: string } = $props();
|
||||||
|
|
||||||
|
const defaultOpen = $derived(depth < 4);
|
||||||
|
let open = $derived(browser && path ? readOpen(path, defaultOpen) : defaultOpen);
|
||||||
|
let flashing = $state(false);
|
||||||
|
|
||||||
|
const isArr = $derived(Array.isArray(value));
|
||||||
|
const isExpandable = $derived(value !== null && typeof value === 'object');
|
||||||
|
const open_bracket = $derived(isArr ? '[' : '{');
|
||||||
|
const close_bracket = $derived(isArr ? ']' : '}');
|
||||||
|
const items = $derived.by(() => {
|
||||||
|
if (isArr) {
|
||||||
|
return (value as unknown[]).map((v, i) => [String(i), v] as [string, unknown]);
|
||||||
|
}
|
||||||
|
if (value !== null && typeof value === 'object') {
|
||||||
|
return Object.entries(value as Record<string, unknown>).filter(
|
||||||
|
([, v]) => v !== undefined
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return [] as [string, unknown][];
|
||||||
|
});
|
||||||
|
const showKeys = $derived(!isArr || typeof items[0]?.[1] === "object")
|
||||||
|
|
||||||
|
function toggle(next: boolean) {
|
||||||
|
open = next;
|
||||||
|
if (browser && path) writeOpen(path, next);
|
||||||
|
}
|
||||||
|
|
||||||
|
let prevJson = '';
|
||||||
|
let flashTimeout: ReturnType<typeof setTimeout> | null = null;
|
||||||
|
|
||||||
|
$effect(() => {
|
||||||
|
const json = JSON.stringify(value);
|
||||||
|
if (prevJson && json !== prevJson) {
|
||||||
|
if (flashTimeout) clearTimeout(flashTimeout);
|
||||||
|
flashing = true;
|
||||||
|
flashTimeout = setTimeout(() => {
|
||||||
|
flashing = false;
|
||||||
|
flashTimeout = null;
|
||||||
|
}, 500);
|
||||||
|
}
|
||||||
|
prevJson = json;
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<span
|
||||||
|
class="font-mono text-xs leading-[1.6] rounded transition-[background-color] duration-500"
|
||||||
|
class:bg-layer-3={flashing}
|
||||||
|
>
|
||||||
|
{#if key !== undefined}
|
||||||
|
<span class="text-text">{key}</span><span class="text-text/40">: </span>
|
||||||
|
{/if}
|
||||||
|
|
||||||
|
{#if isExpandable}
|
||||||
|
{#if items.length === 0}
|
||||||
|
<span class="text-text/50">{open_bracket}{close_bracket}</span>
|
||||||
|
{:else if open}
|
||||||
|
{#if depth > 0}
|
||||||
|
<button class="w-3 text-text/50 hover:text-text" onclick={() => toggle(false)}>
|
||||||
|
▼
|
||||||
|
</button>
|
||||||
|
{/if}
|
||||||
|
<span class="text-text/50">{open_bracket}</span>
|
||||||
|
<div class="pl-4 border-l border-outline">
|
||||||
|
{#each items as [k, v], i (k)}
|
||||||
|
<div>
|
||||||
|
<JsonViewer
|
||||||
|
value={v}
|
||||||
|
key={showKeys ? k : undefined }
|
||||||
|
depth={depth + 1}
|
||||||
|
path={path ? `${path}/${k}` : k}
|
||||||
|
/>{#if i < items.length - 1}<span class="text-text/20">,</span>{/if}
|
||||||
|
</div>
|
||||||
|
{/each}
|
||||||
|
</div>
|
||||||
|
<span class="text-text/50">{close_bracket}</span>
|
||||||
|
{:else}
|
||||||
|
<button
|
||||||
|
class="inline text-text/50 hover:text-text"
|
||||||
|
onclick={() => toggle(true)}
|
||||||
|
>
|
||||||
|
<span class="w-3 inline-block">▶</span>
|
||||||
|
{open_bracket}<span class="text-text/40 mx-1">{items.length}</span>{close_bracket}
|
||||||
|
</button>
|
||||||
|
{/if}
|
||||||
|
{:else if value === null}
|
||||||
|
<span class="text-emerald-500!">null</span>
|
||||||
|
{:else if typeof value === 'boolean'}
|
||||||
|
<span class="text-blue-500!">{value}</span>
|
||||||
|
{:else if typeof value === 'number'}
|
||||||
|
<span class="text-orange-400!">{value}</span>
|
||||||
|
{:else if typeof value === 'string'}
|
||||||
|
<span class="text-emerald-500!">"{value}"</span>
|
||||||
|
{:else}
|
||||||
|
<span class="text-text/70">{String(value)}</span>
|
||||||
|
{/if}
|
||||||
|
</span>
|
||||||
@@ -7,6 +7,7 @@ export { default as InputShape } from './inputs/InputShape.svelte';
|
|||||||
export { default as InputVec3 } from './inputs/InputVec3.svelte';
|
export { default as InputVec3 } from './inputs/InputVec3.svelte';
|
||||||
|
|
||||||
export { default as Details } from './Details.svelte';
|
export { default as Details } from './Details.svelte';
|
||||||
|
export { default as JsonViewer } from './JsonViewer.svelte';
|
||||||
export { default as ShortCut } from './ShortCut.svelte';
|
export { default as ShortCut } from './ShortCut.svelte';
|
||||||
|
|
||||||
import Input from './Input.svelte';
|
import Input from './Input.svelte';
|
||||||
|
|||||||
@@ -6,6 +6,13 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
let { options = [], value = $bindable(0), id = '' }: Props = $props();
|
let { options = [], value = $bindable(0), id = '' }: Props = $props();
|
||||||
|
|
||||||
|
$effect(() => {
|
||||||
|
console.log({ options, value });
|
||||||
|
if (typeof value !== typeof options[0]) {
|
||||||
|
console.trace('WARNING: value type does not match options type');
|
||||||
|
}
|
||||||
|
});
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<select {id} bind:value class="bg-layer-2 text-text">
|
<select {id} bind:value class="bg-layer-2 text-text">
|
||||||
|
|||||||
@@ -8,6 +8,7 @@
|
|||||||
InputSelect,
|
InputSelect,
|
||||||
InputShape,
|
InputShape,
|
||||||
InputVec3,
|
InputVec3,
|
||||||
|
JsonViewer,
|
||||||
ShortCut
|
ShortCut
|
||||||
} from '$lib';
|
} from '$lib';
|
||||||
import Section from './Section.svelte';
|
import Section from './Section.svelte';
|
||||||
@@ -25,6 +26,32 @@
|
|||||||
let colorValue = $state<[number, number, number]>([59, 130, 246]);
|
let colorValue = $state<[number, number, number]>([59, 130, 246]);
|
||||||
let mirrorShape = $state(true);
|
let mirrorShape = $state(true);
|
||||||
let detailsOpen = $state(false);
|
let detailsOpen = $state(false);
|
||||||
|
let jsonValue = $state({
|
||||||
|
id: 1,
|
||||||
|
nodes: [{ id: 0, type: 'max/test/node', position: [0, 0] }, {
|
||||||
|
id: 1,
|
||||||
|
type: 'max/test/other',
|
||||||
|
position: [100, 50]
|
||||||
|
}],
|
||||||
|
edges: [[0, 0, 1, 'input']],
|
||||||
|
groups: [],
|
||||||
|
settings: { seed: 42, enabled: true }
|
||||||
|
});
|
||||||
|
|
||||||
|
function randomlyUpdateJson() {
|
||||||
|
const rand = Math.floor(Math.random() * 5);
|
||||||
|
if (rand === 0) {
|
||||||
|
jsonValue.nodes[0].position[0] += 1;
|
||||||
|
} else if (rand === 1) {
|
||||||
|
jsonValue.nodes[0].position[1] += 1;
|
||||||
|
} else if (rand === 2) {
|
||||||
|
jsonValue.settings.seed += 1;
|
||||||
|
} else if (rand === 3) {
|
||||||
|
jsonValue.settings.enabled = !jsonValue.settings.enabled;
|
||||||
|
} else if (rand === 4) {
|
||||||
|
jsonValue.id += Math.floor(Math.random() * 10 - 5);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let points = $state([]);
|
let points = $state([]);
|
||||||
let theme = $state('dark');
|
let theme = $state('dark');
|
||||||
@@ -56,6 +83,7 @@
|
|||||||
</Section>
|
</Section>
|
||||||
|
|
||||||
<Section title="Select" value={d}>
|
<Section title="Select" value={d}>
|
||||||
|
<i>Select with simple values</i>
|
||||||
<InputSelect bind:value={selectValue} {options} />
|
<InputSelect bind:value={selectValue} {options} />
|
||||||
</Section>
|
</Section>
|
||||||
|
|
||||||
@@ -86,6 +114,23 @@
|
|||||||
</Details>
|
</Details>
|
||||||
</Section>
|
</Section>
|
||||||
|
|
||||||
|
<Section title="JsonViewer">
|
||||||
|
{#snippet header()}
|
||||||
|
<button
|
||||||
|
onclick={() => randomlyUpdateJson()}
|
||||||
|
class="-mt-1 bg-layer-2 p-1 px-2 rounded-sm cursor-pointer"
|
||||||
|
>
|
||||||
|
update
|
||||||
|
</button>
|
||||||
|
{/snippet}
|
||||||
|
<div class="w-64 bg-layer-1 p-2 rounded">
|
||||||
|
<JsonViewer
|
||||||
|
value={jsonValue}
|
||||||
|
path="demo"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</Section>
|
||||||
|
|
||||||
<Section title="Shortcut">
|
<Section title="Shortcut">
|
||||||
<div class="flex gap-4">
|
<div class="flex gap-4">
|
||||||
<ShortCut ctrl key="S" />
|
<ShortCut ctrl key="S" />
|
||||||
|
|||||||
Generated
-6
@@ -4,12 +4,6 @@ settings:
|
|||||||
autoInstallPeers: true
|
autoInstallPeers: true
|
||||||
excludeLinksFromLockfile: false
|
excludeLinksFromLockfile: false
|
||||||
|
|
||||||
catalogs:
|
|
||||||
default:
|
|
||||||
chokidar-cli:
|
|
||||||
specifier: github:open-cli-tools/chokidar-cli#semver:v4.0.0
|
|
||||||
version: 4.0.0
|
|
||||||
|
|
||||||
importers:
|
importers:
|
||||||
|
|
||||||
.:
|
.:
|
||||||
|
|||||||
Reference in New Issue
Block a user