duo_splitter 0.1.0 → 0.2.0

data/lib/duo_splitter/commands/repeating_command.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Commands
+    class RepeatingCommand < BaseCommand
+      def run
+        Services::CreateRepeatingSentences.new(context: context).run
+        print_done
+      end
+
+      private
+
+      def default_output_dir
+        Pathname.new('~/Desktop/DUO 3.0 (Repeating)').expand_path
+      end
+
+      def parse_options(argv)
+        parser.banner = %(Usage: #{parser.program_name} [options] repeating "01 Track 01.wav" ... "45 Track 45.wav")
+
+        on_output_dir
+        on_output_format
+        on_output_intro
+        on_prefix_section_number
+        on_show_progress
+        on_ffmpeg_path
+        on_sox_path
+        on_use_notification_sound
+        on_notification_sound_path
+
+        parse!(argv)
+
+        after_parse
+      end
+
+      Runner.register_command(:repeating, self)
+    end
+  end
+end

data/lib/duo_splitter/commands/split_command.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Commands
+    class SplitCommand < BaseCommand
+      def run
+        Services::CreateSentences.new(context: context).run
+        print_done
+      end
+
+      private
+
+      def parse_options(argv)
+        parser.banner = %(Usage: #{parser.program_name} [options] split "01 Track 01.wav" ... "45 Track 45.wav")
+
+        on_output_dir
+        on_output_format
+        on_output_intro
+        on_prefix_section_number
+        on_show_progress
+        on_ffmpeg_path
+
+        parse!(argv)
+
+        after_parse
+      end
+
+      Runner.register_command(:split, self)
+    end
+  end
+end

data/lib/duo_splitter/context.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'pathname'
+
+module DuoSplitter
+  class Context
+    attr_reader :album
+
+    attr_accessor :output_dir
+    attr_accessor :output_format
+    attr_accessor :output_intro
+    attr_accessor :prefix_section_number
+    attr_accessor :show_progress
+    attr_accessor :ffmpeg_path
+    attr_accessor :sox_path
+    attr_accessor :use_notification_sound
+    attr_accessor :notification_sound_path
+    attr_accessor :temp_dir
+
+    def initialize
+      @album = Models::Album.new
+      @output_dir = Pathname.new('~/Desktop/DUO 3.0').expand_path
+      @output_format = 'wav'
+      @output_intro = true
+      @prefix_section_number = true
+      @show_progress = true
+      @ffmpeg_path = 'ffmpeg'
+      @sox_path = 'sox'
+      @use_notification_sound = true
+      @notification_sound_path = ASSETS_DIR.join('C4.wav')
+      @temp_dir = nil
+    end
+  end
+end

data/lib/duo_splitter/encoders/aac_encoder.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Encoders
+    class AACEncoder < BaseEncoder
+      def encoding_options
+        %w[-codec:a aac -b:a 256k]
+      end
+
+      def ext
+        'm4a'
+      end
+
+      BaseEncoder.register_encoder(:aac, self)
+    end
+  end
+end

data/lib/duo_splitter/encoders/base_encoder.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Encoders
+    class BaseEncoder
+      class << self
+        def known_encoders
+          @known_encoders ||= {}
+        end
+
+        def register_encoder(encoder_name, encoder_class)
+          known_encoders[encoder_name.to_sym] = encoder_class
+        end
+      end
+
+      attr_reader :commands
+
+      def initialize(context:)
+        @context = context
+        @commands = build_commands
+      end
+
+      def run
+        return if @commands.empty?
+
+        @context.output_dir.mkpath
+
+        BatchProcessor.new(message: 'encoding...', show_progress: @context.show_progress).run(@commands)
+      end
+
+      private
+
+      def build_commands
+        sentences = @context.album.sentences(intro: @context.output_intro).select(&:source_audio_path)
+
+        sentences.map do |sentence|
+          output_basename = sentence.output_basename(prefix_section_number: @context.prefix_section_number, ext: ext)
+
+          command = [@context.ffmpeg_path, '-y']
+          command += ['-i', sentence.source_audio_path.to_s]
+          command += encoding_options
+          command += [@context.output_dir.join(output_basename).to_s]
+
+          command
+        end
+      end
+
+      def encoding_options
+        raise Error, '#encoding_options must be defined in sub classes'
+      end
+
+      def ext
+        raise Error, '#ext must be defined in sub classes'
+      end
+    end
+  end
+end

data/lib/duo_splitter/encoders/mp3_encoder.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Encoders
+    class MP3Encoder < BaseEncoder
+      def encoding_options
+        %w[-codec:a libmp3lame -b:a 320k]
+      end
+
+      def ext
+        'mp3'
+      end
+
+      BaseEncoder.register_encoder(:mp3, self)
+    end
+  end
+end

data/lib/duo_splitter/encoders/wav_encoder.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Encoders
+    class WAVEncoder < BaseEncoder
+      def encoding_options
+        %w[-codec:a pcm_s16le -ar 44100 -ac 2]
+      end
+
+      def ext
+        'wav'
+      end
+
+      BaseEncoder.register_encoder(:wav, self)
+    end
+  end
+end

data/lib/duo_splitter/models/album.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Models
+    class Album
+      attr_reader :sections
+
+      def initialize
+        @sections = load_sections
+      end
+
+      def sentences(intro: true)
+        sections.map {|section| section.sentences(intro: intro) }.flatten
+      end
+
+      private
+
+      def load_sections
+        1.upto(NUMBER_OF_SECTIONS).map {|n| Section.new(number: n) }
+      end
+    end
+  end
+end

data/lib/duo_splitter/models/section.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Models
+    class Section
+      attr_reader :number
+
+      attr_accessor :audio_path
+
+      def initialize(number:)
+        @number = number
+        @sentences = load_sentences
+      end
+
+      def sentences(intro: true)
+        @sentences.reject {|sentence| !intro && sentence.intro? }
+      end
+
+      private
+
+      def load_sentences
+        label_path = DATA_DIR.join(format('labels/%02<number>d.txt', number: number))
+
+        records = label_path.readlines.map {|line| line.chomp.split(/\t/) }
+
+        records.unshift(%w[0 0 0])
+
+        records.each_cons(2).map do |current_record, next_record|
+          sentence_number = current_record.last.to_i
+          start_pos = current_record.first
+          end_pos = next_record.first
+
+          Sentence.new(number: sentence_number, start_pos: start_pos, end_pos: end_pos, section: self)
+        end
+      end
+    end
+  end
+end

data/lib/duo_splitter/models/sentence.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module DuoSplitter
+  module Models
+    class Sentence
+      attr_reader :number, :start_pos, :end_pos, :section
+
+      attr_accessor :audio_path, :blank_audio_path, :source_audio_path
+
+      def initialize(number:, start_pos:, end_pos:, section:)
+        @number = number
+        @start_pos = start_pos
+        @end_pos = end_pos
+        @section = section
+      end
+
+      def duration
+        Float(@end_pos) - Float(@start_pos)
+      end
+
+      def intro?
+        number.zero?
+      end
+
+      def output_basename(prefix_section_number: true, blank: false, concatenated: false, ext:)
+        basename = format('%03<number>d', number: number)
+        basename = "#{basename}_blank" if blank
+        basename = "#{basename}_concatenated" if concatenated
+        basename = "#{basename}.#{ext}"
+        basename = format('%02<section_number>d-', section_number: @section.number) + basename if prefix_section_number
+        basename
+      end
+    end
+  end
+end

data/lib/duo_splitter/runner.rb

@@ -1,93 +1,72 @@
+# frozen_string_literal: true
+
 require 'optparse'
-require 'pathname'
 
 module DuoSplitter
   class Runner
-    attr_reader :output_dir, :format, :output_intro, :add_section_prefix, :ffmpeg_path, :album
+    class << self
+      def known_commands
+        @known_commands ||= {}
+      end
 
-    def initialize(argv)
-      @output_dir = Pathname.new('~/Desktop/DUO 3.0').expand_path
-      @format = :wav
-      @known_formats = Encoder.known_encoders.keys.sort
-      @output_intro = true
-      @add_section_prefix = true
-      @ffmpeg_path = 'ffmpeg'
-      @album = Album.new
-
-      parse_options(argv)
-      handle_source_paths(argv)
+      def register_command(command_name, command_class)
+        known_commands[command_name.to_sym] = command_class
+      end
     end
 
-    def run
-      encoder_class = Encoder.known_encoders[@format]
-
-      encoder = encoder_class.new(
-        album: @album,
-        output_dir: @output_dir,
-        output_intro: @output_intro,
-        add_section_prefix: @add_section_prefix,
-        ffmpeg_path: @ffmpeg_path
-      )
+    def initialize(argv)
+      @argv = argv.dup
+      @parser = OptionParser.new
 
-      encoder.run
+      parse_options
     end
 
-    private
+    def run
+      command_name = @argv.shift
 
-    def parse_options(argv)
-      parser = OptionParser.new
+      if command_name
+        command_class = self.class.known_commands[command_name.to_sym]
 
-      parser.banner = %(Usage: #{File.basename($PROGRAM_NAME)} [options] "01 Track 01.wav" "02 Track 02.wav" ... "45 Track 45.wav")
+        raise Error, "unknown command: #{command_name}" unless command_class
 
-      parser.on('-d', '--output-dir=PATH', "Output directory (default: #{@output_dir})") do |value|
-        @output_dir = Pathname.new(value)
+        command_class.new(@argv).run
+      else
+        show_help
       end
+    end
 
-      parser.on('-f', '--format=FORMAT', @known_formats, "Output format (default: #{@format}; one of #{@known_formats.join(', ')})") do |value|
-        @format = value.to_sym
-      end
+    private
 
-      parser.on('--[no-]output-intro', "Output intro track (default: #{@output_intro})") do |flag|
-        @output_intro = flag
-      end
+    def parse_options
+      @parser.banner = "Usage: #{@parser.program_name} [options] command"
 
-      parser.on('--[no-]section-number', "Add section number to output name (default: #{@add_section_prefix})") do |flag|
-        @add_section_prefix = flag
+      @parser.on('-h', '--help', 'Print this message') do
+        show_help
+        exit
       end
 
-      parser.on('--ffmpeg-path=PATH', "Path to ffmpeg command (default: #{@ffmpeg_path})") do |value|
-        @ffmpeg_path = value
+      @parser.on('-v', '--version', 'Print version number') do
+        puts DuoSplitter::VERSION
+        exit
       end
 
       begin
-        parser.parse!(argv)
-      rescue OptionParser::ParseError => error
-        raise Error, error.message
-      end
-
-      unless @add_section_prefix
-        @output_intro = false
+        @parser.order!(@argv)
+      rescue OptionParser::ParseError => e
+        raise Error, e.message
       end
     end
 
-    def handle_source_paths(argv)
-      argv.each do |source_path|
-        source_path = Pathname.new(source_path)
-
-        if /\A(\d+)/ =~ source_path.basename.to_s
-          section_number = $1.to_i
+    def show_help
+      puts @parser
+      puts
+      puts <<~MESSAGE
+        Available commands are:
 
-          section = @album.sections.find {|section| section.number == section_number }
-
-          if section
-            section.source_path = source_path
-          else
-            raise Error, "invalid section number: #{section_number}"
-          end
-        else
-          raise Error, "input file name doesn't start with a section number: #{source_path}"
-        end
-      end
+          - split
+          - repeating
+          - overlapping
+      MESSAGE
     end
   end
 end