media/mp2t/mp2t_stream_parser.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef MEDIA_MP2T_MP2T_STREAM_PARSER_H_
   6 #define MEDIA_MP2T_MP2T_STREAM_PARSER_H_
   7
   8 #include <list>
   9 #include <map>
  10
  11 #include "base/memory/ref_counted.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "media/base/audio_decoder_config.h"
  14 #include "media/base/byte_queue.h"
  15 #include "media/base/media_export.h"
  16 #include "media/base/stream_parser.h"
  17 #include "media/base/video_decoder_config.h"
  18
  19 namespace media {
  20
  21 class StreamParserBuffer;
  22
  23 namespace mp2t {
  24
  25 class PidState;
  26
  27 class MEDIA_EXPORT Mp2tStreamParser : public StreamParser {
  28  public:
  29   Mp2tStreamParser();
  30   virtual ~Mp2tStreamParser();
  31
  32   // StreamParser implementation.
  33   virtual void Init(const InitCB& init_cb,
  34                     const NewConfigCB& config_cb,
  35                     const NewBuffersCB& new_buffers_cb,
  36                     const NewTextBuffersCB& text_cb,
  37                     const NeedKeyCB& need_key_cb,
  38                     const AddTextTrackCB& add_text_track_cb,
  39                     const NewMediaSegmentCB& new_segment_cb,
  40                     const base::Closure& end_of_segment_cb,
  41                     const LogCB& log_cb) OVERRIDE;
  42   virtual void Flush() OVERRIDE;
  43   virtual bool Parse(const uint8* buf, int size) OVERRIDE;
  44
  45  private:
  46   typedef std::map<int, PidState*> PidMap;
  47
  48   struct BufferQueueWithConfig {
  49     BufferQueueWithConfig(bool is_cfg_sent,
  50                           const AudioDecoderConfig& audio_cfg,
  51                           const VideoDecoderConfig& video_cfg);
  52     ~BufferQueueWithConfig();
  53
  54     bool is_config_sent;
  55     AudioDecoderConfig audio_config;
  56     StreamParser::BufferQueue audio_queue;
  57     VideoDecoderConfig video_config;
  58     StreamParser::BufferQueue video_queue;
  59   };
  60
  61   // Callback invoked to register a Program Map Table.
  62   // Note: Does nothing if the PID is already registered.
  63   void RegisterPmt(int program_number, int pmt_pid);
  64
  65   // Callback invoked to register a PES pid.
  66   // Possible values for |stream_type| are defined in:
  67   // ISO-13818.1 / ITU H.222 Table 2.34 "Stream type assignments".
  68   // |pes_pid| is part of the Program Map Table refered by |pmt_pid|.
  69   void RegisterPes(int pmt_pid, int pes_pid, int stream_type);
  70
  71   // Since the StreamParser interface allows only one audio & video streams,
  72   // an automatic PID filtering should be applied to select the audio & video
  73   // streams.
  74   void UpdatePidFilter();
  75
  76   // Callback invoked each time the audio/video decoder configuration is
  77   // changed.
  78   void OnVideoConfigChanged(int pes_pid,
  79                             const VideoDecoderConfig& video_decoder_config);
  80   void OnAudioConfigChanged(int pes_pid,
  81                             const AudioDecoderConfig& audio_decoder_config);
  82
  83   // Invoke the initialization callback if needed.
  84   bool FinishInitializationIfNeeded();
  85
  86   // Callback invoked by the ES stream parser
  87   // to emit a new audio/video access unit.
  88   void OnEmitAudioBuffer(
  89       int pes_pid,
  90       scoped_refptr<StreamParserBuffer> stream_parser_buffer);
  91   void OnEmitVideoBuffer(
  92       int pes_pid,
  93       scoped_refptr<StreamParserBuffer> stream_parser_buffer);
  94   bool EmitRemainingBuffers();
  95
  96   // List of callbacks.
  97   InitCB init_cb_;
  98   NewConfigCB config_cb_;
  99   NewBuffersCB new_buffers_cb_;
 100   NeedKeyCB need_key_cb_;
 101   NewMediaSegmentCB new_segment_cb_;
 102   base::Closure end_of_segment_cb_;
 103   LogCB log_cb_;
 104
 105   // Bytes of the TS stream.
 106   ByteQueue ts_byte_queue_;
 107
 108   // List of PIDs and their state.
 109   PidMap pids_;
 110
 111   // Selected audio and video PIDs.
 112   int selected_audio_pid_;
 113   int selected_video_pid_;
 114
 115   // Pending audio & video buffers.
 116   std::list<BufferQueueWithConfig> buffer_queue_chain_;
 117
 118   // Whether |init_cb_| has been invoked.
 119   bool is_initialized_;
 120
 121   // Indicate whether a segment was started.
 122   bool segment_started_;
 123   bool first_video_frame_in_segment_;
 124   base::TimeDelta time_offset_;
 125
 126   DISALLOW_COPY_AND_ASSIGN(Mp2tStreamParser);
 127 };
 128
 129 }  // namespace mp2t
 130 }  // namespace media
 131
 132 #endif
 133