rx-player/src/transports/smooth/isobmff/create_video_init_segment.ts

Canal+ HTML5 Video Player

/**
 * Copyright 2015 CANAL+ Group
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { createBoxWithChildren } from "../../../parsers/containers/isobmff";
import { hexToBytes } from "../../../utils/string_parsing";
import {
  createAVC1Box,
  createAVCCBox,
  createENCVBox,
  createFRMABox,
  createSCHMBox,
  createSTSDBox,
  createTENCBox,
  createVMHDBox,
} from "./create_boxes";
import createInitSegment from "./create_init_segment";

/**
 * Return full video Init segment as Uint8Array
 * @param {Number} timescale - lowest number, this one will be set into mdhd
 * *10000 in mvhd, e.g. 1000
 * @param {Number} width
 * @param {Number} height
 * @param {Number} hRes
 * @param {Number} vRes
 * @param {Number} nalLength (1, 2 or 4)
 * @param {string} codecPrivateData
 * @param {Uint8Array} [keyId]
 * @returns {Uint8Array}
 */
export default function createVideoInitSegment(
  timescale: number,
  width: number,
  height: number,
  hRes: number,
  vRes: number,
  nalLength: number,
  codecPrivateData: string,
  keyId?: Uint8Array,
): Uint8Array<ArrayBuffer> {
  const [, spsHex, ppsHex] = codecPrivateData.split("00000001");
  if (spsHex === undefined || ppsHex === undefined) {
    throw new Error("Smooth: unsupported codec private data.");
  }
  const sps = hexToBytes(spsHex);
  const pps = hexToBytes(ppsHex);

  // TODO NAL length is forced to 4
  const avcc = createAVCCBox(sps, pps, nalLength);
  let stsd;
  if (keyId === undefined) {
    const avc1 = createAVC1Box(width, height, hRes, vRes, "AVC Coding", 24, avcc);
    stsd = createSTSDBox([avc1]);
  } else {
    const tenc = createTENCBox(1, 8, keyId);
    const schi = createBoxWithChildren("schi", [tenc]);
    const schm = createSCHMBox("cenc", 65536);
    const frma = createFRMABox("avc1");
    const sinf = createBoxWithChildren("sinf", [frma, schm, schi]);
    const encv = createENCVBox(width, height, hRes, vRes, "AVC Coding", 24, avcc, sinf);
    stsd = createSTSDBox([encv]);
  }

  return createInitSegment(timescale, "video", stsd, createVMHDBox(), width, height);
}