fix prev/next_subtitle
[HTML5TV.git] / bin / mplayer.pl
1 #!/usr/bin/perl
2
3 use warnings;
4 use strict;
5
6 use IPC::Open3 qw(open3);
7 use IO::Epoll;
8 use Data::Dump qw(dump);
9 use File::Slurp;
10 use YAML;
11 use JSON;
12 use HTML::TreeBuilder;
13 use Imager;
14
15 my $debug = 0;
16
17 my $movie = shift @ARGV;
18
19 sub base_dir  { $1 if $_[0] =~ m{^(.+)/[^/]+$} }
20 sub base_name { $1 if $_[0] =~ m{^.+/([^/]+)$} }
21
22 if ( ! $movie && -e 'media/_editing' ) {
23         $movie = 'media/' . readlink('media/_editing') . '/video.ogv';
24         warn "using media/_editing -> $movie\n";
25 } elsif ( -d $movie && $movie =~ m{media/} ) {
26         $movie .= '/video.ogv';
27 } elsif ( -f $movie && $movie !~ m{video\.ogv} ) {
28         my $movie_master = $movie;
29         $movie = base_dir($movie) . '/video.ogv';
30         if ( ! -e $movie ) {
31                 symlink base_name($movie_master), $movie;
32                 warn "symlink video.ogv -> $movie_master";
33         } else {
34                 warn "using symlink video.ogv -> ", readlink $movie;
35         }
36 } elsif ( -f $movie ) {
37         warn "using video $movie";
38 } else {
39         die "Usage: $0 media/conference-Title_of_talk[/video.ogv'\n";
40 }
41
42 my $media_part = my $media_dir = base_dir($movie);
43 $media_part =~ s{media/}{};
44 $media_part =~ s{/$}{};
45
46 unlink 'media/_editing';
47 symlink $media_part, 'media/_editing';
48
49 warn "# media_part $media_part\n";
50
51 my $edl = "/dev/shm/edl";
52 my $subtitles = $movie;
53 $subtitles =~ s{\.\w+$}{.srt};
54
55 my $preroll = 3;
56
57 my $slide_factor = 4; # 1/4 size of video
58
59 our $to_mplayer;
60 our $from_mplayer;
61 our $err_mplayer;
62 our $prop = {};
63
64 my $pid = open3( $to_mplayer, $from_mplayer, $err_mplayer,
65          'mplayer',
66                 '-slave', '-idle',
67                 '-quiet',
68                 '-edlout', $edl,
69                 '-osdlevel', 3,
70                 '-vf' => 'screenshot',
71 );
72
73 my $epfd = epoll_create(10);
74
75 epoll_ctl($epfd, EPOLL_CTL_ADD, fileno STDIN         , EPOLLIN  ) >= 0 || die $!;
76 epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $from_mplayer , EPOLLIN  ) >= 0 || die $!;
77 #epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $to_mplayer   , EPOLLOUT ) >= 0 || die $!;
78
79 sub load_subtitles;
80
81 sub load_movie {
82         warn "$movie ", -s $movie, " bytes $edl\n";
83         print $to_mplayer qq|loadfile "$movie"\n|;
84         print $to_mplayer "get_property $_\n" foreach ( qw/metadata video_codec video_bitrate width height fps length/ );
85         load_subtitles;
86 }
87
88
89 my $term_id = `xdotool getwindowfocus`;
90 our $mplayer_id;
91
92
93 sub focus_mplayer {
94         $mplayer_id ||= `xdotool search mplayer`;
95         warn "focus_mplayer $mplayer_id\n";
96         system "xdotool windowactivate $mplayer_id"
97 }
98
99 sub focus_term {
100         warn "focus_term $term_id\n";
101         system "xdotool windowactivate $term_id";
102 }
103
104 sub preroll {
105         my ( $pos, $osd ) = @_;
106         $osd =~ s{\W+}{ }gs;
107         warn "PREROLL $pos $osd\n";
108         print $to_mplayer "osd_show_text \"PREROLL $osd\" ", $preroll * 1000, "\n";
109         my $to = $pos - $preroll;
110         $to = 0 if $to < 0;
111         print $to_mplayer "set_property time_pos $to\n";
112         print $to_mplayer "get_property time_pos\n";
113         print $to_mplayer "play\n";
114 }
115
116
117 $|=1;
118
119 my $line;
120 my $repl;
121
122 sub repl {
123         print "> ";
124         my $cmd = <STDIN>;
125         warn ">>> $cmd\n";
126         print $to_mplayer $cmd;
127 }
128
129
130 our @subtitles;
131
132 sub slide_jpg {
133         sprintf "%s/s/%d/%03d.jpg", $media_dir, @_;
134 }
135
136 sub oggThumb {
137         my ($video,$t,$file) = @_;
138         system "oggvideotools/src/oggThumb -t $t -o jpg -n $file $video";
139         warn "oggThumb $file at $t from $video\n";
140 }
141
142 sub html5tv {
143
144         if ( ! $prop->{width} || ! $prop->{height} ) {
145                 warn "SKIP no size yet\n";
146                 return;
147         }
148
149         if ( ! @subtitles ) {
150                 warn "SKIP no subtitles yet\n";
151                 return;
152         }
153
154
155         my $sync;
156
157         my @slide_t;
158
159         my @videos;
160
161         foreach my $s ( @subtitles ) {
162                 push @{ $sync->{htmlEvents}->{'#subtitle'} }, {
163                         startTime => $s->[0],
164                         endTime   => $s->[1],
165                         html      => $s->[2],
166                 };
167
168                 if ( $s->[2] =~ m{video:(.+)} ) {
169                         my $video = $1;
170                         my $path = "$media_dir/$video";
171                         if ( ! -e $path ) {
172                                 warn "MISSING $path: $!\n";
173                         } else {
174                                 my $frame_dir = "$media_dir/s/$video";
175                                 system "mplayer -vo jpeg:outdir=$frame_dir,quality=95 -frames 1 -ss 0 -ao null -really-quiet $media_dir/$video"
176                                         if ! -e $frame_dir;
177                                 push @videos, [ @$s, $video ];
178                         }
179                 } elsif ( $s->[2] =~ m{slide:(\d+)\s+shot:(\d+\.\d+)} ) {
180                         my ( $nr, $t ) = ( $1, $2 );
181
182                         my $hires = "$media_dir/s/hires";
183                         mkdir $hires unless -e $hires;
184
185                         my $shot_path = sprintf "$hires/s%03d.jpg", $nr;
186                         oggThumb $movie, $t, $shot_path unless -e $shot_path;
187                         next;
188                 }
189
190                 next unless $s->[2] =~ m{\[(\d+)\]};
191
192
193                 push @{ $sync->{customEvents} }, {
194                         startTime => $s->[0],
195                         endTime   => $s->[1],
196                         action    => 'chapterChange',
197                         args => {
198                                 carousel => 'theCarousel',
199                                 id => "chapter$1",
200                                 title => $s->[2],
201                                 description => $s->[2],
202                                 src => slide_jpg( 4 => $1 ),
203                                 href => '',
204                         },
205                 };
206
207                 push @slide_t, $s->[0];
208         }
209
210         foreach ( 0 .. $#slide_t ) {
211                 my $slide_nr = $_ + 1;
212                 push @{ $sync->{htmlEvents}->{'#slide'} }, {
213                         startTime => $slide_t[$_],
214                         endTime   => $slide_t[$_ + 1] || $prop->{length},
215                         html      => '<img src=' . slide_jpg( 1 => $slide_nr ) . '>',
216                 };
217         }
218
219         my @slides_hires = glob "$media_dir/s/hires/*";
220
221         foreach my $factor ( 4, 2, 1 ) {
222                 my $w = $prop->{width}  / $factor;
223                 my $h = $prop->{height} / $factor;
224
225                 my $path = "$media_dir/s";
226                 if ( ! -d $path ) {
227                         warn "create slides imaes in $path";
228                         mkdir $path;
229                 }
230
231                 $path .= '/' . $factor;
232
233                 if ( ! -d $path ) {
234                         mkdir $path;
235                         warn "created $path\n";
236
237                 }
238
239                 foreach my $hires ( @slides_hires ) {
240
241                         my $nr = $1 if $hires =~ m{(\d+)\.\w+$} || warn "can't find number in $hires";
242                         next unless $nr;
243                         my $file = slide_jpg( $factor => $nr );
244                         warn "slide $hires -> $file\n";
245                         next if -e $file;
246
247                         if ( my $im = Imager->new( file => $hires ) ) {
248                                 $im->scale( xpixels => $w, ypixels => $h, type => 'min' )->write( file => $file );
249                                 warn "resized $file ", -s $file, " bytes\n";
250                         } else {
251                                 die "can't open $hires: $!";
252                         }
253                 }
254
255         }
256
257         my ( $slide_width, $slide_height );
258
259         my $im = Imager->new( file => slide_jpg( 1 => 1 ) );
260
261         if ( $im ) {
262                 $slide_width  = $im->getwidth  / $slide_factor;
263                 $slide_height = $im->getheight / $slide_factor;
264         } else {
265                 warn "can't find first slide default to 1/$slide_factor of video size\n";
266                 $slide_width  = $prop->{width}  / $slide_factor;
267                 $slide_height = $prop->{height} / $slide_factor;
268         }
269
270         my $html5tv = {
271                 sync => $sync,
272                 video => $prop,
273                 slide => {
274                         width => $slide_width,
275                         height => $slide_height,
276                 },
277         };
278
279         $html5tv->{html}->{video_tags} =
280                 join("\n",
281                         map {
282                                 my $s = $_;
283                                 my $id = $s->[3];
284                                 $id =~ s{\W+}{_}g;
285
286                                 push @{ $html5tv->{sync}->{customEvents} }, {
287                                         startTime => $s->[0],
288                                         endTime   => $s->[1],
289                                         action    => 'additional_video',
290                                         args => {
291                                                 id => $id,
292                                                 title => $s->[2],
293                                                 description => $s->[2],
294                                                 src => "$media_dir/s/1/00000001.jpg",
295                                                 href => '',
296                                         },
297                                 };
298
299                                 qq|
300                                         <video id="$id" style="display: none" controls="controls" width="$html5tv->{video}->{width}px" height="$html5tv->{video}->{height}px">
301                                         <source src="$media_dir/$_->[3]" />
302                                         </video>
303                                 |
304                         } @videos
305                 )
306         ;
307
308         sub customEvents_sorted {
309
310                 if ( ref $html5tv->{sync}->{customEvents} ne 'ARRAY' ) {
311                         my $max = 
312                         warn "ERROR: no slide markers [1] .. [", scalar @slides_hires, "] in subtitles\n";
313                         return;
314                 }
315
316                 sort { $a->{startTime} <=> $b->{startTime} }
317                 @{ $html5tv->{sync}->{customEvents} }
318         }
319
320         my $index = 1;
321
322         $_->{args}->{index} = $index++ foreach customEvents_sorted;
323
324         warn "last customEvent $index\n";
325
326         $html5tv->{html}->{subtitles_table}
327                 = qq|<table id="subtitles">|
328                 . join("\n",
329                         map { qq|
330                                 <tr id="sub_$_->{index}">
331                                         <td class="seek_video">$_->{startTime}</td>
332                                         <td class="seek_video">$_->{endTime}</td>
333                                         <td>$_->{args}->{title}</td>
334                                 </tr>
335                         | }
336                         customEvents_sorted
337                 )
338                 . qq|</table><a href="$media_dir/video.srt">download subtitles</a>|
339                 ;
340
341         my $hCalendar = '<div style="color: red">Create <tt>hCalendar.html</tt> to fill this space</div>';
342         my $hcal_path = '$media_dir/hCalendar.html';
343         if ( -e $hcal_path ) {
344                 $html5tv->{hCalendar} = read_file $hcal_path;
345                 my $tree = HTML::TreeBuilder->new;
346                 $tree->parse_file($hcal_path);
347                 if ( my $vevent = $tree->look_down( class => 'vevent' ) ) {
348                         $html5tv->{title} = $vevent->look_down( class=> 'summary' )->as_trimmed_text;
349                 }
350         }
351
352         warn "# html5tv ", dump $html5tv if $debug;
353
354         my $sync_path = "$media_dir/video.js";
355         write_file $sync_path, "var html5tv = " . to_json($html5tv) . " ;\n";
356         warn "sync $sync_path ", -s $sync_path, " bytes\n";
357
358         my $html = read_file 'www/tv.html';
359         $html =~ s|{([^}]+)}|my $n = $1; $n =~ s(\.)(}->{)g; eval "\$html5tv->{$n}"|egs ||
360                 warn "no interpolation in template!";
361
362         write_file "www/_editing.html", $html;
363         $html =~ s{media/_editing}{media/$media_part}gs;
364         write_file "www/$media_part.html", $html;
365
366         my $carousel_width = $prop->{width} + $slide_width - 80;
367         $carousel_width -= $carousel_width % ( $slide_width + 6 ); # round to full slide
368         my $carousel_height =   $slide_height + 2;
369
370         write_file "$media_dir/video.css", qq|
371
372 .jcarousel-skin-ie7 .jcarousel-container-horizontal,
373 .jcarousel-skin-ie7 .jcarousel-clip-horizontal {
374         width: ${carousel_width}px;
375         height: ${carousel_height}px;
376 }
377
378 .jcarousel-skin-ie7 .jcarousel-item {
379         width: ${slide_width}px;
380         height: ${slide_height}px;
381         margin: 0 2px 0 2px;
382 }
383
384 .active {
385         background-color: #ffc;
386 }
387
388 div#videoContainer {
389         width: $prop->{width}px;
390         height: $prop->{height}px;
391         font-family: Arial, Helvetica, sans-serif;
392         margin: 0 10px 0px 0;
393         position: relative;
394         display: inline;
395 }
396
397
398 div#subtitle {
399         bottom: 24px;
400         color: white;
401         font-size: 100%;
402         font-weight: bold;
403         height: 22px;
404         line-height: 1em;
405         margin: 0  0 0 0;
406         padding: 1px 10px 5px 10px ;
407         position: absolute;
408         text-align: center;
409         width: $html5tv->{video}->{width}px;
410 }
411
412
413
414 .jcarousel-skin-ie7 .jcarousel-item img:hover {
415 //      border-color: #555 !important;
416 }
417
418 .jcarousel-skin-ie7 .jcarousel-item:hover div.thumbnailOverlay {
419         visibility: visible !important;
420 }
421
422 .jcarousel-skin-ie7 .jcarousel-item div.thumbnailOverlay {
423         background: black;
424         bottom: 1px;
425         color: #00EEFF;
426         visibility: hidden;
427         font-size: 10px;
428         font-family: Arial, Verdana;
429         font-weight: bold;
430         line-height: 0.9em;
431         opacity: 0.5;
432         position: absolute;
433         text-align: center;
434         z-index: 10;
435         padding: 2px 0 2px 0;
436         width: ${slide_width}px;
437 }
438
439 .seek_video {
440         text-align: right;
441         font-family: monospace;
442 }
443
444         |;
445
446         return 1;
447 }
448
449
450 sub t_srt {
451         my $t = shift;
452         my $hh = int( $t / 3600 );
453         $t -= $hh * 3600;
454         my $mm = int( $t / 60 );
455         $t -= $mm * 60;
456         my $srt = sprintf "%02d:%02d:%04.1f", $hh, $mm, $t;
457         $srt =~ s{\.}{,};
458         return $srt;
459 }
460
461 sub save_subtitles {
462
463         html5tv || return;
464
465         my $nr = 0;
466         my $srt = "\n";
467         foreach my $s ( @subtitles ) {
468                 $srt .= $nr++ . "\n"
469                         . t_srt( $s->[0] ) . " --> " . t_srt( $s->[1] ) . "\n"
470                         . $s->[2] . "\n\n"
471                         ;
472         }
473         warn $srt;
474
475         write_file $subtitles, $srt;
476         YAML::DumpFile "$subtitles.yaml", sort { $a->[0] <=> $b->[0] } @subtitles;
477
478         print $to_mplayer "sub_remove\n";
479         print $to_mplayer qq|sub_load "$subtitles"\n|;
480         print $to_mplayer "sub_select 1\n";
481 }
482
483 sub load_subtitles {
484         if ( ! -e "$subtitles.yaml" ) {
485                 warn "no subtitles $subtitles to load\n";
486                 return;
487         }
488         @subtitles = YAML::LoadFile "$subtitles.yaml";
489         warn "subtitles ", dump @subtitles;
490         save_subtitles;
491 }
492
493 sub edit_subtitles {
494         print $to_mplayer qq|pause\n|;
495         focus_term;
496         system( qq|vi "$subtitles.yaml"| ) == 0 and load_subtitles;
497         focus_mplayer;
498 }
499
500 sub add_subtitle {
501         print $to_mplayer qq|pause\n|;
502
503         focus_term;
504
505         warn "subtitles ", dump( @subtitles );
506         print "## ";
507         my $line = <STDIN>;
508         $subtitles[ $#subtitles ]->[2] = $line if defined $line;
509
510         save_subtitles;
511
512         focus_mplayer;
513
514         preroll $subtitles[ $#subtitles ]->[0], $line;
515 }
516
517 sub time_pos {
518         print $to_mplayer qq|get_property time_pos\n|;
519         my $pos = <$from_mplayer>;
520         if ( $pos =~ m{^ANS_time_pos=(\d+\.\d+)} ) {
521                 warn "# time_pos $1\n";
522                 return $1;
523         }
524 }
525
526 sub sub_fmt {
527         my $s = shift;
528         sprintf "%1.5f - %1.5f %s %s\n", @$s, join(' | ',@_);
529 }
530
531 sub prev_subtitle {
532         my $pos = time_pos;
533         my $s = ( grep { $_->[0] < $pos } @subtitles )[-1] || return;
534         warn "<<<< subtitle ", sub_fmt $s;
535         preroll $s->[0], $s->[2];
536 #       print $to_mplayer "set_property time_pos $s->[0]\n";
537 }
538
539 sub next_subtitle {
540         my $pos = time_pos;
541         my $s = ( grep { $_->[0] > $pos } @subtitles )[0] || return;
542         warn ">>>> subtitle ", sub_fmt $s;
543         preroll $s->[0], $s->[2];
544 #       print $to_mplayer "set_property time_pos $s->[0]\n";
545 }
546
547 sub current_subtitle {
548         my $callback = shift;
549         my $pos = time_pos;
550         my $visible;
551         foreach my $nr ( 0 .. $#subtitles ) {
552                 my $s = $subtitles[$nr];
553                 if ( $s->[0] <= $pos && $s->[1] >= $pos ) {
554                         warn sub_fmt $s, $pos;
555                         $visible = $nr;
556                         $callback->( $visible, $pos ) if $callback;
557                         return ( $visible, $pos );
558                 }
559         }
560         warn "# $pos no visible subtitle\n";
561 }
562
563 sub move_subtitle {
564         my $offset = shift;
565         current_subtitle( sub {
566                 my ( $nr, $pos ) = @_;
567                 my $new_start = $subtitles[$nr]->[0] += $offset;
568                 warn "subtitle $nr $pos $offset $new_start\n";
569                 save_subtitles;
570                 preroll $new_start, "$pos $offset $new_start";
571         } );
572 }
573
574
575
576 # XXX main epool loop
577
578 load_movie;
579
580 while ( my $events = epoll_wait($epfd, 10, 1000) ) { # Max 10 events returned, 1s timeout
581
582         warn "no events" unless $events;
583
584         foreach my $e ( @$events ) {
585 #               warn "# event: ", dump($e), $/;
586
587                 my $fileno = $e->[0];
588
589                 if ( $fileno == fileno STDIN ) {
590                         my $chr;
591                         sysread STDIN, $chr, 1;
592                         print $chr;
593                 } elsif ( $fileno == fileno $from_mplayer ) {
594                         my $chr;
595                         sysread $from_mplayer, $chr, 1;
596                         print $chr;
597
598                         if ( $chr =~ m{[\n\r]} ) {
599
600                                 exit if $line =~ m{Exiting};
601
602                                 if ( $line =~ m{ANS_(\w+)=(\S+)} ) {
603                                         $prop->{$1} = $2;
604                                         warn "prop $1 = $2\n";
605                                 } elsif ( $line =~ m{No bind found for key '(.+)'} ) {
606
607                                         # XXX keyboard shortcuts
608
609                                           $1 eq 'c'  ? repl
610                                         : $1 eq ','  ? add_subtitle
611                                         : $1 eq 'F1' ? prev_subtitle
612                                         : $1 eq 'F2' ? move_subtitle( -0.3 )
613                                         : $1 eq 'F3' ? move_subtitle( +0.3 )
614                                         : $1 eq 'F4' ? next_subtitle
615                                         : $1 eq 'F5' ? save_subtitles
616                                         : $1 eq 'F9' ? add_subtitle
617                                         : $1 eq 'F12' ? edit_subtitles
618                                         : warn "CUSTOM $1\n"
619                                         ;
620
621                                 } elsif ( $line =~ m{EDL}i ) {
622
623                                         print $to_mplayer qq|osd_show_text "$line"\n|;
624
625                                         if ( my $pos = time_pos ) {
626                                                 if ( $line =~ m{start}i ) {
627                                                         push @subtitles, [ $pos, $pos, '-' ];
628                                                 } else {
629                                                         $subtitles[ $#subtitles ]->[1] = $pos;
630                                                 }
631                                         }
632                                 } elsif ( $line =~ m{(shot\d+.png)} ) {
633                                         my $shot = $1;
634                                         my $t = time_pos;
635                                         warn "shot $t $shot\n";
636
637                                         my @existing_slides = glob("$media_dir/s/hires/*");
638                                         my $nr = $#existing_slides + 2;
639
640                                         push @subtitles, [ $t, $t, "slide:$nr shot:$t" ];
641
642                                         warn "slide $nr from video $t file $shot\n";
643                                         save_subtitles;
644                                 }
645
646                                 $line = '';
647                         } else {
648                                 $line .= $chr;
649                         }
650
651
652                 } elsif ( $fileno == fileno $to_mplayer ) {
653 #                       warn "command";
654                 } else {
655                         die "invalid fileno $fileno";
656                 }
657         }
658
659 }
660