Context Navigation

source: trunk/gotcha.cpp@ 65

Last change on this file since 65 was 65, checked in by Gregg Young, 6 years ago
Fix #ifdef change from CS [59]
Property svn:eol-style set to `native`
File size: 12.1 KB

Rev	Line
[2]	1	/***
	2	Main source of the Gotcha! screencapture program.
	3	Copyright (C) 1998-2002 Thorsten Thielen <thth@c2226.de>
	4
	5	This program is free software; you can redistribute it and/or modify
	6	it under the terms of the GNU General Public License as published by
	7	the Free Software Foundation; either version 2 of the License, or
	8	(at your option) any later version.
	9
	10	This program is distributed in the hope that it will be useful,
	11	but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	GNU General Public License for more details.
	14
	15	You should have received a copy of the GNU General Public License
	16	along with this program; if not, write to the Free Software
	17	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	18	***/
	19
	20	#define INCL_WINHOOKS
	21
	22	#include "gotcha.h"
[10]	23	#include "dll/gotchdll.h"
[2]	24	#include "settings.h"
	25	#include "string.h"
	26
	27	#include "io.h"
[13]	28	#define INCL_EXCEPTQ_CLASS
	29	#define INCL_LOADEXCEPTQ
	30	#include "exceptq.h"
[2]	31
	32	HAB hab;
	33	HWND hwndFrame, hwndSnapshot;
	34	PFNWP OldFrameWP, wpOldButton;
	35	PSETTINGS pset;
[38]	36	BOOL g_fIdle = FALSE, g_fSetPathName = FALSE;
[2]	37	HWND g_hwndMenuSSW;
	38	HMODULE g_hmod = NULLHANDLE;
	39	Helper *g_phelp = NULL;
[57]	40	BOOL g_usePMps = FALSE;
[2]	41
	42	#include "bitmap.cpp"
	43	#include "mainwin.cpp"
	44	#include "snapshot.cpp"
	45	#include "savebmp.cpp"
	46
[27]	47	// CheckCmdlineArgs ************************************************** /FOLD00/
[2]	48
	49	BOOL CheckCmdlineArgs (int argc, char *argv[])
	50	{
	51	BOOL fAuto = FALSE;
	52
	53	for (USHORT i = 1; i < argc; i++)
	54	{
[59]	55	#ifndef _QUIET_
[38]	56	// batch mode
	57	if ((stricmp (argv[i], "-a") == 0))
[2]	58	{
	59	if (pset->QuerySaveStyle () != FSS_FORCEFILE)
	60	pset->SetFileSaveStyle (FSS_NUMFILES);
	61	fAuto = TRUE;
	62
	63	if (i < argc-1)
	64	if (argv[i+1][0] != '-')
	65	{
	66	i++;
	67	pset->SetFileSaveStyle (FSS_NUMFILES);
	68	pset->SetNumSaveDir (argv[i]);
	69	}
	70	}
	71	// force saving to the given file name
	72	else if ((stricmp (argv[i], "-f") == 0) && (i < argc-1))
	73	{
	74	if (argv[i+1][0] != '-')
	75	{
	76	i++;
	77	pset->SetFileSaveStyle (FSS_FORCEFILE);
	78	pset->SetForceSaveFile (argv[i]);
[27]	79	g_fSetPathName = TRUE;
[2]	80	}
	81	}
	82	// set to idle priority
	83	else if (stricmp (argv[i], "-i") == 0)
	84	{
[27]	85	g_fIdle = TRUE;
[2]	86	pset->SetFlag (SEI_IDLEPRIORITY, TRUE);
	87	}
[59]	88	#else
[57]	89	// use default PM print screen
[65]	90	if (stricmp (argv[i], "-p") == 0)
[57]	91	{
	92	g_usePMps = TRUE;
	93	}
	94	#endif
[2]	95	}
	96
	97	return fAuto;
	98	}
	99
	100	// main ************************************************************** /FOLD00/
	101
	102	int main (int argc, PSZ argv[])
	103	{
[13]	104	ScopedExceptqLoader sel;
[2]	105	#ifdef _DOLOGDEBUG_
	106	LogDebug( "Gotcha! start" );
	107	#endif
	108	#ifdef _DOLOGMEM_
	109	LogMem("main", TRUE);
	110	#endif
	111
	112	// init system and msg queue
	113	hab = WinInitialize (0);
	114	HMQ hmq = WinCreateMsgQueue (hab, 0);
	115
	116	//DisplayError("DEBUG", "%d", Version());
[30]	117	#ifdef _QUIET_
[50]	118	if ((Version() < 2))
[2]	119	{
[55]	120	DisplayError("GOTCHDLL.DLL Outdated",
	121	"The file gotchdll.dll is is outdated. You should have "
	122	"received a new version with the program, check for an "
	123	"older version of gotchdll.dll in your LIBPATH. Is the "
	124	"new gotchdll.dll in a directory in your LIBPATH?");
[2]	125	exit (0);
	126	}
[27]	127	int rc;
[50]	128	HMTX hmtx = NULLHANDLE;
	129	// running multiple instances orphans all but the first closed hook unloaded
	130	rc = DosOpenMutexSem("\\SEM32\\GOTCHA", &hmtx);
	131	if (!rc) {
[55]	132	DisplayError("GOTCHA QUIET Already running",
	133	"Another instance of Gotcha Quiet is running. You can "
	134	"open Gotcha Quiet's settings using the Alt+PrintScreen "
	135	"key combination");
[50]	136	exit (0);
[27]	137	}
[50]	138	rc = DosCreateMutexSem("\\SEM32\\GOTCHA", &hmtx, 0, FALSE);
	139	if (rc) {
	140	DisplayError("Semaphore creation failed",
	141	"Try restarting Gotcha Quiet");
	142	exit (0);
	143	}
[30]	144	#endif
[27]	145
[2]	146	// register our window classes
	147	WinRegisterClass (hab, "thth.wc.gotcha.main", WindowProcedure, 0L,
	148	sizeof (ULONG)*2L);
	149	WinRegisterClass (hab, "thth.wc.gotcha.snapshot", wpSnapshot,
	150	CS_SIZEREDRAW, sizeof (ULONG)*2L);
	151
	152	// load the settings
	153	pset = new SETTINGS;
	154
[27]	155	pset->idleSetInIni = pset->QueryFlag(SEI_IDLEPRIORITY);
	156	pset->saveStyle = pset->QuerySaveStyle ();
	157	pset->pNumSaveDir = pset->QueryNumSaveDir ();
	158	pset->pForceSaveFile = pset->QueryForceSaveFile();
	159	pset->bSerialCapture = pset->SerialCapture ();
[50]	160
[2]	161	// check cmd line args and if "-a" found take screenshot and exit
	162	if (CheckCmdlineArgs (argc, argv))
	163	{
	164	CaptureWindow (HWND_DESKTOP, HWND_DESKTOP, NULL, TRUE);
[27]	165	pset->SetFileSaveStyle(pset->saveStyle);
[38]	166	pset->SetNumSaveDir (pset->pNumSaveDir);
[2]	167	delete pset;
	168	WinDestroyMsgQueue (hmq);
	169	WinTerminate (hab);
	170	exit (0);
	171	}
	172
	173	SetIdlePriority(pset->QueryFlag(SEI_IDLEPRIORITY));
	174
	175	// create the windows
	176	hwndFrame = CreateMainWindow ();
	177	hwndSnapshot = CreateSnapshotWindow ();
	178
	179	SWP swp;
	180	USHORT us[7];
	181
	182	// position main window
	183	pset->QueryWindowData (&swp, us, FALSE);
	184	WinSetWindowPos (hwndFrame, HWND_DESKTOP, swp.x,swp.y, 0,0,
	185	SWP_SHOW \| SWP_MOVE);
	186
[38]	187
[30]	188	#ifdef _QUIET_
[57]	189	bool fPrtScr = FALSE;
	190	if (!g_usePMps) {
	191	// always turn it off while running
	192	// WinSet does not effect the ini setting
	193	WinSetSysValue(HWND_DESKTOP, SV_PRINTSCREEN , fPrtScr);
	194	}
	195	InitDLL (hab, hwndFrame, g_usePMps);
	196	StartInputHook ();
[2]	197	#endif
	198
	199	// position snapshot window
	200	pset->QueryWindowData (&swp, us);
	201
	202	// size, activate & show window
	203	WinSetWindowPos (hwndSnapshot, HWND_DESKTOP, swp.x,swp.y, swp.cx,swp.cy,
	204	SWP_SHOW \| SWP_SIZE \| SWP_MOVE);
	205	if (! pset->SnapshotWindow ())
	206	WinShowWindow (hwndSnapshot, FALSE);
	207	else
	208	WinShowWindow (hwndSnapshot, TRUE);
	209
	210	WinSetWindowPos (hwndFrame, NULLHANDLE, 0,0, 0,0, SWP_SHOW);
	211	WinSetWindowPos (WinWindowFromID (hwndFrame, FID_CLIENT), NULLHANDLE,
	212	0,0, 0,0, SWP_SHOW);
	213
	214	g_phelp = new Helper(hwndFrame);
	215
	216
	217	// do the main msg loop
	218	QMSG qmsg;
	219	while (WinGetMsg (hab, &qmsg, 0L, 0, 0))
	220	WinDispatchMsg (hab, &qmsg);
	221
[27]	222
	223	// Set priorty to unless user set regular since running -q sets it to idle in the ini
[38]	224	if (g_fIdle) {
[27]	225	pset->SetFlag (SEI_IDLEPRIORITY, pset->idleSetInIni);
	226	}
	227	// Don't change the ini stored paths if the paths were set from the command line
	228	if (g_fSetPathName) {
	229	pset->SetFileSaveStyle(pset->saveStyle);
	230	pset->SetForceSaveFile (pset->pForceSaveFile);
	231	pset->SetNumSaveDir (pset->pNumSaveDir);
	232	}
[2]	233	// save size, etc. of snapshot window
	234	WinQueryWindowPos (hwndSnapshot, &swp);
	235	pset->SetWindowData (&swp);
	236
	237	// save size, etc. of main window
	238	WinQueryWindowPos (hwndFrame, &swp);
	239	pset->SetWindowData (&swp, FALSE);
	240
	241	// goodbye windows!
	242	WinDestroyWindow (hwndSnapshot);
	243	WinDestroyWindow (hwndFrame);
	244
[30]	245	#ifdef _QUIET_
[52]	246	// Reset to user PM print screen choice
[57]	247	if (!g_usePMps) {
	248	ULONG ulDataSize = 0;
	249	rc = PrfQueryProfileSize(HINI_USERPROFILE, "PM_ControlPanel",
	250	"PrintScreen", &ulDataSize );
	251	rc = PrfQueryProfileData(HINI_USERPROFILE, "PM_ControlPanel",
	252	"PrintScreen", &fPrtScr, &ulDataSize);
	253	if (!rc) // Print screen is on by default (no ini entry)
	254	fPrtScr = TRUE;
	255	WinSetSysValue(HWND_DESKTOP, SV_PRINTSCREEN , fPrtScr);
	256	}
[38]	257	StopInputHook ();
[50]	258	DosCloseMutexSem(hmtx);
[2]	259	#endif
[50]	260
[2]	261	delete g_phelp;
	262	delete pset;
	263
	264	WinDestroyMsgQueue (hmq);
	265	WinTerminate (hab);
	266
	267	#ifdef _DOLOGMEM_
	268	LogMem("main", FALSE);
	269	#endif
	270	#ifdef _DOLOGDEBUG_
	271	LogDebug( "Gotcha! end" );
	272	#endif
	273	}
	274
	275	// DisplayError ****************************************************** /FOLD00/
	276
	277	VOID DisplayError (PSZ pszTitle, PSZ psz, ...)
	278	{
	279	CHAR dstring[401];
	280	va_list valst;
	281
	282	va_start (valst, psz);
	283	vsnprintf (dstring, 401, psz, valst);
	284	va_end (valst);
	285
	286	WinMessageBox (HWND_DESKTOP, WinQueryActiveWindow (HWND_DESKTOP), dstring,
	287	pszTitle, 0, MB_OK \| MB_ERROR \| MB_APPLMODAL \| MB_MOVEABLE);
	288	}
	289
[38]	290	// saymsg2 was adapted from code in FM/2
[27]	291	APIRET saymsg2(int DefaultButton, HWND hwnd, PCSZ pszTitle, PCSZ pszFmt, ...)
	292	{
	293	ULONG i;
	294	APIRET rc;
	295	CHAR szMsg[4096];
	296	va_list va;
	297	MB2INFO *pmbInfo;
	298	MB2D mb2dBut[3];
	299	ULONG ulInfoSize = (sizeof(MB2INFO) + (sizeof(MB2D) * 2));
[2]	300
[27]	301	va_start(va, pszFmt);
	302	szMsg[sizeof(szMsg) - 1] = 0;
	303	vsprintf(szMsg, pszFmt, va);
	304	va_end(va);
	305
	306	if (szMsg[sizeof(szMsg) - 1]) {
	307	fprintf(stderr, "Buffer overflow in saymsg2 - need %u bytes\n", strlen(szMsg) + 1);
	308	fflush(stderr);
	309	}
	310
[59]	311	memset(mb2dBut, 0, sizeof(MB2D) * 2);
[55]	312	strcpy(mb2dBut[0].achText,RSTR(IDS_OK));
	313	strcpy(mb2dBut[1].achText,RSTR(IDS_CANCEL));
[59]	314	//strcpy(mb2dBut[2].achText,RSTR(IDS_SETTINGS));
[27]	315	mb2dBut[0].idButton = 1;
	316	mb2dBut[1].idButton = 2;
[59]	317	//mb2dBut[2].idButton = 3;
[27]	318	if (DefaultButton)
	319	mb2dBut[DefaultButton - 1].flStyle = BS_DEFAULT;
	320	pmbInfo = (MB2INFO *) malloc(ulInfoSize);
	321	memset(pmbInfo, 0, ulInfoSize);
	322	if (pmbInfo) {
	323	pmbInfo->cb = ulInfoSize;
	324	pmbInfo->hIcon = 0;
[59]	325	pmbInfo->cButtons = 2;
[27]	326	pmbInfo->flStyle = MB_MOVEABLE \| MB_ICONQUESTION ;
	327	pmbInfo->hwndNotify = NULLHANDLE;
[59]	328	for (i = 0; i < 2; i++) {
[27]	329	memcpy( pmbInfo->mb2d+i , mb2dBut+i , sizeof(MB2D));
	330	}
	331	rc = WinMessageBox2(HWND_DESKTOP, hwnd,
	332	szMsg, pszTitle, SM2_DIALOG,
	333	pmbInfo);
	334	WinSetFocus(HWND_DESKTOP, SM2_DIALOG);
	335	free(pmbInfo);
	336	return rc;
	337	}
	338	return MBID_ERROR;
	339	}
	340	// AddSysMenuItem **************************************************** /FOLD00/
	341
[2]	342	VOID AddSysMenuItem (HWND hwndFrame, MENUITEM *Item, PSZ Text)
	343	{
	344	HWND hwndSysMenu = WinWindowFromID (hwndFrame, FID_SYSMENU);
	345	USHORT idSysMenu = SHORT1FROMMR (WinSendMsg (hwndSysMenu,
	346	MM_ITEMIDFROMPOSITION, NULL,
	347	NULL ));
	348	MENUITEM miSysMenu;
	349	WinSendMsg (hwndSysMenu, MM_QUERYITEM,
	350	MPFROM2SHORT (idSysMenu,FALSE), MPFROMP(&miSysMenu));
	351
	352	HWND hwndSysSubMenu = miSysMenu.hwndSubMenu;
	353
	354	WinSendMsg (hwndSysSubMenu, MM_INSERTITEM, MPFROMP(Item), MPFROMP(Text));
	355	}
	356
	357	// DoCountdown ******************************************************* /fold00/
	358
	359	VOID DoCountdown (ULONG ul)
	360	{
	361	if (ul > 10)
	362	DosBeep (4000L-3000L, 20);
	363	else
	364	DosBeep (4000L-ul*300L, 20);
	365	}
	366
	367	// SetIdlePriority ************************************************** /FOLD00/
	368
	369	VOID SetIdlePriority (BOOL f)
	370	{
[52]	371	#ifndef _QUEIT_
[2]	372	if (f)
	373	DosSetPriority(PRTYS_PROCESS, PRTYC_IDLETIME, 0, 0);
	374	else
	375	DosSetPriority(PRTYS_PROCESS, PRTYC_REGULAR, 0, 0);
[27]	376	#else
	377	DosSetPriority(PRTYS_PROCESS, PRTYC_FOREGROUNDSERVER, 0, 0);
	378	#endif
[2]	379	}
	380
	381	// ***********************************************************************
	382
	383	#ifdef _DOLOGMEM_
	384	VOID LogMem (PSZ psz, BOOL f)
	385	{
	386	FILE *pf = fopen("gotcha.mem","ab");
	387	static ULONG TotalPhysicalMemory, ul = 0;
	388	ULONG i;
	389	if (!f)
	390	ul--;
	391	DosQuerySysInfo(QSV_TOTAVAILMEM, QSV_TOTAVAILMEM, &TotalPhysicalMemory,
	392	sizeof(TotalPhysicalMemory));
	393	for (i = 0; i < ul; i++)
	394	fputs(" ", pf);
	395	fprintf(pf,"%10ld (%s)\n", TotalPhysicalMemory, psz);
	396	if (f)
	397	ul++;
	398	fclose(pf);
	399	}
	400	#endif
	401
	402	#ifdef _DOLOGDEBUG_
	403	VOID LogDebug( PSZ psz, ... )
	404	{
	405	CHAR dstring[401];
	406	va_list valst;
	407
	408	va_start (valst, psz);
	409	vsnprintf (dstring, 401, psz, valst);
	410	va_end (valst);
	411
	412	FILE *pf = fopen( "gotcha.log", "ab" );
	413	fprintf( pf, "%s\n", dstring );
	414	fclose( pf );
	415	}
	416	#endif
	417
	418	// ***********************************************************************

Note: See TracBrowser for help on using the repository browser.

Download in other formats: