michael@0: /*
michael@0:  *  Copyright 2012 The LibYuv Project Authors. All rights reserved.
michael@0:  *
michael@0:  *  Use of this source code is governed by a BSD-style license
michael@0:  *  that can be found in the LICENSE file in the root of the source
michael@0:  *  tree. An additional intellectual property rights grant can be found
michael@0:  *  in the file PATENTS. All contributing project authors may
michael@0:  *  be found in the AUTHORS file in the root of the source tree.
michael@0:  */
michael@0: 
michael@0: #include <stdlib.h>
michael@0: #include <time.h>
michael@0: 
michael@0: #include "libyuv/cpu_id.h"
michael@0: #include "libyuv/rotate_argb.h"
michael@0: #include "libyuv/row.h"
michael@0: #include "../unit_test/unit_test.h"
michael@0: 
michael@0: namespace libyuv {
michael@0: 
michael@0: void TestRotateBpp(int src_width, int src_height,
michael@0:                    int dst_width, int dst_height,
michael@0:                    libyuv::RotationMode mode,
michael@0:                    int benchmark_iterations,
michael@0:                    const int kBpp) {
michael@0:   if (src_width < 1) {
michael@0:     src_width = 1;
michael@0:   }
michael@0:   if (src_height < 1) {
michael@0:     src_height = 1;
michael@0:   }
michael@0:   if (dst_width < 1) {
michael@0:     dst_width = 1;
michael@0:   }
michael@0:   if (dst_height < 1) {
michael@0:     dst_height = 1;
michael@0:   }
michael@0:   int src_stride_argb = src_width * kBpp;
michael@0:   int src_argb_plane_size = src_stride_argb * src_height;
michael@0:   align_buffer_64(src_argb, src_argb_plane_size);
michael@0:   for (int i = 0; i < src_argb_plane_size; ++i) {
michael@0:     src_argb[i] = random() & 0xff;
michael@0:   }
michael@0: 
michael@0:   int dst_stride_argb = dst_width * kBpp;
michael@0:   int dst_argb_plane_size = dst_stride_argb * dst_height;
michael@0:   align_buffer_64(dst_argb_c, dst_argb_plane_size);
michael@0:   align_buffer_64(dst_argb_opt, dst_argb_plane_size);
michael@0:   memset(dst_argb_c, 2, dst_argb_plane_size);
michael@0:   memset(dst_argb_opt, 3, dst_argb_plane_size);
michael@0: 
michael@0:   if (kBpp == 1) {
michael@0:     MaskCpuFlags(0);  // Disable all CPU optimization.
michael@0:     RotatePlane(src_argb, src_stride_argb,
michael@0:                 dst_argb_c, dst_stride_argb,
michael@0:                 src_width, src_height, mode);
michael@0: 
michael@0:     MaskCpuFlags(-1);  // Enable all CPU optimization.
michael@0:     for (int i = 0; i < benchmark_iterations; ++i) {
michael@0:       RotatePlane(src_argb, src_stride_argb,
michael@0:                   dst_argb_opt, dst_stride_argb,
michael@0:                   src_width, src_height, mode);
michael@0:     }
michael@0:   } else if (kBpp == 4) {
michael@0:     MaskCpuFlags(0);  // Disable all CPU optimization.
michael@0:     ARGBRotate(src_argb, src_stride_argb,
michael@0:                dst_argb_c, dst_stride_argb,
michael@0:                src_width, src_height, mode);
michael@0: 
michael@0:     MaskCpuFlags(-1);  // Enable all CPU optimization.
michael@0:     for (int i = 0; i < benchmark_iterations; ++i) {
michael@0:       ARGBRotate(src_argb, src_stride_argb,
michael@0:                  dst_argb_opt, dst_stride_argb,
michael@0:                  src_width, src_height, mode);
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   // Rotation should be exact.
michael@0:   for (int i = 0; i < dst_argb_plane_size; ++i) {
michael@0:     EXPECT_EQ(dst_argb_c[i], dst_argb_opt[i]);
michael@0:   }
michael@0: 
michael@0:   free_aligned_buffer_64(dst_argb_c);
michael@0:   free_aligned_buffer_64(dst_argb_opt);
michael@0:   free_aligned_buffer_64(src_argb);
michael@0: }
michael@0: 
michael@0: static void ARGBTestRotate(int src_width, int src_height,
michael@0:                            int dst_width, int dst_height,
michael@0:                            libyuv::RotationMode mode,
michael@0:                            int benchmark_iterations) {
michael@0:   TestRotateBpp(src_width, src_height,
michael@0:                 dst_width, dst_height,
michael@0:                 mode, benchmark_iterations, 4);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate0) {
michael@0:   ARGBTestRotate(benchmark_width_, benchmark_height_,
michael@0:                  benchmark_width_, benchmark_height_,
michael@0:                  kRotate0, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate90) {
michael@0:   ARGBTestRotate(benchmark_width_, benchmark_height_,
michael@0:                  benchmark_height_, benchmark_width_,
michael@0:                  kRotate90, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate180) {
michael@0:   ARGBTestRotate(benchmark_width_, benchmark_height_,
michael@0:                  benchmark_width_, benchmark_height_,
michael@0:                  kRotate180, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate270) {
michael@0:   ARGBTestRotate(benchmark_width_, benchmark_height_,
michael@0:                  benchmark_height_, benchmark_width_,
michael@0:                  kRotate270, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate0_Odd) {
michael@0:   ARGBTestRotate(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  kRotate0, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate90_Odd) {
michael@0:   ARGBTestRotate(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  benchmark_height_ - 1, benchmark_width_ - 3,
michael@0:                  kRotate90, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate180_Odd) {
michael@0:   ARGBTestRotate(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  kRotate180, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, ARGBRotate270_Odd) {
michael@0:   ARGBTestRotate(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                  benchmark_height_ - 1, benchmark_width_ - 3,
michael@0:                  kRotate270, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: static void TestRotatePlane(int src_width, int src_height,
michael@0:                             int dst_width, int dst_height,
michael@0:                             libyuv::RotationMode mode,
michael@0:                             int benchmark_iterations) {
michael@0:   TestRotateBpp(src_width, src_height,
michael@0:                 dst_width, dst_height,
michael@0:                 mode, benchmark_iterations, 1);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane0) {
michael@0:   TestRotatePlane(benchmark_width_, benchmark_height_,
michael@0:                   benchmark_width_, benchmark_height_,
michael@0:                   kRotate0, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane90) {
michael@0:   TestRotatePlane(benchmark_width_, benchmark_height_,
michael@0:                   benchmark_height_, benchmark_width_,
michael@0:                   kRotate90, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane180) {
michael@0:   TestRotatePlane(benchmark_width_, benchmark_height_,
michael@0:                   benchmark_width_, benchmark_height_,
michael@0:                   kRotate180, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane270) {
michael@0:   TestRotatePlane(benchmark_width_, benchmark_height_,
michael@0:                   benchmark_height_, benchmark_width_,
michael@0:                   kRotate270, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane0_Odd) {
michael@0:   TestRotatePlane(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   kRotate0, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane90_Odd) {
michael@0:   TestRotatePlane(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   benchmark_height_ - 1, benchmark_width_ - 3,
michael@0:                   kRotate90, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane180_Odd) {
michael@0:   TestRotatePlane(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   kRotate180, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: TEST_F(libyuvTest, RotatePlane270_Odd) {
michael@0:   TestRotatePlane(benchmark_width_ - 3, benchmark_height_ - 1,
michael@0:                   benchmark_height_ - 1, benchmark_width_ - 3,
michael@0:                   kRotate270, benchmark_iterations_);
michael@0: }
michael@0: 
michael@0: }  // namespace libyuv