webmenc.cc

   1 /*
   2  *  Copyright (c) 2014 The WebM 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 #include "./webmenc.h"
  11
  12 #include <string>
  13
  14 #include "third_party/libwebm/mkvmuxer/mkvmuxer.h"
  15 #include "third_party/libwebm/mkvmuxer/mkvmuxerutil.h"
  16 #include "third_party/libwebm/mkvmuxer/mkvwriter.h"
  17
  18 namespace {
  19 const uint64_t kDebugTrackUid = 0xDEADBEEF;
  20 const int kVideoTrackNumber = 1;
  21 }  // namespace
  22
  23 void write_webm_file_header(struct WebmOutputContext *webm_ctx,
  24                             const vpx_codec_enc_cfg_t *cfg,
  25                             const struct vpx_rational *fps,
  26                             stereo_format_t stereo_fmt,
  27                             unsigned int fourcc,
  28                             const struct VpxRational *par) {
  29   mkvmuxer::MkvWriter *const writer = new mkvmuxer::MkvWriter(webm_ctx->stream);
  30   mkvmuxer::Segment *const segment = new mkvmuxer::Segment();
  31   segment->Init(writer);
  32   segment->set_mode(mkvmuxer::Segment::kFile);
  33   segment->OutputCues(true);
  34
  35   mkvmuxer::SegmentInfo *const info = segment->GetSegmentInfo();
  36   const uint64_t kTimecodeScale = 1000000;
  37   info->set_timecode_scale(kTimecodeScale);
  38   std::string version = "vpxenc";
  39   if (!webm_ctx->debug) {
  40     version.append(std::string(" ") + vpx_codec_version_str());
  41   }
  42   info->set_writing_app(version.c_str());
  43
  44   const uint64_t video_track_id =
  45       segment->AddVideoTrack(static_cast<int>(cfg->g_w),
  46                              static_cast<int>(cfg->g_h),
  47                              kVideoTrackNumber);
  48   mkvmuxer::VideoTrack* const video_track =
  49       static_cast<mkvmuxer::VideoTrack*>(
  50           segment->GetTrackByNumber(video_track_id));
  51   video_track->SetStereoMode(stereo_fmt);
  52   const char *codec_id;
  53   switch (fourcc) {
  54   case VP8_FOURCC:
  55     codec_id = "V_VP8";
  56     break;
  57   case VP9_FOURCC:
  58     codec_id = "V_VP9";
  59     break;
  60   case VP10_FOURCC:
  61     codec_id = "V_VP10";
  62     break;
  63   default:
  64     codec_id = "V_VP10";
  65     break;
  66   }
  67   video_track->set_codec_id(codec_id);
  68   if (par->numerator > 1 || par->denominator > 1) {
  69     // TODO(fgalligan): Add support of DisplayUnit, Display Aspect Ratio type
  70     // to WebM format.
  71     const uint64_t display_width =
  72         static_cast<uint64_t>(((cfg->g_w * par->numerator * 1.0) /
  73                                par->denominator) + .5);
  74     video_track->set_display_width(display_width);
  75     video_track->set_display_height(cfg->g_h);
  76   }
  77   if (webm_ctx->debug) {
  78     video_track->set_uid(kDebugTrackUid);
  79   }
  80   webm_ctx->writer = writer;
  81   webm_ctx->segment = segment;
  82 }
  83
  84 void write_webm_block(struct WebmOutputContext *webm_ctx,
  85                       const vpx_codec_enc_cfg_t *cfg,
  86                       const vpx_codec_cx_pkt_t *pkt) {
  87   mkvmuxer::Segment *const segment =
  88       reinterpret_cast<mkvmuxer::Segment*>(webm_ctx->segment);
  89   int64_t pts_ns = pkt->data.frame.pts * 1000000000ll *
  90                    cfg->g_timebase.num / cfg->g_timebase.den;
  91   if (pts_ns <= webm_ctx->last_pts_ns)
  92     pts_ns = webm_ctx->last_pts_ns + 1000000;
  93   webm_ctx->last_pts_ns = pts_ns;
  94
  95   segment->AddFrame(static_cast<uint8_t*>(pkt->data.frame.buf),
  96                     pkt->data.frame.sz,
  97                     kVideoTrackNumber,
  98                     pts_ns,
  99                     pkt->data.frame.flags & VPX_FRAME_IS_KEY);
 100 }
 101
 102 void write_webm_file_footer(struct WebmOutputContext *webm_ctx) {
 103   mkvmuxer::MkvWriter *const writer =
 104       reinterpret_cast<mkvmuxer::MkvWriter*>(webm_ctx->writer);
 105   mkvmuxer::Segment *const segment =
 106       reinterpret_cast<mkvmuxer::Segment*>(webm_ctx->segment);
 107   segment->Finalize();
 108   delete segment;
 109   delete writer;
 110   webm_ctx->writer = NULL;
 111   webm_ctx->segment = NULL;
 112 }