synchronize timestamp offsets between h.264 and AAC streams

@@ -30,18 +30,17 @@ window.videojs.Hls.AacStream = function() {
 
 
   this.tags = [];
   this.tags = [];
 
 
-  // (pts:uint, pes_size:int, dataAligned:Boolean):void
-  this.setNextTimeStamp = function(pts, pes_size, dataAligned) {
-
-    // on the first invocation, capture the starting PTS value
+  // (pts:uint):void
+  this.setTimeStampOffset = function(pts) {
     pts_offset = pts;
     pts_offset = pts;
 
 
     // keep track of the last time a metadata tag was written out
     // keep track of the last time a metadata tag was written out
     // set the initial value so metadata will be generated before any
     // set the initial value so metadata will be generated before any
     // payload data
     // payload data
     lastMetaPts = pts - 1000;
     lastMetaPts = pts - 1000;
+  };
 
 
-    // on subsequent invocations, calculate the PTS based on the starting offset
+  // (pts:uint, pes_size:int, dataAligned:Boolean):void
   this.setNextTimeStamp = function(pts, pes_size, dataAligned) {
   this.setNextTimeStamp = function(pts, pes_size, dataAligned) {
     next_pts = pts - pts_offset;
     next_pts = pts - pts_offset;
     pes_length = pes_size;
     pes_length = pes_size;
@@ -52,9 +51,6 @@ window.videojs.Hls.AacStream = function() {
     }
     }
   };
   };
 
 
-    this.setNextTimeStamp(pts, pes_size, dataAligned);
-  };
-
   // (data:ByteArray, o:int = 0, l:int = 0):void
   // (data:ByteArray, o:int = 0, l:int = 0):void
   this.writeBytes = function(data, offset, length) {
   this.writeBytes = function(data, offset, length) {
     var
     var

@@ -258,12 +258,12 @@
 
 
     this.tags = [];
     this.tags = [];
 
 
-    //(pts:uint, dts:uint, dataAligned:Boolean):void
-    this.setNextTimeStamp = function(pts, dts, dataAligned) {
-      // on the first invocation, capture the starting PTS value
+    //(pts:uint):void
+    this.setTimeStampOffset = function(pts) {
       pts_offset = pts;
       pts_offset = pts;
+    };
 
 
-      // on subsequent invocations, calculate the PTS based on the starting offset
+    //(pts:uint, dts:uint, dataAligned:Boolean):void
     this.setNextTimeStamp = function(pts, dts, dataAligned) {
     this.setNextTimeStamp = function(pts, dts, dataAligned) {
       // We could end up with a DTS less than 0 here. We need to deal with that!
       // We could end up with a DTS less than 0 here. We need to deal with that!
       next_pts = pts - pts_offset;
       next_pts = pts - pts_offset;
@@ -275,9 +275,6 @@
       }
       }
     };
     };
 
 
-      this.setNextTimeStamp(pts, dts, dataAligned);
-    };
-
     this.finishFrame = function() {
     this.finishFrame = function() {
       if (h264Frame) {
       if (h264Frame) {
         // Push SPS before EVERY IDR frame for seeking
         // Push SPS before EVERY IDR frame for seeking

@@ -19,7 +19,10 @@
       streamBuffer = new Uint8Array(MP2T_PACKET_LENGTH),
       streamBuffer = new Uint8Array(MP2T_PACKET_LENGTH),
       streamBufferByteCount = 0,
       streamBufferByteCount = 0,
       h264Stream = new H264Stream(),
       h264Stream = new H264Stream(),
-      aacStream = new AacStream();
+      aacStream = new AacStream(),
+      h264HasTimeStampOffset = false,
+      aacHasTimeStampOffset = false,
+      timeStampOffset;
 
 
     // expose the stream metadata
     // expose the stream metadata
     self.stream = {
     self.stream = {
@@ -344,10 +347,24 @@
           }
           }
 
 
           if (pid === self.stream.programMapTable[STREAM_TYPES.h264]) {
           if (pid === self.stream.programMapTable[STREAM_TYPES.h264]) {
+            if (!h264HasTimeStampOffset) {
+              h264HasTimeStampOffset = true;
+              if (timeStampOffset === undefined) {
+                timeStampOffset = pts;
+              }
+              h264Stream.setTimeStampOffset(timeStampOffset);
+            }
             h264Stream.setNextTimeStamp(pts,
             h264Stream.setNextTimeStamp(pts,
                                         dts,
                                         dts,
                                         dataAlignmentIndicator);
                                         dataAlignmentIndicator);
           } else if (pid === self.stream.programMapTable[STREAM_TYPES.adts]) {
           } else if (pid === self.stream.programMapTable[STREAM_TYPES.adts]) {
+            if (!aacHasTimeStampOffset) {
+              aacHasTimeStampOffset = true;
+              if (timeStampOffset === undefined) {
+                timeStampOffset = pts;
+              }
+              aacStream.setTimeStampOffset(timeStampOffset);
+            }
             aacStream.setNextTimeStamp(pts,
             aacStream.setNextTimeStamp(pts,
                                        pesPacketSize,
                                        pesPacketSize,
                                        dataAlignmentIndicator);
                                        dataAlignmentIndicator);

@@ -70,6 +70,7 @@ test('starting PTS values can be negative', function() {
       nalUnitTypes.access_unit_delimiter_rbsp
       nalUnitTypes.access_unit_delimiter_rbsp
     ]);
     ]);
 
 
+  h264Stream.setTimeStampOffset(-100);
   h264Stream.setNextTimeStamp(-100, -100, true);
   h264Stream.setNextTimeStamp(-100, -100, true);
   h264Stream.writeBytes(accessUnitDelimiter, 0, accessUnitDelimiter.byteLength);
   h264Stream.writeBytes(accessUnitDelimiter, 0, accessUnitDelimiter.byteLength);
   h264Stream.setNextTimeStamp(-99, -99, true);
   h264Stream.setNextTimeStamp(-99, -99, true);