starboard/nplb/player_test_fixture.h - cobalt - Git at Google

 // Copyright 2023 The Cobalt Authors. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //     http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.

 #ifndef STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_
 #define STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_

 #include <atomic>
 #include <deque>
 #include <set>
 #include <string>
 #include <vector>

 #include "starboard/common/queue.h"
 #include "starboard/common/scoped_ptr.h"
 #include "starboard/drm.h"
 #include "starboard/nplb/player_test_util.h"
 #include "starboard/player.h"
 #include "starboard/shared/starboard/player/video_dmp_reader.h"
 #include "starboard/shared/starboard/thread_checker.h"
 #include "starboard/testing/fake_graphics_context_provider.h"

 namespace starboard {
 namespace nplb {

 class SbPlayerTestFixture {
  public:
   // A simple encapsulation of grouped samples.
   class GroupedSamplesIterator;
   class GroupedSamples {
    public:
     struct AudioSamplesDescriptor {
       int start_index = 0;
       int samples_count = 0;
       int64_t timestamp_offset = 0;
       int64_t discarded_duration_from_front = 0;
       int64_t discarded_duration_from_back = 0;
       bool is_end_of_stream = false;
     };

     struct VideoSamplesDescriptor {
       int start_index = 0;
       int samples_count = 0;
       bool is_end_of_stream = false;
     };

     GroupedSamples& AddAudioSamples(int start_index, int number_of_samples);
     GroupedSamples& AddAudioSamples(int start_index,
                                     int number_of_samples,
                                     int64_t timestamp_offset,
                                     int64_t discarded_duration_from_front,
                                     int64_t discarded_duration_from_back);
     GroupedSamples& AddAudioEOS();
     GroupedSamples& AddVideoSamples(int start_index, int number_of_samples);
     GroupedSamples& AddVideoEOS();

     friend class GroupedSamplesIterator;

    private:
     std::vector<AudioSamplesDescriptor> audio_samples_;
     std::vector<VideoSamplesDescriptor> video_samples_;
   };

   SbPlayerTestFixture(
       const SbPlayerTestConfig& config,
       testing::FakeGraphicsContextProvider* fake_graphics_context_provider);
   ~SbPlayerTestFixture();

   void Seek(const int64_t time);
   // Write audio and video samples. It waits for
   // |kSbPlayerDecoderStateNeedsData| internally. When writing EOS are
   // requested, the function will write EOS after all samples of the same type
   // are written.
   void Write(const GroupedSamples& grouped_samples);
   // Wait until kSbPlayerStatePresenting received.
   void WaitForPlayerPresenting();
   // Wait until kSbPlayerStateEndOfStream received.
   void WaitForPlayerEndOfStream();
   int64_t GetCurrentMediaTime() const;

   void SetAudioWriteDuration(int64_t duration);

   SbPlayer GetPlayer() const { return player_; }
   bool HasAudio() const { return audio_dmp_reader_; }
   bool HasVideo() const { return video_dmp_reader_; }

   int64_t GetAudioSampleTimestamp(int index) const;
   int ConvertDurationToAudioBufferCount(int64_t duration) const;
   int ConvertDurationToVideoBufferCount(int64_t duration) const;

  private:
   static constexpr int64_t kDefaultWaitForPlayerStateTimeout = 5'000'000LL;
   static constexpr int64_t kDefaultWaitForCallbackEventTimeout = 15'000;

   typedef shared::starboard::player::video_dmp::VideoDmpReader VideoDmpReader;

   typedef enum CallbackEventType {
     kEmptyEvent,
     kDecoderStateEvent,
     kPlayerStateEvent,
   } CallbackEventType;

   struct CallbackEvent {
     CallbackEvent();
     CallbackEvent(SbPlayer player,
                   SbMediaType type,
                   SbPlayerDecoderState state,
                   int ticket);
     CallbackEvent(SbPlayer player, SbPlayerState state, int ticket);

     CallbackEventType event_type;
     SbPlayer player = kSbPlayerInvalid;
     SbMediaType media_type;
     SbPlayerDecoderState decoder_state;
     SbPlayerState player_state;
     int ticket = SB_PLAYER_INITIAL_TICKET;
   };

   static void DecoderStatusCallback(SbPlayer player,
                                     void* context,
                                     SbMediaType type,
                                     SbPlayerDecoderState state,
                                     int ticket);

   static void PlayerStatusCallback(SbPlayer player,
                                    void* context,
                                    SbPlayerState state,
                                    int ticket);

   static void ErrorCallback(SbPlayer player,
                             void* context,
                             SbPlayerError error,
                             const char* message);

   // Callback methods from the underlying player. Note that callbacks may be
   // invoked on underlying player's thread.
   void OnDecoderState(SbPlayer player,
                       SbMediaType media_type,
                       SbPlayerDecoderState state,
                       int ticket);
   void OnPlayerState(SbPlayer player, SbPlayerState state, int ticket);
   void OnError(SbPlayer player, SbPlayerError error, const char* message);

   void Initialize();
   void TearDown();

   bool CanWriteMoreAudioData();
   bool CanWriteMoreVideoData();

   void WriteAudioSamples(int start_index,
                          int samples_to_write,
                          int64_t timestamp_offset,
                          int64_t discarded_duration_from_front,
                          int64_t discarded_duration_from_back);
   void WriteVideoSamples(int start_index, int samples_to_write);
   void WriteEndOfStream(SbMediaType media_type);

   // Checks if there are pending callback events and, if so, logs the received
   // state update in the appropriate tracking container:
   // * |can_accept_more_*_data_| for SbPlayerDecoderState updates.
   // * |player_state_set_| for SbPlayerState updates.
   // Executes a blocking wait for any new CallbackEvents to be enqueued.
   void WaitAndProcessNextEvent(
       int64_t timeout = kDefaultWaitForCallbackEventTimeout);

   // Waits for |kSbPlayerDecoderStateNeedsData| to be sent.
   void WaitForDecoderStateNeedsData(
       const int64_t timeout = kDefaultWaitForCallbackEventTimeout);

   // Waits for desired player state update to be sent.
   void WaitForPlayerState(
       const SbPlayerState desired_state,
       const int64_t timeout = kDefaultWaitForPlayerStateTimeout);

   // When the |output_mode| is decoding to texture, then this method is used to
   // advance the decoded frames.
   void GetDecodeTargetWhenSupported();

   // Determine if the the current event is valid based on previously received
   // player state updates, or other inputs to the player.
   void AssertPlayerStateIsValid(SbPlayerState state) const;

   bool HasReceivedPlayerState(SbPlayerState state) const {
     return player_state_set_.find(state) != player_state_set_.end();
   }

   shared::starboard::ThreadChecker thread_checker_;
   const SbPlayerOutputMode output_mode_;
   std::string key_system_;
   std::string max_video_capabilities_;
   scoped_ptr<VideoDmpReader> audio_dmp_reader_;
   scoped_ptr<VideoDmpReader> video_dmp_reader_;
   testing::FakeGraphicsContextProvider* fake_graphics_context_provider_;

   SbPlayer player_ = kSbPlayerInvalid;
   SbDrmSystem drm_system_ = kSbDrmSystemInvalid;

   // Queue of events from the underlying player.
   Queue<CallbackEvent> callback_event_queue_;

   // States of if decoder can accept more inputs.
   bool can_accept_more_audio_data_ = false;
   bool can_accept_more_video_data_ = false;

   // The duration of how far past the current playback position we will write
   // audio samples, in microseconds.
   int64_t audio_write_duration_ = 0;
   int64_t last_written_audio_timestamp_ = 0;

   // Set of received player state updates from the underlying player. This is
   // used to check that the state updates occur in a valid order during normal
   // playback.
   std::set<SbPlayerState> player_state_set_;

   bool destroy_player_called_ = false;
   bool audio_end_of_stream_written_ = false;
   bool video_end_of_stream_written_ = false;
   std::atomic_bool error_occurred_{false};
   int ticket_ = SB_PLAYER_INITIAL_TICKET;
 };

 }  // namespace nplb
 }  // namespace starboard

 #endif  // STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_
	// Copyright 2023 The Cobalt Authors. All Rights Reserved.
	//
	// Licensed under the Apache License, Version 2.0 (the "License");
	// you may not use this file except in compliance with the License.
	// You may obtain a copy of the License at
	//
	// http://www.apache.org/licenses/LICENSE-2.0
	//
	// Unless required by applicable law or agreed to in writing, software
	// distributed under the License is distributed on an "AS IS" BASIS,
	// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	// See the License for the specific language governing permissions and
	// limitations under the License.

	#ifndef STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_
	#define STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_

	#include <atomic>
	#include <deque>
	#include <set>
	#include <string>
	#include <vector>

	#include "starboard/common/queue.h"
	#include "starboard/common/scoped_ptr.h"
	#include "starboard/drm.h"
	#include "starboard/nplb/player_test_util.h"
	#include "starboard/player.h"
	#include "starboard/shared/starboard/player/video_dmp_reader.h"
	#include "starboard/shared/starboard/thread_checker.h"
	#include "starboard/testing/fake_graphics_context_provider.h"

	namespace starboard {
	namespace nplb {

	class SbPlayerTestFixture {
	public:
	// A simple encapsulation of grouped samples.
	class GroupedSamplesIterator;
	class GroupedSamples {
	public:
	struct AudioSamplesDescriptor {
	int start_index = 0;
	int samples_count = 0;
	int64_t timestamp_offset = 0;
	int64_t discarded_duration_from_front = 0;
	int64_t discarded_duration_from_back = 0;
	bool is_end_of_stream = false;
	};

	struct VideoSamplesDescriptor {
	int start_index = 0;
	int samples_count = 0;
	bool is_end_of_stream = false;
	};

	GroupedSamples& AddAudioSamples(int start_index, int number_of_samples);
	GroupedSamples& AddAudioSamples(int start_index,
	int number_of_samples,
	int64_t timestamp_offset,
	int64_t discarded_duration_from_front,
	int64_t discarded_duration_from_back);
	GroupedSamples& AddAudioEOS();
	GroupedSamples& AddVideoSamples(int start_index, int number_of_samples);
	GroupedSamples& AddVideoEOS();

	friend class GroupedSamplesIterator;

	private:
	std::vector<AudioSamplesDescriptor> audio_samples_;
	std::vector<VideoSamplesDescriptor> video_samples_;
	};

	SbPlayerTestFixture(
	const SbPlayerTestConfig& config,
	testing::FakeGraphicsContextProvider* fake_graphics_context_provider);
	~SbPlayerTestFixture();

	void Seek(const int64_t time);
	// Write audio and video samples. It waits for
	// \|kSbPlayerDecoderStateNeedsData\| internally. When writing EOS are
	// requested, the function will write EOS after all samples of the same type
	// are written.
	void Write(const GroupedSamples& grouped_samples);
	// Wait until kSbPlayerStatePresenting received.
	void WaitForPlayerPresenting();
	// Wait until kSbPlayerStateEndOfStream received.
	void WaitForPlayerEndOfStream();
	int64_t GetCurrentMediaTime() const;

	void SetAudioWriteDuration(int64_t duration);

	SbPlayer GetPlayer() const { return player_; }
	bool HasAudio() const { return audio_dmp_reader_; }
	bool HasVideo() const { return video_dmp_reader_; }

	int64_t GetAudioSampleTimestamp(int index) const;
	int ConvertDurationToAudioBufferCount(int64_t duration) const;
	int ConvertDurationToVideoBufferCount(int64_t duration) const;

	private:
	static constexpr int64_t kDefaultWaitForPlayerStateTimeout = 5'000'000LL;
	static constexpr int64_t kDefaultWaitForCallbackEventTimeout = 15'000;

	typedef shared::starboard::player::video_dmp::VideoDmpReader VideoDmpReader;

	typedef enum CallbackEventType {
	kEmptyEvent,
	kDecoderStateEvent,
	kPlayerStateEvent,
	} CallbackEventType;

	struct CallbackEvent {
	CallbackEvent();
	CallbackEvent(SbPlayer player,
	SbMediaType type,
	SbPlayerDecoderState state,
	int ticket);
	CallbackEvent(SbPlayer player, SbPlayerState state, int ticket);

	CallbackEventType event_type;
	SbPlayer player = kSbPlayerInvalid;
	SbMediaType media_type;
	SbPlayerDecoderState decoder_state;
	SbPlayerState player_state;
	int ticket = SB_PLAYER_INITIAL_TICKET;
	};

	static void DecoderStatusCallback(SbPlayer player,
	void* context,
	SbMediaType type,
	SbPlayerDecoderState state,
	int ticket);

	static void PlayerStatusCallback(SbPlayer player,
	void* context,
	SbPlayerState state,
	int ticket);

	static void ErrorCallback(SbPlayer player,
	void* context,
	SbPlayerError error,
	const char* message);

	// Callback methods from the underlying player. Note that callbacks may be
	// invoked on underlying player's thread.
	void OnDecoderState(SbPlayer player,
	SbMediaType media_type,
	SbPlayerDecoderState state,
	int ticket);
	void OnPlayerState(SbPlayer player, SbPlayerState state, int ticket);
	void OnError(SbPlayer player, SbPlayerError error, const char* message);

	void Initialize();
	void TearDown();

	bool CanWriteMoreAudioData();
	bool CanWriteMoreVideoData();

	void WriteAudioSamples(int start_index,
	int samples_to_write,
	int64_t timestamp_offset,
	int64_t discarded_duration_from_front,
	int64_t discarded_duration_from_back);
	void WriteVideoSamples(int start_index, int samples_to_write);
	void WriteEndOfStream(SbMediaType media_type);

	// Checks if there are pending callback events and, if so, logs the received
	// state update in the appropriate tracking container:
	// * \|can_accept_more_*_data_\| for SbPlayerDecoderState updates.
	// * \|player_state_set_\| for SbPlayerState updates.
	// Executes a blocking wait for any new CallbackEvents to be enqueued.
	void WaitAndProcessNextEvent(
	int64_t timeout = kDefaultWaitForCallbackEventTimeout);

	// Waits for \|kSbPlayerDecoderStateNeedsData\| to be sent.
	void WaitForDecoderStateNeedsData(
	const int64_t timeout = kDefaultWaitForCallbackEventTimeout);

	// Waits for desired player state update to be sent.
	void WaitForPlayerState(
	const SbPlayerState desired_state,
	const int64_t timeout = kDefaultWaitForPlayerStateTimeout);

	// When the \|output_mode\| is decoding to texture, then this method is used to
	// advance the decoded frames.
	void GetDecodeTargetWhenSupported();

	// Determine if the the current event is valid based on previously received
	// player state updates, or other inputs to the player.
	void AssertPlayerStateIsValid(SbPlayerState state) const;

	bool HasReceivedPlayerState(SbPlayerState state) const {
	return player_state_set_.find(state) != player_state_set_.end();
	}

	shared::starboard::ThreadChecker thread_checker_;
	const SbPlayerOutputMode output_mode_;
	std::string key_system_;
	std::string max_video_capabilities_;
	scoped_ptr<VideoDmpReader> audio_dmp_reader_;
	scoped_ptr<VideoDmpReader> video_dmp_reader_;
	testing::FakeGraphicsContextProvider* fake_graphics_context_provider_;

	SbPlayer player_ = kSbPlayerInvalid;
	SbDrmSystem drm_system_ = kSbDrmSystemInvalid;

	// Queue of events from the underlying player.
	Queue<CallbackEvent> callback_event_queue_;

	// States of if decoder can accept more inputs.
	bool can_accept_more_audio_data_ = false;
	bool can_accept_more_video_data_ = false;

	// The duration of how far past the current playback position we will write
	// audio samples, in microseconds.
	int64_t audio_write_duration_ = 0;
	int64_t last_written_audio_timestamp_ = 0;

	// Set of received player state updates from the underlying player. This is
	// used to check that the state updates occur in a valid order during normal
	// playback.
	std::set<SbPlayerState> player_state_set_;

	bool destroy_player_called_ = false;
	bool audio_end_of_stream_written_ = false;
	bool video_end_of_stream_written_ = false;
	std::atomic_bool error_occurred_{false};
	int ticket_ = SB_PLAYER_INITIAL_TICKET;
	};

	} // namespace nplb
	} // namespace starboard

	#endif // STARBOARD_NPLB_PLAYER_TEST_FIXTURE_H_