sps_parser.h 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. /*
  2. * Copyright (c) 2016 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. #ifndef COMMON_VIDEO_H264_SPS_PARSER_H_
  11. #define COMMON_VIDEO_H264_SPS_PARSER_H_
  12. #include "absl/types/optional.h"
  13. namespace rtc {
  14. class BitBuffer;
  15. }
  16. namespace webrtc {
  17. // A class for parsing out sequence parameter set (SPS) data from an H264 NALU.
  18. class SpsParser {
  19. public:
  20. // The parsed state of the SPS. Only some select values are stored.
  21. // Add more as they are actually needed.
  22. struct SpsState {
  23. SpsState();
  24. SpsState(const SpsState&);
  25. ~SpsState();
  26. uint32_t width = 0;
  27. uint32_t height = 0;
  28. uint32_t delta_pic_order_always_zero_flag = 0;
  29. uint32_t separate_colour_plane_flag = 0;
  30. uint32_t frame_mbs_only_flag = 0;
  31. uint32_t log2_max_frame_num = 4; // Smallest valid value.
  32. uint32_t log2_max_pic_order_cnt_lsb = 4; // Smallest valid value.
  33. uint32_t pic_order_cnt_type = 0;
  34. uint32_t max_num_ref_frames = 0;
  35. uint32_t vui_params_present = 0;
  36. uint32_t id = 0;
  37. };
  38. // Unpack RBSP and parse SPS state from the supplied buffer.
  39. static absl::optional<SpsState> ParseSps(const uint8_t* data, size_t length);
  40. protected:
  41. // Parse the SPS state, up till the VUI part, for a bit buffer where RBSP
  42. // decoding has already been performed.
  43. static absl::optional<SpsState> ParseSpsUpToVui(rtc::BitBuffer* buffer);
  44. };
  45. } // namespace webrtc
  46. #endif // COMMON_VIDEO_H264_SPS_PARSER_H_