]> git.jsancho.org Git - lugaru.git/blob - Source/OpenGL_Windows.cpp
Some debug about textures.
[lugaru.git] / Source / OpenGL_Windows.cpp
1 /*
2 Copyright (C) 2003, 2010 - Wolfire Games
3
4 This file is part of Lugaru.
5
6 Lugaru is free software; you can redistribute it and/or
7 modify it under the terms of the GNU General Public License
8 as published by the Free Software Foundation; either version 2
9 of the License, or (at your option) any later version.
10
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
14
15 See the GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
20 */
21
22
23 #ifdef WIN32
24 #define UINT8 WIN32API_UINT8
25 #define UINT16 WIN32API_UINT16
26 #define boolean WIN32API_boolean
27 #include <windows.h>
28 #undef UINT8
29 #undef UINT16
30 #undef boolean
31 #endif
32
33
34
35 #include "Game.h"
36 extern "C" {
37         #include "zlib.h"
38         #include "png.h"
39    #ifdef WIN32
40                 #define INT32 INT32_jpeg
41                 #include "jpeglib.h"
42                 #undef INT32
43         #else
44                 #include "jpeglib.h"
45         #endif
46 }
47
48 using namespace Game;
49
50 static bool load_image(const char * fname, TGAImageRec & tex);
51 static bool load_png(const char * fname, TGAImageRec & tex);
52 static bool load_jpg(const char * fname, TGAImageRec & tex);
53 bool save_image(const char * fname);
54 static bool save_png(const char * fname);
55
56
57 #include "openal_wrapper.h"
58
59 extern float multiplier;
60 extern float sps;
61 extern float realmultiplier;
62 extern int slomo;
63 extern bool cellophane;
64 extern float texdetail;
65
66 extern bool osx;
67 extern bool freeze;
68 extern bool stillloading;
69 extern int mainmenu;
70 /*extern*/ bool gameFocused;
71
72 extern float slomospeed;
73 extern float slomofreq;
74
75
76
77 #include <math.h>
78 #include <stdio.h>
79 #include <string.h>
80 #include <fstream>
81 #include <iostream>
82 #include "gamegl.h"
83 #include "MacCompatibility.h"
84 #include "Settings.h"
85
86 #ifdef WIN32
87 #include <shellapi.h>
88 #include "win-res/resource.h"
89 #endif
90
91 using namespace std;
92
93 SDL_Rect **resolutions = NULL;
94 static SDL_Rect rect_1024_768 = { 0, 0, 1024, 768 };
95 static SDL_Rect rect_800_600  = { 0, 0, 800,  600 };
96 static SDL_Rect rect_640_480  = { 0, 0, 640,  480 };
97 static SDL_Rect *hardcoded_resolutions[] = {
98     &rect_1024_768,
99     &rect_800_600,
100     &rect_640_480,
101     NULL
102 };
103
104 Boolean SetUp ();
105 void DoUpdate ();
106
107 void CleanUp (void);
108
109 // statics/globals (internal only) ------------------------------------------
110
111 #ifdef _MSC_VER
112 #pragma warning(push)
113 #pragma warning(disable: 4273)
114 #endif
115
116 #define GL_FUNC(ret,fn,params,call,rt) \
117     extern "C" { \
118         static ret (GLAPIENTRY *p##fn) params = NULL; \
119         ret GLAPIENTRY fn params { rt p##fn call; } \
120     }
121 #include "glstubs.h"
122 #undef GL_FUNC
123
124 #ifdef _MSC_VER
125 #pragma warning(pop)
126 #endif
127
128 static bool lookup_glsym(const char *funcname, void **func)
129 {
130     *func = SDL_GL_GetProcAddress(funcname);
131     if (*func == NULL)
132     {
133         fprintf(stderr, "Failed to find OpenGL symbol \"%s\"\n", funcname);
134         return false;
135     }
136     return true;
137 }
138
139 static bool lookup_all_glsyms(void)
140 {
141     bool retval = true;
142     #define GL_FUNC(ret,fn,params,call,rt) \
143         if (!lookup_glsym(#fn, (void **) &p##fn)) retval = false;
144     #include "glstubs.h"
145     #undef GL_FUNC
146     return retval;
147 }
148
149 static void GLAPIENTRY glDeleteTextures_doNothing(GLsizei n, const GLuint *textures)
150 {
151     // no-op.
152 }
153
154 #ifdef MessageBox
155 #undef MessageBox
156 #endif
157 #define MessageBox(hwnd,text,title,flags) STUBBED("msgbox")
158
159 // Menu defs
160
161 int kContextWidth;
162 int kContextHeight;
163
164 Boolean gDone = false;
165
166 #ifndef __MINGW32__
167 static int _argc = 0;
168 static char **_argv = NULL;
169 #endif
170
171 bool cmdline(const char *cmd)
172 {
173     for (int i = 1; i < _argc; i++)
174     {
175         char *arg = _argv[i];
176         while (*arg == '-')
177             arg++;
178         if (strcasecmp(arg, cmd) == 0)
179             return true;
180     }
181
182     return false;
183 }
184
185 //-----------------------------------------------------------------------------------------------------------------------
186
187 // OpenGL Drawing
188
189 void initGL(){
190         glClear( GL_COLOR_BUFFER_BIT );
191         swap_gl_buffers();
192
193         // clear all states
194         glDisable( GL_ALPHA_TEST);
195         glDisable( GL_BLEND);
196         glDisable( GL_DEPTH_TEST);
197         //      glDisable( GL_DITHER);
198         glDisable( GL_FOG);
199         glDisable( GL_LIGHTING);
200         glDisable( GL_LOGIC_OP);
201         glDisable( GL_TEXTURE_1D);
202         glDisable( GL_TEXTURE_2D);
203         glPixelTransferi( GL_MAP_COLOR, GL_FALSE);
204         glPixelTransferi( GL_RED_SCALE, 1);
205         glPixelTransferi( GL_RED_BIAS, 0);
206         glPixelTransferi( GL_GREEN_SCALE, 1);
207         glPixelTransferi( GL_GREEN_BIAS, 0);
208         glPixelTransferi( GL_BLUE_SCALE, 1);
209         glPixelTransferi( GL_BLUE_BIAS, 0);
210         glPixelTransferi( GL_ALPHA_SCALE, 1);
211         glPixelTransferi( GL_ALPHA_BIAS, 0);
212
213         // set initial rendering states
214         glShadeModel( GL_SMOOTH);
215         glClearDepth( 1.0f);
216         glDepthFunc( GL_LEQUAL);
217         glDepthMask( GL_TRUE);
218         //      glDepthRange( FRONT_CLIP, BACK_CLIP);
219         glEnable( GL_DEPTH_TEST);
220         glHint( GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST);
221         glCullFace( GL_FRONT);
222         glEnable( GL_CULL_FACE);
223         glEnable( GL_LIGHTING);
224 //      glEnable( GL_LIGHT_MODEL_AMBIENT);
225         glEnable( GL_DITHER);
226         glEnable( GL_COLOR_MATERIAL);
227         glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
228         glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
229         glAlphaFunc( GL_GREATER, 0.5f);
230
231         if ( CanInitStereo(stereomode) ) {
232                 InitStereo(stereomode);
233         } else {
234                 fprintf(stderr, "Failed to initialize stereo, disabling.\n");
235                 stereomode = stereoNone;
236         }
237 }
238
239 static void toggleFullscreen(){
240         if(!SDL_WM_ToggleFullScreen(SDL_GetVideoSurface())){
241         SDL_Surface* screen=SDL_GetVideoSurface();
242         Uint32 flags=screen->flags;
243         screen=SDL_SetVideoMode(0,0,0,flags^SDL_FULLSCREEN);
244         if(!screen)
245             screen=SDL_SetVideoMode(0,0,0,flags);
246         if(!screen)
247             exit(1);
248         //reload opengl state
249         initGL();
250         Texture::reloadAll();
251         if(text)
252             text->BuildFont();
253         if(firstload){
254             screentexture=0;
255             LoadScreenTexture();
256         }
257         screentexture2=0;
258     }
259 }
260
261 static void sdlEventProc(const SDL_Event &e)
262 {
263     switch(e.type) {
264         case SDL_MOUSEMOTION:
265             deltah += e.motion.xrel;
266             deltav += e.motion.yrel;
267             break;
268
269         case SDL_KEYDOWN:
270             if ((e.key.keysym.sym == SDLK_g) &&
271                                 (e.key.keysym.mod & KMOD_CTRL) &&
272                                 !(SDL_GetVideoSurface()->flags & SDL_FULLSCREEN) ) {
273                                 SDL_WM_GrabInput( ((SDL_WM_GrabInput(SDL_GRAB_QUERY)==SDL_GRAB_ON) ? SDL_GRAB_OFF:SDL_GRAB_ON) );
274             } else if ( (e.key.keysym.sym == SDLK_RETURN) && (e.key.keysym.mod & KMOD_ALT) ) {
275                                 toggleFullscreen();
276             }
277             break;
278     }
279 }
280
281
282
283 // --------------------------------------------------------------------------
284
285 static Point gMidPoint;
286
287 Boolean SetUp ()
288 {
289         char string[10];
290
291         LOGFUNC;
292
293         osx = 0;
294         cellophane=0;
295         texdetail=4;
296         slomospeed=0.25;
297         slomofreq=8012;
298         numplayers=1;
299         
300         DefaultSettings();
301
302     if (!SDL_WasInit(SDL_INIT_VIDEO))
303         if (SDL_Init(SDL_INIT_VIDEO) == -1)
304         {
305             fprintf(stderr, "SDL_Init() failed: %s\n", SDL_GetError());
306             return false;
307         }
308         if(!LoadSettings()) {
309                 fprintf(stderr, "Failed to load config, creating default\n");
310                 SaveSettings();
311         }
312         if(kBitsPerPixel!=32&&kBitsPerPixel!=16){
313                 kBitsPerPixel=16;
314         }
315
316         if (SDL_GL_LoadLibrary(NULL) == -1)
317         {
318                 fprintf(stderr, "SDL_GL_LoadLibrary() failed: %s\n", SDL_GetError());
319                 SDL_Quit();
320                 return false;
321         }
322
323         SDL_Rect **res = SDL_ListModes(NULL, SDL_FULLSCREEN|SDL_OPENGL);
324         if ( (res == NULL) || (res == ((SDL_Rect **)-1)) || (res[0] == NULL) || (res[0]->w < 640) || (res[0]->h < 480) )
325                 res = hardcoded_resolutions;
326
327         // reverse list (it was sorted biggest to smallest by SDL)...
328         int count;
329         for (count = 0; res[count]; count++)
330         {
331                 if ((res[count]->w < 640) || (res[count]->h < 480))
332                         break;   // sane lower limit.
333         }
334
335         static SDL_Rect *resolutions_block = NULL;
336         resolutions_block = (SDL_Rect*) realloc(resolutions_block, sizeof (SDL_Rect) * count);
337         resolutions = (SDL_Rect**) realloc(resolutions, sizeof (SDL_Rect *) * (count + 1));
338         if ((resolutions_block == NULL) || (resolutions == NULL))
339         {
340                 SDL_Quit();
341                 fprintf(stderr, "Out of memory!\n");
342                 return false;
343         }
344
345         resolutions[count--] = NULL;
346         for (int i = 0; count >= 0; i++, count--)
347         {
348                 memcpy(&resolutions_block[count], res[i], sizeof (SDL_Rect));
349                 resolutions[count] = &resolutions_block[count];
350         }
351
352         if (cmdline("showresolutions"))
353         {
354                 printf("Resolutions we think are okay:\n");
355                 for (int i = 0; resolutions[i]; i++)
356                         printf("  %d x %d\n", (int) resolutions[i]->w, (int) resolutions[i]->h);
357         }
358
359     Uint32 sdlflags = SDL_OPENGL;
360     
361     if (!cmdline("windowed"))
362         sdlflags |= SDL_FULLSCREEN;
363
364     SDL_WM_SetCaption("Lugaru", "Lugaru");
365
366     SDL_ShowCursor(0);
367
368     SDL_GL_SetAttribute(SDL_GL_DOUBLEBUFFER, 1);
369     SDL_GL_SetAttribute(SDL_GL_STENCIL_SIZE, 1);
370 #if SDL_VERSION_ATLEAST(1, 2, 10)
371     SDL_GL_SetAttribute(SDL_GL_SWAP_CONTROL, vblsync);
372 #endif
373     
374     if (SDL_SetVideoMode(kContextWidth, kContextHeight, 0, sdlflags) == NULL)
375     {
376         fprintf(stderr, "SDL_SetVideoMode() failed: %s\n", SDL_GetError());
377         fprintf(stderr, "forcing 640x480...\n");
378         kContextWidth = 640;
379         kContextHeight = 480;
380         if (SDL_SetVideoMode(kContextWidth, kContextHeight, 0, sdlflags) == NULL)
381         {
382             fprintf(stderr, "SDL_SetVideoMode() failed: %s\n", SDL_GetError());
383             fprintf(stderr, "forcing 640x480 windowed mode...\n");
384             sdlflags &= ~SDL_FULLSCREEN;
385             if (SDL_SetVideoMode(kContextWidth, kContextHeight, 0, sdlflags) == NULL)
386             {
387                 fprintf(stderr, "SDL_SetVideoMode() failed: %s\n", SDL_GetError());
388                 return false;
389             }
390         }
391     }
392
393     int dblbuf = 0;
394     if ((SDL_GL_GetAttribute(SDL_GL_DOUBLEBUFFER, &dblbuf) == -1) || (!dblbuf))
395     {
396         fprintf(stderr, "Failed to get double buffered GL context!\n");
397         SDL_Quit();
398         return false;
399     }
400
401     if (!lookup_all_glsyms())
402     {
403         SDL_Quit();
404         return false;
405     }
406
407     if (!cmdline("nomousegrab"))
408         SDL_WM_GrabInput(SDL_GRAB_ON);
409
410
411     initGL();
412
413         GLint width = kContextWidth;
414         GLint height = kContextHeight;
415         gMidPoint.h = width / 2;
416         gMidPoint.v = height / 2;
417         screenwidth=width;
418         screenheight=height;
419
420         newdetail=detail;
421         newscreenwidth=screenwidth;
422         newscreenheight=screenheight;
423
424         InitGame();
425
426         return true;
427 }
428
429
430 static void DoMouse()
431 {
432
433         if(mainmenu|| ( (abs(deltah)<10*realmultiplier*1000) && (abs(deltav)<10*realmultiplier*1000) ))
434         {
435                 deltah *= usermousesensitivity;
436                 deltav *= usermousesensitivity;
437                 mousecoordh += deltah;
438                 mousecoordv += deltav;
439         if (mousecoordh < 0)
440             mousecoordh = 0;
441         else if (mousecoordh >= kContextWidth)
442             mousecoordh = kContextWidth - 1;
443         if (mousecoordv < 0)
444             mousecoordv = 0;
445         else if (mousecoordv >= kContextHeight)
446             mousecoordv = kContextHeight - 1;
447         }
448
449 }
450
451 void DoFrameRate (int update)
452 {       
453         static long frames = 0;
454
455         static AbsoluteTime time = {0,0};
456         static AbsoluteTime frametime = {0,0};
457         AbsoluteTime currTime = UpTime ();
458         double deltaTime = (float) AbsoluteDeltaToDuration (currTime, frametime);
459
460         if (0 > deltaTime)      // if negative microseconds
461                 deltaTime /= -1000000.0;
462         else                            // else milliseconds
463                 deltaTime /= 1000.0;
464
465         multiplier=deltaTime;
466         if(multiplier<.001) multiplier=.001;
467         if(multiplier>10) multiplier=10;
468         if(update) frametime = currTime;        // reset for next time interval
469
470         deltaTime = (float) AbsoluteDeltaToDuration (currTime, time);
471
472         if (0 > deltaTime)      // if negative microseconds
473                 deltaTime /= -1000000.0;
474         else                            // else milliseconds
475                 deltaTime /= 1000.0;
476         frames++;
477         if (0.001 <= deltaTime) // has update interval passed
478         {
479                 if(update){
480                         time = currTime;        // reset for next time interval
481                         frames = 0;
482                 }
483         }
484 }
485
486
487 void DoUpdate ()
488 {
489         static float sps=200;
490         static int count;
491         static float oldmult;
492
493         DoFrameRate(1);
494         if(multiplier>.6)multiplier=.6;
495
496         fps=1/multiplier;
497
498         count = multiplier*sps;
499         if(count<2)count=2;
500
501         realmultiplier=multiplier;
502         multiplier*=gamespeed;
503         if(difficulty==1)multiplier*=.9;
504         if(difficulty==0)multiplier*=.8;
505
506         if(loading==4)multiplier*=.00001;
507         if(slomo&&!mainmenu)multiplier*=slomospeed;
508         oldmult=multiplier;
509         multiplier/=(float)count;
510
511         DoMouse();
512
513         TickOnce();
514
515         for(int i=0;i<count;i++)
516         {
517                 Tick();
518         }
519         multiplier=oldmult;
520
521         TickOnceAfter();
522 /* - Debug code to test how many channels were active on average per frame
523         static long frames = 0;
524
525         static AbsoluteTime start = {0,0};
526         AbsoluteTime currTime = UpTime ();
527         static int num_channels = 0;
528         
529         num_channels += OPENAL_GetChannelsPlaying();
530         double deltaTime = (float) AbsoluteDeltaToDuration (currTime, start);
531
532         if (0 > deltaTime)      // if negative microseconds
533                 deltaTime /= -1000000.0;
534         else                            // else milliseconds
535                 deltaTime /= 1000.0;
536
537         ++frames;
538
539         if (deltaTime >= 1)
540         {
541                 start = currTime;
542                 float avg_channels = (float)num_channels / (float)frames;
543
544                 ofstream opstream("log.txt",ios::app); 
545                 opstream << "Average frame count: ";
546                 opstream << frames;
547                 opstream << " frames - ";
548                 opstream << avg_channels;
549                 opstream << " per frame.\n";
550                 opstream.close();
551
552                 frames = 0;
553                 num_channels = 0;
554         }
555 */
556         if ( stereomode == stereoNone ) {
557                 DrawGLScene(stereoCenter);
558         } else {
559                 DrawGLScene(stereoLeft);
560                 DrawGLScene(stereoRight);
561         }
562 }
563
564 // --------------------------------------------------------------------------
565
566
567 void CleanUp (void)
568 {
569         LOGFUNC;
570
571     SDL_Quit();
572     #define GL_FUNC(ret,fn,params,call,rt) p##fn = NULL;
573     #include "glstubs.h"
574     #undef GL_FUNC
575     // cheat here...static destructors are calling glDeleteTexture() after
576     //  the context is destroyed and libGL unloaded by SDL_Quit().
577     pglDeleteTextures = glDeleteTextures_doNothing;
578
579 }
580
581 // --------------------------------------------------------------------------
582
583 static bool IsFocused()
584 {
585     return ((SDL_GetAppState() & SDL_APPINPUTFOCUS) != 0);
586 }
587
588
589
590 #ifndef WIN32
591 // (code lifted from physfs: http://icculus.org/physfs/ ... zlib license.)
592 static char *findBinaryInPath(const char *bin, char *envr)
593 {
594     size_t alloc_size = 0;
595     char *exe = NULL;
596     char *start = envr;
597     char *ptr;
598
599     do
600     {
601         size_t size;
602         ptr = strchr(start, ':');  /* find next $PATH separator. */
603         if (ptr)
604             *ptr = '\0';
605
606         size = strlen(start) + strlen(bin) + 2;
607         if (size > alloc_size)
608         {
609             char *x = (char *) realloc(exe, size);
610             if (x == NULL)
611             {
612                 if (exe != NULL)
613                     free(exe);
614                 return(NULL);
615             } /* if */
616
617             alloc_size = size;
618             exe = x;
619         } /* if */
620
621         /* build full binary path... */
622         strcpy(exe, start);
623         if ((exe[0] == '\0') || (exe[strlen(exe) - 1] != '/'))
624             strcat(exe, "/");
625         strcat(exe, bin);
626
627         if (access(exe, X_OK) == 0)  /* Exists as executable? We're done. */
628         {
629             strcpy(exe, start);  /* i'm lazy. piss off. */
630             return(exe);
631         } /* if */
632
633         start = ptr + 1;  /* start points to beginning of next element. */
634     } while (ptr != NULL);
635
636     if (exe != NULL)
637         free(exe);
638
639     return(NULL);  /* doesn't exist in path. */
640 } /* findBinaryInPath */
641
642
643 char *calcBaseDir(const char *argv0)
644 {
645     /* If there isn't a path on argv0, then look through the $PATH for it. */
646     char *retval;
647     char *envr;
648
649     const char *ptr = strrchr((char *)argv0, '/');
650     if (strchr(argv0, '/'))
651     {
652         retval = strdup(argv0);
653         if (retval)
654             *((char *) strrchr(retval, '/')) = '\0';
655         return(retval);
656     }
657
658     envr = getenv("PATH");
659     if (!envr) return NULL;
660     envr = strdup(envr);
661     if (!envr) return NULL;
662     retval = findBinaryInPath(argv0, envr);
663     free(envr);
664     return(retval);
665 }
666
667 static inline void chdirToAppPath(const char *argv0)
668 {
669     char *dir = calcBaseDir(argv0);
670     if (dir)
671     {
672         #if (defined(__APPLE__) && defined(__MACH__))
673         // Chop off /Contents/MacOS if it's at the end of the string, so we
674         //  land in the base of the app bundle.
675         const size_t len = strlen(dir);
676         const char *bundledirs = "/Contents/MacOS";
677         const size_t bundledirslen = strlen(bundledirs);
678         if (len > bundledirslen)
679         {
680             char *ptr = (dir + len) - bundledirslen;
681             if (strcasecmp(ptr, bundledirs) == 0)
682                 *ptr = '\0';
683         }
684         #endif
685         chdir(dir);
686         free(dir);
687     }
688 }
689 #endif
690
691
692 int main(int argc, char **argv)
693 {
694 #ifndef __MINGW32__
695     _argc = argc;
696     _argv = argv;
697 #endif
698
699     // !!! FIXME: we could use a Win32 API for this.  --ryan.
700 #ifndef WIN32
701     chdirToAppPath(argv[0]);
702 #endif
703
704         LOGFUNC;
705
706         try
707         {
708                 {
709                         newGame();
710
711                         //ofstream os("error.txt");
712                         //os.close();
713                         //ofstream os("log.txt");
714                         //os.close();
715
716                         if (!SetUp ())
717                 return 42;
718
719                         while (!gDone&&!tryquit)
720                         {
721                                         if (IsFocused())
722                                         {
723                                                         gameFocused = true;
724
725                                                         // check windows messages
726                         
727                                                         deltah = 0;
728                                                         deltav = 0;
729                                                         SDL_Event e;
730                                                         if(!waiting) {
731                                                                         // message pump
732                                                                         while( SDL_PollEvent( &e ) )
733                                                                         {
734                                                                                         if( e.type == SDL_QUIT )
735                                                                                         {
736                                                                                                         gDone=true;
737                                                                                                         break;
738                                                                                         }
739                                                                                         sdlEventProc(e);
740                                                                         }
741                                                         }
742
743                                                         // game
744                                                         DoUpdate();
745                                         }
746                                         else
747                                         {
748                                                         if (gameFocused)
749                                                         {
750                                                                         // allow game chance to pause
751                                                                         gameFocused = false;
752                                                                         DoUpdate();
753                                                         }
754
755                                                         // game is not in focus, give CPU time to other apps by waiting for messages instead of 'peeking'
756                                                         SDL_ActiveEvent evt;
757                                                         SDL_WaitEvent((SDL_Event*)&evt);
758                                                         if (evt.type == SDL_ACTIVEEVENT && evt.gain == 1)
759                                                                         gameFocused = true;
760                                                         else if (evt.type == SDL_QUIT)
761                                                                         gDone = true;
762                                         }
763                         }
764
765             deleteGame();
766                 }
767
768                 CleanUp ();
769
770                 return 0;
771         }
772         catch (const std::exception& error)
773         {
774                 CleanUp();
775
776                 std::string e = "Caught exception: ";
777                 e += error.what();
778
779                 LOG(e);
780
781                 MessageBox(g_windowHandle, error.what(), "ERROR", MB_OK | MB_ICONEXCLAMATION);
782         }
783
784         CleanUp();
785
786         return -1;
787 }
788
789
790
791 // --------------------------------------------------------------------------
792
793
794 bool LoadImage(const char * fname, TGAImageRec & tex)
795 {
796         if ( tex.data == NULL )
797                 return false;
798         else
799                 return load_image(fname, tex);
800 }
801
802 void ScreenShot(const char * fname)
803 {
804         
805 }
806
807
808
809 static bool load_image(const char *file_name, TGAImageRec &tex)
810 {
811     const char *ptr = strrchr((char *)file_name, '.');
812     if (ptr)
813     {
814         if (strcasecmp(ptr+1, "png") == 0)
815             return load_png(file_name, tex);
816         else if (strcasecmp(ptr+1, "jpg") == 0)
817             return load_jpg(file_name, tex);
818     }
819
820     STUBBED("Unsupported image type");
821     return false;
822 }
823
824
825 struct my_error_mgr {
826   struct jpeg_error_mgr pub;    /* "public" fields */
827   jmp_buf setjmp_buffer;        /* for return to caller */
828 };
829 typedef struct my_error_mgr * my_error_ptr;
830
831
832 static void my_error_exit(j_common_ptr cinfo)
833 {
834         struct my_error_mgr *err = (struct my_error_mgr *)cinfo->err;
835         longjmp(err->setjmp_buffer, 1);
836 }
837
838 /* stolen from public domain example.c code in libjpg distribution. */
839 static bool load_jpg(const char *file_name, TGAImageRec &tex)
840 {
841     struct jpeg_decompress_struct cinfo;
842     struct my_error_mgr jerr;
843     JSAMPROW buffer[1];         /* Output row buffer */
844     int row_stride;             /* physical row width in output buffer */
845     FILE *infile = fopen(file_name, "rb");
846
847     if (infile == NULL)
848         return false;
849
850     cinfo.err = jpeg_std_error(&jerr.pub);
851     jerr.pub.error_exit = my_error_exit;
852     if (setjmp(jerr.setjmp_buffer)) {
853         jpeg_destroy_decompress(&cinfo);
854         fclose(infile);
855         return false;
856     }
857
858     jpeg_create_decompress(&cinfo);
859     jpeg_stdio_src(&cinfo, infile);
860     (void) jpeg_read_header(&cinfo, TRUE);
861
862     cinfo.out_color_space = JCS_RGB;
863     cinfo.quantize_colors = 0;
864     (void) jpeg_calc_output_dimensions(&cinfo);
865     (void) jpeg_start_decompress(&cinfo);
866
867     row_stride = cinfo.output_width * cinfo.output_components;
868     tex.sizeX = cinfo.output_width;
869     tex.sizeY = cinfo.output_height;
870     tex.bpp = 24;
871
872     while (cinfo.output_scanline < cinfo.output_height) {
873         buffer[0] = (JSAMPROW)(char *)tex.data +
874                         ((cinfo.output_height-1) - cinfo.output_scanline) * row_stride;
875         (void) jpeg_read_scanlines(&cinfo, buffer, 1);
876     }
877
878     (void) jpeg_finish_decompress(&cinfo);
879     jpeg_destroy_decompress(&cinfo);
880     fclose(infile);
881
882     return true;
883 }
884
885
886 /* stolen from public domain example.c code in libpng distribution. */
887 static bool load_png(const char *file_name, TGAImageRec &tex)
888 {
889     bool hasalpha = false;
890     png_structp png_ptr = NULL;
891     png_infop info_ptr = NULL;
892     png_uint_32 width, height;
893     int bit_depth, color_type, interlace_type;
894     png_byte **rows = NULL;
895     bool retval = false;
896     png_byte **row_pointers = NULL;
897     FILE *fp = fopen(file_name, "rb");
898
899     if (fp == NULL) {
900                 cerr << file_name << " not found" << endl;
901         return(NULL);
902         }
903
904     png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
905     if (png_ptr == NULL)
906         goto png_done;
907
908     info_ptr = png_create_info_struct(png_ptr);
909     if (info_ptr == NULL)
910         goto png_done;
911
912     if (setjmp(png_jmpbuf(png_ptr)))
913         goto png_done;
914
915     png_init_io(png_ptr, fp);
916     png_read_png(png_ptr, info_ptr,
917                  PNG_TRANSFORM_STRIP_16 | PNG_TRANSFORM_PACKING,
918                  NULL);
919     png_get_IHDR(png_ptr, info_ptr, &width, &height,
920                  &bit_depth, &color_type, &interlace_type, NULL, NULL);
921
922     if (bit_depth != 8)  // transform SHOULD handle this...
923         goto png_done;
924
925     if (color_type & PNG_COLOR_MASK_PALETTE)  // !!! FIXME?
926         goto png_done;
927
928     if ((color_type & PNG_COLOR_MASK_COLOR) == 0)  // !!! FIXME?
929         goto png_done;
930
931     hasalpha = ((color_type & PNG_COLOR_MASK_ALPHA) != 0);
932     row_pointers = png_get_rows(png_ptr, info_ptr);
933     if (!row_pointers)
934         goto png_done;
935
936     if (!hasalpha)
937     {
938         png_byte *dst = tex.data;
939         for (int i = height-1; i >= 0; i--)
940         {
941             png_byte *src = row_pointers[i];
942             for (int j = 0; j < width; j++)
943             {
944                 dst[0] = src[0];
945                 dst[1] = src[1];
946                 dst[2] = src[2];
947                 dst[3] = 0xFF;
948                 src += 3;
949                 dst += 4;
950             }
951         }
952     }
953
954     else
955     {
956         png_byte *dst = tex.data;
957         int pitch = width * 4;
958         for (int i = height-1; i >= 0; i--, dst += pitch)
959             memcpy(dst, row_pointers[i], pitch);
960     }
961
962     tex.sizeX = width;
963     tex.sizeY = height;
964     tex.bpp = 32;
965     retval = true;
966     
967 png_done:
968         if(!retval) {
969                 cerr << "There was a problem loading " << file_name << endl;
970         }
971     png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
972     if (fp)
973         fclose(fp);
974     return (retval);
975 }
976
977
978 bool save_image(const char *file_name)
979 {
980     const char *ptr = strrchr((char *)file_name, '.');
981     if (ptr)
982     {
983         if (strcasecmp(ptr+1, "png") == 0)
984             return save_png(file_name);
985     }
986
987     STUBBED("Unsupported image type");
988     return false;
989 }
990
991
992 static bool save_png(const char *file_name)
993 {
994     FILE *fp = NULL;
995     png_structp png_ptr = NULL;
996     png_infop info_ptr = NULL;
997     bool retval = false;
998
999     fp = fopen(file_name, "wb");
1000     if (fp == NULL)
1001         return false;
1002
1003     png_bytep *row_pointers = new png_bytep[kContextHeight];
1004     png_bytep screenshot = new png_byte[kContextWidth * kContextHeight * 3];
1005     if ((!screenshot) || (!row_pointers))
1006         goto save_png_done;
1007
1008     glGetError();
1009     glReadPixels(0, 0, kContextWidth, kContextHeight,
1010                  GL_RGB, GL_UNSIGNED_BYTE, screenshot);
1011     if (glGetError() != GL_NO_ERROR)
1012         goto save_png_done;
1013
1014     for (int i = 0; i < kContextHeight; i++)
1015         row_pointers[i] = screenshot + ((kContextWidth * ((kContextHeight-1) - i)) * 3);
1016
1017     png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
1018     if (png_ptr == NULL)
1019         goto save_png_done;
1020
1021     info_ptr = png_create_info_struct(png_ptr);
1022     if (info_ptr == NULL)
1023         goto save_png_done;
1024
1025     if (setjmp(png_jmpbuf(png_ptr)))
1026         goto save_png_done;
1027
1028     png_init_io(png_ptr, fp);
1029
1030     if (setjmp(png_jmpbuf(png_ptr)))
1031         goto save_png_done;
1032
1033     png_set_IHDR(png_ptr, info_ptr, kContextWidth, kContextHeight,
1034                  8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
1035                  PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
1036
1037     png_write_info(png_ptr, info_ptr);
1038
1039     if (setjmp(png_jmpbuf(png_ptr)))
1040         goto save_png_done;
1041
1042         png_write_image(png_ptr, row_pointers);
1043
1044         if (setjmp(png_jmpbuf(png_ptr)))
1045         goto save_png_done;
1046
1047     png_write_end(png_ptr, NULL);
1048     retval = true;
1049
1050 save_png_done:
1051     png_destroy_write_struct(&png_ptr, &info_ptr);
1052     delete[] screenshot;
1053     delete[] row_pointers;
1054     if (fp)
1055         fclose(fp);
1056     if (!retval)
1057         unlink(ConvertFileName(file_name));
1058     return retval;
1059 }
1060
1061
1062