Source code
Revision control
Copy as Markdown
Other Tools
/*
* Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "modules/desktop_capture/desktop_frame_rotation.h"
#include <stdint.h>
#include "modules/desktop_capture/desktop_frame.h"
#include "modules/desktop_capture/desktop_region.h"
#include "modules/desktop_capture/test_utils.h"
#include "test/gtest.h"
namespace webrtc {
namespace {
// A DesktopFrame implementation which stores data in an external int array.
class ArrayDesktopFrame : public DesktopFrame {
public:
ArrayDesktopFrame(DesktopSize size, uint32_t* data);
~ArrayDesktopFrame() override;
};
ArrayDesktopFrame::ArrayDesktopFrame(DesktopSize size, uint32_t* data)
: DesktopFrame(size,
size.width() * kBytesPerPixel,
reinterpret_cast<uint8_t*>(data),
nullptr) {}
ArrayDesktopFrame::~ArrayDesktopFrame() = default;
} // namespace
TEST(DesktopFrameRotationTest, CopyRect3x4) {
// A DesktopFrame of 4-pixel width by 3-pixel height.
static uint32_t frame_pixels[] = {
0, 1, 2, 3, //
4, 5, 6, 7, //
8, 9, 10, 11, //
};
ArrayDesktopFrame frame(DesktopSize(4, 3), frame_pixels);
{
BasicDesktopFrame target(DesktopSize(4, 3));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_0, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(frame, target));
}
// After Rotating clock-wise 90 degree
{
static uint32_t expected_pixels[] = {
8, 4, 0, //
9, 5, 1, //
10, 6, 2, //
11, 7, 3, //
};
ArrayDesktopFrame expected(DesktopSize(3, 4), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 4));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 180 degree
{
static uint32_t expected_pixels[] = {
11, 10, 9, 8, //
7, 6, 5, 4, //
3, 2, 1, 0, //
};
ArrayDesktopFrame expected(DesktopSize(4, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(4, 3));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_180, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 270 degree
{
static uint32_t expected_pixels[] = {
3, 7, 11, //
2, 6, 10, //
1, 5, 9, //
0, 4, 8, //
};
ArrayDesktopFrame expected(DesktopSize(3, 4), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 4));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
}
TEST(DesktopFrameRotationTest, CopyRect3x5) {
// A DesktopFrame of 5-pixel width by 3-pixel height.
static uint32_t frame_pixels[] = {
0, 1, 2, 3, 4, //
5, 6, 7, 8, 9, //
10, 11, 12, 13, 14, //
};
ArrayDesktopFrame frame(DesktopSize(5, 3), frame_pixels);
{
BasicDesktopFrame target(DesktopSize(5, 3));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_0, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, frame));
}
// After Rotating clock-wise 90 degree
{
static uint32_t expected_pixels[] = {
10, 5, 0, //
11, 6, 1, //
12, 7, 2, //
13, 8, 3, //
14, 9, 4, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 180 degree
{
static uint32_t expected_pixels[]{
14, 13, 12, 11, 10, //
9, 8, 7, 6, 5, //
4, 3, 2, 1, 0, //
};
ArrayDesktopFrame expected(DesktopSize(5, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(5, 3));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_180, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 270 degree
{
static uint32_t expected_pixels[] = {
4, 9, 14, //
3, 8, 13, //
2, 7, 12, //
1, 6, 11, //
0, 5, 10, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
}
TEST(DesktopFrameRotationTest, PartialCopyRect3x5) {
// A DesktopFrame of 5-pixel width by 3-pixel height.
static uint32_t frame_pixels[] = {
0, 1, 2, 3, 4, //
5, 6, 7, 8, 9, //
10, 11, 12, 13, 14, //
};
ArrayDesktopFrame frame(DesktopSize(5, 3), frame_pixels);
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, //
0, 6, 7, 8, 0, //
0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(5, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(5, 3));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 1),
Rotation::CLOCK_WISE_0, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
{
static uint32_t expected_pixels[] = {
0, 1, 2, 3, 0, //
0, 6, 7, 8, 0, //
0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(5, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(5, 3));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 0, 3, 2),
Rotation::CLOCK_WISE_0, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 90 degree
{
static uint32_t expected_pixels[] = {
0, 0, 0, //
0, 6, 0, //
0, 7, 0, //
0, 8, 0, //
0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 1),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
{
static uint32_t expected_pixels[] = {
0, 0, 0, //
11, 6, 0, //
12, 7, 0, //
13, 8, 0, //
0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 2),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 180 degree
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, //
0, 8, 7, 6, 0, //
0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(5, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(5, 3));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 1),
Rotation::CLOCK_WISE_180, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
{
static uint32_t expected_pixels[] = {
0, 13, 12, 11, 0, //
0, 8, 7, 6, 0, //
0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(5, 3), expected_pixels);
BasicDesktopFrame target(DesktopSize(5, 3));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 2),
Rotation::CLOCK_WISE_180, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
// After Rotating clock-wise 270 degree
{
static uint32_t expected_pixels[] = {
0, 0, 0, //
0, 8, 0, //
0, 7, 0, //
0, 6, 0, //
0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 1, 3, 1),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
{
static uint32_t expected_pixels[] = {
0, 0, 0, //
3, 8, 0, //
2, 7, 0, //
1, 6, 0, //
0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(3, 5), expected_pixels);
BasicDesktopFrame target(DesktopSize(3, 5));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeXYWH(1, 0, 3, 2),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
}
}
TEST(DesktopFrameRotationTest, WithOffset) {
// A DesktopFrame of 4-pixel width by 3-pixel height.
static uint32_t frame_pixels[] = {
0, 1, 2, 3, //
4, 5, 6, 7, //
8, 9, 10, 11, //
};
ArrayDesktopFrame frame(DesktopSize(4, 3), frame_pixels);
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, 0, 0, 0, //
0, 0, 1, 2, 3, 0, 0, 0, //
0, 4, 5, 6, 7, 0, 0, 0, //
0, 8, 9, 10, 11, 0, 0, 0, //
0, 0, 0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(8, 6), expected_pixels);
BasicDesktopFrame target(DesktopSize(8, 6));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_0, DesktopVector(1, 1), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
target.mutable_updated_region()->Subtract(
DesktopRect::MakeOriginSize(DesktopVector(1, 1), frame.size()));
ASSERT_TRUE(target.updated_region().is_empty());
}
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, 0, 0, 0, //
0, 11, 10, 9, 8, 0, 0, 0, //
0, 7, 6, 5, 4, 0, 0, 0, //
0, 3, 2, 1, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(8, 6), expected_pixels);
BasicDesktopFrame target(DesktopSize(8, 6));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_180, DesktopVector(1, 1), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
target.mutable_updated_region()->Subtract(
DesktopRect::MakeOriginSize(DesktopVector(1, 1), frame.size()));
ASSERT_TRUE(target.updated_region().is_empty());
}
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, 0, //
0, 8, 4, 0, 0, 0, //
0, 9, 5, 1, 0, 0, //
0, 10, 6, 2, 0, 0, //
0, 11, 7, 3, 0, 0, //
0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(6, 8), expected_pixels);
BasicDesktopFrame target(DesktopSize(6, 8));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_90, DesktopVector(1, 1), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
target.mutable_updated_region()->Subtract(
DesktopRect::MakeXYWH(1, 1, 3, 4));
ASSERT_TRUE(target.updated_region().is_empty());
}
{
static uint32_t expected_pixels[] = {
0, 0, 0, 0, 0, 0, //
0, 3, 7, 11, 0, 0, //
0, 2, 6, 10, 0, 0, //
0, 1, 5, 9, 0, 0, //
0, 0, 4, 8, 0, 0, //
0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, //
0, 0, 0, 0, 0, 0, //
};
ArrayDesktopFrame expected(DesktopSize(6, 8), expected_pixels);
BasicDesktopFrame target(DesktopSize(6, 8));
ClearDesktopFrame(&target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_270, DesktopVector(1, 1), &target);
ASSERT_TRUE(DesktopFrameDataEquals(target, expected));
target.mutable_updated_region()->Subtract(
DesktopRect::MakeXYWH(1, 1, 3, 4));
ASSERT_TRUE(target.updated_region().is_empty());
}
}
// On a typical machine (Intel(R) Xeon(R) E5-1650 v3 @ 3.50GHz, with O2
// optimization, the following case uses ~1.4s to finish. It means entirely
// rotating one 2048 x 1536 frame, which is a large enough number to cover most
// of desktop computer users, uses around 14ms.
TEST(DesktopFrameRotationTest, DISABLED_PerformanceTest) {
BasicDesktopFrame frame(DesktopSize(2048, 1536));
BasicDesktopFrame target(DesktopSize(1536, 2048));
BasicDesktopFrame target2(DesktopSize(2048, 1536));
for (int i = 0; i < 100; i++) {
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_0, DesktopVector(), &target2);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_180, DesktopVector(), &target2);
}
}
// On a typical machine (Intel(R) Xeon(R) E5-1650 v3 @ 3.50GHz, with O2
// optimization, the following case uses ~6.7s to finish. It means entirely
// rotating one 4096 x 3072 frame uses around 67ms.
TEST(DesktopFrameRotationTest, DISABLED_PerformanceTestOnLargeScreen) {
BasicDesktopFrame frame(DesktopSize(4096, 3072));
BasicDesktopFrame target(DesktopSize(3072, 4096));
BasicDesktopFrame target2(DesktopSize(4096, 3072));
for (int i = 0; i < 100; i++) {
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_90, DesktopVector(), &target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_270, DesktopVector(), &target);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_0, DesktopVector(), &target2);
RotateDesktopFrame(frame, DesktopRect::MakeSize(frame.size()),
Rotation::CLOCK_WISE_180, DesktopVector(), &target2);
}
}
} // namespace webrtc