luboslenco
138 lines
3.4 KiB
Haxe
138 lines
3.4 KiB
Haxe
package arm.logic;
|
|
|
|
import zui.Nodes;
|
|
import arm.logic.LogicNode;
|
|
import arm.logic.LogicParser.f32;
|
|
import arm.Translator._tr;
|
|
|
|
@:keep
|
|
class TilingNode extends LogicNode {
|
|
|
|
var result: kha.Image = null;
|
|
static var image: kha.Image = null;
|
|
static var prompt = "";
|
|
static var strength = 0.5;
|
|
static var auto = true;
|
|
|
|
public function new(tree: LogicTree) {
|
|
super(tree);
|
|
|
|
init();
|
|
}
|
|
|
|
public static function init() {
|
|
if (image == null) {
|
|
image = kha.Image.createRenderTarget(Config.getTextureResX(), Config.getTextureResY());
|
|
}
|
|
}
|
|
|
|
public static function buttons(ui: zui.Zui, nodes: zui.Nodes, node: zui.Nodes.TNode) {
|
|
auto = node.buttons[0].default_value;
|
|
if (!auto) {
|
|
strength = ui.slider(zui.Zui.handle("tilingnode_0", {value: strength}), tr("strength"), 0, 1, true);
|
|
prompt = zui.Ext.textArea(ui, zui.Zui.handle("tilingnode_1"), true, tr("prompt"), true);
|
|
node.buttons[1].height = 1 + prompt.split("\n").length;
|
|
}
|
|
else node.buttons[1].height = 0;
|
|
}
|
|
|
|
override function getAsImage(from: Int, done: kha.Image->Void) {
|
|
inputs[0].getAsImage(function(source: kha.Image) {
|
|
image.g2.begin(false);
|
|
image.g2.drawScaledImage(source, 0, 0, Config.getTextureResX(), Config.getTextureResY());
|
|
image.g2.end();
|
|
|
|
Console.progress(tr("Processing") + " - " + tr("Tiling"));
|
|
App.notifyOnNextFrame(function() {
|
|
function _done(image: kha.Image) {
|
|
result = image;
|
|
done(image);
|
|
}
|
|
auto ? InpaintNode.texsynthInpaint(image, true, null, _done) : sdTiling(image, -1, _done);
|
|
});
|
|
});
|
|
}
|
|
|
|
override public function getCachedImage(): kha.Image {
|
|
return result;
|
|
}
|
|
|
|
public static function sdTiling(image: kha.Image, seed: Int/* = -1*/, done: kha.Image->Void) {
|
|
@:privateAccess TextToPhotoNode.tiling = false;
|
|
var tile = kha.Image.createRenderTarget(512, 512);
|
|
tile.g2.begin(false);
|
|
tile.g2.drawScaledImage(image, -256, -256, 512, 512);
|
|
tile.g2.drawScaledImage(image, 256, -256, 512, 512);
|
|
tile.g2.drawScaledImage(image, -256, 256, 512, 512);
|
|
tile.g2.drawScaledImage(image, 256, 256, 512, 512);
|
|
tile.g2.end();
|
|
|
|
var bytes = haxe.io.Bytes.alloc(512 * 512);
|
|
for (i in 0...512 * 512) {
|
|
var x = i % 512;
|
|
var y = Std.int(i / 512);
|
|
var l = y < 256 ? y : (511 - y);
|
|
bytes.set(i, (x > 256 - l && x < 256 + l) ? 0 : 255);
|
|
}
|
|
// for (i in 0...512 * 512) bytes.set(i, 255);
|
|
// for (x in (256 - 32)...(256 + 32)) {
|
|
// for (y in 0...512) {
|
|
// bytes.set(y * 512 + x, 0);
|
|
// }
|
|
// }
|
|
// for (x in 0...512) {
|
|
// for (y in (256 - 32)...(256 + 32)) {
|
|
// bytes.set(y * 512 + x, 0);
|
|
// }
|
|
// }
|
|
var mask = kha.Image.fromBytes(bytes, 512, 512, kha.graphics4.TextureFormat.L8);
|
|
|
|
@:privateAccess InpaintNode.prompt = prompt;
|
|
@:privateAccess InpaintNode.strength = strength;
|
|
if (seed >= 0) RandomNode.setSeed(seed);
|
|
InpaintNode.sdInpaint(tile, mask, done);
|
|
}
|
|
|
|
public static var def: TNode = {
|
|
id: 0,
|
|
name: _tr("Tiling"),
|
|
type: "TilingNode",
|
|
x: 0,
|
|
y: 0,
|
|
color: 0xff4982a0,
|
|
inputs: [
|
|
{
|
|
id: 0,
|
|
node_id: 0,
|
|
name: _tr("Color"),
|
|
type: "RGBA",
|
|
color: 0xffc7c729,
|
|
default_value: f32([0.0, 0.0, 0.0, 1.0])
|
|
}
|
|
],
|
|
outputs: [
|
|
{
|
|
id: 0,
|
|
node_id: 0,
|
|
name: _tr("Color"),
|
|
type: "RGBA",
|
|
color: 0xffc7c729,
|
|
default_value: f32([0.0, 0.0, 0.0, 1.0])
|
|
}
|
|
],
|
|
buttons: [
|
|
{
|
|
name: _tr("auto"),
|
|
type: "BOOL",
|
|
default_value: true,
|
|
output: 0
|
|
},
|
|
{
|
|
name: "arm.logic.TilingNode.buttons",
|
|
type: "CUSTOM",
|
|
height: 0
|
|
}
|
|
]
|
|
};
|
|
}
|