]> diplodocus.org Git - flac-archive/blobdiff - flac2mp3
Fix a job in Jobs::run (@jobs < $maxjobs, not <=) and icky icky
[flac-archive] / flac2mp3
index 6a496bf7aae3461d426b22d4e2d771e769718098..842f0cd8041e9e61eb554258c0c50bc44ef13a6e 100755 (executable)
--- a/flac2mp3
+++ b/flac2mp3
@@ -8,17 +8,118 @@ B<flac2mp3> - transcode FLAC file to MP3 files
 
 =head1 SYNOPSIS
 
-B<flac2mp3> [B<--lame-options> I<lame-options>] [B<-q>] [B<-v>] I<file>
+B<flac2mp3> [B<--lame-options> I<lame-options>] [B<-j> I<jobs>] [B<-q>] [B<-v>] I<file> [...]
 
 =head1 DESCRIPTION
 
-B<flac2mp3> transcodes the FLAC file I<file> to MP3 files.  I<file> is
-the kind of FLAC file B<fa-flacd> generates.  That is, it contains a
-cue sheet, one TITLE tag per track listed therein, and ARTIST, ALBUM,
-and DATE tags.
+B<flac2mp3> transcodes the FLAC files I<file> to MP3 files.  I<file>
+may be the kind of FLAC file B<fa-flacd> generates.  That is, it
+contains a cue sheet, one TITLE tag per track listed therein, and
+ARTIST, ALBUM, and DATE tags.
 
 =cut
 
+package Jobs;
+
+use strict;
+use warnings;
+
+use Errno;
+use POSIX ':sys_wait_h';
+
+sub newjob {
+    my $f = shift;
+    my $jobs = shift;
+    my $debug = shift;
+    my $pid;
+
+    if (not $debug) {
+        $pid = fork();
+        if (not defined($pid)) {
+            die("fork: $!");
+        }
+    }
+
+    if ($debug or $pid == 0) {
+        exit($f->());
+    }
+
+    if ($pid == 0) {
+        exit($f->());
+    }
+
+    push(@$jobs, $pid);
+
+    return $pid;
+}
+
+sub deljob {
+    my $pid = shift;
+    my $status = shift;
+    my $jobs = shift;
+
+    for (my $i = 0; $i <= $#$jobs; $i++) {
+        if ($pid == $jobs->[$i]) {
+            splice(@$jobs, $i, 1);
+            last;
+        }
+    }
+
+    return ($pid, $status);
+}
+
+sub run {
+    my %o = @_;
+    my $maxjobs = $o{'max-jobs'};
+    my $get_job = $o{'get-job'};
+    my $notify_start = $o{'notify-start'};
+    my $notify_finish = $o{'notify-finish'};
+    my @jobs;
+    my $pid;
+
+    # Call notifier function if given.
+    sub call {
+        my $f = shift or return;
+        ref($f) eq 'CODE' or return;
+        $f->(@_);
+    }
+
+    while (1) {
+        if (@jobs < $maxjobs) {
+            my $job;
+            while (defined($job = $get_job->())) {
+                $pid = newjob($job, \@jobs, $o{'debug'});
+                call($notify_start, $pid, @jobs);
+                @jobs < $maxjobs or last;
+            }
+
+            # No jobs running and get-job returned undef; we're finished.
+            if (@jobs == 0 and not defined($job)) {
+                return;
+            }
+        }
+
+        # Now running as many jobs as we can, block waiting for one to die.
+        do {
+            $pid = waitpid(-1, 0);
+        } while ($pid == 0
+                 or ($pid == -1 and ($!{ECHILD} or $!{EINTR})));
+        $pid == -1 and die("waitpid(-1): $!");
+
+        # Before starting more, see if any others have finished.
+        do {
+            call($notify_finish, deljob($pid, $?, \@jobs), @jobs);
+        } while (($pid = waitpid(-1, WNOHANG)) > 0);
+        if ($pid == -1) {
+            $!{ECHILD} or $!{EINTR} or die("waitpid(-1): $!");
+        }
+    }
+}
+
+\f
+################################################################################
+package main;
+
 use strict;
 use warnings;
 
@@ -85,6 +186,11 @@ sub get_decode_args {
         push(@args, $arg);
     }
 
+    # If no cue sheet, stick a dummy in here.
+    if (@args == 0) {
+        @args = ([]);
+    }
+
     return @args;
 }
 
@@ -99,6 +205,8 @@ sub get_tags {
     my $artist;
     my $album;
     my $date;
+    my $discnum;
+    my $track;
 
     open(TAGS, '-|', 'metaflac', '--export-vc-to=-', $fn)
       or die("open(metaflac --export-vc-to=- $fn): $!");
@@ -113,8 +221,12 @@ sub get_tags {
             $album = $value;
         } elsif (/^DATE=/i) {
             $date = $value;
+        } elsif (/^DISCNUMBER=/i) {
+            $discnum = int($value);
         } elsif (/^ARTIST\[/i) {
             push(@$artists, $value);
+        } elsif (/^TRACKNUMBER=/i) {
+            $track = $value;
 
         # Intentionally don't match the = on this one, to support the
         # TITLE[1] .. TITLE[n] tag style.
@@ -124,16 +236,32 @@ sub get_tags {
     }
     close(TAGS) or die("close(metaflac --export-vc-to=- $fn): $?");
 
-    return ($artist, $album, $date);
+    # If no TITLEs, stick a dummy in here.
+    if (@$titles == 0) {
+        push(@$titles, undef);
+    }
+
+    return ($artist, $album, $date, $discnum, $track);
+}
+
+sub arg {
+    my $arg = shift;
+    my $var = shift;
+
+    if (defined($$var)) {
+        $$var = "$arg '$$var'";
+    } else {
+        $$var = ''
+    }
 }
 
 sub flac2mp3 {
     my $fn = shift;
-    my $title = shift;
-    my $artist = shift;
-    my $album = shift;
-    my $date = shift;
-    my $track = shift;
+    my $title = (shift or 'unknown');
+    my $artist = (shift or 'unknown');
+    my $album = (shift or 'unknown');
+    my $date = (shift or 'unknown');
+    my $track = int(shift);
     my $skip_arg = shift;
     my $until_arg = shift;
     my @tmp;
@@ -158,21 +286,32 @@ sub flac2mp3 {
     # any single quotes in the filename by closing the quote ('),
     # putting an escaped quote (\'), and then reopening the quote (').
     for ($fn, $title, $artist, $album, $date) {
-        s/'/'\\''/g;
+        defined and s/'/'\\''/g;
     }
 
     $outfile = sprintf("$artist ($album) \%02s $title.mp3", $track);
     $outfile =~ s/\//_/g;
 
+    arg('--tt', \$title);
+    arg('--ta', \$artist);
+    arg('--tl', \$album);
+    arg('--ty', \$date);
+    arg('--tn', \$track);
+
+    $skip_arg ||= '';
     $until_arg ||= '';
     run_or_die(join(' ', "flac $flac_options -cd $skip_arg $until_arg '$fn'",
-                    " | lame $lame_options --tt '$title' --ta '$artist'",
-                    " --tl '$album' --ty '$date' --tn $track - '$outfile'"));
+                    " | lame $lame_options $title $artist $album $date $track",
+                    " - '$outfile'"));
 }
 
 MAIN: {
     my $help;
+    my $debug;
+    my $maxjobs = 1;
     GetOptions(
+               'debug|X' => \$debug,
+               'jobs|j=i' => \$maxjobs,
                'lame-options=s', \$lame_options,
                'quiet|q' => \$quiet,
                'verbose|v' => \$verbose,
@@ -180,15 +319,37 @@ MAIN: {
               ) or pod2usage();
     $help and pod2usage(-exitstatus=>0, -verbose=>1);
 
-    my $fn = shift or pod2usage();
-    my @args = get_decode_args($fn);
-    my (@artists, @titles);
-    my ($artist, $album, $date) = get_tags($fn, \@artists, \@titles);
+    @ARGV > 0 or pod2usage();
+
+    my @jobs;
+    for my $fn (@ARGV) {
+        my @args = get_decode_args($fn);
+        my (@artists, @titles);
+        my ($artist, $album, $date, $discnum, $track) = get_tags($fn, \@artists,
+                                                                 \@titles);
 
-    for my $i (0..$#titles) {
-        flac2mp3($fn, $titles[$i], ($artists[$i] or $artist), $album, $date,
-                 $i + 1, @{$args[$i]});
+        # lame doesn't seem to support disc number.
+        defined($discnum) and $album .= " (disc $discnum)";
+
+        # Stupid hack: only a single-track file should have the
+        # TRACKNUMBER tag, so use it if set for the first pass through
+        # the loop.  At the end of the loop, we'll set $track for the
+        # next run, so this continues to work for multi-track files.
+        $track ||= 1;
+
+        for my $i (0..$#titles) {
+            push(@jobs, [$fn, $titles[$i], ($artists[$i] or $artist), $album,
+                         $date, $track, @{$args[$i]}]);
+            $track = $i + 2;
+        }
     }
+
+    Jobs::run('max-jobs'=>$maxjobs,
+              'debug'=>$debug,
+              'get-job'=>sub {
+                  my $job = shift(@jobs) or return;
+                  return sub { flac2mp3(@$job) }
+              });
 }
 
 \f
@@ -204,6 +365,10 @@ Pass I<lame-options> to B<lame>.  This ends up being passed to the
 shell, so feel free to take advantage of that.  You'll almost
 certainly have to put I<lame-options> in single quotes.
 
+=item B<-j> [B<--jobs>] I<jobs>
+
+Run up to I<jobs> jobs instead of the default 1.
+
 =item B<-q> [B<--quiet>]
 
 Suppress status information.  This option is passed along to B<flac>