PlaybackManager.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655
  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. class DefaultPlaybackTimer final : public PlaybackTimer {
  24. public:
  25. static ErrorOr<NonnullOwnPtr<DefaultPlaybackTimer>> create(int interval_ms, Function<void()>&& timeout_handler)
  26. {
  27. auto timer = TRY(Core::Timer::create_single_shot(interval_ms, move(timeout_handler)));
  28. return adopt_nonnull_own_or_enomem(new (nothrow) DefaultPlaybackTimer(move(timer)));
  29. }
  30. virtual void start() override { m_timer->start(); }
  31. virtual void start(int interval_ms) override { m_timer->start(interval_ms); }
  32. private:
  33. explicit DefaultPlaybackTimer(NonnullRefPtr<Core::Timer> timer)
  34. : m_timer(move(timer))
  35. {
  36. }
  37. NonnullRefPtr<Core::Timer> m_timer;
  38. };
  39. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::from_file(StringView filename, PlaybackTimerCreator playback_timer_creator)
  40. {
  41. auto demuxer = TRY(Matroska::MatroskaDemuxer::from_file(filename));
  42. return create_with_demuxer(move(demuxer), move(playback_timer_creator));
  43. }
  44. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::from_data(ReadonlyBytes data, PlaybackTimerCreator playback_timer_creator)
  45. {
  46. auto demuxer = TRY(Matroska::MatroskaDemuxer::from_data(data));
  47. return create_with_demuxer(move(demuxer), move(playback_timer_creator));
  48. }
  49. DecoderErrorOr<NonnullOwnPtr<PlaybackManager>> PlaybackManager::create_with_demuxer(NonnullOwnPtr<Demuxer> demuxer, PlaybackTimerCreator playback_timer_creator)
  50. {
  51. auto video_tracks = TRY(demuxer->get_tracks_for_type(TrackType::Video));
  52. if (video_tracks.is_empty())
  53. return DecoderError::with_description(DecoderErrorCategory::Invalid, "No video track is present"sv);
  54. auto track = video_tracks[0];
  55. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Selecting video track number {}", track.identifier());
  56. return make<PlaybackManager>(demuxer, track, make<VP9::Decoder>(), move(playback_timer_creator));
  57. }
  58. PlaybackManager::PlaybackManager(NonnullOwnPtr<Demuxer>& demuxer, Track video_track, NonnullOwnPtr<VideoDecoder>&& decoder, PlaybackTimerCreator playback_timer_creator)
  59. : m_demuxer(move(demuxer))
  60. , m_selected_video_track(video_track)
  61. , m_decoder(move(decoder))
  62. , m_frame_queue(make<VideoFrameQueue>())
  63. , m_playback_handler(make<SeekingStateHandler>(*this, false, Time::zero(), SeekMode::Fast))
  64. {
  65. if (playback_timer_creator) {
  66. m_present_timer = playback_timer_creator(0, [&] { timer_callback(); }).release_value_but_fixme_should_propagate_errors();
  67. m_decode_timer = playback_timer_creator(0, [&] { on_decode_timer(); }).release_value_but_fixme_should_propagate_errors();
  68. } else {
  69. m_present_timer = DefaultPlaybackTimer::create(0, [&] { timer_callback(); }).release_value_but_fixme_should_propagate_errors();
  70. m_decode_timer = DefaultPlaybackTimer::create(0, [&] { on_decode_timer(); }).release_value_but_fixme_should_propagate_errors();
  71. }
  72. TRY_OR_FATAL_ERROR(m_playback_handler->on_enter());
  73. }
  74. void PlaybackManager::resume_playback()
  75. {
  76. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Resuming playback.");
  77. TRY_OR_FATAL_ERROR(m_playback_handler->play());
  78. }
  79. void PlaybackManager::pause_playback()
  80. {
  81. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Pausing playback.");
  82. if (!m_playback_handler->is_playing())
  83. warnln("Cannot pause.");
  84. TRY_OR_FATAL_ERROR(m_playback_handler->pause());
  85. }
  86. Time PlaybackManager::current_playback_time()
  87. {
  88. return m_playback_handler->current_time();
  89. }
  90. Time PlaybackManager::duration()
  91. {
  92. auto duration_result = m_demuxer->duration();
  93. if (duration_result.is_error())
  94. dispatch_decoder_error(duration_result.release_error());
  95. return duration_result.release_value();
  96. }
  97. void PlaybackManager::dispatch_fatal_error(Error error)
  98. {
  99. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Encountered fatal error: {}", error.string_literal());
  100. // FIXME: For threading, this will have to use a pre-allocated event to send to the main loop
  101. // to be able to gracefully handle OOM.
  102. if (on_fatal_playback_error)
  103. on_fatal_playback_error(move(error));
  104. }
  105. void PlaybackManager::dispatch_decoder_error(DecoderError error)
  106. {
  107. switch (error.category()) {
  108. case DecoderErrorCategory::EndOfStream:
  109. dbgln_if(PLAYBACK_MANAGER_DEBUG, "{}", error.string_literal());
  110. TRY_OR_FATAL_ERROR(m_playback_handler->stop());
  111. break;
  112. default:
  113. dbgln("Playback error encountered: {}", error.string_literal());
  114. TRY_OR_FATAL_ERROR(m_playback_handler->stop());
  115. if (on_decoder_error)
  116. on_decoder_error(move(error));
  117. break;
  118. }
  119. }
  120. void PlaybackManager::dispatch_new_frame(RefPtr<Gfx::Bitmap> frame)
  121. {
  122. if (on_video_frame)
  123. on_video_frame(move(frame));
  124. }
  125. bool PlaybackManager::dispatch_frame_queue_item(FrameQueueItem&& item)
  126. {
  127. if (item.is_error()) {
  128. dispatch_decoder_error(item.release_error());
  129. return true;
  130. }
  131. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Sent frame for presentation with timestamp {}ms", item.timestamp().to_milliseconds());
  132. dispatch_new_frame(item.bitmap());
  133. return false;
  134. }
  135. void PlaybackManager::dispatch_state_change()
  136. {
  137. if (on_playback_state_change)
  138. on_playback_state_change();
  139. }
  140. void PlaybackManager::timer_callback()
  141. {
  142. TRY_OR_FATAL_ERROR(m_playback_handler->on_timer_callback());
  143. }
  144. void PlaybackManager::seek_to_timestamp(Time target_timestamp, SeekMode seek_mode)
  145. {
  146. TRY_OR_FATAL_ERROR(m_playback_handler->seek(target_timestamp, seek_mode));
  147. }
  148. Optional<Time> PlaybackManager::seek_demuxer_to_most_recent_keyframe(Time timestamp, Optional<Time> earliest_available_sample)
  149. {
  150. // FIXME: When the demuxer is getting samples off the main thread in the future, this needs to
  151. // mutex so that seeking can't happen while that thread is getting a sample.
  152. auto result = m_demuxer->seek_to_most_recent_keyframe(m_selected_video_track, timestamp, move(earliest_available_sample));
  153. if (result.is_error())
  154. dispatch_decoder_error(result.release_error());
  155. return result.release_value();
  156. }
  157. void PlaybackManager::restart_playback()
  158. {
  159. seek_to_timestamp(Time::zero());
  160. }
  161. void PlaybackManager::start_timer(int milliseconds)
  162. {
  163. m_present_timer->start(milliseconds);
  164. }
  165. bool PlaybackManager::decode_and_queue_one_sample()
  166. {
  167. if (m_frame_queue->size() >= FRAME_BUFFER_COUNT) {
  168. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Frame queue is full, stopping");
  169. return false;
  170. }
  171. #if PLAYBACK_MANAGER_DEBUG
  172. auto start_time = Time::now_monotonic();
  173. #endif
  174. auto enqueue_error = [&](DecoderError&& error, Time timestamp) {
  175. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Enqueued decoder error: {}", error.string_literal());
  176. m_frame_queue->enqueue(FrameQueueItem::error_marker(move(error), timestamp));
  177. };
  178. #define TRY_OR_ENQUEUE_ERROR(expression, timestamp) \
  179. ({ \
  180. auto&& _temporary_result = (expression); \
  181. if (_temporary_result.is_error()) { \
  182. enqueue_error(_temporary_result.release_error(), (timestamp)); \
  183. return false; \
  184. } \
  185. static_assert(!::AK::Detail::IsLvalueReference<decltype(_temporary_result.release_value())>, \
  186. "Do not return a reference from a fallible expression"); \
  187. _temporary_result.release_value(); \
  188. })
  189. auto frame_sample = TRY_OR_ENQUEUE_ERROR(m_demuxer->get_next_video_sample_for_track(m_selected_video_track), Time::min());
  190. OwnPtr<VideoFrame> decoded_frame = nullptr;
  191. while (!decoded_frame) {
  192. TRY_OR_ENQUEUE_ERROR(m_decoder->receive_sample(frame_sample->data()), frame_sample->timestamp());
  193. while (true) {
  194. auto frame_result = m_decoder->get_decoded_frame();
  195. if (frame_result.is_error()) {
  196. if (frame_result.error().category() == DecoderErrorCategory::NeedsMoreInput)
  197. break;
  198. enqueue_error(frame_result.release_error(), frame_sample->timestamp());
  199. return false;
  200. }
  201. decoded_frame = frame_result.release_value();
  202. VERIFY(decoded_frame);
  203. }
  204. }
  205. auto& cicp = decoded_frame->cicp();
  206. cicp.adopt_specified_values(frame_sample->container_cicp());
  207. cicp.default_code_points_if_unspecified({ ColorPrimaries::BT709, TransferCharacteristics::BT709, MatrixCoefficients::BT709, VideoFullRangeFlag::Studio });
  208. // BT.601, BT.709 and BT.2020 have a similar transfer function to sRGB, so other applications
  209. // (Chromium, VLC) forgo transfer characteristics conversion. We will emulate that behavior by
  210. // handling those as sRGB instead, which causes no transfer function change in the output,
  211. // unless display color management is later implemented.
  212. switch (cicp.transfer_characteristics()) {
  213. case TransferCharacteristics::BT601:
  214. case TransferCharacteristics::BT709:
  215. case TransferCharacteristics::BT2020BitDepth10:
  216. case TransferCharacteristics::BT2020BitDepth12:
  217. cicp.set_transfer_characteristics(TransferCharacteristics::SRGB);
  218. break;
  219. default:
  220. break;
  221. }
  222. auto bitmap = TRY_OR_ENQUEUE_ERROR(decoded_frame->to_bitmap(), frame_sample->timestamp());
  223. m_frame_queue->enqueue(FrameQueueItem::frame(bitmap, frame_sample->timestamp()));
  224. #if PLAYBACK_MANAGER_DEBUG
  225. auto end_time = Time::now_monotonic();
  226. dbgln("Decoding sample at {}ms took {}ms, queue contains {} items", frame_sample->timestamp().to_milliseconds(), (end_time - start_time).to_milliseconds(), m_frame_queue->size());
  227. #endif
  228. return true;
  229. }
  230. void PlaybackManager::on_decode_timer()
  231. {
  232. if (!decode_and_queue_one_sample()) {
  233. // Note: When threading is implemented, this must be dispatched via an event loop.
  234. TRY_OR_FATAL_ERROR(m_playback_handler->on_buffer_filled());
  235. return;
  236. }
  237. // Continually decode until buffering is complete
  238. m_decode_timer->start(0);
  239. }
  240. Time PlaybackManager::PlaybackStateHandler::current_time() const
  241. {
  242. return m_manager.m_last_present_in_media_time;
  243. }
  244. ErrorOr<void> PlaybackManager::PlaybackStateHandler::seek(Time target_timestamp, SeekMode seek_mode)
  245. {
  246. return replace_handler_and_delete_this<SeekingStateHandler>(is_playing(), target_timestamp, seek_mode);
  247. }
  248. ErrorOr<void> PlaybackManager::PlaybackStateHandler::stop()
  249. {
  250. return replace_handler_and_delete_this<StoppedStateHandler>();
  251. }
  252. template<class T, class... Args>
  253. ErrorOr<void> PlaybackManager::PlaybackStateHandler::replace_handler_and_delete_this(Args... args)
  254. {
  255. auto temp_handler = TRY(adopt_nonnull_own_or_enomem<PlaybackStateHandler>(new (nothrow) T(m_manager, args...)));
  256. m_manager.m_playback_handler.swap(temp_handler);
  257. #if PLAYBACK_MANAGER_DEBUG
  258. m_has_exited = true;
  259. dbgln("Changing state from {} to {}", temp_handler->name(), m_manager.m_playback_handler->name());
  260. #endif
  261. TRY(m_manager.m_playback_handler->on_enter());
  262. m_manager.dispatch_state_change();
  263. return {};
  264. }
  265. PlaybackManager& PlaybackManager::PlaybackStateHandler::manager() const
  266. {
  267. #if PLAYBACK_MANAGER_DEBUG
  268. VERIFY(!m_has_exited);
  269. #endif
  270. return m_manager;
  271. }
  272. class PlaybackManager::ResumingStateHandler : public PlaybackManager::PlaybackStateHandler {
  273. public:
  274. ResumingStateHandler(PlaybackManager& manager, bool playing)
  275. : PlaybackStateHandler(manager)
  276. , m_playing(playing)
  277. {
  278. }
  279. ~ResumingStateHandler() override = default;
  280. protected:
  281. ErrorOr<void> assume_next_state()
  282. {
  283. if (!m_playing)
  284. return replace_handler_and_delete_this<PausedStateHandler>();
  285. return replace_handler_and_delete_this<PlayingStateHandler>();
  286. }
  287. ErrorOr<void> play() override
  288. {
  289. m_playing = true;
  290. manager().dispatch_state_change();
  291. return {};
  292. }
  293. bool is_playing() const override { return m_playing; }
  294. ErrorOr<void> pause() override
  295. {
  296. m_playing = false;
  297. manager().dispatch_state_change();
  298. return {};
  299. }
  300. bool m_playing { false };
  301. };
  302. class PlaybackManager::PlayingStateHandler : public PlaybackManager::PlaybackStateHandler {
  303. public:
  304. PlayingStateHandler(PlaybackManager& manager)
  305. : PlaybackStateHandler(manager)
  306. {
  307. }
  308. ~PlayingStateHandler() override = default;
  309. private:
  310. ErrorOr<void> on_enter() override
  311. {
  312. m_last_present_in_real_time = Time::now_monotonic();
  313. return on_timer_callback();
  314. }
  315. StringView name() override { return "Playing"sv; }
  316. bool is_playing() const override { return true; };
  317. PlaybackState get_state() const override { return PlaybackState::Playing; }
  318. ErrorOr<void> pause() override
  319. {
  320. manager().m_last_present_in_media_time = current_time();
  321. return replace_handler_and_delete_this<PausedStateHandler>();
  322. }
  323. ErrorOr<void> buffer() override
  324. {
  325. manager().m_last_present_in_media_time = current_time();
  326. return replace_handler_and_delete_this<BufferingStateHandler>(true);
  327. }
  328. Time current_time() const override
  329. {
  330. return manager().m_last_present_in_media_time + (Time::now_monotonic() - m_last_present_in_real_time);
  331. }
  332. ErrorOr<void> on_timer_callback() override
  333. {
  334. auto set_presentation_timer = [&]() {
  335. auto frame_time_ms = (manager().m_next_frame->timestamp() - current_time()).to_milliseconds();
  336. VERIFY(frame_time_ms <= NumericLimits<int>::max());
  337. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Time until next frame is {}ms", frame_time_ms);
  338. manager().start_timer(max(static_cast<int>(frame_time_ms), 0));
  339. };
  340. if (manager().m_next_frame.has_value() && current_time() < manager().m_next_frame->timestamp()) {
  341. 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());
  342. set_presentation_timer();
  343. return {};
  344. }
  345. Optional<FrameQueueItem> future_frame_item;
  346. bool should_present_frame = false;
  347. // Skip frames until we find a frame past the current playback time, and keep the one that precedes it to display.
  348. while (!manager().m_frame_queue->is_empty()) {
  349. future_frame_item.emplace(manager().m_frame_queue->dequeue());
  350. manager().m_decode_timer->start(0);
  351. if (future_frame_item->timestamp() >= current_time() || future_frame_item->timestamp() == FrameQueueItem::no_timestamp) {
  352. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Should present frame, future {} is error or after {}ms", future_frame_item->debug_string(), current_time().to_milliseconds());
  353. should_present_frame = true;
  354. break;
  355. }
  356. if (manager().m_next_frame.has_value()) {
  357. 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());
  358. manager().m_skipped_frames++;
  359. }
  360. manager().m_next_frame.emplace(future_frame_item.release_value());
  361. }
  362. // If we don't have both of these items, we can't present, since we need to set a timer for
  363. // the next frame. Check if we need to buffer based on the current state.
  364. if (!manager().m_next_frame.has_value() || !future_frame_item.has_value()) {
  365. #if PLAYBACK_MANAGER_DEBUG
  366. StringBuilder debug_string_builder;
  367. debug_string_builder.append("We don't have "sv);
  368. if (!manager().m_next_frame.has_value()) {
  369. debug_string_builder.append("a frame to present"sv);
  370. if (!future_frame_item.has_value())
  371. debug_string_builder.append(" or a future frame"sv);
  372. } else {
  373. debug_string_builder.append("a future frame"sv);
  374. }
  375. debug_string_builder.append(", checking for error and buffering"sv);
  376. dbgln_if(PLAYBACK_MANAGER_DEBUG, debug_string_builder.to_deprecated_string());
  377. #endif
  378. if (future_frame_item.has_value()) {
  379. if (future_frame_item->is_error()) {
  380. manager().dispatch_decoder_error(future_frame_item.release_value().release_error());
  381. return {};
  382. }
  383. manager().m_next_frame.emplace(future_frame_item.release_value());
  384. }
  385. TRY(buffer());
  386. return {};
  387. }
  388. // If we have a frame, send it for presentation.
  389. if (should_present_frame) {
  390. auto now = Time::now_monotonic();
  391. manager().m_last_present_in_media_time += now - m_last_present_in_real_time;
  392. m_last_present_in_real_time = now;
  393. if (manager().dispatch_frame_queue_item(manager().m_next_frame.release_value()))
  394. return {};
  395. }
  396. // Now that we've presented the current frame, we can throw whatever error is next in queue.
  397. // This way, we always display a frame before the stream ends, and should also show any frames
  398. // we already had when a real error occurs.
  399. if (future_frame_item->is_error()) {
  400. manager().dispatch_decoder_error(future_frame_item.release_value().release_error());
  401. return {};
  402. }
  403. // The future frame item becomes the next one to present.
  404. manager().m_next_frame.emplace(future_frame_item.release_value());
  405. set_presentation_timer();
  406. return {};
  407. }
  408. Time m_last_present_in_real_time = Time::zero();
  409. };
  410. class PlaybackManager::PausedStateHandler : public PlaybackManager::PlaybackStateHandler {
  411. public:
  412. PausedStateHandler(PlaybackManager& manager)
  413. : PlaybackStateHandler(manager)
  414. {
  415. }
  416. ~PausedStateHandler() override = default;
  417. private:
  418. StringView name() override { return "Paused"sv; }
  419. ErrorOr<void> play() override
  420. {
  421. return replace_handler_and_delete_this<PlayingStateHandler>();
  422. }
  423. bool is_playing() const override { return false; };
  424. PlaybackState get_state() const override { return PlaybackState::Paused; }
  425. };
  426. class PlaybackManager::BufferingStateHandler : public PlaybackManager::ResumingStateHandler {
  427. using PlaybackManager::ResumingStateHandler::ResumingStateHandler;
  428. ErrorOr<void> on_enter() override
  429. {
  430. manager().m_decode_timer->start(0);
  431. return {};
  432. }
  433. StringView name() override { return "Buffering"sv; }
  434. ErrorOr<void> on_buffer_filled() override
  435. {
  436. return assume_next_state();
  437. }
  438. PlaybackState get_state() const override { return PlaybackState::Buffering; }
  439. };
  440. class PlaybackManager::SeekingStateHandler : public PlaybackManager::ResumingStateHandler {
  441. public:
  442. SeekingStateHandler(PlaybackManager& manager, bool playing, Time target_timestamp, SeekMode seek_mode)
  443. : ResumingStateHandler(manager, playing)
  444. , m_target_timestamp(target_timestamp)
  445. , m_seek_mode(seek_mode)
  446. {
  447. }
  448. ~SeekingStateHandler() override = default;
  449. private:
  450. ErrorOr<void> on_enter() override
  451. {
  452. auto earliest_available_sample = manager().m_last_present_in_media_time;
  453. if (manager().m_next_frame.has_value() && manager().m_next_frame->timestamp() != FrameQueueItem::no_timestamp) {
  454. earliest_available_sample = min(earliest_available_sample, manager().m_next_frame->timestamp());
  455. }
  456. auto keyframe_timestamp = manager().seek_demuxer_to_most_recent_keyframe(m_target_timestamp, earliest_available_sample);
  457. #if PLAYBACK_MANAGER_DEBUG
  458. auto seek_mode_name = m_seek_mode == SeekMode::Accurate ? "Accurate"sv : "Fast"sv;
  459. if (keyframe_timestamp.has_value()) {
  460. dbgln("{} seeking to timestamp target {}ms, selected keyframe at {}ms", seek_mode_name, m_target_timestamp.to_milliseconds(), keyframe_timestamp->to_milliseconds());
  461. } else {
  462. 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());
  463. }
  464. #endif
  465. if (m_seek_mode == SeekMode::Fast) {
  466. m_target_timestamp = keyframe_timestamp.value_or(manager().m_last_present_in_media_time);
  467. }
  468. if (keyframe_timestamp.has_value()) {
  469. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Keyframe is nearer to the target than the current frames, clearing queue");
  470. manager().m_frame_queue->clear();
  471. manager().m_next_frame.clear();
  472. manager().m_last_present_in_media_time = keyframe_timestamp.value();
  473. } 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) {
  474. 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());
  475. manager().m_last_present_in_media_time = m_target_timestamp;
  476. return assume_next_state();
  477. }
  478. return skip_samples_until_timestamp();
  479. }
  480. ErrorOr<void> skip_samples_until_timestamp()
  481. {
  482. while (!manager().m_frame_queue->is_empty()) {
  483. auto item = manager().m_frame_queue->dequeue();
  484. manager().m_decode_timer->start(0);
  485. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Dequeuing frame at {}ms and comparing to seek target {}ms", item.timestamp().to_milliseconds(), m_target_timestamp.to_milliseconds());
  486. if (manager().m_next_frame.has_value() && (item.timestamp() > m_target_timestamp || item.timestamp() == FrameQueueItem::no_timestamp)) {
  487. // If the frame we're presenting is later than the target timestamp, skip the timestamp forward to it.
  488. if (manager().m_next_frame->timestamp() > m_target_timestamp) {
  489. manager().m_last_present_in_media_time = manager().m_next_frame->timestamp();
  490. } else {
  491. manager().m_last_present_in_media_time = m_target_timestamp;
  492. }
  493. if (manager().dispatch_frame_queue_item(manager().m_next_frame.release_value()))
  494. return {};
  495. manager().m_next_frame.emplace(item);
  496. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Exiting seek to {} state at {}ms", m_playing ? "Playing" : "Paused", manager().m_last_present_in_media_time.to_milliseconds());
  497. return assume_next_state();
  498. }
  499. manager().m_next_frame.emplace(item);
  500. }
  501. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Frame queue is empty while seeking, waiting for buffer fill.");
  502. manager().m_decode_timer->start(0);
  503. return {};
  504. }
  505. StringView name() override { return "Seeking"sv; }
  506. ErrorOr<void> seek(Time target_timestamp, SeekMode seek_mode) override
  507. {
  508. m_target_timestamp = target_timestamp;
  509. m_seek_mode = seek_mode;
  510. return on_enter();
  511. }
  512. Time current_time() const override
  513. {
  514. return m_target_timestamp;
  515. }
  516. // We won't need this override when threaded, the queue can pause us in on_enter().
  517. ErrorOr<void> on_buffer_filled() override
  518. {
  519. dbgln_if(PLAYBACK_MANAGER_DEBUG, "Buffer filled while seeking, dequeuing until timestamp.");
  520. return skip_samples_until_timestamp();
  521. }
  522. PlaybackState get_state() const override { return PlaybackState::Seeking; }
  523. Time m_target_timestamp { Time::zero() };
  524. SeekMode m_seek_mode { SeekMode::Accurate };
  525. };
  526. class PlaybackManager::StoppedStateHandler : public PlaybackManager::PlaybackStateHandler {
  527. public:
  528. StoppedStateHandler(PlaybackManager& manager)
  529. : PlaybackStateHandler(manager)
  530. {
  531. }
  532. ~StoppedStateHandler() override = default;
  533. private:
  534. ErrorOr<void> on_enter() override
  535. {
  536. return {};
  537. }
  538. StringView name() override { return "Stopped"sv; }
  539. ErrorOr<void> play() override
  540. {
  541. manager().m_next_frame.clear();
  542. manager().m_frame_queue->clear();
  543. auto start_timestamp = manager().seek_demuxer_to_most_recent_keyframe(Time::zero());
  544. VERIFY(start_timestamp.has_value());
  545. manager().m_last_present_in_media_time = start_timestamp.release_value();
  546. return replace_handler_and_delete_this<SeekingStateHandler>(true, Time::zero(), SeekMode::Fast);
  547. }
  548. bool is_playing() const override { return false; };
  549. PlaybackState get_state() const override { return PlaybackState::Stopped; }
  550. };
  551. }