webmenc.cc 3.63 KB
Newer Older
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3
 *
4 5 6 7 8 9 10 11
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
*/

12 13 14 15 16 17 18 19 20 21 22 23 24 25
#include "./webmenc.h"

#include <string>

#include "third_party/libwebm/mkvmuxer.hpp"
#include "third_party/libwebm/mkvmuxerutil.hpp"
#include "third_party/libwebm/mkvwriter.hpp"

namespace {
const uint64_t kDebugTrackUid = 0xDEADBEEF;
const int kVideoTrackNumber = 1;
}  // namespace

void write_webm_file_header(struct EbmlGlobal *glob,
Adrian Grange's avatar
Adrian Grange committed
26 27
                            const aom_codec_enc_cfg_t *cfg,
                            const struct aom_rational *fps,
clang-format's avatar
clang-format committed
28
                            stereo_format_t stereo_fmt, unsigned int fourcc,
Adrian Grange's avatar
Adrian Grange committed
29
                            const struct AvxRational *par) {
30 31 32 33 34 35 36 37 38
  mkvmuxer::MkvWriter *const writer = new mkvmuxer::MkvWriter(glob->stream);
  mkvmuxer::Segment *const segment = new mkvmuxer::Segment();
  segment->Init(writer);
  segment->set_mode(mkvmuxer::Segment::kFile);
  segment->OutputCues(true);

  mkvmuxer::SegmentInfo *const info = segment->GetSegmentInfo();
  const uint64_t kTimecodeScale = 1000000;
  info->set_timecode_scale(kTimecodeScale);
Adrian Grange's avatar
Adrian Grange committed
39
  std::string version = "aomenc";
40
  if (!glob->debug) {
Adrian Grange's avatar
Adrian Grange committed
41
    version.append(std::string(" ") + aom_codec_version_str());
42 43 44 45 46
  }
  info->set_writing_app(version.c_str());

  const uint64_t video_track_id =
      segment->AddVideoTrack(static_cast<int>(cfg->g_w),
clang-format's avatar
clang-format committed
47 48 49
                             static_cast<int>(cfg->g_h), kVideoTrackNumber);
  mkvmuxer::VideoTrack *const video_track = static_cast<mkvmuxer::VideoTrack *>(
      segment->GetTrackByNumber(video_track_id));
50
  video_track->SetStereoMode(stereo_fmt);
51 52
  const char *codec_id;
  switch (fourcc) {
53 54
    case AV1_FOURCC: codec_id = "V_AV1"; break;
    default: codec_id = "V_AV1"; break;
55 56
  }
  video_track->set_codec_id(codec_id);
57 58 59
  if (par->numerator > 1 || par->denominator > 1) {
    // TODO(fgalligan): Add support of DisplayUnit, Display Aspect Ratio type
    // to WebM format.
clang-format's avatar
clang-format committed
60 61
    const uint64_t display_width = static_cast<uint64_t>(
        ((cfg->g_w * par->numerator * 1.0) / par->denominator) + .5);
62 63 64
    video_track->set_display_width(display_width);
    video_track->set_display_height(cfg->g_h);
  }
65 66 67 68 69 70 71
  if (glob->debug) {
    video_track->set_uid(kDebugTrackUid);
  }
  glob->writer = writer;
  glob->segment = segment;
}

Adrian Grange's avatar
Adrian Grange committed
72 73
void write_webm_block(struct EbmlGlobal *glob, const aom_codec_enc_cfg_t *cfg,
                      const aom_codec_cx_pkt_t *pkt) {
74
  mkvmuxer::Segment *const segment =
clang-format's avatar
clang-format committed
75 76 77 78
      reinterpret_cast<mkvmuxer::Segment *>(glob->segment);
  int64_t pts_ns = pkt->data.frame.pts * 1000000000ll * cfg->g_timebase.num /
                   cfg->g_timebase.den;
  if (pts_ns <= glob->last_pts_ns) pts_ns = glob->last_pts_ns + 1000000;
79 80
  glob->last_pts_ns = pts_ns;

clang-format's avatar
clang-format committed
81 82
  segment->AddFrame(static_cast<uint8_t *>(pkt->data.frame.buf),
                    pkt->data.frame.sz, kVideoTrackNumber, pts_ns,
Adrian Grange's avatar
Adrian Grange committed
83
                    pkt->data.frame.flags & AOM_FRAME_IS_KEY);
84 85 86 87
}

void write_webm_file_footer(struct EbmlGlobal *glob) {
  mkvmuxer::MkvWriter *const writer =
clang-format's avatar
clang-format committed
88
      reinterpret_cast<mkvmuxer::MkvWriter *>(glob->writer);
89
  mkvmuxer::Segment *const segment =
clang-format's avatar
clang-format committed
90
      reinterpret_cast<mkvmuxer::Segment *>(glob->segment);
91 92 93 94 95 96
  segment->Finalize();
  delete segment;
  delete writer;
  glob->writer = NULL;
  glob->segment = NULL;
}