samples/opengl/penguin/penguin.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        penguin.cpp
   3 // Purpose:     wxGLCanvas demo program
   4 // Author:      Robert Roebling
   5 // Modified by: Sandro Sigala
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Robert Roebling
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // For compilers that support precompilation, includes "wx.h".
  13 #include "wx/wxprec.h"
  14
  15 #ifdef __BORLANDC__
  16 #pragma hdrstop
  17 #endif
  18
  19 #ifndef WX_PRECOMP
  20 #include "wx/wx.h"
  21 #endif
  22
  23 #if !wxUSE_GLCANVAS
  24     #error "OpenGL required: set wxUSE_GLCANVAS to 1 and rebuild the library"
  25 #endif
  26
  27 #include "penguin.h"
  28 #ifdef __DARWIN__
  29     #include <OpenGL/glu.h>
  30 #else
  31     #include <GL/glu.h>
  32 #endif
  33
  34 #include "../../sample.xpm"
  35
  36 // ---------------------------------------------------------------------------
  37 // MyApp
  38 // ---------------------------------------------------------------------------
  39
  40 // `Main program' equivalent, creating windows and returning main app frame
  41 bool MyApp::OnInit()
  42 {
  43     if ( !wxApp::OnInit() )
  44         return false;
  45
  46     // Create the main frame window
  47     MyFrame *frame = new MyFrame(NULL, wxT("wxWidgets Penguin Sample"),
  48         wxDefaultPosition, wxDefaultSize);
  49
  50 #if wxUSE_ZLIB
  51     if (wxFileExists(wxT("penguin.dxf.gz")))
  52         frame->GetCanvas()->LoadDXF(wxT("penguin.dxf.gz"));
  53 #else
  54     if (wxFileExists(wxT("penguin.dxf")))
  55         frame->GetCanvas()->LoadDXF(wxT("penguin.dxf"));
  56 #endif
  57
  58     /* Show the frame */
  59     frame->Show(true);
  60
  61     return true;
  62 }
  63
  64 IMPLEMENT_APP(MyApp)
  65
  66 // ---------------------------------------------------------------------------
  67 // MyFrame
  68 // ---------------------------------------------------------------------------
  69
  70 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
  71     EVT_MENU(wxID_OPEN, MyFrame::OnMenuFileOpen)
  72     EVT_MENU(wxID_EXIT, MyFrame::OnMenuFileExit)
  73     EVT_MENU(wxID_HELP, MyFrame::OnMenuHelpAbout)
  74 END_EVENT_TABLE()
  75
  76 // MyFrame constructor
  77 MyFrame::MyFrame(wxFrame *frame, const wxString& title, const wxPoint& pos,
  78     const wxSize& size, long style)
  79     : wxFrame(frame, wxID_ANY, title, pos, size, style)
  80 {
  81     SetIcon(wxICON(sample));
  82
  83     // Make the "File" menu
  84     wxMenu *fileMenu = new wxMenu;
  85     fileMenu->Append(wxID_OPEN, wxT("&Open..."));
  86     fileMenu->AppendSeparator();
  87     fileMenu->Append(wxID_EXIT, wxT("E&xit\tALT-X"));
  88     // Make the "Help" menu
  89     wxMenu *helpMenu = new wxMenu;
  90     helpMenu->Append(wxID_HELP, wxT("&About"));
  91
  92     wxMenuBar *menuBar = new wxMenuBar;
  93     menuBar->Append(fileMenu, wxT("&File"));
  94     menuBar->Append(helpMenu, wxT("&Help"));
  95     SetMenuBar(menuBar);
  96
  97     Show(true);
  98
  99     m_canvas = new TestGLCanvas(this, wxID_ANY, wxDefaultPosition,
 100         GetClientSize(), wxSUNKEN_BORDER);
 101 }
 102
 103 // File|Open... command
 104 void MyFrame::OnMenuFileOpen( wxCommandEvent& WXUNUSED(event) )
 105 {
 106     wxString filename = wxFileSelector(wxT("Choose DXF Model"), wxT(""), wxT(""), wxT(""),
 107 #if wxUSE_ZLIB
 108         wxT("DXF Drawing (*.dxf;*.dxf.gz)|*.dxf;*.dxf.gz|All files (*.*)|*.*"),
 109 #else
 110         wxT("DXF Drawing (*.dxf)|*.dxf)|All files (*.*)|*.*"),
 111 #endif
 112         wxFD_OPEN);
 113     if (!filename.IsEmpty())
 114     {
 115         m_canvas->LoadDXF(filename);
 116         m_canvas->Refresh(false);
 117     }
 118 }
 119
 120 // File|Exit command
 121 void MyFrame::OnMenuFileExit( wxCommandEvent& WXUNUSED(event) )
 122 {
 123     // true is to force the frame to close
 124     Close(true);
 125 }
 126
 127 // Help|About command
 128 void MyFrame::OnMenuHelpAbout( wxCommandEvent& WXUNUSED(event) )
 129 {
 130     wxMessageBox(wxT("OpenGL Penguin Sample (c) Robert Roebling, Sandro Sigala et al"));
 131 }
 132
 133 // ---------------------------------------------------------------------------
 134 // TestGLCanvas
 135 // ---------------------------------------------------------------------------
 136
 137 BEGIN_EVENT_TABLE(TestGLCanvas, wxGLCanvas)
 138     EVT_SIZE(TestGLCanvas::OnSize)
 139     EVT_PAINT(TestGLCanvas::OnPaint)
 140     EVT_ERASE_BACKGROUND(TestGLCanvas::OnEraseBackground)
 141     EVT_MOUSE_EVENTS(TestGLCanvas::OnMouse)
 142 END_EVENT_TABLE()
 143
 144 TestGLCanvas::TestGLCanvas(wxWindow *parent,
 145                            wxWindowID id,
 146                            const wxPoint& pos,
 147                            const wxSize& size,
 148                            long style,
 149                            const wxString& name)
 150     : wxGLCanvas(parent, id, NULL, pos, size,
 151                  style | wxFULL_REPAINT_ON_RESIZE, name)
 152 {
 153     // Explicitly create a new rendering context instance for this canvas.
 154     m_glRC = new wxGLContext(this);
 155
 156     m_gldata.initialized = false;
 157
 158     // initialize view matrix
 159     m_gldata.beginx = 0.0f;
 160     m_gldata.beginy = 0.0f;
 161     m_gldata.zoom   = 45.0f;
 162     trackball(m_gldata.quat, 0.0f, 0.0f, 0.0f, 0.0f);
 163 }
 164
 165 TestGLCanvas::~TestGLCanvas()
 166 {
 167     delete m_glRC;
 168 }
 169
 170 void TestGLCanvas::OnPaint( wxPaintEvent& WXUNUSED(event) )
 171 {
 172     // must always be here
 173     wxPaintDC dc(this);
 174
 175     SetCurrent(*m_glRC);
 176
 177     // Initialize OpenGL
 178     if (!m_gldata.initialized)
 179     {
 180         InitGL();
 181         ResetProjectionMode();
 182         m_gldata.initialized = true;
 183     }
 184
 185     // Clear
 186     glClearColor( 0.3f, 0.4f, 0.6f, 1.0f );
 187     glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT );
 188
 189     // Transformations
 190     glLoadIdentity();
 191     glTranslatef( 0.0f, 0.0f, -20.0f );
 192     GLfloat m[4][4];
 193     build_rotmatrix( m, m_gldata.quat );
 194     glMultMatrixf( &m[0][0] );
 195
 196     m_renderer.Render();
 197
 198     // Flush
 199     glFlush();
 200
 201     // Swap
 202     SwapBuffers();
 203 }
 204
 205 void TestGLCanvas::OnSize(wxSizeEvent& WXUNUSED(event))
 206 {
 207     // Reset the OpenGL view aspect.
 208     // This is OK only because there is only one canvas that uses the context.
 209     // See the cube sample for that case that multiple canvases are made current with one context.
 210     ResetProjectionMode();
 211 }
 212
 213 void TestGLCanvas::OnEraseBackground(wxEraseEvent& WXUNUSED(event))
 214 {
 215     // Do nothing, to avoid flashing on MSW
 216 }
 217
 218 // Load the DXF file.  If the zlib support is compiled in wxWidgets,
 219 // supports also the ".dxf.gz" gzip compressed files.
 220 void TestGLCanvas::LoadDXF(const wxString& filename)
 221 {
 222     wxFileInputStream stream(filename);
 223     if (stream.IsOk())
 224 #if wxUSE_ZLIB
 225     {
 226         if (filename.Right(3).Lower() == wxT(".gz"))
 227         {
 228             wxZlibInputStream zstream(stream);
 229             m_renderer.Load(zstream);
 230         } else
 231         {
 232             m_renderer.Load(stream);
 233         }
 234     }
 235 #else
 236     {
 237         m_renderer.Load(stream);
 238     }
 239 #endif
 240 }
 241
 242 void TestGLCanvas::OnMouse(wxMouseEvent& event)
 243 {
 244     if (event.Dragging())
 245     {
 246         wxSize sz(GetClientSize());
 247
 248         /* drag in progress, simulate trackball */
 249         float spin_quat[4];
 250         trackball(spin_quat,
 251             (2.0*m_gldata.beginx - sz.x) / sz.x,
 252             (sz.y - 2.0*m_gldata.beginy) / sz.y,
 253             (2.0*event.GetX() - sz.x)    / sz.x,
 254             (sz.y - 2.0*event.GetY())    / sz.y);
 255
 256         add_quats(spin_quat, m_gldata.quat, m_gldata.quat);
 257
 258         /* orientation has changed, redraw mesh */
 259         Refresh(false);
 260     }
 261
 262     m_gldata.beginx = event.GetX();
 263     m_gldata.beginy = event.GetY();
 264 }
 265
 266 void TestGLCanvas::InitGL()
 267 {
 268     static const GLfloat light0_pos[4]   = { -50.0f, 50.0f, 0.0f, 0.0f };
 269
 270     // white light
 271     static const GLfloat light0_color[4] = { 0.6f, 0.6f, 0.6f, 1.0f };
 272
 273     static const GLfloat light1_pos[4]   = {  50.0f, 50.0f, 0.0f, 0.0f };
 274
 275     // cold blue light
 276     static const GLfloat light1_color[4] = { 0.4f, 0.4f, 1.0f, 1.0f };
 277
 278     /* remove back faces */
 279     glDisable(GL_CULL_FACE);
 280     glEnable(GL_DEPTH_TEST);
 281
 282     /* speedups */
 283     glEnable(GL_DITHER);
 284     glShadeModel(GL_SMOOTH);
 285     glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_FASTEST);
 286     glHint(GL_POLYGON_SMOOTH_HINT, GL_FASTEST);
 287
 288     /* light */
 289     glLightfv(GL_LIGHT0, GL_POSITION, light0_pos);
 290     glLightfv(GL_LIGHT0, GL_DIFFUSE,  light0_color);
 291     glLightfv(GL_LIGHT1, GL_POSITION, light1_pos);
 292     glLightfv(GL_LIGHT1, GL_DIFFUSE,  light1_color);
 293     glEnable(GL_LIGHT0);
 294     glEnable(GL_LIGHT1);
 295     glEnable(GL_LIGHTING);
 296
 297     glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE);
 298     glEnable(GL_COLOR_MATERIAL);
 299 }
 300
 301 void TestGLCanvas::ResetProjectionMode()
 302 {
 303     if ( !IsShownOnScreen() )
 304         return;
 305
 306     // This is normally only necessary if there is more than one wxGLCanvas
 307     // or more than one wxGLContext in the application.
 308     SetCurrent(*m_glRC);
 309
 310     int w, h;
 311     GetClientSize(&w, &h);
 312
 313     // It's up to the application code to update the OpenGL viewport settings.
 314     // In order to avoid extensive context switching, consider doing this in
 315     // OnPaint() rather than here, though.
 316     glViewport(0, 0, (GLint) w, (GLint) h);
 317
 318     glMatrixMode(GL_PROJECTION);
 319     glLoadIdentity();
 320     gluPerspective(45.0f, (GLfloat)w/h, 1.0, 100.0);
 321     glMatrixMode(GL_MODELVIEW);
 322     glLoadIdentity();
 323 }