git.mutantstargoat.com Git - winnie/blob - src/gfx.cc

   1 #include <errno.h>
   2 #include <limits.h>
   3 #include <stdio.h>
   4 #include <string.h>
   5
   6 #include <fcntl.h>
   7 #include <sys/ioctl.h>
   8 #include <sys/mman.h>
   9 #include <unistd.h>
  10
  11 #include <linux/fb.h>
  12
  13 #include "gfx.h"
  14
  15 #define FRAMEBUFFER_SIZE(xsz, ysz, bpp) ((xsz) * (ysz) * (bpp) / CHAR_BIT)
  16
  17 static unsigned char *framebuffer;
  18 static int dev_fd = -1;
  19
  20 static Rect screen_rect;
  21 static int color_depth; //bits per pixel
  22
  23 bool init_gfx()
  24 {
  25         if((dev_fd = open("/dev/fb0", O_RDWR)) == -1) {
  26                 fprintf(stderr, "Cannot open /dev/fb0 : %s\n", strerror(errno));
  27                 return false;
  28         }
  29
  30         fb_var_screeninfo sinfo;
  31         if(ioctl(dev_fd, FBIOGET_VSCREENINFO, &sinfo) == -1) {
  32                 close(dev_fd);
  33                 dev_fd = -1;
  34                 fprintf(stderr, "Unable to get screen info : %s\n", strerror(errno));
  35                 return false;
  36         }
  37
  38         printf("width : %d height : %d\n : bpp : %d\n", sinfo.xres, sinfo.yres, sinfo.bits_per_pixel);
  39         printf("virtual w: %d virtual h: %d\n", sinfo.xres_virtual, sinfo.yres_virtual);
  40
  41         screen_rect.x = screen_rect.y = 0;
  42         screen_rect.width = sinfo.xres_virtual;
  43         screen_rect.height = sinfo.yres_virtual;
  44         color_depth = sinfo.bits_per_pixel;
  45
  46         int sz = FRAMEBUFFER_SIZE(screen_rect.width, screen_rect.height, color_depth);
  47         framebuffer = (unsigned char*)mmap(0, sz, PROT_READ | PROT_WRITE, MAP_SHARED, dev_fd, 0);
  48
  49         if(framebuffer == (void*)-1) {
  50                 close(dev_fd);
  51                 dev_fd = -1;
  52                 fprintf(stderr, "Cannot map the framebuffer to memory : %s\n", strerror(errno));
  53                 return false;
  54         }
  55
  56         return true;
  57 }
  58
  59 void destroy_gfx()
  60 {
  61         clear_screen(0, 0, 0);
  62         close(dev_fd);
  63         dev_fd = -1;
  64
  65         munmap(framebuffer, FRAMEBUFFER_SIZE(screen_rect.width, screen_rect.height, color_depth));
  66         framebuffer = 0;
  67 }
  68
  69 unsigned char *get_framebuffer()
  70 {
  71         return framebuffer;
  72 }
  73
  74 Rect get_screen_size()
  75 {
  76         return screen_rect;
  77 }
  78
  79 int get_color_depth()
  80 {
  81         return color_depth;
  82 }
  83
  84 void clear_screen(int r, int g, int b)
  85 {
  86         fill_rect(screen_rect, r, g, b);
  87 }
  88
  89 void fill_rect(const Rect &rect, int r, int g, int b)
  90 {
  91         unsigned char *fb = framebuffer + (rect.x + screen_rect.width * rect.y) * 4;
  92         for(int i=0; i<rect.height; i++) {
  93                 for(int j=0; j<rect.width; j++) {
  94                         fb[j * 4] = b;
  95                         fb[j * 4 + 1] = g;
  96                         fb[j * 4 + 2] = r;
  97                 }
  98                 fb += screen_rect.width * 4;
  99         }
 100 }
 101
 102 void set_cursor_visibility(bool visible)
 103 {
 104         fb_cursor curs;
 105         curs.enable = visible ? 1 : 0;
 106
 107         if(ioctl(dev_fd, FBIO_CURSOR, &curs) == -1) {
 108                 fprintf(stderr, "Cannot toggle cursor visibility : %s\n", strerror(errno));
 109         }
 110 }