nodes/bruefire-ComfyUI-SeqImageLoader
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added logic to display a loading frame number.

Browse Source
This commit is contained in:
bruefire
2023-12-24 12:08:43 +09:00
parent b29a08a8f1
commit 18f08eb766
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
js/main.js
+8 -3
View File
@@ -263,7 +263,8 @@ app.registerExtension({
async function getFramesFromVideo(file) {
const imageType = 'image/png';
let frameCount = 0;
let loadFrameCount = 0;
let blobFrameCount = 0;
let tWidth, tHeight;
let videoUrl = URL.createObjectURL(file);
let fCanvases = [];
@@ -278,6 +279,8 @@ app.registerExtension({
fCanvas.getContext('2d').drawImage(vframe, 0, 0, fCanvas.width, fCanvas.height);
vframe.close();
fCanvases.push(fCanvas);
idWidget.value = (++loadFrameCount) + " frames loaded..";
},
onConfig(config) {
tWidth = config.codedWidth;
@@ -287,12 +290,14 @@ app.registerExtension({
},
});
let blobFrameCount2 = 0;
let frames = await Promise.all(
fCanvases.map(fCanvas =>
new Promise((resolve, reject) => {
const fid = frameCount++;
const fid = blobFrameCount++;
fCanvas.toBlob(blob => {
if (blob) {
idWidget.value = (++blobFrameCount2) + "/" + (loadFrameCount) + " frames prepared..";
blob.name = ("00000000" + fid).slice(-8) + ".png";
resolve(blob);
} else {
@@ -302,7 +307,7 @@ app.registerExtension({
})));
URL.revokeObjectURL(videoUrl); // revoke URL to prevent memory leak
console.log("Frame extraction completed with " + frameCount + " frames");
console.log("Frame extraction completed with " + blobFrameCount + " frames");
return frames;
}