1 /*
2  *  Copyright (c) 2019 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 #ifndef MODULES_RTP_RTCP_SOURCE_RTP_PACKETIZER_AV1_H_
12 #define MODULES_RTP_RTCP_SOURCE_RTP_PACKETIZER_AV1_H_
13 
14 #include <stddef.h>
15 #include <stdint.h>
16 
17 #include <vector>
18 
19 #include "api/array_view.h"
20 #include "api/video/video_frame_type.h"
21 #include "modules/rtp_rtcp/source/rtp_format.h"
22 
23 namespace webrtc {
24 
25 class RtpPacketizerAv1 : public RtpPacketizer {
26  public:
27   RtpPacketizerAv1(rtc::ArrayView<const uint8_t> payload,
28                    PayloadSizeLimits limits,
29                    VideoFrameType frame_type,
30                    bool is_last_frame_in_picture);
31   ~RtpPacketizerAv1() override = default;
32 
NumPackets()33   size_t NumPackets() const override { return packets_.size() - packet_index_; }
34   bool NextPacket(RtpPacketToSend* packet) override;
35 
36  private:
37   struct Obu {
38     uint8_t header;
39     uint8_t extension_header;  // undefined if (header & kXbit) == 0
40     rtc::ArrayView<const uint8_t> payload;
41     int size;  // size of the header and payload combined.
42   };
43   struct Packet {
PacketPacket44     explicit Packet(int first_obu_index) : first_obu(first_obu_index) {}
45     // Indexes into obus_ vector of the first and last obus that should put into
46     // the packet.
47     int first_obu;
48     int num_obu_elements = 0;
49     int first_obu_offset = 0;
50     int last_obu_size;
51     // Total size consumed by the packet.
52     int packet_size = 0;
53   };
54 
55   // Parses the payload into serie of OBUs.
56   static std::vector<Obu> ParseObus(rtc::ArrayView<const uint8_t> payload);
57   // Returns the number of additional bytes needed to store the previous OBU
58   // element if an additonal OBU element is added to the packet.
59   static int AdditionalBytesForPreviousObuElement(const Packet& packet);
60   static std::vector<Packet> Packetize(rtc::ArrayView<const Obu> obus,
61                                        PayloadSizeLimits limits);
62   uint8_t AggregationHeader() const;
63 
64   const VideoFrameType frame_type_;
65   const std::vector<Obu> obus_;
66   const std::vector<Packet> packets_;
67   const bool is_last_frame_in_picture_;
68   size_t packet_index_ = 0;
69 };
70 
71 }  // namespace webrtc
72 #endif  // MODULES_RTP_RTCP_SOURCE_RTP_PACKETIZER_AV1_H_
73