luboslenco
62 lines
2.0 KiB
TypeScript
62 lines
2.0 KiB
TypeScript
/// <reference path="./nodes/image_texture_node.ts"/>
|
|
/// <reference path="./nodes/rgb_node.ts"/>
|
|
/// <reference path="./nodes/inpaint_node.ts"/>
|
|
/// <reference path="./nodes/photo_to_pbr_node.ts"/>
|
|
/// <reference path="./nodes/text_to_photo_node.ts"/>
|
|
/// <reference path="./nodes/tiling_node.ts"/>
|
|
/// <reference path="./nodes/upscale_node.ts"/>
|
|
/// <reference path="./nodes/variance_node.ts"/>
|
|
|
|
let nodes_brush_categories: string[] = [_tr("Input"), _tr("Model")];
|
|
|
|
let nodes_brush_input: zui_node_t[] = [
|
|
image_texture_node_def,
|
|
rgb_node_def,
|
|
];
|
|
|
|
let nodes_brush_model: zui_node_t[] = [
|
|
inpaint_node_def,
|
|
photo_to_pbr_node_def,
|
|
text_to_photo_node_def,
|
|
tiling_node_def,
|
|
upscale_node_def,
|
|
variance_node_def,
|
|
];
|
|
|
|
let nodes_brush_list: node_list_t[] = [
|
|
nodes_brush_input,
|
|
nodes_brush_model
|
|
]
|
|
|
|
let nodes_brush_creates: map_t<string, any>;
|
|
|
|
function nodes_brush_init() {
|
|
nodes_brush_creates = map_create();
|
|
map_set(nodes_brush_creates, "brush_output_node", brush_output_node_create);
|
|
map_set(nodes_brush_creates, "image_texture_node", image_texture_node_create);
|
|
map_set(nodes_brush_creates, "rgb_node", rgb_node_create);
|
|
map_set(nodes_brush_creates, "inpaint_node", inpaint_node_create);
|
|
map_set(nodes_brush_creates, "photo_to_pbr_node", photo_to_pbr_node_create);
|
|
map_set(nodes_brush_creates, "text_to_photo_node", text_to_photo_node_create);
|
|
map_set(nodes_brush_creates, "tiling_node", tiling_node_create);
|
|
map_set(nodes_brush_creates, "upscale_node", upscale_node_create);
|
|
map_set(nodes_brush_creates, "variance_node", variance_node_create);
|
|
}
|
|
|
|
function nodes_brush_create_node(node_type: string): zui_node_t {
|
|
for (let i: i32 = 0; i < nodes_brush_list.length; ++i) {
|
|
let c = nodes_brush_list[i];
|
|
for (let j: i32 = 0; j < c.length; ++j) {
|
|
let n = c[j];
|
|
if (n.type == node_type) {
|
|
let canvas = project_canvas;
|
|
let nodes = project_nodes;
|
|
let node = ui_nodes_make_node(n, nodes, canvas);
|
|
array_push(canvas.nodes, node);
|
|
return node;
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|