segmenter 0.7.0

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Emil Palm
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Manifest

@@ -0,0 +1,12 @@
+COPYING
+ext/extconf.rb
+ext/Makefile
+ext/segmenter.bundle
+ext/segmenter.c
+ext/segmenter.o
+ext/test.rb
+NEWS
+Rakefile
+README.rdoc
+lib/segmenter.rb
+

data/README.rdoc

@@ -0,0 +1,19 @@
+= segmenter
+
+Description goes here.
+
+== Contributing to segmenter
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it
+* Fork the project
+* Start a feature/bugfix branch
+* Commit and push until you are happy with your contribution
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+== Copyright
+
+Copyright (c) 2011 Emil Palm. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,37 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "segmenter"
+  gem.homepage = "http://github.com/mrevilme/segmenter"
+  gem.license = "MIT"
+  gem.summary = %Q{}
+  gem.description = %Q{}
+  gem.email = "emil@x86.nu"
+  gem.authors = ["Emil Palm"]
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  #  gem.add_runtime_dependency 'jabber4r', '> 0.1'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+  gem.add_development_dependency "shoulda", ">= 0"
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  #  gem.add_runtime_dependency 'jabber4r', '> 0.1'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+  gem.add_development_dependency "bundler", "~> 1.0.0"
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  #  gem.add_runtime_dependency 'jabber4r', '> 0.1'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+  gem.add_development_dependency "jeweler", "~> 1.6.3"
+  # Include your dependencies below. Runtime dependencies are required when using your gem,
+  # and development dependencies are only needed for development (ie running rake tasks, tests, etc)
+  #  gem.add_runtime_dependency 'jabber4r', '> 0.1'
+  #  gem.add_development_dependency 'rspec', '> 1.2.3'
+  gem.add_development_dependency "rcov", ">= 0"
+end
+Jeweler::RubygemsDotOrgTasks.new

data/VERSION

@@ -0,0 +1 @@
+0.7.0

data/ext/Makefile

@@ -0,0 +1,187 @@
+
+SHELL = /bin/sh
+
+#### Start of system configuration section. ####
+
+srcdir = .
+topdir = /usr/local/rvm/rubies/ruby-1.9.2-p180/include/ruby-1.9.1
+hdrdir = /usr/local/rvm/rubies/ruby-1.9.2-p180/include/ruby-1.9.1
+arch_hdrdir = /usr/local/rvm/rubies/ruby-1.9.2-p180/include/ruby-1.9.1/$(arch)
+VPATH = $(srcdir):$(arch_hdrdir)/ruby:$(hdrdir)/ruby
+prefix = $(DESTDIR)/usr/local/rvm/rubies/ruby-1.9.2-p180
+rubylibprefix = $(libdir)/$(RUBY_BASE_NAME)
+exec_prefix = $(prefix)
+vendorhdrdir = $(rubyhdrdir)/vendor_ruby
+sitehdrdir = $(rubyhdrdir)/site_ruby
+rubyhdrdir = $(includedir)/$(RUBY_BASE_NAME)-$(ruby_version)
+vendordir = $(rubylibprefix)/vendor_ruby
+sitedir = $(rubylibprefix)/site_ruby
+ridir = $(datarootdir)/$(RI_BASE_NAME)
+mandir = $(datarootdir)/man
+localedir = $(datarootdir)/locale
+libdir = $(exec_prefix)/lib
+psdir = $(docdir)
+pdfdir = $(docdir)
+dvidir = $(docdir)
+htmldir = $(docdir)
+infodir = $(datarootdir)/info
+docdir = $(datarootdir)/doc/$(PACKAGE)
+oldincludedir = $(DESTDIR)/usr/include
+includedir = $(prefix)/include
+localstatedir = $(prefix)/var
+sharedstatedir = $(prefix)/com
+sysconfdir = $(prefix)/etc
+datadir = $(datarootdir)
+datarootdir = $(prefix)/share
+libexecdir = $(exec_prefix)/libexec
+sbindir = $(exec_prefix)/sbin
+bindir = $(exec_prefix)/bin
+rubylibdir = $(rubylibprefix)/$(ruby_version)
+archdir = $(rubylibdir)/$(arch)
+sitelibdir = $(sitedir)/$(ruby_version)
+sitearchdir = $(sitelibdir)/$(sitearch)
+vendorlibdir = $(vendordir)/$(ruby_version)
+vendorarchdir = $(vendorlibdir)/$(sitearch)
+
+CC = gcc
+CXX = g++
+LIBRUBY = $(LIBRUBY_SO)
+LIBRUBY_A = lib$(RUBY_SO_NAME)-static.a
+LIBRUBYARG_SHARED = -l$(RUBY_SO_NAME)
+LIBRUBYARG_STATIC = -l$(RUBY_SO_NAME)-static
+OUTFLAG = -o 
+COUTFLAG = -o 
+
+RUBY_EXTCONF_H = 
+cflags   =  $(optflags) $(debugflags) $(warnflags)
+optflags = -O3
+debugflags = -ggdb
+warnflags = -Wextra -Wno-unused-parameter -Wno-parentheses -Wpointer-arith -Wwrite-strings -Wno-missing-field-initializers -Wshorten-64-to-32 -Wno-long-long
+CFLAGS   = -fno-common $(cflags)  -fno-common -pipe -I/opt/local/include -W -Wall 
+INCFLAGS = -I. -I$(arch_hdrdir) -I$(hdrdir)/ruby/backward -I$(hdrdir) -I$(srcdir)
+DEFS     = 
+CPPFLAGS =  -D_XOPEN_SOURCE -D_DARWIN_C_SOURCE $(DEFS) $(cppflags)
+CXXFLAGS = $(CFLAGS) $(cxxflags)
+ldflags  = -L. -L/opt/local/lib -L/usr/local/lib -lavfilter
+dldflags = -Wl,-undefined,dynamic_lookup -Wl,-multiply_defined,suppress -Wl,-flat_namespace
+ARCH_FLAG = 
+DLDFLAGS = $(ldflags) $(dldflags)
+LDSHARED = $(CC) -dynamic -bundle
+LDSHAREDXX = $(CXX) -dynamic -bundle
+AR = ar
+EXEEXT = 
+
+RUBY_BASE_NAME = ruby
+RUBY_INSTALL_NAME = ruby
+RUBY_SO_NAME = ruby.1.9.1
+arch = x86_64-darwin10.7.0
+sitearch = $(arch)
+ruby_version = 1.9.1
+ruby = /usr/local/rvm/rubies/ruby-1.9.2-p180/bin/ruby
+RUBY = $(ruby)
+RM = rm -f
+RM_RF = $(RUBY) -run -e rm -- -rf
+RMDIRS = $(RUBY) -run -e rmdir -- -p
+MAKEDIRS = mkdir -p
+INSTALL = /usr/bin/install -c
+INSTALL_PROG = $(INSTALL) -m 0755
+INSTALL_DATA = $(INSTALL) -m 644
+COPY = cp
+
+#### End of system configuration section. ####
+
+preload = 
+
+libpath = . $(libdir)
+LIBPATH =  -L. -L$(libdir)
+DEFFILE = 
+
+CLEANFILES = mkmf.log
+DISTCLEANFILES = 
+DISTCLEANDIRS = 
+
+extout = 
+extout_prefix = 
+target_prefix = 
+LOCAL_LIBS = 
+LIBS = $(LIBRUBYARG_SHARED) -lavformat  -lpthread -ldl -lobjc 
+SRCS = segmenter.c
+OBJS = segmenter.o
+TARGET = segmenter_ext
+DLLIB = $(TARGET).bundle
+EXTSTATIC = 
+STATIC_LIB = 
+
+BINDIR        = $(bindir)
+RUBYCOMMONDIR = $(sitedir)$(target_prefix)
+RUBYLIBDIR    = $(sitelibdir)$(target_prefix)
+RUBYARCHDIR   = $(sitearchdir)$(target_prefix)
+HDRDIR        = $(rubyhdrdir)/ruby$(target_prefix)
+ARCHHDRDIR    = $(rubyhdrdir)/$(arch)/ruby$(target_prefix)
+
+TARGET_SO     = $(DLLIB)
+CLEANLIBS     = $(TARGET).bundle 
+CLEANOBJS     = *.o  *.bak
+
+all:    $(DLLIB)
+static: $(STATIC_LIB)
+.PHONY: all install static install-so install-rb
+.PHONY: clean clean-so clean-rb
+
+clean-rb-default::
+clean-rb::
+clean-so::
+clean: clean-so clean-rb-default clean-rb
+		@-$(RM) $(CLEANLIBS) $(CLEANOBJS) $(CLEANFILES)
+
+distclean-rb-default::
+distclean-rb::
+distclean-so::
+distclean: clean distclean-so distclean-rb-default distclean-rb
+		@-$(RM) Makefile $(RUBY_EXTCONF_H) conftest.* mkmf.log
+		@-$(RM) core ruby$(EXEEXT) *~ $(DISTCLEANFILES)
+		@-$(RMDIRS) $(DISTCLEANDIRS)
+
+realclean: distclean
+install: install-so install-rb
+
+install-so: $(RUBYARCHDIR)
+install-so: $(RUBYARCHDIR)/$(DLLIB)
+$(RUBYARCHDIR)/$(DLLIB): $(DLLIB)
+	@-$(MAKEDIRS) $(@D)
+	$(INSTALL_PROG) $(DLLIB) $(@D)
+install-rb: pre-install-rb install-rb-default
+install-rb-default: pre-install-rb-default
+pre-install-rb: Makefile
+pre-install-rb-default: Makefile
+$(RUBYARCHDIR):
+	$(MAKEDIRS) $@
+
+site-install: site-install-so site-install-rb
+site-install-so: install-so
+site-install-rb: install-rb
+
+.SUFFIXES: .c .m .cc .cxx .cpp .C .o
+
+.cc.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.cxx.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.cpp.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.C.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.c.o:
+	$(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -c $<
+
+$(DLLIB): $(OBJS) Makefile
+	@-$(RM) $(@)
+	$(LDSHARED) -o $@ $(OBJS) $(LIBPATH) $(DLDFLAGS) $(LOCAL_LIBS) $(LIBS)
+
+
+
+$(OBJS): $(hdrdir)/ruby.h $(hdrdir)/ruby/defines.h $(arch_hdrdir)/ruby/config.h

data/ext/extconf.rb

@@ -0,0 +1,24 @@
+require 'mkmf'
+
+
+if RUBY_PLATFORM == "universal-darwin10.0"
+  # TODO Set the archflags to -arch x86_64 ONLY if it's a 64-bit snow leopard machine.
+   #ENV['ARCHFLAGS'] = "-arch i386 -arch x86_64"
+   $CFLAGS.sub!("-arch x86_64", "")
+end
+
+if find_executable('pkg-config')
+$CFLAGS << ' ' + `pkg-config libavfilter --cflags`.strip
+$LDFLAGS << ' ' + `pkg-config libavfilter --libs`.strip
+end
+
+$CFLAGS << " -W -Wall"
+
+if 
+    have_library("avformat") and find_header('libavformat/avformat.h') and find_header('libavutil/log.h')  then
+    $objs = %w(segmenter.o)
+    create_makefile("segmenter_ext")
+else
+    STDERR.puts "missing library"
+    exit 1
+end

data/ext/segmenter.c

@@ -0,0 +1,465 @@
+//
+//  iosegmenter.c
+//  iosegmenter
+//
+//  Created by Emil Palm on 4/5/11.
+//  Copyright 2011 none. All rights reserved.
+//
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdbool.h>
+
+#include "ruby.h"
+#include "libavformat/avformat.h"
+#include "libavutil/log.h"
+
+
+char * dirname2(const char *path) {
+    static char dname[PATH_MAX];
+    size_t len;
+    const char *endp;
+	
+    /* Empty or NULL string gets treated as "." */
+    if (path == NULL || *path == '\0') {
+        dname[0] = '.';
+        dname[1] = '\0';
+        return (dname);
+    }
+	
+    /* Strip any trailing slashes */
+    endp = path + strlen(path) - 1;
+    while (endp > path && *endp == '/')
+        endp--;
+	
+    /* Find the start of the dir */
+    while (endp > path && *endp != '/')
+        endp--;
+	
+    /* Either the dir is "/" or there are no slashes */
+    if (endp == path) {
+        dname[0] = *endp == '/' ? '/' : '.';
+        dname[1] = '\0';
+        return (dname);
+    } else {
+        /* Move forward past the separating slashes */
+        do {
+            endp--;
+        } while (endp > path && *endp == '/');
+    }
+	
+    len = endp - path + 1;
+    if (len >= sizeof(dname)) {
+        errno = ENAMETOOLONG;
+        return (NULL);
+    }
+    memcpy(dname, path, len);
+    dname[len] = '\0';
+    return (dname);
+}
+
+
+typedef struct _segment {
+    int index;
+    long duration;
+    char *filename;
+} Segment;
+
+static VALUE rb_mAvSegmenter;
+static VALUE rb_cAvSegmenter;
+static VALUE rb_cAvSegment;
+
+/* Segment */
+
+static void segment_free(Segment *segment)
+{
+    if (NULL == segment)
+        return;
+    
+    segment->index = 0;
+    segment->duration = 0;
+    segment->filename = NULL;
+    ruby_xfree(segment);
+}
+
+struct config_info
+{
+    const char *input_filename;
+    int segment_length;
+    const char *temp_directory;
+    const char *filename_prefix;
+    const char *encoding_profile;
+};
+
+static AVStream *add_output_stream(AVFormatContext *output_format_context, AVStream *input_stream) 
+{
+    AVCodecContext *input_codec_context;
+    AVCodecContext *output_codec_context;
+    AVStream *output_stream;
+    
+    output_stream = av_new_stream(output_format_context, 0);
+    if (!output_stream) 
+    {
+        fprintf(stderr, "Segmenter error: Could not allocate stream\n");
+        exit(1);
+    }
+    
+    input_codec_context = input_stream->codec;
+    output_codec_context = output_stream->codec;
+    
+    output_codec_context->codec_id = input_codec_context->codec_id;
+    output_codec_context->codec_type = input_codec_context->codec_type;
+    output_codec_context->codec_tag = input_codec_context->codec_tag;
+    output_codec_context->bit_rate = input_codec_context->bit_rate;
+    output_codec_context->extradata = input_codec_context->extradata;
+    output_codec_context->extradata_size = input_codec_context->extradata_size;
+    
+    if(av_q2d(input_codec_context->time_base) * input_codec_context->ticks_per_frame > av_q2d(input_stream->time_base) && av_q2d(input_stream->time_base) < 1.0/1000) 
+    {
+        output_codec_context->time_base = input_codec_context->time_base;
+        output_codec_context->time_base.num *= input_codec_context->ticks_per_frame;
+    }
+    else 
+    {
+        output_codec_context->time_base = input_stream->time_base;
+    }
+    
+    switch (input_codec_context->codec_type) 
+    {
+        case CODEC_TYPE_AUDIO:
+            output_codec_context->channel_layout = input_codec_context->channel_layout;
+            output_codec_context->sample_rate = input_codec_context->sample_rate;
+            output_codec_context->channels = input_codec_context->channels;
+            output_codec_context->frame_size = input_codec_context->frame_size;
+            if ((input_codec_context->block_align == 1 && input_codec_context->codec_id == CODEC_ID_MP3) || input_codec_context->codec_id == CODEC_ID_AC3) 
+            {
+                output_codec_context->block_align = 0;
+            }
+            else 
+            {
+                output_codec_context->block_align = input_codec_context->block_align;
+            }
+            break;
+        case CODEC_TYPE_VIDEO:
+            output_codec_context->pix_fmt = input_codec_context->pix_fmt;
+            output_codec_context->width = input_codec_context->width;
+            output_codec_context->height = input_codec_context->height;
+            output_codec_context->has_b_frames = input_codec_context->has_b_frames;
+            
+            if (output_format_context->oformat->flags & AVFMT_GLOBALHEADER) 
+            {
+                output_codec_context->flags |= CODEC_FLAG_GLOBAL_HEADER;
+            }
+            break;
+        default:
+            break;
+    }
+    
+    return output_stream;
+}
+
+void output_transfer_command(const unsigned int first_segment, const unsigned int last_segment, const int end, const char *encoding_profile)
+{
+    char buffer[1024 * 10];
+    memset(buffer, 0, sizeof(char) * 1024 * 10);
+    
+    sprintf(buffer, "%d, %d, %d, %s", first_segment, last_segment, end, encoding_profile);
+    
+    fprintf(stderr, "segmenter: %s\n\r", buffer);
+}
+
+static VALUE segmenter_segment(VALUE klass, VALUE input_, VALUE output_prefix_, VALUE duration_ ) {
+
+    const char *input;
+    const char *output_prefix;
+    int segment_duration;
+    long max_tsfiles = 0;
+    double prev_segment_time = 0;
+    unsigned int output_index = 1;
+    AVInputFormat *ifmt;
+    AVOutputFormat *ofmt;
+    AVFormatContext *ic = NULL;
+    AVFormatContext *oc;
+    AVStream *video_st;
+    AVStream *audio_st;
+    AVCodec *codec;
+    char *output_filename;
+    char *remove_filename;
+    int video_index;
+    int audio_index;
+    unsigned int first_segment = 1;
+    unsigned int last_segment = 0;
+    int decode_done;
+    int ret;
+    unsigned int i;
+    int remove_file;
+    
+	bool soundOnly = false;
+
+    VALUE sArray = rb_ary_new();
+    av_register_all();
+    av_log_set_level(AV_LOG_PANIC);
+
+    input = RSTRING_PTR(input_);
+    output_prefix = RSTRING_PTR(output_prefix_);
+    segment_duration = (FIX2INT(duration_));
+	
+    char *folder = dirname2(strdup(input));
+    
+    remove_filename = malloc(sizeof(char) * (strlen(output_prefix) + 15));
+    if (!remove_filename) {
+        rb_raise(rb_eNoMemError, "Could not allocate space for remove filenames");
+    }
+    
+    output_filename = malloc(sizeof(char) * (strlen(output_prefix) + strlen(folder) + 15));
+    if (!output_filename) {
+        rb_raise(rb_eNoMemError, "Could not allocate space for output filenames");
+    }
+    
+        
+    ifmt = av_find_input_format("mpegts");
+    if (!ifmt) {
+        rb_raise(rb_eException, "Could not find MPEG-TS demuxer");
+    }
+    
+    ret = av_open_input_file(&ic, input, ifmt, 0, NULL);
+    if (ret != 0) {
+        rb_raise(rb_eException, "Could not open input file, make sure it is an mpegts file: %d %s", ret, input);
+    }
+    
+    if (av_find_stream_info(ic) < 0) {
+        rb_raise(rb_eException, "Could not read stream information");
+    }
+    
+    ofmt = av_guess_format("mpegts", NULL, NULL);
+    if (!ofmt) {
+        rb_raise(rb_eException, "Could not find MPEG-TS muxer");
+    }
+    
+    oc = avformat_alloc_context();
+    if (!oc) {
+        rb_raise(rb_eException, "Could not allocated output context");
+    }
+    oc->oformat = ofmt;
+
+    ic->flags |= AVFMT_FLAG_IGNDTS;
+    
+    video_index = -1;
+    audio_index = -1;
+    
+    for (i = 0; i < ic->nb_streams && (video_index < 0 || audio_index < 0); i++) {
+        switch (ic->streams[i]->codec->codec_type) {
+            case CODEC_TYPE_VIDEO:
+                video_index = i;
+                ic->streams[i]->discard = AVDISCARD_NONE;
+                video_st = add_output_stream(oc, ic->streams[i]);
+                break;
+            case CODEC_TYPE_AUDIO:
+                audio_index = i;
+                ic->streams[i]->discard = AVDISCARD_NONE;
+                audio_st = add_output_stream(oc, ic->streams[i]);
+                break;
+            default:
+                ic->streams[i]->discard = AVDISCARD_ALL;
+                break;
+        }
+    }
+
+	if(!soundOnly) {
+		soundOnly = (video_st != NULL);
+	}
+
+    if (av_set_parameters(oc, NULL) < 0) {
+        rb_raise(rb_eException, "Invalid output format parameters");
+    }
+    
+    dump_format(oc, 0, output_prefix, 1);
+
+
+
+	if (!soundOnly) {
+		codec = avcodec_find_decoder(video_st->codec->codec_id);
+	    if (!codec) {
+	        rb_raise(rb_eException, "Could not find video decoder, key frames will not be honored");
+	    }
+		
+	    if (avcodec_open(video_st->codec, codec) < 0) {
+	        rb_raise(rb_eException, "Could not open video decoder, key frames will not be honored");
+	    }
+    
+    
+	    if (video_st->codec->ticks_per_frame > 1) {
+	        // h264 sets the ticks_per_frame and time_base.den but not time_base.num
+	        // since we don't use ticks_per_frame, adjust time_base.num accordingly.
+	        video_st->codec->time_base.num *= video_st->codec->ticks_per_frame;
+	    }
+	}
+    
+    snprintf(output_filename, strlen(output_prefix) + strlen(folder) + 15, "%s/%s-%u.ts", folder, output_prefix, output_index++);
+    if (url_fopen(&oc->pb, output_filename, URL_WRONLY) < 0) {
+        rb_raise(rb_eException, "Could not open '%s'", output_filename);
+    }
+    
+    if (av_write_header(oc)) {
+        rb_raise(rb_eException, "Could not write mpegts header to first output file");
+    }
+    
+    //write_index = !write_index_file(index, tmp_index, segment_duration, output_prefix, http_prefix, first_segment, last_segment, 0, max_tsfiles);
+    int64_t initial_audio_pts = -1;
+    int64_t initial_video_pts = -1;
+	double segment_time;
+    do {
+		
+        AVPacket packet;
+        //av_init_packet(&packet);
+        decode_done = av_read_frame(ic, &packet);
+        if (decode_done < 0) {
+            break;
+        }
+        
+        if (av_dup_packet(&packet) < 0) {
+            rb_raise(rb_eException, "Could not duplicate packet");
+            av_free_packet(&packet);
+            break;
+        }
+		
+		if (!soundOnly) {
+	        if (packet.stream_index == video_index) {
+	            if (initial_video_pts < 0) 
+					initial_video_pts = packet.pts;
+					
+	            packet.pts -= initial_video_pts;
+	            packet.dts = packet.pts;
+	            if (packet.flags & AV_PKT_FLAG_KEY) {
+	                segment_time = (double)packet.pts * video_st->time_base.num / video_st->time_base.den;
+	            } else {
+	                segment_time = prev_segment_time;            }
+	        } else if (packet.stream_index == audio_index) {
+	            if (initial_audio_pts < 0) initial_audio_pts = packet.pts;
+	            packet.pts -= initial_audio_pts;
+	            packet.dts = packet.pts;
+	            segment_time = prev_segment_time;
+	        } else {
+	            segment_time = prev_segment_time;
+	            segment_time = prev_segment_time;
+	        }			
+		} else {
+
+			if (packet.stream_index == audio_index) {
+				
+	            if (initial_audio_pts < 0) 
+					initial_audio_pts = packet.pts;
+				
+				if (packet.flags & AV_PKT_FLAG_KEY) {
+					segment_time = (double)packet.pts * audio_st->time_base.num / audio_st->time_base.den;
+				} else {
+	            	segment_time = prev_segment_time;	
+				}
+					
+	            packet.pts -= initial_audio_pts;
+	            packet.dts = packet.pts;
+
+	        } else {
+	            segment_time = prev_segment_time;
+	            segment_time = prev_segment_time;
+	        }
+		
+		}
+
+
+
+        if (segment_time - prev_segment_time >= segment_duration) {
+            put_flush_packet(oc->pb);
+            url_fclose(oc->pb);
+            
+            if (max_tsfiles && (int)(last_segment - first_segment) >= max_tsfiles - 1) {
+                remove_file = 1;
+                first_segment++;
+            }
+            else {
+                remove_file = 0;
+            }
+            
+            
+            // Create Segment object
+            VALUE seg = rb_obj_alloc(rb_cAvSegment);            
+
+            rb_obj_call_init(seg, 0, 0);
+            rb_iv_set(seg, "@index", INT2FIX(++last_segment));
+            rb_iv_set(seg, "@duration",INT2FIX((int)floor((segment_time - prev_segment_time))));
+            rb_iv_set(seg, "@filename", rb_str_new2(output_filename));
+            
+            rb_ary_push(sArray, seg);
+            
+            if (remove_file) {
+                snprintf(remove_filename, strlen(output_prefix) + strlen(folder) + 15, "%s/%s-%u.ts", folder, output_prefix, first_segment - 1);
+                //snprintf(remove_filename, strlen(output_prefix) + 15, "%s-%u.ts", output_prefix, first_segment - 1);
+                remove(remove_filename);
+            }
+            
+//            snprintf(output_filename, strlen(output_prefix) + 15, "%s-%u.ts", output_prefix, output_index++);
+            snprintf(output_filename, strlen(output_prefix) + strlen(folder) + 15, "%s/%s-%u.ts", folder, output_prefix, output_index++);
+            if (url_fopen(&oc->pb, output_filename, URL_WRONLY) < 0) {
+                fprintf(stderr, "Could not open '%s'\n", output_filename);
+                break;
+            }
+            
+            prev_segment_time = segment_time;
+        }
+
+        ret = av_interleaved_write_frame(oc, &packet);
+        if (ret < 0) {
+            fprintf(stderr, "Warning: Could not write frame of stream\n");
+        }
+        else if (ret > 0) {
+            fprintf(stderr, "End of stream requested\n");
+            av_free_packet(&packet);
+            break;
+        }
+        
+        av_free_packet(&packet);
+    } while (!decode_done);
+
+    av_write_trailer(oc);
+
+	if (!soundOnly)
+    	avcodec_close(video_st->codec);
+
+    for(i = 0; i < oc->nb_streams; i++) {		
+		if(&oc->streams[i]->codec != NULL)
+        	av_freep(&oc->streams[i]->codec);
+
+		if (&oc->streams[i] != NULL)
+        	av_freep(&oc->streams[i]);
+    }
+
+    url_fclose(oc->pb);
+    av_free(oc);
+    
+    if (max_tsfiles && (int)(last_segment - first_segment) >= max_tsfiles - 1) {
+        remove_file = 1;
+        first_segment++;
+    }
+    else {
+        remove_file = 0;
+    }
+ 
+    if (remove_file) {
+        snprintf(remove_filename, strlen(output_prefix) + strlen(folder) + 15, "%s/%s-%u.ts", folder, output_prefix, first_segment - 1);
+        remove(remove_filename);
+    }
+    return sArray;
+}
+
+void Init_segmenter_ext() {
+    rb_mAvSegmenter = rb_define_module("Segmenter");
+    rb_define_module_function(rb_mAvSegmenter, "segment", segmenter_segment, 3);
+    
+    rb_cAvSegment = rb_define_class_under(rb_mAvSegmenter, "Segment", rb_cObject);
+    rb_define_attr(rb_cAvSegment, "duration", 1, 1);
+    rb_define_attr(rb_cAvSegment, "index", 1, 1);
+    rb_define_attr(rb_cAvSegment, "filename", 1, 1);
+}                                       
+
+

data/ext/test.rb

@@ -0,0 +1,3 @@
+$:.unshift "#{File.dirname(__FILE__)}";require "segmenter_ext";a = Segmenter::segment("audio_only.ts","najs",3); a.each do |b| puts "#{b.index} - #{b.duration} - #{b.filename}\n";end
+#$:.unshift "#{File.dirname(__FILE__)}";require "segmenter_ext";a = Segmenter::segment("movie.ts","najs",3); a.each do |b| puts "#{b.index} - #{b.duration} - #{b.filename}\n";end
+#$:.unshift "#{File.dirname(__FILE__)}";require "segmenter_ext";a = Segmenter::segment("bajs/bajs.ts","09a7ca73-9ee6-4664-b73a-cc342c556fba",40);b = a[0]; puts "#{b.index} - #{b.duration} - #{b.filename}\n"

data/lib/segmenter.rb

@@ -0,0 +1,4 @@
+$:.unshift(File.dirname(__FILE__))
+
+require File.dirname(__FILE__) + '/../ext/segmenter_ext'
+

data/segmenter.gemspec

@@ -0,0 +1,63 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{segmenter}
+  s.version = "0.7.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Emil Palm"]
+  s.date = %q{2011-07-08}
+  s.description = %q{}
+  s.email = %q{emil@x86.nu}
+  s.extensions = ["ext/extconf.rb"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    "COPYING",
+    "LICENSE.txt",
+    "Manifest",
+    "NEWS",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "ext/Makefile",
+    "ext/extconf.rb",
+    "ext/segmenter.c",
+    "ext/segmenter.o",
+    "ext/test.rb",
+    "lib/segmenter.rb",
+    "segmenter.gemspec"
+  ]
+  s.homepage = %q{http://github.com/mrevilme/segmenter}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.6.2}
+  s.summary = %q{}
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.6.3"])
+      s.add_development_dependency(%q<rcov>, [">= 0"])
+    else
+      s.add_dependency(%q<shoulda>, [">= 0"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.6.3"])
+      s.add_dependency(%q<rcov>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<shoulda>, [">= 0"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.6.3"])
+    s.add_dependency(%q<rcov>, [">= 0"])
+  end
+end
+

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification 
+name: segmenter
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.7.0
+platform: ruby
+authors: 
+- Emil Palm
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-08 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.0.0
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: jeweler
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.6.3
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: rcov
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  version_requirements: *id004
+description: ""
+email: emil@x86.nu
+executables: []
+
+extensions: 
+- ext/extconf.rb
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- COPYING
+- LICENSE.txt
+- Manifest
+- NEWS
+- README.rdoc
+- Rakefile
+- VERSION
+- ext/Makefile
+- ext/extconf.rb
+- ext/segmenter.c
+- ext/segmenter.o
+- ext/test.rb
+- lib/segmenter.rb
+- segmenter.gemspec
+has_rdoc: true
+homepage: http://github.com/mrevilme/segmenter
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: ""
+test_files: []
+