Class: AppleTvConverter::MediaConverterAdapter

Inherits:
Object
  • Object
show all
Includes:
AppleTvConverter
Defined in:
lib/apple_tv_converter/media_converter_adapter.rb

Constant Summary

Constants included from AppleTvConverter

Timer, VERSION

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from AppleTvConverter

atomic_parsley_binary, atomic_parsley_binary=, copy, data_path, get_language_name, is_macosx?, is_windows?, logger, logger=, mp4box_binary, mp4box_binary=

Constructor Details

#initialize(options) ⇒ MediaConverterAdapter

Returns a new instance of MediaConverterAdapter.



7
8
9
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 7

def initialize(options)
  self.conversion_options = options
end

Instance Attribute Details

#conversion_options ⇒ Object

Returns the value of attribute conversion_options.



5
6
7
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 5

def conversion_options
  @conversion_options
end

Instance Method Details

#add_subtitles(media) ⇒ Object

Raises:

  • (NotImplementedYetException) 


171
172
173
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 171

def add_subtitles(media)
  raise NotImplementedYetException
end

#add_to_itunes(media) ⇒ Object

Raises:

  • (NotImplementedYetException) 


274
275
276
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 274

def add_to_itunes(media)
  raise NotImplementedYetException
end

#clean_up(media) ⇒ Object 



278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 278

def clean_up(media)
  printf "* Cleaning up"
  begin
    if media.needs_transcoding?
      FileUtils.rm media.original_filename

      if media.converted_filename_equals_original_filename?
        FileUtils.mv media.converted_filename, media.original_filename
      end
    end

    # Always clean up subtitles and artwork, and backup
    FileUtils.rm(media.artwork_filename) if File.exists?(media.artwork_filename)
    FileUtils.rm_r list_files(media.original_filename.gsub(File.extname(media.original_filename), '*.srt'))
    FileUtils.rm(media.backup_filename) if File.exists?(media.backup_filename)

    puts " [DONE]"
  rescue
    puts " [ERROR]"
  end
end

#download_subtitles(media, languages) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 20

def download_subtitles(media, languages)
  AppleTvConverter::SubtitlesFetcher::Opensubtitles.new(languages) do |fetcher|
    if fetcher.has_found_subtitles?(media)
      printf "* Downloading subtitles"
      fetcher.download_subtitles media do |step, subtitles|
        case step
          when :search        then puts %Q[ (#{subtitles.map { |l, subs| "#{subs.count} #{AppleTvConverter.get_language_name(l)}" }.join(', ') })]
          when :downloading   then printf "  * Downloading: \##{subtitles['IDSubtitleFile']} (#{AppleTvConverter.get_language_name(subtitles['SubLanguageID'])}) - #{subtitles['SubFileName']}"
          when :downloaded    then puts " [DONE]"
        end
      end
      puts "  * All subtitles downloaded"
    else
      puts "* No subtitles found to download"
    end
  end
end

#extract_subtitles(media, languages) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 38

def extract_subtitles(media, languages)
  puts "* Extracting subtitles"

  languages.map!(&:to_sym)

  if media.has_embedded_subtitles?(languages)
    last_destination_filename = nil

    options = "-scodec subrip"

    media.ffmpeg_data.streams.each do |stream|
      next unless stream.type == :subtitle && (languages.empty? || languages.include?(stream.language.to_sym))
      filename = File.join(File.dirname(media.original_filename), "#{File.basename(media.original_filename).gsub(File.extname(media.original_filename), %Q[.#{stream.stream_number}.#{stream.language}.srt])}")

      begin
        printf "  * #{File.basename(filename)}: Progress:     0%"
        start_time = Time.now.to_i
        transcoded = media.ffmpeg_data.transcode(filename, "#{options} -map #{stream.input_number}:#{stream.stream_number}", :validate_output => false) do |progress|
          elapsed = Time.now.to_i - start_time
          printf "\r" + (" " * 40)
          printf "\r  * #{File.basename(filename)}: Progress: #{(progress * 10000).round.to_s.gsub(/(\d{2})$/, '.\1').gsub(/^\./, '0.').rjust(6)}%% (#{(elapsed / 60).to_s.rjust(2, '0')}:#{(elapsed % 60).to_s.rjust(2, '0')})     "
        end
        puts ""
      rescue Interrupt
        puts "\nProcess canceled!"
        exit!
      end
    end

    puts "  * Extracted all subtitles"
  else
    puts "  * No subtitles to extract"
  end
end

#get_imdb_info(media, feedback = true) ⇒ Object 



210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 210

def get_imdb_info(media, feedback = true)
  printf "* Getting info from IMDB" if feedback

  if media.imdb_id
    media.imdb_movie = Imdb::Movie.new(media.imdb_id)
  elsif Dir[File.join(File.dirname(media.original_filename), '*.imdb')].any?
    media.imdb_movie = Imdb::Movie.new(File.basename(Dir[File.join(File.dirname(media.original_filename), '*.imdb')].first).gsub(/\.imdb$/i, ''))
  else

    search = Imdb::Search.new(media.show)

    search.movies.delete_if do |item|
      item.title.strip =~ /(?:(?:\(TV\s*(?:Movie|(?:Mini.?)?Series|Episode))|(?:Video(?:\s*Game)?))/i
    end

    media.imdb_id = if search.movies.length > 1 && conversion_options.interactive
      choice = 0
      puts "\n  *"
      while true
        puts %Q[  | Several movies found, choose the intended one#{" (showing only the first 20 of #{search.movies.length} results)" if search.movies.length > 20}:]

        search.movies[0...20].each_with_index do |item, index|
          puts "  | #{(index + 1).to_s.rjust(search.movies.length.to_s.length)} - #{item.title.strip} (id: #{item.id})"
          puts "  | #{' '.rjust(search.movies.length.to_s.length)}   AKA: #{item.also_known_as.join(', ').strip}" if item.also_known_as.any?
        end

        printf "  |\n  *- What's your choice (1..#{[search.movies.length, 20].min})? "
        choice = STDIN.gets.chomp.to_i

        break if choice.between?(1, [search.movies.length, 20].min)

        puts "  | Invalid choice!"
        puts "  |"
      end

      printf "  * Getting info from IMDB"
      search.movies[choice - 1].id
    else
      search.movies.first.id rescue nil
    end

    media.imdb_movie = Imdb::Movie.new(media.imdb_id) if media.imdb_id
  end

  begin
    media.imdb_movie.year
    puts " [DONE]" if feedback
  rescue OpenURI::HTTPError => e
    media.imdb_id = nil
    media.imdb_movie = nil
    puts (e.message =~ /404/ ? " [NOT FOUND]" : " [ERROR]") if feedback
  rescue
    if media.imdb_id.nil?
      puts " [NOT FOUND]" if feedback
    else
      raise e
    end
  end
end

#get_metadata(media) ⇒ Object 



175
176
177
178
179
180
181
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 175

def get_metadata(media)
  if media.is_tv_show_episode?
    get_tv_show_db_info media
  else
    get_imdb_info media
  end
end

#get_tv_show_db_info(media) ⇒ Object 



183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 183

def get_tv_show_db_info(media)
  printf "* Getting info from TheTVDB"

  media.tvdb_movie = TvDbFetcher.search(media, conversion_options.interactive)
  if media.tvdb_movie
    media.imdb_id = media.tvdb_movie[:show][:series]['IMDB_ID'] if media.tvdb_movie.has_key?(:show) && media.tvdb_movie[:show].has_key?(:series)
    media.imdb_id = media.imdb_id.gsub(/\D+/, '') if media.imdb_id

    # Update the episode name, if available
    media.episode_title = media.tvdb_movie_data('EpisodeName')
    media.tvdb_id = media.tvdb_movie_data('seriesid')
    media.tvdb_season_id = media.tvdb_movie_data('seasonid')
    media.tvdb_episode_id = media.tvdb_movie_data('id')
    media.imdb_episode_id = media.tvdb_movie_data('IMDB_ID')
    media.imdb_episode_id = media.imdb_episode_id.gsub(/\D+/, '') if media.imdb_episode_id

    unless media.imdb_id.nil? || media.imdb_id.blank?
      puts "" # Line break just for output layout
      get_imdb_info(media, false)
    end

    puts " [DONE]"
  else
    puts " [NOT FOUND]"
  end
end

#rename_to_plex_format(media) ⇒ Object 



300
301
302
303
304
305
306
307
308
309
310
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 300

def rename_to_plex_format(media)
  printf "* Renaming to PLEX format"
  begin
    plex_format_filename = media.plex_format_filename
    FileUtils.mv(media.converted_filename_equals_original_filename? ? media.original_filename : media.converted_filename, plex_format_filename) unless media.converted_filename == plex_format_filename

    puts " [DONE]"
  rescue => e
    puts " [ERROR]"
  end
end

#search_subtitles(media, languages) ⇒ Object 



11
12
13
14
15
16
17
18
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 11

def search_subtitles(media, languages)
  # Load the subtitles into memory and get IMDB id from them
  AppleTvConverter::SubtitlesFetcher::Opensubtitles.new(languages) do |fetcher|
    fetcher.search_subtitles media do |subtitles|
      media.imdb_id = subtitles.first['IDMovieImdb'] if media.imdb_id.nil? || media.imdb_id.to_s.strip == ''
    end
  end
end

#tag(media) ⇒ Object

Raises:

  • (NotImplementedYetException) 


270
271
272
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 270

def tag(media)
  raise NotImplementedYetException
end

#transcode(media, languages = nil) ⇒ Object 



73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
 
# File 'lib/apple_tv_converter/media_converter_adapter.rb', line 73

def transcode(media, languages = nil)
  if media.needs_transcoding?
    puts "* Transcoding"

    options = {}
    options[:codecs] = get_transcode_options(media)
    options[:files] = ""
    options[:metadata] = ""
    options[:map] = ""
    options[:extra] = ""

    # Better video and audio transcoding quality
    if media.needs_video_conversion?
      # Ensure divisible by 2 width and height
      dimensions = "-s #{(media.ffmpeg_data.width % 2 > 0) ? (media.ffmpeg_data.width + 1) : media.ffmpeg_data.width}x#{(media.ffmpeg_data.height % 2 > 0) ? (media.ffmpeg_data.height + 1) : media.ffmpeg_data.height}" if media.ffmpeg_data.width % 2 > 0 || media.ffmpeg_data.height % 2 > 0

      options[:extra] << " #{dimensions} -mbd rd -flags +mv4+aic -trellis 2 -cmp 2 -subcmp 2 -g 300 -pass 1 -q:v 1 -r 23.98"
    end

    if media.needs_audio_conversion?
      if media.ffmpeg_data.audio_codec =~ /mp3/i
        # Ensure that destination audio bitrate is 128k for Stereo MP3 (couldn't convert with higher bitrate)
        audio_bitrate = 128 if media.ffmpeg_data.audio_channels == 2
      elsif media.ffmpeg_data.audio_codec =~ /pcm_s16le/i
        # Ensure that destination audio bitrate is 128k for PCM signed 16-bit little-endian (couldn't convert with higher bitrate)
        audio_bitrate = 128
      elsif media.ffmpeg_data.audio_codec =~ /ac3/i
        # Ensure that maximum destination audio bitrate is 576k for AC3 (couldn't convert with higher bitrate)
        audio_bitrate = [media.ffmpeg_data.audio_bitrate || 576, 576].min
      end

      audio_bitrate ||= media.ffmpeg_data.audio_bitrate || 448

      options[:extra] << " -af volume=2.000000" # Increase the volume when transcoding
      options[:extra] << " -ac #{media.ffmpeg_data.audio_channels} -ar #{media.ffmpeg_data.audio_sample_rate}"
      options[:extra] << " -ab #{[audio_bitrate, (media.ffmpeg_data.audio_bitrate || 1000000)].min}k"
    end

    # Ensure the languages are 'stored' as symbols, for comparison
    languages = (languages || []).map(&:to_sym)

    # If we're filtering by language, ensure that unknown and undetermined language
    # streams are also mapped
    languages += [nil, :und, :unk] if languages.any?

    # If the file has more than one audio track, map all tracks but subtitles when transcoding
    if media.audio_streams.length > 0
      # Check whether to filter audio tracks (ie, we have at least one of the languages we're filtering)
      filter_by_language = false

      if languages.any?
        media.streams.each do |stream|
          if stream.type == :audio && languages.include?(stream.language ? stream.language.to_sym : stream.language)
            filter_by_language = true
            break
          end
        end
      end

      media.streams.each do |stream|
        options[:map] << " -map #{stream.input_number}:#{stream.stream_number}" if stream.type == :video || (stream.type == :audio && (!filter_by_language || languages.include?(stream.language ? stream.language.to_sym : stream.language)))
      end
    end

    options = "#{options[:files]} #{options[:codecs]} #{options[:map]} #{options[:metadata]} #{options[:extra]}"

    transcoded = nil

    begin
      start_time = Time.now.to_i
      transcoded = media.ffmpeg_data.transcode(media.converted_filename, options) do |progress|
        elapsed = Time.now.to_i - start_time
        printf "\r" + (" " * 40)
        printf %Q[\r  * Progress: #{(progress * 10000).round.to_s.gsub(/(\d{2})$/, '.\1').gsub(/^\./, '0.').rjust(6)}%% (#{(elapsed / 60).to_s.rjust(2, '0')}:#{(elapsed % 60).to_s.rjust(2, '0')})]
      end
    rescue Interrupt
      puts "\nProcess canceled!"
      exit!
    end

    status = transcoded.valid?

    printf "\r" + (" " * 40)

    if status
      puts "\r  * Progress: [DONE]#{' ' * 20}"
    else
      puts "\r  * Progress: [ERROR]#{' ' * 20}"
      exit!
    end
  else
    puts "* Encoding: [UNNECESSARY]"
    status = true
  end

  return status
end