diff --git a/src/main.ts b/src/main.ts index b9b55dc..43b984f 100644 --- a/src/main.ts +++ b/src/main.ts @@ -1,4 +1,4 @@ -import { PluginTranscodingManager } from "@peertube/peertube-types" +import { PluginSettingsManager, PluginTranscodingManager } from "@peertube/peertube-types" import { EncoderOptions, EncoderOptionsBuilderParams, RegisterServerOptions, VideoResolution } from "@peertube/peertube-types" import { Logger } from 'winston' @@ -7,11 +7,7 @@ let transcodingManager : PluginTranscodingManager const DEFAULT_HARDWARE_DECODE : boolean = false const DEFAULT_QUALITY : number = -1 - -let hardwareDecode : boolean = DEFAULT_HARDWARE_DECODE -let quality : number = DEFAULT_QUALITY - -let baseBitrates : Map = new Map([ +const DEFAULT_BITRATES : Map = new Map([ [VideoResolution.H_NOVIDEO, 64 * 1000], [VideoResolution.H_144P, 320 * 1000], [VideoResolution.H_360P, 780 * 1000], @@ -22,11 +18,22 @@ let baseBitrates : Map = new Map([ [VideoResolution.H_4K, 22_000 * 1000] ]) +interface PluginSettings { + hardwareDecode : boolean + quality: number + baseBitrate: Map +} +let pluginSettings : PluginSettings = { + hardwareDecode: DEFAULT_HARDWARE_DECODE, + quality: DEFAULT_QUALITY, + baseBitrate: structuredClone(DEFAULT_BITRATES) +} + let latestStreamNum = 9999 -export async function register(options :RegisterServerOptions) { - logger = options.peertubeHelpers.logger - transcodingManager = options.transcodingManager +export async function register({settingsManager, peertubeHelpers, transcodingManager: transcode, registerSetting} :RegisterServerOptions) { + logger = peertubeHelpers.logger + transcodingManager = transcode logger.info("Registering peertube-plugin-hardware-encode"); @@ -40,22 +47,10 @@ export async function register(options :RegisterServerOptions) { transcodingManager.addLiveProfile(encoder, profileName, liveBuilder) transcodingManager.addLiveEncoderPriority('video', encoder, 1000) - // Get stored data from the database, default to constants if not found - hardwareDecode = await options.storageManager.getData('hardware-decode') == "true" // ?? DEFAULT_HARDWARE_DECODE // not needed, since undefined == "true" -> false - quality = parseInt(await options.storageManager.getData('compression-level')) || DEFAULT_QUALITY + // Load existing settings and default to constants if not present + await loadSettings(settingsManager) - for (const [resolution, bitrate] of baseBitrates) { - const key = `base-bitrate-${resolution}` - const storedValue = await options.storageManager.getData(key) - if (storedValue) { - baseBitrates.set(resolution, parseInt(storedValue) || bitrate) - } - } - - logger.info(`Hardware decode: ${hardwareDecode}`) - logger.info(`Quality: ${quality}`) - - options.registerSetting({ + registerSetting({ name: 'hardware-decode', label: 'Hardware decode', @@ -63,10 +58,10 @@ export async function register(options :RegisterServerOptions) { 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: hardwareDecode, + default: DEFAULT_HARDWARE_DECODE, private: false }) - options.registerSetting({ + registerSetting({ name: 'quality', label: 'Quality', @@ -84,48 +79,37 @@ export async function register(options :RegisterServerOptions) { descriptionHTML: 'This parameter controls the speed / quality tradeoff. Lower values mean better quality but slower encoding. Higher values mean faster encoding but lower quality. This setting is hardware dependent, you may need to experiment to find the best value for your hardware. Some hardware may have less than 7 levels of compression.', - default: quality.toString(), + default: DEFAULT_QUALITY.toString(), private: false }) - options.registerSetting({ + registerSetting({ name: 'base-bitrate-description', label: 'Base bitrate', type: 'html', html: '', - descriptionHTML: `The base bitrate for video in bits. 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.`, + descriptionHTML: `The base bitrate for video in bits. We take the min bitrate between the bitrate setting and video bitrate.
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, }) - for (const [resolution, bitrate] of baseBitrates) { - options.registerSetting({ + for (const [resolution, bitrate] of pluginSettings.baseBitrate) { + logger.info("registering bitrate setting: "+ bitrate.toString()) + registerSetting({ name: `base-bitrate-${resolution}`, label: `Base bitrate for ${printResolution(resolution)}`, type: 'input', - default: bitrate.toString(), + default: DEFAULT_BITRATES.get(resolution)?.toString(), + descriptionHTML: `Default value: ${DEFAULT_BITRATES.get(resolution)}`, private: false }) } - options.settingsManager.onSettingsChange(async (settings) => { - hardwareDecode = settings['hardware-decode'] as boolean - quality = parseInt(settings['quality'] as string) || DEFAULT_QUALITY - - for (const [resolution, bitrate] of baseBitrates) { - const key = `base-bitrate-${resolution}` - const storedValue = settings[key] as string - if (storedValue) { - baseBitrates.set(resolution, parseInt(storedValue) || bitrate) - logger.info(`New base bitrate for ${resolution}: ${baseBitrates.get(resolution)}`) - } - } - - logger.info(`New hardware decode: ${hardwareDecode}`) - logger.info(`New quality: ${quality}`) + settingsManager.onSettingsChange(async (settings) => { + loadSettings(settingsManager) }) } @@ -135,6 +119,21 @@ export async function unregister() { return true } +async function loadSettings(settingsManager: PluginSettingsManager) { + pluginSettings.hardwareDecode = await settingsManager.getSetting('hardware-decode') == "true" + pluginSettings.quality = parseInt(await settingsManager.getSetting('quality') as string) || DEFAULT_QUALITY + + for (const [resolution, bitrate] of DEFAULT_BITRATES) { + const key = `base-bitrate-${resolution}` + const storedValue = await settingsManager.getSetting(key) as string + pluginSettings.baseBitrate.set(resolution, parseInt(storedValue) || bitrate) + logger.info(`Bitrate ${printResolution(resolution)}: ${pluginSettings.baseBitrate.get(resolution)}`) + } + + logger.info(`Hardware decode: ${pluginSettings.hardwareDecode}`) + logger.info(`Quality: ${pluginSettings.quality}`) +} + function printResolution(resolution : VideoResolution) : string { switch (resolution) { case VideoResolution.H_NOVIDEO: return 'audio only' @@ -152,7 +151,7 @@ function printResolution(resolution : VideoResolution) : string { } function buildInitOptions() { - if (hardwareDecode) { + if (pluginSettings.hardwareDecode) { return [ '-hwaccel vaapi', '-vaapi_device /dev/dri/renderD128', @@ -184,11 +183,11 @@ async function vodBuilder(params: EncoderOptionsBuilderParams) : Promise