Ticket #8206: stream_flv.pl.diff

/var/www/mythweb/modules/stream/stream_flv.pl

@@ -2 +2 @@
 #
 # MythWeb Streaming/Download module
 #
-# @url $URL$
-# @date $Date$
-# @version $Revision$
-# @author $Author$
+# @url       $URL: http://svn.mythtv.org/svn/branches/release-0-23-fixes/mythplugins/mythweb/modules/stream/stream_flv.pl $
+# @date      $Date: 2009-10-30 00:58:53 +0000 (Fri, 30 Oct 2009) $
+# @version   $Revision: 22646 $
+# @author    $Author: kormoc $
 #
 
     use Math::Round qw(round_even);
@@ -38 +38 @@
 # Load some conversion settings from the database
     $sh = $dbh->prepare('SELECT data FROM settings WHERE value=? AND hostname IS NULL');
     $sh->execute('WebFLV_w');
-    my ($width) = $sh->fetchrow_array;
+    my ($width)    = $sh->fetchrow_array;
     $sh->execute('WebFLV_vb');
     my ($vbitrate) = $sh->fetchrow_array;
     $sh->execute('WebFLV_ab');
     my ($abitrate) = $sh->fetchrow_array;
     $sh->finish();
+
 # auto-detect height based on aspect ratio
     $sh = $dbh->prepare('SELECT data FROM recordedmarkup WHERE chanid=? AND starttime=FROM_UNIXTIME(?) AND data IS NOT NULL ORDER BY data DESC');
     $sh->execute($chanid,$starttime);
@@ -57 +58 @@
     }
     $sh->finish();
 
-    $width = 320 unless ($width && $width > 1);
-    $height = 240 unless ($height && $height > 1);
+    $width    = 320 unless ($width    && $width    > 1);
+    $height   = 240 unless ($height   && $height   > 1);
     $vbitrate = 256 unless ($vbitrate && $vbitrate > 1);
-    $abitrate = 64 unless ($abitrate && $abitrate > 1);
+    $abitrate = 64  unless ($abitrate && $abitrate > 1);
+
 
-    my $ffmpeg_command = $ffmpeg
+    if ("$ENV{'REQUEST_URI'}" =~ /start\=(\d+)$/i) { 
+            $start = $1; } 
+    else { 
+            $start = 0; } 
+    warn("Specified start time: ".$start);
+
+# Are we still recording, and how long should the program be?
+    $sh = $dbh->prepare('SELECT UNIX_TIMESTAMP(endtime) - UNIX_TIMESTAMP(starttime), endtime-now()
+                             FROM recorded
+                             WHERE starttime=FROM_UNIXTIME(?)
+                               AND recorded.chanid   = ?');
+    $sh->execute($starttime, $chanid);
+    our ($lengthSec, $inprogress) = $sh->fetchrow_array();
+    $sh->finish;
+
+# Calculate the offset given the start time
+    $frameNum = $start * 24;
+    warn("Looking for frame number: ".$frameNum);
+    $sh = $dbh->prepare('SELECT count(*), offset 
+                             FROM recordedseek 
+                             WHERE chanid=? 
+                               AND starttime=FROM_UNIXTIME(?) 
+                               AND mark >= ? 
+                               AND mark < ?');
+    $sh->execute($chanid,$starttime, $frameNum, $frameNum + 12);
+    our($numRows, $byteOffset) = $sh->fetchrow_array();
+    $sh->finish();
+    if ($numRows < 1) {
+        warn("Got no rows, need to wait for recording to catchup to offset, sleeping");
+        print header(), "Seeked beyond the file";
+        exit;
+    }
+
+    warn("Looked up a byte offset of: ".$byteOffset);
+
+
+    my $ffmpeg_command = 'tail -fc +'
+                        .$byteOffset
+                        .' '
+                        .shell_escape($filename)
+                        .' | '
+                        .$ffmpeg
                         .' -y'
-                        .' -i '.shell_escape($filename)
+                        .' -i pipe:0'
                         .' -s '.shell_escape("${width}x${height}")
                         .' -g 30'
-                        .' -r 24'
+                        .' -r 15'
+                        .' -qmin 8'
+                        .' -qmax 15'
                         .' -f flv'
                         .' -deinterlace'
                         .' -ac 2'
-                        .' -ar 11025'
+                        .' -ar 22050'
                         .' -ab '.shell_escape("${abitrate}k")
                         .' -b '.shell_escape("${vbitrate}k")
                         .' /dev/stdout 2>/dev/null |';
 
+    warn("Command to be used is: ".$ffmpeg_command);
+
 # Print the movie
     $ffmpeg_pid = open(DATA, $ffmpeg_command);
+    warn("PID is: ".$ffmpeg_pid);
     unless ($ffmpeg_pid) {
         print header(),
                 "Can't do ffmpeg: $!\n${ffmpeg_command}";
         exit;
     }
-    print header(-type => 'video/x-flv');
+
+# Guess the filesize based on duration and bitrate. This allows for progressive download behavior
+    warn("Length in seconds from DB is: ".$lengthSec);
+    if ($inprogress > 0) {
+        warn("Recording is in progress");
+    } else {
+        warn("Recording is finished");
+        $dur = `ffmpeg -i $filename 2>&1 | grep "Duration" | cut -d ' ' -f 4 | sed s/,//`;    
+        if ($dur && $dur =~ /\d*:\d*:.*/) {
+            warn("Accurate duration from ffmpeg is: ".$dur);
+            @times = split(':',$dur);
+            $lengthSec = $times[0]*3600+$times[1]*60+$times[2] - $start;
+        }
+    }
+
+    warn("Using a duration of: ".$lengthSec." seconds");
+    if ($lengthSec > 0) { #This should always be true since we use the DB runtime.
+        #Get the content size as a whole interger.
+        $size = int(1.05*$lengthSec*($vbitrate*1024+$abitrate*1024)/8);
+        warn("Determined content size is: ".$size);
+        print header(-type => 'video/x-flv','Content-Length' => $size);
+    } else {
+        print header(-type => 'video/x-flv');
+    }
+
     my $buffer;
-    while (read DATA, $buffer, 262144) {
+    if (read DATA, $buffer, 53) {
         print $buffer;
+        read DATA, $buffer, 8; 
+        $durPrint = reverse pack("d",$lengthSec); 
+        print $durPrint; 
+        while (read DATA, $buffer, 262144) { 
+            print $buffer; 
+        } 
+
     }
     close DATA;