js-encoding-samples/ves.js

// --------------------------------------------------------------------------------
// Development options (for user options, see config.json)
global.debug = false;


// --------------------------------------------------------------------------------
// Actual Code

/*
Comparing VMAF instantly:
- Saves disk space - no need to keep copies around!
fn: .\ffmpeg.exe -i "..\..\cache\arma_3-002-1280x720x60.00.mkv" -i "..\..\videos\arma_3-002.mkv" -filter_complex_threads 8 -filter_complex [0:v:0]scale=flags=bicubic+full_chroma_inp+full_chroma_int:w=1920:h=1080,colorspace=all=bt709:range=pc,format=pix_fmts=yuv444p[main];[1:v:0]colorspace=all=bt709:range=pc,format=pix_fmts=yuv444p[ref];[main][ref]libvmaf=model_path=../vmaf/vmaf_4k_rb_v0.6.2.pkl:log_fmt=json:log_path=here2.json:enable_conf_interval=1:shortest=1[out] -map [out] -f null -

*/

// Import modules
const ffmpeg = require('./ffmpeg.js');
const poolqueue = require('./poolqueue.js');

const fs = require('fs');
const path = require('path');
const process = require('process');
const util = require('util');
const { config } = require('process');

// Define some Helpers
function float_eq(a, b, edge) { return (Math.abs(a - b) <= edge); }
function float_lt(a, b, edge) { return ((a + edge) < b); }
function float_le(a, b, edge) { return float_lt(a, b, edge) || float_eq(a, b, edge); }
function float_gt(a, b, edge) { return ((a - edge) > b); }
function float_ge(a, b, edge) { return float_gt(a, b, edge) || float_eq(a, b, edge); }
Object.size = function (obj) { var size = 0, key; for (key in obj) { if (obj.hasOwnProperty(key)) { size++; } } return size; };

// Actual Code
async function load_config() { // Load Configuration
	console.group("Loading configuration...");
	console.time("Total");

	let config = require('./config.json');
	// Resolve and create directories.
	for (let name in config.paths) {
		config.paths[name] = path.resolve(config.paths[name]);
		if (!fs.existsSync(config.paths[name])) {
			fs.mkdirSync(config.paths[name]);
		}
	}

	console.timeEnd("Total");
	console.groupEnd();
	return config;
}

async function load_encoders(config, ff) { // Load Encoders
	console.group("Loading encoders...");
	console.time("Total");

	let encoders = new Map();
	for (let name in config.encoders) {
		let encoder = config.encoders[name];

		// Is the encoder disabled, or invalid?
		if (!encoder || !encoder.enabled) {
			if (global.debug) console.debug(`${name} is disabled or invalid.`);
			continue;
		}

		// Does the encoder exist?
		if (!fs.existsSync(path.join(".", `encoder_${name}.js`))) {
			console.error(`${name} does not exist.`);
			continue;
		}

		// Attempt to load the encoder, if possible.
		console.group(name);
		console.time("Subtotal");
		try {
			let instance = new (require(`./encoder_${name}.js`))(ff, config, encoder);
			instance.load();
			encoders.set(name, instance);
		} catch (ex) {
			console.log(ex);
		}
		console.timeEnd("Subtotal");
		console.groupEnd();
	}

	console.timeEnd("Total");
	console.groupEnd();
	return encoders;
}

async function load_videos(config, ff) { // Load Videos
	console.group("Loading videos...");
	console.time("Total");

	let videos = new Map();
	let promises = []; // Asynchronous loading.
	for (let name in config.videos) {
		let video = config.videos[name];

		// Is the video disabled, or invalid?
		if (!video || !video.enabled) {
			if (global.debug) console.debug(`${name} is disabled or invalid.`);
			continue;
		}

		// Does the video exist?
		if (!fs.existsSync(path.join(config.paths.videos, `${name}.mkv`))) {
			console.error(`${name} does not exist.`);
			continue;
		}

		// Load video information asynchronously.
		promises.push(new Promise((resolve, reject) => {
			console.time(name);

			let file_name = path.join(config.paths.videos, `${name}.mkv`);
			let probeprom = ff.probe(file_name);
			probeprom.then((json) => {
				let data = new Object();

				data.name = name;
				data.info = json;
				data.caches = new Map();

				// File Information
				data.file_name = `${name}.mkv`;
				data.file_ext = path.extname(data.file_name);
				data.file_base = path.basename(data.file_name, data.file_ext);
				data.file = path.join(config.paths.videos, data.file_name);

				// Video Information
				data.resolution = { width: data.info.streams[0].width, height: data.info.streams[0].height };
				data.framerate = eval(data.info.streams[0].r_frame_rate);
				data.duration = data.info.streams[0].duration;

				// Color Information
				data.color = {};
				data.color.range = data.info.streams[0].color_range ? data.info.streams[0].color_range : 'tv';
				data.color.trc = data.info.streams[0].color_transfer ? data.info.streams[0].color_transfer : 'bt709';
				data.color.primaries = data.info.streams[0].color_primaries ? data.info.streams[0].color_primaries : 'bt709';
				data.color.matrix = data.info.streams[0].color_space ? data.info.streams[0].color_space : 'bt709';

				videos.set(data.name, data);
				console.timeEnd(name);
				resolve(data);
			});
		}));
	}

	await Promise.allSettled(promises);

	// Sort the videos list again. (We don't care about what order the user wants!)
	videos = new Map([...videos].sort((a, b) => a[0] > b[0] ? 1 : -1));

	console.timeEnd("Total");
	console.groupEnd();
	return videos;
}

async function create_caches(config, ff, videos, encoders) { // Create Caches
	console.group("Creating caches...");
	console.time("Total");

	for (let name of videos.keys()) {
		let video = videos.get(name);

		console.group(name);
		console.time("Subtotal");

		video.caches = new Map();
		for (let resolution of config.options.resolutions) {
			let aspect_ratio = (video.resolution.height / video.resolution.width);
			for (let framerate_scaling of config.options.framerate_scalings) {
				let height = (Math.round(resolution[0] * aspect_ratio * 0.5) * 2);
				// Skip resolutions that are too large horizontally.
				if (resolution[0] > video.resolution.width) {
					if (global.debug) console.debug(`${resolution[0]}x${height}x${(video.framerate * framerate_scaling)} skipped.`);
					continue;
				}

				// Create Cache Information
				let cache = {
					framerate: (video.framerate * framerate_scaling),
					width: resolution[0],
					height: height
				};
				cache.framerate_s = cache.framerate.toFixed(2);
				cache.duration = Math.floor(video.duration * video.framerate * framerate_scaling) / cache.framerate;
				let key = `${cache.width}x${cache.height}x${cache.framerate_s}`;
				cache.file = path.join(config.paths.cache, `${video.file_base}-${key}.mkv`);
				cache.queues = new Map();

				video.caches.set(`${key}`, cache);
			}
		}

		for (let key of video.caches.keys()) {
			let cache = video.caches.get(key);

			// Check if cache file already exists and is correct.
			if (fs.existsSync(cache.file)) {
				let info = ff.probeSync(cache.file);
				if ((info.streams)
					&& (info.streams.length > 0)
					&& (info.streams[0].width == cache.width)
					&& (info.streams[0].height == cache.height)
					&& float_eq(eval(info.streams[0].r_frame_rate), cache.framerate, 0.01)
					&& float_eq(info.streams[0].duration, video.duration, 0.1)) {
					if (global.debug) console.debug(`${key} already exists.`);
					continue;
				}
			}

			console.time(`${key} created`);
			let command = [
				"-y",
				"-hide_banner",
				"-v", "error",
				"-i", video.file,
				"-filter_complex", `fps=fps=${cache.framerate_s},scale=flags=bicubic+full_chroma_inp+full_chroma_int:w=${cache.width}:h=${cache.height},colorspace=all=bt709:range=tv:format=yuv420p`,
				"-an",
			];
			if (encoders.has("h264_nvenc")) {
				command.push(
					"-c:v", "h264_nvenc",
					"-profile:v", "high",
					"-preset", "p1",
					"-tune", "lossless",
					"-rc", "constqp",
					"-rc-lookahead", "0",
					"-multipass", "0",
					"-b:v", "0",
					"-minrate", "0",
					"-maxrate", "0",
					"-bufsize", "0",
					"-qp", "0",
					"-init_qpI", "0",
					"-init_qpP", "0",
					"-init_qpB", "0",
					"-bf", "0",
					"-g", `15`,
				);
			} else {
				command.push(
					"-c:v", "libx264",
					"-preset", "veryfast",
					"-crf", "0",
					"-b:v", "0",
					"-minrate", "0",
					"-maxrate", "0",
					"-bufsize", "0",
					"-g", `15`,
				);
			}
			command.push(cache.file);
			let res = ff.ffmpegSync(command);
			if (res.status != 0) {
				console.log(res.stderr.toString());
			}
			if (global.debug) console.log(res.stdout.toString(), res.stderr.toString());
			console.timeEnd(`${key} created`);
		}

		console.timeEnd("Subtotal");
		console.groupEnd(name);
	}

	console.timeEnd("Total");
	console.groupEnd();
	return videos;
}

async function transcode(config, ff, videos, encoders) {
	console.group("Transcoding...");
	console.time("Total");

	/* Needs a massive overhaul:
	- Check if the .json file for a configuration already exists, if not continue.
	- Do not store transcodes for longer than needed, we only need to compare against source.
	- Use poolqueue to still allow for parallelization. 
	- '-g' and bitrate options are controlled _from here_.
	*/

	// Build queues per cache.
	console.group("Queueing...")
	console.time("Subtotal");
	for (let video_key of videos.keys()) {
		console.group(video_key);
		console.time(video_key);
		let video = videos.get(video_key);
		for (let cache_key of video.caches.keys()) {
			let cache = video.caches.get(cache_key);
			console.time(cache_key);

			for (let encoder_key of encoders.keys()) {
				let encoder = encoders.get(encoder_key);

				let queue_promises = [];
				let queue_commands = new poolqueue();
				let queue_files = new poolqueue();
				for (let idx = 0; idx < encoder.count(); idx++) {
					let command_promises = [];
					let command = encoder.get(idx, cache.width, cache.height, cache.framerate);
					for (let bitrate of config.options.bitrates) {
						for (let kfinterval of config.options.keyframeinterval) {
							command_promises.push(new Promise((resolve, reject) => {
								let file = path.join(
									config.paths.output,
									video.name,
									cache_key,
									encoder_key,
									bitrate.toFixed(0),
									(cache.framerate * kfinterval).toFixed(0),
									`${command.hash}.mkv`
								);
								let file_json = path.join(
									config.paths.output,
									video.name,
									cache_key,
									encoder_key,
									bitrate.toFixed(0),
									(cache.framerate * kfinterval).toFixed(0),
									`${command.hash}.json`
								);

								// Check if this file already exists at the target location.
								if (fs.existsSync(file_json)) {
									// Don't need to check for a video here, since we only care about results.
									if (global.debug) console.debug(`${file} already completed.`);
									resolve();
									return;
								}

								let line = [
									"-map", "0:v:0",
									"-an",
									"-c:v", encoder_key,
									"-g", (cache.framerate * kfinterval).toFixed(0),
									"-b:v", `${bitrate}k`,
									"-minrate", "0",
									"-maxrate", "0",
									"-bufsize", `${2 * bitrate}k`,
								].concat(command.options).concat(encoder.extra()).concat([file]);

								queue_commands.push(
									encoder.pool(),
									line,
									command.cost,
								);
								queue_files.push(
									encoder.pool(),
									[file, file_json],
									command.cost,
								);

								resolve();
							}));
						}
					}
					queue_promises.push(async function () {
						await Promise.allSettled(command_promises);
					});
				}
				await Promise.allSettled(queue_promises);

				cache.queues.set(encoder_key, {
					commands: queue_commands.finalize(),
					files: queue_files.finalize(),
				});
			}

			console.timeEnd(cache_key);
		}
		console.timeEnd(video_key);
		console.groupEnd();
	}
	console.timeEnd("Subtotal");
	console.groupEnd();

	// Process from here on out.
	// LOOP
	// 1. Pull out front of the command and file queue.
	// 2. Encode using the given command(s).
	// 3. Compare resulting files with real input (libvmaf).
	// 4. Delete encoded files.
	// 5. Repeat until queues empty, no more caches for video, and no more videos.

/*
	console.group("Queueing...")
	console.time("Subtotal");
	for (let video_key of videos.keys()) {
		console.group(video_key);
		console.time(video_key);
		let video = videos.get(video_key);
		for (let cache_key of video.caches.keys()) {
			let cache = video.caches.get(cache_key);
			console.time(cache_key);

			console.timeEnd(cache_key);
		}
		console.timeEnd(video_key);
		console.groupEnd();
	}
	console.timeEnd("Subtotal");
	console.groupEnd();
*/
	/*
	for (let video_name in videos) {
		console.time(video_name);
		console.group(video_name);
		let video = videos[video_name];
		for (let cache of video.caches) {
			let key_cache = `${cache.x}x${cache.y}x${cache.fps.toFixed(2)}`;
			console.time(key_cache);
			console.group(key_cache);
			for (let cmd of cache.commands) {
				let opts = [
					"-y",
					"-hide_banner",
					"-v", "error",
					"-hwaccel", "auto",
					"-i", cache.file
				].concat(cmd);
				let res = ff.ffmpegSync(opts);
				console.log(res.stdout.toString(), res.stderr.toString());
			}
			console.groupEnd();
			console.timeEnd(key_cache);
		}
		console.groupEnd();
		console.timeEnd(video_name);
	}
	*/

	console.timeEnd("Total");
	console.groupEnd();
}

async function main() {
	let config;
	let ff;
	let encoders;
	let videos;

	await load_config().then((p) => { config = p; });
	ff = new ffmpeg(config.paths.ffmpeg);
	await load_encoders(config, ff).then((p) => { encoders = p; });
	await load_videos(config, ff).then((p) => { videos = p; });
	await create_caches(config, ff, videos, encoders);
	await transcode(config, ff, videos, encoders);


	// Queue
	//await queue_transcodes(config, ff, videos, encoders);

	// Transcode
}

main();