#!/usr/bin/perl
#nathan jahnke <njahnke@gmail.com>
use warnings;
use strict 'subs';
package sasami;
#system("export DYLD_LIBRARY_PATH=/Volumes/l/anrix");
my $really_quiet = "";
$really_quiet = "-really-quiet";
#to move to anri config file later
$gba_crop_left = 43;
$gba_crop_top = 39;
$gba_crop_right = 44;
$gba_crop_bottom = 41;
$gb_crop_left = 80;
$gb_crop_top = 48;
$gb_crop_right = 80;
$gb_crop_bottom = 48;
sub pipe_prepare($) {
$basename = shift;
$avs = "${basename}.pipe";
system("rm ${avs}") if -e $avs;
system("mkfifo ${avs}");
open(AVS, "${basename}.avs"); #must be unix text when running sasami under unix ...
@avs = <AVS>;
close(AVS);
@sources = ();
@trims = ();
for (@avs) {
chomp;
next if m/^#/;
push(@sources,"${1}.VOB") if m/^.*MPEG2source\("(.+)\.d2v".*$/;
push(@sources,$1) if m/^.*avisource\("([^"]+)".*$/;
push(@sources,$1) if m/^.*directshowsource\("([^"]+)".*$/;
$fieldorder = 1 if m/^AssumeTFF\(\)/;
$fieldorder = 0 if m/^AssumeBFF\(\)/;
$d = 1 if m/^global d1 = true/;
$d = 4 if m/^global d1 = false/;
$f = $1 if m/^changefps\(f1\/([0-9])\)/;
$deflicker = 1 if m/^nate_retard_bob_2/;
$vhs = 1 if m/^nate_vhs_head_change_erase/;
$onepixel = 1 if m/^nate_1_pixel_bob_fix/;
$nes = 1 if m/^nate_nes$/;
push(@trims,"$1,$2") if m/trim\(([0-9]+),([0-9]+)\)/;
$gba = 1 if m/^nate_gba$/;
$gb = 1 if m/^nate_gb$/;
$resize = 1 if m/^lanczos4resize\(d1,last\.height\)/;
$statidlines[0] = $1 if m/^statid1="""(.*)"""$/;
$statidlines[1] = $1 if m/^statid2="""(.*)"""$/;
$statidlines[2] = $1 if m/^statid3="""(.*)"""$/;
}
die 'sasami: no sources detected in .avs!' if @sources == 0;
$newframerate = ($pal ? (50/$f) : (59.94/$f)); #required for trimming
@mencoderargslist = ();
my $sourcelength = 0;
my $previous_sourcelength = 0;
my @trimargsarr = ();
foreach $source (@sources) {
#get vital info about source
$mplayeridentify = <<SABRAC;
mplayer -identify -vo null -slave ${source} 2>&1 <<ENDOFSCRIPT
quit
ENDOFSCRIPT
SABRAC
for (qx($mplayeridentify)) {
chomp;
$width = $_ if s/ID_VIDEO_WIDTH=([0-9]+)/$1/;
$height = $_ if s/ID_VIDEO_HEIGHT=([0-9]+)/$1/;
$framerate = $_ if s/ID_VIDEO_FPS=([0-9.]+)/$1/;
$maxaudiobitrate = $_ if s/ID_AUDIO_BITRATE=([0-9]+)/$1/;
$sourcelength += ($framerate * $_) if s/ID_LENGTH=([0-9.]+)/$1/;
}
# print "$width\n";
# print "$height\n";
# print "$framerate\n";
# print "$maxaudiobitrate\n";
$pal = ($height eq "576");
#build mencoder command to feed fifo
$mencoderargs = "./mencoder ${really_quiet}";
#trimming
# $framerate = 29.97;
# $framerate = 25 if $pal;
### BUG: a trim in a later source followed by a trim in an earlier source does not work as expected
my $source_intermediate = 1;
$source_intermediate = 0 if !@trims;
my $ss = 0;
my $sourcecount = 1; #need to immediately cat a source declaration if a trimpoint is identified
my $trimargs = '';
for (@trims) {
m/^(.*),(.*)$/;
if ($1 || $2) {
$source_intermediate = 0; #1 for killing trailing unused sources - there are still trim points so don't need to worry about that this time around
}
if ($1) {
if ($1 ne 0) { #no need to do anything if it's the initial frame - unlike in avisynth, mencoder's first trim argument is optional
$source_intermediate = 1; #this source might have been trimmed out - see if ($2) block below for more
if ($1 < $sourcelength) {
if ($sourcecount) {
push(@trimargsarr, $trimargs) if $trimargs;
$trimargs = '';
$sourcecount = 0;
}
print "($1 / $framerate) - $previous_sourcelength\n";
$trimargs .= " -ss ".
(
$1 - $previous_sourcelength
) / $framerate;
$ss = ($1 - $previous_sourcelength); #remember how far in we started since -endpos is relative to this
$_ = ",$2"; #wipe this trim point out so we don't accidentally use it twice
$source_intermediate = 0; #this source was not trimmed out
}
} else {
$_ = ",$2"; #wipe this trim point out
$source_intermediate = 0; #this source was not trimmed out
}
}
if ($2) {
if ($2 ne 0) { #no need to do anything if it's final frame - unlike avisynth, mencoder's second trim argument is optional
if (!($1)) {
$source_intermediate = 1; #this source was trimmed out
}
if ($2 < $sourcelength) {
#have to insert a conversion factor ($newframerate / $framerate) to handle the fact that the output framerate can differ from the input framerate - -endpos is not based on output framerate while -ss is
print "((($2 - $previous_sourcelength) - $ss) / $framerate) * ($newframerate / $framerate)\n";
$trimargs .= " -endpos ".
(
(
($2 - $previous_sourcelength) - $ss
) / $framerate
)
*
($newframerate / $framerate);
$_ = ","; #wipe this trim point out so we don't accidentally use it twice
$source_intermediate = 0; #this source was not trimmed out
$sourcecount = 1; #this source declaration can't take any more -ss and/or -endpos
}
} else {
$_ = ","; #wipe this trim point out
$source_intermediate = 0; #this source was not trimmed out
}
}
}
$previous_sourcelength = $sourcelength;
next if $source_intermediate; #this source was trimmed out entirely
push(@trimargsarr, $trimargs) if $trimargs; #in case we had no complete trims
#one pixel bob
if ($onepixel) {
$mencoderargs .= " -vf-add crop=${width}:".($height-1).":0:0,expand=${width}:${height}:0:1";
$fieldorder = 0 if $fieldorder == 1;
$fieldorder = 1 if $fieldorder == 0;
}
#nes
$mencoderargs .= " -vf-add crop=".($width-8).":${height}:8:0,expand=${width}:${height}:8:0" if $nes;
#-af channels=6:4:0:0:0:1:0:2:0:3 media.avi
#Would change the number of channels to 6 and set up 4 routes that copy channel 0 to channels 0 to 3. Channel 4 and 5 will contain silence.
$mencoderargs .= " -vf-add yadif=3 -vf-add mcdeint=1:${fieldorder}" if $d == 1;
$tfields = 0;
$tfields = 4 if $deflicker;
$mencoderargs .= " -vf-add tfields=${tfields}" if $d == 4;
#i just eyeballed this - someone will need to sync the avisynth and mencoder sharpen filters
$mencoderargs .= " -vf-add unsharp=l3x3:0.6:c3x3:0.6" if $deflicker;
$mencoderargs .= " -vf-add crop=${width}:".($height-12).":0:0,expand=0:-12:0:0" if $vhs;
if ($gba) {
$mencoderargs .= " -vf-add scale=320:240";
$mencoderargs .= " -vf-add crop=".(320-$gba_crop_right-$gba_crop_left).":".(240-$gba_crop_bottom-$gba_crop_top).":${gba_crop_left}:${gba_crop_top}";
$mencoderargs .= " -vf-add scale=240:160";
$newwidth = 240;
$newheight = 160;
} elsif ($gb) {
$mencoderargs .= " -vf-add scale=320:240";
$mencoderargs .= " -vf-add crop=".(320-$gb_crop_right-$gb_crop_left).":".(240-$gb_crop_bottom-$gb_crop_top).":${gb_crop_left}:${gb_crop_top}";
$newwidth = (320-$gb_crop_right-$gb_crop_left);
$newheight = (240-$gb_crop_bottom-$gb_crop_top);
}
$x264newframerate = "--fps ${newframerate}";
$mp4boxnewframerate = "-fps ${newframerate}";
$mencoderargs .= " -vf-add framestep=${f}";
$newwidth = (($d == 1) ? ($pal ? 704 : 640) : ($pal ? 352 : 320)) unless $newwidth;
$newheight = (($d == 1) ? ($pal ? 576 : 480) : ($pal ? 288 : 240)) unless $newheight;
$mencoderargs .= " -vf-add scale=${newwidth}:${newheight}" if $resize;
$mencoderargs .= " -vf-add format=i420 -ovc raw -of rawvideo -nosound";
my $mencoderargstemp = '';
while (my $trimargs = shift(@trimargsarr)) {
$mencoderargstemp .= $mencoderargs." ${source}".$trimargs." -o ${avs}\n";
}
$mencoderargs = $mencoderargstemp;
push(@mencoderargslist,$mencoderargs);
}
if (defined $statidlines[0]) {
use Imager;
$img = Imager->new;
$img->read(file=>"ntsc_d1.tga") or die "Cannot load statid image: ", $img->errstr;
$font = Imager::Font->new(file=>"/Library/Fonts/Verdana.ttf"); #will need to be changed for non-mac unix ...
&statid_lines_write($statidlines[0], $img->getheight/2, 36, 'white' ) if defined $statidlines[0];
&statid_lines_write($statidlines[1], 360, 36, 'white' ) if defined $statidlines[1];
&statid_lines_write($statidlines[2], 400, 36, 'white' ) if defined $statidlines[2];
&statid_lines_write("Audio commentary on track 2", 460, 28, '#E1CE8B') if defined $audio_commentary;
$img->write(file=>'statid.tga')
or die 'Cannot save statid.tga: ', $img->errstr;
open(MF,">statid.txt") or die "Could not open statid.txt for appending!";
for (1..5) { #once for each second of the statid
for (1..$newframerate) { #once for each frame this second
print MF "statid.tga\n";
}
}
close(MF);
$statidmencoderargs = "./mencoder ${really_quiet} -vf-add scale=${newwidth}:${newheight} -vf-add format=i420 -ovc raw -of rawvideo -nosound -ofps ${newframerate} mf://\@statid.txt -mf fps=${newframerate}:type=tga -o ${avs}";
}
}
sub pipe_write {
open(FIFO,">${avs}");
print("$statidmencoderargs\n") if defined $statidlines[0];
system($statidmencoderargs) if defined $statidlines[0];
for (@mencoderargslist) {
print("$_\n");
system($_);
}
print("$statidmencoderargs\n") if defined $statidlines[0];
system($statidmencoderargs) if defined $statidlines[0];
close(FIFO);
}
$basename='pokemon'; #remove in anri
for(my $N = 1; $N <= 2 ; $N++) {
&pipe_prepare($basename);# if $os eq 'unix'; #replace with $_[4] or whatever
if ($N == 2) {
# this is a hack to deal with the fact that sometimes x264 takes a while to finish encoding and write out the statsfile - thanks grenola!
while (!(-e "${basename}.stats")) {
print("Waiting for stats file ...\n");
sleep(1);
}
}
$x264command = "./x264 --pass ${N} --stats ${basename}.stats --bitrate 5000 --qpmin 0 ${x264newframerate} -o ${basename}.264 ${avs} ${newwidth}x${newheight} &";
print("$x264command\n");
system($x264command);
&pipe_write;
}
$mp4boxcommand = "./mp4box ${mp4boxnewframerate} -new -add ${basename}.264 ${basename}.mp4";
print ("$mp4boxcommand\n");
system($mp4boxcommand);
exit 0;
#statid line text, y-offset, text size, color
sub statid_lines_write {
$font->align(string => $_[0],
aa => 1,
size => $_[2],
color => $_[3],
x => $img->getwidth/2,
y => $_[1],
halign => 'center',
valign => 'center',
image => $img);
}
