packager.h

// Copyright 2017 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

#ifndef PACKAGER_PACKAGER_H_
#define PACKAGER_PACKAGER_H_

#include <memory>
#include <string>
#include <vector>

#include "packager/file/public/buffer_callback_params.h"
#include "packager/hls/public/hls_params.h"
#include "packager/media/public/ad_cue_generator_params.h"
#include "packager/media/public/chunking_params.h"
#include "packager/media/public/crypto_params.h"
#include "packager/media/public/mp4_output_params.h"
#include "packager/mpd/public/mpd_params.h"
#include "packager/status.h"

namespace shaka {

struct TestParams {
  bool dump_stream_info = false;
  bool inject_fake_clock = false;
  std::string injected_library_version;
};

struct PackagingParams {
  std::string temp_dir;
  Mp4OutputParams mp4_output_params;
  uint32_t transport_stream_timestamp_offset_ms = 0;
  ChunkingParams chunking_params;

  AdCueGeneratorParams ad_cue_generator_params;

  bool output_media_info = false;
  MpdParams mpd_params;
  HlsParams hls_params;

  EncryptionParams encryption_params;
  DecryptionParams decryption_params;

  BufferCallbackParams buffer_callback_params;

  // Parameters for testing. Do not use in production.
  TestParams test_params;
};

struct StreamDescriptor {
  std::string input;

  std::string stream_selector;

  std::string output;
  std::string segment_template;

  std::string output_format;
  bool skip_encryption = false;
  std::string drm_label;
  uint32_t trick_play_factor = 0;
  uint32_t bandwidth = 0;
  std::string language;
  std::string hls_name;
  std::string hls_group_id;
  std::string hls_playlist_name;
  std::string hls_iframe_playlist_name;
};

class SHAKA_EXPORT Packager {
 public:
  Packager();
  ~Packager();

  Status Initialize(
      const PackagingParams& packaging_params,
      const std::vector<StreamDescriptor>& stream_descriptors);

  Status Run();

  void Cancel();

  static std::string GetLibraryVersion();

  static std::string DefaultStreamLabelFunction(
      int max_sd_pixels,
      int max_hd_pixels,
      int max_uhd1_pixels,
      const EncryptionParams::EncryptedStreamAttributes& stream_attributes);

 private:
  Packager(const Packager&) = delete;
  Packager& operator=(const Packager&) = delete;

  struct PackagerInternal;
  std::unique_ptr<PackagerInternal> internal_;
};

}  // namespace shaka

#endif  // PACKAGER_PACKAGER_H_