shaka-packager/mpd/base/simple_mpd_notifier.h

78 lines
2.1 KiB
C++

// Copyright 2014 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 MPD_BASE_SIMPLE_MPD_NOTIFIER_H_
#define MPD_BASE_SIMPLE_MPD_NOTIFIER_H_
#include <map>
#include <string>
#include <vector>
#include "base/memory/scoped_ptr.h"
#include "base/synchronization/lock.h"
#include "mpd/base/mpd_notifier.h"
namespace dash_packager {
class AdaptationSet;
class MpdBuilder;
class Representation;
struct MpdOptions;
/// A simple MpdNotifier implementation which receives muxer listener event and
/// generates an Mpd file.
class SimpleMpdNotifier : public MpdNotifier {
public:
SimpleMpdNotifier(DashProfile dash_profile,
const MpdOptions& mpd_options,
const std::vector<std::string>& base_urls,
const std::string& output_path);
virtual ~SimpleMpdNotifier();
/// @name MpdNotifier implemetation overrides.
/// @{
virtual bool Init() OVERRIDE;
virtual bool NotifyNewContainer(const MediaInfo& media_info,
uint32* id) OVERRIDE;
virtual bool NotifyNewSegment(uint32 id,
uint64 start_time,
uint64 duration,
uint64 size) OVERRIDE;
virtual bool AddContentProtectionElement(
uint32 id,
const ContentProtectionElement& content_protection_element) OVERRIDE;
/// @}
private:
enum ContentType {
kUnknown,
kVideo,
kAudio,
kText
};
ContentType GetContentType(const MediaInfo& media_info);
bool WriteMpdToFile();
std::string output_path_;
scoped_ptr<MpdBuilder> mpd_builder_;
base::Lock lock_;
typedef std::map<ContentType, AdaptationSet*> AdaptationSetMap;
AdaptationSetMap adaptation_set_map_;
typedef std::map<uint32, Representation*> RepresentationMap;
RepresentationMap representation_map_;
DISALLOW_COPY_AND_ASSIGN(SimpleMpdNotifier);
};
} // namespace dash_packager
#endif // MPD_BASE_SIMPLE_MPD_NOTIFIER_H_