/* General DSI hardware state. */
 struct vc4_dsi {
+       struct vc4_encoder encoder;
+       struct mipi_dsi_host dsi_host;
+
        struct platform_device *pdev;
 
-       struct mipi_dsi_host dsi_host;
-       struct drm_encoder *encoder;
        struct drm_bridge *bridge;
        struct list_head bridge_chain;
 
 
 #define host_to_dsi(host) container_of(host, struct vc4_dsi, dsi_host)
 
+static inline struct vc4_dsi *
+to_vc4_dsi(struct drm_encoder *encoder)
+{
+       return container_of(encoder, struct vc4_dsi, encoder.base);
+}
+
 static inline void
 dsi_dma_workaround_write(struct vc4_dsi *dsi, u32 offset, u32 val)
 {
        DSI_WRITE(dsi->variant->port ? DSI1_##offset : DSI0_##offset, val)
 #define DSI_PORT_BIT(bit) (dsi->variant->port ? DSI1_##bit : DSI0_##bit)
 
-/* VC4 DSI encoder KMS struct */
-struct vc4_dsi_encoder {
-       struct vc4_encoder base;
-       struct vc4_dsi *dsi;
-};
-
-static inline struct vc4_dsi_encoder *
-to_vc4_dsi_encoder(struct drm_encoder *encoder)
-{
-       return container_of(encoder, struct vc4_dsi_encoder, base.base);
-}
-
 static const struct debugfs_reg32 dsi0_regs[] = {
        VC4_REG32(DSI0_CTRL),
        VC4_REG32(DSI0_STAT),
 
 static void vc4_dsi_encoder_disable(struct drm_encoder *encoder)
 {
-       struct vc4_dsi_encoder *vc4_encoder = to_vc4_dsi_encoder(encoder);
-       struct vc4_dsi *dsi = vc4_encoder->dsi;
+       struct vc4_dsi *dsi = to_vc4_dsi(encoder);
        struct device *dev = &dsi->pdev->dev;
        struct drm_bridge *iter;
 
                                       const struct drm_display_mode *mode,
                                       struct drm_display_mode *adjusted_mode)
 {
-       struct vc4_dsi_encoder *vc4_encoder = to_vc4_dsi_encoder(encoder);
-       struct vc4_dsi *dsi = vc4_encoder->dsi;
+       struct vc4_dsi *dsi = to_vc4_dsi(encoder);
        struct clk *phy_parent = clk_get_parent(dsi->pll_phy_clock);
        unsigned long parent_rate = clk_get_rate(phy_parent);
        unsigned long pixel_clock_hz = mode->clock * 1000;
 static void vc4_dsi_encoder_enable(struct drm_encoder *encoder)
 {
        struct drm_display_mode *mode = &encoder->crtc->state->adjusted_mode;
-       struct vc4_dsi_encoder *vc4_encoder = to_vc4_dsi_encoder(encoder);
-       struct vc4_dsi *dsi = vc4_encoder->dsi;
+       struct vc4_dsi *dsi = to_vc4_dsi(encoder);
        struct device *dev = &dsi->pdev->dev;
        bool debug_dump_regs = false;
        struct drm_bridge *iter;
        struct platform_device *pdev = to_platform_device(dev);
        struct drm_device *drm = dev_get_drvdata(master);
        struct vc4_dsi *dsi = dev_get_drvdata(dev);
-       struct vc4_dsi_encoder *vc4_dsi_encoder;
+       struct drm_encoder *encoder = &dsi->encoder.base;
        int ret;
 
        dsi->variant = of_device_get_match_data(dev);
 
-       vc4_dsi_encoder = devm_kzalloc(dev, sizeof(*vc4_dsi_encoder),
-                                      GFP_KERNEL);
-       if (!vc4_dsi_encoder)
-               return -ENOMEM;
-
        INIT_LIST_HEAD(&dsi->bridge_chain);
-       vc4_dsi_encoder->base.type = dsi->variant->port ?
-                       VC4_ENCODER_TYPE_DSI1 : VC4_ENCODER_TYPE_DSI0;
-       vc4_dsi_encoder->dsi = dsi;
-       dsi->encoder = &vc4_dsi_encoder->base.base;
+       dsi->encoder.type = dsi->variant->port ?
+               VC4_ENCODER_TYPE_DSI1 : VC4_ENCODER_TYPE_DSI0;
 
        dsi->regs = vc4_ioremap_regs(pdev, 0);
        if (IS_ERR(dsi->regs))
        if (ret)
                return ret;
 
-       drm_simple_encoder_init(drm, dsi->encoder, DRM_MODE_ENCODER_DSI);
-       drm_encoder_helper_add(dsi->encoder, &vc4_dsi_encoder_helper_funcs);
+       drm_simple_encoder_init(drm, encoder, DRM_MODE_ENCODER_DSI);
+       drm_encoder_helper_add(encoder, &vc4_dsi_encoder_helper_funcs);
 
-       ret = drm_bridge_attach(dsi->encoder, dsi->bridge, NULL, 0);
+       ret = drm_bridge_attach(encoder, dsi->bridge, NULL, 0);
        if (ret)
                return ret;
        /* Disable the atomic helper calls into the bridge.  We
         * from our driver, since we need to sequence them within the
         * encoder's enable/disable paths.
         */
-       list_splice_init(&dsi->encoder->bridge_chain, &dsi->bridge_chain);
+       list_splice_init(&encoder->bridge_chain, &dsi->bridge_chain);
 
        vc4_debugfs_add_regset32(drm, dsi->variant->debugfs_name, &dsi->regset);
 
                           void *data)
 {
        struct vc4_dsi *dsi = dev_get_drvdata(dev);
+       struct drm_encoder *encoder = &dsi->encoder.base;
 
        pm_runtime_disable(dev);
 
         * Restore the bridge_chain so the bridge detach procedure can happen
         * normally.
         */
-       list_splice_init(&dsi->bridge_chain, &dsi->encoder->bridge_chain);
-       drm_encoder_cleanup(dsi->encoder);
+       list_splice_init(&dsi->bridge_chain, &encoder->bridge_chain);
+       drm_encoder_cleanup(encoder);
 }
 
 static const struct component_ops vc4_dsi_ops = {