winmain.cpp #4

#ifdef WIN_SCREENSAVER
//Windows screensaver code borrowed from Rachel Grey
// http://www.cityintherain.com/howtoscr.html

#include <direct.h>
#include <windows.h>
#include <scrnsave.h>
#include <GL/gl.h>

//Settings dialog (includes globals)
//#include <qapplication.h>
//#include "SettingsDialog.h"

#include "hive.h"
#include "vanillahive.h"

#include "settings.h"

Hive* hive;

//globals used by the function below to hold the screen size
int Width;	
int Height;

//Windows timer IDs
#define FRAME_TIMER 1 
#define EVENT_TIMER 2

static void InitGL(HWND hWnd, HDC & hDC, HGLRC & hRC)
{
  PIXELFORMATDESCRIPTOR pfd;
  ZeroMemory( &pfd, sizeof pfd );
  pfd.nSize = sizeof pfd;
  pfd.nVersion = 1;
  pfd.dwFlags = PFD_SUPPORT_OPENGL | PFD_DOUBLEBUFFER;
  pfd.iPixelType = PFD_TYPE_RGBA;
  pfd.cColorBits = 24;
  
  hDC = GetDC( hWnd );
  
  int i = ChoosePixelFormat( hDC, &pfd );  
  SetPixelFormat( hDC, i, &pfd );

  hRC = wglCreateContext( hDC );
  wglMakeCurrent( hDC, hRC );
}
 
static void CloseGL(HWND hWnd, HDC hDC, HGLRC hRC)
{
  wglMakeCurrent( NULL, NULL );
  wglDeleteContext( hRC );
  ReleaseDC( hWnd, hDC );
}

// Screen Saver Procedure
LRESULT WINAPI ScreenSaverProc(HWND hWnd, UINT message, 
                               WPARAM wParam, LPARAM lParam)
{
  static HDC hDC;
  static HGLRC hRC;
  static RECT rect;

  int argc = 0; 
  char** argv = 0;

  switch ( message ) {

  case WM_CREATE: 
    // get window dimensions
    GetClientRect( hWnd, &rect );
    Width = rect.right;		
    Height = rect.bottom;

	//get configuration from config file if applicable
	chdir( getenv( "SystemRoot" ) );
	chdir( "SYSTEM32" );
	loadsettings();

	if ( Width < 400 )
	{
		//mini-preview settings
		settings.numbees = 12;
		settings.numcells = 3;
		settings.eventms = 1000;
		settings.launchdist = -5;
		settings.basesize = 5;
		settings.randsize = 1;
	}

    //set up OpenGL
    InitGL( hWnd, hDC, hRC );

    //Initialize perspective, viewpoint, and
    //any objects you wish to animate
	hive = new VanillaHive( Width, Height );

    // Initialize timers
    SetTimer( hWnd, FRAME_TIMER, 10, NULL ); 
	SetTimer( hWnd, EVENT_TIMER, settings.eventms, NULL ); 
    return 0;
 
  case WM_DESTROY:
    KillTimer( hWnd, FRAME_TIMER );
	KillTimer( hWnd, EVENT_TIMER );
    
	delete hive;

    CloseGL( hWnd, hDC, hRC );
    return 0;

  case WM_TIMER:
	switch( wParam )
	{
	case FRAME_TIMER:
		hive->step();
		hive->paint();
		SwapBuffers( hDC );
		glFlush();
		break;
	case EVENT_TIMER:
		hive->doEvent();
		break;
	}
	return 0;

  //handle special keypresses
  case WM_KEYDOWN:
	  	switch ( wParam )
		{
		case 'c':
		case 'C':
			//stuff would go here if C did something special
			//return 0;
			break;
		}
  case WM_KEYUP:
	  switch ( wParam )
	  {
		case 'c':
		case 'C':
			//if C did something special, return 0 here
			//return 0;
			break;
	  }
  //let the screensaver library take care of any
  //other messages
  }
  return DefScreenSaverProc( 
    hWnd, message, wParam, lParam );
}

BOOL WINAPI
ScreenSaverConfigureDialog(HWND hDlg, UINT message, 
                           WPARAM wParam, LPARAM lParam)
{
	return true;
}

// needed for SCRNSAVE.LIB
BOOL WINAPI RegisterDialogClasses(HANDLE hInst)
{
	//Notepad config "dialog" -- temporary measure
	//until a proper one is in place.

	chdir( getenv( "SystemRoot" ) );
 	chdir( "system32" );
  	loadsettings();
	savesettings();

	STARTUPINFO si;
    PROCESS_INFORMATION pi;

    ZeroMemory( &si, sizeof(si) );
    si.cb = sizeof(si);
	GetStartupInfo( &si );
	si.dwFlags |= STARTF_USESHOWWINDOW;
	si.wShowWindow = SW_SHOWNORMAL;

    ZeroMemory( &pi, sizeof(pi) );

	WCHAR FAR u[18];
	MultiByteToWideChar( CP_ACP, MB_PRECOMPOSED, "notepad Hive.cfg", -1, u, 18 );

    // Start the child process. 
    if( !CreateProcess( NULL, // No module name (use command line). 
        u, // Command line. 
        NULL,             // Process handle not inheritable. 
        NULL,             // Thread handle not inheritable. 
        FALSE,            // Set handle inheritance to FALSE. 
        0,                // No creation flags. 
        NULL,             // Use parent's environment block. 
        NULL,             // Use parent's starting directory. 
        &si,              // Pointer to STARTUPINFO structure.
        &pi )             // Pointer to PROCESS_INFORMATION structure.
    ) 
    {
        return false;
    }

    // Wait until child process exits.
    WaitForSingleObject( pi.hProcess, INFINITE );

    // Close process and thread handles. 
    CloseHandle( pi.hProcess );
    CloseHandle( pi.hThread );

	return true;

	//Here's where the real dialog will go.
	/*
	chdir( getenv( "SystemRoot" ) );
	chdir( "SYSTEM32" );
  	loadsettings();

	int argc = 1;
	char* argv[1];
	argv[0] = "Hive";

	QApplication app( argc, argv );
	SettingsDialog diag;
	app.setMainWidget( &diag );
	diag.show();
	return ! (app.exec() == 0);
	*/
}

#endif //WIN_SCREENSAVER