// Copyright 2015 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 #include #include #include #include #include #include #include #include #include namespace shaka { SimpleMpdNotifier::SimpleMpdNotifier(const MpdOptions& mpd_options) : MpdNotifier(mpd_options), output_path_(mpd_options.mpd_params.mpd_output), mpd_builder_(new MpdBuilder(mpd_options)), content_protection_in_adaptation_set_( mpd_options.mpd_params.generate_dash_if_iop_compliant_mpd) { for (const std::string& base_url : mpd_options.mpd_params.base_urls) mpd_builder_->AddBaseUrl(base_url); } SimpleMpdNotifier::~SimpleMpdNotifier() {} bool SimpleMpdNotifier::Init() { return true; } bool SimpleMpdNotifier::NotifyNewContainer(const MediaInfo& media_info, uint32_t* container_id) { DCHECK(container_id); ContentType content_type = GetContentType(media_info); if (content_type == kContentTypeUnknown) return false; MediaInfo adjusted_media_info(media_info); MpdBuilder::MakePathsRelativeToMpd(output_path_, &adjusted_media_info); absl::MutexLock auto_lock(&lock_); const double kPeriodStartTimeSeconds = 0.0; Period* period = mpd_builder_->GetOrCreatePeriod(kPeriodStartTimeSeconds); DCHECK(period); AdaptationSet* adaptation_set = period->GetOrCreateAdaptationSet( media_info, content_protection_in_adaptation_set_); DCHECK(adaptation_set); if (!adaptation_set->has_id()) adaptation_set->set_id(next_adaptation_set_id_++); Representation* representation = adaptation_set->AddRepresentation(adjusted_media_info); if (!representation) return false; *container_id = representation->id(); if (content_protection_in_adaptation_set_) { // ContentProtection elements are already added to AdaptationSet above. // Use RepresentationId to AdaptationSet map to update ContentProtection // in AdaptationSet in NotifyEncryptionUpdate. representation_id_to_adaptation_set_[representation->id()] = adaptation_set; } else { AddContentProtectionElements(media_info, representation); } representation_map_[representation->id()] = representation; return true; } bool SimpleMpdNotifier::NotifyAvailabilityTimeOffset(uint32_t container_id) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } it->second->SetAvailabilityTimeOffset(); return true; } bool SimpleMpdNotifier::NotifySampleDuration(uint32_t container_id, int32_t sample_duration) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } it->second->SetSampleDuration(sample_duration); return true; } bool SimpleMpdNotifier::NotifySegmentDuration(uint32_t container_id) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } it->second->SetSegmentDuration(); return true; } bool SimpleMpdNotifier::NotifyNewSegment(uint32_t container_id, int64_t start_time, int64_t duration, uint64_t size, int64_t segment_number) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } it->second->AddNewSegment(start_time, duration, size, segment_number); return true; } bool SimpleMpdNotifier::NotifyCompletedSegment(uint32_t container_id, int64_t duration, uint64_t size) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } it->second->UpdateCompletedSegment(duration, size); return true; } bool SimpleMpdNotifier::NotifyCueEvent(uint32_t container_id, int64_t timestamp) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } Representation* original_representation = it->second; AdaptationSet* original_adaptation_set = representation_id_to_adaptation_set_[container_id]; const MediaInfo& media_info = original_representation->GetMediaInfo(); const double period_start_time_seconds = static_cast(timestamp) / media_info.reference_time_scale(); Period* period = mpd_builder_->GetOrCreatePeriod(period_start_time_seconds); DCHECK(period); AdaptationSet* adaptation_set = period->GetOrCreateAdaptationSet( media_info, content_protection_in_adaptation_set_); DCHECK(adaptation_set); if (!adaptation_set->has_id()) { adaptation_set->set_id(original_adaptation_set->id()); } else { DCHECK_EQ(adaptation_set->id(), original_adaptation_set->id()); } Representation* representation = adaptation_set->CopyRepresentation(*original_representation); if (!representation) return false; if (content_protection_in_adaptation_set_) { // ContentProtection elements are already added to AdaptationSet above. // Use RepresentationId to AdaptationSet map to update ContentProtection // in AdaptationSet in NotifyEncryptionUpdate. representation_id_to_adaptation_set_[representation->id()] = adaptation_set; } else { AddContentProtectionElements(media_info, representation); } representation_map_[representation->id()] = representation; return true; } bool SimpleMpdNotifier::NotifyEncryptionUpdate( uint32_t container_id, const std::string& drm_uuid, const std::vector& new_key_id, const std::vector& new_pssh) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } if (content_protection_in_adaptation_set_) { AdaptationSet* adaptation_set_for_representation = representation_id_to_adaptation_set_[it->second->id()]; adaptation_set_for_representation->UpdateContentProtectionPssh( drm_uuid, Uint8VectorToBase64(new_pssh)); } else { it->second->UpdateContentProtectionPssh(drm_uuid, Uint8VectorToBase64(new_pssh)); } return true; } bool SimpleMpdNotifier::NotifyMediaInfoUpdate(uint32_t container_id, const MediaInfo& media_info) { absl::MutexLock lock(&lock_); auto it = representation_map_.find(container_id); if (it == representation_map_.end()) { LOG(ERROR) << "Unexpected container_id: " << container_id; return false; } MediaInfo adjusted_media_info(media_info); MpdBuilder::MakePathsRelativeToMpd(output_path_, &adjusted_media_info); it->second->set_media_info(adjusted_media_info); return true; } bool SimpleMpdNotifier::Flush() { absl::MutexLock lock(&lock_); return WriteMpdToFile(output_path_, mpd_builder_.get()); } } // namespace shaka