// Copyright 2015 Google Inc. 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

#include "packager/media/formats/webm/webm_muxer.h"

#include "packager/media/base/fourccs.h"
#include "packager/media/base/media_sample.h"
#include "packager/media/base/media_stream.h"
#include "packager/media/base/stream_info.h"
#include "packager/media/formats/webm/mkv_writer.h"
#include "packager/media/formats/webm/multi_segment_segmenter.h"
#include "packager/media/formats/webm/single_segment_segmenter.h"
#include "packager/media/formats/webm/two_pass_single_segment_segmenter.h"

namespace shaka {
namespace media {
namespace webm {

WebMMuxer::WebMMuxer(const MuxerOptions& options) : Muxer(options) {}
WebMMuxer::~WebMMuxer() {}

Status WebMMuxer::Initialize() {
  CHECK_EQ(streams().size(), 1U);

  if (crypto_period_duration_in_seconds() > 0) {
    NOTIMPLEMENTED() << "Key rotation is not implemented for WebM";
    return Status(error::UNIMPLEMENTED,
                  "Key rotation is not implemented for WebM");
  }

  if (encryption_key_source() && (protection_scheme() != FOURCC_cenc)) {
    NOTIMPLEMENTED()
        << "WebM does not support protection scheme other than 'cenc'.";
    return Status(error::UNIMPLEMENTED,
                  "WebM does not support protection scheme other than 'cenc'.");
  }

  std::unique_ptr<MkvWriter> writer(new MkvWriter);
  Status status = writer->Open(options().output_file_name);
  if (!status.ok())
    return status;

  if (!options().single_segment) {
    segmenter_.reset(new MultiSegmentSegmenter(options()));
  } else {
    segmenter_.reset(new TwoPassSingleSegmentSegmenter(options()));
  }

  Status initialized = segmenter_->Initialize(
      std::move(writer), streams()[0]->info().get(), progress_listener(),
      muxer_listener(), encryption_key_source(), max_sd_pixels(),
      clear_lead_in_seconds());

  if (!initialized.ok())
    return initialized;

  FireOnMediaStartEvent();
  return Status::OK;
}

Status WebMMuxer::Finalize() {
  DCHECK(segmenter_);
  Status segmenter_finalized = segmenter_->Finalize();

  if (!segmenter_finalized.ok())
    return segmenter_finalized;

  FireOnMediaEndEvent();
  LOG(INFO) << "WEBM file '" << options().output_file_name << "' finalized.";
  return Status::OK;
}

Status WebMMuxer::DoAddSample(const MediaStream* stream,
                              scoped_refptr<MediaSample> sample) {
  DCHECK(segmenter_);
  DCHECK(stream == streams()[0]);
  return segmenter_->AddSample(sample);
}

void WebMMuxer::FireOnMediaStartEvent() {
  if (!muxer_listener())
    return;

  DCHECK(!streams().empty()) << "Media started without a stream.";

  const uint32_t timescale = streams().front()->info()->time_scale();
  muxer_listener()->OnMediaStart(options(), *streams().front()->info(),
                                 timescale, MuxerListener::kContainerWebM);
}

void WebMMuxer::FireOnMediaEndEvent() {
  if (!muxer_listener())
    return;

  uint64_t init_range_start = 0;
  uint64_t init_range_end = 0;
  const bool has_init_range =
      segmenter_->GetInitRangeStartAndEnd(&init_range_start, &init_range_end);

  uint64_t index_range_start = 0;
  uint64_t index_range_end = 0;
  const bool has_index_range = segmenter_->GetIndexRangeStartAndEnd(
      &index_range_start, &index_range_end);

  const float duration_seconds = segmenter_->GetDuration();

  const int64_t file_size =
      File::GetFileSize(options().output_file_name.c_str());
  if (file_size <= 0) {
    LOG(ERROR) << "Invalid file size: " << file_size;
    return;
  }

  muxer_listener()->OnMediaEnd(has_init_range, init_range_start, init_range_end,
                               has_index_range, index_range_start,
                               index_range_end, duration_seconds, file_size);
}

}  // namespace webm
}  // namespace media
}  // namespace shaka