#include "kirin_drm_drv.h"
 #include "kirin_ade_reg.h"
 
-#define PRIMARY_CH     ADE_CH1 /* primary plane */
 #define OUT_OVLY       ADE_OVLY2 /* output overlay compositor */
 #define ADE_DEBUG      1
 
        struct kirin_plane *kplane;
        enum drm_plane_type type;
        int ret;
-       int i;
+       u32 ch;
 
        ade = devm_kzalloc(dev->dev, sizeof(*ade), GFP_KERNEL);
        if (!ade) {
         * TODO: Now only support primary plane, overlay planes
         * need to do.
         */
-       for (i = 0; i < ADE_CH_NUM; i++) {
-               kplane = &ade->planes[i];
-               kplane->ch = i;
+       for (ch = 0; ch < ade_driver_data.num_planes; ch++) {
+               kplane = &ade->planes[ch];
+               kplane->ch = ch;
                kplane->hw_ctx = ctx;
-               type = i == PRIMARY_CH ? DRM_PLANE_TYPE_PRIMARY :
-                       DRM_PLANE_TYPE_OVERLAY;
+
+               if (ch == ade_driver_data.prim_plane)
+                       type = DRM_PLANE_TYPE_PRIMARY;
+               else
+                       type = DRM_PLANE_TYPE_OVERLAY;
 
                ret = ade_plane_init(dev, kplane, type);
                if (ret)
        }
 
        /* crtc init */
-       ret = ade_crtc_init(dev, &kcrtc->base, &ade->planes[PRIMARY_CH].base);
+       ret = ade_crtc_init(dev, &kcrtc->base,
+                           &ade->planes[ade_driver_data.prim_plane].base);
        if (ret)
                return ret;
 
 };
 
 struct kirin_drm_data ade_driver_data = {
+       .num_planes = ADE_CH_NUM,
+       .prim_plane = ADE_CH1,
        .channel_formats = channel_formats,
        .channel_formats_cnt = ARRAY_SIZE(channel_formats),
        .crtc_helper_funcs = &ade_crtc_helper_funcs,