make_mkv

   1 #!/usr/bin/perl
   2 # Author: Alan J. Pippin
   3 # Description: Take a list of video files and create a single mkv video file from them
   4
   5 # Pre-requisites:
   6 # mkvtoolnix - http://www.bunkus.org/videotools/mkvtoolnix/
   7 # ffmpeg
   8
   9 ####################################################################################################
  10 # Includes
  11 use File::Copy;
  12 use File::Basename;
  13 use Getopt::Std;
  14 use File::stat;
  15 use DateTime;
  16 use DateTime::Duration;
  17 use DateTime::Format::Duration;
  18
  19 ####################################################################################################
  20 # Configuration parameters
  21 $mydir = `cd \$(dirname $0) 2>/dev/null; pwd`; chomp($mydir); unshift @INC,("$mydir");
  22 # Default configuration values
  23 require "organize_videos.conf";
  24 # Override defaults with local customizations
  25 if( -f "$mydir/organize_videos.conf.local") { require "organize_videos.conf.local"; }
  26
  27 ####################################################################################################
  28 # Command Line Options
  29 getopts("sqzt:o:i:h");
  30
  31 if(! defined $opt_t) { &usage(); die "-E- Missing required title: -t <title>\n"; }
  32 if(! defined $opt_o) { &usage(); die "-E- Missing required argument output video names: -o <output.mkv>\n"; }
  33 if(! defined $opt_i) { &usage(); die "-E- Missing required argument input video names: -i <input,input,...>\n"; }
  34
  35 sub usage {
  36     print "usage: $0 -t <title> -o <output.mkv> -i <input,input,...>\n";
  37     print "  -t <title>            Sets the general title for the output video file\n";
  38     print "  -o <output.mkv>       Sets the name of the output mkv file\n";
  39     print "  -i <input,input,...>  Sets the name of the input files to make into an mkv file\n";
  40     print "  -q                    Requantize input videos to decrease output video size (requires HandBrakeCLI)\n";
  41     print "  -z                    Recompress input videos to decrease output video size (requires HandBrakeCLI)\n";
  42     print "  -s                    Simulate mode. Don't actually make the video, but tell us what you will do\n";
  43     print "\n";
  44     return 1;
  45 }
  46
  47 ####################################################################################################
  48 # Helper Subroutines
  49 sub epoch_to_date {
  50     my ($epoch) = @_;
  51     my $mtime = DateTime->from_epoch(epoch => $epoch);
  52     $mtime->set_time_zone($timezone);
  53     return sprintf("%4d",$mtime->year)."-".sprintf("%02d",$mtime->month)."-".sprintf("%02d",$mtime->day)." ".$mtime->hms;
  54 }
  55
  56 ####################################################################################################
  57 # MAIN
  58
  59 # Turn the list of input videos into a hash with a value equal to the modification time in epoch seconds
  60 my %videos;
  61 foreach $video (split(/,/, $opt_i)) {
  62     if(! -r "$video") { die "-E- Unable to read input video file: $video\n"; }
  63     my $mtime_epoch = stat("$video")->mtime;
  64     my $mdate = epoch_to_date($mtime_epoch);
  65     $videos{$video} = $mtime_epoch;
  66 }
  67
  68 # Create the chapters file for the mkv file.
  69 # Make each video file it's own chapter in the MKV file.
  70 # Name the chapter with the date and time the video clip was taken (modified date).
  71 print "-> Creating $opt_o with title '$opt_t' from the following video files in last modified date order:\n";
  72 open(CHAPTERS,">$chapter_file") || die "-E- Unable to create chapter file: $chapter_file\n";
  73 my $chapter_num = 0;
  74 my $chapter_timecode = DateTime::Duration->new(years => 2000, hours => 0, minutes => 0, seconds => 0);
  75 my $timecode_format = DateTime::Format::Duration->new(pattern => '%H:%M:%S.%3N', normalize => 1);
  76 foreach my $video (sort{$videos{$a} <=> $videos{$b}} keys %videos) {
  77     $chapter_num++;
  78     my $hour = 0;
  79     my $min  = 5;
  80     my $sec  = 0;
  81     my $mtime_epoch = stat("$video")->mtime;
  82     my $mdate = epoch_to_date($mtime_epoch);
  83     my $duration = `$ffmpeg -i "$video" 2>&1 | grep Duration`;
  84     if($duration =~ /Duration: (\d+):(\d+):(\d+)\.(\d+)/) {
  85         $hour = $1;
  86         $min = $2;
  87         $sec = "$3.$4";
  88     }
  89     my $timecode = $timecode_format->format_duration($chapter_timecode);
  90     print "$mdate $hour:$min:$sec  -> $video \n";
  91     print CHAPTERS "CHAPTER".sprintf("%02d",$chapter_num)."=".$timecode."\n";
  92     print CHAPTERS "CHAPTER".sprintf("%02d",$chapter_num)."NAME=".$mdate."\n";
  93     my $dt = DateTime::Duration->new(years => 2000, hours => $hour, minutes => $min, seconds => $sec);
  94     $chapter_timecode = $chapter_timecode + $dt;
  95 }
  96 close(CHAPTERS);
  97 print "\n-> Creating the following chapter file for this video:\n";
  98 $chapter_file_contents = `cat $chapter_file`;
  99 print "$chapter_file_contents\n";
 100
 101 # Create the mkv video
 102 print "-> Creating the MKV video file '$opt_o'\n";
 103
 104 my $cmd = "$mkvmerge --title \"$opt_t\" $output_file_options -o \"$opt_o\"";
 105
 106 # Make our input file command line options for all the videos
 107 my $video_count = 0;
 108 my @video_tmp_files;
 109 foreach my $video (sort{$videos{$a} <=> $videos{$b}} keys %videos) {
 110     # Make a note of the video extension
 111     my $video_ext = $video;
 112     $video_ext =~ s/.*\.(\S+)$/$1/;
 113
 114     # Detect if the input file is interlaced or not.
 115     # There is a bug in mkvmerge 5.0.1 and earlier where if the video content is 1080i, it doesn't mux it properly, and it stutters.
 116     # The quick solution to this issue is to run the interlaced video file through ffmpeg and tell it to copy the video/audio streams to a mkv container.
 117     # We will then merge this temporarily created mkv container into the final mkv container instead of the original interlaced video.
 118     # http://ubuntuforums.org/showthread.php?t=1627194
 119     # http://forum.doom9.org/showthread.php?t=155732&page=31
 120     my $progressive = system('$ffmpeg -i "$video" 2>&1 | grep -q "frame rate differs"');
 121     if(!$progressive) {
 122         my $video_mkv = $video;
 123         print "   Detected interlaced video content: $video\n";
 124         if($video_ext !~ /mkv/i) {
 125             $video_mkv =~ s/\.[^.]*$//; $video_mkv .= ".ffmpeg.mkv";
 126             print "   Re-muxing the interlaced video content as an mkv file: $video_mkv\n";
 127             my $make_mkv_cmd = "$ffmpeg -y -i \"$video\" -scodec copy -acodec copy -vcodec copy -f matroska \"$video_mkv\" >> \"$tmpfile\" 2>&1";
 128             if(! defined $opt_s) {
 129                 my $errno = system("$make_mkv_cmd");
 130                 $errno = $errno >> 8;
 131                 if($errno > 1) {
 132                     unlink "$video_mkv";
 133                     die "-E- ffmpeg encountered some errors with exit code $errno\n";
 134                 }
 135             }
 136             # Push the name of our intermediate video file onto a list of files to delete on exit
 137             push(@video_tmp_files, "$video_mkv");
 138             # Update the name of our video file to equal the name of our new intermediate video file name
 139             $video = $video_mkv;
 140         }
 141     } else {
 142         print "   Detected progressive video content: $video\n";
 143     }
 144
 145     # Re-quantize or re-compress the input video to reduce the resulting output filesize
 146     # This also gives us a chance to deinterlace the video as well
 147     # Only re-quantize for .MTS videos
 148     # We can re-compress any input video
 149     if(((defined $opt_q) && ($video_ext =~ /mts/i)) || (defined $opt_z)) {
 150
 151         my $handbrake_options = "";
 152
 153         # Set our output video filename
 154         my $video_mp4 = $video; $video_mp4 =~ s/\.[^.]*$//; $video_mp4 .= ".hb.mp4";
 155
 156         # Set our requantize factor accordingly
 157         if(defined $opt_q) {
 158             print "   Re-quantizing input video content to: $video_mp4\n";
 159             $handbrake_options = $handbrake_requantize_options;
 160             if(!$progressive) { $handbrake_options = "-q $interlaced_requantize_quality"; }
 161             else { $handbrake_options = "-q $progressive_requantize_quality"; }
 162         };
 163
 164         # Set our recompress options accordingly
 165         if(defined $opt_z) {
 166             print "   Re-compressing input video content to: $video_mp4\n";
 167             $handbrake_options = $handbrake_recompress_options;
 168             # We want our audio to be passed-through by default, so detect how the audio of the input is encoded, and tell handbrake to make the output match
 169             $AUDIO_CODEC=`$ffmpeg -i "$video" 2>&1 | grep "Audio" | sed -r -e 's/.*?Audio: (\\S+),.*?/\\1/'`; chomp($AUDIO_CODEC);
 170             if($AUDIO_CODEC eq "") { die "-E- Unable to extract audio track encoding from input video file: $video\n"; }
 171             $handbrake_options .= " -E copy:$AUDIO_CODEC";
 172         }
 173
 174         # Set our de-interlace option accordingly
 175         my $deinterlace_option = "";
 176         if(!$progressive) { $deinterlace_option = "-d"; }
 177
 178         # Use HandBrake to requantize/recompress/deinterlace the input video
 179         my $handbrake_cmd = "$handbrake $deinterlace_option $handbrake_options -i \"$video\" -o \"$video_mp4\" >> \"$tmpfile\" 2>&1";
 180         #print "   $handbrake_cmd\n";
 181         if(! defined $opt_s) {
 182             my $errno = system("$handbrake_cmd");
 183             $errno = $errno >> 8;
 184             if($errno > 1) {
 185                 unlink "$video_mp4";
 186                 die "-E- handbrake encountered some errors with exit code $errno\n";
 187             }
 188         }
 189
 190         # Push the name of our intermediate video file onto a list of files to delete on exit
 191         push(@video_tmp_files, "$video_mp4");
 192         # Update the name of our video file to equal the name of our new intermediate video file name
 193         $video = $video_mp4;
 194     }
 195
 196     # Create our input file command line options for this video
 197     if($video_count == 0) {
 198         $cmd .= " $input_file_options \"$video\"";
 199     } else {
 200         $cmd .= " $input_file_options + \"$video\"";
 201     }
 202     $video_count++;
 203 }
 204
 205 # Execute our command line
 206 print "\n$cmd\n";
 207 if(! defined $opt_s) {
 208     my $errno = system("$cmd");
 209     $errno = $errno >> 8;
 210     if($errno > 1) {
 211         unlink "$opt_o";
 212         die "-E- mkvmerge encountered some errors with exit code $errno\n";
 213     }
 214 }
 215
 216 # Remove the temporary file used for ffmpeg and handbrake output
 217 if(-e "$tmpfile") { unlink "$tmpfile"; }
 218 # Remove the temporary file used for the chapter generation
 219 if(-e "$chapter_file") { unlink "$chapter_file"; }
 220
 221 # Remove any temporary video files created during the merge process
 222 foreach my $video (@video_tmp_files) {
 223     if(-e "$video") { unlink "$video"; }
 224 }
 225
 226 ####################################################################################################
 227