138 lines
3.3 KiB
Rust
138 lines
3.3 KiB
Rust
use std::os::fd::RawFd;
|
|
|
|
use drm_fourcc::{DrmFormat, DrmModifier};
|
|
|
|
#[cfg(feature = "egl")]
|
|
#[rustfmt::skip]
|
|
const EGL_DMABUF_PLANE_ATTRS: [isize; 20] = [
|
|
// FD Offset Stride ModLo ModHi
|
|
0x3272,0x3273,0x3274,0x3443,0x3444,
|
|
0x3275,0x3276,0x3277,0x3445,0x3446,
|
|
0x3278,0x3279,0x327A,0x3447,0x3448,
|
|
0x3440,0x3441,0x3442,0x3449,0x344A,
|
|
];
|
|
|
|
pub enum WlxFrame {
|
|
Dmabuf(DmabufFrame),
|
|
MemFd(MemFdFrame),
|
|
MemPtr(MemPtrFrame),
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, Default, PartialEq)]
|
|
pub enum Transform {
|
|
#[default]
|
|
Undefined,
|
|
Normal,
|
|
Rotated90,
|
|
Rotated180,
|
|
Rotated270,
|
|
Flipped,
|
|
Flipped90,
|
|
Flipped180,
|
|
Flipped270,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
pub struct FrameFormat {
|
|
pub width: u32,
|
|
pub height: u32,
|
|
pub drm_format: DrmFormat,
|
|
pub transform: Transform,
|
|
}
|
|
|
|
impl FrameFormat {
|
|
#[must_use]
|
|
pub fn get_mod_hi(&self) -> u32 {
|
|
let m = u64::from(self.drm_format.modifier);
|
|
(m >> 32) as _
|
|
}
|
|
#[must_use]
|
|
pub fn get_mod_lo(&self) -> u32 {
|
|
let m = u64::from(self.drm_format.modifier);
|
|
(m & 0xFFFFFFFF) as _
|
|
}
|
|
pub fn set_mod(&mut self, mod_hi: u32, mod_low: u32) {
|
|
self.drm_format.modifier = DrmModifier::from(((mod_hi as u64) << 32) + mod_low as u64);
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Copy, Default)]
|
|
pub struct FramePlane {
|
|
pub fd: Option<RawFd>,
|
|
pub offset: u32,
|
|
pub stride: i32,
|
|
}
|
|
|
|
pub struct DmabufFrame {
|
|
pub format: FrameFormat,
|
|
pub num_planes: usize,
|
|
pub planes: [FramePlane; 4],
|
|
pub mouse: Option<MouseMeta>,
|
|
}
|
|
|
|
impl DmabufFrame {
|
|
#[cfg(feature = "egl")]
|
|
/// Get the attributes for creating an EGLImage.
|
|
/// Pacics if fd is None; check using `is_valid` first.
|
|
pub fn get_egl_image_attribs(&self) -> Vec<isize> {
|
|
let mut vec: Vec<isize> = vec![
|
|
0x3057, // WIDTH
|
|
self.format.width as _,
|
|
0x3056, // HEIGHT
|
|
self.format.height as _,
|
|
0x3271, // LINUX_DRM_FOURCC_EXT,
|
|
self.format.drm_format.code as _,
|
|
];
|
|
|
|
for i in 0..self.num_planes {
|
|
let mut a = i * 5usize;
|
|
vec.push(EGL_DMABUF_PLANE_ATTRS[a]);
|
|
vec.push(self.planes[i].fd.unwrap() as _); // safe to unwrap due to contract
|
|
a += 1;
|
|
vec.push(EGL_DMABUF_PLANE_ATTRS[a]);
|
|
vec.push(self.planes[i].offset as _);
|
|
a += 1;
|
|
vec.push(EGL_DMABUF_PLANE_ATTRS[a]);
|
|
vec.push(self.planes[i].stride as _);
|
|
a += 1;
|
|
vec.push(EGL_DMABUF_PLANE_ATTRS[a]);
|
|
vec.push(self.format.get_mod_lo() as _);
|
|
a += 1;
|
|
vec.push(EGL_DMABUF_PLANE_ATTRS[a]);
|
|
vec.push(self.format.get_mod_hi() as _);
|
|
}
|
|
vec.push(0x3038); // NONE
|
|
|
|
vec
|
|
}
|
|
|
|
/// Returns true if all planes have a valid file descriptor.
|
|
pub fn is_valid(&self) -> bool {
|
|
for i in 0..self.num_planes {
|
|
if self.planes[i].fd.is_none() {
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
}
|
|
|
|
pub struct MemFdFrame {
|
|
pub format: FrameFormat,
|
|
pub plane: FramePlane,
|
|
pub mouse: Option<MouseMeta>,
|
|
}
|
|
|
|
pub struct MemPtrFrame {
|
|
pub format: FrameFormat,
|
|
pub ptr: usize,
|
|
pub size: usize,
|
|
pub mouse: Option<MouseMeta>,
|
|
}
|
|
|
|
#[derive(Default, Clone, PartialEq)]
|
|
pub struct MouseMeta {
|
|
pub x: f32,
|
|
pub y: f32,
|
|
}
|