src/screen.cpp

   1 #include "common.h"
   2 #include "screen.h"
   3
   4 #include "SDL_opengl.h"
   5 #include <fcntl.h> // for access(2)
   6 #include <unistd.h> // for access(2)
   7
   8 using namespace IPDF;
   9 using namespace std;
  10
  11 Screen::Screen()
  12 {
  13         SDL_Init(SDL_INIT_VIDEO);
  14         m_window = SDL_CreateWindow("IPDF", SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED,
  15                         800, 600, SDL_WINDOW_OPENGL | SDL_WINDOW_RESIZABLE);
  16
  17         if (!m_window)
  18         {
  19                 Fatal("Couldn't create window!");
  20         }
  21
  22         m_gl_context = SDL_GL_CreateContext(m_window);
  23
  24         ResizeViewport(800, 600);
  25
  26         Clear();
  27         Present();
  28
  29
  30 }
  31
  32 Screen::~Screen()
  33 {
  34         SDL_GL_DeleteContext(m_gl_context);
  35         SDL_DestroyWindow(m_window);
  36         SDL_Quit();
  37 }
  38
  39 void Screen::Clear(float r, float g, float b, float a)
  40 {
  41         glClearColor(r,g,b,a);
  42         glClear(GL_COLOR_BUFFER_BIT);
  43         DebugFontClear();
  44 }
  45
  46 void Screen::ResizeViewport(int width, int height)
  47 {
  48         glViewport(0, 0, width, height);
  49         m_viewport_width = width;
  50         m_viewport_height = height;
  51 }
  52
  53 bool Screen::PumpEvents()
  54 {
  55         SDL_Event evt;
  56         bool no_quit_requested = true;
  57         while (SDL_PollEvent(&evt))
  58         {
  59                 switch (evt.type)
  60                 {
  61                 case SDL_QUIT:
  62                         no_quit_requested = false;
  63                         break;
  64                 case SDL_WINDOWEVENT:
  65                         switch (evt.window.event)
  66                         {
  67                         case SDL_WINDOWEVENT_RESIZED:
  68                         case SDL_WINDOWEVENT_SIZE_CHANGED:
  69                                 ResizeViewport(evt.window.data1, evt.window.data2);
  70                                 break;
  71                         }
  72                         break;
  73                 case SDL_MOUSEMOTION:
  74                         m_last_mouse_x = evt.motion.x;
  75                         m_last_mouse_y = evt.motion.y;
  76                         if (m_mouse_handler)
  77                         {
  78                                 m_mouse_handler(evt.motion.x, evt.motion.y,evt.motion.state, 0);
  79                         }
  80                         break;
  81                 case SDL_MOUSEBUTTONDOWN:
  82                 case SDL_MOUSEBUTTONUP:
  83                         m_last_mouse_x = evt.button.x;
  84                         m_last_mouse_y = evt.button.y;
  85                         if (m_mouse_handler)
  86                         {
  87                                 m_mouse_handler(evt.button.x, evt.button.y, evt.button.state?evt.button.button:0, 0);
  88                         }
  89                         break;
  90                 case SDL_MOUSEWHEEL:
  91                         if (m_mouse_handler)
  92                         {
  93                                 m_mouse_handler(m_last_mouse_x, m_last_mouse_y, 0, evt.wheel.y);
  94                         }
  95                         break;
  96                 case SDL_KEYDOWN:
  97                 {
  98                         Debug("Key %c down", (char)evt.key.keysym.sym);
  99                         if (isalnum((char)evt.key.keysym.sym))
 100                         {
 101                                 char filename[] = "0.bmp";
 102                                 filename[0] = (char)evt.key.keysym.sym;
 103                                 ScreenShot(filename);
 104                         }
 105                         break;
 106                 }
 107                 default:
 108                         break;
 109                 }
 110         }
 111         return no_quit_requested;
 112 }
 113
 114 void Screen::SetMouseCursor(Screen::MouseCursors cursor)
 115 {
 116         SDL_SystemCursor system_cursor_id = SDL_SYSTEM_CURSOR_ARROW;
 117         switch (cursor)
 118         {
 119         case CursorArrow: system_cursor_id = SDL_SYSTEM_CURSOR_ARROW; break;
 120         case CursorWait: system_cursor_id = SDL_SYSTEM_CURSOR_WAIT; break;
 121         case CursorWaitArrow: system_cursor_id = SDL_SYSTEM_CURSOR_WAITARROW; break;
 122         case CursorMove: system_cursor_id = SDL_SYSTEM_CURSOR_SIZEALL; break;
 123         case CursorHand: system_cursor_id = SDL_SYSTEM_CURSOR_HAND; break;
 124         default: break;
 125         }
 126         SDL_Cursor *system_cursor = SDL_CreateSystemCursor(system_cursor_id);
 127         SDL_SetCursor(system_cursor);
 128         //TODO: Check if we need to free the system cursors.
 129 }
 130
 131 void Screen::Present()
 132 {
 133         SDL_GL_SwapWindow(m_window);
 134 }
 135
 136 void Screen::ScreenShot(const char * filename) const
 137 {
 138         Debug("Attempting to save BMP to file %s", filename);
 139
 140         int w = ViewportWidth();
 141         int h = ViewportHeight();
 142         unsigned char * pixels = new unsigned char[w*h*4];
 143         if (pixels == NULL)
 144                 Fatal("Failed to allocate %d x %d x 4 = %d pixel array", w, h, w*h*4);
 145
 146         for (int y = 0; y < h; ++y)
 147         {
 148                 glReadPixels(0,h-y-1,w, 1, GL_RGBA, GL_UNSIGNED_BYTE, &pixels[y*w*4]);
 149         }
 150
 151 #if SDL_BYTEORDER == SDL_LIL_ENDIAN
 152         SDL_Surface * surf = SDL_CreateRGBSurfaceFrom(pixels, w, h, 8*4, w*4, 0x000000ff,0x0000ff00,0x00ff0000,0xff000000);
 153 #else
 154         SDL_Surface * surf = SDL_CreateRGBSurfaceFrom(pixels, w, h, 8*4, w*4, 0xff000000,0x00ff0000,0x0000ff00,0x000000ff);
 155 #endif
 156         if (surf == NULL)
 157                 Fatal("Failed to create SDL_Surface from pixel data - %s", SDL_GetError());
 158
 159         GLenum texture_format = (surf->format->Rmask == 0x000000FF) ? GL_RGBA : GL_BGRA;
 160         Debug("SDL_Surface %d BytesPerPixel, format %d (RGB = %d, BGR = %d, RGBA = %d, BGRA = %d)", surf->format->BytesPerPixel, texture_format, GL_RGB, GL_BGR, GL_RGBA, GL_BGRA);
 161
 162         if (SDL_SaveBMP(surf, filename) != 0)
 163                 Fatal("SDL_SaveBMP failed - %s", SDL_GetError());
 164
 165         SDL_FreeSurface(surf);
 166         delete [] pixels;
 167         Debug("Succeeded!");
 168 }
 169
 170 /**
 171  * Render a BMP
 172  * NOT PART OF THE DOCUMENT FORMAT
 173  */
 174 void Screen::RenderBMP(const char * filename) const
 175 {
 176         if (access(filename, R_OK) == -1)
 177         {
 178                 Error("No such file \"%s\" - Nothing to render - You might have done this deliberately?", filename);
 179                 return;
 180         }
 181         SDL_Surface * bmp = SDL_LoadBMP(filename);
 182         if (bmp == NULL)
 183                 Fatal("Failed to load BMP from %s - %s", filename, SDL_GetError());
 184
 185         int w = bmp->w;
 186         int h = bmp->h;
 187
 188         GLenum texture_format;
 189         switch (bmp->format->BytesPerPixel)
 190         {
 191                 case 4: //contains alpha
 192                         texture_format = (bmp->format->Rmask == 0x000000FF) ? GL_RGBA : GL_BGRA;
 193                         break;
 194                 case 3: //does not contain alpha
 195                         texture_format = (bmp->format->Rmask == 0x000000FF) ? GL_RGB : GL_BGR;
 196                         break;
 197                 default:
 198                         Fatal("Could not understand SDL_Surface format (%d colours)", bmp->format->BytesPerPixel);
 199                         break;
 200         }
 201
 202         //Debug("SDL_Surface %d BytesPerPixel, format %d (RGB = %d, BGR = %d, RGBA = %d, BGRA = %d)", bmp->format->BytesPerPixel, texture_format, GL_RGB, GL_BGR, GL_RGBA, GL_BGRA);
 203
 204
 205         GLuint texID;
 206         glEnable(GL_TEXTURE_2D);
 207         glGenTextures(1, &texID);
 208         glBindTexture(GL_TEXTURE_2D, texID);
 209
 210         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
 211         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 212         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
 213         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
 214
 215         glTexImage2D(GL_TEXTURE_2D, 0, bmp->format->BytesPerPixel, w, h, 0, texture_format, GL_UNSIGNED_BYTE, bmp->pixels);
 216
 217         glMatrixMode(GL_PROJECTION);
 218         glLoadIdentity();
 219         glOrtho(0.0, 1.0, 1.0, 0.0, -1.f, 1.f);
 220         glMatrixMode(GL_MODELVIEW);
 221         glLoadIdentity();
 222
 223         glBegin(GL_QUADS);
 224                 glTexCoord2i(0,0); glVertex2f(0,0);
 225                 glTexCoord2i(1,0); glVertex2f(1,0);
 226                 glTexCoord2i(1,1); glVertex2f(1,1);
 227                 glTexCoord2i(0,1); glVertex2f(0,1);
 228         glEnd();
 229
 230         glDisable(GL_TEXTURE_2D);
 231         SDL_FreeSurface(bmp);
 232 }
 233
 234 void Screen::DebugFontInit(const char *name, float font_size)
 235 {
 236         unsigned char font_atlas_data[1024*1024];
 237         FILE *font_file = fopen(name, "rb");
 238         fseek(font_file, 0, SEEK_END);
 239         size_t font_file_size = ftell(font_file);
 240         fseek(font_file, 0, SEEK_SET);
 241         unsigned char *font_file_data = (unsigned char*)malloc(font_file_size);
 242         fread(font_file_data, 1, font_file_size, font_file);
 243         fclose(font_file);
 244         stbtt_BakeFontBitmap(font_file_data,0, font_size, font_atlas_data,1024,1024, 32,96, m_debug_font_rects);
 245         free(font_file_data);
 246         glGenTextures(1, &m_debug_font_atlas);
 247         glBindTexture(GL_TEXTURE_2D, m_debug_font_atlas);
 248         glTexImage2D(GL_TEXTURE_2D, 0, GL_ALPHA, 1024,1024, 0, GL_ALPHA, GL_UNSIGNED_BYTE, font_atlas_data);
 249         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 250         m_debug_font_size = font_size;
 251 }
 252
 253 void Screen::DebugFontClear()
 254 {
 255         m_debug_font_x = m_debug_font_y = 0;
 256         DebugFontPrint("\n");
 257 }
 258
 259 void Screen::DebugFontPrint(const char* str)
 260 {
 261         glMatrixMode(GL_PROJECTION);
 262         glPushMatrix();
 263         glLoadIdentity();
 264         glOrtho(0,ViewportWidth(), ViewportHeight(), 0, -1, 1);
 265         glMatrixMode(GL_MODELVIEW);
 266         glPushMatrix();
 267         glLoadIdentity();
 268
 269
 270
 271         glEnable(GL_TEXTURE_2D);
 272         glEnable(GL_BLEND);
 273         glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
 274         glBindTexture(GL_TEXTURE_2D, m_debug_font_atlas);
 275         glBegin(GL_QUADS);
 276         while (*str) {
 277                 if (*str >= 32 && *str < 128) {
 278                         stbtt_aligned_quad q;
 279                         stbtt_GetBakedQuad(m_debug_font_rects, 1024,1024, *str-32, &m_debug_font_x,&m_debug_font_y,&q,1);
 280                         glTexCoord2f(q.s0,q.t0); glVertex2f(q.x0,q.y0);
 281                         glTexCoord2f(q.s1,q.t0); glVertex2f(q.x1,q.y0);
 282                         glTexCoord2f(q.s1,q.t1); glVertex2f(q.x1,q.y1);
 283                         glTexCoord2f(q.s0,q.t1); glVertex2f(q.x0,q.y1);
 284                 }
 285                 else if (*str == '\n')
 286                 {
 287                         m_debug_font_x = 0;
 288                         m_debug_font_y += m_debug_font_size;
 289                 }
 290                 ++str;
 291         }
 292         glEnd();
 293         glDisable(GL_BLEND);
 294         glDisable(GL_TEXTURE_2D);
 295         glPopMatrix();
 296         glMatrixMode(GL_PROJECTION);
 297         glPopMatrix();
 298 }
 299
 300 void Screen::DebugFontPrintF(const char *fmt, ...)
 301 {
 302         char buffer[BUFSIZ];
 303         va_list va;
 304         va_start(va, fmt);
 305         vsnprintf(buffer, BUFSIZ, fmt,va);
 306         va_end(va);
 307         DebugFontPrint(buffer);
 308 }