mkv_extract_chapter

   1 #!/usr/bin/perl
   2 # Author: Alan J. Pippin
   3 # Description: Extract the given chapter(s) from an mkv file into separate video files
   4
   5 # MOV:
   6 # major_brand : qt
   7 # Video: h264 (avc1 / 0x31637661)
   8 # Audio: pcm_s16le
   9
  10 # MTS:
  11 # Video: h264 (High) (HDMV / 0x564D4448)
  12 # Audio: ac3 (AC-3 / 0x332D4341)
  13
  14 # MP4:
  15 # Video: h264 (Baseline) (avc1 / 0x31637661)
  16 # Audio: aac (mp4a / 0x6134706D)
  17
  18 # 3GP:
  19 # Video: h264 (Baseline) (avc1 / 0x31637661)
  20 # Audio: aac (mp4a / 0x6134706D)
  21
  22 # FFMPEG EXAMPLE CMD:
  23 #ffmpeg -ss 00:00:00.000 -t 00:00:38.299 -i 2010-12-04.000.mkv -map 0 -vcodec copy -acodec copy test.mov
  24
  25 ####################################################################################################
  26 # Includes
  27 use File::Copy;
  28 use File::Basename;
  29 use Getopt::Std;
  30
  31 ####################################################################################################
  32 # Configuration parameters - CHANGE THESE TO SUITE YOUR NEEDS
  33 my $ffmpeg=`which ffmpeg`; chomp($ffmpeg);
  34 my $tmpfile = `tempfile`; chomp($tmpfile);
  35 ####################################################################################################
  36
  37 ####################################################################################################
  38 # Command Line Options
  39 getopts("c:i:o:se:");
  40
  41 if(! -x $ffmpeg) { die "-E- Unable to find required program: ffmpeg\n"; }
  42 if(! defined $opt_i) { &usage(); die "-E- Missing required argument input video name: -i <input.mkv>\n"; }
  43 if(! defined $opt_c) { &usage(); die "-E- Missing required chapter: -c <all|1,2,...>"; }
  44 if(! -r $opt_i) { &usage(); die "-E- Unable to open/find the input mkv file: $opt_i\n"; }
  45 if(! defined $opt_o) {
  46     if($opt_i =~ /^(\d+-\d+-\d+)/) { $opt_o = $1; }
  47     elsif($opt_i =~ /^(.*?)\./) { $opt_o = $1; }
  48 }
  49
  50 sub usage {
  51     print "usage: $0 -i <input.mkv> -o <output_base_name> -c <chapter,chapter,...>\n";
  52     print "  -c <chapter,chapter,...>  Specify which chapter from the mkv file to extract (all or 1,2,3, ...)\n";
  53     print "  -i <input.mkv>            Specify the name of the input mkv file\n";
  54     print "  -o <output_base_name>     Sets the base name to use when naming the output videos extracted from the chapters\n";
  55     print "                            If not specified, the base name will be taken from the name of your input video\n";
  56     print "  -e <ext>                  Specify the extension that should be used on the output files.\n";
  57     print "                            This is automatically determined. You only need to specify it if autodetection fails.\n";
  58     print "  -s                        Simulate mode. Don't actually make the video, but tell us what you will do\n";
  59     print "\n";
  60     return 1;
  61 }
  62
  63 ####################################################################################################
  64 # SUBROUTINES
  65
  66 sub detect_ext {
  67     my ($ffmpeg_info) = @_;
  68
  69     my $h264 = 0;
  70     my $h264_high = 0;
  71     my $ac3 = 0;
  72     my $aac = 0;
  73     my $pcm_s16le = 0;
  74
  75     foreach $line (@{$ffmpeg_info}) {
  76         if($line =~ /Stream/) {
  77             if($line =~ /h264/i) { $h264 = 1; }
  78             if($line =~ /h264.*High/i) { $h264_high = 1; }
  79             if($line =~ /ac3/i) { $ac3 = 1; }
  80             if($line =~ /aac/i) { $aac = 1; }
  81             if($line =~ /pcm_s16le/i) { $pcm_s16le = 1; }
  82         }
  83     }
  84
  85     if($h264 && $pcm_s16le) { return "mov"; }
  86     if($h264_high && $ac3) { return "mts"; }
  87     if($h264 && $aac) { return "mp4"; }
  88
  89     return "UNKNOWN";
  90 }
  91
  92 sub export_chapter {
  93     my ($chapters, $chapter) = @_;
  94     foreach $ch (@{$chapters}) {
  95         if($ch == $chapter) { return 1; }
  96     }
  97     return 0;
  98 }
  99
 100 ####################################################################################################
 101 # MAIN
 102
 103 # Turn our list of chapters into an ordered array
 104 my @chapters = sort(split(/,/, $opt_c));
 105 my $all_chapters = 0;
 106 if($opt_c =~ /all/) { $all_chapters = 1; }
 107 print "-> Extracting the following chapters from $opt_i: @chapters\n";
 108
 109 # Use ffmpeg to extract the list of chapters available to rip
 110 # For each chapter specified on the command line, use ffmpeg to extract a video clip from that chapter
 111 my @ffmpeg_info = `$ffmpeg -i $opt_i 2>&1`;
 112 foreach $line (@ffmpeg_info) {
 113     if($line =~ /Chapter #\d+\.(\d+): start (\S+), end (\S+)/) {
 114         $chapter = $1;
 115         $start = $2;
 116         $end = $3;
 117         $duration = $end - $start;
 118         if($duration < 0) { die "-E- Unexpected negative duration detected for chapter $chapter\n"; }
 119         if($all_chapters || &export_chapter(\@chapters,$chapter)) {
 120             $ext = "UNKNOWN";
 121             if(defined $opt_e) { $ext = $opt_e; }
 122             else { $ext = &detect_ext(\@ffmpeg_info); }
 123             if($ext =~ /UNKNOWN/) { die "-E- Unable to determine the file type/extension to use for the output videos. Specify it with the -e <ext> option.\n"; }
 124             $dstfile = $opt_o . ".c" . sprintf("%03d",$chapter) . "." . $ext;
 125             print "-> Exporting $chapter to $dstfile: ";
 126             $cmd = "ffmpeg -ss $start -t $duration -i $opt_i -map 0 -vcodec copy -acodec copy $dstfile 2>&1";
 127             if(defined $opt_s) {
 128                 print "$cmd\n";
 129             } else {
 130                 print "\n";
 131                 $errno = system("$cmd");
 132                 $errno = $errno >> 8;
 133                 if($errno > 0) { die "-E- ffmpeg encountered some errors with exit code $errno\n"; }
 134             }
 135         }
 136     }
 137 }
 138
 139 ####################################################################################################