simple_mpd_notifier.h

// Copyright 2014 Google LLC. 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 MPD_BASE_SIMPLE_MPD_NOTIFIER_H_
#define MPD_BASE_SIMPLE_MPD_NOTIFIER_H_
 
#include <cstdint>
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
#include <absl/synchronization/mutex.h>
 
#include <packager/mpd/base/mpd_builder.h>
#include <packager/mpd/base/mpd_notifier.h>
 
namespace shaka {
 
class AdaptationSet;
class MpdBuilder;
class Representation;
 
struct MpdOptions;
 
class SimpleMpdNotifier : public MpdNotifier {
 public:
  explicit SimpleMpdNotifier(const MpdOptions& mpd_options);
  ~SimpleMpdNotifier() override;
 
  bool Init() override;
  bool NotifyNewContainer(const MediaInfo& media_info, uint32_t* id) override;
  bool NotifyAvailabilityTimeOffset(uint32_t container_id) override;
  bool NotifySampleDuration(uint32_t container_id,
                            int32_t sample_duration) override;
  bool NotifySegmentDuration(uint32_t container_id) override;
  bool NotifyNewSegment(uint32_t container_id,
                        int64_t start_time,
                        int64_t duration,
                        uint64_t size,
                        int64_t segment_number) override;
  bool NotifyCompletedSegment(uint32_t container_id,
                              int64_t duration,
                              uint64_t size) override;
  bool NotifyCueEvent(uint32_t container_id, int64_t timestamp) override;
  bool NotifyEncryptionUpdate(uint32_t container_id,
                              const std::string& drm_uuid,
                              const std::vector<uint8_t>& new_key_id,
                              const std::vector<uint8_t>& new_pssh) override;
  bool NotifyMediaInfoUpdate(uint32_t container_id,
                             const MediaInfo& media_info) override;
 
  bool NotifyEndOfStream() override;
 
  bool Flush() override;
 
 private:
  SimpleMpdNotifier(const SimpleMpdNotifier&) = delete;
  SimpleMpdNotifier& operator=(const SimpleMpdNotifier&) = delete;
 
  friend class SimpleMpdNotifierTest;
 
  // Testing only method. Returns a pointer to MpdBuilder.
  MpdBuilder* MpdBuilderForTesting() const { return mpd_builder_.get(); }
 
  // Testing only method. Sets mpd_builder_.
  void SetMpdBuilderForTesting(std::unique_ptr<MpdBuilder> mpd_builder) {
    mpd_builder_ = std::move(mpd_builder);
  }
 
  // MPD output path.
  std::string output_path_;
  std::unique_ptr<MpdBuilder> mpd_builder_;
  bool content_protection_in_adaptation_set_ = true;
  absl::Mutex lock_;
 
  uint32_t next_adaptation_set_id_ = 0;
  // Maps Representation ID to Representation.
  std::map<uint32_t, Representation*> representation_map_;
  // Maps Representation ID to AdaptationSet. This is for updating the PSSH.
  std::map<uint32_t, AdaptationSet*> representation_id_to_adaptation_set_;
};
 
}  // namespace shaka
 
#endif  // MPD_BASE_SIMPLE_MPD_NOTIFIER_H_