mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-04-30 19:21:54 +00:00
fix
This commit is contained in:
tamat
@@ -3409,262 +3409,6 @@ LGraphTextureKuwaharaFilter.pixel_shader = "\n\
|
|||||||
}
|
}
|
||||||
|
|
||||||
LGraphLensFX.pixel_shader = "precision highp float;\n\
|
LGraphLensFX.pixel_shader = "precision highp float;\n\
|
||||||
=======
|
|
||||||
LiteGraph.registerNodeType(
|
|
||||||
"texture/kuwahara",
|
|
||||||
LGraphTextureKuwaharaFilter
|
|
||||||
);
|
|
||||||
|
|
||||||
// Texture Webcam *****************************************
|
|
||||||
function LGraphTextureWebcam() {
|
|
||||||
this.addOutput("Webcam", "Texture");
|
|
||||||
this.properties = { texture_name: "", facingMode: "user" };
|
|
||||||
this.boxcolor = "black";
|
|
||||||
this.version = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
LGraphTextureWebcam.title = "Webcam";
|
|
||||||
LGraphTextureWebcam.desc = "Webcam texture";
|
|
||||||
|
|
||||||
LGraphTextureWebcam.is_webcam_open = false;
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.openStream = function() {
|
|
||||||
if (!navigator.getUserMedia) {
|
|
||||||
//console.log('getUserMedia() is not supported in your browser, use chrome and enable WebRTC from about://flags');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
this._waiting_confirmation = true;
|
|
||||||
|
|
||||||
// Not showing vendor prefixes.
|
|
||||||
var constraints = {
|
|
||||||
audio: false,
|
|
||||||
video: { facingMode: this.properties.facingMode }
|
|
||||||
};
|
|
||||||
navigator.mediaDevices
|
|
||||||
.getUserMedia(constraints)
|
|
||||||
.then(this.streamReady.bind(this))
|
|
||||||
.catch(onFailSoHard);
|
|
||||||
|
|
||||||
var that = this;
|
|
||||||
function onFailSoHard(e) {
|
|
||||||
LGraphTextureWebcam.is_webcam_open = false;
|
|
||||||
console.log("Webcam rejected", e);
|
|
||||||
that._webcam_stream = false;
|
|
||||||
that.boxcolor = "red";
|
|
||||||
that.trigger("stream_error");
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.closeStream = function() {
|
|
||||||
if (this._webcam_stream) {
|
|
||||||
var tracks = this._webcam_stream.getTracks();
|
|
||||||
if (tracks.length) {
|
|
||||||
for (var i = 0; i < tracks.length; ++i) tracks[i].stop();
|
|
||||||
}
|
|
||||||
LGraphTextureWebcam.is_webcam_open = false;
|
|
||||||
this._webcam_stream = null;
|
|
||||||
this._video = null;
|
|
||||||
this.boxcolor = "black";
|
|
||||||
this.trigger("stream_closed");
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.streamReady = function(localMediaStream) {
|
|
||||||
this._webcam_stream = localMediaStream;
|
|
||||||
//this._waiting_confirmation = false;
|
|
||||||
this.boxcolor = "green";
|
|
||||||
var video = this._video;
|
|
||||||
if (!video) {
|
|
||||||
video = document.createElement("video");
|
|
||||||
video.autoplay = true;
|
|
||||||
video.srcObject = localMediaStream;
|
|
||||||
this._video = video;
|
|
||||||
//document.body.appendChild( video ); //debug
|
|
||||||
//when video info is loaded (size and so)
|
|
||||||
video.onloadedmetadata = function(e) {
|
|
||||||
// Ready to go. Do some stuff.
|
|
||||||
LGraphTextureWebcam.is_webcam_open = true;
|
|
||||||
console.log(e);
|
|
||||||
};
|
|
||||||
}
|
|
||||||
this.trigger("stream_ready", video);
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.onPropertyChanged = function(
|
|
||||||
name,
|
|
||||||
value
|
|
||||||
) {
|
|
||||||
if (name == "facingMode") {
|
|
||||||
this.properties.facingMode = value;
|
|
||||||
this.closeStream();
|
|
||||||
this.openStream();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.onRemoved = function() {
|
|
||||||
if (!this._webcam_stream) return;
|
|
||||||
|
|
||||||
var tracks = this._webcam_stream.getTracks();
|
|
||||||
if (tracks.length) {
|
|
||||||
for (var i = 0; i < tracks.length; ++i) tracks[i].stop();
|
|
||||||
}
|
|
||||||
|
|
||||||
this._webcam_stream = null;
|
|
||||||
this._video = null;
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.onDrawBackground = function(ctx) {
|
|
||||||
if (this.flags.collapsed || this.size[1] <= 20) return;
|
|
||||||
|
|
||||||
if (!this._video) return;
|
|
||||||
|
|
||||||
//render to graph canvas
|
|
||||||
ctx.save();
|
|
||||||
if (!ctx.webgl)
|
|
||||||
//reverse image
|
|
||||||
ctx.drawImage(this._video, 0, 0, this.size[0], this.size[1]);
|
|
||||||
else {
|
|
||||||
if (this._video_texture)
|
|
||||||
ctx.drawImage(
|
|
||||||
this._video_texture,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
this.size[0],
|
|
||||||
this.size[1]
|
|
||||||
);
|
|
||||||
}
|
|
||||||
ctx.restore();
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.onExecute = function() {
|
|
||||||
if (this._webcam_stream == null && !this._waiting_confirmation)
|
|
||||||
this.openStream();
|
|
||||||
|
|
||||||
if (!this._video || !this._video.videoWidth) return;
|
|
||||||
|
|
||||||
var width = this._video.videoWidth;
|
|
||||||
var height = this._video.videoHeight;
|
|
||||||
|
|
||||||
var temp = this._video_texture;
|
|
||||||
if (!temp || temp.width != width || temp.height != height)
|
|
||||||
this._video_texture = new GL.Texture(width, height, {
|
|
||||||
format: gl.RGB,
|
|
||||||
filter: gl.LINEAR
|
|
||||||
});
|
|
||||||
|
|
||||||
this._video_texture.uploadImage(this._video);
|
|
||||||
this._video_texture.version = ++this.version;
|
|
||||||
|
|
||||||
if (this.properties.texture_name) {
|
|
||||||
var container = LGraphTexture.getTexturesContainer();
|
|
||||||
container[this.properties.texture_name] = this._video_texture;
|
|
||||||
}
|
|
||||||
|
|
||||||
this.setOutputData(0, this._video_texture);
|
|
||||||
for (var i = 1; i < this.outputs.length; ++i) {
|
|
||||||
if (!this.outputs[i]) continue;
|
|
||||||
switch (this.outputs[i].name) {
|
|
||||||
case "width":
|
|
||||||
this.setOutputData(i, this._video.videoWidth);
|
|
||||||
break;
|
|
||||||
case "height":
|
|
||||||
this.setOutputData(i, this._video.videoHeight);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphTextureWebcam.prototype.onGetOutputs = function() {
|
|
||||||
return [
|
|
||||||
["width", "number"],
|
|
||||||
["height", "number"],
|
|
||||||
["stream_ready", LiteGraph.EVENT],
|
|
||||||
["stream_closed", LiteGraph.EVENT],
|
|
||||||
["stream_error", LiteGraph.EVENT]
|
|
||||||
];
|
|
||||||
};
|
|
||||||
|
|
||||||
LiteGraph.registerNodeType("texture/webcam", LGraphTextureWebcam);
|
|
||||||
|
|
||||||
//from https://github.com/spite/Wagner
|
|
||||||
function LGraphLensFX() {
|
|
||||||
this.addInput("in", "Texture");
|
|
||||||
this.addInput("f", "number");
|
|
||||||
this.addOutput("out", "Texture");
|
|
||||||
this.properties = {
|
|
||||||
enabled: true,
|
|
||||||
factor: 1,
|
|
||||||
precision: LGraphTexture.LOW
|
|
||||||
};
|
|
||||||
|
|
||||||
this._uniforms = { u_texture: 0, u_factor: 1 };
|
|
||||||
}
|
|
||||||
|
|
||||||
LGraphLensFX.title = "Lens FX";
|
|
||||||
LGraphLensFX.desc = "distortion and chromatic aberration";
|
|
||||||
|
|
||||||
LGraphLensFX.widgets_info = {
|
|
||||||
precision: { widget: "combo", values: LGraphTexture.MODE_VALUES }
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphLensFX.prototype.onGetInputs = function() {
|
|
||||||
return [["enabled", "boolean"]];
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphLensFX.prototype.onExecute = function() {
|
|
||||||
var tex = this.getInputData(0);
|
|
||||||
if (!tex) return;
|
|
||||||
|
|
||||||
if (!this.isOutputConnected(0)) return; //saves work
|
|
||||||
|
|
||||||
if (
|
|
||||||
this.properties.precision === LGraphTexture.PASS_THROUGH ||
|
|
||||||
this.getInputOrProperty("enabled") === false
|
|
||||||
) {
|
|
||||||
this.setOutputData(0, tex);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
var temp = this._temp_texture;
|
|
||||||
if (
|
|
||||||
!temp ||
|
|
||||||
temp.width != tex.width ||
|
|
||||||
temp.height != tex.height ||
|
|
||||||
temp.type != tex.type
|
|
||||||
)
|
|
||||||
temp = this._temp_texture = new GL.Texture(
|
|
||||||
tex.width,
|
|
||||||
tex.height,
|
|
||||||
{ type: tex.type, format: gl.RGBA, filter: gl.LINEAR }
|
|
||||||
);
|
|
||||||
|
|
||||||
var shader = LGraphLensFX._shader;
|
|
||||||
if (!shader)
|
|
||||||
shader = LGraphLensFX._shader = new GL.Shader(
|
|
||||||
GL.Shader.SCREEN_VERTEX_SHADER,
|
|
||||||
LGraphLensFX.pixel_shader
|
|
||||||
);
|
|
||||||
|
|
||||||
var factor = this.getInputData(1);
|
|
||||||
if (factor == null) factor = this.properties.factor;
|
|
||||||
|
|
||||||
var uniforms = this._uniforms;
|
|
||||||
uniforms.u_factor = factor;
|
|
||||||
|
|
||||||
//apply shader
|
|
||||||
gl.disable(gl.DEPTH_TEST);
|
|
||||||
temp.drawTo(function() {
|
|
||||||
tex.bind(0);
|
|
||||||
shader.uniforms(uniforms).draw(GL.Mesh.getScreenQuad());
|
|
||||||
});
|
|
||||||
|
|
||||||
this.setOutputData(0, temp);
|
|
||||||
};
|
|
||||||
|
|
||||||
LGraphLensFX.pixel_shader =
|
|
||||||
"precision highp float;\n\
|
|
||||||
>>>>>>> c6b16f04e0c2859d0b9dfe07104a594713305912
|
|
||||||
varying vec2 v_coord;\n\
|
varying vec2 v_coord;\n\
|
||||||
uniform sampler2D u_texture;\n\
|
uniform sampler2D u_texture;\n\
|
||||||
uniform float u_factor;\n\
|
uniform float u_factor;\n\
|
||||||
|
|||||||
Reference in New Issue
Block a user