mirror of https://github.com/mifi/lossless-cut
Mikael Finstad
4 years ago
parent
572a0caf1a
commit
75f768dad7
@ -1,93 +0,0 @@
|
||||
import dataUriToBuffer from 'data-uri-to-buffer';
|
||||
import pMap from 'p-map';
|
||||
|
||||
import { getSuffixedOutPath, getOutDir, transferTimestamps, getSuffixedFileName, getOutPath, escapeRegExp } from './util';
|
||||
import { formatDuration } from './util/duration';
|
||||
|
||||
import { captureFrame as ffmpegCaptureFrame, captureFrames as ffmpegCaptureFrames } from './ffmpeg';
|
||||
|
||||
const fs = window.require('fs-extra');
|
||||
const mime = window.require('mime-types');
|
||||
const { rename, readdir } = window.require('fs/promises');
|
||||
|
||||
function getFrameFromVideo(video, format, quality) {
|
||||
const canvas = document.createElement('canvas');
|
||||
canvas.width = video.videoWidth;
|
||||
canvas.height = video.videoHeight;
|
||||
|
||||
canvas.getContext('2d').drawImage(video, 0, 0);
|
||||
|
||||
const dataUri = canvas.toDataURL(`image/${format}`, quality);
|
||||
|
||||
return dataUriToBuffer(dataUri);
|
||||
}
|
||||
|
||||
export async function captureFramesRange({ customOutDir, filePath, fps, fromTime, toTime, estimatedMaxNumFiles, captureFormat, quality, filter, onProgress, outputTimestamps }) {
|
||||
const getSuffix = (prefix) => `${prefix}.${captureFormat}`;
|
||||
|
||||
if (!outputTimestamps) {
|
||||
const numDigits = Math.floor(Math.log10(estimatedMaxNumFiles)) + 1;
|
||||
const nameTemplateSuffix = getSuffix(`%0${numDigits}d`);
|
||||
const nameSuffix = getSuffix(`${'1'.padStart(numDigits, '0')}`); // mimic ffmpeg output
|
||||
const outPathTemplate = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: nameTemplateSuffix });
|
||||
const firstFileOutPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix });
|
||||
|
||||
await ffmpegCaptureFrames({ from: fromTime, to: toTime, videoPath: filePath, outPathTemplate, quality, filter, onProgress });
|
||||
|
||||
return firstFileOutPath;
|
||||
}
|
||||
|
||||
// see https://github.com/mifi/lossless-cut/issues/1139
|
||||
const tmpSuffix = 'llc-tmp-frame-capture-';
|
||||
const outPathTemplate = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: getSuffix(`${tmpSuffix}%d`) });
|
||||
await ffmpegCaptureFrames({ from: fromTime, to: toTime, videoPath: filePath, outPathTemplate, quality, filter, framePts: true, onProgress });
|
||||
|
||||
const outDir = getOutDir(customOutDir, filePath);
|
||||
const files = await readdir(outDir);
|
||||
|
||||
// https://github.com/mifi/lossless-cut/issues/1139
|
||||
const matches = files.map((fileName) => {
|
||||
const escapedRegexp = escapeRegExp(getSuffixedFileName(filePath, tmpSuffix));
|
||||
const regexp = `^${escapedRegexp}(\\d+)`;
|
||||
console.log(regexp);
|
||||
const match = fileName.match(new RegExp(regexp));
|
||||
if (!match) return undefined;
|
||||
const frameNum = parseInt(match[1], 10);
|
||||
if (Number.isNaN(frameNum) || frameNum < 0) return undefined;
|
||||
return { fileName, frameNum };
|
||||
}).filter((it) => it != null);
|
||||
|
||||
console.log('Renaming temp files...');
|
||||
const outPaths = await pMap(matches, async ({ fileName, frameNum }) => {
|
||||
const duration = formatDuration({ seconds: fromTime + (frameNum / fps), fileNameFriendly: true });
|
||||
const renameFromPath = getOutPath({ customOutDir, filePath, fileName });
|
||||
const renameToPath = getOutPath({ customOutDir, filePath, fileName: getSuffixedFileName(filePath, getSuffix(duration, captureFormat)) });
|
||||
await rename(renameFromPath, renameToPath);
|
||||
return renameToPath;
|
||||
}, { concurrency: 1 });
|
||||
|
||||
return outPaths[0];
|
||||
}
|
||||
|
||||
export async function captureFrameFromFfmpeg({ customOutDir, filePath, fromTime, captureFormat, enableTransferTimestamps, quality }) {
|
||||
const time = formatDuration({ seconds: fromTime, fileNameFriendly: true });
|
||||
const nameSuffix = `${time}.${captureFormat}`;
|
||||
const outPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix });
|
||||
await ffmpegCaptureFrame({ timestamp: fromTime, videoPath: filePath, outPath, quality });
|
||||
|
||||
if (enableTransferTimestamps) await transferTimestamps(filePath, outPath, fromTime);
|
||||
return outPath;
|
||||
}
|
||||
|
||||
export async function captureFrameFromTag({ customOutDir, filePath, currentTime, captureFormat, video, enableTransferTimestamps, quality }) {
|
||||
const buf = getFrameFromVideo(video, captureFormat, quality);
|
||||
|
||||
const ext = mime.extension(buf.type);
|
||||
const time = formatDuration({ seconds: currentTime, fileNameFriendly: true });
|
||||
|
||||
const outPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: `${time}.${ext}` });
|
||||
await fs.writeFile(outPath, buf);
|
||||
|
||||
if (enableTransferTimestamps) await transferTimestamps(filePath, outPath, currentTime);
|
||||
return outPath;
|
||||
}
|
||||
@ -0,0 +1,101 @@
|
||||
import dataUriToBuffer from 'data-uri-to-buffer';
|
||||
import pMap from 'p-map';
|
||||
|
||||
import { getSuffixedOutPath, getOutDir, transferTimestamps, getSuffixedFileName, getOutPath, escapeRegExp } from '../util';
|
||||
|
||||
import { captureFrame as ffmpegCaptureFrame, captureFrames as ffmpegCaptureFrames } from '../ffmpeg';
|
||||
|
||||
const fs = window.require('fs-extra');
|
||||
const mime = window.require('mime-types');
|
||||
const { rename, readdir } = window.require('fs/promises');
|
||||
|
||||
|
||||
function getFrameFromVideo(video, format, quality) {
|
||||
const canvas = document.createElement('canvas');
|
||||
canvas.width = video.videoWidth;
|
||||
canvas.height = video.videoHeight;
|
||||
|
||||
canvas.getContext('2d').drawImage(video, 0, 0);
|
||||
|
||||
const dataUri = canvas.toDataURL(`image/${format}`, quality);
|
||||
|
||||
return dataUriToBuffer(dataUri);
|
||||
}
|
||||
|
||||
export default ({ formatTimecode }) => {
|
||||
async function captureFramesRange({ customOutDir, filePath, fps, fromTime, toTime, estimatedMaxNumFiles, captureFormat, quality, filter, onProgress, outputTimestamps }) {
|
||||
const getSuffix = (prefix) => `${prefix}.${captureFormat}`;
|
||||
|
||||
if (!outputTimestamps) {
|
||||
const numDigits = Math.floor(Math.log10(estimatedMaxNumFiles)) + 1;
|
||||
const nameTemplateSuffix = getSuffix(`%0${numDigits}d`);
|
||||
const nameSuffix = getSuffix(`${'1'.padStart(numDigits, '0')}`); // mimic ffmpeg output
|
||||
const outPathTemplate = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: nameTemplateSuffix });
|
||||
const firstFileOutPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix });
|
||||
|
||||
await ffmpegCaptureFrames({ from: fromTime, to: toTime, videoPath: filePath, outPathTemplate, quality, filter, onProgress });
|
||||
|
||||
return firstFileOutPath;
|
||||
}
|
||||
|
||||
// see https://github.com/mifi/lossless-cut/issues/1139
|
||||
const tmpSuffix = 'llc-tmp-frame-capture-';
|
||||
const outPathTemplate = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: getSuffix(`${tmpSuffix}%d`) });
|
||||
await ffmpegCaptureFrames({ from: fromTime, to: toTime, videoPath: filePath, outPathTemplate, quality, filter, framePts: true, onProgress });
|
||||
|
||||
const outDir = getOutDir(customOutDir, filePath);
|
||||
const files = await readdir(outDir);
|
||||
|
||||
// https://github.com/mifi/lossless-cut/issues/1139
|
||||
const matches = files.map((fileName) => {
|
||||
const escapedRegexp = escapeRegExp(getSuffixedFileName(filePath, tmpSuffix));
|
||||
const regexp = `^${escapedRegexp}(\\d+)`;
|
||||
console.log(regexp);
|
||||
const match = fileName.match(new RegExp(regexp));
|
||||
if (!match) return undefined;
|
||||
const frameNum = parseInt(match[1], 10);
|
||||
if (Number.isNaN(frameNum) || frameNum < 0) return undefined;
|
||||
return { fileName, frameNum };
|
||||
}).filter((it) => it != null);
|
||||
|
||||
console.log('Renaming temp files...');
|
||||
const outPaths = await pMap(matches, async ({ fileName, frameNum }) => {
|
||||
const duration = formatTimecode({ seconds: fromTime + (frameNum / fps), fileNameFriendly: true });
|
||||
const renameFromPath = getOutPath({ customOutDir, filePath, fileName });
|
||||
const renameToPath = getOutPath({ customOutDir, filePath, fileName: getSuffixedFileName(filePath, getSuffix(duration, captureFormat)) });
|
||||
await rename(renameFromPath, renameToPath);
|
||||
return renameToPath;
|
||||
}, { concurrency: 1 });
|
||||
|
||||
return outPaths[0];
|
||||
}
|
||||
|
||||
async function captureFrameFromFfmpeg({ customOutDir, filePath, fromTime, captureFormat, enableTransferTimestamps, quality }) {
|
||||
const time = formatTimecode({ seconds: fromTime, fileNameFriendly: true });
|
||||
const nameSuffix = `${time}.${captureFormat}`;
|
||||
const outPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix });
|
||||
await ffmpegCaptureFrame({ timestamp: fromTime, videoPath: filePath, outPath, quality });
|
||||
|
||||
if (enableTransferTimestamps) await transferTimestamps(filePath, outPath, fromTime);
|
||||
return outPath;
|
||||
}
|
||||
|
||||
async function captureFrameFromTag({ customOutDir, filePath, currentTime, captureFormat, video, enableTransferTimestamps, quality }) {
|
||||
const buf = getFrameFromVideo(video, captureFormat, quality);
|
||||
|
||||
const ext = mime.extension(buf.type);
|
||||
const time = formatTimecode({ seconds: currentTime, fileNameFriendly: true });
|
||||
|
||||
const outPath = getSuffixedOutPath({ customOutDir, filePath, nameSuffix: `${time}.${ext}` });
|
||||
await fs.writeFile(outPath, buf);
|
||||
|
||||
if (enableTransferTimestamps) await transferTimestamps(filePath, outPath, currentTime);
|
||||
return outPath;
|
||||
}
|
||||
|
||||
return {
|
||||
captureFramesRange,
|
||||
captureFrameFromFfmpeg,
|
||||
captureFrameFromTag,
|
||||
};
|
||||
};
|
||||
Loading…
Reference in New Issue