// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "media/base/android/extract_sps_and_pps.h"

#include <array>

#include "media/formats/mp4/box_definitions.h"

namespace media {

void ExtractSpsAndPps(const std::vector<uint8_t>& avcc,
                      std::vector<uint8_t>* sps_out,
                      std::vector<uint8_t>* pps_out) {
  if (avcc.empty())
    return;

  mp4::AVCDecoderConfigurationRecord record;
  if (!record.Parse(avcc.data(), avcc.size())) {
    DVLOG(1) << "Failed to extract SPS and PPS";
    return;
  }

  constexpr std::array<uint8_t, 4> prefix = {{0, 0, 0, 1}};
  for (const std::vector<uint8_t>& sps : record.sps_list) {
    sps_out->insert(sps_out->end(), prefix.begin(), prefix.end());
    sps_out->insert(sps_out->end(), sps.begin(), sps.end());
  }

  for (const std::vector<uint8_t>& pps : record.pps_list) {
    pps_out->insert(pps_out->end(), prefix.begin(), prefix.end());
    pps_out->insert(pps_out->end(), pps.begin(), pps.end());
  }
}

}  // namespace media