example_v2.rb

require 'axlsx'
require 'pry'
require './source/source'
require './parser/parser'
require './mp4/mp4'
require './layout/layout'

# mp4_file_path = './examples/keyint_30_02_converted.mp4'
# mp4_file_path = './examples/small_linus_example.mp4'
# mp4_file_path = './examples/03_ctts.mp4'
# mp4_file_path = './examples/keyint_30_example_google.mp4'
mp4_file_path = './examples/02.mp4'
# mp4_file_path = './examples/example_google.mp4'
file_name = File.basename(mp4_file_path)

source = Source.load_from_local(mp4_file_path)
parsed_mp4 = Parser.parse(source)
mp4 = MP4.load_v2(parsed_mp4)

# res = mp4.parts.all


mp4_layout = Layout.build(mp4, source)
File.write("./out/result_#{file_name}", mp4_layout.pack)
# return
package = Axlsx::Package.new

package.workbook do |workbook|
  mp4.get_tracks.each do |track|
    workbook.add_worksheet name: "trak_#{track['track_id']} (#{track['track_type']})" do |sheet|
      # sheet.add_row(['keyframe', 'sample_number', 'sample_size', 'chunk_number', 'chunk_offset', 'stts', 'timestamp'])
      sheet.add_row(['Chunk ID', 'Chunk Offset', 'Sample Number', 'Sample Size', 'Sample Delta', 'Sample C Delta', 'timestamp', 'Keyframe'])

      delta_counter = 0.0
      samples_list = mp4.get_samples_of(track['track_id'])

      until (row = samples_list.next).nil?
        sheet.add_row([
          row['chunk_index'],
          row['chunk_offset'],
          row['sample_index'],
          row['size'],
          row['delta'],
          row['c_delta'],
          Time.at(row['delta_sum'] / row['timescale'].to_f).utc.strftime('%H:%M:%S,%L'),
          row['keyframe']
        ])
      end

      grouped_chunks = mp4.cache_db.execute('SELECT chunk_index, min(sample_index) as min_sample_index, max(sample_index) as max_sample_index from samples WHERE track_id = ? GROUP BY chunk_index', track['track_id'])
      grouped_chunks.each do |grouped_chunk|
        _chunk_index, min_sample_index, max_sample_index = *grouped_chunk.values

        sheet.merge_cells "A#{min_sample_index + 1}:A#{max_sample_index + 1}"
        sheet.merge_cells "B#{min_sample_index + 1}:B#{max_sample_index + 1}"
      end
    end
  end
end

package.serialize "fragmented/new.fragmented_#{file_name}.xlsx"