You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
337 lines
9.1 KiB
337 lines
9.1 KiB
/*
|
|
* Copyright (c) 2019 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.h"
|
|
|
|
#include <memory>
|
|
|
|
#include "modules/desktop_capture/desktop_region.h"
|
|
#include "modules/desktop_capture/test_utils.h"
|
|
#include "rtc_base/arraysize.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
std::unique_ptr<DesktopFrame> CreateTestFrame(DesktopRect rect,
|
|
int pixels_value) {
|
|
DesktopSize size = rect.size();
|
|
auto frame = std::make_unique<BasicDesktopFrame>(size);
|
|
frame->set_top_left(rect.top_left());
|
|
memset(frame->data(), pixels_value, frame->stride() * size.height());
|
|
return frame;
|
|
}
|
|
|
|
struct TestData {
|
|
const char* description;
|
|
DesktopRect dest_frame_rect;
|
|
DesktopRect src_frame_rect;
|
|
double horizontal_scale;
|
|
double vertical_scale;
|
|
DesktopRect expected_overlap_rect;
|
|
};
|
|
|
|
void RunTest(const TestData& test) {
|
|
// Copy a source frame with all bits set into a dest frame with none set.
|
|
auto dest_frame = CreateTestFrame(test.dest_frame_rect, 0);
|
|
auto src_frame = CreateTestFrame(test.src_frame_rect, 0xff);
|
|
|
|
dest_frame->CopyIntersectingPixelsFrom(
|
|
*src_frame, test.horizontal_scale, test.vertical_scale);
|
|
|
|
// Translate the expected overlap rect to be relative to the dest frame/rect.
|
|
DesktopVector dest_frame_origin = test.dest_frame_rect.top_left();
|
|
DesktopRect relative_expected_overlap_rect = test.expected_overlap_rect;
|
|
relative_expected_overlap_rect.Translate(-dest_frame_origin.x(),
|
|
-dest_frame_origin.y());
|
|
|
|
// Confirm bits are now set in the dest frame if & only if they fall in the
|
|
// expected range.
|
|
for (int y = 0; y < dest_frame->size().height(); ++y) {
|
|
SCOPED_TRACE(y);
|
|
|
|
for (int x = 0; x < dest_frame->size().width(); ++x) {
|
|
SCOPED_TRACE(x);
|
|
|
|
DesktopVector point(x, y);
|
|
uint8_t* data = dest_frame->GetFrameDataAtPos(point);
|
|
uint32_t pixel_value = *reinterpret_cast<uint32_t*>(data);
|
|
bool was_copied = pixel_value == 0xffffffff;
|
|
ASSERT_TRUE(was_copied || pixel_value == 0);
|
|
|
|
bool expected_to_be_copied =
|
|
relative_expected_overlap_rect.Contains(point);
|
|
|
|
ASSERT_EQ(was_copied, expected_to_be_copied);
|
|
}
|
|
}
|
|
}
|
|
|
|
void RunTests(const TestData* tests, int num_tests) {
|
|
for (int i = 0; i < num_tests; i++) {
|
|
const TestData& test = tests[i];
|
|
|
|
SCOPED_TRACE(test.description);
|
|
|
|
RunTest(test);
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsMatchingRects) {
|
|
const TestData tests[] = {
|
|
{"0 origin",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2)},
|
|
|
|
{"Negative origin",
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsMatchingRectsScaled) {
|
|
// The scale factors shouldn't affect matching rects (they're only applied
|
|
// to any difference between the origins)
|
|
const TestData tests[] = {
|
|
{"0 origin 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2)},
|
|
|
|
{"0 origin 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2)},
|
|
|
|
{"Negative origin 2x",
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2)},
|
|
|
|
{"Negative origin 0.5x",
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsFullyContainedRects) {
|
|
const TestData tests[] = {
|
|
{"0 origin top left",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1)},
|
|
|
|
{"0 origin bottom right",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(1, 1, 1, 1),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(1, 1, 1, 1)},
|
|
|
|
{"Negative origin bottom left",
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, 0, 1, 1),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(-1, 0, 1, 1)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsFullyContainedRectsScaled) {
|
|
const TestData tests[] = {
|
|
{"0 origin top left 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1)},
|
|
|
|
{"0 origin top left 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1)},
|
|
|
|
{"0 origin bottom left 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 4, 4),
|
|
DesktopRect::MakeXYWH(1, 1, 2, 2),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(2, 2, 2, 2)},
|
|
|
|
{"0 origin bottom middle 2x/1x",
|
|
DesktopRect::MakeXYWH(0, 0, 4, 3),
|
|
DesktopRect::MakeXYWH(1, 1, 2, 2),
|
|
2.0, 1.0,
|
|
DesktopRect::MakeXYWH(2, 1, 2, 2)},
|
|
|
|
{"0 origin middle 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 3, 3),
|
|
DesktopRect::MakeXYWH(2, 2, 1, 1),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(1, 1, 1, 1)},
|
|
|
|
{"Negative origin bottom left 2x",
|
|
DesktopRect::MakeXYWH(-1, -1, 3, 3),
|
|
DesktopRect::MakeXYWH(-1, 0, 1, 1),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(-1, 1, 1, 1)},
|
|
|
|
{"Negative origin near middle 0.5x",
|
|
DesktopRect::MakeXYWH(-2, -2, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(-1, -1, 1, 1)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsPartiallyContainedRects) {
|
|
const TestData tests[] = {
|
|
{"Top left",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, -1, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1)},
|
|
|
|
{"Top right",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(1, -1, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(1, 0, 1, 1)},
|
|
|
|
{"Bottom right",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(1, 1, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(1, 1, 1, 1)},
|
|
|
|
{"Bottom left",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, 1, 2, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 1, 1, 1)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsPartiallyContainedRectsScaled) {
|
|
const TestData tests[] = {
|
|
{"Top left 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, -1, 3, 3),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 0, 1, 1)},
|
|
|
|
{"Top right 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(2, -2, 2, 2),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(1, 0, 1, 1)},
|
|
|
|
{"Bottom right 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 3, 3),
|
|
DesktopRect::MakeXYWH(-1, 1, 3, 3),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 2, 1, 1)},
|
|
|
|
{"Bottom left 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-2, 2, 2, 2),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(0, 1, 1, 1)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsUncontainedRects) {
|
|
const TestData tests[] = {
|
|
{"Left",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, 0, 1, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
{"Top",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, -1, 2, 1),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
{"Right",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(2, 0, 1, 2),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
|
|
{"Bottom",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 2, 2, 1),
|
|
1.0, 1.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
TEST(DesktopFrameTest, CopyIntersectingPixelsUncontainedRectsScaled) {
|
|
const TestData tests[] = {
|
|
{"Left 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(-1, 0, 2, 2),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
{"Top 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, -2, 2, 1),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
{"Right 2x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(1, 0, 1, 2),
|
|
2.0, 2.0,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)},
|
|
|
|
|
|
{"Bottom 0.5x",
|
|
DesktopRect::MakeXYWH(0, 0, 2, 2),
|
|
DesktopRect::MakeXYWH(0, 4, 2, 1),
|
|
0.5, 0.5,
|
|
DesktopRect::MakeXYWH(0, 0, 0, 0)}
|
|
};
|
|
|
|
RunTests(tests, arraysize(tests));
|
|
}
|
|
|
|
} // namespace webrtc
|