edgevoid/armorpaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Refactor

Browse Source
This commit is contained in:
luboslenco
2024-02-06 19:32:21 +01:00
parent fd9bdbeec5
commit 779218d9b8
110 changed files with 2367 additions and 2360 deletions
Download Patch File Download Diff File Expand all files Collapse all files
armorlab/Sources/nodes/ImageTextureNode.ts
+1 -1
View File
@@ -20,7 +20,7 @@ class ImageTextureNode extends LogicNode {
return image;
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Image Texture"),
type: "ImageTextureNode",
armorlab/Sources/nodes/InpaintNode.ts
+8 -8
View File
@@ -40,11 +40,11 @@ class InpaintNode extends LogicNode {
}
}
static buttons = (ui: ZuiRaw, nodes: NodesRaw, node: TNode) => {
static buttons = (ui: zui_t, nodes: zui_nodes_t, node: zui_node_t) => {
InpaintNode.auto = node.buttons[0].default_value == 0 ? false : true;
if (!InpaintNode.auto) {
InpaintNode.strength = Zui.slider(Zui.handle("inpaintnode_0", {value: InpaintNode.strength}), tr("strength"), 0, 1, true);
InpaintNode.prompt = Zui.textArea(Zui.handle("inpaintnode_1"), Align.Left, true, tr("prompt"), true);
InpaintNode.strength = zui_slider(zui_handle("inpaintnode_0", { value: InpaintNode.strength }), tr("strength"), 0, 1, true);
InpaintNode.prompt = zui_text_area(zui_handle("inpaintnode_1"), Align.Left, true, tr("prompt"), true);
node.buttons[1].height = 1 + InpaintNode.prompt.split("\n").length;
}
else node.buttons[1].height = 0;
@@ -68,13 +68,13 @@ class InpaintNode extends LogicNode {
Base.notifyOnNextFrame(() => {
this.inputs[0].getAsImage((source: image_t) => {
if (Base.pipeCopy == null) Base.makePipe();
if (ConstData.screenAlignedVB == null) ConstData.createScreenAlignedData();
if (const_data_screen_aligned_vb == null) const_data_create_screen_aligned_data();
g4_begin(InpaintNode.image.g4);
g4_set_pipeline(Base.pipeInpaintPreview);
g4_set_tex(Base.tex0InpaintPreview, source);
g4_set_tex(Base.texaInpaintPreview, InpaintNode.mask);
g4_set_vertex_buffer(ConstData.screenAlignedVB);
g4_set_index_buffer(ConstData.screenAlignedIB);
g4_set_vertex_buffer(const_data_screen_aligned_vb);
g4_set_index_buffer(const_data_screen_aligned_ib);
g4_draw();
g4_end();
});
@@ -106,7 +106,7 @@ class InpaintNode extends LogicNode {
let u8 = new Uint8Array(bytes_img);
let f32mask = new Float32Array(4 * 64 * 64);
Data.getBlob("models/sd_vae_encoder.quant.onnx", (vae_encoder_blob: ArrayBuffer) => {
data_get_blob("models/sd_vae_encoder.quant.onnx", (vae_encoder_blob: ArrayBuffer) => {
// for (let x = 0; x < Math.floor(image.width / 512); ++x) {
// for (let y = 0; y < Math.floor(image.height / 512); ++y) {
let x = 0;
@@ -175,7 +175,7 @@ class InpaintNode extends LogicNode {
});
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Inpaint"),
type: "InpaintNode",
armorlab/Sources/nodes/PhotoToPBRNode.ts
+2 -2
View File
@@ -67,7 +67,7 @@ class PhotoToPBRNode extends LogicNode {
f32a[i + PhotoToPBRNode.tileWithBorderW * PhotoToPBRNode.tileWithBorderW * 2] = (u8a[i * 4 + 2] / 255 - 0.5) / 0.5;
}
Data.getBlob("models/photo_to_" + PhotoToPBRNode.modelNames[from] + ".quant.onnx", (model_blob: ArrayBuffer) => {
data_get_blob("models/photo_to_" + PhotoToPBRNode.modelNames[from] + ".quant.onnx", (model_blob: ArrayBuffer) => {
let buf = Krom.mlInference(model_blob, [f32a.buffer], null, null, Config.raw.gpu_inference);
let ar = new Float32Array(buf);
let u8a = new Uint8Array(4 * PhotoToPBRNode.tileW * PhotoToPBRNode.tileW);
@@ -168,7 +168,7 @@ class PhotoToPBRNode extends LogicNode {
}
///end
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Photo to PBR"),
type: "PhotoToPBRNode",
armorlab/Sources/nodes/RGBNode.ts
+1 -1
View File
@@ -30,7 +30,7 @@ class RGBNode extends LogicNode {
return this.image;
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("RGB"),
type: "RGBNode",
armorlab/Sources/nodes/TextToPhotoNode.ts
+8 -8
View File
@@ -23,16 +23,16 @@ class TextToPhotoNode extends LogicNode {
return TextToPhotoNode.image;
}
static buttons = (ui: ZuiRaw, nodes: NodesRaw, node: TNode) => {
static buttons = (ui: zui_t, nodes: zui_nodes_t, node: zui_node_t) => {
TextToPhotoNode.tiling = node.buttons[0].default_value == 0 ? false : true;
TextToPhotoNode.prompt = Zui.textArea(Zui.handle("texttophotonode_0"), Align.Left, true, tr("prompt"), true);
TextToPhotoNode.prompt = zui_text_area(zui_handle("texttophotonode_0"), Align.Left, true, tr("prompt"), true);
node.buttons[1].height = TextToPhotoNode.prompt.split("\n").length;
}
static stableDiffusion = (prompt: string, done: (img: image_t)=>void, inpaintLatents: Float32Array = null, offset = 0, upscale = true, mask: Float32Array = null, latents_orig: Float32Array = null) => {
Data.getBlob("models/sd_text_encoder.quant.onnx", (_text_encoder_blob: ArrayBuffer) => {
Data.getBlob("models/sd_unet.quant.onnx", (_unet_blob: ArrayBuffer) => {
Data.getBlob("models/sd_vae_decoder.quant.onnx", (_vae_decoder_blob: ArrayBuffer) => {
data_get_blob("models/sd_text_encoder.quant.onnx", (_text_encoder_blob: ArrayBuffer) => {
data_get_blob("models/sd_unet.quant.onnx", (_unet_blob: ArrayBuffer) => {
data_get_blob("models/sd_vae_decoder.quant.onnx", (_vae_decoder_blob: ArrayBuffer) => {
TextToPhotoNode.text_encoder_blob = _text_encoder_blob;
TextToPhotoNode.unet_blob = _unet_blob;
TextToPhotoNode.vae_decoder_blob = _vae_decoder_blob;
@@ -188,11 +188,11 @@ class TextToPhotoNode extends LogicNode {
}
if (counter == (51 - offset)) {
App.removeRender2D(processing);
app_remove_render_2d(processing);
done(latents);
}
}
App.notifyOnRender2D(processing);
app_notify_on_render_2d(processing);
}
static vaeDecoder = (latents: Float32Array, upscale: bool, done: (img: image_t)=>void) => {
@@ -241,7 +241,7 @@ class TextToPhotoNode extends LogicNode {
});
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Text to Photo"),
type: "TextToPhotoNode",
armorlab/Sources/nodes/TilingNode.ts
+4 -4
View File
@@ -18,11 +18,11 @@ class TilingNode extends LogicNode {
}
}
static buttons = (ui: ZuiRaw, nodes: NodesRaw, node: TNode) => {
static buttons = (ui: zui_t, nodes: zui_nodes_t, node: zui_node_t) => {
TilingNode.auto = node.buttons[0].default_value == 0 ? false : true;
if (!TilingNode.auto) {
TilingNode.strength = Zui.slider(Zui.handle("tilingnode_0", {value: TilingNode.strength}), tr("strength"), 0, 1, true);
TilingNode.prompt = Zui.textArea(Zui.handle("tilingnode_1"), Align.Left, true, tr("prompt"), true);
TilingNode.strength = zui_slider(zui_handle("tilingnode_0", { value: TilingNode.strength }), tr("strength"), 0, 1, true);
TilingNode.prompt = zui_text_area(zui_handle("tilingnode_1"), Align.Left, true, tr("prompt"), true);
node.buttons[1].height = 1 + TilingNode.prompt.split("\n").length;
}
else node.buttons[1].height = 0;
@@ -85,7 +85,7 @@ class TilingNode extends LogicNode {
InpaintNode.sdInpaint(tile, mask, done);
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Tiling"),
type: "TilingNode",
armorlab/Sources/nodes/UpscaleNode.ts
+2 -2
View File
@@ -31,7 +31,7 @@ class UpscaleNode extends LogicNode {
}
static loadBlob = (done: ()=>void) => {
Data.getBlob("models/esrgan.quant.onnx", (_esrgan_blob: ArrayBuffer) => {
data_get_blob("models/esrgan.quant.onnx", (_esrgan_blob: ArrayBuffer) => {
UpscaleNode.esrgan_blob = _esrgan_blob;
done();
});
@@ -119,7 +119,7 @@ class UpscaleNode extends LogicNode {
return result;
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Upscale"),
type: "UpscaleNode",
armorlab/Sources/nodes/VarianceNode.ts
+4 -4
View File
@@ -18,8 +18,8 @@ class VarianceNode extends LogicNode {
}
}
static buttons = (ui: ZuiRaw, nodes: NodesRaw, node: TNode) => {
VarianceNode.prompt = Zui.textArea(Zui.handle("variancenode_0"), Align.Left, true, tr("prompt"), true);
static buttons = (ui: zui_t, nodes: zui_nodes_t, node: zui_node_t) => {
VarianceNode.prompt = zui_text_area(zui_handle("variancenode_0"), Align.Left, true, tr("prompt"), true);
node.buttons[0].height = VarianceNode.prompt.split("\n").length;
}
@@ -42,7 +42,7 @@ class VarianceNode extends LogicNode {
Console.progress(tr("Processing") + " - " + tr("Variance"));
Base.notifyOnNextFrame(() => {
Data.getBlob("models/sd_vae_encoder.quant.onnx", (vae_encoder_blob: ArrayBuffer) => {
data_get_blob("models/sd_vae_encoder.quant.onnx", (vae_encoder_blob: ArrayBuffer) => {
let latents_buf = Krom.mlInference(vae_encoder_blob, [f32a.buffer], [[1, 3, 512, 512]], [1, 4, 64, 64], Config.raw.gpu_inference);
let latents = new Float32Array(latents_buf);
for (let i = 0; i < latents.length; ++i) {
@@ -75,7 +75,7 @@ class VarianceNode extends LogicNode {
return VarianceNode.image;
}
static def: TNode = {
static def: zui_node_t = {
id: 0,
name: _tr("Variance"),
type: "VarianceNode",