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