mirror of https://github.com/Chocobozzz/PeerTube
119 lines
3.1 KiB
TypeScript
119 lines
3.1 KiB
TypeScript
import { VideoResolution, VideoResolutionType } from '@peertube/peertube-models'
|
|
|
|
type BitPerPixel = { [ id in VideoResolutionType ]: number }
|
|
|
|
// https://bitmovin.com/video-bitrate-streaming-hls-dash/
|
|
|
|
const minLimitBitPerPixel: BitPerPixel = {
|
|
[VideoResolution.H_NOVIDEO]: 0,
|
|
[VideoResolution.H_144P]: 0.02,
|
|
[VideoResolution.H_240P]: 0.02,
|
|
[VideoResolution.H_360P]: 0.02,
|
|
[VideoResolution.H_480P]: 0.02,
|
|
[VideoResolution.H_720P]: 0.02,
|
|
[VideoResolution.H_1080P]: 0.02,
|
|
[VideoResolution.H_1440P]: 0.02,
|
|
[VideoResolution.H_4K]: 0.02
|
|
}
|
|
|
|
const averageBitPerPixel: BitPerPixel = {
|
|
[VideoResolution.H_NOVIDEO]: 0,
|
|
[VideoResolution.H_144P]: 0.19,
|
|
[VideoResolution.H_240P]: 0.17,
|
|
[VideoResolution.H_360P]: 0.15,
|
|
[VideoResolution.H_480P]: 0.12,
|
|
[VideoResolution.H_720P]: 0.11,
|
|
[VideoResolution.H_1080P]: 0.10,
|
|
[VideoResolution.H_1440P]: 0.09,
|
|
[VideoResolution.H_4K]: 0.08
|
|
}
|
|
|
|
const maxBitPerPixel: BitPerPixel = {
|
|
[VideoResolution.H_NOVIDEO]: 0,
|
|
[VideoResolution.H_144P]: 0.32,
|
|
[VideoResolution.H_240P]: 0.29,
|
|
[VideoResolution.H_360P]: 0.26,
|
|
[VideoResolution.H_480P]: 0.22,
|
|
[VideoResolution.H_720P]: 0.19,
|
|
[VideoResolution.H_1080P]: 0.17,
|
|
[VideoResolution.H_1440P]: 0.16,
|
|
[VideoResolution.H_4K]: 0.14
|
|
}
|
|
|
|
function getAverageTheoreticalBitrate (options: {
|
|
resolution: number
|
|
ratio: number
|
|
fps: number
|
|
}) {
|
|
const targetBitrate = calculateBitrate({ ...options, bitPerPixel: averageBitPerPixel })
|
|
if (!targetBitrate) return 192 * 1000
|
|
|
|
return targetBitrate
|
|
}
|
|
|
|
function getMaxTheoreticalBitrate (options: {
|
|
resolution: number
|
|
ratio: number
|
|
fps: number
|
|
}) {
|
|
const targetBitrate = calculateBitrate({ ...options, bitPerPixel: maxBitPerPixel })
|
|
if (!targetBitrate) return 256 * 1000
|
|
|
|
return targetBitrate
|
|
}
|
|
|
|
function getMinTheoreticalBitrate (options: {
|
|
resolution: number
|
|
ratio: number
|
|
fps: number
|
|
}) {
|
|
const minLimitBitrate = calculateBitrate({ ...options, bitPerPixel: minLimitBitPerPixel })
|
|
if (!minLimitBitrate) return 10 * 1000
|
|
|
|
return minLimitBitrate
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
getAverageTheoreticalBitrate,
|
|
getMaxTheoreticalBitrate,
|
|
getMinTheoreticalBitrate
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
function calculateBitrate (options: {
|
|
bitPerPixel: BitPerPixel
|
|
resolution: number
|
|
ratio: number
|
|
fps: number
|
|
}) {
|
|
const { bitPerPixel, resolution, ratio, fps } = options
|
|
|
|
const resolutionsOrder = [
|
|
VideoResolution.H_4K,
|
|
VideoResolution.H_1440P,
|
|
VideoResolution.H_1080P,
|
|
VideoResolution.H_720P,
|
|
VideoResolution.H_480P,
|
|
VideoResolution.H_360P,
|
|
VideoResolution.H_240P,
|
|
VideoResolution.H_144P,
|
|
VideoResolution.H_NOVIDEO
|
|
]
|
|
|
|
const size1 = resolution
|
|
const size2 = ratio < 1 && ratio > 0
|
|
? resolution / ratio // Portrait mode
|
|
: resolution * ratio
|
|
|
|
for (const toTestResolution of resolutionsOrder) {
|
|
if (toTestResolution <= resolution) {
|
|
return Math.floor(size1 * size2 * fps * bitPerPixel[toTestResolution])
|
|
}
|
|
}
|
|
|
|
throw new Error('Unknown resolution ' + resolution)
|
|
}
|