luboslenco
58 lines
1.6 KiB
JavaScript
58 lines
1.6 KiB
JavaScript
// Example-based texture synthesis written in Rust
|
|
// https://github.com/EmbarkStudios/texture-synthesis
|
|
|
|
// texsynth library is linked with armorcore until multi-threaded execution is supported via webassembly
|
|
// https://github.com/armory3d/armorpaint_plugins/tree/master/plugin_texsynth
|
|
|
|
let plugin = new arm.Plugin();
|
|
let h1 = new zui.Handle();
|
|
|
|
plugin.drawUI = function(ui) {
|
|
if (ui.panel(h1, "Texture Synthesis")) {
|
|
if (ui.button("Inpaint")) {
|
|
texsynthInpaint(false);
|
|
}
|
|
if (ui.button("Tiling")) {
|
|
texsynthInpaint(true);
|
|
}
|
|
}
|
|
};
|
|
|
|
function texsynthInpaint(tiling) {
|
|
let w = arm.Config.getTextureResX();
|
|
let h = arm.Config.getTextureResY();
|
|
let l = arm.Context.layer;
|
|
|
|
let bytes_img = l.texpaint.getPixels().b.bufferValue;
|
|
let bytes_mask = l.texpaint_mask != null ? l.texpaint_mask.getPixels().b.bufferValue : new ArrayBuffer(w * h);
|
|
let bytes_out = new arm.Bytes(new ArrayBuffer(w * h * 4));
|
|
Krom.texsynthInpaint(w, h, bytes_out.b.bufferValue, bytes_img, bytes_mask, tiling);
|
|
let image = arm.Image.fromBytes(bytes_out, w, h);
|
|
|
|
function apply(g) {
|
|
g.end();
|
|
|
|
arm.Context.layerIsMask = false;
|
|
arm.History.applyFilter();
|
|
|
|
l.deleteMask();
|
|
l.texpaint.unload();
|
|
|
|
l.texpaint = arm.Image.createRenderTarget(w, h);
|
|
let g2 = l.texpaint.get_g2();
|
|
g2.begin(false);
|
|
g2.drawImage(image, 0, 0);
|
|
g2.end();
|
|
|
|
let rts = iron.RenderPath.active.renderTargets;
|
|
rts.h["texpaint" + l.ext].image = l.texpaint;
|
|
arm.MakeMaterial.parseMeshMaterial();
|
|
arm.App.redrawUI();
|
|
arm.Context.layerPreviewDirty = true;
|
|
|
|
g.begin();
|
|
iron.App.removeRender(apply);
|
|
}
|
|
iron.App.notifyOnRender(apply);
|
|
}
|