/*
Copyright (C) 2018 drebrez <drebrez@gmail.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <linux/fb.h>
#include <sys/ioctl.h>

void usage(char* appname)
{
    printf("Usage: %s [-d DEV] [-i] [-p] [-h]\n\
    -d  Framebuffer device (default /dev/fb0)\n\
    -i  Show fixed and variable screen info\n\
    -p  Perform the panning of the display\n\
    -h  Show this help\n", appname);
}

void print_fix_screeninfo(struct fb_fix_screeninfo* scr_fix)
{
    // ref: http://elixir.free-electrons.com/linux/latest/ident/fb_fix_screeninfo
    printf("Fixed screen info:\n");
    printf("\tid           = %s\n", scr_fix->id);
    printf("\tsmem_start   = %lu\n", scr_fix->smem_start);
    printf("\tsmem_len     = %u\n", scr_fix->smem_len);
    printf("\ttype         = %u\n", scr_fix->type);
    printf("\ttype_aux     = %u\n", scr_fix->type_aux);
    printf("\tvisual       = %u\n", scr_fix->visual);
    printf("\txpanstep     = %hu\n", scr_fix->xpanstep);
    printf("\typanstep     = %hu\n", scr_fix->ypanstep);
    printf("\tywrapstep    = %hu\n", scr_fix->ywrapstep);
    printf("\tline_length  = %u\n", scr_fix->line_length);
    printf("\tmmio_start   = %lu\n", scr_fix->mmio_start);
    printf("\tmmio_len     = %u\n", scr_fix->mmio_len);
    printf("\taccel        = %u\n", scr_fix->accel);
    printf("\tcapabilities = %hu\n", scr_fix->capabilities);
    printf("\treserved[0]  = %hu\n", scr_fix->reserved[0]);
    printf("\treserved[1]  = %hu\n", scr_fix->reserved[1]);
}

void print_var_screeninfo(struct fb_var_screeninfo* scr_var)
{
    // ref: http://elixir.free-electrons.com/linux/latest/ident/fb_var_screeninfo
    printf("Variable screen info:\n");
    printf("\txres             = %u\n", scr_var->xres);
    printf("\tyres             = %u\n", scr_var->yres);
    printf("\txres_virtual     = %u\n", scr_var->xres_virtual);
    printf("\tyres_virtual     = %u\n", scr_var->yres_virtual);
    printf("\txoffset          = %u\n", scr_var->xoffset);
    printf("\tyoffset          = %u\n", scr_var->yoffset);
    printf("\tbits_per_pixel   = %u\n", scr_var->bits_per_pixel);
    printf("\tgrayscale        = %u\n", scr_var->grayscale);
    printf("\tred.offset       = %u\n", scr_var->red.offset);
    printf("\tred.length       = %u\n", scr_var->red.length);
    printf("\tred.msb_right    = %u\n", scr_var->red.msb_right);
    printf("\tgreen.offset     = %u\n", scr_var->green.offset);
    printf("\tgreen.length     = %u\n", scr_var->green.length);
    printf("\tgreen.msb_right  = %u\n", scr_var->green.msb_right);
    printf("\tblue.offset      = %u\n", scr_var->blue.offset);
    printf("\tblue.length      = %u\n", scr_var->blue.length);
    printf("\tblue.msb_right   = %u\n", scr_var->blue.msb_right);
    printf("\ttransp.offset    = %u\n", scr_var->transp.offset);
    printf("\ttransp.length    = %u\n", scr_var->transp.length);
    printf("\ttransp.msb_right = %u\n", scr_var->transp.msb_right);
    printf("\tnonstd           = %u\n", scr_var->nonstd);
    printf("\tactivate         = %u\n", scr_var->activate);
    printf("\theight           = %u\n", scr_var->height);
    printf("\twidth            = %u\n", scr_var->width);
    printf("\taccel_flags      = %u\n", scr_var->accel_flags);
    printf("\tpixclock         = %u\n", scr_var->pixclock);
    printf("\tleft_margin      = %u\n", scr_var->left_margin);
    printf("\tright_margin     = %u\n", scr_var->right_margin);
    printf("\tupper_margin     = %u\n", scr_var->upper_margin);
    printf("\tlower_margin     = %u\n", scr_var->lower_margin);
    printf("\thsync_len        = %u\n", scr_var->hsync_len);
    printf("\tvsync_len        = %u\n", scr_var->vsync_len);
    printf("\tsync             = %u\n", scr_var->sync);
    printf("\tvmode            = %u\n", scr_var->vmode);
    printf("\trotate           = %u\n", scr_var->rotate);
    printf("\tcolorspace       = %u\n", scr_var->colorspace);
    printf("\treserved[0]      = %u\n", scr_var->reserved[0]);
    printf("\treserved[1]      = %u\n", scr_var->reserved[1]);
    printf("\treserved[2]      = %u\n", scr_var->reserved[2]);
    printf("\treserved[3]      = %u\n", scr_var->reserved[3]);
}

int main(int argc, char** argv)
{
    const char *fb_device;
    struct fb_fix_screeninfo scr_fix;
    struct fb_var_screeninfo scr_var;
    int fb_fd;

    // parse command line options
    fb_device = "/dev/fb0";
    bool show_info = false;
    bool pan_display = false;

    if (argc < 2)
    {
        usage(argv[0]);
        exit(1);
    }
    int opt;
    while ((opt = getopt(argc, argv, "d:iph")) != -1)
    {
        switch (opt)
        {
            case 'd':
                fb_device = optarg;
                break;
            case 'i':
                show_info = true;
                break;
            case 'p':
                pan_display = true;
                break;
            case 'h':
            default:
                usage(argv[0]);
                exit(1);
        }
    }

    // open the framebuffer device
    fb_fd = open(fb_device, O_RDWR);
    if (fb_fd < 0)
    {
        printf("Unable to open %s.\n", fb_device);
        return 1;
    }

    // call ioctl. retrieve fixed screen info
    if (ioctl(fb_fd, FBIOGET_FSCREENINFO, &scr_fix) < 0)
    {
        printf("Unable to retrieve fixed screen info: %s\n",
            strerror(errno));
        close(fb_fd);
        return 1;
    }

    // print all the fixed screen info values
    if (show_info)
        print_fix_screeninfo(&scr_fix);

    // call ioctl. retrieve variable screen info
    if (ioctl(fb_fd, FBIOGET_VSCREENINFO, &scr_var) < 0)
    {
        printf("Unable to retrieve variable screen info: %s\n",
            strerror(errno));
        close(fb_fd);
        return 1;
    }

    // print all the variable screen info values
    if (show_info)
        print_var_screeninfo(&scr_var);

    if (pan_display)
    {
        // call ioctl. pan the display
        if (ioctl(fb_fd, FBIOPAN_DISPLAY, &scr_var) < 0)
        {
            printf("Unable to pan the display: %s\n",
                strerror(errno));
            close(fb_fd);
            return 1;
        }
    }

    // close the framebuffer device
    close(fb_fd);
}