xref: /aosp_15_r20/external/webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h (revision d9f758449e529ab9291ac668be2861e7a55c2422)
1 /*
2  *  Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 //
12 // This file contains the declaration of the VP9 packetizer class.
13 // A packetizer object is created for each encoded video frame. The
14 // constructor is called with the payload data and size.
15 //
16 // After creating the packetizer, the method NextPacket is called
17 // repeatedly to get all packets for the frame. The method returns
18 // false as long as there are more packets left to fetch.
19 //
20 
21 #ifndef MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP9_H_
22 #define MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP9_H_
23 
24 #include <stddef.h>
25 #include <stdint.h>
26 
27 #include <vector>
28 
29 #include "api/array_view.h"
30 #include "modules/rtp_rtcp/source/rtp_format.h"
31 #include "modules/rtp_rtcp/source/rtp_packet_to_send.h"
32 #include "modules/video_coding/codecs/vp9/include/vp9_globals.h"
33 
34 namespace webrtc {
35 
36 class RtpPacketizerVp9 : public RtpPacketizer {
37  public:
38   // The `payload` must be one encoded VP9 layer frame.
39   RtpPacketizerVp9(rtc::ArrayView<const uint8_t> payload,
40                    PayloadSizeLimits limits,
41                    const RTPVideoHeaderVP9& hdr);
42 
43   ~RtpPacketizerVp9() override;
44 
45   RtpPacketizerVp9(const RtpPacketizerVp9&) = delete;
46   RtpPacketizerVp9& operator=(const RtpPacketizerVp9&) = delete;
47 
48   size_t NumPackets() const override;
49 
50   // Gets the next payload with VP9 payload header.
51   // Write payload and set marker bit of the `packet`.
52   // Returns true on success, false otherwise.
53   bool NextPacket(RtpPacketToSend* packet) override;
54 
55  private:
56   // Writes the payload descriptor header.
57   // `layer_begin` and `layer_end` indicates the postision of the packet in
58   // the layer frame. Returns false on failure.
59   bool WriteHeader(bool layer_begin,
60                    bool layer_end,
61                    rtc::ArrayView<uint8_t> rtp_payload) const;
62 
63   const RTPVideoHeaderVP9 hdr_;
64   const int header_size_;
65   const int first_packet_extra_header_size_;
66   rtc::ArrayView<const uint8_t> remaining_payload_;
67   std::vector<int> payload_sizes_;
68   std::vector<int>::const_iterator current_packet_;
69 };
70 
71 }  // namespace webrtc
72 #endif  // MODULES_RTP_RTCP_SOURCE_RTP_FORMAT_VP9_H_
73