Source/WebCore/Modules/mediasource/SampleMap.h

   1 /*
   2  * Copyright (C) 2013 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef SampleMap_h
  27 #define SampleMap_h
  28
  29 #if ENABLE(MEDIA_SOURCE)
  30
  31 #include <map>
  32 #include <wtf/MediaTime.h>
  33 #include <wtf/RefPtr.h>
  34
  35 namespace WebCore {
  36
  37 class MediaSample;
  38 class SampleMap;
  39
  40 class PresentationOrderSampleMap {
  41     friend class SampleMap;
  42 public:
  43     typedef std::map<MediaTime, RefPtr<MediaSample>> MapType;
  44     typedef MapType::iterator iterator;
  45     typedef MapType::reverse_iterator reverse_iterator;
  46     typedef std::pair<iterator, iterator> iterator_range;
  47
  48     iterator begin() { return m_samples.begin(); }
  49     iterator end() { return m_samples.end(); }
  50     reverse_iterator rbegin() { return m_samples.rbegin(); }
  51     reverse_iterator rend() { return m_samples.rend(); }
  52
  53     iterator findSampleWithPresentationTime(const MediaTime&);
  54     iterator findSampleContainingPresentationTime(const MediaTime&);
  55     iterator findSampleOnOrAfterPresentationTime(const MediaTime&);
  56     reverse_iterator reverseFindSampleContainingPresentationTime(const MediaTime&);
  57     reverse_iterator reverseFindSampleBeforePresentationTime(const MediaTime&);
  58     iterator_range findSamplesBetweenPresentationTimes(const MediaTime&, const MediaTime&);
  59     iterator_range findSamplesWithinPresentationRange(const MediaTime&, const MediaTime&);
  60     iterator_range findSamplesWithinPresentationRangeFromEnd(const MediaTime&, const MediaTime&);
  61
  62 private:
  63     MapType m_samples;
  64 };
  65
  66 class DecodeOrderSampleMap {
  67     friend class SampleMap;
  68 public:
  69     typedef std::pair<MediaTime, MediaTime> KeyType;
  70     typedef std::map<KeyType, RefPtr<MediaSample>> MapType;
  71     typedef MapType::iterator iterator;
  72     typedef MapType::reverse_iterator reverse_iterator;
  73     typedef std::pair<reverse_iterator, reverse_iterator> reverse_iterator_range;
  74
  75     iterator begin() { return m_samples.begin(); }
  76     iterator end() { return m_samples.end(); }
  77     reverse_iterator rbegin() { return m_samples.rbegin(); }
  78     reverse_iterator rend() { return m_samples.rend(); }
  79
  80     iterator findSampleWithDecodeKey(const KeyType&);
  81     reverse_iterator reverseFindSampleWithDecodeKey(const KeyType&);
  82     reverse_iterator findSyncSamplePriorToPresentationTime(const MediaTime&, const MediaTime& threshold = MediaTime::positiveInfiniteTime());
  83     reverse_iterator findSyncSamplePriorToDecodeIterator(reverse_iterator);
  84     iterator findSyncSampleAfterPresentationTime(const MediaTime&, const MediaTime& threshold = MediaTime::positiveInfiniteTime());
  85     iterator findSyncSampleAfterDecodeIterator(iterator);
  86     reverse_iterator_range findDependentSamples(MediaSample*);
  87
  88 private:
  89     MapType m_samples;
  90     PresentationOrderSampleMap m_presentationOrder;
  91 };
  92
  93 class SampleMap {
  94 public:
  95     SampleMap()
  96         : m_totalSize(0)
  97     {
  98     }
  99
 100     bool empty() const;
 101     void clear();
 102     void addSample(MediaSample&);
 103     void removeSample(MediaSample*);
 104     size_t sizeInBytes() const { return m_totalSize; }
 105
 106     template<typename I>
 107     void addRange(I begin, I end);
 108
 109     DecodeOrderSampleMap& decodeOrder() { return m_decodeOrder; }
 110     const DecodeOrderSampleMap& decodeOrder() const { return m_decodeOrder; }
 111     PresentationOrderSampleMap& presentationOrder() { return m_decodeOrder.m_presentationOrder; }
 112     const PresentationOrderSampleMap& presentationOrder() const { return m_decodeOrder.m_presentationOrder; }
 113
 114 private:
 115     DecodeOrderSampleMap m_decodeOrder;
 116     size_t m_totalSize;
 117 };
 118
 119 template<typename I>
 120 void SampleMap::addRange(I begin, I end)
 121 {
 122     for (I iter = begin; iter != end; ++iter)
 123         addSample(*iter->second);
 124 }
 125
 126 }
 127
 128 #endif
 129
 130 #endif // SampleMap_h