feat: extract graph-interface into seperate package
This commit is contained in:
56
app/src/lib/graph-templates/tree.ts
Normal file
56
app/src/lib/graph-templates/tree.ts
Normal file
@@ -0,0 +1,56 @@
|
||||
import type { Graph, Node } from "@nodes/types";
|
||||
|
||||
export function tree(depth: number): Graph {
|
||||
|
||||
const nodes: Node[] = [
|
||||
{
|
||||
id: 0,
|
||||
type: "max/plantarium/output",
|
||||
position: [0, 0]
|
||||
},
|
||||
{
|
||||
id: 1,
|
||||
type: "max/plantarium/math",
|
||||
position: [-40, -10]
|
||||
}
|
||||
]
|
||||
|
||||
const edges: [number, number, number, string][] = [
|
||||
[1, 0, 0, "input"]
|
||||
];
|
||||
|
||||
for (let d = 0; d < depth; d++) {
|
||||
const amount = Math.pow(2, d);
|
||||
for (let i = 0; i < amount; i++) {
|
||||
|
||||
const id0 = amount * 2 + i * 2;
|
||||
const id1 = amount * 2 + i * 2 + 1;
|
||||
|
||||
const parent = Math.floor(id0 / 2);
|
||||
|
||||
const x = -(d + 1) * 50 - 40;
|
||||
const y = i * 80 - amount * 35;
|
||||
|
||||
nodes.push({
|
||||
id: id0,
|
||||
type: "max/plantarium/math",
|
||||
position: [x, y],
|
||||
});
|
||||
edges.push([id0, 0, parent, "a"]);
|
||||
nodes.push({
|
||||
id: id1,
|
||||
type: "max/plantarium/math",
|
||||
position: [x, y + 35],
|
||||
});
|
||||
edges.push([id1, 0, parent, "b"]);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return {
|
||||
id: Math.floor(Math.random() * 100000),
|
||||
nodes,
|
||||
edges
|
||||
};
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user