eleni.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
  63         if(dev_fd != -1) {
  64                 close(dev_fd);
  65         }
  66
  67         dev_fd = -1;
  68
  69         munmap(framebuffer, FRAMEBUFFER_SIZE(screen_rect.width, screen_rect.height, color_depth));
  70         framebuffer = 0;
  71 }
  72
  73 unsigned char *get_framebuffer()
  74 {
  75         return framebuffer;
  76 }
  77
  78 Rect get_screen_size()
  79 {
  80         return screen_rect;
  81 }
  82
  83 int get_color_depth()
  84 {
  85         return color_depth;
  86 }
  87
  88 void clear_screen(int r, int g, int b)
  89 {
  90         fill_rect(screen_rect, r, g, b);
  91 }
  92
  93 void fill_rect(const Rect &rect, int r, int g, int b)
  94 {
  95         unsigned char *fb = framebuffer + (rect.x + screen_rect.width * rect.y) * 4;
  96         for(int i=0; i<rect.height; i++) {
  97                 for(int j=0; j<rect.width; j++) {
  98                         fb[j * 4] = b;
  99                         fb[j * 4 + 1] = g;
 100                         fb[j * 4 + 2] = r;
 101                 }
 102                 fb += screen_rect.width * 4;
 103         }
 104 }
 105
 106 void set_cursor_visibility(bool visible)
 107 {
 108         fb_cursor curs;
 109         curs.enable = visible ? 1 : 0;
 110
 111         if(ioctl(dev_fd, FBIO_CURSOR, &curs) == -1) {
 112                 fprintf(stderr, "Cannot toggle cursor visibility : %s\n", strerror(errno));
 113         }
 114 }
 115
 116 void blit(unsigned char *src_img, const Rect &src_rect, unsigned char* dest_img,
 117                 const Rect &dest_rect, int dest_x, int dest_y)
 118 {
 119         int width = src_rect.width;
 120         int height = src_rect.height;
 121
 122         int xoffs = dest_x - dest_rect.x;
 123         if(xoffs < 0) {
 124                 dest_x = dest_rect.x;
 125                 width += xoffs;
 126         }
 127
 128         int yoffs = dest_y - dest_rect.y;
 129         if(yoffs < 0) {
 130                 dest_y = dest_rect.y;
 131                 height += yoffs;
 132         }
 133
 134         int xend = dest_x + width;
 135         if(xend >= dest_rect.width) {
 136                 width -= xend - dest_rect.width;
 137         }
 138
 139         int yend = dest_y + height;
 140         if(yend >= dest_rect.height) {
 141                 height -= yend - dest_rect.height;
 142         }
 143
 144         if(width <= 0 || height <= 0) {
 145                 return;
 146         }
 147
 148         unsigned char *sptr = src_img + (src_rect.y * src_rect.width + src_rect.x) * 4;
 149         unsigned char *dptr = dest_img + (dest_y * dest_rect.width + dest_x) * 4;
 150
 151         for(int i=0; i<height; i++) {
 152                 memcpy(dptr, sptr, width * 4);
 153                 sptr += src_rect.width * 4;
 154                 dptr += dest_rect.width * 4;
 155         }
 156 }
 157
 158 void blit_key(unsigned char *src_img, const Rect &src_rect, unsigned char* dest_img,
 159                 const Rect &dest_rect, int dest_x, int dest_y, int key_r, int key_g, int key_b)
 160 {
 161         int width = src_rect.width;
 162         int height = src_rect.height;
 163
 164         int xoffs = dest_x - dest_rect.x;
 165         if(xoffs < 0) {
 166                 dest_x = dest_rect.x;
 167                 width += xoffs;
 168         }
 169
 170         int yoffs = dest_y - dest_rect.y;
 171         if(yoffs < 0) {
 172                 dest_y = dest_rect.y;
 173                 height += yoffs;
 174         }
 175
 176         int xend = dest_x + width;
 177         if(xend >= dest_rect.width) {
 178                 width -= xend - dest_rect.width;
 179         }
 180
 181         int yend = dest_y + height;
 182         if(yend >= dest_rect.height) {
 183                 height -= yend - dest_rect.height;
 184         }
 185
 186         if(width <= 0 || height <= 0) {
 187                 return;
 188         }
 189
 190         unsigned char *sptr = src_img + (src_rect.y * src_rect.width + src_rect.x) * 4;
 191         unsigned char *dptr = dest_img + (dest_y * dest_rect.width + dest_x) * 4;
 192
 193         for(int i=0; i<height; i++) {
 194                 for(int j=0; j<width; j++) {
 195                         int r = sptr[j * 4];
 196                         int g = sptr[j * 4 + 1];
 197                         int b = sptr[j * 4 + 2];
 198
 199                         if(r != key_r || g != key_g || b != key_b) {
 200                                 dptr[j * 4] = r;
 201                                 dptr[j * 4 + 1] = g;
 202                                 dptr[j * 4 + 2] = b;
 203                         }
 204                 }
 205
 206                 sptr += src_rect.width * 4;
 207                 dptr += dest_rect.width * 4;
 208         }
 209 }