#include <drm/drm_atomic_helper.h>
 #include <drm/drm_crtc.h>
 #include <drm/drm_crtc_helper.h>
+#include <drm/drm_encoder.h>
 #include <drm/drm_modes.h>
 #include <drm/drm_of.h>
 
+#include <uapi/drm/drm_mode.h>
+
 #include <linux/component.h>
 #include <linux/ioport.h>
 #include <linux/of_address.h>
 void sun4i_tcon_set_mux(struct sun4i_tcon *tcon, int channel,
                        struct drm_encoder *encoder)
 {
-       u32 val;
-
-       if (!tcon->quirks->has_unknown_mux)
-               return;
+       int ret = -ENOTSUPP;
 
-       if (channel != 1)
-               return;
-
-       if (encoder->encoder_type == DRM_MODE_ENCODER_TVDAC)
-               val = 1;
-       else
-               val = 0;
+       if (tcon->quirks->set_mux)
+               ret = tcon->quirks->set_mux(tcon, encoder);
 
-       /*
-        * FIXME: Undocumented bits
-        */
-       regmap_write(tcon->regs, SUN4I_TCON_MUX_CTRL_REG, val);
+       DRM_DEBUG_DRIVER("Muxing encoder %s to CRTC %s: %d\n",
+                        encoder->name, encoder->crtc->name, ret);
 }
 EXPORT_SYMBOL(sun4i_tcon_set_mux);
 
        return 0;
 }
 
+/* platform specific TCON muxing callbacks */
+static int sun5i_a13_tcon_set_mux(struct sun4i_tcon *tcon,
+                                 struct drm_encoder *encoder)
+{
+       u32 val;
+
+       if (encoder->encoder_type == DRM_MODE_ENCODER_TVDAC)
+               val = 1;
+       else
+               val = 0;
+
+       /*
+        * FIXME: Undocumented bits
+        */
+       return regmap_write(tcon->regs, SUN4I_TCON_MUX_CTRL_REG, val);
+}
+
 static const struct sun4i_tcon_quirks sun5i_a13_quirks = {
-       .has_unknown_mux = true,
-       .has_channel_1  = true,
+       .has_channel_1          = true,
+       .set_mux                = sun5i_a13_tcon_set_mux,
 };
 
 static const struct sun4i_tcon_quirks sun6i_a31_quirks = {
 
 
 #define SUN4I_TCON_MAX_CHANNELS                2
 
+struct sun4i_tcon;
+
 struct sun4i_tcon_quirks {
-       bool    has_unknown_mux; /* sun5i has undocumented mux */
        bool    has_channel_1;  /* a33 does not have channel 1 */
        bool    needs_de_be_mux; /* sun6i needs mux to select backend */
+
+       /* callback to handle tcon muxing options */
+       int     (*set_mux)(struct sun4i_tcon *, struct drm_encoder *);
 };
 
 struct sun4i_tcon {