To differentiate between when a packet should be dropped or when a keyframe is missing its SPS and/or SPS CopyAndFixBitstream now returns: - kInsert, the packet should be inserted into the PacketBuffer. - kDrop, the packet should be dropped. - kRequestKeyframe, a keyframe should be requested. BUG=webrtc:5514 Review-Url: https://codereview.webrtc.org/2477343002 Cr-Commit-Position: refs/heads/master@{#14948}
52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
/*
|
|
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef WEBRTC_MODULES_VIDEO_CODING_H264_SPS_PPS_TRACKER_H_
|
|
#define WEBRTC_MODULES_VIDEO_CODING_H264_SPS_PPS_TRACKER_H_
|
|
|
|
#include <cstdint>
|
|
#include <map>
|
|
#include <memory>
|
|
|
|
#include "webrtc/modules/include/module_common_types.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class VCMPacket;
|
|
|
|
namespace video_coding {
|
|
|
|
class H264SpsPpsTracker {
|
|
public:
|
|
enum PacketAction { kInsert, kDrop, kRequestKeyframe };
|
|
|
|
PacketAction CopyAndFixBitstream(VCMPacket* packet);
|
|
|
|
private:
|
|
struct PpsInfo {
|
|
int sps_id = -1;
|
|
size_t size = 0;
|
|
std::unique_ptr<uint8_t[]> data;
|
|
};
|
|
|
|
struct SpsInfo {
|
|
size_t size = 0;
|
|
std::unique_ptr<uint8_t[]> data;
|
|
};
|
|
|
|
std::map<uint32_t, PpsInfo> pps_data_;
|
|
std::map<uint32_t, SpsInfo> sps_data_;
|
|
};
|
|
|
|
} // namespace video_coding
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_MODULES_VIDEO_CODING_H264_SPS_PPS_TRACKER_H_
|