/* Public domain. */

#include <drm/drm_gem.h>
#include <drm/drm_framebuffer.h>
#include <drm/drm_modeset_helper.h>

void
drm_gem_fb_destroy(struct drm_framebuffer *fb)
{
        int i;

        for (i = 0; i < 4; i++)
                drm_gem_object_put(fb->obj[i]);
        drm_framebuffer_cleanup(fb);
        free(fb, M_DRM, 0);
}

int
drm_gem_fb_create_handle(struct drm_framebuffer *fb, struct drm_file *file,
    unsigned int *handle)
{
        return drm_gem_handle_create(file, fb->obj[0], handle);
}

const struct drm_framebuffer_funcs drm_gem_fb_funcs = {
        .create_handle = drm_gem_fb_create_handle,
        .destroy = drm_gem_fb_destroy,
};

struct drm_framebuffer *
drm_gem_fb_create(struct drm_device *dev, struct drm_file *file,
    const struct drm_format_info *info, const struct drm_mode_fb_cmd2 *cmd)
{
        struct drm_framebuffer *fb;
        struct drm_gem_object *gem_obj;
        int error;

        KASSERT(info->num_planes == 1);

        gem_obj = drm_gem_object_lookup(file, cmd->handles[0]);
        if (gem_obj == NULL)
                return ERR_PTR(-ENOENT);

        fb = malloc(sizeof(*fb), M_DRM, M_ZERO | M_WAITOK);

        drm_helper_mode_fill_fb_struct(dev, fb, info, cmd);
        fb->obj[0] = gem_obj;

        error = drm_framebuffer_init(dev, fb, &drm_gem_fb_funcs);
        if (error != 0)
                goto dealloc;

        return fb;

dealloc:
        drm_framebuffer_cleanup(fb);
        free(fb, M_DRM, sizeof(*fb));
        drm_gem_object_put(gem_obj);

        return ERR_PTR(error);
}

struct drm_gem_object *
drm_gem_fb_get_obj(struct drm_framebuffer *fb, unsigned int plane)
{
        if (plane < nitems(fb->obj))
                return fb->obj[plane];
        return NULL;
}