1// SPDX-License-Identifier: GPL-2.0
2
3#include <drm/drm_atomic_state_helper.h>
4#include <drm/drm_atomic_uapi.h>
5#include <drm/drm_connector.h>
6#include <drm/drm_crtc.h>
7#include <drm/drm_encoder.h>
8#include <drm/drm_modeset_helper_vtables.h>
9
10#include <kunit/test.h>
11
12#include "vc4_mock.h"
13
14static const struct drm_connector_helper_funcs vc4_dummy_connector_helper_funcs = {
15};
16
17static const struct drm_connector_funcs vc4_dummy_connector_funcs = {
18	.atomic_destroy_state	= drm_atomic_helper_connector_destroy_state,
19	.atomic_duplicate_state	= drm_atomic_helper_connector_duplicate_state,
20	.reset			= drm_atomic_helper_connector_reset,
21};
22
23struct vc4_dummy_output *vc4_dummy_output(struct kunit *test,
24					  struct drm_device *drm,
25					  struct drm_crtc *crtc,
26					  enum vc4_encoder_type vc4_encoder_type,
27					  unsigned int kms_encoder_type,
28					  unsigned int connector_type)
29{
30	struct vc4_dummy_output *dummy_output;
31	struct drm_connector *conn;
32	struct drm_encoder *enc;
33	int ret;
34
35	dummy_output = drmm_kzalloc(drm, sizeof(*dummy_output), GFP_KERNEL);
36	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, dummy_output);
37	dummy_output->encoder.type = vc4_encoder_type;
38
39	enc = &dummy_output->encoder.base;
40	ret = drmm_encoder_init(drm, enc,
41				NULL,
42				kms_encoder_type,
43				NULL);
44	KUNIT_ASSERT_EQ(test, ret, 0);
45	enc->possible_crtcs = drm_crtc_mask(crtc);
46
47	conn = &dummy_output->connector;
48	ret = drmm_connector_init(drm, conn,
49				  &vc4_dummy_connector_funcs,
50				  connector_type,
51				  NULL);
52	KUNIT_ASSERT_EQ(test, ret, 0);
53
54	drm_connector_helper_add(conn, &vc4_dummy_connector_helper_funcs);
55	drm_connector_attach_encoder(conn, enc);
56
57	return dummy_output;
58}
59
60static const struct drm_display_mode default_mode = {
61	DRM_SIMPLE_MODE(640, 480, 64, 48)
62};
63
64int vc4_mock_atomic_add_output(struct kunit *test,
65			       struct drm_atomic_state *state,
66			       enum vc4_encoder_type type)
67{
68	struct drm_device *drm = state->dev;
69	struct drm_connector_state *conn_state;
70	struct drm_crtc_state *crtc_state;
71	struct vc4_dummy_output *output;
72	struct drm_connector *conn;
73	struct drm_encoder *encoder;
74	struct drm_crtc *crtc;
75	int ret;
76
77	encoder = vc4_find_encoder_by_type(drm, type);
78	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, encoder);
79
80	crtc = vc4_find_crtc_for_encoder(test, drm, encoder);
81	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, crtc);
82
83	output = encoder_to_vc4_dummy_output(encoder);
84	conn = &output->connector;
85	conn_state = drm_atomic_get_connector_state(state, conn);
86	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, conn_state);
87
88	ret = drm_atomic_set_crtc_for_connector(conn_state, crtc);
89	KUNIT_EXPECT_EQ(test, ret, 0);
90
91	crtc_state = drm_atomic_get_crtc_state(state, crtc);
92	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, crtc_state);
93
94	ret = drm_atomic_set_mode_for_crtc(crtc_state, &default_mode);
95	KUNIT_EXPECT_EQ(test, ret, 0);
96
97	crtc_state->active = true;
98
99	return 0;
100}
101
102int vc4_mock_atomic_del_output(struct kunit *test,
103			       struct drm_atomic_state *state,
104			       enum vc4_encoder_type type)
105{
106	struct drm_device *drm = state->dev;
107	struct drm_connector_state *conn_state;
108	struct drm_crtc_state *crtc_state;
109	struct vc4_dummy_output *output;
110	struct drm_connector *conn;
111	struct drm_encoder *encoder;
112	struct drm_crtc *crtc;
113	int ret;
114
115	encoder = vc4_find_encoder_by_type(drm, type);
116	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, encoder);
117
118	crtc = vc4_find_crtc_for_encoder(test, drm, encoder);
119	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, crtc);
120
121	crtc_state = drm_atomic_get_crtc_state(state, crtc);
122	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, crtc_state);
123
124	crtc_state->active = false;
125
126	ret = drm_atomic_set_mode_for_crtc(crtc_state, NULL);
127	KUNIT_ASSERT_EQ(test, ret, 0);
128
129	output = encoder_to_vc4_dummy_output(encoder);
130	conn = &output->connector;
131	conn_state = drm_atomic_get_connector_state(state, conn);
132	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, conn_state);
133
134	ret = drm_atomic_set_crtc_for_connector(conn_state, NULL);
135	KUNIT_ASSERT_EQ(test, ret, 0);
136
137	return 0;
138}
139