sort customEvents just once
[HTML5TV.git] / bin / mplayer.pl
index 9d60b39..9dd342d 100755 (executable)
@@ -4,15 +4,20 @@ use warnings;
 use strict;
 
 use IPC::Open3 qw(open3);
-use IO::Epoll;
+use IO::Select;
 use Data::Dump qw(dump);
 use File::Slurp;
 use YAML;
 use JSON;
 use HTML::TreeBuilder;
-use Imager;
+use Graphics::Magick;
+use Time::HiRes qw(time);
+use File::Path qw(rmtree);
 
-my $debug = 0;
+use lib 'lib';
+use HTML5TV::Slides;
+
+my $debug = $ENV{DEBUG} || 0;
 
 my $movie = shift @ARGV;
 
@@ -56,6 +61,8 @@ my $preroll = 3;
 
 my $slide_factor = 4; # 1/4 size of video
 
+my $min_slide_height = 480;
+
 our $to_mplayer;
 our $from_mplayer;
 our $err_mplayer;
@@ -64,24 +71,23 @@ our $prop = {};
 my $pid = open3( $to_mplayer, $from_mplayer, $err_mplayer,
         'mplayer',
                '-slave', '-idle',
-               '-quiet',
+#              '-quiet',
+#              '-msglevel', 'demux=9', '-msgmodule',
                '-edlout', $edl,
                '-osdlevel', 3,
                '-vf' => 'screenshot',
 );
 
-my $epfd = epoll_create(10);
-
-epoll_ctl($epfd, EPOLL_CTL_ADD, fileno STDIN         , EPOLLIN  ) >= 0 || die $!;
-epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $from_mplayer , EPOLLIN  ) >= 0 || die $!;
-#epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $to_mplayer   , EPOLLOUT ) >= 0 || die $!;
+my $select = IO::Select->new();
+#$select->add( \*STDIN );
+$select->add( $from_mplayer );
+#$select->add( $err_mplayer );
 
 sub load_subtitles;
 
 sub load_movie {
        warn "$movie ", -s $movie, " bytes $edl\n";
        print $to_mplayer qq|loadfile "$movie"\n|;
-       print $to_mplayer "get_property $_\n" foreach ( qw/metadata video_codec video_bitrate width height fps length/ );
        load_subtitles;
 }
 
@@ -133,6 +139,18 @@ sub slide_jpg {
        sprintf "%s/s/%d/%03d.jpg", $media_dir, @_;
 }
 
+sub oggThumb {
+       my $video = shift;
+       my $file  = shift;
+       my $t = join(',', @_);
+       system "oggThumb -t $t -o jpg -n $file $video";
+}
+
+sub fmt_mmss {
+       my $t = shift;
+       return sprintf('%02d:%02d', int($t/60), int($t%60));
+}
+
 sub html5tv {
 
        if ( ! $prop->{width} || ! $prop->{height} ) {
@@ -145,12 +163,14 @@ sub html5tv {
                return;
        }
 
+       warn "html5tv";
 
        my $sync;
 
        my @slide_t;
 
        my @videos;
+       my @frames;
 
        foreach my $s ( @subtitles ) {
                push @{ $sync->{htmlEvents}->{'#subtitle'} }, {
@@ -171,18 +191,8 @@ sub html5tv {
                                push @videos, [ @$s, $video ];
                        }
                } elsif ( $s->[2] =~ m{slide:(\d+)\s+shot:(\d+\.\d+)} ) {
-                       my ( $nr, $t ) = ( $1, $2 );
-
-                       my $hires = "$media_dir/s/hires";
-                       mkdir $hires unless -e $hires;
-
-                       my $shot_path = sprintf "$hires/s%03d.jpg", $nr;
-                       if ( ! -e $shot_path ) {
-                               my $frame_dir = "$media_dir/s/shot/";
-                               system "mplayer -vo jpeg:outdir=$frame_dir,quality=95 -frames 1 -ss $t -ao null -really-quiet $movie";
-                               rename "$media_dir/s/shot/00000001.jpg", $shot_path;
-                               warn "created $shot_path from $movie at $t for slide $nr\n";
-                       }
+                       push @frames, [ $2, $1 ];
+                       next;
                }
 
                next unless $s->[2] =~ m{\[(\d+)\]};
@@ -205,6 +215,21 @@ sub html5tv {
                push @slide_t, $s->[0];
        }
 
+       if ( @frames ) {
+               rmdir $_ foreach glob "$media_dir/s/[124]";
+               my $hires = "$media_dir/s/hires";
+               mkdir $hires unless -e $hires;
+
+               oggThumb $movie, "$hires/.f%.jpg", map { $_->[0] } @frames;
+
+               foreach my $i ( 0 .. $#frames ) {
+                       my $from = "$hires/.f$i.jpg";
+                       my $to   = "$hires/f" . $frames[$i]->[1] . '.jpg';
+                       rename $from, $to || warn "can't rename $from -> $to: $!";
+               }
+
+       }
+
        foreach ( 0 .. $#slide_t ) {
                my $slide_nr = $_ + 1;
                push @{ $sync->{htmlEvents}->{'#slide'} }, {
@@ -216,54 +241,62 @@ sub html5tv {
 
        my @slides_hires = glob "$media_dir/s/hires/*";
 
-       foreach my $factor ( 4, 2, 1 ) {
-               my $w = $prop->{width}  / $factor;
-               my $h = $prop->{height} / $factor;
+       my $path = "$media_dir/s";
 
-               my $path = "$media_dir/s";
-               if ( ! -d $path ) {
-                       warn "create slides imaes in $path";
-                       mkdir $path;
-               }
+       if ( ! -d $path ) {
+               warn "create slides images in $path";
+               mkdir $path;
+       }
 
-               $path .= '/' . $factor;
+       foreach my $hires ( @slides_hires ) {
 
-               if ( ! -d $path ) {
-                       mkdir $path;
-                       warn "created $path\n";
+               my $im = Graphics::Magick->new;
+               $im->ReadImage( $hires );
 
-               }
+               my ($slide_width, $slide_height) = Graphics::Magick->new->Ping( $hires );
+               my $slide_aspect = $slide_width / $slide_height;
+
+               my $nr = $1 if $hires =~ m{(\d+)\.\w+$} || warn "can't find number in $hires";
+               next unless $nr;
+
+               foreach my $factor ( 1, 2, 4 ) {
 
-               foreach my $hires ( @slides_hires ) {
+                       mkdir "$path/$factor" unless -e "$path/$factor";
 
-                       my $nr = $1 if $hires =~ m{(\d+)\.\w+$} || warn "can't find number in $hires";
-                       next unless $nr;
                        my $file = slide_jpg( $factor => $nr );
-                       warn "slide $hires -> $file\n";
                        next if -e $file;
 
-                       if ( my $im = Imager->new( file => $hires ) ) {
-                               $im->scale( xpixels => $w, ypixels => $h, type => 'min' )->write( file => $file );
-                               warn "resized $file ", -s $file, " bytes\n";
-                       } else {
-                               die "can't open $hires: $!";
-                       }
+                       $min_slide_height = $prop->{height} if $prop->{height} > $min_slide_height;
+
+                       my $w = $min_slide_height / $factor * $slide_aspect;
+                       my $h = $min_slide_height / $factor;
+
+                       warn "slide [$nr] $hires -> ${w}x${h} $file\n";
+
+                       $im->Resize( width => $w, height => $h, filter => 13, blur => 0.9 );
+
+                       my $c = 1; # $h / 10;
+                       my %info = (
+                               font => 'Sans', pointsize => $h / 10,
+                               #text => "$factor = $w*$h",
+                               text => " $nr ",
+                               y => $c,
+                               x => $c,
+                       );
+                       #warn "# info ", dump %info;
+                       #warn dump $im->QueryFontMetrics( %info );
+                       my ($x_ppem, $y_ppem, $ascender, $descender, $width, $height, $max_advance) = $im->QueryFontMetrics( %info );
+                       my $background = Graphics::Magick->new( size => $width . 'x' . $height );
+                       $background->ReadImage( 'xc:black' );
+                       $im->Composite( image => $background, compose => 'Over', x => $c, y => $c, opacity => 75 );
+                       $info{y} += $ascender;
+                       $im->Annotate( fill => 'yellow', %info );
+                       $im->Write( filename => $file );
                }
 
        }
 
-       my ( $slide_width, $slide_height );
-
-       my $im = Imager->new( file => slide_jpg( 1 => 1 ) );
-
-       if ( $im ) {
-               $slide_width  = $im->getwidth  / $slide_factor;
-               $slide_height = $im->getheight / $slide_factor;
-       } else {
-               warn "can't find first slide default to 1/$slide_factor of video size\n";
-               $slide_width  = $prop->{width}  / $slide_factor;
-               $slide_height = $prop->{height} / $slide_factor;
-       }
+       my ($slide_width, $slide_height, $size, $format) = Graphics::Magick->new->Ping( slide_jpg( $slide_factor => 1 ) );
 
        my $html5tv = {
                sync => $sync,
@@ -303,41 +336,46 @@ sub html5tv {
                )
        ;
 
-       sub customEvents_sorted {
-
-               if ( ref $html5tv->{sync}->{customEvents} ne 'ARRAY' ) {
-                       my $max = 
-                       warn "ERROR: no slide markers [1] .. [", scalar @slides_hires, "] in subtitles\n";
-                       return;
-               }
+       my @customEvents_sorted;
 
+       if ( ref $html5tv->{sync}->{customEvents} ne 'ARRAY' ) {
+               my $max = 
+               warn "ERROR: no slide markers [1] .. [", scalar @slides_hires, "] in subtitles\n";
+               return;
+       } else {
+               @customEvents_sorted =
                sort { $a->{startTime} <=> $b->{startTime} }
                @{ $html5tv->{sync}->{customEvents} }
+               ;
        }
 
        my $index = 1;
 
-       $_->{args}->{index} = $index++ foreach customEvents_sorted;
+       $_->{args}->{index} = $index++ foreach @customEvents_sorted;
 
        warn "last customEvent $index\n";
 
        $html5tv->{html}->{subtitles_table}
                = qq|<table id="subtitles">|
                . join("\n",
-                       map { qq|
+                       map {
+                               my $s = fmt_mmss( $_->{startTime} );
+                               my $e = fmt_mmss( $_->{endTime} );
+                               qq|
                                <tr id="sub_$_->{index}">
-                                       <td class="seek_video">$_->{startTime}</td>
-                                       <td class="seek_video">$_->{endTime}</td>
+                                       <td class="seek_video"><a href="#$_->{startTime}">$s</a></td>
+                                       <td class="seek_video"><a href="#$_->{endTime}">$e</a></td>
                                        <td>$_->{args}->{title}</td>
                                </tr>
-                       | }
-                       customEvents_sorted
+                               |
+                       }
+                       @customEvents_sorted
                )
                . qq|</table><a href="$media_dir/video.srt">download subtitles</a>|
                ;
 
        my $hCalendar = '<div style="color: red">Create <tt>hCalendar.html</tt> to fill this space</div>';
-       my $hcal_path = '$media_dir/hCalendar.html';
+       my $hcal_path = "$media_dir/hCalendar.html";
        if ( -e $hcal_path ) {
                $html5tv->{hCalendar} = read_file $hcal_path;
                my $tree = HTML::TreeBuilder->new;
@@ -456,6 +494,8 @@ sub t_srt {
        return $srt;
 }
 
+my @to_mplayer;
+
 sub save_subtitles {
 
        html5tv || return;
@@ -473,9 +513,11 @@ sub save_subtitles {
        write_file $subtitles, $srt;
        YAML::DumpFile "$subtitles.yaml", sort { $a->[0] <=> $b->[0] } @subtitles;
 
-       print $to_mplayer "sub_remove\n";
-       print $to_mplayer qq|sub_load "$subtitles"\n|;
-       print $to_mplayer "sub_select 1\n";
+       push @to_mplayer
+               , "sub_remove\n"
+               , qq|sub_load "$subtitles"\n|
+               , "sub_select 1\n"
+               ;
 }
 
 sub load_subtitles {
@@ -512,6 +554,8 @@ sub add_subtitle {
        preroll $subtitles[ $#subtitles ]->[0], $line;
 }
 
+our $pos;
+
 sub time_pos {
        print $to_mplayer qq|get_property time_pos\n|;
        my $pos = <$from_mplayer>;
@@ -527,15 +571,13 @@ sub sub_fmt {
 }
 
 sub prev_subtitle {
-       my $pos = time_pos;
-       my $s = ( grep { $_->[0] < $pos } @subtitles )[0] || return;
+       my $s = ( grep { $_->[0] < $pos } @subtitles )[-1] || return;
        warn "<<<< subtitle ", sub_fmt $s;
        preroll $s->[0], $s->[2];
 #      print $to_mplayer "set_property time_pos $s->[0]\n";
 }
 
 sub next_subtitle {
-       my $pos = time_pos + $preroll;
        my $s = ( grep { $_->[0] > $pos } @subtitles )[0] || return;
        warn ">>>> subtitle ", sub_fmt $s;
        preroll $s->[0], $s->[2];
@@ -544,7 +586,6 @@ sub next_subtitle {
 
 sub current_subtitle {
        my $callback = shift;
-       my $pos = time_pos;
        my $visible;
        foreach my $nr ( 0 .. $#subtitles ) {
                my $s = $subtitles[$nr];
@@ -575,84 +616,105 @@ sub move_subtitle {
 
 load_movie;
 
-while ( my $events = epoll_wait($epfd, 10, 1000) ) { # Max 10 events returned, 1s timeout
-
-       warn "no events" unless $events;
-
-       foreach my $e ( @$events ) {
-#              warn "# event: ", dump($e), $/;
+my $slides = HTML5TV::Slides->new(
+       sub {
+               my $t = shift;
+               my $nr = 0;
+               foreach my $s ( @subtitles ) {
+                       $nr = $1 if $s->[2] =~ m{\[(\d+)\]} && $s->[0] < $t;
+               }
+               return $nr;
+       }
+);
 
-               my $fileno = $e->[0];
 
-               if ( $fileno == fileno STDIN ) {
-                       my $chr;
-                       sysread STDIN, $chr, 1;
-                       print $chr;
-               } elsif ( $fileno == fileno $from_mplayer ) {
-                       my $chr;
-                       sysread $from_mplayer, $chr, 1;
-                       print $chr;
+sub from_mplayer {
+       my $line = shift;
+
+       if ( $line =~ m{V:\s+(\d+\.\d+)\s+} ) {
+               $pos = $1;
+               print "$pos\r";
+#              $pos = $1 if $1 > 0.2; # mplayer demuxer report fake position while seeking
+       } elsif ( $line =~ m{Exiting} ) {
+               exit;
+       } elsif ( $line =~ m{ANS_(\w+)=(\S+)} ) {
+               $prop->{$1} = $2;
+               warn "prop $1 = $2\n";
+       } elsif ( $line =~ m{No bind found for key '(.+)'} ) {
+
+               # XXX keyboard shortcuts
+
+                 $1 eq 'c'  ? repl
+               : $1 eq ','  ? add_subtitle
+               : $1 eq 'F1' ? prev_subtitle
+               : $1 eq 'F2' ? move_subtitle( -0.3 )
+               : $1 eq 'F3' ? move_subtitle( +0.3 )
+               : $1 eq 'F4' ? next_subtitle
+               : $1 eq 'F5' ? save_subtitles
+               : $1 eq 'F9' ? add_subtitle
+               : $1 eq 'F12' ? edit_subtitles
+               : warn "CUSTOM $1\n"
+               ;
 
-                       if ( $chr =~ m{[\n\r]} ) {
+       } elsif ( $line =~ m{EDL}i ) {
 
-                               exit if $line =~ m{Exiting};
+               print $to_mplayer qq|osd_show_text "$line"\n|;
 
-                               if ( $line =~ m{ANS_(\w+)=(\S+)} ) {
-                                       $prop->{$1} = $2;
-                                       warn "prop $1 = $2\n";
-                               } elsif ( $line =~ m{No bind found for key '(.+)'} ) {
+               if ( $line =~ m{start}i ) {
+                       push @subtitles, [ $pos, $pos, '-' ];
+               } else {
+                       $subtitles[ $#subtitles ]->[1] = $pos;
+               }
+       } elsif ( $line =~ m{(shot\d+.png)} ) {
+               my $shot = $1;
+               warn "shot $pos $shot\n";
 
-                                       # XXX keyboard shortcuts
+               my @existing_slides = glob("$media_dir/s/hires/*");
+               my $nr = $#existing_slides + 2;
 
-                                         $1 eq 'c'  ? repl
-                                       : $1 eq ','  ? add_subtitle
-                                       : $1 eq 'F1' ? prev_subtitle
-                                       : $1 eq 'F2' ? move_subtitle( -0.3 )
-                                       : $1 eq 'F3' ? move_subtitle( +0.3 )
-                                       : $1 eq 'F4' ? next_subtitle
-                                       : $1 eq 'F5' ? save_subtitles
-                                       : $1 eq 'F9' ? add_subtitle
-                                       : $1 eq 'F12' ? edit_subtitles
-                                       : warn "CUSTOM $1\n"
-                                       ;
+               push @subtitles, [ $pos, $pos, "slide:$nr shot:$pos" ];
 
-                               } elsif ( $line =~ m{EDL}i ) {
+               warn "slide $nr from video $pos file $shot\n";
+               save_subtitles;
+       } else {
+               warn "IGNORE $line";
+       }
 
-                                       print $to_mplayer qq|osd_show_text "$line"\n|;
+}
 
-                                       if ( my $pos = time_pos ) {
-                                               if ( $line =~ m{start}i ) {
-                                                       push @subtitles, [ $pos, $pos, '-' ];
-                                               } else {
-                                                       $subtitles[ $#subtitles ]->[1] = $pos;
-                                               }
-                                       }
-                               } elsif ( $line =~ m{(shot\d+.png)} ) {
-                                       my $shot = $1;
-                                       my $t = time_pos;
-                                       warn "shot $t $shot\n";
+push @to_mplayer, "get_property $_\n" foreach grep { ! $prop->{$_} } ( qw/metadata video_codec video_bitrate width height fps length/ );
 
-                                       my @existing_slides = glob("$media_dir/s/hires/*");
-                                       my $nr = $#existing_slides + 2;
+my $t = time;
+my $line;
 
-                                       push @subtitles, [ $t, $t, "slide:$nr shot:$t" ];
+while ( 1 ) {
 
-                                       warn "slide $nr from video $t file $shot\n";
-                                       save_subtitles;
-                               }
+       my $dt = time - $t;
+       if ( abs($dt) > 0.7 ) {
+warn "dt $dt\n";
+               $slides->show( $pos );
+               $t = time;
+       }
 
+       foreach my $fd ( $select->can_read(0.1) ) {
+               if ( $fd == $from_mplayer ) {
+                       my $ch;
+                       sysread $from_mplayer, $ch, 1;
+                       if ( $ch =~ m{[\n\r]} ) {
+                               from_mplayer $line;
                                $line = '';
                        } else {
-                               $line .= $chr;
+                               $line .= $ch;
                        }
-
-
-               } elsif ( $fileno == fileno $to_mplayer ) {
-#                      warn "command";
                } else {
-                       die "invalid fileno $fileno";
+                       warn "unknown fd $fd";
                }
        }
 
+       if ( my $cmd = shift @to_mplayer ) {
+               warn ">>>> $cmd\n";
+               print $to_mplayer $cmd;
+       }
+
 }