matty/peertube-plugin-nctv-nvenc-transcode
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make target bitrate whatever the input bitrate is

Browse Source
This commit is contained in:
matty 2024-09-10 23:26:38 -04:00
parent 016b5600e4
commit 98c7fc5eac
1 changed files with 323 additions and 299 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

284
dist/main.js vendored
View File

@ -18,7 +18,7 @@ const DEFAULT_BITRATES = new Map([
[720, 2800 * 1000], [720, 2800 * 1000],
[1080, 5200 * 1000], [1080, 5200 * 1000],
[1440, 10000 * 1000], [1440, 10000 * 1000],
[2160, 22000 * 1000] [2160, 22000 * 1000],
]); ]);
let pluginSettings = { let pluginSettings = {
hardwareDecode: DEFAULT_HARDWARE_DECODE, hardwareDecode: DEFAULT_HARDWARE_DECODE,
@ -28,114 +28,124 @@ let pluginSettings = {
cqH264: DEFAULT_CQ_H264, cqH264: DEFAULT_CQ_H264,
cqHEVC: DEFAULT_CQ_HEVC, cqHEVC: DEFAULT_CQ_HEVC,
h264Profile: DEFAULT_H264_PROFILE, h264Profile: DEFAULT_H264_PROFILE,
baseBitrate: new Map(DEFAULT_BITRATES) baseBitrate: new Map(DEFAULT_BITRATES),
}; };
let latestStreamNum = 9999; let latestStreamNum = 9999;
async function register({ settingsManager, peertubeHelpers, transcodingManager: transcode, registerSetting }) { async function register({
settingsManager,
peertubeHelpers,
transcodingManager: transcode,
registerSetting,
}) {
logger = peertubeHelpers.logger; logger = peertubeHelpers.logger;
transcodingManager = transcode; transcodingManager = transcode;
logger.info("Registering peertube-plugin-nctv-hardware-encode"); logger.info("Registering peertube-plugin-nctv-hardware-encode");
const encoder = 'h264_nvenc'; const encoder = "h264_nvenc";
const hevc = 'hevc_nvenc'; const hevc = "hevc_nvenc";
const profileName = 'nctv-nvenc'; const profileName = "nctv-nvenc";
const hevcProfile = 'nctv-hevc'; const hevcProfile = "nctv-hevc";
transcodingManager.addVODProfile(encoder, profileName, vodBuilder); transcodingManager.addVODProfile(encoder, profileName, vodBuilder);
transcodingManager.addVODEncoderPriority('video', encoder, 1000); transcodingManager.addVODEncoderPriority("video", encoder, 1000);
transcodingManager.addLiveProfile(encoder, profileName, liveBuilder); transcodingManager.addLiveProfile(encoder, profileName, liveBuilder);
transcodingManager.addLiveEncoderPriority('video', encoder, 1000); transcodingManager.addLiveEncoderPriority("video", encoder, 1000);
transcodingManager.addVODProfile(hevc, hevcProfile, hevcVODBuilder); transcodingManager.addVODProfile(hevc, hevcProfile, hevcVODBuilder);
transcodingManager.addVODEncoderPriority('video', hevc, 900); transcodingManager.addVODEncoderPriority("video", hevc, 900);
transcodingManager.addLiveProfile(hevc, hevcProfile, hevcLiveBuilder); transcodingManager.addLiveProfile(hevc, hevcProfile, hevcLiveBuilder);
transcodingManager.addLiveEncoderPriority('video', hevc, 900); transcodingManager.addLiveEncoderPriority("video", hevc, 900);
await loadSettings(settingsManager); await loadSettings(settingsManager);
registerSetting({ registerSetting({
name: 'hardware-decode', name: "hardware-decode",
label: 'Hardware decode', label: "Hardware decode",
type: 'input-checkbox', type: "input-checkbox",
descriptionHTML: 'Use hardware video decoder instead of software decoder. This will slightly improve performance but may cause some issues with some videos. If you encounter issues, disable this option and restart failed jobs.', descriptionHTML:
"Use hardware video decoder instead of software decoder. This will slightly improve performance but may cause some issues with some videos. If you encounter issues, disable this option and restart failed jobs.",
default: DEFAULT_HARDWARE_DECODE, default: DEFAULT_HARDWARE_DECODE,
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'cq-h264', name: "cq-h264",
label: 'CQ Value for H264_nvenc', label: "CQ Value for H264_nvenc",
type: 'input', type: "input",
descriptionHTML: 'Sets the -cq value for h264_nvenc encoder. Valid values are between 0 and 51 (lossess and AIDS, respectively)', descriptionHTML:
"Sets the -cq value for h264_nvenc encoder. Valid values are between 0 and 51 (lossess and AIDS, respectively)",
default: DEFAULT_CQ_H264, default: DEFAULT_CQ_H264,
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'cq-hevc', name: "cq-hevc",
label: 'CQ Value for hevc_nvenc', label: "CQ Value for hevc_nvenc",
type: 'input', type: "input",
descriptionHTML: 'Sets the -cq value for hevc_nvenc encoder. Valid values are between 0 and 51 (lossess and AIDS, respectively)', descriptionHTML:
"Sets the -cq value for hevc_nvenc encoder. Valid values are between 0 and 51 (lossess and AIDS, respectively)",
default: DEFAULT_CQ_HEVC, default: DEFAULT_CQ_HEVC,
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'vod-quality', name: "vod-quality",
label: 'VOD Quality', label: "VOD Quality",
type: 'select', type: "select",
options: [ options: [
{ label: 'fastest', value: 'p1' }, { label: "fastest", value: "p1" },
{ label: 'faster', value: 'p2' }, { label: "faster", value: "p2" },
{ label: 'fast', value: 'p3' }, { label: "fast", value: "p3" },
{ label: 'medium (default)', value: 'p4' }, { label: "medium (default)", value: "p4" },
{ label: 'slow', value: 'p5' }, { label: "slow", value: "p5" },
{ label: 'slower', value: 'p6' }, { label: "slower", value: "p6" },
{ label: 'slowest', value: 'p7' } { label: "slowest", value: "p7" },
], ],
descriptionHTML: 'This parameter controls the speed / quality tradeoff. Slower speed mean better quality. Faster speed mean lower quality. This setting is hardware dependent, you may need to experiment to find the best value for your hardware.', descriptionHTML:
"This parameter controls the speed / quality tradeoff. Slower speed mean better quality. Faster speed mean lower quality. This setting is hardware dependent, you may need to experiment to find the best value for your hardware.",
default: DEFAULT_VOD_QUALITY.toString(), default: DEFAULT_VOD_QUALITY.toString(),
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'hevc-profile', name: "hevc-profile",
label: 'HEVC Profile', label: "HEVC Profile",
type: 'select', type: "select",
options: [ options: [
{ label: 'main (default)', value: 'main' }, { label: "main (default)", value: "main" },
{ label: 'main10', value: 'main10' }, { label: "main10", value: "main10" },
{ label: 'rext', value: 'rext' } { label: "rext", value: "rext" },
], ],
descriptionHTML: 'Set the HEVC profile', descriptionHTML: "Set the HEVC profile",
default: DEFAULT_HEVC_PROFILE.toString(), default: DEFAULT_HEVC_PROFILE.toString(),
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'h264-profile', name: "h264-profile",
label: 'H264 Profile', label: "H264 Profile",
type: 'select', type: "select",
options: [ options: [
{ label: 'main (default)', value: 'main' }, { label: "main (default)", value: "main" },
{ label: 'high', value: 'high' }, { label: "high", value: "high" },
{ label: 'high444p', value: 'high444p' } { label: "high444p", value: "high444p" },
], ],
descriptionHTML: 'Set the H264 profile', descriptionHTML: "Set the H264 profile",
default: DEFAULT_H264_PROFILE.toString(), default: DEFAULT_H264_PROFILE.toString(),
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'live-quality', name: "live-quality",
label: 'Live Quality', label: "Live Quality",
type: 'select', type: "select",
options: [ options: [
{ label: 'low latency', value: 'll' }, { label: "low latency", value: "ll" },
{ label: 'high quality (default)', value: 'hq' }, { label: "high quality (default)", value: "hq" },
{ label: 'low latency high performance', value: 'ull' } { label: "low latency high performance", value: "ull" },
], ],
descriptionHTML: 'This parameter controls the speed / quality tradeoff. High performance mean lower quality.', descriptionHTML:
"This parameter controls the speed / quality tradeoff. High performance mean lower quality.",
default: DEFAULT_LIVE_QUALITY.toString(), default: DEFAULT_LIVE_QUALITY.toString(),
private: false private: false,
}); });
registerSetting({ registerSetting({
name: 'base-bitrate-description', name: "base-bitrate-description",
label: 'Base bitrate', label: "Base bitrate",
type: 'html', type: "html",
html: '', html: "",
descriptionHTML: `The base bitrate for video in bits. We take the min bitrate between the bitrate setting and video bitrate.<br/>This is the bitrate used when the video is transcoded at 30 FPS. The bitrate will be scaled linearly between this value and the maximum bitrate when the video is transcoded at 60 FPS. Wrong values are replaced by default values.`, descriptionHTML: `The base bitrate for video in bits. We take the min bitrate between the bitrate setting and video bitrate.<br/>This is the bitrate used when the video is transcoded at 30 FPS. The bitrate will be scaled linearly between this value and the maximum bitrate when the video is transcoded at 60 FPS. Wrong values are replaced by default values.`,
private: true, private: true,
}); });
@ -144,10 +154,10 @@ async function register({ settingsManager, peertubeHelpers, transcodingManager:
registerSetting({ registerSetting({
name: `base-bitrate-${resolution}`, name: `base-bitrate-${resolution}`,
label: `Base bitrate for ${printResolution(resolution)}`, label: `Base bitrate for ${printResolution(resolution)}`,
type: 'input', type: "input",
default: DEFAULT_BITRATES.get(resolution)?.toString(), default: DEFAULT_BITRATES.get(resolution)?.toString(),
descriptionHTML: `Default value: ${DEFAULT_BITRATES.get(resolution)}`, descriptionHTML: `Default value: ${DEFAULT_BITRATES.get(resolution)}`,
private: false private: false,
}); });
} }
settingsManager.onSettingsChange(async (settings) => { settingsManager.onSettingsChange(async (settings) => {
@ -162,19 +172,34 @@ async function unregister() {
} }
exports.unregister = unregister; exports.unregister = unregister;
async function loadSettings(settingsManager) { async function loadSettings(settingsManager) {
pluginSettings.hardwareDecode = await settingsManager.getSetting('hardware-decode') || DEFAULT_HARDWARE_DECODE; pluginSettings.hardwareDecode =
pluginSettings.vodQuality = await settingsManager.getSetting('vod-quality') || DEFAULT_VOD_QUALITY; (await settingsManager.getSetting("hardware-decode")) ||
pluginSettings.liveQuality = await settingsManager.getSetting('live-quality') || DEFAULT_LIVE_QUALITY; DEFAULT_HARDWARE_DECODE;
pluginSettings.hevcProfile = await settingsManager.getSetting('hevc-profile') || DEFAULT_HEVC_PROFILE; pluginSettings.vodQuality =
pluginSettings.cqH264 = parseInt(await settingsManager.getSetting('cq-h264')) || DEFAULT_CQ_H264; (await settingsManager.getSetting("vod-quality")) || DEFAULT_VOD_QUALITY;
pluginSettings.cqHEVC = parseInt(await settingsManager.getSetting('cq-hevc')) || DEFAULT_CQ_HEVC; pluginSettings.liveQuality =
pluginSettings.h264Profile = await settingsManager.getSetting('h264-profile') || DEFAULT_H264_PROFILE; (await settingsManager.getSetting("live-quality")) || DEFAULT_LIVE_QUALITY;
pluginSettings.hevcProfile =
(await settingsManager.getSetting("hevc-profile")) || DEFAULT_HEVC_PROFILE;
pluginSettings.cqH264 =
parseInt(await settingsManager.getSetting("cq-h264")) || DEFAULT_CQ_H264;
pluginSettings.cqHEVC =
parseInt(await settingsManager.getSetting("cq-hevc")) || DEFAULT_CQ_HEVC;
pluginSettings.h264Profile =
(await settingsManager.getSetting("h264-profile")) || DEFAULT_H264_PROFILE;
for (const [resolution, bitrate] of DEFAULT_BITRATES) { for (const [resolution, bitrate] of DEFAULT_BITRATES) {
const key = `base-bitrate-${resolution}`; const key = `base-bitrate-${resolution}`;
const storedValue = await settingsManager.getSetting(key); const storedValue = await settingsManager.getSetting(key);
pluginSettings.baseBitrate.set(resolution, parseInt(storedValue) || bitrate); pluginSettings.baseBitrate.set(
logger.info(`Bitrate ${printResolution(resolution)}: ${pluginSettings.baseBitrate.get(resolution)}`); resolution,
parseInt(storedValue) || bitrate
);
logger.info(
`Bitrate ${printResolution(resolution)}: ${pluginSettings.baseBitrate.get(
resolution
)}`
);
} }
logger.info(`Hardware decode: ${pluginSettings.hardwareDecode}`); logger.info(`Hardware decode: ${pluginSettings.hardwareDecode}`);
logger.info(`VOD Quality: ${pluginSettings.vodQuality}`); logger.info(`VOD Quality: ${pluginSettings.vodQuality}`);
@ -184,7 +209,8 @@ async function loadSettings(settingsManager) {
} }
function printResolution(resolution) { function printResolution(resolution) {
switch (resolution) { switch (resolution) {
case 0: return 'audio only'; case 0:
return "audio only";
case 144: case 144:
case 360: case 360:
case 480: case 480:
@ -192,21 +218,17 @@ function printResolution(resolution) {
case 1080: case 1080:
case 1440: case 1440:
return `${resolution}p`; return `${resolution}p`;
case 2160: return '4K'; case 2160:
default: return 'Unknown'; return "4K";
default:
return "Unknown";
} }
} }
function buildInitOptions() { function buildInitOptions() {
if (pluginSettings.hardwareDecode) { if (pluginSettings.hardwareDecode) {
return [ return ["-hwaccel cuda", "-hwaccel_output_format cuda"];
'-hwaccel cuda', } else {
'-hwaccel_output_format cuda' return ["-hwaccel cuda"];
];
}
else {
return [
'-hwaccel cuda'
];
} }
} }
@ -214,20 +236,20 @@ function buildInitOptions() {
async function vodBuilder(params) { async function vodBuilder(params) {
const { resolution, fps, streamNum, inputBitrate } = params; const { resolution, fps, streamNum, inputBitrate } = params;
const streamSuffix = streamNum == undefined ? '' : `:${streamNum}`; const streamSuffix = streamNum == undefined ? "" : `:${streamNum}`;
let targetBitrate = getTargetBitrate(resolution, fps); // let targetBitrate = getTargetBitrate(resolution, fps);
// let targetBitrate = inputBitrate; let targetBitrate = inputBitrate;
let shouldInitVaapi = (streamNum == undefined || streamNum <= latestStreamNum); let shouldInitVaapi = streamNum == undefined || streamNum <= latestStreamNum;
if (targetBitrate > inputBitrate) { // if (targetBitrate > inputBitrate) {
targetBitrate = inputBitrate; // targetBitrate = inputBitrate;
} // }
logger.info(`Building encoder options, received ${JSON.stringify(params)}`); logger.info(`Building encoder options, received ${JSON.stringify(params)}`);
if (shouldInitVaapi && streamNum != undefined) { if (shouldInitVaapi && streamNum != undefined) {
latestStreamNum = streamNum; latestStreamNum = streamNum;
} }
let options = { let options = {
scaleFilter: { scaleFilter: {
name: 'scale' name: "scale",
}, },
inputOptions: shouldInitVaapi ? buildInitOptions() : [], inputOptions: shouldInitVaapi ? buildInitOptions() : [],
outputOptions: [ outputOptions: [
@ -236,8 +258,8 @@ async function vodBuilder(params) {
`-bufsize ${targetBitrate * 2}`, `-bufsize ${targetBitrate * 2}`,
`-profile:v${streamSuffix} ${pluginSettings.h264Profile}`, `-profile:v${streamSuffix} ${pluginSettings.h264Profile}`,
`-cq ${pluginSettings.cqH264}`, `-cq ${pluginSettings.cqH264}`,
`-bf 4` `-bf 4`,
] ],
}; };
logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: false`); logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: false`);
@ -248,12 +270,13 @@ async function vodBuilder(params) {
async function liveBuilder(params) { async function liveBuilder(params) {
const { resolution, fps, streamNum, inputBitrate } = params; const { resolution, fps, streamNum, inputBitrate } = params;
const streamSuffix = streamNum == undefined ? '' : `:${streamNum}`; const streamSuffix = streamNum == undefined ? "" : `:${streamNum}`;
let targetBitrate = getTargetBitrate(resolution, fps); //let targetBitrate = getTargetBitrate(resolution, fps);
let shouldInitVaapi = (streamNum == undefined || streamNum <= latestStreamNum); let targetBitrate = inputBitrate;
if (targetBitrate > inputBitrate) { let shouldInitVaapi = streamNum == undefined || streamNum <= latestStreamNum;
targetBitrate = inputBitrate; //if (targetBitrate > inputBitrate) {
} // targetBitrate = inputBitrate;
// }
logger.info(`Building encoder options, received ${JSON.stringify(params)}`); logger.info(`Building encoder options, received ${JSON.stringify(params)}`);
if (shouldInitVaapi && streamNum != undefined) { if (shouldInitVaapi && streamNum != undefined) {
latestStreamNum = streamNum; latestStreamNum = streamNum;
@ -261,7 +284,7 @@ async function liveBuilder(params) {
let options = { let options = {
scaleFilter: { scaleFilter: {
name: 'scale' name: "scale",
}, },
inputOptions: shouldInitVaapi ? buildInitOptions() : [], inputOptions: shouldInitVaapi ? buildInitOptions() : [],
outputOptions: [ outputOptions: [
@ -272,8 +295,8 @@ async function liveBuilder(params) {
`-g:v${streamSuffix} ${fps * 2}`, `-g:v${streamSuffix} ${fps * 2}`,
`-b:v${streamSuffix} ${targetBitrate}`, `-b:v${streamSuffix} ${targetBitrate}`,
`-bufsize ${targetBitrate * 2}`, `-bufsize ${targetBitrate * 2}`,
`-bf 4` `-bf 4`,
] ],
}; };
logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: false`); logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: false`);
@ -283,20 +306,20 @@ async function liveBuilder(params) {
//HEVC VOD builder //HEVC VOD builder
async function hevcVODBuilder(params) { async function hevcVODBuilder(params) {
const { resolution, fps, streamNum, inputBitrate } = params; const { resolution, fps, streamNum, inputBitrate } = params;
const streamSuffix = streamNum == undefined ? '' : `:${streamNum}`; const streamSuffix = streamNum == undefined ? "" : `:${streamNum}`;
let targetBitrate = getTargetBitrate(resolution, fps); // let targetBitrate = getTargetBitrate(resolution, fps);
// let targetBitrate = inputBitrate; let targetBitrate = inputBitrate;
let shouldInitVaapi = (streamNum == undefined || streamNum <= latestStreamNum); let shouldInitVaapi = streamNum == undefined || streamNum <= latestStreamNum;
if (targetBitrate > inputBitrate) { // if (targetBitrate > inputBitrate) {
targetBitrate = inputBitrate; // targetBitrate = inputBitrate;
} // }
logger.info(`Building encoder options, received ${JSON.stringify(params)}`); logger.info(`Building encoder options, received ${JSON.stringify(params)}`);
if (shouldInitVaapi && streamNum != undefined) { if (shouldInitVaapi && streamNum != undefined) {
latestStreamNum = streamNum; latestStreamNum = streamNum;
} }
let options = { let options = {
scaleFilter: { scaleFilter: {
name: 'scale' name: "scale",
}, },
inputOptions: shouldInitVaapi ? buildInitOptions() : [], inputOptions: shouldInitVaapi ? buildInitOptions() : [],
outputOptions: [ outputOptions: [
@ -304,26 +327,25 @@ async function hevcVODBuilder(params) {
`-b:v${streamSuffix} ${targetBitrate}`, `-b:v${streamSuffix} ${targetBitrate}`,
`-bufsize ${targetBitrate * 2}`, `-bufsize ${targetBitrate * 2}`,
`-profile:v${streamSuffix} ${pluginSettings.hevcProfile}`, `-profile:v${streamSuffix} ${pluginSettings.hevcProfile}`,
`-cq ${pluginSettings.cqHEVC}` `-cq ${pluginSettings.cqHEVC}`,
] ],
}; };
logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: true`); logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: true`);
return options; return options;
} }
//HEVC Live builder //HEVC Live builder
async function hevcLiveBuilder(params) { async function hevcLiveBuilder(params) {
const { resolution, fps, streamNum, inputBitrate } = params; const { resolution, fps, streamNum, inputBitrate } = params;
const streamSuffix = streamNum == undefined ? '' : `:${streamNum}`; const streamSuffix = streamNum == undefined ? "" : `:${streamNum}`;
let targetBitrate = getTargetBitrate(resolution, fps); // let targetBitrate = getTargetBitrate(resolution, fps);
// let targetBitrate = inputBitrate; let targetBitrate = inputBitrate;
let shouldInitVaapi = (streamNum == undefined || streamNum <= latestStreamNum); let shouldInitVaapi = streamNum == undefined || streamNum <= latestStreamNum;
if (targetBitrate > inputBitrate) { // if (targetBitrate > inputBitrate) {
targetBitrate = inputBitrate; // targetBitrate = inputBitrate;
} // }
logger.info(`Building encoder options, received ${JSON.stringify(params)}`); logger.info(`Building encoder options, received ${JSON.stringify(params)}`);
if (shouldInitVaapi && streamNum != undefined) { if (shouldInitVaapi && streamNum != undefined) {
latestStreamNum = streamNum; latestStreamNum = streamNum;
@ -331,7 +353,7 @@ async function hevcLiveBuilder(params) {
let options = { let options = {
scaleFilter: { scaleFilter: {
name: 'scale' name: "scale",
}, },
inputOptions: shouldInitVaapi ? buildInitOptions() : [], inputOptions: shouldInitVaapi ? buildInitOptions() : [],
outputOptions: [ outputOptions: [
@ -341,8 +363,8 @@ async function hevcLiveBuilder(params) {
`-g:v${streamSuffix} ${fps * 2}`, `-g:v${streamSuffix} ${fps * 2}`,
`-b:v${streamSuffix} ${targetBitrate}`, `-b:v${streamSuffix} ${targetBitrate}`,
`-cq ${pluginSettings.cqHEVC}`, `-cq ${pluginSettings.cqHEVC}`,
`-bufsize ${targetBitrate * 2}` `-bufsize ${targetBitrate * 2}`,
] ],
}; };
logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: true`); logger.info(`EncoderOptions: ${JSON.stringify(options)}, HEVC: true`);
@ -354,6 +376,8 @@ function getTargetBitrate(resolution, fps) {
const maxBitrate = baseBitrate * 1.6; const maxBitrate = baseBitrate * 1.6;
const maxBitrateDifference = maxBitrate - baseBitrate; const maxBitrateDifference = maxBitrate - baseBitrate;
const maxFpsDifference = 60 - 30; const maxFpsDifference = 60 - 30;
return Math.floor(baseBitrate + (fps - 30) * (maxBitrateDifference / maxFpsDifference)); return Math.floor(
baseBitrate + (fps - 30) * (maxBitrateDifference / maxFpsDifference)
);
} }
//# sourceMappingURL=main.js.map //# sourceMappingURL=main.js.map