MSE: Add StreamParser buffer remuxing utility and tests

media/base/stream_parser.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef MEDIA_BASE_STREAM_PARSER_H_
 #define MEDIA_BASE_STREAM_PARSER_H_
 
 #include <deque>
 #include <map>
 #include <string>
+#include <vector>
 
 #include "base/callback_forward.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/time/time.h"
 #include "media/base/media_export.h"
 #include "media/base/media_log.h"
 
 namespace media {
 
 class AudioDecoderConfig;
 class StreamParserBuffer;
 class TextTrackConfig;
 class VideoDecoderConfig;
 
 // Abstract interface for parsing media byte streams.
 class MEDIA_EXPORT StreamParser {
  public:
   typedef std::deque<scoped_refptr<StreamParserBuffer> > BufferQueue;
 
-  // Map of text track number to the track configuration.
-  typedef std::map<int, TextTrackConfig> TextTrackConfigMap;
+  // Range of |TrackId| is dependent upon stream parsers. It is currently
+  // the key for the buffer's text track config in the applicable
+  // TextTrackConfigMap (which is passed in StreamParser::NewConfigCB), or
+  // 0 for other media types that currently allow at most one track.
+  // WebMTracksParser uses -1 as an invalid text track number.
+  // TODO(wolenetz/acolwell): Change to size_type while fixing stream parsers to
+  // emit validated track configuration and buffer vectors rather than max 1
+  // audio, max 1 video, and N text tracks in a map keyed by
+  // bytestream-specific-ranged track numbers. See http://crbug.com/341581.
+  typedef int TrackId;
 
-  // Map of text track number to decode-timestamp-ordered buffers for the track.
-  typedef std::map<int, const BufferQueue> TextBufferQueueMap;
+  // Map of text track ID to the track configuration.
+  typedef std::map<TrackId, TextTrackConfig> TextTrackConfigMap;
+
+  // Map of text track ID to decode-timestamp-ordered buffers for the track.
+  typedef std::map<TrackId, const BufferQueue> TextBufferQueueMap;
 
   StreamParser();
   virtual ~StreamParser();
 
   // Indicates completion of parser initialization.
   // First parameter - Indicates initialization success. Set to true if
   //                   initialization was successful. False if an error
   //                   occurred.
   // Second parameter -  Indicates the stream duration. Only contains a valid
   //                     value if the first parameter is true.
@@ skipping 58 matching lines @@
 
   // Called when there is new data to parse.
   //
   // Returns true if the parse succeeds.
   virtual bool Parse(const uint8* buf, int size) = 0;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(StreamParser);
 };
 
+// Appends to |merged_buffers| the provided buffers in decode-timestamp order.
+// Any previous contents of |merged_buffers| is assumed to have lower
+// decode timestamps versus the provided buffers. All provided buffer queues
+// are assumed to already be in decode-timestamp order.
+// Returns false if any of the provided audio/video/text buffers are found
+// to not be in decode timestamp order, or have a decode timestamp less than
+// the last buffer, if any, in |merged_buffers|. Partial results may exist
+// in |merged_buffers| in this case. Returns true on success.
+// No validation of media type within the various buffer queues is done here.
+// TODO(wolenetz/acolwell): Merge incrementally in parsers to eliminate
+// subtle issues with tie-breaking. See http://crbug.com/338484.
+MEDIA_EXPORT bool MergeBufferQueues(
+    const StreamParser::BufferQueue& audio_buffers,
+    const StreamParser::BufferQueue& video_buffers,
+    const StreamParser::TextBufferQueueMap& text_buffers,
+    StreamParser::BufferQueue* merged_buffers);
+
 }  // namespace media
 
 #endif  // MEDIA_BASE_STREAM_PARSER_H_