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 }
  44
  45 void Screen::ResizeViewport(int width, int height)
  46 {
  47         glViewport(0, 0, width, height);
  48         m_viewport_width = width;
  49         m_viewport_height = height;
  50 }
  51
  52 bool Screen::PumpEvents()
  53 {
  54         SDL_Event evt;
  55         bool no_quit_requested = true;
  56         while (SDL_PollEvent(&evt))
  57         {
  58                 switch (evt.type)
  59                 {
  60                 case SDL_QUIT:
  61                         no_quit_requested = false;
  62                         break;
  63                 case SDL_WINDOWEVENT:
  64                         switch (evt.window.event)
  65                         {
  66                         case SDL_WINDOWEVENT_RESIZED:
  67                         case SDL_WINDOWEVENT_SIZE_CHANGED:
  68                                 ResizeViewport(evt.window.data1, evt.window.data2);
  69                                 break;
  70                         }
  71                         break;
  72                 case SDL_MOUSEMOTION:
  73                         m_last_mouse_x = evt.motion.x;
  74                         m_last_mouse_y = evt.motion.y;
  75                         if (m_mouse_handler)
  76                         {
  77                                 m_mouse_handler(evt.motion.x, evt.motion.y,evt.motion.state, 0);
  78                         }
  79                         break;
  80                 case SDL_MOUSEBUTTONDOWN:
  81                 case SDL_MOUSEBUTTONUP:
  82                         m_last_mouse_x = evt.button.x;
  83                         m_last_mouse_y = evt.button.y;
  84                         if (m_mouse_handler)
  85                         {
  86                                 m_mouse_handler(evt.button.x, evt.button.y, evt.button.state, 0);
  87                         }
  88                         break;
  89                 case SDL_MOUSEWHEEL:
  90                         if (m_mouse_handler)
  91                         {
  92                                 m_mouse_handler(m_last_mouse_x, m_last_mouse_y, 0, evt.wheel.y);
  93                         }
  94                         break;
  95                 case SDL_KEYDOWN:
  96                 {
  97                         Debug("Key %c down", (char)evt.key.keysym.sym);
  98                         if (isalnum((char)evt.key.keysym.sym))
  99                         {
 100                                 char filename[] = "0.bmp";
 101                                 filename[0] = (char)evt.key.keysym.sym;
 102                                 ScreenShot(filename);
 103                         }
 104                 }
 105                 default:
 106                         break;
 107                 }
 108         }
 109         return no_quit_requested;
 110 }
 111
 112 void Screen::SetMouseCursor(Screen::MouseCursors cursor)
 113 {
 114         SDL_SystemCursor system_cursor_id = SDL_SYSTEM_CURSOR_ARROW;
 115         switch (cursor)
 116         {
 117         case CursorArrow: system_cursor_id = SDL_SYSTEM_CURSOR_ARROW; break;
 118         case CursorWait: system_cursor_id = SDL_SYSTEM_CURSOR_WAIT; break;
 119         case CursorWaitArrow: system_cursor_id = SDL_SYSTEM_CURSOR_WAITARROW; break;
 120         case CursorMove: system_cursor_id = SDL_SYSTEM_CURSOR_SIZEALL; break;
 121         case CursorHand: system_cursor_id = SDL_SYSTEM_CURSOR_HAND; break;
 122         default: break;
 123         }
 124         SDL_Cursor *system_cursor = SDL_CreateSystemCursor(system_cursor_id);
 125         SDL_SetCursor(system_cursor);
 126         //TODO: Check if we need to free the system cursors.
 127 }
 128
 129 void Screen::Present()
 130 {
 131         SDL_GL_SwapWindow(m_window);
 132 }
 133
 134 void Screen::ScreenShot(const char * filename) const
 135 {
 136         Debug("Attempting to save BMP to file %s", filename);
 137
 138         int w = ViewportWidth();
 139         int h = ViewportHeight();
 140         unsigned char * pixels = new unsigned char[w*h*4];
 141         if (pixels == NULL)
 142                 Fatal("Failed to allocate %d x %d x 4 = %d pixel array", w, h, w*h*4);
 143
 144
 145         for (int y = 0; y < h; ++y)
 146         {
 147                 glReadPixels(0,h-y-1,w, 1, GL_RGBA, GL_UNSIGNED_BYTE, &pixels[y*w*4]);
 148         }
 149
 150 #if SDL_BYTEORDER == SDL_LIL_ENDIAN
 151         SDL_Surface * surf = SDL_CreateRGBSurfaceFrom(pixels, w, h, 8*4, w*4, 0x000000ff,0x0000ff00,0x00ff0000,0xff000000);
 152 #else
 153         SDL_Surface * surf = SDL_CreateRGBSurfaceFrom(pixels, w, h, 8*4, w*4, 0xff000000,0x00ff0000,0x0000ff00,0x000000ff);
 154 #endif
 155         if (surf == NULL)
 156                 Fatal("Failed to create SDL_Surface from pixel data - %s", SDL_GetError());
 157
 158         GLenum texture_format = (surf->format->Rmask == 0x000000FF) ? GL_RGBA : GL_BGRA;
 159         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);
 160
 161         if (SDL_SaveBMP(surf, filename) != 0)
 162                 Fatal("SDL_SaveBMP failed - %s", SDL_GetError());
 163
 164         SDL_FreeSurface(surf);
 165         delete [] pixels;
 166         Debug("Succeeded!");
 167 }
 168
 169 /**
 170  * Render a BMP
 171  * NOT PART OF THE DOCUMENT FORMAT
 172  */
 173 void Screen::RenderBMP(const char * filename) const
 174 {
 175         if (access(filename, R_OK) == -1)
 176         {
 177                 Error("No such file \"%s\" - Nothing to render - You might have done this deliberately?", filename);
 178                 return;
 179         }
 180         SDL_Surface * bmp = SDL_LoadBMP(filename);
 181         if (bmp == NULL)
 182                 Fatal("Failed to load BMP from %s - %s", filename, SDL_GetError());
 183
 184         int w = bmp->w;
 185         int h = bmp->h;
 186
 187         GLenum texture_format;
 188         switch (bmp->format->BytesPerPixel)
 189         {
 190                 case 4: //contains alpha
 191                         texture_format = (bmp->format->Rmask == 0x000000FF) ? GL_RGBA : GL_BGRA;
 192                         break;
 193                 case 3: //does not contain alpha
 194                         texture_format = (bmp->format->Rmask == 0x000000FF) ? GL_RGB : GL_BGR;
 195                         break;
 196                 default:
 197                         Fatal("Could not understand SDL_Surface format (%d colours)", bmp->format->BytesPerPixel);
 198                         break;
 199         }
 200
 201         //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);
 202
 203
 204         GLuint texID;
 205         glEnable(GL_TEXTURE_2D);
 206         glGenTextures(1, &texID);
 207         glBindTexture(GL_TEXTURE_2D, texID);
 208
 209         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
 210         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 211         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
 212         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
 213
 214         glTexImage2D(GL_TEXTURE_2D, 0, bmp->format->BytesPerPixel, w, h, 0, texture_format, GL_UNSIGNED_BYTE, bmp->pixels);
 215
 216         glMatrixMode(GL_PROJECTION);
 217         glLoadIdentity();
 218         glOrtho(0.0, 1.0, 1.0, 0.0, -1.f, 1.f);
 219         glMatrixMode(GL_MODELVIEW);
 220         glLoadIdentity();
 221
 222         glBegin(GL_QUADS);
 223                 glTexCoord2i(0,0); glVertex2f(0,0);
 224                 glTexCoord2i(1,0); glVertex2f(1,0);
 225                 glTexCoord2i(1,1); glVertex2f(1,1);
 226                 glTexCoord2i(0,1); glVertex2f(0,1);
 227         glEnd();
 228
 229         glDisable(GL_TEXTURE_2D);
 230         SDL_FreeSurface(bmp);
 231 }