Quelle  drm_rect_test.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * Test cases for the drm_rect functions
 *
 * Copyright (c) 2022 Maíra Canal <mairacanal@riseup.net>
 */

#include <kunit/test.h>

#include <drm/drm_rect.h>
#include <drm/drm_mode.h>

#include <linux/string_helpers.h>
#include <linux/errno.h>

static void drm_rect_compare(struct kunit *test, const struct drm_rect *r,
        const struct drm_rect *expected)
{
 KUNIT_EXPECT_EQ(test, r->x1, expected->x1);
 KUNIT_EXPECT_EQ(test, r->y1, expected->y1);
 KUNIT_EXPECT_EQ(test, drm_rect_width(r), drm_rect_width(expected));
 KUNIT_EXPECT_EQ(test, drm_rect_height(r), drm_rect_height(expected));
}

static void drm_test_rect_clip_scaled_div_by_zero(struct kunit *test)
{
 struct drm_rect src, dst, clip;
 bool visible;

 /*
 * Make sure we don't divide by zero when dst
 * width/height is zero and dst and clip do not intersect.
 */
 drm_rect_init(&src, 0, 0, 0, 0);
 drm_rect_init(&dst, 0, 0, 0, 0);
 drm_rect_init(&clip, 1, 1, 1, 1);
 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, visible, "Destination not be visible\n");
 KUNIT_EXPECT_FALSE_MSG(test, drm_rect_visible(&src), "Source should not be visible\n");

 drm_rect_init(&src, 0, 0, 0, 0);
 drm_rect_init(&dst, 3, 3, 0, 0);
 drm_rect_init(&clip, 1, 1, 1, 1);
 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, visible, "Destination not be visible\n");
 KUNIT_EXPECT_FALSE_MSG(test, drm_rect_visible(&src), "Source should not be visible\n");
}

static void drm_test_rect_clip_scaled_not_clipped(struct kunit *test)
{
 struct drm_rect src, dst, clip;
 bool visible;

 /* 1:1 scaling */
 drm_rect_init(&src, 0, 0, 1 << 16, 1 << 16);
 drm_rect_init(&dst, 0, 0, 1, 1);
 drm_rect_init(&clip, 0, 0, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 1 << 16 ||
          src.y1 != 0 || src.y2 != 1 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 1 ||
          dst.y1 != 0 || dst.y2 != 1, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 2:1 scaling */
 drm_rect_init(&src, 0, 0, 2 << 16, 2 << 16);
 drm_rect_init(&dst, 0, 0, 1, 1);
 drm_rect_init(&clip, 0, 0, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 2 << 16 ||
          src.y1 != 0 || src.y2 != 2 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 1 ||
          dst.y1 != 0 || dst.y2 != 1, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 1:2 scaling */
 drm_rect_init(&src, 0, 0, 1 << 16, 1 << 16);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 0, 0, 2, 2);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 1 << 16 ||
          src.y1 != 0 || src.y2 != 1 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 2 ||
          dst.y1 != 0 || dst.y2 != 2, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");
}

static void drm_test_rect_clip_scaled_clipped(struct kunit *test)
{
 struct drm_rect src, dst, clip;
 bool visible;

 /* 1:1 scaling top/left clip */
 drm_rect_init(&src, 0, 0, 2 << 16, 2 << 16);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 0, 0, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 1 << 16 ||
          src.y1 != 0 || src.y2 != 1 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 1 ||
          dst.y1 != 0 || dst.y2 != 1, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 1:1 scaling bottom/right clip */
 drm_rect_init(&src, 0, 0, 2 << 16, 2 << 16);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 1, 1, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 1 << 16 || src.x2 != 2 << 16 ||
          src.y1 != 1 << 16 || src.y2 != 2 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 1 || dst.x2 != 2 || dst.y1 != 1 ||
          dst.y2 != 2, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 2:1 scaling top/left clip */
 drm_rect_init(&src, 0, 0, 4 << 16, 4 << 16);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 0, 0, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 2 << 16 ||
          src.y1 != 0 || src.y2 != 2 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 1 || dst.y1 != 0 ||
          dst.y2 != 1, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 2:1 scaling bottom/right clip */
 drm_rect_init(&src, 0, 0, 4 << 16, 4 << 16);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 1, 1, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 2 << 16 || src.x2 != 4 << 16 ||
          src.y1 != 2 << 16 || src.y2 != 4 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 1 || dst.x2 != 2 || dst.y1 != 1 ||
          dst.y2 != 2, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 1:2 scaling top/left clip */
 drm_rect_init(&src, 0, 0, 2 << 16, 2 << 16);
 drm_rect_init(&dst, 0, 0, 4, 4);
 drm_rect_init(&clip, 0, 0, 2, 2);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 0 || src.x2 != 1 << 16 ||
          src.y1 != 0 || src.y2 != 1 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 0 || dst.x2 != 2 || dst.y1 != 0 ||
          dst.y2 != 2, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");

 /* 1:2 scaling bottom/right clip */
 drm_rect_init(&src, 0, 0, 2 << 16, 2 << 16);
 drm_rect_init(&dst, 0, 0, 4, 4);
 drm_rect_init(&clip, 2, 2, 2, 2);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, src.x1 != 1 << 16 || src.x2 != 2 << 16 ||
          src.y1 != 1 << 16 || src.y2 != 2 << 16, "Source badly clipped\n");
 KUNIT_EXPECT_FALSE_MSG(test, dst.x1 != 2 || dst.x2 != 4 || dst.y1 != 2 ||
          dst.y2 != 4, "Destination badly clipped\n");
 KUNIT_EXPECT_TRUE_MSG(test, visible, "Destination should be visible\n");
 KUNIT_EXPECT_TRUE_MSG(test, drm_rect_visible(&src), "Source should be visible\n");
}

static void drm_test_rect_clip_scaled_signed_vs_unsigned(struct kunit *test)
{
 struct drm_rect src, dst, clip;
 bool visible;

 /*
 * 'clip.x2 - dst.x1 >= dst width' could result a negative
 * src rectangle width which is no longer expected by the
 * code as it's using unsigned types. This could lead to
 * the clipped source rectangle appering visible when it
 * should have been fully clipped. Make sure both rectangles
 * end up invisible.
 */
 drm_rect_init(&src, 0, 0, INT_MAX, INT_MAX);
 drm_rect_init(&dst, 0, 0, 2, 2);
 drm_rect_init(&clip, 3, 3, 1, 1);

 visible = drm_rect_clip_scaled(&src, &dst, &clip);

 KUNIT_EXPECT_FALSE_MSG(test, visible, "Destination should not be visible\n");
 KUNIT_EXPECT_FALSE_MSG(test, drm_rect_visible(&src), "Source should not be visible\n");
}

struct drm_rect_intersect_case {
 const char *description;
 struct drm_rect r1, r2;
 bool should_be_visible;
 struct drm_rect expected_intersection;
};

static const struct drm_rect_intersect_case drm_rect_intersect_cases[] = {
 {
  .description = "top-left x bottom-right",
  .r1 = DRM_RECT_INIT(1, 1, 2, 2),
  .r2 = DRM_RECT_INIT(0, 0, 2, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 1, 1, 1),
 },
 {
  .description = "top-right x bottom-left",
  .r1 = DRM_RECT_INIT(0, 0, 2, 2),
  .r2 = DRM_RECT_INIT(1, -1, 2, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 0, 1, 1),
 },
 {
  .description = "bottom-left x top-right",
  .r1 = DRM_RECT_INIT(1, -1, 2, 2),
  .r2 = DRM_RECT_INIT(0, 0, 2, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 0, 1, 1),
 },
 {
  .description = "bottom-right x top-left",
  .r1 = DRM_RECT_INIT(0, 0, 2, 2),
  .r2 = DRM_RECT_INIT(1, 1, 2, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 1, 1, 1),
 },
 {
  .description = "right x left",
  .r1 = DRM_RECT_INIT(0, 0, 2, 1),
  .r2 = DRM_RECT_INIT(1, 0, 3, 1),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 0, 1, 1),
 },
 {
  .description = "left x right",
  .r1 = DRM_RECT_INIT(1, 0, 3, 1),
  .r2 = DRM_RECT_INIT(0, 0, 2, 1),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 0, 1, 1),
 },
 {
  .description = "up x bottom",
  .r1 = DRM_RECT_INIT(0, 0, 1, 2),
  .r2 = DRM_RECT_INIT(0, -1, 1, 3),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(0, 0, 1, 2),
 },
 {
  .description = "bottom x up",
  .r1 = DRM_RECT_INIT(0, -1, 1, 3),
  .r2 = DRM_RECT_INIT(0, 0, 1, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(0, 0, 1, 2),
 },
 {
  .description = "touching corner",
  .r1 = DRM_RECT_INIT(0, 0, 1, 1),
  .r2 = DRM_RECT_INIT(1, 1, 2, 2),
  .should_be_visible = false,
  .expected_intersection = DRM_RECT_INIT(1, 1, 0, 0),
 },
 {
  .description = "touching side",
  .r1 = DRM_RECT_INIT(0, 0, 1, 1),
  .r2 = DRM_RECT_INIT(1, 0, 1, 1),
  .should_be_visible = false,
  .expected_intersection = DRM_RECT_INIT(1, 0, 0, 1),
 },
 {
  .description = "equal rects",
  .r1 = DRM_RECT_INIT(0, 0, 2, 2),
  .r2 = DRM_RECT_INIT(0, 0, 2, 2),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(0, 0, 2, 2),
 },
 {
  .description = "inside another",
  .r1 = DRM_RECT_INIT(0, 0, 2, 2),
  .r2 = DRM_RECT_INIT(1, 1, 1, 1),
  .should_be_visible = true,
  .expected_intersection = DRM_RECT_INIT(1, 1, 1, 1),
 },
 {
  .description = "far away",
  .r1 = DRM_RECT_INIT(0, 0, 1, 1),
  .r2 = DRM_RECT_INIT(3, 6, 1, 1),
  .should_be_visible = false,
  .expected_intersection = DRM_RECT_INIT(3, 6, -2, -5),
 },
 {
  .description = "points intersecting",
  .r1 = DRM_RECT_INIT(5, 10, 0, 0),
  .r2 = DRM_RECT_INIT(5, 10, 0, 0),
  .should_be_visible = false,
  .expected_intersection = DRM_RECT_INIT(5, 10, 0, 0),
 },
 {
  .description = "points not intersecting",
  .r1 = DRM_RECT_INIT(0, 0, 0, 0),
  .r2 = DRM_RECT_INIT(5, 10, 0, 0),
  .should_be_visible = false,
  .expected_intersection = DRM_RECT_INIT(5, 10, -5, -10),
 },
};

static void drm_rect_intersect_case_desc(const struct drm_rect_intersect_case *t, char *desc)
{
 snprintf(desc, KUNIT_PARAM_DESC_SIZE,
   "%s: " DRM_RECT_FMT " x " DRM_RECT_FMT,
   t->description, DRM_RECT_ARG(&t->r1), DRM_RECT_ARG(&t->r2));
}

KUNIT_ARRAY_PARAM(drm_rect_intersect, drm_rect_intersect_cases, drm_rect_intersect_case_desc);

static void drm_test_rect_intersect(struct kunit *test)
{
 const struct drm_rect_intersect_case *params = test->param_value;
 struct drm_rect r1_aux = params->r1;
 bool visible;

 visible = drm_rect_intersect(&r1_aux, ¶ms->r2);

 KUNIT_EXPECT_EQ(test, visible, params->should_be_visible);
 drm_rect_compare(test, &r1_aux, ¶ms->expected_intersection);
}

struct drm_rect_scale_case {
 const char *name;
 struct drm_rect src, dst;
 int min_range, max_range;
 int expected_scaling_factor;
};

static const struct drm_rect_scale_case drm_rect_scale_cases[] = {
 {
  .name = "normal use",
  .src = DRM_RECT_INIT(0, 0, 2 << 16, 2 << 16),
  .dst = DRM_RECT_INIT(0, 0, 1 << 16, 1 << 16),
  .min_range = 0, .max_range = INT_MAX,
  .expected_scaling_factor = 2,
 },
 {
  .name = "out of max range",
  .src = DRM_RECT_INIT(0, 0, 10 << 16, 10 << 16),
  .dst = DRM_RECT_INIT(0, 0, 1 << 16, 1 << 16),
  .min_range = 3, .max_range = 5,
  .expected_scaling_factor = -ERANGE,
 },
 {
  .name = "out of min range",
  .src = DRM_RECT_INIT(0, 0, 2 << 16, 2 << 16),
  .dst = DRM_RECT_INIT(0, 0, 1 << 16, 1 << 16),
  .min_range = 3, .max_range = 5,
  .expected_scaling_factor = -ERANGE,
 },
 {
  .name = "zero dst",
  .src = DRM_RECT_INIT(0, 0, 2 << 16, 2 << 16),
  .dst = DRM_RECT_INIT(0, 0, 0 << 16, 0 << 16),
  .min_range = 0, .max_range = INT_MAX,
  .expected_scaling_factor = 0,
 },
 {
  .name = "negative src",
  .src = DRM_RECT_INIT(0, 0, -(1 << 16), -(1 << 16)),
  .dst = DRM_RECT_INIT(0, 0, 1 << 16, 1 << 16),
  .min_range = 0, .max_range = INT_MAX,
  .expected_scaling_factor = -EINVAL,
 },
 {
  .name = "negative dst",
  .src = DRM_RECT_INIT(0, 0, 1 << 16, 1 << 16),
  .dst = DRM_RECT_INIT(0, 0, -(1 << 16), -(1 << 16)),
  .min_range = 0, .max_range = INT_MAX,
  .expected_scaling_factor = -EINVAL,
 },
};

static void drm_rect_scale_case_desc(const struct drm_rect_scale_case *t, char *desc)
{
 strscpy(desc, t->name, KUNIT_PARAM_DESC_SIZE);
}

KUNIT_ARRAY_PARAM(drm_rect_scale, drm_rect_scale_cases, drm_rect_scale_case_desc);

static void drm_test_rect_calc_hscale(struct kunit *test)
{
 const struct drm_rect_scale_case *params = test->param_value;
 int scaling_factor;

 scaling_factor = drm_rect_calc_hscale(¶ms->src, ¶ms->dst,
           params->min_range, params->max_range);

 KUNIT_EXPECT_EQ(test, scaling_factor, params->expected_scaling_factor);
}

static void drm_test_rect_calc_vscale(struct kunit *test)
{
 const struct drm_rect_scale_case *params = test->param_value;
 int scaling_factor;

 scaling_factor = drm_rect_calc_vscale(¶ms->src, ¶ms->dst,
           params->min_range, params->max_range);

 KUNIT_EXPECT_EQ(test, scaling_factor, params->expected_scaling_factor);
}

struct drm_rect_rotate_case {
 const char *name;
 unsigned int rotation;
 struct drm_rect rect;
 int width, height;
 struct drm_rect expected;
};

static const struct drm_rect_rotate_case drm_rect_rotate_cases[] = {
 {
  .name = "reflect-x",
  .rotation = DRM_MODE_REFLECT_X,
  .rect = DRM_RECT_INIT(0, 0, 5, 5),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(0, 0, 5, 5),
 },
 {
  .name = "reflect-y",
  .rotation = DRM_MODE_REFLECT_Y,
  .rect = DRM_RECT_INIT(2, 0, 5, 5),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(2, 5, 5, 5),
 },
 {
  .name = "rotate-0",
  .rotation = DRM_MODE_ROTATE_0,
  .rect = DRM_RECT_INIT(0, 2, 5, 5),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(0, 2, 5, 5),
 },
 {
  .name = "rotate-90",
  .rotation = DRM_MODE_ROTATE_90,
  .rect = DRM_RECT_INIT(0, 0, 5, 10),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(0, 0, 10, 5),
 },
 {
  .name = "rotate-180",
  .rotation = DRM_MODE_ROTATE_180,
  .rect = DRM_RECT_INIT(11, 3, 5, 10),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(-11, -3, 5, 10),
 },
 {
  .name = "rotate-270",
  .rotation = DRM_MODE_ROTATE_270,
  .rect = DRM_RECT_INIT(6, 3, 5, 10),
  .width = 5, .height = 10,
  .expected = DRM_RECT_INIT(-3, 6, 10, 5),
 },
};

static void drm_rect_rotate_case_desc(const struct drm_rect_rotate_case *t, char *desc)
{
 strscpy(desc, t->name, KUNIT_PARAM_DESC_SIZE);
}

KUNIT_ARRAY_PARAM(drm_rect_rotate, drm_rect_rotate_cases, drm_rect_rotate_case_desc);

static void drm_test_rect_rotate(struct kunit *test)
{
 const struct drm_rect_rotate_case *params = test->param_value;
 struct drm_rect r = params->rect;

 drm_rect_rotate(&r, params->width, params->height, params->rotation);

 drm_rect_compare(test, &r, ¶ms->expected);
}

static void drm_test_rect_rotate_inv(struct kunit *test)
{
 const struct drm_rect_rotate_case *params = test->param_value;
 struct drm_rect r = params->expected;

 drm_rect_rotate_inv(&r, params->width, params->height, params->rotation);

 drm_rect_compare(test, &r, ¶ms->rect);
}

static struct kunit_case drm_rect_tests[] = {
 KUNIT_CASE(drm_test_rect_clip_scaled_div_by_zero),
 KUNIT_CASE(drm_test_rect_clip_scaled_not_clipped),
 KUNIT_CASE(drm_test_rect_clip_scaled_clipped),
 KUNIT_CASE(drm_test_rect_clip_scaled_signed_vs_unsigned),
 KUNIT_CASE_PARAM(drm_test_rect_intersect, drm_rect_intersect_gen_params),
 KUNIT_CASE_PARAM(drm_test_rect_calc_hscale, drm_rect_scale_gen_params),
 KUNIT_CASE_PARAM(drm_test_rect_calc_vscale, drm_rect_scale_gen_params),
 KUNIT_CASE_PARAM(drm_test_rect_rotate, drm_rect_rotate_gen_params),
 KUNIT_CASE_PARAM(drm_test_rect_rotate_inv, drm_rect_rotate_gen_params),
 { }
};

static struct kunit_suite drm_rect_test_suite = {
 .name = "drm_rect",
 .test_cases = drm_rect_tests,
};

kunit_test_suite(drm_rect_test_suite);

MODULE_DESCRIPTION("Test cases for the drm_rect functions");
MODULE_LICENSE("GPL");