bare-ffmpeg 1.0.0-25 → 1.0.0-27

package/index.js

@@ -10,6 +10,10 @@ const {
   InputFormatContext,
   OutputFormatContext
 } = require('./lib/format-context')
+const Filter = require('./lib/filter')
+const FilterContext = require('./lib/filter-context')
+const FilterGraph = require('./lib/filter-graph')
+const FilterInOut = require('./lib/filter-inout')
 const Frame = require('./lib/frame')
 const IOContext = require('./lib/io-context')
 const Image = require('./lib/image')
@@ -31,6 +35,10 @@ exports.CodecParameters = CodecParameters
 exports.Decoder = Decoder
 exports.Dictionary = Dictionary
 exports.Encoder = Encoder
+exports.Filter = Filter
+exports.FilterContext = FilterContext
+exports.FilterGraph = FilterGraph
+exports.FilterInOut = FilterInOut
 exports.Frame = Frame
 exports.IOContext = IOContext
 exports.Image = Image

package/lib/codec-context.js

@@ -1,6 +1,7 @@
 const binding = require('../binding')
 const Rational = require('./rational')
 const ChannelLayout = require('./channel-layout')
+const { codecConfig } = require('./constants')
 
 module.exports = class FFmpegCodecContext {
   constructor(codec) {
@@ -107,6 +108,27 @@ module.exports = class FFmpegCodecContext {
     binding.setCodecContextFlags(this._handle, value)
   }
 
+  get extraData() {
+    return Buffer.from(binding.getCodecContextExtraData(this._handle))
+  }
+
+  set extraData(value) {
+    binding.setCodecContextExtraData(
+      this._handle,
+      value.buffer,
+      value.byteOffset,
+      value.byteLength
+    )
+  }
+
+  get frameSize() {
+    return binding.getCodecContextFrameSize(this._handle)
+  }
+
+  get frameNum() {
+    return binding.getCodecContextFrameNum(this._handle)
+  }
+
   open(options) {
     if (this._opened) return
     this._opened = true
@@ -138,6 +160,45 @@ module.exports = class FFmpegCodecContext {
     return binding.receiveCodecContextFrame(this._handle, frame._handle)
   }
 
+  getSupportedConfig(config) {
+    if (config === codecConfig.FRAME_RATE) {
+      const data = binding.getSupportedFrameRates(
+        this._handle,
+        this._codec._handle
+      )
+      if (!data || data.length === 0) return null
+
+      const view = new Int32Array(data)
+      const rates = []
+      for (let i = 0; i < view.length; i += 2) {
+        rates.push(new Rational(view[i], view[i + 1]))
+      }
+      return rates
+    }
+
+    if (config === codecConfig.CHANNEL_LAYOUT) {
+      const handles = binding.getSupportedChannelLayouts(
+        this._handle,
+        this._codec._handle
+      )
+      if (!handles || handles.length === 0) return null
+
+      return handles.map((handle) => new ChannelLayout(handle))
+    }
+
+    const data = binding.getSupportedConfig(
+      this._handle,
+      this._codec._handle,
+      config
+    )
+
+    if (config === codecConfig.SAMPLE_RATE && !data) {
+      return null
+    }
+
+    return new Int32Array(data)
+  }
+
   [Symbol.dispose]() {
     this.destroy()
   }
@@ -155,7 +216,8 @@ module.exports = class FFmpegCodecContext {
       sampleRate: this.sampleRate,
       timeBase: this.timeBase,
       channelLayout: this.channelLayout,
-      gopSize: this.gopSize
+      gopSize: this.gopSize,
+      extraData: this.extraData
     }
   }
 }

package/lib/codec-parameters.js

@@ -72,6 +72,14 @@ module.exports = class FFmpegCodecParameters {
     )
   }
 
+  get frameSize() {
+    return binding.getCodecParametersFrameSize(this._handle)
+  }
+
+  set frameSize(size) {
+    binding.setCodecParametersFrameSize(this._handle, size)
+  }
+
   get videoDelay() {
     return binding.getCodecParametersVideoDelay(this._handle)
   }
@@ -235,22 +243,26 @@ module.exports = class FFmpegCodecParameters {
       type: this.type,
       id: this.id,
       format: this.format,
+      tag: this.tag,
+      frameRate: this.frameRate,
+      videoDelay: this.videoDelay,
       profile: this.profile,
       level: this.level,
       width: this.width,
       height: this.height,
+      sampleAsectRatio: this.sampleAspectRatio,
       bitRate: this.bitRate,
       bitsPerCodedSample: this.bitsPerCodedSample,
       bitsPerRawSample: this.bitsPerRawSample,
       sampleRate: this.sampleRate,
       nbChannels: this.nbChannels,
       channelLayout: this.channelLayout,
-      frameRate: this.frameRate,
       extraData: this.extraData,
       blockAlign: this.blockAlign,
       initialPadding: this.initialPadding,
       trailingPadding: this.trailingPadding,
-      seekPreroll: this.seekPreroll
+      seekPreroll: this.seekPreroll,
+      frameSize: this.frameSize
     }
   }
 

package/lib/constants.js

@@ -140,6 +140,59 @@ module.exports = exports = {
     CUR: binding.SEEK_CUR,
     SET: binding.SEEK_SET,
     END: binding.SEEK_END
+  },
+  packetSideDataType: {
+    PALETTE: binding.AV_PKT_DATA_PALETTE,
+    NEW_EXTRADATA: binding.AV_PKT_DATA_NEW_EXTRADATA,
+    PARAM_CHANGE: binding.AV_PKT_DATA_PARAM_CHANGE,
+    H263_MB_INFO: binding.AV_PKT_DATA_H263_MB_INFO,
+    REPLAYGAIN: binding.AV_PKT_DATA_REPLAYGAIN,
+    DISPLAYMATRIX: binding.AV_PKT_DATA_DISPLAYMATRIX,
+    STEREO3D: binding.AV_PKT_DATA_STEREO3D,
+    AUDIO_SERVICE_TYPE: binding.AV_PKT_DATA_AUDIO_SERVICE_TYPE,
+    QUALITY_STATS: binding.AV_PKT_DATA_QUALITY_STATS,
+    FALLBACK_TRACK: binding.AV_PKT_DATA_FALLBACK_TRACK,
+    CPB_PROPERTIES: binding.AV_PKT_DATA_CPB_PROPERTIES,
+    SKIP_SAMPLES: binding.AV_PKT_DATA_SKIP_SAMPLES,
+    JP_DUALMONO: binding.AV_PKT_DATA_JP_DUALMONO,
+    STRINGS_METADATA: binding.AV_PKT_DATA_STRINGS_METADATA,
+    SUBTITLE_POSITION: binding.AV_PKT_DATA_SUBTITLE_POSITION,
+    MATROSKA_BLOCKADDITIONAL: binding.AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL,
+    WEBVTT_IDENTIFIER: binding.AV_PKT_DATA_WEBVTT_IDENTIFIER,
+    WEBVTT_SETTINGS: binding.AV_PKT_DATA_WEBVTT_SETTINGS,
+    METADATA_UPDATE: binding.AV_PKT_DATA_METADATA_UPDATE,
+    MPEGTS_STREAM_ID: binding.AV_PKT_DATA_MPEGTS_STREAM_ID,
+    MASTERING_DISPLAY_METADATA: binding.AV_PKT_DATA_MASTERING_DISPLAY_METADATA,
+    SPHERICAL: binding.AV_PKT_DATA_SPHERICAL,
+    CONTENT_LIGHT_LEVEL: binding.AV_PKT_DATA_CONTENT_LIGHT_LEVEL,
+    A53_CC: binding.AV_PKT_DATA_A53_CC,
+    ENCRYPTION_INIT_INFO: binding.AV_PKT_DATA_ENCRYPTION_INIT_INFO,
+    ENCRYPTION_INFO: binding.AV_PKT_DATA_ENCRYPTION_INFO,
+    AFD: binding.AV_PKT_DATA_AFD,
+    PRFT: binding.AV_PKT_DATA_PRFT,
+    ICC_PROFILE: binding.AV_PKT_DATA_ICC_PROFILE,
+    DOVI_CONF: binding.AV_PKT_DATA_DOVI_CONF,
+    S12M_TIMECODE: binding.AV_PKT_DATA_S12M_TIMECODE,
+    DYNAMIC_HDR10_PLUS: binding.AV_PKT_DATA_DYNAMIC_HDR10_PLUS,
+    IAMF_MIX_GAIN_PARAM: binding.AV_PKT_DATA_IAMF_MIX_GAIN_PARAM,
+    IAMF_DEMIXING_INFO_PARAM: binding.AV_PKT_DATA_IAMF_DEMIXING_INFO_PARAM,
+    IAMF_RECON_GAIN_INFO_PARAM: binding.AV_PKT_DATA_IAMF_RECON_GAIN_INFO_PARAM,
+    AMBIENT_VIEWING_ENVIRONMENT:
+      binding.AV_PKT_DATA_AMBIENT_VIEWING_ENVIRONMENT,
+    FRAME_CROPPING: binding.AV_PKT_DATA_FRAME_CROPPING,
+    LCEVC: binding.AV_PKT_DATA_LCEVC,
+    '3D_REFERENCE_DISPLAYS': binding.AV_PKT_DATA_3D_REFERENCE_DISPLAYS,
+    RTCP_SR: binding.AV_PKT_DATA_RTCP_SR,
+    NB: binding.AV_PKT_DATA_NB
+  },
+  codecConfig: {
+    PIX_FORMAT: binding.AV_CODEC_CONFIG_PIX_FORMAT,
+    FRAME_RATE: binding.AV_CODEC_CONFIG_FRAME_RATE,
+    SAMPLE_RATE: binding.AV_CODEC_CONFIG_SAMPLE_RATE,
+    SAMPLE_FORMAT: binding.AV_CODEC_CONFIG_SAMPLE_FORMAT,
+    CHANNEL_LAYOUT: binding.AV_CODEC_CONFIG_CHANNEL_LAYOUT,
+    COLOR_RANGE: binding.AV_CODEC_CONFIG_COLOR_RANGE,
+    COLOR_SPACE: binding.AV_CODEC_CONFIG_COLOR_SPACE
   }
 }
 
@@ -190,3 +243,7 @@ exports.toChannelLayout = function toChannelLayout(layout) {
     `Channel layout must be a number or string. Received ${typeof layout} (${layout})`
   )
 }
+
+exports.getSampleFormatName = function (sampleFormat) {
+  return binding.getSampleFormatNameByID(sampleFormat)
+}

package/lib/filter-context.js

@@ -0,0 +1,7 @@
+const binding = require('../binding')
+
+module.exports = class FilterContext {
+  constructor() {
+    this._handle = binding.initFilterContext()
+  }
+}

package/lib/filter-graph.js

@@ -0,0 +1,54 @@
+const binding = require('../binding')
+
+module.exports = class FilterGraph {
+  constructor() {
+    this._handle = binding.initFilterGraph()
+  }
+
+  destroy() {
+    binding.destroyFilterGraph(this._handle)
+    this._handle = null
+  }
+
+  createFilter(context, filter, name, args) {
+    const argumentsString =
+      args &&
+      `video_size=${args.width}x${args.height}` +
+        `:pix_fmt=${args.pixelFormat}` +
+        `:time_base=${args.timeBase.numerator}/${args.timeBase.denominator}` +
+        `:pixel_aspect=${args.aspectRatio.numerator}/${args.aspectRatio.denominator}`
+
+    return binding.createFilterGraphFilter(
+      this._handle,
+      context._handle,
+      filter._handle,
+      name,
+      argumentsString
+    )
+  }
+
+  parse(filterDescription, inputs, outputs) {
+    binding.parseFilterGraph(
+      this._handle,
+      inputs._handle,
+      outputs._handle,
+      filterDescription
+    )
+  }
+
+  configure() {
+    binding.configureFilterGraph(this._handle)
+  }
+
+  pushFrame(ctx, frame) {
+    return binding.pushFilterGraphFrame(ctx._handle, frame._handle)
+  }
+
+  pullFrame(ctx, frame) {
+    return binding.pullFilterGraphFrame(ctx._handle, frame._handle)
+  }
+
+  [Symbol.dispose]() {
+    this.destroy()
+  }
+}

package/lib/filter-inout.js

@@ -0,0 +1,55 @@
+const binding = require('../binding')
+
+module.exports = class FilterInOut {
+  constructor(handle = binding.initFilterInout()) {
+    this._handle = handle
+    this._next = null
+    this._filterContext = null
+  }
+
+  destroy() {
+    binding.destroyFilterInOut(this._handle)
+    this._handle = null
+  }
+
+  get name() {
+    return binding.getFilterInOutName(this._handle)
+  }
+
+  set name(value) {
+    return binding.setFilterInOutName(this._handle, value)
+  }
+
+  get filterContext() {
+    return this._filterContext
+  }
+
+  set filterContext(value) {
+    this._filterContext = value
+    return binding.setFilterInOutFilterContext(
+      this._handle,
+      this._filterContext._handle
+    )
+  }
+
+  get padIdx() {
+    return binding.getFilterInOutPadIdx(this._handle)
+  }
+
+  set padIdx(value) {
+    return binding.setFilterInOutPadIdx(this._handle, value)
+  }
+
+  get next() {
+    return this._next
+  }
+
+  set next(value) {
+    this._next = value
+    return binding.setFilterInOutNext(this._handle, this._next._handle)
+  }
+
+  [Symbol.dispose]() {
+    this.destroy()
+  }
+}

package/lib/filter.js

@@ -0,0 +1,7 @@
+const binding = require('../binding')
+
+module.exports = class Filter {
+  constructor(name) {
+    this._handle = binding.getFilterByName(name)
+  }
+}

package/lib/packet.js

@@ -1,6 +1,42 @@
 const binding = require('../binding')
 const Rational = require('./rational')
 
+class SideData {
+  constructor(handle, opts = {}) {
+    this._handle = handle
+    this._type = opts.type || null
+    this._data = opts.data || null
+  }
+
+  static fromData(data, type) {
+    return new SideData(null, { data, type })
+  }
+
+  get type() {
+    if (this._type) return this._type
+    return binding.getSideDataType(this._handle)
+  }
+
+  get name() {
+    if (!this._handle) return null
+    return binding.getSideDataName(this._handle)
+  }
+
+  get data() {
+    if (this._data) return this._data
+    return Buffer.from(binding.getSideDataBuffer(this._handle))
+  }
+
+  [Symbol.for('bare.inspect')]() {
+    return {
+      __proto__: { constructor: SideData },
+      type: this.type,
+      name: this.name,
+      data: this.data
+    }
+  }
+}
+
 module.exports = class FFmpegPacket {
   constructor(buffer) {
     if (buffer) {
@@ -44,10 +80,31 @@ module.exports = class FFmpegPacket {
     )
   }
 
+  get sideData() {
+    const handles = binding.getPacketSideData(this._handle)
+    return handles.map((handle) => new SideData(handle))
+  }
+
+  set sideData(values) {
+    binding.setPacketSideData(
+      this._handle,
+      values.map((value) => ({
+        buffer: value.data.buffer,
+        offset: value.data.byteOffset,
+        length: value.data.byteLength,
+        type: value.type
+      }))
+    )
+  }
+
   get isKeyframe() {
     return binding.isPacketKeyframe(this._handle)
   }
 
+  set isKeyframe(value) {
+    binding.setPacketIsKeyFrame(this._handle, value)
+  }
+
   get dts() {
     return binding.getPacketDTS(this._handle)
   }
@@ -81,6 +138,10 @@ module.exports = class FFmpegPacket {
     )
   }
 
+  copyPropsFrom(sourcePacket) {
+    binding.copyPacketProps(this._handle, sourcePacket._handle)
+  }
+
   get duration() {
     return binding.getPacketDuration(this._handle)
   }
@@ -107,7 +168,8 @@ module.exports = class FFmpegPacket {
       dts: this.dts,
       pts: this.pts,
       duration: this.duration,
-      data: this.data
+      data: this.data,
+      sideData: this.sideData
     }
   }
 
@@ -115,3 +177,5 @@ module.exports = class FFmpegPacket {
     this.destroy()
   }
 }
+
+module.exports.SideData = SideData

package/lib/rational.js

@@ -26,15 +26,27 @@ module.exports = class FFmpegRational {
     return this.numerator / this.denominator
   }
 
+  equals(other) {
+    return (
+      this.numerator === other.numerator &&
+      this.denominator === other.denominator
+    )
+  }
+
   static from(num) {
     const view = new Int32Array(binding.rationalD2Q(num))
     return new FFmpegRational(view[0], view[1])
   }
 
   static rescaleQ(n, src, dst) {
-    return (
-      (n * (src.numerator / src.denominator)) /
-      (dst.numerator / dst.denominator)
+    if (src.equals(dst)) return n
+
+    return binding.rationalRescaleQ(
+      n,
+      src.numerator,
+      src.denominator,
+      dst.numerator,
+      dst.denominator
     )
   }
 }

package/lib/samples.js

@@ -1,49 +1,64 @@
 const binding = require('../binding')
-const constants = require('./constants.js')
 
 module.exports = class FFmpegSamples {
-  constructor(sampleFormat, nbChannels, nbSamples, align = 0) {
-    sampleFormat = constants.toSampleFormat(sampleFormat)
+  _data
+  _frame
 
-    this._sampleFormat = sampleFormat
-    this._nbChannels = nbChannels
-    this._nbSamples = nbSamples
-    this._align = align
+  constructor({ noAlignment = false } = {}) {
+    this._noAlignment = noAlignment
+  }
 
-    this._data = Buffer.from(
-      binding.initSamples(sampleFormat, nbChannels, nbSamples, align)
-    )
+  get data() {
+    return this._data
   }
 
-  get sampleFormat() {
-    return this._sampleFormat
+  get format() {
+    return this._frame.format
   }
 
-  get nbChannels() {
-    return this._nbChannels
+  get channelLayout() {
+    return this._frame.channelLayout
   }
 
   get nbSamples() {
-    return this._nbSamples
+    return this._frame.nbSamples
   }
 
-  get align() {
-    return this._align
+  get nbChannels() {
+    return this.channelLayout && this.channelLayout.nbChannels
   }
 
-  get data() {
-    return this._data
+  get pts() {
+    return this._frame.pts
   }
 
   fill(frame) {
-    binding.fillSamples(
-      this._sampleFormat,
-      this._nbChannels,
-      this._nbSamples,
-      this._align,
+    const { format, channelLayout, nbSamples } = frame
+
+    const len = FFmpegSamples.bufferSize(
+      format,
+      channelLayout.nbChannels,
+      nbSamples,
+      this._noAlignment
+    )
+
+    if (!this._data || len !== this._data.length) {
+      this._data = Buffer.allocUnsafe(len)
+    }
+
+    const res = binding.fillSamples(
+      frame._handle,
       this._data.buffer,
       this._data.byteOffset,
-      frame._handle
+      this._noAlignment
     )
+
+    this._frame = frame
+
+    return res
+  }
+
+  static bufferSize(format, nbChannels, nbSamples, noAlignment = false) {
+    return binding.samplesBufferSize(format, nbChannels, nbSamples, noAlignment)
   }
 }

package/lib/stream.js

@@ -66,4 +66,15 @@ module.exports = class FFmpegStream {
     this._codecParameters.toContext(context)
     return context.open()
   }
+
+  [Symbol.for('bare.inspect')]() {
+    return {
+      __proto__: { constructor: FFmpegStream },
+      id: this.id,
+      index: this.index,
+      timeBase: this.timeBase,
+      avgFramerate: this.avgFramerate,
+      codecParameters: this.codecParameters
+    }
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bare-ffmpeg",
-  "version": "1.0.0-25",
+  "version": "1.0.0-27",
   "description": "Low-level FFmpeg bindings for Bare",
   "exports": {
     ".": "./index.js",
@@ -18,7 +18,8 @@
   "addon": true,
   "scripts": {
     "test": "prettier . --check && bare test.js",
-    "format": "prettier . --write && clang-format -i binding.cc"
+    "format": "prettier . --write && clang-format -i binding.cc",
+    "example": "bare ./examples/index.js"
   },
   "repository": {
     "type": "git",
@@ -31,6 +32,7 @@
   },
   "homepage": "https://github.com/holepunchto/bare-ffmpeg#readme",
   "devDependencies": {
+    "bare-assert": "^1.1.0",
     "brittle": "^3.4.0",
     "cmake-bare": "^1.1.2",
     "cmake-harden": "^1.0.2",