Hi Lyude > On 30 Sep 2024, at 20:10, Lyude Paul <lyude@xxxxxxxxxx> wrote: > > Same thing as RawCrtc and RawCrtcState, but for DRM planes now > > Signed-off-by: Lyude Paul <lyude@xxxxxxxxxx> > --- > rust/kernel/drm/kms/plane.rs | 35 +++++++++++++++++++++++++++++++++++ > 1 file changed, 35 insertions(+) > > diff --git a/rust/kernel/drm/kms/plane.rs b/rust/kernel/drm/kms/plane.rs > index 3ace487316d46..1c151ae3b3dcc 100644 > --- a/rust/kernel/drm/kms/plane.rs > +++ b/rust/kernel/drm/kms/plane.rs > @@ -312,6 +312,27 @@ unsafe impl<T: DriverPlane> Send for Plane<T> {} > // SAFETY: Our interface is thread-safe. > unsafe impl<T: DriverPlane> Sync for Plane<T> {} > > +/// Common methods available on any type which implements [`AsRawPlane`]. > +/// > +/// This is implemented internally by DRM, and provides many of the basic methods for working with > +/// planes. > +pub trait RawPlane: AsRawPlane { > + /// Return the index of this DRM plane > + #[inline] > + fn index(&self) -> u32 { > + // SAFETY: The index is initialized by the time we expose `Plane` objects to users, and is > + // invariant throughout the lifetime of the `Plane` > + unsafe { (*self.as_raw()).index } > + } > + > + /// Return the index of this DRM plane in the form of a bitmask > + #[inline] > + fn mask(&self) -> u32 { > + 1 << self.index() > + } > +} > +impl<T: AsRawPlane> RawPlane for T {} > + > /// A [`struct drm_plane`] without a known [`DriverPlane`] implementation. > /// > /// This is mainly for situations where our bindings can't infer the [`DriverPlane`] implementation > @@ -426,6 +447,20 @@ pub trait FromRawPlaneState: AsRawPlaneState { > unsafe fn from_raw_mut<'a>(ptr: *mut bindings::drm_plane_state) -> &'a mut Self; > } > > +/// Common methods available on any type which implements [`AsRawPlane`]. > +/// > +/// This is implemented internally by DRM, and provides many of the basic methods for working with > +/// the atomic state of [`Plane`]s. > +pub trait RawPlaneState: AsRawPlaneState { > + /// Return the plane that this plane state belongs to. > + fn plane(&self) -> &Self::Plane { > + // SAFETY: The index is initialized by the time we expose Plane objects to users, and is > + // invariant throughout the lifetime of the Plane > + unsafe { Self::Plane::from_raw(self.as_raw().plane) } > + } > +} > +impl<T: AsRawPlaneState + ?Sized> RawPlaneState for T {} > + > /// The main interface for a [`struct drm_plane_state`]. > /// > /// This type is the main interface for dealing with the atomic state of DRM planes. In addition, it > -- > 2.46.1 > LGTM — Daniel