bare-ffmpeg 1.0.0-2 → 1.0.0-3

package/binding.c

@@ -1504,6 +1504,35 @@ bare_ffmpeg_frame_set_pixel_format(js_env_t *env, js_callback_info_t *info) {
   return NULL;
 }
 
+static js_value_t *
+bare_ffmpeg_frame_get_linesize(js_env_t *env, js_callback_info_t *info) {
+  int err;
+
+  size_t argc = 2;
+  js_value_t *argv[2];
+
+  err = js_get_callback_info(env, info, &argc, argv, NULL, NULL);
+  assert(err == 0);
+  assert(argc == 2);
+
+  bare_ffmpeg_frame_t *frame;
+  err = js_get_arraybuffer_info(env, argv[0], (void **) &frame, NULL);
+  assert(err == 0);
+
+  uint32_t channel;
+  err = js_get_value_uint32(env, argv[1], &channel);
+  assert(err == 0);
+  assert(channel < AV_NUM_DATA_POINTERS);
+
+  int32_t linesize = frame->handle->linesize[channel];
+
+  js_value_t *result;
+  err = js_create_int32(env, linesize, &result);
+  assert(err == 0);
+
+  return result;
+}
+
 static js_value_t *
 bare_ffmpeg_frame_alloc(js_env_t *env, js_callback_info_t *info) {
   int err;
@@ -1535,6 +1564,49 @@ bare_ffmpeg_frame_alloc(js_env_t *env, js_callback_info_t *info) {
   return NULL;
 }
 
+static js_value_t *
+bare_ffmpeg_frame_get_data(js_env_t *env, js_callback_info_t *info) {
+  int err;
+
+  size_t argc = 1;
+  js_value_t *argv[1];
+
+  err = js_get_callback_info(env, info, &argc, argv, NULL, NULL);
+  assert(err == 0);
+  assert(argc == 1);
+
+  bare_ffmpeg_frame_t *frame;
+  err = js_get_arraybuffer_info(env, argv[0], (void **) &frame, NULL);
+  assert(err == 0);
+
+  int size = av_image_get_buffer_size(
+    frame->handle->format,
+    frame->handle->width,
+    frame->handle->height,
+    1
+  );
+  assert(size > 0);
+
+  js_value_t *result;
+  uint8_t *buffer;
+  err = js_create_arraybuffer(env, size, (void **) &buffer, &result);
+  assert(err == 0);
+
+  err = av_image_copy_to_buffer(
+    buffer,
+    size,
+    (const uint8_t *const *) frame->handle->data,
+    frame->handle->linesize,
+    frame->handle->format,
+    frame->handle->width,
+    frame->handle->height,
+    1
+  );
+  assert(err >= 0);
+
+  return result;
+}
+
 static js_value_t *
 bare_ffmpeg_image_init(js_env_t *env, js_callback_info_t *info) {
   int err;
@@ -2075,7 +2147,9 @@ bare_ffmpeg_exports(js_env_t *env, js_value_t *exports) {
   V("setFrameHeight", bare_ffmpeg_frame_set_height)
   V("getFramePixelFormat", bare_ffmpeg_frame_get_pixel_format)
   V("setFramePixelFormat", bare_ffmpeg_frame_set_pixel_format)
+  V("getFrameLineSize", bare_ffmpeg_frame_get_linesize)
   V("allocFrame", bare_ffmpeg_frame_alloc)
+  V("getFrameData", bare_ffmpeg_frame_get_data)
 
   V("initImage", bare_ffmpeg_image_init)
   V("fillImage", bare_ffmpeg_image_fill)
@@ -2110,6 +2184,7 @@ bare_ffmpeg_exports(js_env_t *env, js_value_t *exports) {
   V(AV_CODEC_ID_H264)
 
   V(AV_PIX_FMT_RGBA)
+  V(AV_PIX_FMT_RGB24)
   V(AV_PIX_FMT_YUVJ420P)
   V(AV_PIX_FMT_YUV420P)
 

package/lib/frame.js

@@ -45,7 +45,15 @@ module.exports = class FFmpegFrame extends ReferenceCounted {
     return binding.getFramePixelFormat(this._handle)
   }
 
+  lineSize(channel) {
+    return binding.getFrameLineSize(this._handle, channel)
+  }
+
   alloc() {
     binding.allocFrame(this._handle, 32)
   }
+
+  get data() {
+    return Buffer.from(binding.getFrameData(this._handle))
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bare-ffmpeg",
-  "version": "1.0.0-2",
+  "version": "1.0.0-3",
   "description": "Low-level FFmpeg bindings for Bare",
   "exports": {
     ".": "./index.js",