Major rewrite of video import scripts to handle videos from any device
[videoscripts/.git] / merge_videos_by_day
index 1a153b00e63ad4bfed95eeb85838666cc9d22b27..cda301dae72bcf37e2d184d0e017b9196f7557b0 100755 (executable)
@@ -34,7 +34,9 @@ sub usage {
     return 1;
 }
 if(defined $opt_h) { usage(); exit 1; }
+$SIG{'INT'} = sub {die "-E- Killed by CTRL-C\n"};
 
+####################################################################################################
 # Sanity checks
 if((defined $opt_q || defined $opt_z) && !$use_compute_host && ! -x "$handbrake") { die "-E- Unable to find required program: handbrake\n"; }
 if(! -d $srcpathname) { &usage; print "-E- Can't find srcpath: $srcpathname\n"; exit 1; }
@@ -136,137 +138,82 @@ foreach $file (sort `$find_cmd`) {
     $dstdir = $srcdir;
     $dstfile = $dstdir . "/" . $year . "-" . $monthnum . "-" . $day;
 
-    # Check for duplicate filenames at the destination
-    $newfile = $dstfile . "." . $video_suffix;
-    if(-e "$newfile.$ext") {
-       foreach $i ($video_suffix+1 .. '999') {
-           $newfile = $dstfile . "." . sprintf("%03d",$i);
-           if(! -e "$newfile.$ext") { last; }
-       }
-       $dstfile = $newfile;
-    }
-
-    # Set the name of our unique destination file
-    $dstfile = "$newfile.$ext";
-
-    # You can only merge videos into a single destination that have the same extension/type
-    push(@{$videos{"$srcext"}{"$dstfile"}}, "\"$srcdir/$srcfile\"");
-}
-
-# Check for duplicate filenames in the dstfiles being created for other exts
-my $last_dstnum = 0;
-my $changed_dst = 0;
-foreach $ext (sort keys %videos) {
-    #print "checking: $ext\n";
-    foreach $video (sort keys %{$videos{$ext}}) {
-       #print "checking: $ext $video\n";
-       # Make sure this video name is not in use as a destination for any other ext
-       foreach $checkext (sort keys %videos) {
-           #print "checking: $ext $video $checkext\n";
-           if($checkext eq $ext) { next; }
-           foreach $checkvideo (sort keys %{$videos{$checkext}}) {
-               #print "checking: $ext $video $checkext $checkvideo\n";
-               if("$video" eq "$checkvideo") {
-                   if($video =~ /(.*?)\.(\d+)\.(\w+)$/) {
-                       $dstfile = $1;
-                       $dstnum = $2;
-                       $dstext = $3;
-                   }
-                   foreach $i ($last_dstnum .. '999') {
-                       $last_dstnum = $i + 1;
-                       $newfile = $dstfile . "." . sprintf("%03d",$i);
-                       if("$video" ne "$newfile.$dstext") { last; }
-                   }
-                   $videos{$ext}{"$newfile.$dstext"} = $videos{$ext}{$video};
-                   #print "for ext $ext changed destination to: $newfile.$dstext: $videos{$ext}{$video};\n";
-                   delete $videos{$ext}{$video};
-                   $changed_dst = 1;
-                   last;
-               }
-           }
-           if($changed_dst) { last; }
-       }
-       if($changed_dst) { last; }
-    }
-    $changed_dst = 0;
+    # Make a note of this video and its merged destination
+    push(@{$videos{"$dstfile"}}, "\"$srcdir/$srcfile\"");
 }
 
 # Tell the user which videos we are going to merge
-foreach $ext (sort keys %videos) {
-    foreach $video (sort keys %{$videos{$ext}}) {
-       foreach $srcfile (@{$videos{$ext}{$video}}) { 
-           print "   merging $ext $srcfile into \"$video\"\n";
-       }
+foreach $dstfile (sort keys %{$videos}) {
+    foreach $srcfile (@{$videos{$dstfile}}) { 
+       print "   merging $srcfile into \"$dstfile\"\n";
     }
 }
 
 # Now actually do the merging
 print "\n";
-foreach $ext (sort keys %videos) {
-    foreach $video (sort keys %{$videos{$ext}}) {
+foreach $dstfile (sort keys %videos) {
 
-       my $videos = join(',', @{$videos{$ext}{$video}});
-       
-       if($video =~ /(\d+)-(\d+)-(\d+)/) {
-           $year = $1;
-           $month = $2;
-           $day = sprintf("%02d",$3);
-       }
-       
-       my $pwd = `pwd`; chomp($pwd);
-       my $cmd = "";
-       if($use_compute_host) { $cmd .= "ssh $compute_host 'cd \"$pwd\";"; }
-       $cmd .= "$make_mkv -t \"$video_title_prefix $year-$month-$day\" -o \"$video\" -i $videos";
-       if($requantize_input_video) { $cmd .= ' -q'; }
-       if($recompress_input_video) { $cmd .= ' -z'; }
-       if($opt_v) { $cmd .= ' -v'; }
-       if($use_compute_host) { $cmd .= "'"; }
-       if(defined $opt_t) {
-           # Print what will be done, but don't actually do it
-           print "\n-> Creating \"$video\"\n";
-           print "$cmd\n";
-           if(!defined $opt_k) { 
-               foreach $video (@{$videos{$ext}{$video}}) {
-                   if(($save_originals) && ($video =~ /\.$originals_file_ext/)) {
-                       print "-> Saving the original video $video\n";
-                       if(index($video, basename(dirname($video))) == -1) { 
-                           print("mv $video \"$origpathname/".basename(dirname($video))."_".basename($video)."\n");
-                       } else {
-                           print("mv $video \"$origpathname/".basename($video)."\n");
-                       }
+    my $videos = join(',', @{$videos{$dstfile}});
+    
+    if($dstfile =~ /(\d+)-(\d+)-(\d+)/) {
+       $year = $1;
+       $month = $2;
+       $day = sprintf("%02d",$3);
+    }
+    
+    my $pwd = `pwd`; chomp($pwd);
+    my $cmd = "";
+    if($use_compute_host) { $cmd .= "ssh $compute_host 'cd \"$pwd\";"; }
+    $cmd .= "$make_mkv -t \"$video_title_prefix $year-$month-$day\" -o \"$dstfile\" -i $videos";
+    if($requantize_input_video) { $cmd .= ' -q'; }
+    if($recompress_input_video) { $cmd .= ' -z'; }
+    if($opt_v) { $cmd .= ' -v'; }
+    if($use_compute_host) { $cmd .= "'"; }
+    if(defined $opt_t) {
+       # Print what will be done, but don't actually do it
+       print "\n-> Creating \"$dstfile\"\n";
+       print "$cmd\n";
+       if(!defined $opt_k) { 
+           foreach $video (@{$videos{$dstfile}}) {
+               if(($save_originals) && ($video =~ /\.$originals_file_ext/)) {
+                   print "-> Saving the original video $video\n";
+                   if(index($video, basename(dirname($video))) == -1) { 
+                       print("mv $video \"$origpathname/".basename(dirname($video))."_".basename($video)."\n");
                    } else {
-                       print "-> Removing the original video $video\n";
-                       print("/bin/bash -c '[[ -e $video ]] && rm -f $video'\n");
+                       print("mv $video \"$origpathname/".basename($video)."\n");
                    }
+               } else {
+                   print "-> Removing the original video $video\n";
+                   print("/bin/bash -c '[[ -e $video ]] && rm -f $video'\n");
                }
            }
-       } else {
-           # Create the merged video
-           print "$ext: $cmd" if($opt_v);
-           my $errno = system("$cmd");
-           $errno = $errno >> 8;
-           if($errno) { die "-E- make_mkv encountered some errors with exit code $errno\n"; }
-           system("ls -l \"$srcpathname/\" > /dev/null"); # Make sure the video file is there
-           # Fix the permissions
-           if(-f "$video") { 
-               system("chown $owner \"$video\"");
-               system("chgrp $group \"$video\"");
-               system("chmod $mode \"$video\"");
-           }
-           # Remove the individual video files
-           if(!defined $opt_k) { 
-               foreach $srcvideo (@{$videos{$ext}{$video}}) {
-                   if(($save_originals) && ($srcvideo =~ /\.$originals_file_ext/)) {
-                       print "-> Saving the original video $srcvideo to $origpathname\n";
-                       if(index($srcvideo, basename(dirname($srcvideo))) == -1) { 
-                           system("mv $srcvideo \"$origpathname/".basename(dirname($srcvideo))."_".basename($srcvideo));
-                       } else {
-                           system("mv $srcvideo \"$origpathname/".basename($srcvideo));
-                       }
+       }
+    } else {
+       # Create the merged video
+       print "$cmd" if($opt_v);
+       my $errno = system("$cmd");
+       $errno = $errno >> 8;
+       if($errno) { die "-E- make_mkv encountered some errors with exit code $errno\n"; }
+       system("ls -l \"$srcpathname/\" > /dev/null"); # Make sure the video file is there
+
+       # Fix the permissions
+       system("chown $owner \"$dstfile\"*");
+       system("chgrp $group \"$dstfile\"*");
+       system("chmod $mode \"$dstfile\"*");
+
+       # Remove the individual video files
+       if(!defined $opt_k) { 
+           foreach $srcvideo (@{$videos{$dstfile}}) {
+               if(($save_originals) && ($srcvideo =~ /\.$originals_file_ext/)) {
+                   print "-> Saving the original video $srcvideo to $origpathname\n";
+                   if(index($srcvideo, basename(dirname($srcvideo))) == -1) { 
+                       system("mv $srcvideo \"$origpathname/".basename(dirname($srcvideo))."_".basename($srcvideo));
                    } else {
-                       print "-> Removing the original video $srcvideo\n";
-                       system("/bin/bash -c '[[ -e $srcvideo ]] && rm -f $srcvideo'");
+                       system("mv $srcvideo \"$origpathname/".basename($srcvideo));
                    }
+               } else {
+                   print "-> Removing the original video $srcvideo\n";
+                   system("/bin/bash -c '[[ -e $srcvideo ]] && rm -f $srcvideo'");
                }
            }
        }