OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 // This file contains an implementation of an H.264 Decoded Picture Buffer | 5 // This file contains an implementation of an H.264 Decoded Picture Buffer |
6 // used in H264 decoders. | 6 // used in H264 decoders. |
7 | 7 |
8 #ifndef CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ | 8 #ifndef CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ |
9 #define CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ | 9 #define CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ |
10 | 10 |
11 #include <vector> | 11 #include <vector> |
12 | 12 |
13 #include "base/basictypes.h" | 13 #include "base/basictypes.h" |
14 #include "base/memory/ref_counted.h" | 14 #include "base/memory/ref_counted.h" |
15 #include "media/filters/h264_parser.h" | 15 #include "media/filters/h264_parser.h" |
16 | 16 |
17 namespace content { | 17 namespace content { |
18 | 18 |
19 class V4L2H264Picture; | 19 class V4L2H264Picture; |
20 class VaapiH264Picture; | |
20 | 21 |
21 // A picture (a frame or a field) in the H.264 spec sense. | 22 // A picture (a frame or a field) in the H.264 spec sense. |
22 // See spec at http://www.itu.int/rec/T-REC-H.264 | 23 // See spec at http://www.itu.int/rec/T-REC-H.264 |
23 struct H264PictureBase { | 24 struct H264Picture : public base::RefCounted<H264Picture> { |
wuchengli
2015/03/30 11:02:54
This is complicated enough to be a class.
Pawel Osciak
2015/04/03 07:06:00
Done.
| |
25 H264Picture(); | |
26 | |
27 virtual V4L2H264Picture* AsV4L2H264Picture(); | |
28 virtual VaapiH264Picture* AsVaapiH264Picture(); | |
29 | |
30 using Vector = std::vector<scoped_refptr<H264Picture>>; | |
31 | |
24 enum Field { | 32 enum Field { |
25 FIELD_NONE, | 33 FIELD_NONE, |
26 FIELD_TOP, | 34 FIELD_TOP, |
27 FIELD_BOTTOM, | 35 FIELD_BOTTOM, |
28 }; | 36 }; |
29 | 37 |
30 H264PictureBase(); | |
31 | |
32 // Values calculated per H.264 specification or taken from slice header. | 38 // Values calculated per H.264 specification or taken from slice header. |
33 // See spec for more details on each (some names have been converted from | 39 // See spec for more details on each (some names have been converted from |
34 // CamelCase in spec to Chromium-style names). | 40 // CamelCase in spec to Chromium-style names). |
35 int top_field_order_cnt; | 41 int top_field_order_cnt; |
36 int bottom_field_order_cnt; | 42 int bottom_field_order_cnt; |
37 int pic_order_cnt; | 43 int pic_order_cnt; |
38 int pic_order_cnt_msb; | 44 int pic_order_cnt_msb; |
39 int pic_order_cnt_lsb; | 45 int pic_order_cnt_lsb; |
40 | 46 |
41 int pic_num; | 47 int pic_num; |
(...skipping 16 matching lines...) Expand all Loading... | |
58 | 64 |
59 // Values from slice_hdr to be used during reference marking and | 65 // Values from slice_hdr to be used during reference marking and |
60 // memory management after finishing this picture. | 66 // memory management after finishing this picture. |
61 bool long_term_reference_flag; | 67 bool long_term_reference_flag; |
62 bool adaptive_ref_pic_marking_mode_flag; | 68 bool adaptive_ref_pic_marking_mode_flag; |
63 media::H264DecRefPicMarking | 69 media::H264DecRefPicMarking |
64 ref_pic_marking[media::H264SliceHeader::kRefListSize]; | 70 ref_pic_marking[media::H264SliceHeader::kRefListSize]; |
65 | 71 |
66 // Position in DPB (i.e. index in DPB). | 72 // Position in DPB (i.e. index in DPB). |
67 int dpb_position; | 73 int dpb_position; |
68 }; | |
69 | |
70 class H264Picture : public H264PictureBase, | |
71 public base::RefCounted<H264Picture> { | |
72 public: | |
73 H264Picture(); | |
74 | |
75 virtual V4L2H264Picture* AsV4L2H264Picture(); | |
76 | |
77 using Vector = std::vector<scoped_refptr<H264Picture>>; | |
78 | 74 |
79 protected: | 75 protected: |
80 friend class base::RefCounted<H264Picture>; | 76 friend class base::RefCounted<H264Picture>; |
81 virtual ~H264Picture(); | 77 virtual ~H264Picture(); |
82 | 78 |
83 DISALLOW_COPY_AND_ASSIGN(H264Picture); | 79 DISALLOW_COPY_AND_ASSIGN(H264Picture); |
wuchengli
2015/03/30 11:02:54
This should be in private section according to bas
Pawel Osciak
2015/04/03 07:06:00
Done.
| |
84 }; | 80 }; |
85 | 81 |
86 // DPB - Decoded Picture Buffer. | 82 // DPB - Decoded Picture Buffer. |
87 // Stores decoded pictures that will be used for future display | 83 // Stores decoded pictures that will be used for future display |
88 // and/or reference. | 84 // and/or reference. |
89 class H264DPB { | 85 class H264DPB { |
90 public: | 86 public: |
91 H264DPB(); | 87 H264DPB(); |
92 ~H264DPB(); | 88 ~H264DPB(); |
93 | 89 |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
132 | 128 |
133 // Append all long term reference pictures to the passed |out| vector. | 129 // Append all long term reference pictures to the passed |out| vector. |
134 void GetLongTermRefPicsAppending(H264Picture::Vector* out); | 130 void GetLongTermRefPicsAppending(H264Picture::Vector* out); |
135 | 131 |
136 // Iterators for direct access to DPB contents. | 132 // Iterators for direct access to DPB contents. |
137 // Will be invalidated after any of Remove* calls. | 133 // Will be invalidated after any of Remove* calls. |
138 H264Picture::Vector::iterator begin() { return pics_.begin(); } | 134 H264Picture::Vector::iterator begin() { return pics_.begin(); } |
139 H264Picture::Vector::iterator end() { return pics_.end(); } | 135 H264Picture::Vector::iterator end() { return pics_.end(); } |
140 H264Picture::Vector::const_iterator begin() const { return pics_.begin(); } | 136 H264Picture::Vector::const_iterator begin() const { return pics_.begin(); } |
141 H264Picture::Vector::const_iterator end() const { return pics_.end(); } | 137 H264Picture::Vector::const_iterator end() const { return pics_.end(); } |
142 H264Picture::Vector::reverse_iterator rbegin() { return pics_.rbegin(); } | 138 H264Picture::Vector::const_reverse_iterator rbegin() const { |
143 H264Picture::Vector::reverse_iterator rend() { return pics_.rend(); } | 139 return pics_.rbegin(); |
140 } | |
141 H264Picture::Vector::const_reverse_iterator rend() const { | |
142 return pics_.rend(); | |
143 } | |
144 | 144 |
145 size_t size() const { return pics_.size(); } | 145 size_t size() const { return pics_.size(); } |
146 bool IsFull() const { return pics_.size() == max_num_pics_; } | 146 bool IsFull() const { return pics_.size() == max_num_pics_; } |
147 | 147 |
148 // Per H264 spec, increase to 32 if interlaced video is supported. | 148 // Per H264 spec, increase to 32 if interlaced video is supported. |
149 enum { kDPBMaxSize = 16, }; | 149 enum { kDPBMaxSize = 16, }; |
150 | 150 |
151 private: | 151 private: |
152 void UpdatePicPositions(); | 152 void UpdatePicPositions(); |
153 | 153 |
154 H264Picture::Vector pics_; | 154 H264Picture::Vector pics_; |
155 size_t max_num_pics_; | 155 size_t max_num_pics_; |
156 | 156 |
157 DISALLOW_COPY_AND_ASSIGN(H264DPB); | 157 DISALLOW_COPY_AND_ASSIGN(H264DPB); |
158 }; | 158 }; |
159 | 159 |
160 } // namespace content | 160 } // namespace content |
161 | 161 |
162 #endif // CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ | 162 #endif // CONTENT_COMMON_GPU_MEDIA_H264_DPB_H_ |
OLD | NEW |