VEA unittest: add a test case for cache line-alignment

media/gpu/video_encode_accelerator_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <inttypes.h>
 #include <stddef.h>
 #include <stdint.h>
 
 #include <algorithm>
 #include <memory>
@@ skipping 185 matching lines @@
   }
 
   void deallocate(pointer p, size_type n) {
     base::AlignedFree(static_cast<void*>(p));
   }
 
   size_type max_size() const {
     return std::numeric_limits<size_t>::max() / sizeof(T);
   }
 };
+typedef std::vector<char, AlignedAllocator<char, kPlatformBufferAlignment>>
+    AlignedCharVector;
 
 struct TestStream {
   TestStream()
       : num_frames(0),
         aligned_buffer_size(0),
         requested_bitrate(0),
         requested_framerate(0),
         requested_subsequent_bitrate(0),
         requested_subsequent_framerate(0) {}
   ~TestStream() {}
 
   gfx::Size visible_size;
   gfx::Size coded_size;
   unsigned int num_frames;
 
   // Original unaligned input file name provided as an argument to the test.
   // And the file must be an I420 (YUV planar) raw stream.
   std::string in_filename;
 
   // A vector used to prepare aligned input buffers of |in_filename|. This
   // makes sure starting addresses of YUV planes are aligned to
   // kPlatformBufferAlignment bytes.
-  std::vector<char, AlignedAllocator<char, kPlatformBufferAlignment>>
-      aligned_in_file_data;
+  AlignedCharVector aligned_in_file_data;
 
   // Byte size of a frame of |aligned_in_file_data|.
   size_t aligned_buffer_size;
 
   // Byte size for each aligned plane of a frame.
   std::vector<size_t> aligned_plane_size;
 
   std::string out_filename;
   VideoCodecProfile requested_profile;
   unsigned int requested_bitrate;
@@ skipping 627 matching lines @@
       }
     }
   }
 
   // Divide the difference by the size of frame.
   difference /= VideoFrame::AllocationSize(kInputFormat, visible_size);
   EXPECT_TRUE(difference <= kDecodeSimilarityThreshold)
       << "difference = " << difference << "  > decode similarity threshold";
 }
 
-class VEAClient : public VideoEncodeAccelerator::Client {
+// Base class for all VEA Clients in this file
+class VEAClientBase : public VideoEncodeAccelerator::Client {
+ public:
+  ~VEAClientBase() {
+    LOG(INFO) << "VEAClientBase destructor";

[wuchengli, 2016/11/30 04:23:40]

This log is not very useful. Remove.

[hywu1, 2016/11/30 07:06:24]

Done.

+    LOG_ASSERT(!has_encoder());
+  }
+  void NotifyError(VideoEncodeAccelerator::Error error) override {
+    DCHECK(thread_checker_.CalledOnValidThread());
+    SetState(CS_ERROR);
+  }
+
+ protected:
+  VEAClientBase(ClientStateNotification<ClientState>* note)
+      : note_(note), next_output_buffer_id_(0) {}
+
+  bool has_encoder() { return encoder_.get(); }
+
+  virtual void SetState(ClientState new_state) = 0;
+
+  std::unique_ptr<VideoEncodeAccelerator> encoder_;
+
+  // Used to notify another thread about the state. VEAClientBase does not own
+  // this.
+  ClientStateNotification<ClientState>* note_;
+
+  // All methods of this class should be run on the same thread.
+  base::ThreadChecker thread_checker_;
+
+  ScopedVector<base::SharedMemory> output_shms_;
+  int32_t next_output_buffer_id_;
+};
+
+class VEAClient : public VEAClientBase {
  public:
   VEAClient(TestStream* test_stream,
             ClientStateNotification<ClientState>* note,
             bool save_to_file,
             unsigned int keyframe_period,
             bool force_bitrate,
             bool test_perf,
             bool mid_stream_bitrate_switch,
             bool mid_stream_framerate_switch,
             bool verify_output,
             bool verify_output_timestamp);
-  ~VEAClient() override;
   void CreateEncoder();
   void DestroyEncoder();
 
   void TryToSetupEncodeOnSeperateThread();
   void DestroyEncodeOnSeperateThread();
 
   // VideoDecodeAccelerator::Client implementation.
   void RequireBitstreamBuffers(unsigned int input_count,
                                const gfx::Size& input_coded_size,
                                size_t output_buffer_size) override;
   void BitstreamBufferReady(int32_t bitstream_buffer_id,
                             size_t payload_size,
                             bool key_frame,
                             base::TimeDelta timestamp) override;
-  void NotifyError(VideoEncodeAccelerator::Error error) override;
 
  private:
   void BitstreamBufferReadyOnMainThread(int32_t bitstream_buffer_id,
                                         size_t payload_size,
                                         bool key_frame,
                                         base::TimeDelta timestamp);
-  bool has_encoder() { return encoder_.get(); }
 
   // Return the number of encoded frames per second.
   double frames_per_second();
 
-  void SetState(ClientState new_state);
+  void SetState(ClientState new_state) override;
 
   // Set current stream parameters to given |bitrate| at |framerate|.
   void SetStreamParameters(unsigned int bitrate, unsigned int framerate);
 
   // Called when encoder is done with a VideoFrame.
   void InputNoLongerNeededCallback(int32_t input_id);
 
   // Feed the encoder with one input frame.
   void FeedEncoderWithOneInput();
 
@@ skipping 44 matching lines @@
   // Called when the quality validator has decoded all the frames.
   void DecodeCompleted();
 
   // Called when the quality validator fails to decode a frame.
   void DecodeFailed();
 
   // Verify that the output timestamp matches input timestamp.
   void VerifyOutputTimestamp(base::TimeDelta timestamp);
 
   ClientState state_;
-  std::unique_ptr<VideoEncodeAccelerator> encoder_;
 
   TestStream* test_stream_;
 
-  // Used to notify another thread about the state. VEAClient does not own this.
-  ClientStateNotification<ClientState>* note_;
-
   // Ids assigned to VideoFrames.
   std::set<int32_t> inputs_at_client_;
   int32_t next_input_id_;
 
   // Encode start time of all encoded frames. The position in the vector is the
   // frame input id.
   std::vector<base::TimeTicks> encode_start_time_;
   // The encode latencies of all encoded frames. We define encode latency as the
   // time delay from input of each VideoFrame (VEA::Encode()) to output of the
   // corresponding BitstreamBuffer (VEA::Client::BitstreamBufferReady()).
   std::vector<base::TimeDelta> encode_latencies_;
 
   // Ids for output BitstreamBuffers.
   typedef std::map<int32_t, base::SharedMemory*> IdToSHM;
-  ScopedVector<base::SharedMemory> output_shms_;
   IdToSHM output_buffers_at_client_;
-  int32_t next_output_buffer_id_;
 
   // Current offset into input stream.
   off_t pos_in_input_stream_;
   gfx::Size input_coded_size_;
   // Requested by encoder.
   unsigned int num_required_input_buffers_;
   size_t output_buffer_size_;
 
   // Number of frames to encode. This may differ from the number of frames in
   // stream if we need more frames for bitrate tests.
@@ skipping 47 matching lines @@
 
   // Used to validate the encoded frame quality.
   std::unique_ptr<VideoFrameQualityValidator> quality_validator_;
 
   // The time when the first frame is submitted for encode.
   base::TimeTicks first_frame_start_time_;
 
   // The time when the last encoded frame is ready.
   base::TimeTicks last_frame_ready_time_;
 
-  // All methods of this class should be run on the same thread.
-  base::ThreadChecker thread_checker_;
-
   // Requested bitrate in bits per second.
   unsigned int requested_bitrate_;
 
   // Requested initial framerate.
   unsigned int requested_framerate_;
 
   // Bitrate to switch to in the middle of the stream.
   unsigned int requested_subsequent_bitrate_;
 
   // Framerate to switch to in the middle of the stream.
@@ skipping 30 matching lines @@
 VEAClient::VEAClient(TestStream* test_stream,
                      ClientStateNotification<ClientState>* note,
                      bool save_to_file,
                      unsigned int keyframe_period,
                      bool force_bitrate,
                      bool test_perf,
                      bool mid_stream_bitrate_switch,
                      bool mid_stream_framerate_switch,
                      bool verify_output,
                      bool verify_output_timestamp)
-    : state_(CS_CREATED),
+    : VEAClientBase(note),
+      state_(CS_CREATED),
       test_stream_(test_stream),
-      note_(note),
       next_input_id_(0),
-      next_output_buffer_id_(0),
       pos_in_input_stream_(0),
       num_required_input_buffers_(0),
       output_buffer_size_(0),
       num_frames_to_encode_(0),
       num_encoded_frames_(0),
       num_frames_since_last_check_(0),
       seen_keyframe_in_this_buffer_(false),
       save_to_file_(save_to_file),
       keyframe_period_(keyframe_period),
       num_keyframes_requested_(0),
@@ skipping 33 matching lines @@
     // Without it, AppendToFile() will not work.
     EXPECT_EQ(0, base::WriteFile(out_filename, NULL, 0));
   }
 
   // Initialize the parameters of the test streams.
   UpdateTestStreamData(mid_stream_bitrate_switch, mid_stream_framerate_switch);
 
   thread_checker_.DetachFromThread();
 }
 
-VEAClient::~VEAClient() {
-  LOG_ASSERT(!has_encoder());
-}
-
 void VEAClient::CreateEncoder() {
   DCHECK(thread_checker_.CalledOnValidThread());
   LOG_ASSERT(!has_encoder());
 
   main_thread_task_runner_ = base::ThreadTaskRunnerHandle::Get();
   encode_task_runner_ = main_thread_task_runner_;
 
   std::unique_ptr<VideoEncodeAccelerator> encoders[] = {
       CreateFakeVEA(), CreateV4L2VEA(), CreateVaapiVEA(), CreateVTVEA(),
       CreateMFVEA()};
@@ skipping 200 matching lines @@
                                      size_t payload_size,
                                      bool key_frame,
                                      base::TimeDelta timestamp) {
   ASSERT_TRUE(encode_task_runner_->BelongsToCurrentThread());
   main_thread_task_runner_->PostTask(
       FROM_HERE, base::Bind(&VEAClient::BitstreamBufferReadyOnMainThread,
                             base::Unretained(this), bitstream_buffer_id,
                             payload_size, key_frame, timestamp));
 }
 
-void VEAClient::NotifyError(VideoEncodeAccelerator::Error error) {
-  DCHECK(thread_checker_.CalledOnValidThread());
-  SetState(CS_ERROR);
-}
-
 void VEAClient::BitstreamBufferReadyOnMainThread(int32_t bitstream_buffer_id,
                                                  size_t payload_size,
                                                  bool key_frame,
                                                  base::TimeDelta timestamp) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
   ASSERT_LE(payload_size, output_buffer_size_);
 
   IdToSHM::iterator it = output_buffers_at_client_.find(bitstream_buffer_id);
   ASSERT_NE(it, output_buffers_at_client_.end());
@@ skipping 330 matching lines @@
   IvfFrameHeader header = {};
 
   header.frame_size = static_cast<uint32_t>(frame_size);
   header.timestamp = frame_index;
   header.ByteSwap();
   EXPECT_TRUE(base::AppendToFile(
       base::FilePath::FromUTF8Unsafe(test_stream_->out_filename),
       reinterpret_cast<char*>(&header), sizeof(header)));
 }
 
-// This client is only used to make sure the encoder does not return an encoded
-// frame before getting any input.
-class VEANoInputClient : public VideoEncodeAccelerator::Client {
+// Base class for simple VEA Clients
+class SimpleVEAClientBase : public VEAClientBase {
  public:
-  explicit VEANoInputClient(ClientStateNotification<ClientState>* note);
-  ~VEANoInputClient() override;
   void CreateEncoder();
   void DestroyEncoder();
 
   // VideoDecodeAccelerator::Client implementation.
   void RequireBitstreamBuffers(unsigned int input_count,
                                const gfx::Size& input_coded_size,
                                size_t output_buffer_size) override;
-  void BitstreamBufferReady(int32_t bitstream_buffer_id,
-                            size_t payload_size,
-                            bool key_frame,
-                            base::TimeDelta timestamp) override;
-  void NotifyError(VideoEncodeAccelerator::Error error) override;
 
- private:
-  bool has_encoder() { return encoder_.get(); }
+ protected:
+  SimpleVEAClientBase(ClientStateNotification<ClientState>* note,
+                      const int width,
+                      const int height);
 
-  void SetState(ClientState new_state);
+  void SetState(ClientState new_state) override;
 
   // Provide the encoder with a new output buffer.
   void FeedEncoderWithOutput(base::SharedMemory* shm, size_t output_size);
 
-  std::unique_ptr<VideoEncodeAccelerator> encoder_;
-
-  // Used to notify another thread about the state. VEAClient does not own this.
-  ClientStateNotification<ClientState>* note_;
-
-  // All methods of this class should be run on the same thread.
-  base::ThreadChecker thread_checker_;
-
-  // Ids for output BitstreamBuffers.
-  ScopedVector<base::SharedMemory> output_shms_;
-  int32_t next_output_buffer_id_;
-
-  // The timer used to monitor the encoder doesn't return an output buffer in
-  // a period of time.
-  std::unique_ptr<base::Timer> timer_;
+  const int width_;
+  const int height_;
+  const int bitrate_;
+  const int fps_;
 };
 
-VEANoInputClient::VEANoInputClient(ClientStateNotification<ClientState>* note)
-    : note_(note), next_output_buffer_id_(0) {
+SimpleVEAClientBase::SimpleVEAClientBase(
+    ClientStateNotification<ClientState>* note,
+    const int width,
+    const int height)
+    : VEAClientBase(note),
+      width_(width),
+      height_(height),
+      bitrate_(200000),
+      fps_(30) {
   thread_checker_.DetachFromThread();
 }
 
-VEANoInputClient::~VEANoInputClient() {
-  LOG_ASSERT(!has_encoder());
-}
-
-void VEANoInputClient::CreateEncoder() {
+void SimpleVEAClientBase::CreateEncoder() {
   DCHECK(thread_checker_.CalledOnValidThread());
   LOG_ASSERT(!has_encoder());
   LOG_ASSERT(g_env->test_streams_.size());
 
-  const int kDefaultWidth = 320;
-  const int kDefaultHeight = 240;
-  const int kDefaultBitrate = 200000;
-  const int kDefaultFps = 30;
-
   std::unique_ptr<VideoEncodeAccelerator> encoders[] = {
       CreateFakeVEA(), CreateV4L2VEA(), CreateVaapiVEA(), CreateVTVEA(),
       CreateMFVEA()};
 
-  // The test case is no input. Use default visible size, bitrate, and fps.
-  gfx::Size visible_size(kDefaultWidth, kDefaultHeight);
+  gfx::Size visible_size(width_, height_);
   for (auto& encoder : encoders) {
     if (!encoder)
       continue;
     encoder_ = std::move(encoder);
     if (encoder_->Initialize(kInputFormat, visible_size,
                              g_env->test_streams_[0]->requested_profile,
-                             kDefaultBitrate, this)) {
-      encoder_->RequestEncodingParametersChange(kDefaultBitrate, kDefaultFps);
+                             bitrate_, this)) {
+      encoder_->RequestEncodingParametersChange(bitrate_, fps_);
       SetState(CS_INITIALIZED);
       return;
     }
   }
   encoder_.reset();
   LOG(ERROR) << "VideoEncodeAccelerator::Initialize() failed";
   SetState(CS_ERROR);
 }
 
-void VEANoInputClient::DestroyEncoder() {
+void SimpleVEAClientBase::DestroyEncoder() {
   DCHECK(thread_checker_.CalledOnValidThread());
   if (!has_encoder())
     return;
   // Clear the objects that should be destroyed on the same thread as creation.
   encoder_.reset();
-  timer_.reset();
 }
 
-void VEANoInputClient::RequireBitstreamBuffers(
+void SimpleVEAClientBase::SetState(ClientState new_state) {
+  DVLOG(4) << "Changing state to " << new_state;
+  note_->Notify(new_state);
+}
+
+void SimpleVEAClientBase::RequireBitstreamBuffers(
     unsigned int input_count,
     const gfx::Size& input_coded_size,
     size_t output_size) {
   DCHECK(thread_checker_.CalledOnValidThread());
   SetState(CS_ENCODING);
   ASSERT_GT(output_size, 0UL);
 
   for (unsigned int i = 0; i < kNumOutputBuffers; ++i) {
     base::SharedMemory* shm = new base::SharedMemory();
     LOG_ASSERT(shm->CreateAndMapAnonymous(output_size));
     output_shms_.push_back(shm);
     FeedEncoderWithOutput(shm, output_size);
   }
+}
+
+void SimpleVEAClientBase::FeedEncoderWithOutput(base::SharedMemory* shm,
+                                                size_t output_size) {
+  if (!has_encoder())
+    return;
+
+  base::SharedMemoryHandle dup_handle;
+  LOG_ASSERT(shm->ShareToProcess(base::GetCurrentProcessHandle(), &dup_handle));
+
+  BitstreamBuffer bitstream_buffer(next_output_buffer_id_++, dup_handle,
+                                   output_size);
+  encoder_->UseOutputBitstreamBuffer(bitstream_buffer);
+}
+
+// This client is only used to make sure the encoder does not return an encoded
+// frame before getting any input.
+class VEANoInputClient : public SimpleVEAClientBase {
+ public:
+  explicit VEANoInputClient(ClientStateNotification<ClientState>* note);
+  void DestroyEncoder();
+
+  // VideoDecodeAccelerator::Client implementation.
+  void RequireBitstreamBuffers(unsigned int input_count,
+                               const gfx::Size& input_coded_size,
+                               size_t output_buffer_size) override;
+  void BitstreamBufferReady(int32_t bitstream_buffer_id,
+                            size_t payload_size,
+                            bool key_frame,
+                            base::TimeDelta timestamp) override;
+
+ private:
+  // The timer used to monitor the encoder doesn't return an output buffer in
+  // a period of time.
+  std::unique_ptr<base::Timer> timer_;
+};
+
+VEANoInputClient::VEANoInputClient(ClientStateNotification<ClientState>* note)
+    : SimpleVEAClientBase(note, 320, 240) {}
+
+void VEANoInputClient::DestroyEncoder() {
+  SimpleVEAClientBase::DestroyEncoder();
+  // Clear the objects that should be destroyed on the same thread as creation.
+  timer_.reset();
+}
+
+void VEANoInputClient::RequireBitstreamBuffers(
+    unsigned int input_count,
+    const gfx::Size& input_coded_size,
+    size_t output_size) {
+  SimpleVEAClientBase::RequireBitstreamBuffers(input_count, input_coded_size,
+                                               output_size);
+
   // Timer is used to make sure there is no output frame in 100ms.
   timer_.reset(new base::Timer(FROM_HERE,
                                base::TimeDelta::FromMilliseconds(100),
                                base::Bind(&VEANoInputClient::SetState,
                                           base::Unretained(this), CS_FINISHED),
                                false));
   timer_->Reset();
 }
 
 void VEANoInputClient::BitstreamBufferReady(int32_t bitstream_buffer_id,
                                             size_t payload_size,
                                             bool key_frame,
                                             base::TimeDelta timestamp) {
   DCHECK(thread_checker_.CalledOnValidThread());
   SetState(CS_ERROR);
 }
 
-void VEANoInputClient::NotifyError(VideoEncodeAccelerator::Error error) {
-  DCHECK(thread_checker_.CalledOnValidThread());
-  SetState(CS_ERROR);
+// This client is only used to test input frame with the size of U and V planes
+// unaligned to cache line.
+// To have both width and height divisible by 16 but not 32 will make the size
+// of U/V plane (width * height / 4) unaligned to 128-byte cache line.
+class VEACacheLineUnalignedInputClient : public SimpleVEAClientBase {
+ public:
+  explicit VEACacheLineUnalignedInputClient(
+      ClientStateNotification<ClientState>* note);
+
+  // VideoDecodeAccelerator::Client implementation.
+  void RequireBitstreamBuffers(unsigned int input_count,
+                               const gfx::Size& input_coded_size,
+                               size_t output_buffer_size) override;
+  void BitstreamBufferReady(int32_t bitstream_buffer_id,
+                            size_t payload_size,
+                            bool key_frame,
+                            base::TimeDelta timestamp) override;
+
+ private:
+  // Feed the encoder with one input frame.
+  void FeedEncoderWithOneInput(const gfx::Size& input_coded_size);
+};
+
+VEACacheLineUnalignedInputClient::VEACacheLineUnalignedInputClient(
+    ClientStateNotification<ClientState>* note)
+    : SimpleVEAClientBase(note, 368, 368) {
+}  // 368 is divisible by 16 but not 32
+
+void VEACacheLineUnalignedInputClient::RequireBitstreamBuffers(
+    unsigned int input_count,
+    const gfx::Size& input_coded_size,
+    size_t output_size) {
+  SimpleVEAClientBase::RequireBitstreamBuffers(input_count, input_coded_size,
+                                               output_size);
+
+  FeedEncoderWithOneInput(input_coded_size);
 }
 
-void VEANoInputClient::SetState(ClientState new_state) {
-  DVLOG(4) << "Changing state to " << new_state;
-  note_->Notify(new_state);
+void VEACacheLineUnalignedInputClient::BitstreamBufferReady(
+    int32_t bitstream_buffer_id,
+    size_t payload_size,
+    bool key_frame,
+    base::TimeDelta timestamp) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  // It's enough to encode just one frame. If plane size is not aligned,
+  // VideoEncodeAccelerator::Encode will fail.
+  SetState(CS_FINISHED);
 }
 
-void VEANoInputClient::FeedEncoderWithOutput(base::SharedMemory* shm,
-                                             size_t output_size) {
+void VEACacheLineUnalignedInputClient::FeedEncoderWithOneInput(
+    const gfx::Size& input_coded_size) {
   if (!has_encoder())
     return;
 
-  base::SharedMemoryHandle dup_handle;
-  LOG_ASSERT(shm->ShareToProcess(base::GetCurrentProcessHandle(), &dup_handle));
+  AlignedCharVector aligned_plane[] = {
+      AlignedCharVector(
+          VideoFrame::PlaneSize(kInputFormat, 0, input_coded_size).GetArea()),
+      AlignedCharVector(
+          VideoFrame::PlaneSize(kInputFormat, 1, input_coded_size).GetArea()),
+      AlignedCharVector(
+          VideoFrame::PlaneSize(kInputFormat, 2, input_coded_size).GetArea())};
+  uint8_t* frame_data_y = reinterpret_cast<uint8_t*>(&aligned_plane[0][0]);
+  uint8_t* frame_data_u = reinterpret_cast<uint8_t*>(&aligned_plane[1][0]);
+  uint8_t* frame_data_v = reinterpret_cast<uint8_t*>(&aligned_plane[2][0]);
 
-  BitstreamBuffer bitstream_buffer(next_output_buffer_id_++, dup_handle,
-                                   output_size);
-  encoder_->UseOutputBitstreamBuffer(bitstream_buffer);
+  scoped_refptr<VideoFrame> video_frame = VideoFrame::WrapExternalYuvData(
+      kInputFormat, input_coded_size, gfx::Rect(input_coded_size),
+      input_coded_size, input_coded_size.width(), input_coded_size.width() / 2,
+      input_coded_size.width() / 2, frame_data_y, frame_data_u, frame_data_v,
+      base::TimeDelta().FromMilliseconds(base::Time::kMillisecondsPerSecond /
+                                         fps_));
+
+  encoder_->Encode(video_frame, false);
 }
 
 // Test parameters:
 // - Number of concurrent encoders. The value takes effect when there is only
 //   one input stream; otherwise, one encoder per input stream will be
 //   instantiated.
 // - If true, save output to file (provided an output filename was supplied).
 // - Force a keyframe every n frames.
 // - Force bitrate; the actual required value is provided as a property
 //   of the input stream, because it depends on stream type/resolution/etc.
@@ skipping 65 matching lines @@
   for (size_t i = 0; i < num_concurrent_encoders; ++i) {
     encoder_thread.task_runner()->PostTask(
         FROM_HERE,
         base::Bind(&VEAClient::DestroyEncoder, base::Unretained(clients[i])));
   }
 
   // This ensures all tasks have finished.
   encoder_thread.Stop();
 }
 
+// Test parameters:
+// - Test type
+//   0: No input test
+//   1: Cache line-unaligned test
+class VideoEncodeAcceleratorSimpleTest : public ::testing::TestWithParam<int> {
+};
+
+template <class TestClient>
+void SimpleTestFunc() {
+  std::unique_ptr<ClientStateNotification<ClientState>> note(
+      new ClientStateNotification<ClientState>());
+  std::unique_ptr<TestClient> client(new TestClient(note.get()));
+  base::Thread encoder_thread("EncoderThread");
+  ASSERT_TRUE(encoder_thread.Start());
+
+  encoder_thread.task_runner()->PostTask(
+      FROM_HERE,
+      base::Bind(&TestClient::CreateEncoder, base::Unretained(client.get())));
+
+  // Encoder must pass through states in this order.
+  enum ClientState state_transitions[] = {CS_INITIALIZED, CS_ENCODING,
+                                          CS_FINISHED};
+
+  for (const auto& state : state_transitions) {
+    EXPECT_EQ(state, note->Wait());
+  }
+
+  encoder_thread.task_runner()->PostTask(
+      FROM_HERE,
+      base::Bind(&TestClient::DestroyEncoder, base::Unretained(client.get())));
+
+  // This ensures all tasks have finished.
+  encoder_thread.Stop();
+}
+
+TEST_P(VideoEncodeAcceleratorSimpleTest, TestSimpleEncode) {
+  const int test_type = GetParam();
+  ASSERT_GT(2, test_type) << "Invalid test type=" << test_type;

[wuchengli, 2016/11/30 04:23:41]

s/ASSERT_GT/ASSERT_GE/

[hywu1, 2016/11/30 07:06:24]

Changed to ASSERT_LT

+
+  if (test_type == 0)
+    SimpleTestFunc<VEANoInputClient>();
+  else if (test_type == 1)
+    SimpleTestFunc<VEACacheLineUnalignedInputClient>();
+}
+
 #if defined(OS_CHROMEOS)
 INSTANTIATE_TEST_CASE_P(
     SimpleEncode,
     VideoEncodeAcceleratorTest,
     ::testing::Values(
         std::make_tuple(1, true, 0, false, false, false, false, false, false),
         std::make_tuple(1, true, 0, false, false, false, false, true, false)));
 
 INSTANTIATE_TEST_CASE_P(
     EncoderPerf,
@@ skipping 38 matching lines @@
         std::make_tuple(3, false, 0, false, false, false, false, false, false),
         std::make_tuple(3, false, 0, true, false, false, true, false, false),
         std::make_tuple(3, false, 0, true, false, true, false, false, false)));
 
 INSTANTIATE_TEST_CASE_P(
     VerifyTimestamp,
     VideoEncodeAcceleratorTest,
     ::testing::Values(
         std::make_tuple(1, false, 0, false, false, false, false, false, true)));
 
-TEST(VEANoInputTest, CheckOutput) {
-  std::unique_ptr<ClientStateNotification<ClientState>> note(
-      new ClientStateNotification<ClientState>());
-  std::unique_ptr<VEANoInputClient> client(new VEANoInputClient(note.get()));
-  base::Thread encoder_thread("EncoderThread");
-  ASSERT_TRUE(encoder_thread.Start());
+INSTANTIATE_TEST_CASE_P(NoInputTest,
+                        VideoEncodeAcceleratorSimpleTest,
+                        ::testing::Values(0));
 
-  encoder_thread.task_runner()->PostTask(
-      FROM_HERE, base::Bind(&VEANoInputClient::CreateEncoder,
-                            base::Unretained(client.get())));
-
-  // Encoder must pass through states in this order.
-  enum ClientState state_transitions[] = {CS_INITIALIZED, CS_ENCODING,
-                                          CS_FINISHED};
-
-  for (const auto& state : state_transitions) {
-    ASSERT_EQ(state, note->Wait());
-  }
-
-  encoder_thread.task_runner()->PostTask(
-      FROM_HERE, base::Bind(&VEANoInputClient::DestroyEncoder,
-                            base::Unretained(client.get())));
-
-  // This ensures all tasks have finished.
-  encoder_thread.Stop();
-}
+INSTANTIATE_TEST_CASE_P(CacheLineUnalignedInputTest,
+                        VideoEncodeAcceleratorSimpleTest,
+                        ::testing::Values(1));
 
 #elif defined(OS_MACOSX) || defined(OS_WIN)
 INSTANTIATE_TEST_CASE_P(
     SimpleEncode,
     VideoEncodeAcceleratorTest,
     ::testing::Values(
         std::make_tuple(1, true, 0, false, false, false, false, false, false),
         std::make_tuple(1, true, 0, false, false, false, false, true, false)));
 
 INSTANTIATE_TEST_CASE_P(
@@ skipping 111 matching lines @@
 
   media::g_env =
       reinterpret_cast<media::VideoEncodeAcceleratorTestEnvironment*>(
           testing::AddGlobalTestEnvironment(
               new media::VideoEncodeAcceleratorTestEnvironment(
                   std::move(test_stream_data), log_path, run_at_fps,
                   needs_encode_latency, verify_all_output)));
 
   return RUN_ALL_TESTS();
 }