decoder_configuration_record.h

// Copyright 2016 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
 
#ifndef PACKAGER_MEDIA_CODECS_DECODER_CONFIGURATION_RECORD_H_
#define PACKAGER_MEDIA_CODECS_DECODER_CONFIGURATION_RECORD_H_
 
#include <cstdint>
#include <vector>
 
#include <absl/log/check.h>
#include <absl/log/log.h>
 
#include <packager/macros/classes.h>
#include <packager/media/codecs/nalu_reader.h>
 
namespace shaka {
namespace media {
 
// Defines a base class for decoder configuration record.
class DecoderConfigurationRecord {
 public:
  virtual ~DecoderConfigurationRecord();
 
  bool Parse(const std::vector<uint8_t>& data) {
    return Parse(data.data(), data.size());
  }
 
  bool Parse(const uint8_t* data, size_t data_size);
 
  uint8_t nalu_length_size() const { return nalu_length_size_; }
 
  size_t nalu_count() const { return nalu_.size(); }
 
  const Nalu& nalu(size_t i) const { return nalu_[i]; }
 
  uint8_t transfer_characteristics() const { return transfer_characteristics_; }
 
  uint8_t color_primaries() const { return color_primaries_; }
 
  uint8_t matrix_coefficients() const { return matrix_coefficients_; }
 
 protected:
  DecoderConfigurationRecord();
 
  void AddNalu(const Nalu& nalu);
 
  const uint8_t* data() const { return data_.data(); }
 
  size_t data_size() const { return data_.size(); }
 
  void set_nalu_length_size(uint8_t nalu_length_size) {
    DCHECK(nalu_length_size <= 2 || nalu_length_size == 4);
    nalu_length_size_ = nalu_length_size;
  }
 
  void set_transfer_characteristics(uint8_t transfer_characteristics) {
    transfer_characteristics_ = transfer_characteristics;
  }
 
  void set_color_primaries(uint8_t color_primaries) {
    color_primaries_ = color_primaries;
  }
  void set_matrix_coefficients(uint8_t matrix_coefficients) {
    matrix_coefficients_ = matrix_coefficients;
  }
 
 private:
  // Performs the actual parsing of the data.
  virtual bool ParseInternal() = 0;
 
  // Contains a copy of the data.  This manages the pointer lifetime so the
  // extracted Nalu can accessed.
  std::vector<uint8_t> data_;
  std::vector<Nalu> nalu_;
  uint8_t nalu_length_size_ = 0;
 
  // Indicates the opto-electronic transfer characteristics of the source
  // picture, which can be used to determine whether the video is HDR or SDR.
  // The parameter is extracted from SPS.
  uint8_t transfer_characteristics_ = 0;
 
  uint8_t color_primaries_ = 0;
  uint8_t matrix_coefficients_ = 0;
 
  DISALLOW_COPY_AND_ASSIGN(DecoderConfigurationRecord);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_CODECS_DECODER_CONFIGURATION_RECORD_H_