Bunch-a-bugfixes!

[ipdf/code.git] / src / screen.h

diff --git a/src/screen.h b/src/screen.h
index 4d92cb3..44dd2c7 100644 (file)
--- a/src/screen.h
+++ b/src/screen.h
@@ -20,12 +20,15 @@ namespace IPDF
                // Returns 'false' if the program should quit.
                bool PumpEvents();
 
+               // Clears the screen to a given colour.
+               void Clear(float r=1.0, float g=1.0, float b=1.0, float a=1.0);
+
                // Finishes rendering a frame, and presents it on the screen.
                void Present();
 
                // Get the current width/height of the window's viewport.
-               int ViewportWidth() { return m_viewport_width; }
-               int ViewportHeight() { return m_viewport_height; }
+               int ViewportWidth() const { return m_viewport_width; }
+               int ViewportHeight() const { return m_viewport_height; }
                
                // Handle mouse input.
                typedef std::function<void(int x, int y, int button, int wheel)> MouseHandler;
@@ -43,6 +46,9 @@ namespace IPDF
                        CursorHand
                };
                void SetMouseCursor(MouseCursors cursor);
+
+               void ScreenShot(const char * filename) const;
+               void RenderBMP(const char * filename) const;
        private:
                void ResizeViewport(int width, int height);