#pragma once
#if ENABLE(MEDIA_SOURCE)
#include <map>
#include <wtf/MediaTime.h>
#include <wtf/RefPtr.h>
namespace WebCore {
class MediaSample;
class SampleMap;
class PresentationOrderSampleMap {
friend class SampleMap;
public:
typedef std::map<MediaTime, RefPtr<MediaSample>> MapType;
typedef MapType::iterator iterator;
typedef MapType::const_iterator const_iterator;
typedef MapType::reverse_iterator reverse_iterator;
typedef MapType::const_reverse_iterator const_reverse_iterator;
typedef std::pair<iterator, iterator> iterator_range;
WEBCORE_EXPORT iterator begin() { return m_samples.begin(); }
WEBCORE_EXPORT const_iterator begin() const { return m_samples.begin(); }
WEBCORE_EXPORT iterator end() { return m_samples.end(); }
WEBCORE_EXPORT const_iterator end() const { return m_samples.end(); }
WEBCORE_EXPORT reverse_iterator rbegin() { return m_samples.rbegin(); }
WEBCORE_EXPORT const_reverse_iterator rbegin() const { return m_samples.rbegin(); }
WEBCORE_EXPORT reverse_iterator rend() { return m_samples.rend(); }
WEBCORE_EXPORT const_reverse_iterator rend() const { return m_samples.rend(); }
WEBCORE_EXPORT iterator findSampleWithPresentationTime(const MediaTime&);
WEBCORE_EXPORT iterator findSampleContainingPresentationTime(const MediaTime&);
WEBCORE_EXPORT iterator findSampleStartingOnOrAfterPresentationTime(const MediaTime&);
WEBCORE_EXPORT reverse_iterator reverseFindSampleContainingPresentationTime(const MediaTime&);
WEBCORE_EXPORT reverse_iterator reverseFindSampleBeforePresentationTime(const MediaTime&);
WEBCORE_EXPORT iterator_range findSamplesBetweenPresentationTimes(const MediaTime&, const MediaTime&);
WEBCORE_EXPORT iterator_range findSamplesWithinPresentationRange(const MediaTime&, const MediaTime&);
WEBCORE_EXPORT iterator_range findSamplesWithinPresentationRangeFromEnd(const MediaTime&, const MediaTime&);
private:
MapType m_samples;
};
class DecodeOrderSampleMap {
friend class SampleMap;
public:
typedef std::pair<MediaTime, MediaTime> KeyType;
typedef std::map<KeyType, RefPtr<MediaSample>> MapType;
typedef MapType::iterator iterator;
typedef MapType::const_iterator const_iterator;
typedef MapType::reverse_iterator reverse_iterator;
typedef MapType::const_reverse_iterator const_reverse_iterator;
typedef std::pair<reverse_iterator, reverse_iterator> reverse_iterator_range;
WEBCORE_EXPORT iterator begin() { return m_samples.begin(); }
WEBCORE_EXPORT const_iterator begin() const { return m_samples.begin(); }
WEBCORE_EXPORT iterator end() { return m_samples.end(); }
WEBCORE_EXPORT const_iterator end() const { return m_samples.end(); }
WEBCORE_EXPORT reverse_iterator rbegin() { return m_samples.rbegin(); }
WEBCORE_EXPORT const_reverse_iterator rbegin() const { return m_samples.rbegin(); }
WEBCORE_EXPORT reverse_iterator rend() { return m_samples.rend(); }
WEBCORE_EXPORT const_reverse_iterator rend() const { return m_samples.rend(); }
WEBCORE_EXPORT iterator findSampleWithDecodeKey(const KeyType&);
WEBCORE_EXPORT reverse_iterator reverseFindSampleWithDecodeKey(const KeyType&);
WEBCORE_EXPORT reverse_iterator findSyncSamplePriorToPresentationTime(const MediaTime&, const MediaTime& threshold = MediaTime::positiveInfiniteTime());
WEBCORE_EXPORT reverse_iterator findSyncSamplePriorToDecodeIterator(reverse_iterator);
WEBCORE_EXPORT iterator findSyncSampleAfterPresentationTime(const MediaTime&, const MediaTime& threshold = MediaTime::positiveInfiniteTime());
WEBCORE_EXPORT iterator findSyncSampleAfterDecodeIterator(iterator);
WEBCORE_EXPORT reverse_iterator_range findDependentSamples(MediaSample*);
private:
MapType m_samples;
PresentationOrderSampleMap m_presentationOrder;
};
class SampleMap {
public:
WEBCORE_EXPORT SampleMap()
: m_totalSize(0)
{
}
WEBCORE_EXPORT bool empty() const;
WEBCORE_EXPORT void clear();
WEBCORE_EXPORT void addSample(MediaSample&);
WEBCORE_EXPORT void removeSample(MediaSample*);
WEBCORE_EXPORT size_t sizeInBytes() const { return m_totalSize; }
template<typename I>
void addRange(I begin, I end);
WEBCORE_EXPORT DecodeOrderSampleMap& decodeOrder() { return m_decodeOrder; }
WEBCORE_EXPORT const DecodeOrderSampleMap& decodeOrder() const { return m_decodeOrder; }
WEBCORE_EXPORT PresentationOrderSampleMap& presentationOrder() { return m_decodeOrder.m_presentationOrder; }
WEBCORE_EXPORT const PresentationOrderSampleMap& presentationOrder() const { return m_decodeOrder.m_presentationOrder; }
private:
DecodeOrderSampleMap m_decodeOrder;
size_t m_totalSize;
};
template<typename I>
void SampleMap::addRange(I begin, I end)
{
for (I iter = begin; iter != end; ++iter)
addSample(*iter->second);
}
}
#endif // ENABLE(MEDIA_SOURCE)