Piped/src/utils/DashUtils.js

205 lines
6.7 KiB
JavaScript
Raw Normal View History

2021-02-24 09:35:41 +00:00
// Based of https://github.com/GilgusMaximus/yt-dash-manifest-generator/blob/master/src/DashGenerator.js
2021-12-29 00:33:15 +00:00
import { Buffer } from "buffer";
window.Buffer = Buffer;
import { json2xml } from "xml-js";
2021-02-24 09:35:41 +00:00
const DashUtils = {
generate_dash_file_from_formats(VideoFormats, VideoLength) {
const generatedJSON = this.generate_xmljs_json_from_data(VideoFormats, VideoLength);
2021-12-29 00:33:15 +00:00
return json2xml(generatedJSON);
2021-02-24 09:35:41 +00:00
},
generate_xmljs_json_from_data(VideoFormatArray, VideoLength) {
const convertJSON = {
declaration: {
attributes: {
version: "1.0",
encoding: "utf-8",
},
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "MPD",
attributes: {
xmlns: "urn:mpeg:dash:schema:mpd:2011",
profiles: "urn:mpeg:dash:profile:full:2011",
minBufferTime: "PT1.5S",
type: "static",
mediaPresentationDuration: `PT${VideoLength}S`,
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "Period",
elements: this.generate_adaptation_set(VideoFormatArray),
},
],
},
],
};
return convertJSON;
2021-02-24 09:35:41 +00:00
},
generate_adaptation_set(VideoFormatArray) {
const adaptationSets = [];
let mimeAudioObjs = [];
VideoFormatArray.forEach(videoFormat => {
2021-02-24 09:35:41 +00:00
// the dual formats should not be used
if (videoFormat.mimeType.indexOf("video") != -1 && !videoFormat.videoOnly) {
return;
2021-02-24 09:35:41 +00:00
}
const audioTrackId = videoFormat.audioTrackId;
const mimeType = videoFormat.mimeType;
for (let i = 0; i < mimeAudioObjs.length; i++) {
const mimeAudioObj = mimeAudioObjs[i];
if (mimeAudioObj.audioTrackId == audioTrackId && mimeAudioObj.mimeType == mimeType) {
mimeAudioObj.videoFormats.push(videoFormat);
return;
}
2021-02-24 09:35:41 +00:00
}
mimeAudioObjs.push({
audioTrackId,
mimeType,
videoFormats: [videoFormat],
});
});
mimeAudioObjs.forEach(mimeAudioObj => {
2021-02-24 09:35:41 +00:00
const adapSet = {
type: "element",
name: "AdaptationSet",
attributes: {
id: mimeAudioObj.audioTrackId,
lang: mimeAudioObj.audioTrackId?.substr(0, 2),
mimeType: mimeAudioObj.mimeType,
startWithSAP: "1",
subsegmentAlignment: "true",
2021-02-24 09:35:41 +00:00
},
elements: [],
};
let isVideoFormat = false;
if (mimeAudioObj.mimeType.includes("video")) {
isVideoFormat = true;
2021-02-24 09:35:41 +00:00
}
if (isVideoFormat) {
adapSet.attributes.scanType = "progressive";
}
for (var i = 0; i < mimeAudioObj.videoFormats.length; i++) {
const videoFormat = mimeAudioObj.videoFormats[i];
2021-02-24 09:35:41 +00:00
if (isVideoFormat) {
adapSet.elements.push(this.generate_representation_video(videoFormat));
2021-02-24 09:35:41 +00:00
} else {
adapSet.elements.push(this.generate_representation_audio(videoFormat));
2021-02-24 09:35:41 +00:00
}
}
adaptationSets.push(adapSet);
});
return adaptationSets;
},
generate_representation_audio(Format) {
const representation = {
type: "element",
name: "Representation",
attributes: {
id: Format.itag,
codecs: Format.codec,
bandwidth: Format.bitrate,
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "AudioChannelConfiguration",
attributes: {
schemeIdUri: "urn:mpeg:dash:23003:3:audio_channel_configuration:2011",
value: "2",
2021-02-24 09:35:41 +00:00
},
},
{
type: "element",
name: "BaseURL",
elements: [
2021-02-24 09:35:41 +00:00
{
type: "text",
text: Format.url,
},
],
2021-02-24 09:35:41 +00:00
},
{
type: "element",
name: "SegmentBase",
attributes: {
indexRange: `${Format.indexStart}-${Format.indexEnd}`,
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "Initialization",
attributes: {
range: `${Format.initStart}-${Format.initEnd}`,
},
},
],
},
],
};
return representation;
2021-02-24 09:35:41 +00:00
},
generate_representation_video(Format) {
const representation = {
type: "element",
name: "Representation",
attributes: {
id: Format.itag,
codecs: Format.codec,
bandwidth: Format.bitrate,
width: Format.width,
height: Format.height,
maxPlayoutRate: "1",
frameRate: Format.fps,
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "BaseURL",
elements: [
2021-02-24 09:35:41 +00:00
{
type: "text",
text: Format.url,
},
],
2021-02-24 09:35:41 +00:00
},
{
type: "element",
name: "SegmentBase",
attributes: {
indexRange: `${Format.indexStart}-${Format.indexEnd}`,
2021-02-24 09:35:41 +00:00
},
elements: [
2021-02-24 09:35:41 +00:00
{
type: "element",
name: "Initialization",
attributes: {
range: `${Format.initStart}-${Format.initEnd}`,
},
},
],
},
],
};
return representation;
},
};
2021-02-24 09:35:41 +00:00
export default DashUtils;