5 files changed, 373 insertions, 0 deletions

diff --git a/include/backend/drm/drm.h b/include/backend/drm/drm.h
new file mode 100644
index 00000000..de5212d3
--- /dev/null
+++ b/include/backend/drm/drm.h
@@ -0,0 +1,162 @@
+#ifndef BACKEND_DRM_DRM_H
+#define BACKEND_DRM_DRM_H
+
+#include <EGL/egl.h>
+#include <gbm.h>
+#include <stdbool.h>
+#include <stddef.h>
+#include <stdint.h>
+#include <time.h>
+#include <wayland-server.h>
+#include <wayland-util.h>
+#include <wlr/backend/drm.h>
+#include <wlr/backend/session.h>
+#include <wlr/render/egl.h>
+#include <xf86drmMode.h>
+#include "iface.h"
+#include "properties.h"
+#include "renderer.h"
+
+struct wlr_drm_plane {
+	uint32_t type;
+	uint32_t id;
+
+	uint32_t possible_crtcs;
+
+	struct wlr_drm_surface surf;
+	struct wlr_drm_surface mgpu_surf;
+
+	// Only used by cursor
+	float matrix[9];
+	struct gbm_bo *cursor_bo;
+	bool cursor_enabled;
+	int32_t cursor_hotspot_x, cursor_hotspot_y;
+
+	union wlr_drm_plane_props props;
+};
+
+struct wlr_drm_crtc {
+	uint32_t id;
+
+	// Atomic modesetting only
+	uint32_t mode_id;
+	uint32_t gamma_lut;
+	drmModeAtomicReq *atomic;
+
+	// Legacy only
+	drmModeCrtc *legacy_crtc;
+
+	union {
+		struct {
+			struct wlr_drm_plane *overlay;
+			struct wlr_drm_plane *primary;
+			struct wlr_drm_plane *cursor;
+		};
+		struct wlr_drm_plane *planes[3];
+	};
+
+	union wlr_drm_crtc_props props;
+
+	struct wl_list connectors;
+
+	uint16_t *gamma_table;
+	size_t gamma_table_size;
+};
+
+struct wlr_drm_backend {
+	struct wlr_backend backend;
+
+	struct wlr_drm_backend *parent;
+	const struct wlr_drm_interface *iface;
+	clockid_t clock;
+
+	int fd;
+
+	size_t num_crtcs;
+	struct wlr_drm_crtc *crtcs;
+	size_t num_planes;
+	struct wlr_drm_plane *planes;
+
+	union {
+		struct {
+			size_t num_overlay_planes;
+			size_t num_primary_planes;
+			size_t num_cursor_planes;
+		};
+		size_t num_type_planes[3];
+	};
+
+	union {
+		struct {
+			struct wlr_drm_plane *overlay_planes;
+			struct wlr_drm_plane *primary_planes;
+			struct wlr_drm_plane *cursor_planes;
+		};
+		struct wlr_drm_plane *type_planes[3];
+	};
+
+	struct wl_display *display;
+	struct wl_event_source *drm_event;
+
+	struct wl_listener display_destroy;
+	struct wl_listener session_signal;
+	struct wl_listener drm_invalidated;
+
+	struct wl_list outputs;
+
+	struct wlr_drm_renderer renderer;
+	struct wlr_session *session;
+};
+
+enum wlr_drm_connector_state {
+	// Connector is available but no output is plugged in
+	WLR_DRM_CONN_DISCONNECTED,
+	// An output just has been plugged in and is waiting for a modeset
+	WLR_DRM_CONN_NEEDS_MODESET,
+	WLR_DRM_CONN_CLEANUP,
+	WLR_DRM_CONN_CONNECTED,
+	// Connector disappeared, waiting for being destroyed on next page-flip
+	WLR_DRM_CONN_DISAPPEARED,
+};
+
+struct wlr_drm_mode {
+	struct wlr_output_mode wlr_mode;
+	drmModeModeInfo drm_mode;
+};
+
+struct wlr_drm_connector {
+	struct wlr_output output;
+
+	enum wlr_drm_connector_state state;
+	struct wlr_output_mode *desired_mode;
+	bool desired_enabled;
+	uint32_t id;
+
+	struct wlr_drm_crtc *crtc;
+	uint32_t possible_crtc;
+
+	union wlr_drm_connector_props props;
+
+	uint32_t width, height;
+	int32_t cursor_x, cursor_y;
+
+	drmModeCrtc *old_crtc;
+
+	bool pageflip_pending;
+	struct wl_event_source *retry_pageflip;
+	struct wl_list link;
+};
+
+struct wlr_drm_backend *get_drm_backend_from_backend(
+	struct wlr_backend *wlr_backend);
+bool check_drm_features(struct wlr_drm_backend *drm);
+bool init_drm_resources(struct wlr_drm_backend *drm);
+void finish_drm_resources(struct wlr_drm_backend *drm);
+void restore_drm_outputs(struct wlr_drm_backend *drm);
+void scan_drm_connectors(struct wlr_drm_backend *state);
+int handle_drm_event(int fd, uint32_t mask, void *data);
+bool enable_drm_connector(struct wlr_output *output, bool enable);
+bool set_drm_connector_gamma(struct wlr_output *output, size_t size,
+	const uint16_t *r, const uint16_t *g, const uint16_t *b);
+
+#endif
diff --git a/include/backend/drm/iface.h b/include/backend/drm/iface.h
new file mode 100644
index 00000000..5308b136
--- /dev/null
+++ b/include/backend/drm/iface.h
@@ -0,0 +1,41 @@
+#ifndef BACKEND_DRM_IFACE_H
+#define BACKEND_DRM_IFACE_H
+
+#include <gbm.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <xf86drm.h>
+#include <xf86drmMode.h>
+
+struct wlr_drm_backend;
+struct wlr_drm_connector;
+struct wlr_drm_crtc;
+
+// Used to provide atomic or legacy DRM functions
+struct wlr_drm_interface {
+	// Enable or disable DPMS for connector
+	bool (*conn_enable)(struct wlr_drm_backend *drm,
+		struct wlr_drm_connector *conn, bool enable);
+	// Pageflip on crtc. If mode is non-NULL perform a full modeset using it.
+	bool (*crtc_pageflip)(struct wlr_drm_backend *drm,
+		struct wlr_drm_connector *conn, struct wlr_drm_crtc *crtc,
+		uint32_t fb_id, drmModeModeInfo *mode);
+	// Enable the cursor buffer on crtc. Set bo to NULL to disable
+	bool (*crtc_set_cursor)(struct wlr_drm_backend *drm,
+		struct wlr_drm_crtc *crtc, struct gbm_bo *bo);
+	// Move the cursor on crtc
+	bool (*crtc_move_cursor)(struct wlr_drm_backend *drm,
+		struct wlr_drm_crtc *crtc, int x, int y);
+	// Set the gamma lut on crtc
+	bool (*crtc_set_gamma)(struct wlr_drm_backend *drm,
+		struct wlr_drm_crtc *crtc, size_t size,
+		uint16_t *r, uint16_t *g, uint16_t *b);
+	// Get the gamma lut size of a crtc
+	size_t (*crtc_get_gamma_size)(struct wlr_drm_backend *drm,
+		struct wlr_drm_crtc *crtc);
+};
+
+extern const struct wlr_drm_interface atomic_iface;
+extern const struct wlr_drm_interface legacy_iface;
+
+#endif
diff --git a/include/backend/drm/properties.h b/include/backend/drm/properties.h
new file mode 100644
index 00000000..b4d43bdd
--- /dev/null
+++ b/include/backend/drm/properties.h
@@ -0,0 +1,72 @@
+#ifndef BACKEND_DRM_PROPERTIES_H
+#define BACKEND_DRM_PROPERTIES_H
+
+#include <stdbool.h>
+#include <stdint.h>
+
+/*
+ * These types contain the property ids for several DRM objects.
+ * See https://01.org/linuxgraphics/gfx-docs/drm/gpu/drm-kms.html#kms-properties
+ * for more details.
+ */
+
+union wlr_drm_connector_props {
+	struct {
+		uint32_t edid;
+		uint32_t dpms;
+		uint32_t link_status; // not guaranteed to exist
+		uint32_t path;
+
+		// atomic-modesetting only
+
+		uint32_t crtc_id;
+	};
+	uint32_t props[4];
+};
+
+union wlr_drm_crtc_props {
+	struct {
+		// Neither of these are guaranteed to exist
+		uint32_t rotation;
+		uint32_t scaling_mode;
+
+		// atomic-modesetting only
+
+		uint32_t active;
+		uint32_t mode_id;
+		uint32_t gamma_lut;
+		uint32_t gamma_lut_size;
+	};
+	uint32_t props[6];
+};
+
+union wlr_drm_plane_props {
+	struct {
+		uint32_t type;
+		uint32_t rotation; // Not guaranteed to exist
+
+		// atomic-modesetting only
+
+		uint32_t src_x;
+		uint32_t src_y;
+		uint32_t src_w;
+		uint32_t src_h;
+		uint32_t crtc_x;
+		uint32_t crtc_y;
+		uint32_t crtc_w;
+		uint32_t crtc_h;
+		uint32_t fb_id;
+		uint32_t crtc_id;
+	};
+	uint32_t props[12];
+};
+
+bool get_drm_connector_props(int fd, uint32_t id,
+	union wlr_drm_connector_props *out);
+bool get_drm_crtc_props(int fd, uint32_t id, union wlr_drm_crtc_props *out);
+bool get_drm_plane_props(int fd, uint32_t id, union wlr_drm_plane_props *out);
+
+bool get_drm_prop(int fd, uint32_t obj, uint32_t prop, uint64_t *ret);
+void *get_drm_prop_blob(int fd, uint32_t obj, uint32_t prop, size_t *ret_len);
+
+#endif
diff --git a/include/backend/drm/renderer.h b/include/backend/drm/renderer.h
new file mode 100644
index 00000000..575758de
--- /dev/null
+++ b/include/backend/drm/renderer.h
@@ -0,0 +1,57 @@
+#ifndef BACKEND_DRM_RENDERER_H
+#define BACKEND_DRM_RENDERER_H
+
+#include <EGL/egl.h>
+#include <gbm.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <wlr/backend.h>
+#include <wlr/render/wlr_renderer.h>
+
+struct wlr_drm_backend;
+struct wlr_drm_plane;
+
+struct wlr_drm_renderer {
+	int fd;
+	struct gbm_device *gbm;
+	struct wlr_egl egl;
+
+	struct wlr_renderer *wlr_rend;
+};
+
+struct wlr_drm_surface {
+	struct wlr_drm_renderer *renderer;
+
+	uint32_t width;
+	uint32_t height;
+
+	struct gbm_surface *gbm;
+	EGLSurface egl;
+
+	struct gbm_bo *front;
+	struct gbm_bo *back;
+};
+
+bool init_drm_renderer(struct wlr_drm_backend *drm,
+	struct wlr_drm_renderer *renderer, wlr_renderer_create_func_t create_render);
+void finish_drm_renderer(struct wlr_drm_renderer *renderer);
+
+bool init_drm_surface(struct wlr_drm_surface *surf,
+	struct wlr_drm_renderer *renderer, uint32_t width, uint32_t height,
+	uint32_t format, uint32_t flags);
+
+bool init_drm_plane_surfaces(struct wlr_drm_plane *plane,
+	struct wlr_drm_backend *drm, int32_t width, uint32_t height,
+	uint32_t format);
+
+void finish_drm_surface(struct wlr_drm_surface *surf);
+bool make_drm_surface_current(struct wlr_drm_surface *surf, int *buffer_age);
+struct gbm_bo *swap_drm_surface_buffers(struct wlr_drm_surface *surf,
+	pixman_region32_t *damage);
+struct gbm_bo *get_drm_surface_front(struct wlr_drm_surface *surf);
+void post_drm_surface(struct wlr_drm_surface *surf);
+struct gbm_bo *copy_drm_surface_mgpu(struct wlr_drm_surface *dest,
+	struct gbm_bo *src);
+bool export_drm_bo(struct gbm_bo *bo, struct wlr_dmabuf_attributes *attribs);
+
+#endif
diff --git a/include/backend/drm/util.h b/include/backend/drm/util.h
new file mode 100644
index 00000000..e159d716
--- /dev/null
+++ b/include/backend/drm/util.h
@@ -0,0 +1,41 @@
+#ifndef BACKEND_DRM_UTIL_H
+#define BACKEND_DRM_UTIL_H
+
+#include <stdint.h>
+#include <wlr/types/wlr_output.h>
+#include <xf86drm.h>
+#include <xf86drmMode.h>
+
+// Calculates a more accurate refresh rate (mHz) than what mode itself provides
+int32_t calculate_refresh_rate(const drmModeModeInfo *mode);
+// Populates the make/model/phys_{width,height} of output from the edid data
+void parse_edid(struct wlr_output *restrict output, size_t len,
+	const uint8_t *data);
+// Returns the string representation of a DRM output type
+const char *conn_get_name(uint32_t type_id);
+// Returns the DRM framebuffer id for a gbm_bo
+uint32_t get_fb_for_bo(struct gbm_bo *bo);
+
+// Part of match_obj
+enum {
+	UNMATCHED = (uint32_t)-1,
+	SKIP = (uint32_t)-2,
+};
+
+/*
+ * Tries to match some DRM objects with some other DRM resource.
+ * e.g. Match CRTCs with Encoders, CRTCs with Planes.
+ *
+ * objs contains a bit array which resources it can be matched with.
+ * e.g. Bit 0 set means can be matched with res[0]
+ *
+ * res contains an index of which objs it is matched with or UNMATCHED.
+ *
+ * This solution is left in out.
+ * Returns the total number of matched solutions.
+ */
+size_t match_obj(size_t num_objs, const uint32_t objs[static restrict num_objs],
+		size_t num_res, const uint32_t res[static restrict num_res],
+		uint32_t out[static restrict num_res]);
+
+#endif