sort subtitles on save instead on load
[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
13
14 my $movie = shift @ARGV
15         || 'media/lpc-2009-network-namespaces/Pavel Emelyanov.ogg';
16 #       || die "usage: $0 path/to/movie.ogv\n";
17
18 my $edl = "/dev/shm/edl";
19 my $subtitles = $movie;
20 $subtitles =~ s{\.\w+$}{.srt};
21
22 my $preroll = 3;
23
24 our $to_mplayer;
25 our $from_mplayer;
26 our $err_mplayer;
27 our $prop = {};
28
29 my $pid = open3( $to_mplayer, $from_mplayer, $err_mplayer,
30          'mplayer',
31                 '-slave', '-idle',
32                 '-quiet',
33                 '-edlout', $edl,
34                 '-osdlevel', 3,
35 );
36
37 my $epfd = epoll_create(10);
38
39 epoll_ctl($epfd, EPOLL_CTL_ADD, fileno STDIN         , EPOLLIN  ) >= 0 || die $!;
40 epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $from_mplayer , EPOLLIN  ) >= 0 || die $!;
41 #epoll_ctl($epfd, EPOLL_CTL_ADD, fileno $to_mplayer   , EPOLLOUT ) >= 0 || die $!;
42
43 sub load_movie {
44         warn "$movie ", -s $movie, " bytes $edl\n";
45         print $to_mplayer qq|loadfile "$movie"\n|;
46         print $to_mplayer "get_property $_\n" foreach ( qw/metadata video_codec video_bitrate width height fps/ );
47 }
48
49
50 my $term_id = `xdotool getwindowfocus`;
51 our $mplayer_id;
52
53
54 sub focus_mplayer {
55         $mplayer_id ||= `xdotool search mplayer`;
56         warn "focus_mplayer $mplayer_id\n";
57         system "xdotool windowactivate $mplayer_id"
58 }
59
60 sub focus_term {
61         warn "focus_term $term_id\n";
62         system "xdotool windowactivate $term_id";
63 }
64
65 sub preroll {
66         my ( $pos, $osd ) = @_;
67         $osd =~ s{\W+}{ }gs;
68         warn "PREROLL $pos $osd\n";
69         print $to_mplayer "osd_show_text \"PREROLL $osd\" ", $preroll * 1000, "\n";
70         my $to = $pos - $preroll;
71         $to = 0 if $to < 0;
72         print $to_mplayer "set_property time_pos $to\n";
73         print $to_mplayer "get_property time_pos\n";
74         print $to_mplayer "play\n";
75 }
76
77
78 $|=1;
79
80 my $line;
81 my $repl;
82
83 sub repl {
84         print "> ";
85         my $cmd = <STDIN>;
86         warn ">>> $cmd\n";
87         print $to_mplayer $cmd;
88 }
89
90
91 our @subtitles;
92
93 sub html5tv {
94         my $sync;
95
96         foreach my $s ( @subtitles ) {
97                 push @{ $sync->{htmlEvents}->{'#subtitle'} }, {
98                         startTime => $s->[0],
99                         endTime => $s->[1],
100                         html => $s->[2],
101                 };
102                 next unless $s->[2] =~ m{\[(\d+)\]};
103
104                 my $res = ( $prop->{width} / 4 ) . 'x' . ( $prop->{height} /4 );
105
106                 push @{ $sync->{customEvents} }, {
107                         startTime => $s->[0],
108                         endTime => $s->[1],
109                         action => 'chapterChange',
110                         args => {
111                                 carousel => 'theCarousel',
112                                 id => "chapter$1",
113                                 index => $1,
114                                 title => $s->[2],
115                                 description => $s->[2],
116                                 src => sprintf('s/%s/p%08d.jpg', $res, $1),
117                                 href => '',
118                         },
119                 }
120         }
121
122         warn "# sync ", dump $sync;
123
124         warn "# prop ", dump $prop;
125
126         my $html5tv = $prop;
127         $html5tv->{sync} = $sync;
128
129         my $sync_path = 'www/video.js';
130         write_file $sync_path, "var html5tv = " . to_json($html5tv) . " ;\n";
131         warn "sync $sync_path ", -s $sync_path, " bytes\n";
132
133         if ( $prop->{width} && $prop->{height} ) {
134                 foreach my $factor ( 1, 2, 4 ) {
135                         my $w = $prop->{width}  / $factor;
136                         my $h = $prop->{height} / $factor;
137                         my $path = "www/s/${w}x${h}";
138                         if ( ! -d $path ) {
139                                 mkdir $path;
140                                 warn "created $path\n";
141                         }
142                 }
143         }
144
145 }
146
147
148 sub t_srt {
149         my $t = shift;
150         my $hh = int( $t / 3600 );
151         $t -= $hh * 3600;
152         my $mm = int( $t / 60 );
153         $t -= $mm * 60;
154         my $srt = sprintf "%02d:%02d:%04.1f", $hh, $mm, $t;
155         $srt =~ s{\.}{,};
156         return $srt;
157 }
158
159 sub save_subtitles {
160         my $nr = 0;
161         my $srt = "\n";
162         foreach my $s ( @subtitles ) {
163                 $srt .= $nr++ . "\n"
164                         . t_srt( $s->[0] ) . " --> " . t_srt( $s->[1] ) . "\n"
165                         . $s->[2] . "\n\n"
166                         ;
167         }
168         warn $srt;
169
170         write_file $subtitles, $srt;
171         YAML::DumpFile "$subtitles.yaml", sort { $a->[0] <=> $b->[0] } @subtitles;
172
173         print $to_mplayer "sub_remove\n";
174         print $to_mplayer qq|sub_load "$subtitles"\n|;
175         print $to_mplayer "sub_select 1\n";
176
177         html5tv;
178 }
179
180 sub load_subtitles {
181         @subtitles = YAML::LoadFile "$subtitles.yaml";
182         warn "subtitles ", dump @subtitles;
183         save_subtitles;
184 }
185
186 load_subtitles if -e "$subtitles.yaml";
187
188 sub edit_subtitles {
189         print $to_mplayer qq|pause\n|;
190         focus_term;
191         system( qq|vi "$subtitles.yaml"| ) == 0 and load_subtitles;
192         focus_mplayer;
193 }
194
195 sub add_subtitle {
196         print $to_mplayer qq|pause\n|;
197
198         focus_term;
199
200         warn "subtitles ", dump( @subtitles );
201         print "## ";
202         my $line = <STDIN>;
203         $subtitles[ $#subtitles ]->[2] = $line if defined $line;
204
205         save_subtitles;
206
207         focus_mplayer;
208
209         preroll $subtitles[ $#subtitles ]->[0], $line;
210 }
211
212 sub time_pos {
213         print $to_mplayer qq|get_property time_pos\n|;
214         my $pos = <$from_mplayer>;
215         if ( $pos =~ m{^ANS_time_pos=(\d+\.\d+)} ) {
216                 warn "# time_pos $1\n";
217                 return $1;
218         }
219 }
220
221 sub sub_fmt {
222         my $s = shift;
223         sprintf "%1.5f - %1.5f %s %s\n", @$s, join(' | ',@_);
224 }
225
226 sub prev_subtitle {
227         my $pos = time_pos;
228         my $s = ( grep { $_->[0] < $pos } @subtitles )[0];
229         warn "<<<< subtitle ", sub_fmt $s;
230         preroll $s->[0], $s->[2];
231 #       print $to_mplayer "set_property time_pos $s->[0]\n";
232 }
233
234 sub next_subtitle {
235         my $pos = time_pos + $preroll;
236         my $s = ( grep { $_->[0] > $pos } @subtitles )[0];
237         warn ">>>> subtitle ", sub_fmt $s;
238         preroll $s->[0], $s->[2];
239 #       print $to_mplayer "set_property time_pos $s->[0]\n";
240 }
241
242 sub current_subtitle {
243         my $callback = shift;
244         my $pos = time_pos;
245         my $visible;
246         foreach my $nr ( 0 .. $#subtitles ) {
247                 my $s = $subtitles[$nr];
248                 if ( $s->[0] <= $pos && $s->[1] >= $pos ) {
249                         warn sub_fmt $s, $pos;
250                         $visible = $nr;
251                         $callback->( $visible, $pos ) if $callback;
252                         return ( $visible, $pos );
253                 }
254         }
255         warn "# $pos no visible subtitle\n";
256 }
257
258 sub move_subtitle {
259         my $offset = shift;
260         current_subtitle( sub {
261                 my ( $nr, $pos ) = @_;
262                 my $new_start = $subtitles[$nr]->[0] += $offset;
263                 warn "subtitle $nr $pos $offset $new_start\n";
264                 save_subtitles;
265                 preroll $new_start, "$pos $offset $new_start";
266         } );
267 }
268
269
270
271 # XXX main epool loop
272
273 load_movie;
274
275 while ( my $events = epoll_wait($epfd, 10, 1000) ) { # Max 10 events returned, 1s timeout
276
277         warn "no events" unless $events;
278
279         foreach my $e ( @$events ) {
280 #               warn "# event: ", dump($e), $/;
281
282                 my $fileno = $e->[0];
283
284                 if ( $fileno == fileno STDIN ) {
285                         my $chr;
286                         sysread STDIN, $chr, 1;
287                         print $chr;
288                 } elsif ( $fileno == fileno $from_mplayer ) {
289                         my $chr;
290                         sysread $from_mplayer, $chr, 1;
291                         print $chr;
292
293                         if ( $chr =~ m{[\n\r]} ) {
294
295                                 exit if $line =~ m{Exiting};
296
297                                 if ( $line =~ m{ANS_(\w+)=(\S+)} ) {
298                                         $prop->{$1} = $2;
299                                         warn "prop $1 = $2\n";
300                                 } elsif ( $line =~ m{No bind found for key '(.+)'} ) {
301
302                                           $1 eq 'c'  ? repl
303                                         : $1 eq ','  ? add_subtitle
304                                         : $1 eq 'F1' ? prev_subtitle
305                                         : $1 eq 'F2' ? move_subtitle( -0.3 )
306                                         : $1 eq 'F3' ? move_subtitle( +0.3 )
307                                         : $1 eq 'F4' ? next_subtitle
308                                         : $1 eq 'F2' ? move_subtitle( -0.3 )
309                                         : $1 eq 'F3' ? move_subtitle( +0.3 )
310                                         : $1 eq 'F9' ? add_subtitle
311                                         : $1 eq 'F12' ? edit_subtitles
312                                         : warn "CUSTOM $1\n"
313                                         ;
314
315                                 } elsif ( $line =~ m{EDL}i ) {
316
317                                         print $to_mplayer qq|osd_show_text "$line"\n|;
318
319                                         if ( my $pos = time_pos ) {
320                                                 if ( $line =~ m{start}i ) {
321                                                         push @subtitles, [ $pos, $pos, '-' ];
322                                                 } else {
323                                                         $subtitles[ $#subtitles ]->[1] = $pos;
324                                                 }
325                                         }
326                                 }
327
328                                 $line = '';
329                         } else {
330                                 $line .= $chr;
331                         }
332
333
334                 } elsif ( $fileno == fileno $to_mplayer ) {
335 #                       warn "command";
336                 } else {
337                         die "invalid fileno $fileno";
338                 }
339         }
340
341 }
342