PlaybackManager.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717
  1. /*
  2. * Copyright (c) 2022, Gregory Bertilson <zaggy1024@gmail.com>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/Format.h>
  7. #include <LibCore/Timer.h>
  8. #include <LibVideo/Containers/Matroska/MatroskaDemuxer.h>
  9. #include <LibVideo/VP9/Decoder.h>
  10. #include "PlaybackManager.h"
  11. namespace Video {
  12. #define TRY_OR_FATAL_ERROR(expression) \
  13. ({ \
  14. auto&& _fatal_expression = (expression); \
  15. if (_fatal_expression.is_error()) { \
  16. dispatch_fatal_error(_fatal_expression.release_error()); \
  17. return; \
  18. } \
  19. static_assert(!::AK::Detail::IsLvalueReference<decltype(_fatal_expression.release_value())>, \
  20. "Do not return a reference from a fallible expression"); \
  21. _fatal_expression.release_value(); \
  22. })
  23. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::from_file(StringView filename)
  24. {
  25. auto demuxer = TRY(Matroska::MatroskaDemuxer::from_file(filename));
  26. return create(move(demuxer));
  27. }
  28. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::from_mapped_file(NonnullOwnPtr<Core::MappedFile> mapped_file)
  29. {
  30. auto demuxer = TRY(Matroska::MatroskaDemuxer::from_mapped_file(move(mapped_file)));
  31. return create(move(demuxer));
  32. }
  33. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::from_data(ReadonlyBytes data)
  34. {
  35. auto demuxer = TRY(Matroska::MatroskaDemuxer::from_data(data));
  36. return create(move(demuxer));
  37. }
  38. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::create(NonnullOwnPtr<Demuxer> demuxer)
  39. {
  40. auto video_tracks = TRY(demuxer->get_tracks_for_type(TrackType::Video));
  41. if (video_tracks.is_empty())
  42. return DecoderError::with_description(DecoderErrorCategory::Invalid, "No video track is present"sv);
  43. auto track = video_tracks[0];
  44. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Selecting video track number {}", track.identifier());
  45. auto decoder = DECODER_TRY_ALLOC(try_make<VP9::Decoder>());
  46. auto frame_queue = DECODER_TRY_ALLOC(VideoFrameQueue::create());
  47. auto playback_manager = DECODER_TRY_ALLOC(try_make<PlaybackManager>(demuxer, track, move(decoder), move(frame_queue)));
  48. playback_manager->m_state_update_timer = DECODER_TRY_ALLOC(Core::Timer::create_single_shot(0, [&self = *playback_manager] { self.timer_callback(); }));
  49. playback_manager->m_decode_thread = DECODER_TRY_ALLOC(Threading::Thread::try_create([&self = *playback_manager] {
  50. while (!self.m_stop_decoding.load())
  51. self.decode_and_queue_one_sample();
  52. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Media Decoder thread ended.");
  53. return 0;
  54. },
  55. "Media Decoder"sv));
  56. playback_manager->m_playback_handler = make<SeekingStateHandler>(*playback_manager, false, Duration::zero(), SeekMode::Fast);
  57. DECODER_TRY_ALLOC(playback_manager->m_playback_handler->on_enter());
  58. playback_manager->m_decode_thread->start();
  59. return playback_manager;
  60. }
  61. PlaybackManager::PlaybackManager(NonnullOwnPtr<Demuxer>& demuxer, Track video_track, NonnullOwnPtr<VideoDecoder>&& decoder, VideoFrameQueue&& frame_queue)
  62. : m_demuxer(move(demuxer))
  63. , m_selected_video_track(video_track)
  64. , m_frame_queue(move(frame_queue))
  65. , m_decoder(move(decoder))
  66. , m_decode_wait_condition(m_decode_wait_mutex)
  67. {
  68. }
  69. PlaybackManager::~PlaybackManager()
  70. {
  71. m_stop_decoding.exchange(true);
  72. m_decode_wait_condition.broadcast();
  73. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Waiting for decode thread to end...");
  74. (void)m_decode_thread->join();
  75. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Successfully destroyed PlaybackManager.");
  76. }
  77. void PlaybackManager::resume_playback()
  78. {
  79. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Resuming playback.");
  80. TRY_OR_FATAL_ERROR(m_playback_handler->play());
  81. }
  82. void PlaybackManager::pause_playback()
  83. {
  84. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Pausing playback.");
  85. if (!m_playback_handler->is_playing())
  86. warnln("Cannot pause.");
  87. TRY_OR_FATAL_ERROR(m_playback_handler->pause());
  88. }
  89. Duration PlaybackManager::current_playback_time()
  90. {
  91. return m_playback_handler->current_time();
  92. }
  93. Duration PlaybackManager::duration()
  94. {
  95. auto duration_result = ({
  96. auto demuxer_locker = Threading::MutexLocker(m_demuxer_mutex);
  97. m_demuxer->duration();
  98. });
  99. if (duration_result.is_error()) {
  100. dispatch_decoder_error(duration_result.release_error());
  101. // FIXME: We should determine the last sample that the demuxer knows is available and
  102. // use that as the current duration. The duration may change if the demuxer doesn't
  103. // know there is a fixed duration.
  104. return Duration::zero();
  105. }
  106. return duration_result.release_value();
  107. }
  108. void PlaybackManager::dispatch_fatal_error(Error error)
  109. {
  110. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Encountered fatal error: {}", error.string_literal());
  111. // FIXME: For threading, this will have to use a pre-allocated event to send to the main loop
  112. // to be able to gracefully handle OOM.
  113. if (on_fatal_playback_error)
  114. on_fatal_playback_error(move(error));
  115. }
  116. void PlaybackManager::dispatch_decoder_error(DecoderError error)
  117. {
  118. switch (error.category()) {
  119. case DecoderErrorCategory::EndOfStream:
  120. dbgln_if(PLAYBACK_MANAGER_DEBUG, "{}", error.string_literal());
  121. TRY_OR_FATAL_ERROR(m_playback_handler->stop());
  122. break;
  123. default:
  124. dbgln("Playback error encountered: {}", error.string_literal());
  125. TRY_OR_FATAL_ERROR(m_playback_handler->stop());
  126. if (on_decoder_error)
  127. on_decoder_error(move(error));
  128. break;
  129. }
  130. }
  131. void PlaybackManager::dispatch_new_frame(RefPtr<Gfx::Bitmap> frame)
  132. {
  133. if (on_video_frame)
  134. on_video_frame(move(frame));
  135. }
  136. bool PlaybackManager::dispatch_frame_queue_item(FrameQueueItem&& item)
  137. {
  138. if (item.is_error()) {
  139. dispatch_decoder_error(item.release_error());
  140. return true;
  141. }
  142. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Sent frame for presentation with timestamp {}ms, late by {}ms", item.timestamp().to_milliseconds(), (current_playback_time() - item.timestamp()).to_milliseconds());
  143. dispatch_new_frame(item.bitmap());
  144. return false;
  145. }
  146. void PlaybackManager::dispatch_state_change()
  147. {
  148. if (on_playback_state_change)
  149. on_playback_state_change();
  150. }
  151. void PlaybackManager::timer_callback()
  152. {
  153. TRY_OR_FATAL_ERROR(m_playback_handler->do_timed_state_update());
  154. }
  155. void PlaybackManager::seek_to_timestamp(Duration target_timestamp, SeekMode seek_mode)
  156. {
  157. TRY_OR_FATAL_ERROR(m_playback_handler->seek(target_timestamp, seek_mode));
  158. }
  159. DecoderErrorOr<Optional<Duration>> PlaybackManager::seek_demuxer_to_most_recent_keyframe(Duration timestamp, Optional<Duration> earliest_available_sample)
  160. {
  161. return m_demuxer->seek_to_most_recent_keyframe(m_selected_video_track, timestamp, move(earliest_available_sample));
  162. }
  163. Optional<FrameQueueItem> PlaybackManager::dequeue_one_frame()
  164. {
  165. auto result = m_frame_queue.dequeue();
  166. m_decode_wait_condition.broadcast();
  167. if (result.is_error()) {
  168. if (result.error() != VideoFrameQueue::QueueStatus::Empty)
  169. dispatch_fatal_error(Error::from_string_literal("Dequeue failed with an unexpected error"));
  170. return {};
  171. }
  172. return result.release_value();
  173. }
  174. void PlaybackManager::set_state_update_timer(int delay_ms)
  175. {
  176. m_state_update_timer->start(delay_ms);
  177. }
  178. void PlaybackManager::restart_playback()
  179. {
  180. seek_to_timestamp(Duration::zero());
  181. }
  182. void PlaybackManager::decode_and_queue_one_sample()
  183. {
  184. #if PLAYBACK_MANAGER_DEBUG
  185. auto start_time = MonotonicTime::now();
  186. #endif
  187. FrameQueueItem item_to_enqueue;
  188. while (item_to_enqueue.is_empty()) {
  189. // Get a sample to decode.
  190. auto sample_result = [&]() {
  191. // FIXME: Implement and use a class to enforce that this field is accessed through a mutex (like Kernel::MutexProtected).
  192. Threading::MutexLocker demuxer_locker(m_demuxer_mutex);
  193. return m_demuxer->get_next_video_sample_for_track(m_selected_video_track);
  194. }();
  195. if (sample_result.is_error()) {
  196. item_to_enqueue = FrameQueueItem::error_marker(sample_result.release_error(), FrameQueueItem::no_timestamp);
  197. break;
  198. }
  199. auto sample = sample_result.release_value();
  200. // Submit the sample to the decoder.
  201. auto decode_result = m_decoder->receive_sample(sample->data());
  202. if (decode_result.is_error()) {
  203. item_to_enqueue = FrameQueueItem::error_marker(decode_result.release_error(), sample->timestamp());
  204. break;
  205. }
  206. // Retrieve the last available frame to present.
  207. OwnPtr<VideoFrame> decoded_frame = nullptr;
  208. while (true) {
  209. auto frame_result = m_decoder->get_decoded_frame();
  210. if (frame_result.is_error()) {
  211. if (frame_result.error().category() == DecoderErrorCategory::NeedsMoreInput) {
  212. break;
  213. }
  214. item_to_enqueue = FrameQueueItem::error_marker(frame_result.release_error(), sample->timestamp());
  215. break;
  216. }
  217. decoded_frame = frame_result.release_value();
  218. }
  219. // Convert the frame for display.
  220. if (decoded_frame != nullptr) {
  221. auto& cicp = decoded_frame->cicp();
  222. cicp.adopt_specified_values(sample->container_cicp());
  223. cicp.default_code_points_if_unspecified({ ColorPrimaries::BT709, TransferCharacteristics::BT709, MatrixCoefficients::BT709, VideoFullRangeFlag::Studio });
  224. // BT.601, BT.709 and BT.2020 have a similar transfer function to sRGB, so other applications
  225. // (Chromium, VLC) forgo transfer characteristics conversion. We will emulate that behavior by
  226. // handling those as sRGB instead, which causes no transfer function change in the output,
  227. // unless display color management is later implemented.
  228. switch (cicp.transfer_characteristics()) {
  229. case TransferCharacteristics::BT601:
  230. case TransferCharacteristics::BT709:
  231. case TransferCharacteristics::BT2020BitDepth10:
  232. case TransferCharacteristics::BT2020BitDepth12:
  233. cicp.set_transfer_characteristics(TransferCharacteristics::SRGB);
  234. break;
  235. default:
  236. break;
  237. }
  238. auto bitmap_result = decoded_frame->to_bitmap();
  239. if (bitmap_result.is_error())
  240. item_to_enqueue = FrameQueueItem::error_marker(bitmap_result.release_error(), sample->timestamp());
  241. else
  242. item_to_enqueue = FrameQueueItem::frame(bitmap_result.release_value(), sample->timestamp());
  243. break;
  244. }
  245. }
  246. VERIFY(!item_to_enqueue.is_empty());
  247. #if PLAYBACK_MANAGER_DEBUG
  248. dbgln("Media Decoder: Sample at {}ms took {}ms to decode, queue contains ~{} items", item_to_enqueue.timestamp().to_milliseconds(), (MonotonicTime::now() - start_time).to_milliseconds(), m_frame_queue.weak_used());
  249. #endif
  250. auto wait = [&] {
  251. auto wait_locker = Threading::MutexLocker(m_decode_wait_mutex);
  252. m_decode_wait_condition.wait();
  253. };
  254. bool had_error = item_to_enqueue.is_error();
  255. while (true) {
  256. if (m_frame_queue.can_enqueue()) {
  257. MUST(m_frame_queue.enqueue(move(item_to_enqueue)));
  258. break;
  259. }
  260. if (m_stop_decoding.load()) {
  261. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Media Decoder: Received signal to stop, exiting decode function...");
  262. return;
  263. }
  264. m_buffer_is_full.exchange(true);
  265. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Media Decoder: Waiting for a frame to be dequeued...");
  266. wait();
  267. }
  268. if (had_error) {
  269. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Media Decoder: Encountered {}, waiting...", item_to_enqueue.error().category() == DecoderErrorCategory::EndOfStream ? "end of stream"sv : "error"sv);
  270. m_buffer_is_full.exchange(true);
  271. wait();
  272. }
  273. m_buffer_is_full.exchange(false);
  274. }
  275. Duration PlaybackManager::PlaybackStateHandler::current_time() const
  276. {
  277. return m_manager.m_last_present_in_media_time;
  278. }
  279. ErrorOr<void> PlaybackManager::PlaybackStateHandler::seek(Duration target_timestamp, SeekMode seek_mode)
  280. {
  281. return replace_handler_and_delete_this<SeekingStateHandler>(is_playing(), target_timestamp, seek_mode);
  282. }
  283. ErrorOr<void> PlaybackManager::PlaybackStateHandler::stop()
  284. {
  285. return replace_handler_and_delete_this<StoppedStateHandler>();
  286. }
  287. template<class T, class... Args>
  288. ErrorOr<void> PlaybackManager::PlaybackStateHandler::replace_handler_and_delete_this(Args... args)
  289. {
  290. OwnPtr<PlaybackStateHandler> temp_handler = TRY(try_make<T>(m_manager, args...));
  291. m_manager.m_playback_handler.swap(temp_handler);
  292. #if PLAYBACK_MANAGER_DEBUG
  293. m_has_exited = true;
  294. dbgln("Changing state from {} to {}", temp_handler->name(), m_manager.m_playback_handler->name());
  295. #endif
  296. TRY(m_manager.m_playback_handler->on_enter());
  297. m_manager.dispatch_state_change();
  298. return {};
  299. }
  300. PlaybackManager& PlaybackManager::PlaybackStateHandler::manager() const
  301. {
  302. #if PLAYBACK_MANAGER_DEBUG
  303. VERIFY(!m_has_exited);
  304. #endif
  305. return m_manager;
  306. }
  307. class PlaybackManager::ResumingStateHandler : public PlaybackManager::PlaybackStateHandler {
  308. public:
  309. ResumingStateHandler(PlaybackManager& manager, bool playing)
  310. : PlaybackStateHandler(manager)
  311. , m_playing(playing)
  312. {
  313. }
  314. ~ResumingStateHandler() override = default;
  315. protected:
  316. ErrorOr<void> assume_next_state()
  317. {
  318. if (!m_playing)
  319. return replace_handler_and_delete_this<PausedStateHandler>();
  320. return replace_handler_and_delete_this<PlayingStateHandler>();
  321. }
  322. ErrorOr<void> play() override
  323. {
  324. m_playing = true;
  325. manager().dispatch_state_change();
  326. return {};
  327. }
  328. bool is_playing() const override { return m_playing; }
  329. ErrorOr<void> pause() override
  330. {
  331. m_playing = false;
  332. manager().dispatch_state_change();
  333. return {};
  334. }
  335. bool m_playing { false };
  336. };
  337. class PlaybackManager::PlayingStateHandler : public PlaybackManager::PlaybackStateHandler {
  338. public:
  339. PlayingStateHandler(PlaybackManager& manager)
  340. : PlaybackStateHandler(manager)
  341. {
  342. }
  343. ~PlayingStateHandler() override = default;
  344. private:
  345. ErrorOr<void> on_enter() override
  346. {
  347. m_last_present_in_real_time = MonotonicTime::now();
  348. return do_timed_state_update();
  349. }
  350. StringView name() override { return "Playing"sv; }
  351. bool is_playing() const override { return true; }
  352. PlaybackState get_state() const override { return PlaybackState::Playing; }
  353. ErrorOr<void> pause() override
  354. {
  355. manager().m_last_present_in_media_time = current_time();
  356. return replace_handler_and_delete_this<PausedStateHandler>();
  357. }
  358. ErrorOr<void> buffer() override
  359. {
  360. manager().m_last_present_in_media_time = current_time();
  361. return replace_handler_and_delete_this<BufferingStateHandler>(true);
  362. }
  363. Duration current_time() const override
  364. {
  365. return manager().m_last_present_in_media_time + (MonotonicTime::now() - m_last_present_in_real_time);
  366. }
  367. ErrorOr<void> do_timed_state_update() override
  368. {
  369. auto set_presentation_timer = [&]() {
  370. auto frame_time_ms = (manager().m_next_frame->timestamp() - current_time()).to_milliseconds();
  371. VERIFY(frame_time_ms <= NumericLimits<int>::max());
  372. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Time until next frame is {}ms", frame_time_ms);
  373. manager().set_state_update_timer(max(static_cast<int>(frame_time_ms), 0));
  374. };
  375. if (manager().m_next_frame.has_value() && current_time() < manager().m_next_frame->timestamp()) {
  376. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Current time {}ms is too early to present the next frame at {}ms, delaying", current_time().to_milliseconds(), manager().m_next_frame->timestamp().to_milliseconds());
  377. set_presentation_timer();
  378. return {};
  379. }
  380. Optional<FrameQueueItem> future_frame_item;
  381. bool should_present_frame = false;
  382. // Skip frames until we find a frame past the current playback time, and keep the one that precedes it to display.
  383. while (true) {
  384. future_frame_item = manager().dequeue_one_frame();
  385. if (!future_frame_item.has_value())
  386. break;
  387. if (future_frame_item->timestamp() >= current_time() || future_frame_item->timestamp() == FrameQueueItem::no_timestamp) {
  388. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Should present frame, future {} is error or after {}ms", future_frame_item->debug_string(), current_time().to_milliseconds());
  389. should_present_frame = true;
  390. break;
  391. }
  392. if (manager().m_next_frame.has_value()) {
  393. dbgln_if(PLAYBACK_MANAGER_DEBUG, "At {}ms: Dropped {} in favor of {}", current_time().to_milliseconds(), manager().m_next_frame->debug_string(), future_frame_item->debug_string());
  394. manager().m_skipped_frames++;
  395. }
  396. manager().m_next_frame.emplace(future_frame_item.release_value());
  397. }
  398. // If we don't have both of these items, we can't present, since we need to set a timer for
  399. // the next frame. Check if we need to buffer based on the current state.
  400. if (!manager().m_next_frame.has_value() || !future_frame_item.has_value()) {
  401. #if PLAYBACK_MANAGER_DEBUG
  402. StringBuilder debug_string_builder;
  403. debug_string_builder.append("We don't have "sv);
  404. if (!manager().m_next_frame.has_value()) {
  405. debug_string_builder.append("a frame to present"sv);
  406. if (!future_frame_item.has_value())
  407. debug_string_builder.append(" or a future frame"sv);
  408. } else {
  409. debug_string_builder.append("a future frame"sv);
  410. }
  411. debug_string_builder.append(", checking for error and buffering"sv);
  412. dbgln_if(PLAYBACK_MANAGER_DEBUG, debug_string_builder.to_deprecated_string());
  413. #endif
  414. if (future_frame_item.has_value()) {
  415. if (future_frame_item->is_error()) {
  416. manager().dispatch_decoder_error(future_frame_item.release_value().release_error());
  417. return {};
  418. }
  419. manager().m_next_frame.emplace(future_frame_item.release_value());
  420. }
  421. TRY(buffer());
  422. return {};
  423. }
  424. // If we have a frame, send it for presentation.
  425. if (should_present_frame) {
  426. auto now = MonotonicTime::now();
  427. manager().m_last_present_in_media_time += now - m_last_present_in_real_time;
  428. m_last_present_in_real_time = now;
  429. if (manager().dispatch_frame_queue_item(manager().m_next_frame.release_value()))
  430. return {};
  431. }
  432. // Now that we've presented the current frame, we can throw whatever error is next in queue.
  433. // This way, we always display a frame before the stream ends, and should also show any frames
  434. // we already had when a real error occurs.
  435. if (future_frame_item->is_error()) {
  436. manager().dispatch_decoder_error(future_frame_item.release_value().release_error());
  437. return {};
  438. }
  439. // The future frame item becomes the next one to present.
  440. manager().m_next_frame.emplace(future_frame_item.release_value());
  441. set_presentation_timer();
  442. return {};
  443. }
  444. MonotonicTime m_last_present_in_real_time = MonotonicTime::now_coarse();
  445. };
  446. class PlaybackManager::PausedStateHandler : public PlaybackManager::PlaybackStateHandler {
  447. public:
  448. PausedStateHandler(PlaybackManager& manager)
  449. : PlaybackStateHandler(manager)
  450. {
  451. }
  452. ~PausedStateHandler() override = default;
  453. private:
  454. StringView name() override { return "Paused"sv; }
  455. ErrorOr<void> play() override
  456. {
  457. return replace_handler_and_delete_this<PlayingStateHandler>();
  458. }
  459. bool is_playing() const override { return false; }
  460. PlaybackState get_state() const override { return PlaybackState::Paused; }
  461. };
  462. // FIXME: This is a placeholder variable that could be scaled based on how long each frame decode takes to
  463. // avoid triggering the timer to check the queue constantly. However, doing so may reduce the speed
  464. // of seeking due to the decode thread having to wait for a signal to continue decoding.
  465. constexpr int buffering_or_seeking_decode_wait_time = 1;
  466. class PlaybackManager::BufferingStateHandler : public PlaybackManager::ResumingStateHandler {
  467. using PlaybackManager::ResumingStateHandler::ResumingStateHandler;
  468. ErrorOr<void> on_enter() override
  469. {
  470. manager().set_state_update_timer(buffering_or_seeking_decode_wait_time);
  471. return {};
  472. }
  473. StringView name() override { return "Buffering"sv; }
  474. ErrorOr<void> do_timed_state_update() override
  475. {
  476. auto buffer_is_full = manager().m_buffer_is_full.load();
  477. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Buffering timer callback has been called. Buffer is {}.", buffer_is_full ? "full, exiting"sv : "not full, waiting"sv);
  478. if (buffer_is_full)
  479. return assume_next_state();
  480. manager().set_state_update_timer(buffering_or_seeking_decode_wait_time);
  481. return {};
  482. }
  483. PlaybackState get_state() const override { return PlaybackState::Buffering; }
  484. };
  485. class PlaybackManager::SeekingStateHandler : public PlaybackManager::ResumingStateHandler {
  486. public:
  487. SeekingStateHandler(PlaybackManager& manager, bool playing, Duration target_timestamp, SeekMode seek_mode)
  488. : ResumingStateHandler(manager, playing)
  489. , m_target_timestamp(target_timestamp)
  490. , m_seek_mode(seek_mode)
  491. {
  492. }
  493. ~SeekingStateHandler() override = default;
  494. private:
  495. ErrorOr<void> on_enter() override
  496. {
  497. auto earliest_available_sample = manager().m_last_present_in_media_time;
  498. if (manager().m_next_frame.has_value() && manager().m_next_frame->timestamp() != FrameQueueItem::no_timestamp) {
  499. earliest_available_sample = min(earliest_available_sample, manager().m_next_frame->timestamp());
  500. }
  501. {
  502. Threading::MutexLocker demuxer_locker(manager().m_demuxer_mutex);
  503. auto demuxer_seek_result = manager().seek_demuxer_to_most_recent_keyframe(m_target_timestamp, earliest_available_sample);
  504. if (demuxer_seek_result.is_error()) {
  505. manager().dispatch_decoder_error(demuxer_seek_result.release_error());
  506. return {};
  507. }
  508. auto keyframe_timestamp = demuxer_seek_result.release_value();
  509. #if PLAYBACK_MANAGER_DEBUG
  510. auto seek_mode_name = m_seek_mode == SeekMode::Accurate ? "Accurate"sv : "Fast"sv;
  511. if (keyframe_timestamp.has_value())
  512. dbgln("{} seeking to timestamp target {}ms, selected keyframe at {}ms", seek_mode_name, m_target_timestamp.to_milliseconds(), keyframe_timestamp->to_milliseconds());
  513. else
  514. dbgln("{} seeking to timestamp target {}ms, demuxer kept its iterator position after {}ms", seek_mode_name, m_target_timestamp.to_milliseconds(), earliest_available_sample.to_milliseconds());
  515. #endif
  516. if (m_seek_mode == SeekMode::Fast)
  517. m_target_timestamp = keyframe_timestamp.value_or(manager().m_last_present_in_media_time);
  518. if (keyframe_timestamp.has_value()) {
  519. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Keyframe is nearer to the target than the current frames, emptying queue");
  520. while (manager().dequeue_one_frame().has_value()) { }
  521. manager().m_next_frame.clear();
  522. manager().m_last_present_in_media_time = keyframe_timestamp.value();
  523. } else if (m_target_timestamp >= manager().m_last_present_in_media_time && manager().m_next_frame.has_value() && manager().m_next_frame.value().timestamp() > m_target_timestamp) {
  524. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Target timestamp is between the last presented frame and the next frame, exiting seek at {}ms", m_target_timestamp.to_milliseconds());
  525. manager().m_last_present_in_media_time = m_target_timestamp;
  526. return assume_next_state();
  527. }
  528. }
  529. return skip_samples_until_timestamp();
  530. }
  531. ErrorOr<void> skip_samples_until_timestamp()
  532. {
  533. while (true) {
  534. auto optional_item = manager().dequeue_one_frame();
  535. if (!optional_item.has_value())
  536. break;
  537. auto item = optional_item.release_value();
  538. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Dequeuing frame at {}ms and comparing to seek target {}ms", item.timestamp().to_milliseconds(), m_target_timestamp.to_milliseconds());
  539. if (manager().m_next_frame.has_value() && (item.timestamp() > m_target_timestamp || item.timestamp() == FrameQueueItem::no_timestamp)) {
  540. // If the frame we're presenting is later than the target timestamp, skip the timestamp forward to it.
  541. if (manager().m_next_frame->timestamp() > m_target_timestamp) {
  542. manager().m_last_present_in_media_time = manager().m_next_frame->timestamp();
  543. } else {
  544. manager().m_last_present_in_media_time = m_target_timestamp;
  545. }
  546. if (manager().dispatch_frame_queue_item(manager().m_next_frame.release_value()))
  547. return {};
  548. manager().m_next_frame.emplace(item);
  549. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Exiting seek to {} state at {}ms", m_playing ? "Playing" : "Paused", manager().m_last_present_in_media_time.to_milliseconds());
  550. return assume_next_state();
  551. }
  552. manager().m_next_frame.emplace(item);
  553. }
  554. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Frame queue is empty while seeking, waiting for buffer to fill.");
  555. manager().set_state_update_timer(buffering_or_seeking_decode_wait_time);
  556. return {};
  557. }
  558. StringView name() override { return "Seeking"sv; }
  559. ErrorOr<void> seek(Duration target_timestamp, SeekMode seek_mode) override
  560. {
  561. m_target_timestamp = target_timestamp;
  562. m_seek_mode = seek_mode;
  563. return on_enter();
  564. }
  565. Duration current_time() const override
  566. {
  567. return m_target_timestamp;
  568. }
  569. // We won't need this override when threaded, the queue can pause us in on_enter().
  570. ErrorOr<void> do_timed_state_update() override
  571. {
  572. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Seeking wait finished, attempting to dequeue until timestamp.");
  573. return skip_samples_until_timestamp();
  574. }
  575. PlaybackState get_state() const override { return PlaybackState::Seeking; }
  576. Duration m_target_timestamp { Duration::zero() };
  577. SeekMode m_seek_mode { SeekMode::Accurate };
  578. };
  579. class PlaybackManager::StoppedStateHandler : public PlaybackManager::PlaybackStateHandler {
  580. public:
  581. StoppedStateHandler(PlaybackManager& manager)
  582. : PlaybackStateHandler(manager)
  583. {
  584. }
  585. ~StoppedStateHandler() override = default;
  586. private:
  587. ErrorOr<void> on_enter() override
  588. {
  589. return {};
  590. }
  591. StringView name() override { return "Stopped"sv; }
  592. ErrorOr<void> play() override
  593. {
  594. // When Stopped, the decoder thread will be waiting for a signal to start its loop going again.
  595. manager().m_decode_wait_condition.broadcast();
  596. return replace_handler_and_delete_this<SeekingStateHandler>(true, Duration::zero(), SeekMode::Fast);
  597. }
  598. bool is_playing() const override { return false; }
  599. PlaybackState get_state() const override { return PlaybackState::Stopped; }
  600. };
  601. }