-#! /usr/bin/env perl
-
-# $Id$
-# $URL$
+#! /usr/bin/env python2.4
+'''
=head1 NAME
B<fa-flacd> - archive CDs to single FLAC files
B<fa-flacd> [B<-j> I<jobs>] [B<-v>]
-=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;
-
-use File::Basename;
-use Getopt::Long qw(:config gnu_getopt no_ignore_case);
-use POSIX ':sys_wait_h';
-use Pod::Usage;
-
-my $debug;
-my $verbose;
-
-sub verbose {
- $verbose and print(STDERR $_) for @_;
-}
-
-# Return the ARTIST, ALBUM, and DATE followed by a list of all the
-# lines in the file FN.
-sub get_tags {
- my $fn = shift;
- my $tag;
- my $value;
- my $artist;
- my $album;
- my $discnum;
- my @tags;
-
- verbose("Opening tags file $fn\n");
- open(TAGS, $fn) or die("open($fn): $!");
- while (<TAGS>) {
- chomp;
- push(@tags, $_);
-
- ($tag, $value) = split(/=/, $_, 2);
-
- if (/^ARTIST=/i) {
- $artist = $value;
- verbose("ARTIST $artist from $fn\n");
- } elsif (/^ALBUM=/i) {
- $album = $value;
- verbose("ALBUM $album from $fn\n"); # cperl-mode sucks "
- } elsif (/^DISCNUMBER=/i) {
- $discnum = int($value);
- verbose("DISCNUMBER $discnum from $fn\n");
- }
- }
- close(TAGS) or die("close($fn): $!");
-
- return ($artist, $album, $discnum, @tags);
-}
-
-sub track_tags {
- my $h = shift;
- my @result;
-
- while (my ($key, $vall) = each(%$h)) {
- for my $val (@$vall) {
- push(@result, "$key=$val")
- }
- }
-
- return @result;
-}
-
-sub run_flac {
- my $infile = shift;
- my $cue = shift;
- my $outfile = shift;
-
- my @cue;
- if (defined($cue)) {
- @cue = ('--cuesheet', $cue);
- }
-
- verbose("Running flac\n");
- my $status = system('flac', '-o', "$outfile.flac-tmp",
- '--delete-input-file', '-V', '--no-padding', '--best',
- @cue,
- map({ ('-T', $_) } @_),
- $infile);
- if (WIFEXITED($status)) {
- if (($status = WEXITSTATUS($status)) != 0) {
- die("flac exited with status $status");
- }
- } elsif (WIFSIGNALED($status)) {
- die("flac killed with signal ", WTERMSIG($status));
- } elsif (WIFSTOPPED($status)) {
- die("flac stopped with signal ", WSTOPSIG($status));
- } else {
- die("Major horkage on system(flac): \$? = $? \$! = $!");
- }
-
- rename("$outfile.flac-tmp", "$outfile.flac")
- or die("rename($outfile.flac-tmp, $outfile.flac): $!");
-}
-
-# Encode a single wav file to a single flac file, whether the wav and
-# flac files represent individual tracks or whole discs.
-sub flac {
- my $dir = shift;
- my $artist = shift;
- my $album = shift;
- my $discnum = shift;
- my $tracknum = shift;
- my $track_tags = shift;
- my $disc_artist = shift;
- my $single_file = not defined($tracknum);
- my @tags = @_;
- my $outdir;
- my $outfile;
- my $outlog;
- my @files;
-
- for ($artist, $album) {
- s|/|_|g;
- }
-
- verbose("mkdir($artist)\n");
- mkdir($artist) or $!{EEXIST} or die("mkdir($artist): $!");
-
- if (not $single_file) {
- $outdir = "$artist/$album";
- verbose("mkdir($outdir)\n");
- mkdir("$outdir") or $!{EEXIST} or die("mkdir($outdir): $!");
- }
-
- verbose("chdir($dir)\n");
- chdir($dir) or die("chdir($dir): $!");
-
- if ($single_file) {
- $outfile = $album;
- defined($discnum) and $outfile .= " (disc $discnum)";
- run_flac('wav', 'cue', "../$artist/$outfile", @tags);
- $outlog = "../$artist/$outfile.log";
- @files = ("$artist/$outfile.flac");
-
- unlink('cue') or die("unlink(cue): $!");
- rename('log', $outlog)
- or die("rename(log, $outlog): $!");
- } else {
- my $title = join(' ', map(split, @{$track_tags->{'TITLE'}}));
- $title =~ s|/|_|g;
- $outfile = join('/',
- $outdir,
- join(' ',
- (defined($discnum)
- ? sprintf('%02d', $discnum)
- : ()),
- sprintf('%02d', $tracknum),
- $title));
-
- # If we have ARTIST[n] tags for this track, set @track_artist
- # to the empty list; they will go in along with the other [n]
- # tags.
- my @track_artist;
- if (exists($track_tags->{'ARTIST'})) {
- @track_artist = ();
- } else {
- @track_artist = @$disc_artist;
- }
-
- run_flac(sprintf('track%02d.cdda.wav', $tracknum), undef,
- "../$outfile",
- @track_artist,
- @tags,
- "TRACKNUMBER=$tracknum",
- track_tags($track_tags));
- $outlog = "../$outfile.log";
- push(@files, "$outfile.flac");
-
- rename("$tracknum.log", $outlog)
- or die("rename($tracknum.log, $outlog): $!");
- }
-
- chdir('..') or die("chdir(..): $!");
-
- if (-x "$dir/post-processor") {
- verbose(join(' ', "Running ./$dir/post-processor", @files), "\n");
- system("./$dir/post-processor", @files);
- unlink("$dir/post-processor") or die("unlink($dir/post-processor): $!");
- }
-
- # Clean up if we're the last job for $dir; for multi-file dirs,
- # it's possible for more than one job to run cleanup at once, so
- # don't fail if things are already clean.
-# if (nothing but using-tags in $dir) {
-# unlink("$dir/using-tags") or $!{ENOENT} or die("unlink($dir/using-tags): $!");
-# rmdir($dir) or or $!{ENOENT} die("rmdir($dir): $!");
-# }
-
- return 0;
-}
-
-sub flacloop {
- my $MAXJOBS = shift;
- my $dir;
- my @jobs;
-
- # Get a job for Jobs::run. On each call, look for new fa-rip
- # directories and append an item to the queue @jobs for each wav
- # file therein. Then, if we have anything in the queue, return a
- # function to call flac for it, otherwise sleep for a bit. This
- # looks forever, never returning undef, so Jobs::run never returns.
- my $getjob = sub {
- # Look for new fa-rip directories.
- while (1) {
- for my $i (glob('*/tags')) {
- $dir = dirname($i);
-
- verbose("Renaming $dir/tags\n");
- rename("$dir/tags", "$dir/using-tags")
- or die("rename($dir/tags, $dir/using-tags): $!");
-
- my ($artist, $album,
- $discnum, @tags) = get_tags("$dir/using-tags");
- if (-e "$dir/wav") {
- # single-file
- push(@jobs,
- [$dir, $artist, $album, $discnum,
- undef, undef, undef, @tags]);
- } else {
- #multi-file
- # Don't need cue file.
- unlink("$dir/cue") or die("unlink($dir/cue): $!");
-
- # Go over @tags, store all [n] tags in a list keyed by
- # n in %tracks_to_tags, store all ARTIST (not
- # ARTIST[n]) tags in @disc_artist, and leave the rest
- # in @tags.
- my %tracks_to_tags;
- my @disc_artist;
- my @tmp;
- for my $tag (@tags) {
- if ($tag =~ /^([^[]+)\[(\d+)]=(.*)/) {
- push(@{$tracks_to_tags{$2}->{$1}}, $3);
- } elsif ($tag =~ /^ARTIST=/) {
- push(@disc_artist, $tag);
- } else {
- push(@tmp, $tag);
- }
- }
- @tags = @tmp;
-
- push(@jobs,
- map {
- [$dir, $artist, $album, $discnum, $_,
- $tracks_to_tags{$_}, \@disc_artist, @tags]
- } sort(map(int, keys(%tracks_to_tags))));
- }
- }
-
- # Return a job if we found any work.
- if (my $job = shift(@jobs)) {
- return sub {
- my $log = defined($job->[4]) ? $job->[4] . '.log' : 'log';
- $dir = $job->[0];
- open(STDERR, ">$dir/$log") or die("open(STDERR, >$dir/$log): $!");
- return flac(@$job);
- }
- }
-
- # Didn't find anything; wait a while and check again.
- sleep(5);
- }
- };
-
- # Never returns (see $getjob comment).
- Jobs::run('max-jobs'=>$MAXJOBS,
- 'debug'=>$debug,
- 'get-job'=>$getjob,
-
- 'notify-start'=>sub {
- my $pid = shift;
- verbose("new job $pid for $dir\n");
- verbose(scalar(@_), " jobs\n");
- },
-
- 'notify-finish'=>sub {
- my $pid = shift;
- my $status = shift;
- verbose("$pid finished (");
- if (WIFEXITED($status)) {
- verbose('exited with status ', WEXITSTATUS($status));
- } elsif (WIFSIGNALED($status)) {
- verbose('killed with signal ', WTERMSIG($status));
- } elsif (WIFSTOPPED($status)) {
- verbose('stopped with signal ', WSTOPSIG($status));
- }
- verbose(")\n");
- });
-}
-
-MAIN: {
- my $jobs;
- my $help;
-
- $jobs = 4;
- GetOptions(
- 'debug|X' => \$debug,
- 'jobs|j=i' => \$jobs,
- 'verbose|v' => \$verbose,
- 'help|h|?' => \$help,
- ) or pod2usage();
- $help and pod2usage(-exitstatus=>0, -verbose=>1);
-
- flacloop($jobs);
-}
-
-\f
-__END__
-
=head1 DESCRIPTION
B<fa-flacd> and B<fa-rip> together comprise B<flac-archive>, a system
=cut
-# Local variables:
-# cperl-indent-level: 4
-# perl-indent-level: 4
-# indent-tabs-mode: nil
-# End:
+''' #' # python-mode is sucks
+
+import os
+import re
+import sys
+import time
+import traceback
+from errno import EEXIST, ENOENT
+from glob import glob
+from optparse import OptionParser
+
+import org.diplodocus.jobs
+from org.diplodocus.structures import ListDictDict
+from org.diplodocus.util import die, flatten, nothing
+from org.diplodocus.util import catch_EnvironmentError as c
-# vi: set tabstop=4 expandtab:
+def spew(*args):
+ if verbose:
+ for i in args:
+ sys.stderr.write(i)
+
+################################################################################
+# The child processes
+
+def run_flac(infile, cue, outfile, tags):
+ argv = ['flac', '-o', outfile + '.flac-tmp',
+ '--delete-input-file', '-V', '--no-padding', '--best']
+ if cue != None:
+ argv.extend(['--cuesheet', cue])
+ for i in tags:
+ argv.extend(['-T', i])
+ argv.append(infile)
+
+ spew('Running flac\n')
+ status = os.spawnvp(os.P_WAIT, argv[0], argv)
+ if status > 0:
+ die(2, 'flac exited with status ', str(status))
+ elif status < 0:
+ die(2, 'flac killed with signal ', str(abs(status)))
+
+ c(os.rename, outfile + '.flac-tmp', outfile + '.flac')
+
+def flac(dir, artist, album, discnum, tracknum, track_tags, disc_artist, tags):
+ '''Encode a single wav file to a single flac file, whether the wav and
+ flac files represent individual tracks or whole discs.'''
+
+ (artist, album) = [x.replace('/', '_') for x in (artist, album)]
+
+ spew('mkdir(%s)\n' % (artist,))
+ try:
+ c(os.mkdir, artist)
+ except EnvironmentError, error:
+ error.errno == EEXIST or die(2, error.msg, traceback.format_exc())
+
+ if tracknum != None:
+ outdir = '/'.join([artist, album])
+ spew('mkdir(%s)\n' % (outdir,))
+ try:
+ c(os.mkdir, outdir)
+ except EnvironmentError, error:
+ error.errno == EEXIST or die(2, error.msg, traceback.format_exc())
+
+ spew('chdir(%s)\n' % (dir,))
+ c(os.chdir, dir)
+
+ if tracknum == None:
+ outfile = album
+ if discnum != None:
+ outfile = ''.join([outfile, ' (disc ', discnum, ')'])
+ run_flac('wav', 'cue', '/'.join(['..', artist, outfile]), tags)
+ files = ['%s/%s.flac' % (artist, outfile)]
+
+ c(os.unlink, 'cue')
+ outlog = '/'.join(['..', artist, outfile + '.log'])
+ c(os.rename, 'log', outlog)
+ else:
+ title = []
+ for i in track_tags['TITLE']:
+ title.extend(i.split())
+ title = ' '.join(title).replace('/', '_')
+ tmp = []
+ if discnum != None:
+ tmp.append('%02d' % (int(discnum),))
+ tmp.extend(['%02d' % (tracknum,), title])
+ outfile = '/'.join([outdir, ' '.join(tmp)])
+
+ tags = tags[:]
+ # If we have ARTIST[n] tags for this track, they'll go in with
+ # the other [n] tags. Else, prepend disc_artist to tags.
+ if 'ARTIST' not in track_tags:
+ for i in disc_artist:
+ tags.insert(0, i)
+
+ tags.append('TRACKNUMBER=%d' % (tracknum,))
+ tags.extend(flatten([['='.join([key, x]) for x in track_tags[key]]
+ for key in track_tags]))
+
+ run_flac('track%02d.cdda.wav' % (tracknum,), None,
+ '../' + outfile, tags)
+ outlog = ''.join(['../', outfile, '.log'])
+ files = [outfile + '.flac']
+
+ c(os.rename, str(tracknum) + '.log', outlog)
+
+ c(os.chdir, '..')
+
+ post = dir + '/post-processor'
+ if os.path.exists(post):
+ spew('Running ', post); spew(files); spew('\n')
+ files.insert(0, post)
+ os.spawnv(os.P_WAIT, post, files)
+ c(os.unlink, post)
+
+ # Clean up if we're the last job for dir; for multi-file dirs,
+ # it's possible for more than one job to run cleanup at once, so
+ # don't fail if things are already clean.
+ if os.listdir(dir) == ['using-tags']:
+ try:
+ os.unlink(dir + '/using-tags')
+ os.rmdir(dir)
+ except EnvironmentError:
+ pass
+
+ return 0
+
+################################################################################
+# The master process
+
+def get_tags(fn):
+ '''Return the ARTIST, ALBUM, and DATE followed by a list of all the
+ lines in the file FN.'''
+
+ artist = album = discnum = None
+ tags = []
+
+ spew('Opening tags file %s\n' % (fn,))
+ fp = file(fn)
+ for line in (x.rstrip() for x in fp):
+ tags.append(line)
+
+ (tag, value) = line.split('=', 1)
+
+ if re.match(r'ARTIST=', line, re.IGNORECASE):
+ artist = value
+ spew('ARTIST %s from %s\n' % (artist, fn))
+ elif re.match(r'ALBUM=', line, re.IGNORECASE):
+ album = value
+ spew('ALBUM %s from %s\n' % (album, fn))
+ elif re.match(r'DISCNUMBER=', line, re.IGNORECASE):
+ discnum = value
+ spew('DISCNUMBER %s from %s\n' % (discnum, fn))
+
+ return (artist, album, discnum, tags)
+
+def flacloop(maxjobs):
+ dir = [None] # [str] instead of str for lame python closures
+ jobs = []
+
+ # Get a job for jobs.run. On each call, look for new fa-rip
+ # directories and append an item to the queue @jobs for each wav
+ # file therein. Then, if we have anything in the queue, return a
+ # function to call flac for it, otherwise sleep for a bit. This
+ # looks forever, never returning None, so jobs.run never returns.
+ def getjob(reap):
+ # Look for new fa-rip directories.
+ while True:
+ for i in glob('*/tags'):
+ try:
+ dir[0] = os.path.dirname(i)
+
+ spew("Renaming %s/tags\n" % (dir[0],))
+ c(os.rename, dir[0] + '/tags', dir[0] + '/using-tags')
+
+ (artist, album, discnum, tags) = get_tags(dir[0] + '/using-tags')
+ if os.path.exists(dir[0] + '/wav'):
+ # single-file
+ jobs.append([dir[0], artist, album, discnum,
+ None, None, None, tags])
+ else:
+ # multi-file
+ # Don't need cue file.
+ c(os.unlink, dir[0] + '/cue')
+
+ # Go over @tags, store all [n] tags in a list keyed by
+ # n in %tracks_to_tags, store all ARTIST (not
+ # ARTIST[n]) tags in @disc_artist, and leave the rest
+ # in @tags.
+ tracks_to_tags = ListDictDict()
+ disc_artist = []
+ tmp = []
+ for tag in tags:
+ m = re.match(r'([^[]+)\[(\d+)]=(.*)', tag)
+ if m != None:
+ tracks_to_tags.append(int(m.group(2)), m.group(1), m.group(3))
+ elif re.match(r'ARTIST=', tag, re.IGNORECASE):
+ disc_artist.append(tag)
+ else:
+ tmp.append(tag)
+ tags = tmp
+
+ jobs.extend([[dir[0], artist, album, discnum, x,
+ tracks_to_tags[x], disc_artist, tags]
+ for x in sorted(map(int,
+ tracks_to_tags.keys()))])
+ except Exception, error:
+ sys.stderr.write(getattr(error, 'msg', ''))
+ traceback.print_exc()
+ sys.stderr.write('Continuing...\n')
+
+ # Return a job if we found any work.
+ try:
+ job = jobs.pop(0)
+ except IndexError:
+ # Didn't find anything; wait a while and check again.
+ time.sleep(5)
+ reap()
+ continue
+
+ def lamb():
+ log = '/'.join([job[0],
+ job[4] == None and 'log' or str(job[4])
+ + '.log'])
+ try:
+ c(os.dup2, c(os.open, log, os.O_CREAT | os.O_WRONLY), 2)
+ return flac(*job)
+ except EnvironmentError, error:
+ sys.stderr.write(getattr(error, 'msg', ''))
+ traceback.print_exc()
+ return 1
+ return lamb
+
+ def notify_start(pid, jobs):
+ # make this print '2 jobs; start 378 for Artist/01 Title.flac'
+ spew('%d jobs; start %d for %s\n' % (len(jobs), pid, dir[0]))
+
+ def notify_finish(pid, status, jobs):
+ spew('%d jobs; %d finished (' % (len(jobs), pid))
+ if os.WIFEXITED(status):
+ spew('exited with status ', str(os.WEXITSTATUS(status)))
+ elif os.WIFSIGNALED(status):
+ spew('killed with signal ', str(os.WTERMSIG(status)))
+ elif os.WIFSTOPPED(status):
+ spew('stopped with signal ', str(os.WSTOPSIG(status)))
+ spew(')\n')
+
+ # Never returns (see getjob comment).
+ org.diplodocus.jobs.run(maxjobs=maxjobs, debug=debug,
+ get_job=getjob,
+ notify_start=notify_start,
+ notify_finish=notify_finish)
+
+def main(argv):
+ # Control the exit code for any uncaught exceptions.
+ try:
+ parser = OptionParser()
+ parser.disable_interspersed_args()
+ parser.add_option('-X', '--debug', action='store_true', default=False)
+ parser.add_option('-j', '--jobs', type='int', default=1)
+ parser.add_option('-v', '--verbose', action='store_true', default=False)
+ except:
+ traceback.print_exc()
+ return 2
+
+ try:
+ # Raises SystemExit on invalid options in argv.
+ (options, args) = parser.parse_args(argv[1:])
+ except Exception, error:
+ if isinstance(error, SystemExit):
+ return 1
+ traceback.print_exc()
+ return 2
+
+ try:
+ global debug, verbose
+ debug = options.debug
+ verbose = options.verbose
+
+ # Never returns...
+ flacloop(options.jobs)
+ except Exception, error:
+ # but might blow up.
+ if isinstance(error, SystemExit):
+ raise
+ sys.stderr.write(getattr(error, 'msg', ''))
+ traceback.print_exc()
+ return 2
+
+if __name__ == '__main__':
+ sys.exit(main(sys.argv))
-#! /usr/bin/env perl
-
-# $Id$
+#! /usr/bin/env python2.4
+'''
=head1 NAME
B<flac2mp3> - transcode FLAC file to MP3 files
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;
-
-use POSIX ':sys_wait_h';
-use Pod::Usage;
-use Getopt::Long qw(:config gnu_getopt no_ignore_case);
-
-my $flac_options;
-my $lame_options;
-my $quiet;
-my $verbose;
-
-sub run_or_die {
- my $command = shift;
- my $status;
-
- $verbose and print(STDERR "$command\n");
- $status = system($command);
-
- if (WIFEXITED($status)) {
- if (($status = WEXITSTATUS($status)) != 0) {
- die("$command exited with status $status");
- }
- } elsif (WIFSIGNALED($status)) {
- die("$command killed with signal ", WTERMSIG($status));
- } elsif (WIFSTOPPED($status)) {
- die("$command stopped with signal ", WSTOPSIG($status));
- } else {
- die("Major horkage on system($command): \$? = $? \$! = $!");
- }
-}
-
-sub tformat {
- return sprintf('%02d:%02d.%02d', @_);
-}
-
-sub get_decode_args {
- my $fn = shift;
- my @l;
-
- open(F, '-|', 'metaflac', '--export-cuesheet-to=-', $fn);
- while (<F>) {
- /INDEX 01 (\d\d):(\d\d):(\d\d)$/ or next;
- push(@l, [$1, $2, $3]);
- }
-
- my @args;
- for my $i (0..$#l) {
- my $arg = ["--skip=" . tformat(@{$l[$i]})];
- my $next = $l[$i+1];
- if (defined($next)) {
- if ($next->[2] == 0) {
- if ($next->[1] == 0) {
- push(@$arg, '--until=' . tformat($next->[0] - 1, 59, 74));
- } else {
- push(@$arg, '--until=' . tformat($next->[0], $next->[1] - 1,
- 74));
- }
- } else {
- push(@$arg, '--until=' . tformat($next->[0], $next->[1],
- $next->[2] - 1));
- }
- }
- push(@args, $arg);
- }
-
- # If no cue sheet, stick a dummy in here.
- if (@args == 0) {
- @args = ([]);
- }
-
- return @args;
-}
-
-# Return the ARTIST, ALBUM, and DATE tags followed by the TITLE tags
-# in the file FN.
-sub get_tags {
- my $fn = shift;
- my $artists = shift;
- my $titles = shift;
- my $tag;
- my $value;
- 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): $!");
- while (<TAGS>) {
- chomp;
-
- ($tag, $value) = split(/=/, $_, 2);
-
- if (/^ARTIST=/i) {
- $artist = $value;
- } elsif (/^ALBUM=/i) {
- $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.
- } elsif (/^TITLE/i) {
- push(@$titles, $value);
- }
- }
- close(TAGS) or die("close(metaflac --export-vc-to=- $fn): $?");
-
- # 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 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;
- my $outfile;
-
- if ($quiet) {
- $flac_options = '--silent';
- } else {
- $flac_options = '';
- }
-
- if ($lame_options) {
- push(@tmp, $lame_options);
- } else {
- push(@tmp, '--preset standard');
- }
- $quiet and push(@tmp, '--quiet');
- $verbose and push(@tmp, '--verbose');
- $lame_options = join(' ', @tmp);
-
- # We'll be putting these in single quotes, so we need to escape
- # 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) {
- 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 $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,
- 'help|h|?' => \$help,
- ) or pod2usage();
- $help and pod2usage(-exitstatus=>0, -verbose=>1);
-
- @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);
-
- # 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
-__END__
-
=head1 OPTIONS
=over 4
=cut
-# Local variables:
-# cperl-indent-level: 4
-# perl-indent-level: 4
-# indent-tabs-mode: nil
-# End:
+''' #' # python-mode is sucks
-# vi: set tabstop=4 expandtab:
+import re, sys, traceback
+from optparse import OptionParser
+from subprocess import Popen, PIPE
+
+import org.diplodocus.jobs
+from org.diplodocus.util import run_or_die
+
+################################################################################
+# The child processes
+
+def flac2mp3(fn, title, artist, album, date, track, skip_until):
+ (title, artist, album, date) = [(x == None and 'unknown') or x
+ for x in (title, artist, album, date)]
+ track = int(track) # XXX caller should int this
+ try:
+ (skip_arg, until_arg) = skip_until
+ except ValueError:
+ skip_arg = until_arg = ''
+
+ if quiet:
+ flac_options = '--silent'
+ else:
+ flac_options = ''
+
+ tmp = []
+ global lame_options
+ if lame_options != None:
+ tmp.append(lame_options)
+ else:
+ tmp.append('--preset standard');
+ quiet and tmp.append('--quiet')
+ verbose and tmp.append('--verbose')
+ lame_options = ' '.join(tmp)
+
+ # Escape any single quotes ' so we can quote this.
+ (fn, title, artist,
+ album, date) = [x.replace("'", r"'\''")
+ for x in (fn, title, artist, album, date)]
+
+ outfile = ('%s (%s) %02d %s.mp3' % (artist, album,
+ track, title)).replace('/', '_')
+
+ run_or_die(3, "flac %s -cd %s %s '%s' | lame %s --tt '%s' --ta '%s' --tl '%s' --ty '%s' --tn %d - '%s'"
+ % (flac_options, skip_arg or '', until_arg or '', fn,
+ lame_options, title, artist, album, date, track, outfile))
+
+ return 0
+
+################################################################################
+# The master process
+
+def tformat(m, s, c):
+ return '%02d:%02d.%02d' % (m, s, c)
+
+def get_decode_args(fn):
+ l = []
+
+ p = Popen(['metaflac', '--export-cuesheet-to=-', fn], stdout=PIPE)
+ for line in (x.rstrip() for x in p.stdout):
+ m = re.search(r'INDEX 01 (\d\d):(\d\d):(\d\d)$', line)
+ if m != None:
+ l.append(map(int, m.groups()))
+ status = p.wait()
+ # XXX dataloss! check status
+
+ args = []
+ for i in xrange(len(l)):
+ arg = ['--skip=' + tformat(*l[i])]
+ try:
+ next = l[i + 1]
+ except IndexError:
+ next = None
+ if next != None:
+ if next[2] == 0:
+ if next[1] == 0:
+ arg.append('--until=' + tformat(next[0] - 1, 59, 74))
+ else:
+ arg.append('--until=' + tformat(next[0], next[1] - 1,
+ 74))
+ else:
+ arg.append('--until=' + tformat(next[0], next[1],
+ next[2] - 1))
+
+ args.append(arg)
+
+ # If no cue sheet, stick a dummy in here.
+ if len(args) == 0:
+ args = [[]]
+
+ return args
+
+def get_tags(fn):
+ '''Return the ARTIST, ALBUM, and DATE tags followed by the TITLE tags
+ in the file FN.'''
+
+ date = discnum = track = None
+ artists = []
+ titles = []
+
+ p = Popen(['metaflac', '--export-vc-to=-', fn], stdout=PIPE)
+ for line in (x.rstrip() for x in p.stdout):
+ (tag, value) = line.split('=', 1)
+
+ if re.match(r'ARTIST=', line, re.IGNORECASE):
+ artist = value
+ elif re.match(r'ALBUM=', line, re.IGNORECASE):
+ album = value
+ elif re.match(r'DATE=', line, re.IGNORECASE):
+ date = value
+ elif re.match(r'DISCNUMBER=', line, re.IGNORECASE):
+ discnum = value
+ elif re.match(r'TRACKNUMBER=', line, re.IGNORECASE):
+ track = value
+
+ # XXX Need to support per-track of everything but album and
+ # discnum, not just TITLE... Also unify with fa-flacd get_tags.
+ elif re.match(r'ARTIST\[', line, re.IGNORECASE):
+ artists.append(value)
+ elif re.match(r'TITLE', line, re.IGNORECASE):
+ titles.append(value)
+ # XXX dataloss! check status
+ status = p.wait()
+
+ # If no TITLEs, stick a dummy in here.
+ if len(titles) == 0:
+ titles.append(None)
+
+ return (artist, album, date, discnum, track, artists, titles)
+
+def main(argv):
+ # Control the exit code for any uncaught exceptions.
+ try:
+ parser = OptionParser()
+ parser.disable_interspersed_args()
+ parser.add_option('-X', '--debug', action='store_true', default=False)
+ parser.add_option('-j', '--jobs', type='int', default=1)
+ parser.add_option('--lame-options')
+ parser.add_option('-q', '--quiet', action='store_true', default=False)
+ parser.add_option('-v', '--verbose', action='store_true', default=False)
+ except:
+ traceback.print_exc()
+ return 2
+
+ try:
+ # Raises SystemExit on invalid options in argv.
+ (options, args) = parser.parse_args(argv[1:])
+ except Exception, error:
+ if isinstance(error, SystemExit):
+ return 1
+ traceback.print_exc()
+ return 2
+
+ try:
+ global debug, flac_options, lame_options, quiet, verbose
+ debug = options.debug
+ lame_options = options.lame_options
+ quiet = options.quiet
+ verbose = options.verbose
+
+ jobs = []
+ for fn in args:
+ try:
+ args = get_decode_args(fn)
+ (artist, album, date, discnum, track,
+ artists, titles) = get_tags(fn)
+
+ # lame doesn't seem to support disc number.
+ if discnum != None:
+ album = '%s (disc %d)' % (album, 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.
+ if track == None:
+ track = 1
+
+ for i in range(len(titles)):
+ if len(artists) > 0:
+ artist = artists[i]
+ jobs.append([fn, titles[i], artist, album,
+ date, track, args[i]])
+ track = i + 2
+ except Exception, error:
+ sys.stderr.write(getattr(error, 'msg', ''))
+ traceback.print_exc()
+ sys.stderr.write('Continuing...\n')
+
+ def getjob(reap):
+ try:
+ job = jobs.pop(0)
+ except IndexError:
+ return
+ return lambda: flac2mp3(*job)
+ org.diplodocus.jobs.run(maxjobs=options.jobs, debug=debug, get_job=getjob)
+ except Exception, error:
+ if isinstance(error, SystemExit):
+ raise
+ # check all print_exc and format_exc in fa-flacd.py; i think
+ # for some i don't do this msg print check
+ sys.stderr.write(getattr(error, 'msg', ''))
+ traceback.print_exc()
+ return 2
+
+ return 0
+
+if __name__ == '__main__':
+ sys.exit(main(sys.argv))
projects / flac-archive / commitdiff