OLD | NEW |
---|---|
(Empty) | |
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 #include "media/cast/framer/framer.h" | |
6 | |
7 #include "base/logging.h" | |
8 | |
9 namespace media { | |
10 namespace cast { | |
11 | |
12 typedef FrameList::const_iterator ConstFrameIterator; | |
13 | |
14 Framer::Framer(RtpPayloadFeedback* incoming_payload_feedback, | |
15 uint32 ssrc, | |
16 bool decoder_faster_than_max_frame_rate, | |
17 int max_unacked_frames) | |
18 : decoder_faster_than_max_frame_rate_(decoder_faster_than_max_frame_rate), | |
19 clock_(&default_tick_clock_), | |
20 cast_msg_builder_(new CastMessageBuilder(incoming_payload_feedback, | |
21 &frame_id_map_, ssrc, decoder_faster_than_max_frame_rate, | |
22 max_unacked_frames)) { | |
23 DCHECK(incoming_payload_feedback) << "Invalid argument"; | |
24 } | |
25 | |
26 Framer::~Framer() {} | |
27 | |
28 void Framer::InsertPacket(const uint8* payload_data, | |
29 int payload_size, | |
30 const RtpCastHeader& rtp_header) { | |
31 bool complete = false; | |
32 if (!frame_id_map_.InsertPacket(rtp_header, &complete)) return; | |
33 | |
34 // Does this packet belong to a new frame? | |
35 FrameList::iterator it = frames_.find(rtp_header.frame_id); | |
36 if (it == frames_.end()) { | |
37 // New frame. | |
38 FrameBuffer* frame_buffer = new FrameBuffer(); | |
Alpha Left Google
2013/08/28 19:13:48
Should be linked_ptr<FrameBuffer> frame_buffer = .
pwestin
2013/08/28 20:37:21
Done.
| |
39 frame_buffer->InsertPacket(payload_data, payload_size, rtp_header); | |
40 frames_.insert(std::make_pair(rtp_header.frame_id, frame_buffer)); | |
41 } else { | |
42 // Insert packet to existing frame buffer. | |
43 it->second->InsertPacket(payload_data, payload_size, rtp_header); | |
44 } | |
45 | |
46 if (complete) { | |
47 // ACK as soon as possible. | |
48 cast_msg_builder_->CompleteFrameReceived(rtp_header.frame_id, | |
49 rtp_header.is_key_frame); | |
50 } | |
51 } | |
52 | |
53 // This does not release the frame. | |
54 bool Framer::GetEncodedAudioFrame(const base::TimeTicks& timeout, | |
55 EncodedAudioFrame* audio_frame, | |
56 uint32* rtp_timestamp, | |
57 bool* next_frame) { | |
58 uint8 frame_id; | |
59 // Find frame id. | |
60 if (frame_id_map_.NextContinuousFrame(&frame_id)) { | |
61 // We have our next frame. | |
62 *next_frame = true; | |
63 } else { | |
64 if (WaitForNextFrame(timeout)) return false; | |
65 | |
66 if (!frame_id_map_.NextAudioFrameAllowingMissingFrames(&frame_id)) { | |
67 return false; | |
68 } | |
69 *next_frame = false; | |
70 } | |
71 | |
72 ConstFrameIterator it = frames_.find(frame_id); | |
73 DCHECK(it != frames_.end()); | |
74 if (it == frames_.end()) return false; | |
75 | |
76 return it->second->GetEncodedAudioFrame(audio_frame, rtp_timestamp); | |
77 } | |
78 | |
79 // This does not release the frame. | |
80 bool Framer::GetEncodedVideoFrame(const base::TimeTicks& timeout, | |
81 EncodedVideoFrame* video_frame, | |
82 uint32* rtp_timestamp, | |
83 bool* next_frame) { | |
84 uint8 frame_id; | |
85 // Find frame id. | |
86 if (frame_id_map_.NextContinuousFrame(&frame_id)) { | |
87 // We have our next frame. | |
88 *next_frame = true; | |
89 } else { | |
90 if (WaitForNextFrame(timeout)) return false; | |
91 | |
92 // Check if we can skip frames when our decoder is too slow. | |
93 if (!decoder_faster_than_max_frame_rate_) return false; | |
94 | |
95 if (!frame_id_map_.NextVideoFrameAllowingSkippingFrames(&frame_id)) { | |
96 return false; | |
97 } | |
98 *next_frame = false; | |
99 } | |
100 | |
101 ConstFrameIterator it = frames_.find(frame_id); | |
102 DCHECK(it != frames_.end()); | |
103 if (it == frames_.end()) return false; | |
104 | |
105 return it->second->GetEncodedVideoFrame(video_frame, rtp_timestamp); | |
106 } | |
107 | |
108 bool Framer::WaitForNextFrame(const base::TimeTicks& timeout) const { | |
109 base::TimeDelta wait_time = timeout - clock_->NowTicks(); | |
110 if (wait_time.InMilliseconds() > 0) | |
111 return true; | |
112 | |
113 return false; | |
114 } | |
115 | |
116 void Framer::Reset() { | |
117 frame_id_map_.Clear(); | |
118 frames_.clear(); | |
119 cast_msg_builder_->Reset(); | |
120 } | |
121 | |
122 void Framer::ReleaseFrame(uint8 frame_id) { | |
123 frame_id_map_.RemoveOldFrames(frame_id); | |
124 frames_.erase(frame_id); | |
125 | |
126 // We have a frame - remove all frames with lower frame id. | |
127 FrameList::iterator it; | |
128 for (it = frames_.begin(); it != frames_.end(); ) { | |
129 if (IsOlderFrameId(it->first, frame_id)) { | |
130 frames_.erase(it++); | |
131 } else { | |
132 ++it; | |
133 } | |
134 } | |
135 } | |
136 | |
137 bool Framer::TimeToSendNextCastMessage(base::TimeTicks* time_to_send) { | |
138 return cast_msg_builder_->TimeToSendNextCastMessage(time_to_send); | |
139 } | |
140 | |
141 void Framer::SendCastMessage() { | |
142 cast_msg_builder_->UpdateCastMessage(); | |
143 } | |
144 | |
145 } // namespace cast | |
146 } // namespace media | |
OLD | NEW |