Home | History | Annotate | Download | only in mp2t 
      1 // Copyright 2014 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_FORMATS_MP2T_ES_PARSER_H264_H_
      6 #define MEDIA_FORMATS_MP2T_ES_PARSER_H264_H_
      7 
      8 #include <list>
      9 #include <utility>
     10 
     11 #include "base/basictypes.h"
     12 #include "base/callback.h"
     13 #include "base/compiler_specific.h"
     14 #include "base/memory/scoped_ptr.h"
     15 #include "base/time/time.h"
     16 #include "media/base/media_export.h"
     17 #include "media/base/video_decoder_config.h"
     18 #include "media/formats/mp2t/es_parser.h"
     19 
     20 namespace media {
     21 class H264Parser;
     22 struct H264SPS;
     23 class OffsetByteQueue;
     24 }
     25 
     26 namespace media {
     27 namespace mp2t {
     28 
     29 // Remark:
     30 // In this h264 parser, frame splitting is based on AUD nals.
     31 // Mpeg2 TS spec: "2.14 Carriage of Rec. ITU-T H.264 | ISO/IEC 14496-10 video"
     32 // "Each AVC access unit shall contain an access unit delimiter NAL Unit;"
     33 //
     34 class MEDIA_EXPORT EsParserH264 : NON_EXPORTED_BASE(public EsParser) {
     35  public:
     36   typedef base::Callback<void(const VideoDecoderConfig&)> NewVideoConfigCB;
     37 
     38   EsParserH264(const NewVideoConfigCB& new_video_config_cb,
     39                const EmitBufferCB& emit_buffer_cb);
     40   virtual ~EsParserH264();
     41 
     42   // EsParser implementation.
     43   virtual bool Parse(const uint8* buf, int size,
     44                      base::TimeDelta pts,
     45                      base::TimeDelta dts) OVERRIDE;
     46   virtual void Flush() OVERRIDE;
     47   virtual void Reset() OVERRIDE;
     48 
     49  private:
     50   struct TimingDesc {
     51     base::TimeDelta dts;
     52     base::TimeDelta pts;
     53   };
     54 
     55   // Find the AUD located at or after |*stream_pos|.
     56   // Return true if an AUD is found.
     57   // If found, |*stream_pos| corresponds to the position of the AUD start code
     58   // in the stream. Otherwise, |*stream_pos| corresponds to the last position
     59   // of the start code parser.
     60   bool FindAUD(int64* stream_pos);
     61 
     62   // Resumes the H264 ES parsing.
     63   // Return true if successful.
     64   bool ParseInternal();
     65 
     66   // Emit a frame whose position in the ES queue starts at |access_unit_pos|.
     67   // Returns true if successful, false if no PTS is available for the frame.
     68   bool EmitFrame(int64 access_unit_pos, int access_unit_size,
     69                  bool is_key_frame, int pps_id);
     70 
     71   // Update the video decoder config based on an H264 SPS.
     72   // Return true if successful.
     73   bool UpdateVideoDecoderConfig(const H264SPS* sps);
     74 
     75   // Callbacks to pass the stream configuration and the frames.
     76   NewVideoConfigCB new_video_config_cb_;
     77   EmitBufferCB emit_buffer_cb_;
     78 
     79   // Bytes of the ES stream that have not been emitted yet.
     80   scoped_ptr<media::OffsetByteQueue> es_queue_;
     81   std::list<std::pair<int64, TimingDesc> > timing_desc_list_;
     82 
     83   // H264 parser state.
     84   // - |current_access_unit_pos_| is pointing to an annexB syncword
     85   // representing the first NALU of an H264 access unit.
     86   scoped_ptr<H264Parser> h264_parser_;
     87   int64 current_access_unit_pos_;
     88   int64 next_access_unit_pos_;
     89 
     90   // Last video decoder config.
     91   VideoDecoderConfig last_video_decoder_config_;
     92 };
     93 
     94 }  // namespace mp2t
     95 }  // namespace media
     96 
     97 #endif
     98 
     99