dom/media/hls/HLSDemuxer.h

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #if !defined(HLSDemuxer_h_)
   8 #define HLSDemuxer_h_
   9
  10 #include "AutoTaskQueue.h"
  11 #include "GeneratedJNINatives.h"
  12 #include "GeneratedJNIWrappers.h"
  13 #include "MediaCodec.h"
  14 #include "MediaDataDemuxer.h"
  15 #include "MediaDecoder.h"
  16 #include "mozilla/Atomics.h"
  17 #include "mozilla/Maybe.h"
  18 #include "mozilla/Mutex.h"
  19
  20 #include "VideoUtils.h"
  21
  22 namespace mozilla {
  23
  24 class AbstractThread;
  25 class MediaResult;
  26 class HLSTrackDemuxer;
  27
  28 DDLoggedTypeDeclNameAndBase(HLSDemuxer, MediaDataDemuxer);
  29 DDLoggedTypeNameAndBase(HLSTrackDemuxer, MediaTrackDemuxer);
  30
  31 class HLSDemuxer final
  32   : public MediaDataDemuxer
  33   , public DecoderDoctorLifeLogger<HLSDemuxer>
  34 {
  35   class HLSDemuxerCallbacksSupport;
  36 public:
  37   explicit HLSDemuxer(int aPlayerId);
  38
  39   RefPtr<InitPromise> Init() override;
  40
  41   uint32_t GetNumberTracks(TrackInfo::TrackType aType) const override;
  42
  43   already_AddRefed<MediaTrackDemuxer>
  44   GetTrackDemuxer(TrackInfo::TrackType aType, uint32_t aTrackNumber) override;
  45
  46   bool IsSeekable() const override;
  47
  48   UniquePtr<EncryptionInfo> GetCrypto() override;
  49
  50   bool ShouldComputeStartTime() const override { return true; }
  51
  52   void NotifyDataArrived() override;
  53
  54   AutoTaskQueue* GetTaskQueue() const { return mTaskQueue; }
  55   void OnInitialized(bool aHasAudio, bool aHasVideo);
  56   void OnError(int aErrorCode);
  57
  58 private:
  59   media::TimeUnit GetNextKeyFrameTime();
  60
  61   bool OnTaskQueue() const;
  62   ~HLSDemuxer();
  63   friend class HLSTrackDemuxer;
  64
  65   const RefPtr<AutoTaskQueue> mTaskQueue;
  66   RefPtr<HLSTrackDemuxer> mAudioDemuxer;
  67   RefPtr<HLSTrackDemuxer> mVideoDemuxer;
  68
  69   MozPromiseHolder<InitPromise> mInitPromise;
  70   RefPtr<HLSDemuxerCallbacksSupport> mCallbackSupport;
  71
  72   java::GeckoHLSDemuxerWrapper::Callbacks::GlobalRef mJavaCallbacks;
  73   java::GeckoHLSDemuxerWrapper::GlobalRef mHLSDemuxerWrapper;
  74 };
  75
  76 class HLSTrackDemuxer
  77   : public MediaTrackDemuxer
  78   , public DecoderDoctorLifeLogger<HLSTrackDemuxer>
  79 {
  80 public:
  81   HLSTrackDemuxer(HLSDemuxer* aParent,
  82                   TrackInfo::TrackType aType,
  83                   UniquePtr<TrackInfo> aTrackInfo);
  84   ~HLSTrackDemuxer();
  85   UniquePtr<TrackInfo> GetInfo() const override;
  86
  87   RefPtr<SeekPromise> Seek(const media::TimeUnit& aTime) override;
  88
  89   RefPtr<SamplesPromise> GetSamples(int32_t aNumSamples = 1) override;
  90
  91   void UpdateMediaInfo(int index);
  92
  93   void Reset() override;
  94
  95   nsresult GetNextRandomAccessPoint(media::TimeUnit* aTime) override;
  96
  97   RefPtr<SkipAccessPointPromise> SkipToNextRandomAccessPoint(
  98     const media::TimeUnit& aTimeThreshold) override;
  99
 100   media::TimeIntervals GetBuffered() override;
 101
 102   void BreakCycles() override;
 103
 104   bool GetSamplesMayBlock() const override
 105   {
 106     return false;
 107   }
 108
 109   bool IsTrackValid() const
 110   {
 111     MutexAutoLock lock(mMutex);
 112     return mTrackInfo->IsValid();
 113   }
 114 private:
 115   // Update the timestamp of the next keyframe if there's one.
 116   void UpdateNextKeyFrameTime();
 117
 118   // Runs on HLSDemuxer's task queue.
 119   RefPtr<SeekPromise> DoSeek(const media::TimeUnit& aTime);
 120   RefPtr<SamplesPromise> DoGetSamples(int32_t aNumSamples);
 121   RefPtr<SkipAccessPointPromise> DoSkipToNextRandomAccessPoint(
 122     const media::TimeUnit& aTimeThreshold);
 123
 124   CryptoSample ExtractCryptoSample(size_t aSampleSize,
 125                                    java::sdk::CryptoInfo::LocalRef aCryptoInfo);
 126   RefPtr<MediaRawData> ConvertToMediaRawData(java::GeckoHLSSample::LocalRef aSample);
 127
 128   RefPtr<HLSDemuxer> mParent;
 129   TrackInfo::TrackType mType;
 130   Maybe<media::TimeUnit> mNextKeyframeTime;
 131   int32_t mLastFormatIndex = -1;
 132   // Queued samples extracted by the demuxer, but not yet returned.
 133   RefPtr<MediaRawData> mQueuedSample;
 134
 135   // Mutex to protect members below across multiple threads.
 136   mutable Mutex mMutex;
 137   UniquePtr<TrackInfo> mTrackInfo;
 138 };
 139
 140 } // namespace mozilla
 141
 142 #endif