rmovie 0.5.0

data/ext/rmovie/rmovie_movie.c

@@ -0,0 +1,349 @@
+#include <ruby.h>
+
+#include "qp_movie.h"
+#include "rmovie_movie.h"
+
+extern VALUE rb_cFrame;
+
+extern VALUE wrap_frame_ctx(VALUE klass, qp_frame_context *frame);
+
+static void rm_mark_movie_ctx(qp_movie_context *movie) {}
+
+static void rm_free_movie_ctx(qp_movie_context *movie) {
+    qp_free_movie_ctx(movie, NULL);
+}
+
+VALUE movie_allocate(VALUE klass)
+{
+    qp_movie_context* movie = qp_alloc_movie_ctx(NULL);
+    return Data_Wrap_Struct (klass, rm_mark_movie_ctx, 
+            rm_free_movie_ctx, movie);
+}
+
+
+VALUE movie_initialize(VALUE self, VALUE filename)
+{
+    qp_movie_context *movie_ctx;
+
+    SafeStringValue(filename);
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    if (qp_open_movie_file(movie_ctx, StringValueCStr(filename))) {
+        rb_sys_fail((StringValueCStr(filename)));
+    }
+
+    return self;
+}
+
+
+VALUE Movie_get_duration(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_float_new(qp_get_duration(movie_ctx));
+}
+
+
+VALUE Movie_get_frame_count(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_frame_count(movie_ctx));
+}
+
+
+VALUE Movie_get_frame_rate(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_float_new(qp_get_frame_rate(movie_ctx));
+}
+
+
+VALUE Movie_get_file_name(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_file_name(movie_ctx));
+}
+
+
+VALUE Movie_get_frame_width(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_movie_width(movie_ctx));
+}
+
+
+VALUE Movie_get_frame_height(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_movie_height(movie_ctx));
+}
+
+
+VALUE Movie_get_comment(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_comment(movie_ctx));
+}
+
+
+VALUE Movie_get_title(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_title(movie_ctx));
+}
+
+
+// getartist is an alias of this
+VALUE Movie_get_author(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_author(movie_ctx));
+}
+
+
+VALUE Movie_get_copyright(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_copyright(movie_ctx));
+}
+
+
+VALUE Movie_get_album(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_album(movie_ctx));
+}
+
+
+VALUE Movie_get_genre(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_tainted_str_new2(qp_get_genre(movie_ctx));
+}
+
+
+VALUE Movie_get_year(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_year(movie_ctx));
+}
+
+
+VALUE Movie_get_track(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_track(movie_ctx));
+}
+
+
+VALUE Movie_get_frame_number(VALUE self) 
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_current_frame_number(movie_ctx));
+}
+
+
+VALUE Movie_get_pixel_format(VALUE self)
+{
+    const char *fmt;
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    fmt = qp_get_movie_pixel_format_name(movie_ctx);
+
+    if (fmt) {
+        return rb_tainted_str_new2((char *)fmt);
+    } else {
+        return Qfalse; // TODO: Should this be Qnil or Qundef instead?
+    }
+}
+
+
+VALUE Movie_get_bit_rate(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_bit_rate(movie_ctx));
+}
+
+
+VALUE Movie_has_audio(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return qp_has_audio(movie_ctx) ? Qtrue : Qfalse;
+}
+
+
+VALUE Movie_get_video_codec(VALUE self)
+{
+    char *codec_name;
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    codec_name = (char*)qp_get_codec_name(movie_ctx, CODEC_TYPE_VIDEO);
+
+    if (codec_name) {
+        return rb_tainted_str_new2((char *)codec_name);
+    } else {
+        return Qfalse; // TODO: Should this be Qnil or Qundef instead?
+    }
+}
+
+
+VALUE Movie_get_audio_codec(VALUE self)
+{
+    char *codec_name;
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    codec_name = (char*)qp_get_codec_name(movie_ctx, CODEC_TYPE_AUDIO);
+
+    if (codec_name) {
+        return rb_tainted_str_new2((char *)codec_name);
+    } else {
+        return Qfalse; // TODO: Should this be Qnil or Qundef instead?
+    }
+}
+
+
+VALUE Movie_get_num_audio_channels(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_int_new(qp_get_num_audio_channels(movie_ctx));
+}
+
+
+VALUE Movie_get_pixel_aspect_ratio(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    return rb_float_new(qp_get_pixel_aspect_ratio(movie_ctx));
+}
+
+
+VALUE Movie_get_frame(int argc, VALUE *argv, VALUE self)
+{
+    unsigned int wanted_frame = 0;
+    qp_movie_context *movie_ctx;
+    qp_frame_context *frame_ctx = NULL;
+    int err;
+
+    switch (argc) {
+        case 1:
+            wanted_frame = NUM2UINT(argv[0]);
+            if (wanted_frame < 1) {
+                rb_raise(rb_eArgError, 
+                        "frame number must be greater than zero");
+            }
+            break;
+        case 0:
+            wanted_frame = 0;
+            break;
+        default:
+            rb_raise(rb_eArgError, 
+                    "wrong number of arguments (%d for 0 or 1)", argc);
+            break;
+    }
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    if (qp_get_frame(movie_ctx, wanted_frame, &frame_ctx)) {
+        rb_warn("couldn't get frame %d", wanted_frame);
+    } else {
+        return wrap_frame_ctx(rb_cFrame, frame_ctx);
+    }
+    return Qnil;
+}
+
+
+VALUE Movie_get_next_key_frame(VALUE self)
+{
+    qp_movie_context *movie_ctx;
+    qp_frame_context *frame_ctx = NULL;
+ 
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+    
+    qp_get_frame(movie_ctx, QP_GETFRAME_KEYFRAME, &frame_ctx);
+
+    if (frame_ctx == NULL) {
+        // TODO: Exception or false?
+        rb_warn("couldn't get key frame");
+    } else {
+        return wrap_frame_ctx(rb_cFrame, frame_ctx);
+    }
+    return Qnil;
+}
+
+VALUE Movie_export(VALUE self, VALUE filename)
+{
+    qp_movie_context *movie_ctx;
+
+    SafeStringValue(filename);
+
+    Data_Get_Struct(self, qp_movie_context, movie_ctx);
+
+    if (qp_export_movie(movie_ctx, StringValueCStr(filename)) != 0) {
+        rb_warn("couldn't export movie to %s", filename);
+        return Qfalse;
+    }
+
+    return Qtrue;
+}

data/ext/rmovie/rmovie_movie.h

@@ -0,0 +1,28 @@
+VALUE movie_allocate(VALUE klass);
+VALUE movie_initialize(VALUE self, VALUE filename);
+VALUE Movie_get_duration(VALUE);
+VALUE Movie_get_frame_count(VALUE);
+VALUE Movie_get_frame_rate(VALUE);
+VALUE Movie_get_file_name(VALUE);
+VALUE Movie_get_comment(VALUE);
+VALUE Movie_get_title(VALUE);
+VALUE Movie_get_author(VALUE); // get_artist is alias of this
+VALUE Movie_get_artist(VALUE);
+VALUE Movie_get_copyright(VALUE);
+VALUE Movie_get_album(VALUE);
+VALUE Movie_get_genre(VALUE);
+VALUE Movie_get_year(VALUE);
+VALUE Movie_get_track(VALUE);
+VALUE Movie_get_frame_width(VALUE);
+VALUE Movie_get_frame_height(VALUE);
+VALUE Movie_get_frame_number(VALUE);
+VALUE Movie_get_pixel_format(VALUE);
+VALUE Movie_get_bit_rate(VALUE);
+VALUE Movie_has_audio(VALUE);
+VALUE Movie_get_next_key_frame(VALUE);
+VALUE Movie_get_video_codec(VALUE);
+VALUE Movie_get_audio_codec(VALUE);
+VALUE Movie_get_num_audio_channels(VALUE);
+VALUE Movie_get_pixel_aspect_ratio(VALUE);
+VALUE Movie_get_frame(int argc, VALUE *argv, VALUE self);
+VALUE Movie_export(VALUE self, VALUE filename);

data/lib/rmovie/version.rb

@@ -0,0 +1,45 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, Jamis Buck (jgb3@email.byu.edu)
+# All rights reserved.
+# 
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+# 
+#     * Redistributions of source code must retain the above copyright notice,
+#       this list of conditions and the following disclaimer.
+# 
+#     * Redistributions in binary form must reproduce the above copyright
+#       notice, this list of conditions and the following disclaimer in the
+#       documentation and/or other materials provided with the distribution.
+# 
+#     * The names of its contributors may not be used to endorse or promote
+#       products derived from this software without specific prior written
+#       permission.
+# 
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+# =============================================================================
+#++
+
+module RMovie
+
+  module Version
+
+    MAJOR = 0
+    MINOR = 5
+    TINY  = 0
+
+    STRING = [ MAJOR, MINOR, TINY ].join( "." )
+
+  end
+
+end

data/test/export_movie.rb

@@ -0,0 +1,7 @@
+#!/usr/bin/ruby
+
+require 'rmovie'
+
+mov = RMovie::Movie.new("/home/tkirby/test/video/vimeo.todd.289.mov")
+
+mov.export('out.mpg')

data/test/get_frame.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/ruby
+
+require 'rmovie'
+
+$script_dir = File.dirname(File.expand_path($0))
+$media_dir = "#{$script_dir}/media"
+
+mov = RMovie::Movie.new("#{$media_dir}/1701-D.mov")
+
+print mov.frame(1).inspect, "\n"

data/test/get_frames_reverse.rb

@@ -0,0 +1,11 @@
+#!/usr/bin/ruby
+
+require 'rmovie'
+
+$script_dir = File.dirname(File.expand_path($0))
+$media_dir = "#{$script_dir}/media"
+
+mov = RMovie::Movie.new("#{$media_dir}/1701-D.mov")
+
+mov.frame(6).inspect
+mov.frame(5).inspect