dasha 2.2.6 → 2.3.2

package/README.md

@@ -37,7 +37,7 @@ type VideoTrack = {
   size: number; // MB
   width: number;
   height: number;
-  quality: 'SD' | 'HD' | 'Full HD' | '4K' : '8K';
+  qualityLabel: '144p' | '240p' | '360p' | '480p' | '576p' | '720p' | '1080p' | '2160p';
 };
 
 type AudioTrack = {

package/lib/manifest.js

@@ -2,7 +2,7 @@
 
 const { parseXml } = require('./xml');
 const { processManifest } = require('./processor');
-const { getQualityLabel } = require('./utils');
+const { getQualityLabel, getHeightByWidth, getClosest } = require('./utils');
 const { CONTENT_TYPE, KEY_SYSTEMS } = require('./constants');
 
 const isUrl = (value) => {
@@ -18,6 +18,11 @@ class Manifest {
     for (const key of Object.keys(manifest)) this[key] = manifest[key];
   }
 
+  addBaseUrl(value) {
+    if (Array.isArray(this.baseUrls)) this.baseUrls.push({ value });
+    else this.baseUrls = [{ value }];
+  }
+
   getVideoTrack(height) {
     const isVideoAdaptationSet = (adaptationSet) =>
       adaptationSet.contentType === CONTENT_TYPE.video ||
@@ -29,11 +34,31 @@ class Manifest {
       .flat(1)
       .filter(isVideoAdaptationSet);
     const representations = adaptationSets.map((a) => a.representations).flat(2);
-    const matchesInHeight = representations.filter((r) => r.height === height);
-    const matchesInId = representations.filter((r) => r.id.includes(height));
-    const matches = matchesInHeight?.length ? matchesInHeight : matchesInId;
-    const representation = this.findBestRepresentation(matches.length ? matches : representations);
+
+    let suitableRepresentations = [];
+    if (height) {
+      const resolutions = representations.map((r) => ({
+        height: getHeightByWidth(r.width),
+        width: r.width,
+      }));
+      const heights = resolutions.map((r) => r.height);
+      const matchedHeight = height ? getClosest(height, heights) : Math.max(...heights);
+      const matchedResolution = resolutions.find((r) => r.height === matchedHeight);
+      const matchesInHeight = representations.filter((r) => matchedResolution.width === r.width);
+      const matchesInId = representations.filter((r) => r.id.includes(height.toString()));
+      suitableRepresentations = matchesInHeight?.length ? matchesInHeight : matchesInId;
+    }
+
+    const representation = this.findBestRepresentation(
+      suitableRepresentations.length ? suitableRepresentations : representations
+    );
     const adaptationSet = adaptationSets.find((a) => a.representations.includes(representation));
+    let codec = 'x264';
+    if (representation.codecs.includes('hvc') || representation.codecs.includes('hev'))
+      codec = 'x265';
+    let bitDepth = '8bit';
+    if (representation.codecs.includes('.2.4.')) bitDepth = '10bit';
+    if (representation.codecs.includes('.4.16.')) bitDepth = '12bit';
     return {
       id: 0,
       type: CONTENT_TYPE.video,
@@ -44,7 +69,11 @@ class Manifest {
       size: Math.ceil((representation.bandwidth / 8e6) * this.mediaPresentationDuration), // MB
       width: representation.width,
       height: representation.height,
-      quality: getQualityLabel(representation.width),
+      qualityLabel: getQualityLabel(representation.width),
+      hevc: representation.codecs.includes('hvc') || representation.codecs.includes('hev'),
+      codec,
+      codecs: representation.codecs,
+      bitDepth,
     };
   }
 
@@ -65,6 +94,7 @@ class Manifest {
     const tracks = [];
     for (const adaptationSet of adaptationSets) {
       const representation = this.findBestRepresentation(adaptationSet.representations);
+      let codec = '';
       const track = {
         id: tracks.length,
         type: CONTENT_TYPE.audio,
@@ -75,7 +105,8 @@ class Manifest {
         segments: this.getSegments(adaptationSet, representation),
         bitrate: Math.round(representation.bandwidth / 1000), // Kbps
         size: Math.ceil((representation.bandwidth / 8e6) * this.mediaPresentationDuration), // MB
-        audioSampleRate: representation.audioSamplingRate / 1000,
+        audioSamplingRate: representation.audioSamplingRate / 1000,
+        codecs: representation.codecs,
       };
       tracks.push(track);
     }

package/lib/utils.js

@@ -34,16 +34,24 @@ const parseDuration = (durationString) => {
   return isFinite(d) ? d : null;
 };
 
-const getQualityLabel = (videoWidth) => {
+const getHeightByWidth = (videoWidth) => {
   const width = typeof videoWidth === 'number' ? videoWidth : parseInt(videoWidth);
   if (isNaN(width)) return;
-  let label = 'SD';
-  if (width >= 1280) label = 'HD';
-  if (width >= 1920) label = 'Full HD';
-  if (width >= 3840) label = 'UHD';
-  if (width >= 4096) label = '4K';
-  if (width >= 7680) label = '8K';
-  return label;
+  let height = 144;
+  if (width >= 426) height = 240;
+  if (width >= 640) height = 360;
+  if (width >= 854) height = 480;
+  if (width >= 1024) height = 576;
+  if (width >= 1280) height = 720;
+  if (width >= 1920) height = 1080;
+  if (width >= 3840) height = 2160;
+  if (width >= 7680) height = 4320;
+  return height;
 };
 
-module.exports = { parseDuration, getQualityLabel };
+const getClosest = (value, array) =>
+  array.reduce((prev, curr) => (Math.abs(curr - value) < Math.abs(prev - value) ? curr : prev));
+
+const getQualityLabel = (videoWidth) => getHeightByWidth(videoWidth) + 'p';
+
+module.exports = { parseDuration, getHeightByWidth, getQualityLabel, getClosest };

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "dasha",
-  "version": "2.2.6",
-  "author": "Vitaliy Gashkov <vitnore@gmail.com>",
-  "description": "Parser of MPD-manifests for MPEG DASH",
+  "version": "2.3.2",
+  "author": "Vitaly Gashkov <vitalygashkov@bk.ru>",
+  "description": "MPD-manifest parser for MPEG DASH",
   "license": "Apache-2.0",
   "keywords": [
     "mpeg",
@@ -19,11 +19,11 @@
   ],
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/vitnore/dasha.git"
+    "url": "git+https://github.com/vitalygashkov/dasha.git"
   },
   "funding": {
     "type": "individual",
-    "url": "https://boosty.to/vitnore"
+    "url": "https://boosty.to/vitalygashkov"
   },
   "scripts": {
     "test": "jest",