format_parser 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e49394510e191185e3be1f3560ca182a103b9446225a1163294f2db6a912ae76
-  data.tar.gz: 4f52d761857020a0957711528ad4d10ee885c74745897fa14a5e69292abf0cc2
+  metadata.gz: d8bdd6daa59e43bbe6033562b804d3b1b9685c6e2aa5ef3f8d527c74516e1f6c
+  data.tar.gz: d6035f0b3c819085ffd23c4988e48145976ba1f98547d16589b2a4b4aa3e7aa7
 SHA512:
-  metadata.gz: 829bd488d021cb0db5ce2b718aa661bdf3d4f86a55e55808c8279367e955d35b2bdccdd2bdf805ff736cef6d49f8c0f425c1ec65c0d7f07ada2db684581d4c1f
-  data.tar.gz: ed76f04c157a865de03df0d614d0d353b53e1e9c9b0a164ac6b2a6612a358643ae210b00fd92b91fec6cefd8c8c71e6ff559e63df3343b13ce161175425dbd3a
+  metadata.gz: 1a2f6243d295589972c63e45b719f19a5761c872103c32bfaacaef9cee9b85551c18ddaaf3c4095567c358958e7b7bf78800fc3e5a8e6c9f774eab383dc3160c
+  data.tar.gz: 73ba4f1099ccb4c490b50a8f531be843679159a6417980b2a61b905724905cf6b7e884127b66096e2237982dc6743e7f057bea0062472c1779e4d754dd5388ff

data/CHANGELOG.md

@@ -1,3 +1,6 @@
+## 1.2.0
+* Add support for `codecs` in moov_parser for video metadata
+
 ## 1.1.0
 * Add support for `frame_rate` in moov_parser
 

data/Gemfile

@@ -3,6 +3,7 @@ source 'https://rubygems.org'
 gem 'ruby-debug-ide'
 gem 'debase'
 gem 'solargraph', group: :development
+gem 'pry', group: :development
 
 # Gem dependencies specified in the gemspec
 gemspec

data/lib/format_parser/version.rb

@@ -1,3 +1,3 @@
 module FormatParser
-  VERSION = '1.1.0'
+  VERSION = '1.2.0'
 end

data/lib/parsers/moov_parser/decoder.rb

@@ -130,6 +130,23 @@ class FormatParser::MOOVParser::Decoder
     stts
   end
 
+  def parse_stsd_atom(io, _)
+    version = read_byte_value(io)
+    is_v1 = version == 1
+    stsd = {
+      version: version,
+      flags: read_bytes(io, 3),
+      number_of_entries: is_v1 ? read_64bit_uint(io) : read_32bit_uint(io),
+      codecs: []
+    }
+    stsd[:number_of_entries].times {
+      codec_length = read_32bit_uint(io)
+      stsd[:codecs] << read_bytes(io, 4)
+      io.seek(io.pos + codec_length - 8) # 8 bytes is the header length containing the codec length and the codec name that we just did read
+    }
+    stsd
+  end
+
   def parse_mdhd_atom(io, _)
     version = read_byte_value(io)
     is_v1 = version == 1

data/lib/parsers/moov_parser.rb

@@ -64,14 +64,14 @@ class FormatParser::MOOVParser
         intrinsics: atom_tree,
       )
     else
-      frame_rate = parse_sample_atom(decoder, atom_tree)&.truncate(2)
       FormatParser::Video.new(
         format: format_from_moov_type(file_type),
         width_px: width,
         height_px: height,
-        frame_rate: frame_rate,
+        frame_rate: parse_time_to_sample_atom(decoder, atom_tree)&.truncate(2),
         media_duration_seconds: media_duration_s,
         content_type: MP4_MIXED_MIME_TYPE,
+        codecs: parse_sample_description_atom(decoder, atom_tree),
         intrinsics: atom_tree
       )
     end
@@ -119,7 +119,7 @@ class FormatParser::MOOVParser
 
   # Sample information is found in the 'time-to-sample' stts atom.
   # The media atom mdhd is needed too in order to get the movie timescale
-  def parse_sample_atom(decoder, atom_tree)
+  def parse_time_to_sample_atom(decoder, atom_tree)
     video_trak_atom = decoder.find_video_trak_atom(atom_tree)
 
     stts = if video_trak_atom
@@ -147,5 +147,21 @@ class FormatParser::MOOVParser
     end
   end
 
+  def parse_sample_description_atom(decoder, atom_tree)
+    video_trak_atom = decoder.find_video_trak_atom(atom_tree)
+
+    stsd = if video_trak_atom
+      decoder.find_first_atom_by_path([video_trak_atom], 'trak', 'mdia', 'minf', 'stbl', 'stsd')
+    else
+      decoder.find_first_atom_by_path(atom_tree, 'moov', 'trak', 'mdia', 'minf', 'stbl', 'stsd')
+    end
+
+    if stsd
+      stsd.field_value(:codecs)
+    else
+      nil
+    end
+  end
+
   FormatParser.register_parser new, natures: :video, formats: FTYP_MAP.values, priority: 1
 end

data/lib/video.rb

@@ -13,6 +13,8 @@ module FormatParser
     # Type of the file (e.g :mp3)
     attr_accessor :format
 
+    attr_accessor :codecs
+
     # Duration of the media object (be it audio or video) in seconds,
     # as a Float
     attr_accessor :media_duration_seconds

data/spec/parsers/moov_parser_spec.rb

@@ -94,6 +94,7 @@ describe FormatParser::MOOVParser do
     expect(result.format).to eq(:mov)
     expect(result.width_px).to eq(1920)
     expect(result.height_px).to eq(1080)
+    expect(result.codecs).to eq(['apcn'])
   end
 
   it 'parses an MP4 video file and provides the necessary metadata' do
@@ -107,6 +108,7 @@ describe FormatParser::MOOVParser do
     expect(result.width_px).to eq(160)
     expect(result.height_px).to eq(90)
     expect(result.frame_rate).to eq(14.98)
+    expect(result.codecs).to eq(['avc1'])
   end
 
   it 'provides filename hints' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: format_parser
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Noah Berman
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-04-01 00:00:00.000000000 Z
+date: 2022-04-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ks