@libmedia/avutil 0.12.0 → 1.0.1

package/src/util/encryption.ts

@@ -0,0 +1,154 @@
+/*
+ * libmedia encryption util
+ *
+ * 版权所有 (C) 2024 赵高兴
+ * Copyright (C) 2024 Gaoxing Zhao
+ *
+ * 此文件是 libmedia 的一部分
+ * This file is part of libmedia.
+ * 
+ * libmedia 是自由软件；您可以根据 GNU Lesser General Public License（GNU LGPL）3.1
+ * 或任何其更新的版本条款重新分发或修改它
+ * libmedia is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3.1 of the License, or (at your option) any later version.
+ * 
+ * libmedia 希望能够为您提供帮助，但不提供任何明示或暗示的担保，包括但不限于适销性或特定用途的保证
+ * 您应自行承担使用 libmedia 的风险，并且需要遵守 GNU Lesser General Public License 中的条款和条件。
+ * libmedia is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+import type { EncryptionInfo, EncryptionInitInfo } from '../struct/encryption'
+
+import {
+  BufferReader,
+  BufferWriter,
+  type Uint8ArrayInterface
+} from '@libmedia/common/io'
+
+// The format of the AVEncryptionInfo side data:
+// u32be scheme
+// u32be crypt_byte_block
+// u32be skip_byte_block
+// u32be key_id_size
+// u32be iv_size
+// u32be subsample_count
+// u8[key_id_size] key_id
+// u8[iv_size] iv
+// {
+//   u32be bytes_of_clear_data
+//   u32be bytes_of_protected_data
+// }[subsample_count]
+
+export function encryptionSideData2Info(buffer: Uint8ArrayInterface): EncryptionInfo {
+  const bufferReader = new BufferReader(buffer, true)
+
+  const scheme = bufferReader.readUint32()
+  const cryptByteBlock = bufferReader.readUint32()
+  const skipByteBlock = bufferReader.readUint32()
+  const keyIdSize = bufferReader.readUint32()
+  const ivSize = bufferReader.readUint32()
+  const subsampleCount = bufferReader.readUint32()
+  const info: EncryptionInfo = {
+    scheme,
+    cryptByteBlock,
+    skipByteBlock,
+    keyId: bufferReader.readBuffer(keyIdSize),
+    iv: bufferReader.readBuffer(ivSize),
+    subsamples: []
+  }
+  if (subsampleCount) {
+    for (let i = 0; i < subsampleCount; i++) {
+      info.subsamples.push({
+        bytesOfClearData: bufferReader.readUint32(),
+        bytesOfProtectedData: bufferReader.readUint32()
+      })
+    }
+  }
+  return info
+}
+
+export function encryptionInfo2SideData(info: EncryptionInfo): Uint8Array {
+  const buffer = new Uint8Array(4 * 6 + info.keyId.length + info.iv.length + info.subsamples.length * 8)
+  const writer = new BufferWriter(buffer, true)
+  writer.writeUint32(info.scheme)
+  writer.writeUint32(info.cryptByteBlock)
+  writer.writeUint32(info.skipByteBlock)
+  writer.writeUint32(info.keyId.length)
+  writer.writeUint32(info.iv.length)
+  writer.writeUint32(info.subsamples.length)
+  writer.writeBuffer(info.keyId)
+  writer.writeBuffer(info.iv)
+  info.subsamples.forEach((item) => {
+    writer.writeUint32(item.bytesOfClearData)
+    writer.writeUint32(item.bytesOfProtectedData)
+  })
+
+  return buffer
+}
+
+// The format of the AVEncryptionInitInfo side data:
+// u32be init_info_count
+// {
+//   u32be system_id_size
+//   u32be num_key_ids
+//   u32be key_id_size
+//   u32be data_size
+//   u8[system_id_size] system_id
+//   u8[key_id_size][num_key_id] key_ids
+//   u8[data_size] data
+// }[init_info_count]
+
+export function encryptionSideData2InitInfo(buffer: Uint8ArrayInterface): EncryptionInitInfo[] {
+  const bufferReader = new BufferReader(buffer, true)
+
+  const count = bufferReader.readUint32()
+
+  const infos: EncryptionInitInfo[] = []
+  for (let i = 0; i < count; i++) {
+    const systemIdSize = bufferReader.readUint32()
+    const numKeyIds = bufferReader.readUint32()
+    const keyIdSize = bufferReader.readUint32()
+    const dataSize = bufferReader.readUint32()
+    const info: EncryptionInitInfo = {
+      systemId: bufferReader.readBuffer(systemIdSize),
+      keyIds: [],
+      data: null
+    }
+    if (numKeyIds) {
+      for (let i = 0; i < numKeyIds; i++) {
+        info.keyIds.push(bufferReader.readBuffer(keyIdSize))
+      }
+    }
+    info.data = bufferReader.readBuffer(dataSize)
+    infos.push(info)
+  }
+  return infos
+}
+
+export function encryptionInitInfo2SideData(infos: EncryptionInitInfo[]): Uint8Array {
+  let size = 4
+  infos.forEach((info) => {
+    size += 4 * 4 + info.systemId.length + info.keyIds.length * (info.keyIds.length ? info.keyIds[0].length : 0) + info.data.length
+  })
+  const buffer = new Uint8Array(size)
+  const writer = new BufferWriter(buffer, true)
+  writer.writeUint32(infos.length)
+  infos.forEach((info) => {
+    writer.writeUint32(info.systemId.length)
+    writer.writeUint32(info.keyIds.length)
+    writer.writeUint32(info.keyIds.length ? info.keyIds[0].length : 0)
+    writer.writeUint32(info.data.length)
+    writer.writeBuffer(info.systemId)
+    info.keyIds.forEach((keyId) => {
+      writer.writeBuffer(keyId)
+    })
+    writer.writeBuffer(info.data)
+  })
+  return buffer
+}

package/src/util/expgolomb.ts

@@ -0,0 +1,170 @@
+/*
+ * libmedia expgolomb util
+ *
+ * 版权所有 (C) 2024 赵高兴
+ * Copyright (C) 2024 Gaoxing Zhao
+ *
+ * 此文件是 libmedia 的一部分
+ * This file is part of libmedia.
+ * 
+ * libmedia 是自由软件；您可以根据 GNU Lesser General Public License（GNU LGPL）3.1
+ * 或任何其更新的版本条款重新分发或修改它
+ * libmedia is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3.1 of the License, or (at your option) any later version.
+ * 
+ * libmedia 希望能够为您提供帮助，但不提供任何明示或暗示的担保，包括但不限于适销性或特定用途的保证
+ * 您应自行承担使用 libmedia 的风险，并且需要遵守 GNU Lesser General Public License 中的条款和条件。
+ * libmedia is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+import {
+  type BitWriter,
+  type BitReader
+} from '@libmedia/common/io'
+
+const UESizeTable = [
+  // 0 的二进制所需的比特个数
+  1,
+  // 1 的二进制所需的比特个数    
+  1,
+  // 2~3 的二进制所需的比特个数   
+  2, 2,
+  // 4~7 的二进制所需的比特个数
+  3, 3, 3, 3,
+  // 8~15 的二进制所需的比特个数
+  4, 4, 4, 4, 4, 4, 4, 4,
+  // 16~31 的二进制所需的比特个数
+  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
+  // 32~63 的二进制所需的比特个数
+  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
+  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
+  // 64~127 的二进制所需的比特个数
+  7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
+  7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
+  7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
+  7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
+  // 128~255 的二进制所需的比特个数
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8,
+  8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8
+]
+
+/**
+ * ue(v) 指数哥伦布解码
+ */
+export function readUE(bitReader: BitReader) {
+  let result = 0
+  // leadingZeroBits
+  let i = 0
+
+  while (i < 32 && bitReader.readU1() === 0) {
+    i++
+  }
+  // 计算 read_bits ( leadingZeroBits )
+  result = bitReader.readU(i)
+  // 计算 codeNum，1 << i 即为 2 的 i 次幂
+  result += (1 << i) - 1
+
+  return result
+}
+
+/**
+ * se(v) 有符号指数哥伦布解码
+ */
+export function readSE(bitReader: BitReader) {
+  let result = readUE(bitReader)
+
+  // 判断 result 的奇偶性
+  if (result & 0x01) {
+    // 如果为奇数，说明编码前 > 0
+    result = (result + 1) / 2
+  }
+  else {
+    // 如果为偶数，说明编码前 <= 0
+    result = -result / 2
+  }
+  return result
+}
+
+/**
+ * te(v) 截断指数哥伦布解码
+ */
+export function readTE(bitReader: BitReader, x: number) {
+  let result = 0
+  // 判断取值上限
+  if (x === 1) {
+    // 如果为 1 则将读取到的比特值取反
+    result = 1 - bitReader.readU1()
+  }
+  else if (x > 1) {
+    // 否则按照 ue(v) 进行解码
+    result = readUE(bitReader)
+  }
+  return result
+}
+
+/**
+ * ue(v) 指数哥伦布编码
+ */
+export function writeUE(bitWriter: BitWriter, value: number) {
+  let size = 0
+
+  if (value === 0) {
+    // 0 直接编码为 1
+    bitWriter.writeU1(1)
+  }
+  else {
+    let tmp = ++value
+    // 判断所需比特个数是否大于 16 位
+    if ( tmp >= 0x00010000 ) {
+      size += 16
+      tmp >>= 16
+    }
+
+    // 判断此时所需比特个数是否大于 8 位
+    if ( tmp >= 0x100 ) {
+      size += 8
+      tmp >>= 8
+    }
+    // 最终 tmp 移位至 8 位以内，去查表
+    size += UESizeTable[tmp]
+
+    // 最终得出编码 value 所需的总比特数：2 * size - 1
+    bitWriter.writeU(2 * size - 1, value)
+  }
+}
+
+/**
+ * se(v) 有符号指数哥伦布编码
+ */
+export function writeSE(bitWriter: BitWriter, value: number) {
+  if (value <= 0) {
+    writeUE(bitWriter, -value * 2)
+  }
+  else {
+    writeUE(bitWriter, value * 2 - 1)
+  }
+}
+
+/**
+ * te(v) 截断指数哥伦布编码
+ */
+export function writeTE(bitWriter: BitWriter, x: number, value: number) {
+  if (x === 1) {
+    bitWriter.writeU1(1 & ~value)
+  }
+  else if (x > 1) {
+    writeUE(bitWriter, value)
+  }
+}

package/src/util/intread.ts

@@ -0,0 +1,60 @@
+/*
+ * libmedia int read util
+ *
+ * 版权所有 (C) 2024 赵高兴
+ * Copyright (C) 2024 Gaoxing Zhao
+ *
+ * 此文件是 libmedia 的一部分
+ * This file is part of libmedia.
+ * 
+ * libmedia 是自由软件；您可以根据 GNU Lesser General Public License（GNU LGPL）3.1
+ * 或任何其更新的版本条款重新分发或修改它
+ * libmedia is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3.1 of the License, or (at your option) any later version.
+ * 
+ * libmedia 希望能够为您提供帮助，但不提供任何明示或暗示的担保，包括但不限于适销性或特定用途的保证
+ * 您应自行承担使用 libmedia 的风险，并且需要遵守 GNU Lesser General Public License 中的条款和条件。
+ * libmedia is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+export function r8(p: pointer<void>): uint8 {
+  return accessof(reinterpret_cast<pointer<uint8>>(p))
+}
+
+export function rl16(p: pointer<void>): uint16 {
+  return (r8(reinterpret_cast<pointer<uint8>>(p + 1)) << 8) | r8(p)
+}
+
+export function rb16(p: pointer<void>): uint16 {
+  return (r8(p) << 8) | r8(reinterpret_cast<pointer<uint8>>(p + 1))
+}
+
+export function rl24(p: pointer<void>): int32 {
+  return (r8(reinterpret_cast<pointer<uint8>>(p + 2)) << 16) | (r8(reinterpret_cast<pointer<uint8>>(p + 1)) << 8) + r8(p)
+}
+
+export function rb24(p: pointer<void>): int32 {
+  return (r8(p) << 16) | (r8(reinterpret_cast<pointer<uint8>>(p + 1)) << 8) | r8(reinterpret_cast<pointer<uint8>>(p + 2))
+}
+
+export function rl32(p: pointer<void>): int32 {
+  return (rl16(reinterpret_cast<pointer<uint8>>(p + 2)) << 16) | rl16(p)
+}
+
+export function rb32(p: pointer<void>): int32 {
+  return (rb16(p) << 16) | rb16(reinterpret_cast<pointer<uint8>>(p + 2))
+}
+
+export function rl64(p: pointer<void>): int64 {
+  return (BigInt(rl32(reinterpret_cast<pointer<uint8>>(p + 4))) << 32n) | BigInt(rl32(p))
+}
+
+export function rb64(p: pointer<void>): int64 {
+  return (BigInt(rb32(p)) << 32n) | BigInt(rb32(reinterpret_cast<pointer<uint8>>(p + 4)))
+}

package/src/util/intwrite.ts

@@ -0,0 +1,71 @@
+
+/*
+ * libmedia int write util
+ *
+ * 版权所有 (C) 2024 赵高兴
+ * Copyright (C) 2024 Gaoxing Zhao
+ *
+ * 此文件是 libmedia 的一部分
+ * This file is part of libmedia.
+ * 
+ * libmedia 是自由软件；您可以根据 GNU Lesser General Public License（GNU LGPL）3.1
+ * 或任何其更新的版本条款重新分发或修改它
+ * libmedia is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3.1 of the License, or (at your option) any later version.
+ * 
+ * libmedia 希望能够为您提供帮助，但不提供任何明示或暗示的担保，包括但不限于适销性或特定用途的保证
+ * 您应自行承担使用 libmedia 的风险，并且需要遵守 GNU Lesser General Public License 中的条款和条件。
+ * libmedia is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+export function w8(p: pointer<void>, value: number) {
+  accessof(reinterpret_cast<pointer<uint8>>(p)) <- reinterpret_cast<uint8>(value & 0xff)
+}
+
+export function wl16(p: pointer<void>, value: uint16) {
+  w8(p, value)
+  w8(p + 1, value >>> 8)
+}
+
+export function wb16(p: pointer<void>, value: uint16) {
+  w8(p, value >>> 8)
+  w8(p + 1, value)
+}
+
+export function wl24(p: pointer<void>, value: uint32) {
+  w8(p, value)
+  w8(p + 1, value >>> 8)
+  w8(p + 2, value >>> 16)
+}
+
+export function wb24(p: pointer<void>, value: uint32) {
+  w8(p, value >>> 16)
+  w8(p + 1, value >>> 8)
+  w8(p + 2, value)
+}
+
+export function wl32(p: pointer<void>, value: uint32) {
+  wl16(p, value & 0xffff)
+  wl16(p + 2, value >>> 16)
+}
+
+export function wb32(p: pointer<void>, value: uint32) {
+  wb16(p, value >>> 16)
+  wb16(p + 2, value & 0xffff)
+}
+
+export function wl64(p: pointer<void>, value: uint64) {
+  wl32(p, static_cast<uint32>(value))
+  wl32(p + 4, static_cast<uint32>(value >> 32n))
+}
+
+export function wb64(p: pointer<void>, value: uint64) {
+  wb32(p, static_cast<uint32>(value >> 32n))
+  wb32(p + 4, static_cast<uint32>(value))
+}

package/src/util/mem.ts

@@ -0,0 +1,60 @@
+/*
+ * libmedia mem util
+ *
+ * 版权所有 (C) 2024 赵高兴
+ * Copyright (C) 2024 Gaoxing Zhao
+ *
+ * 此文件是 libmedia 的一部分
+ * This file is part of libmedia.
+ * 
+ * libmedia 是自由软件；您可以根据 GNU Lesser General Public License（GNU LGPL）3.1
+ * 或任何其更新的版本条款重新分发或修改它
+ * libmedia is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3.1 of the License, or (at your option) any later version.
+ * 
+ * libmedia 希望能够为您提供帮助，但不提供任何明示或暗示的担保，包括但不限于适销性或特定用途的保证
+ * 您应自行承担使用 libmedia 的风险，并且需要遵守 GNU Lesser General Public License 中的条款和条件。
+ * libmedia is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+import { memset } from '@libmedia/cheap'
+import { logger } from '@libmedia/common'
+
+export function avMalloc<T = void>(len: size): pointer<T> {
+  const p = malloc(len)
+  if (!p) {
+    logger.fatal('can not alloc buffer')
+  }
+  return p as pointer<T>
+}
+
+export function avMallocz<T = void>(len: size): pointer<T> {
+  const p = malloc(len)
+  if (!p) {
+    logger.fatal('can not alloc buffer')
+  }
+
+  memset(p, 0, len)
+  return p as pointer<T>
+}
+
+export function avFree(p: pointer<void>) {
+  assert(p, 'can not free empty pointer')
+  free(p)
+}
+
+export function avFreep(p: pointer<pointer<void>>) {
+  assert(p, 'can not free empty pointer')
+  free(accessof(p))
+  accessof(p) <- nullptr
+}
+
+export function avRealloc<T = void>(p: pointer<T>, size: size): pointer<T> {
+  return realloc(p as pointer<void>, size) as pointer<T>
+}