src/screen.cpp

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