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_xpm));
  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     // Make the new context current (activate it for use) with this canvas.
 157     SetCurrent(*m_glRC);
 158
 159     m_gldata.initialized = false;
 160
 161     // initialize view matrix
 162     m_gldata.beginx = 0.0f;
 163     m_gldata.beginy = 0.0f;
 164     m_gldata.zoom   = 45.0f;
 165     trackball(m_gldata.quat, 0.0f, 0.0f, 0.0f, 0.0f);
 166 }
 167
 168 TestGLCanvas::~TestGLCanvas()
 169 {
 170     delete m_glRC;
 171 }
 172
 173 void TestGLCanvas::OnPaint( wxPaintEvent& WXUNUSED(event) )
 174 {
 175     // must always be here
 176     wxPaintDC dc(this);
 177
 178     SetCurrent(*m_glRC);
 179
 180     // Initialize OpenGL
 181     if (!m_gldata.initialized)
 182     {
 183         InitGL();
 184         ResetProjectionMode();
 185         m_gldata.initialized = true;
 186     }
 187
 188     // Clear
 189     glClearColor( 0.3f, 0.4f, 0.6f, 1.0f );
 190     glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT );
 191
 192     // Transformations
 193     glLoadIdentity();
 194     glTranslatef( 0.0f, 0.0f, -20.0f );
 195     GLfloat m[4][4];
 196     build_rotmatrix( m, m_gldata.quat );
 197     glMultMatrixf( &m[0][0] );
 198
 199     m_renderer.Render();
 200
 201     // Flush
 202     glFlush();
 203
 204     // Swap
 205     SwapBuffers();
 206 }
 207
 208 void TestGLCanvas::OnSize(wxSizeEvent& WXUNUSED(event))
 209 {
 210     // Reset the OpenGL view aspect.
 211     // This is OK only because there is only one canvas that uses the context.
 212     // See the cube sample for that case that multiple canvases are made current with one context.
 213     ResetProjectionMode();
 214 }
 215
 216 void TestGLCanvas::OnEraseBackground(wxEraseEvent& WXUNUSED(event))
 217 {
 218     // Do nothing, to avoid flashing on MSW
 219 }
 220
 221 // Load the DXF file.  If the zlib support is compiled in wxWidgets,
 222 // supports also the ".dxf.gz" gzip compressed files.
 223 void TestGLCanvas::LoadDXF(const wxString& filename)
 224 {
 225     wxFileInputStream stream(filename);
 226     if (stream.Ok())
 227 #if wxUSE_ZLIB
 228     {
 229         if (filename.Right(3).Lower() == wxT(".gz"))
 230         {
 231             wxZlibInputStream zstream(stream);
 232             m_renderer.Load(zstream);
 233         } else
 234         {
 235             m_renderer.Load(stream);
 236         }
 237     }
 238 #else
 239     {
 240         m_renderer.Load(stream);
 241     }
 242 #endif
 243 }
 244
 245 void TestGLCanvas::OnMouse(wxMouseEvent& event)
 246 {
 247     if (event.Dragging())
 248     {
 249         wxSize sz(GetClientSize());
 250
 251         /* drag in progress, simulate trackball */
 252         float spin_quat[4];
 253         trackball(spin_quat,
 254             (2.0*m_gldata.beginx - sz.x) / sz.x,
 255             (sz.y - 2.0*m_gldata.beginy) / sz.y,
 256             (2.0*event.GetX() - sz.x)    / sz.x,
 257             (sz.y - 2.0*event.GetY())    / sz.y);
 258
 259         add_quats(spin_quat, m_gldata.quat, m_gldata.quat);
 260
 261         /* orientation has changed, redraw mesh */
 262         Refresh(false);
 263     }
 264
 265     m_gldata.beginx = event.GetX();
 266     m_gldata.beginy = event.GetY();
 267 }
 268
 269 void TestGLCanvas::InitGL()
 270 {
 271     static const GLfloat light0_pos[4]   = { -50.0f, 50.0f, 0.0f, 0.0f };
 272
 273     // white light
 274     static const GLfloat light0_color[4] = { 0.6f, 0.6f, 0.6f, 1.0f };
 275
 276     static const GLfloat light1_pos[4]   = {  50.0f, 50.0f, 0.0f, 0.0f };
 277
 278     // cold blue light
 279     static const GLfloat light1_color[4] = { 0.4f, 0.4f, 1.0f, 1.0f };
 280
 281     /* remove back faces */
 282     glDisable(GL_CULL_FACE);
 283     glEnable(GL_DEPTH_TEST);
 284
 285     /* speedups */
 286     glEnable(GL_DITHER);
 287     glShadeModel(GL_SMOOTH);
 288     glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_FASTEST);
 289     glHint(GL_POLYGON_SMOOTH_HINT, GL_FASTEST);
 290
 291     /* light */
 292     glLightfv(GL_LIGHT0, GL_POSITION, light0_pos);
 293     glLightfv(GL_LIGHT0, GL_DIFFUSE,  light0_color);
 294     glLightfv(GL_LIGHT1, GL_POSITION, light1_pos);
 295     glLightfv(GL_LIGHT1, GL_DIFFUSE,  light1_color);
 296     glEnable(GL_LIGHT0);
 297     glEnable(GL_LIGHT1);
 298     glEnable(GL_LIGHTING);
 299
 300     glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE);
 301     glEnable(GL_COLOR_MATERIAL);
 302 }
 303
 304 void TestGLCanvas::ResetProjectionMode()
 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 }