tsparser 0.0.0

data/lib/epg.rb

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+module TSparser
+  class EPG
+    include Enumerable
+
+    def initialize(epg_hash=Hash.new)
+      @epg = epg_hash
+    end
+
+    def add(event_id, attr_hash)
+      @epg[event_id] = attr_hash
+    end
+
+    attr_reader :epg
+
+    def +(other)
+      new_hash = @epg.merge(other.epg) do |event_id, attr_hash1, attr_hash2|
+        attr_hash1.merge(attr_hash2) do |attr_name, val1, val2|
+          [val1, val2].max{|a, b| a.to_s.size <=> b.to_s.size}
+        end
+      end
+      return EPG.new(new_hash)
+    end
+
+    def each(&block)
+      @epg.each_value(&block)
+    end
+  end
+end

data/lib/parsing.rb

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+module TSparser
+  module Parsing
+    
+    def initialize(binary, parsing_proc=nil, delay=true)
+      @binary = binary
+      @parsed_variable = Hash.new
+      if delay
+        @parsing_proc = parsing_proc
+      else
+        self.instance_eval(&parsing_proc) if parsing_proc
+      end
+    end
+
+    def read(name, type, bit)
+      if type == Integer
+        val = @binary.read_bit_as_integer(bit)
+        @parsed_variable[name] = val
+      else
+        val = @binary.read_bit_as_binary(bit)
+        @parsed_variable[name] = Proc.new{ type.new(val) }
+      end
+    rescue => error
+      STDERR.puts "Parsing error occurred at class: #{self.class}, attr: #{name}(type:#{type}, bitlen:#{bit})"
+      raise error
+    end
+
+    def rest_all
+      return @binary.rest_readable_bit_length
+    end
+
+    def method_missing(name, *args)
+      if @parsing_proc
+        parsing_proc = @parsing_proc
+        @parsing_proc = nil
+        self.instance_eval(&parsing_proc)
+      end
+      if @parsed_variable[name]
+        if @parsed_variable[name].instance_of?(Proc)
+          @parsed_variable[name] = @parsed_variable[name].call
+        end
+        return @parsed_variable[name]
+      end
+      super
+    end
+
+    def self.included(klass)
+      klass.extend ClassExtension
+    end
+
+    module ClassExtension
+
+      def def_parsing(delay=true, &block)
+        @delay = delay
+        @parsing_definition = block
+      end
+
+      def new(binary)
+        super(binary, @parsing_definition, @delay)
+      end
+    end
+  end
+end

data/lib/psi_section_reader.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+module TSparser
+  class PSISectionReader
+
+    def initialize(pid, ts)
+      @ts = ts.filter(pid)
+    end
+
+    def read
+      cont_packets = []
+      while packet = @ts.read
+        if packet.payload_unit_start_indicator == 1
+          if @start_packet && continuity_check(@start_packet, *cont_packets, packet)
+            binary = @start_packet.payload.from(@start_packet.payload.b(0) + 1)
+            binary = binary.join(*cont_packets.map{|packet| packet.payload})
+            @start_packet = packet
+            return binary
+          else
+            @start_packet = packet
+            cont_packets = []
+            next
+          end
+        end
+        cont_packets << packet
+      end
+      return nil
+    end
+
+    private
+    
+    def continuity_check(first_packet, *packets)
+      counter = first_packet.continuity_counter
+      return packets.all? do |packet|
+        counter += 1
+        packet.continuity_counter == counter % 16
+      end
+    end
+  end
+end

data/lib/ts.rb

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+module TSparser
+  class TS
+
+    def initialize(io, filtering_procs=[])
+      @io = io
+      @filtering_procs = filtering_procs
+    end
+
+    def filter(*pids, &block)
+      new_filtering_procs = @filtering_procs
+      if pids.length > 0
+        new_filtering_procs = [Proc.new{|packet| pids.include?(packet.pid)}] + new_filtering_procs
+      end
+      if block
+        new_filtering_procs = new_filtering_procs + [block]
+      end
+      return TS.new(@io, new_filtering_procs)
+    end
+
+    def read
+      loop do
+        return nil if eof?
+        packet_binary = Binary.new(@io.read(188))
+        if packet_binary.length != 188
+          raise "Bytes less than 188byte (#{packet_bytes.length}byte) were read from TS file."
+        end
+        packet = TransportPacket.new(packet_binary)
+        return packet if @filtering_procs.all?{|filter| filter.call(packet)}
+      end
+    end
+
+    def eof?
+      return @io.eof?
+    end
+
+    def close
+      @io.close
+    end
+  end
+end

data/lib/tsparser.rb

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+module TSparser
+  extend self
+
+  LIBRARY_ROOT_PATH = File.expand_path(File.dirname(__FILE__))
+
+  MAIN_FILES =
+    ['/binary.rb',
+     '/parsing.rb',
+     '/ts.rb',
+     '/psi_section_reader.rb',
+     '/arib_string_decoder.rb',
+     '/epg.rb'
+    ]
+
+
+  DEFINITION_DESCRIPTOR_FILES =
+    ['/short_event_descriptor.rb',
+     '/extended_event_descriptor.rb',
+     '/component_descriptor.rb',
+     '/content_descriptor.rb',
+     '/digital_copy_control_descriptor.rb',
+     '/audio_component_descriptor.rb',
+     '/data_content_descriptor.rb',
+     '/ca_contract_information_descriptor.rb',
+     '/event_group_descriptor.rb',
+     '/unknown_descriptor.rb'
+    ]
+
+  DEFINITION_FILES =
+    ['/descriptor.rb',
+     '/descriptor_list.rb',
+     '/eit_event.rb',
+     '/event_information_section.rb',
+     '/event_list.rb',
+     '/transport_packet.rb',
+     '/arib_duration.rb',
+     '/arib_time.rb',
+     '/arib_string.rb'
+    ]
+
+  MAIN_FILES.each do |path|
+    require LIBRARY_ROOT_PATH + path
+  end
+
+  DEFINITION_DESCRIPTOR_FILES.each do |path|
+    require LIBRARY_ROOT_PATH + '/definition/descriptor' + path
+  end
+
+  DEFINITION_FILES.each do |path|
+    require LIBRARY_ROOT_PATH + '/definition' + path
+  end
+
+  def parse_epg(input)
+    epg = EPG.new
+    section_stream = PSISectionReader.new(0x12, open_ts(input))
+    while section_binary = section_stream.read
+      next unless EventInformationSection.section_length_enough?(section_binary)        
+      eis = EventInformationSection.new(section_binary)
+      epg = epg + eis.to_epg
+    end
+    return epg
+  end
+
+  def open_ts(input)
+    case input
+    when String
+      return TS.new(File.open(input, "rb"))
+    when IO
+      return TS.new(input)
+    else
+      raise "arugument should be TS file path(String) or IO"
+    end
+  end
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification
+name: tsparser
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- rokugatsu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-02 00:00:00.000000000 Z
+dependencies: []
+description: |2
+        This is a general-purpose MPEG-TS parser.
+email: sasasawada@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/arib_string_decoder.rb
+- lib/binary.rb
+- lib/definition/arib_duration.rb
+- lib/definition/arib_string.rb
+- lib/definition/arib_time.rb
+- lib/definition/descriptor.rb
+- lib/definition/descriptor/audio_component_descriptor.rb
+- lib/definition/descriptor/ca_contract_information_descriptor.rb
+- lib/definition/descriptor/component_descriptor.rb
+- lib/definition/descriptor/content_descriptor.rb
+- lib/definition/descriptor/data_content_descriptor.rb
+- lib/definition/descriptor/digital_copy_control_descriptor.rb
+- lib/definition/descriptor/event_group_descriptor.rb
+- lib/definition/descriptor/extended_event_descriptor.rb
+- lib/definition/descriptor/short_event_descriptor.rb
+- lib/definition/descriptor/unknown_descriptor.rb
+- lib/definition/descriptor_list.rb
+- lib/definition/eit_event.rb
+- lib/definition/event_information_section.rb
+- lib/definition/event_list.rb
+- lib/definition/transport_packet.rb
+- lib/epg.rb
+- lib/parsing.rb
+- lib/psi_section_reader.rb
+- lib/ts.rb
+- lib/tsparser.rb
+- LICENSE
+- README.rdoc
+homepage: http://github.com/rokugatsu/tsparser
+licenses:
+- MIT-LICENSE
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Library to read MPEG2-TS.Mainly, as japanese digital broadcasting in ARIB-format.
+test_files: []