1 |
// Vance Thrift and Biller File Utility 2 |
2 |
// |
3 |
// Douglas Thrift |
4 |
// |
5 |
// $Id: ScanUtility.cxx,v 1.6 2003/08/21 06:29:24 douglas Exp $ |
6 |
|
7 |
#include "ScanUtility.h" |
8 |
|
9 |
ScanUtility::ScanUtility() |
10 |
{ |
11 |
number = count++; |
12 |
utilities.insert(pair<unsigned, ScanUtility*>(number, this)); |
13 |
title = new char[programName.length() + 16]; |
14 |
|
15 |
sprintf(title, "%s - Scan Utility", programName.c_str()); |
16 |
loadDirs(); |
17 |
|
18 |
// start |
19 |
wizard[0].dwSize = sizeof(wizard[0]); |
20 |
wizard[0].dwFlags = PSP_DEFAULT | PSP_USEHICON | PSP_USETITLE | |
21 |
PSP_USEHEADERTITLE | PSP_USEHEADERSUBTITLE; |
22 |
wizard[0].hInstance = gui.instance; |
23 |
wizard[0].pszTemplate = MAKEINTRESOURCE(IDD_START); |
24 |
wizard[0].hIcon = gui.icon; |
25 |
wizard[0].pszTitle = title; |
26 |
wizard[0].pfnDlgProc = start; |
27 |
wizard[0].lParam = number; |
28 |
wizard[0].pszHeaderTitle = "Start"; |
29 |
wizard[0].pszHeaderSubTitle = "Change any settings before scanning."; |
30 |
|
31 |
// select |
32 |
wizard[1].dwSize = sizeof(wizard[1]); |
33 |
wizard[1].dwFlags = PSP_DEFAULT | PSP_USEHICON | PSP_USETITLE | |
34 |
PSP_USEHEADERTITLE | PSP_USEHEADERSUBTITLE; |
35 |
wizard[1].hInstance = gui.instance; |
36 |
wizard[1].pszTemplate = MAKEINTRESOURCE(IDD_SELECT); |
37 |
wizard[1].hIcon = gui.icon; |
38 |
wizard[1].pszTitle = title; |
39 |
wizard[1].pfnDlgProc = select; |
40 |
wizard[1].lParam = number; |
41 |
wizard[1].pszHeaderTitle = "Select"; |
42 |
wizard[1].pszHeaderSubTitle = "Choose the scanned document to save."; |
43 |
} |
44 |
|
45 |
ScanUtility::~ScanUtility() |
46 |
{ |
47 |
utilities.erase(number); |
48 |
saveDirs(); |
49 |
|
50 |
delete [] title; |
51 |
} |
52 |
|
53 |
void ScanUtility::run(void) |
54 |
{ |
55 |
PROPSHEETHEADER header; |
56 |
|
57 |
// header |
58 |
header.dwSize = sizeof(header); |
59 |
header.dwFlags = PSH_DEFAULT | PSH_HEADER | PSH_PROPSHEETPAGE | |
60 |
PSH_USEICONID | PSH_WIZARD97; |
61 |
header.hwndParent = NULL; |
62 |
header.hInstance = gui.instance; |
63 |
header.pszIcon = MAKEINTRESOURCE(IDI_VTB_ICON); |
64 |
header.nPages = 2; |
65 |
header.nStartPage = 0; |
66 |
header.ppsp = wizard; |
67 |
header.pszbmHeader = MAKEINTRESOURCE(IDB_VTB_BMP); |
68 |
|
69 |
PropertySheet(&header); |
70 |
} |
71 |
|
72 |
unsigned ScanUtility::count = 0; |
73 |
map<unsigned, ScanUtility*> ScanUtility::utilities; |
74 |
map<HWND, ScanUtility*> ScanUtility::windows; |
75 |
|
76 |
void ScanUtility::loadDirs(void) |
77 |
{ |
78 |
HKEY key; |
79 |
|
80 |
if (RegOpenKeyEx(HKEY_CURRENT_USER, |
81 |
"Software\\DouglasThrift\\VTBFileUtil2", 0, KEY_QUERY_VALUE, &key) == |
82 |
ERROR_SUCCESS) |
83 |
{ |
84 |
DWORD type; |
85 |
char data[MAX_PATH]; |
86 |
DWORD size = MAX_PATH; |
87 |
|
88 |
if (RegQueryValueEx(key, "ScanDir", NULL, &type, LPBYTE(data), &size) |
89 |
== ERROR_SUCCESS) |
90 |
{ |
91 |
data[size - 1] = '\0'; |
92 |
|
93 |
switch (type) |
94 |
{ |
95 |
case REG_EXPAND_SZ: |
96 |
{ |
97 |
char folder[MAX_PATH]; |
98 |
|
99 |
ExpandEnvironmentStrings(data, folder, MAX_PATH); |
100 |
|
101 |
scanDir = folder; |
102 |
} |
103 |
break; |
104 |
case REG_SZ: |
105 |
scanDir = data; |
106 |
break; |
107 |
default: |
108 |
setScanDir(); |
109 |
break; |
110 |
} |
111 |
} |
112 |
else |
113 |
{ |
114 |
setScanDir(); |
115 |
} |
116 |
|
117 |
size = MAX_PATH; |
118 |
|
119 |
if (RegQueryValueEx(key, "SaveDir", NULL, &type, LPBYTE(data), &size) |
120 |
== ERROR_SUCCESS) |
121 |
{ |
122 |
data[size - 1] = '\0'; |
123 |
|
124 |
switch (type) |
125 |
{ |
126 |
case REG_EXPAND_SZ: |
127 |
{ |
128 |
char folder[MAX_PATH]; |
129 |
|
130 |
ExpandEnvironmentStrings(data, folder, MAX_PATH); |
131 |
|
132 |
saveDir = folder; |
133 |
} |
134 |
break; |
135 |
case REG_SZ: |
136 |
saveDir = data; |
137 |
break; |
138 |
default: |
139 |
setSaveDir(); |
140 |
break; |
141 |
} |
142 |
} |
143 |
else |
144 |
{ |
145 |
setSaveDir(); |
146 |
} |
147 |
|
148 |
RegCloseKey(key); |
149 |
} |
150 |
else |
151 |
{ |
152 |
setScanDir(); |
153 |
setSaveDir(); |
154 |
} |
155 |
|
156 |
if (debug) cerr << "scanDir = " << scanDir << "\n" |
157 |
<< "saveDir = " << saveDir << "\n"; |
158 |
} |
159 |
|
160 |
void ScanUtility::saveDirs(void) |
161 |
{ |
162 |
HKEY key; |
163 |
|
164 |
if (RegCreateKeyEx(HKEY_CURRENT_USER, |
165 |
"Software\\DouglasThrift\\VTBFileUtil2", 0, NULL, |
166 |
REG_OPTION_NON_VOLATILE, KEY_QUERY_VALUE | KEY_SET_VALUE, NULL, &key, |
167 |
NULL) == ERROR_SUCCESS) |
168 |
{ |
169 |
DWORD type; |
170 |
char data[MAX_PATH]; |
171 |
DWORD size = MAX_PATH; |
172 |
|
173 |
if (RegQueryValueEx(key, "ScanDir", NULL, &type, LPBYTE(data), &size) |
174 |
== ERROR_SUCCESS) |
175 |
{ |
176 |
data[size - 1] = '\0'; |
177 |
} |
178 |
else |
179 |
{ |
180 |
data[0] = '\0'; |
181 |
} |
182 |
|
183 |
if (scanDir != data || type != REG_SZ) |
184 |
{ |
185 |
if (RegSetValueEx(key, "ScanDir", 0, REG_SZ, |
186 |
LPBYTE(scanDir.c_str()), scanDir.length() + 1) != |
187 |
ERROR_SUCCESS) |
188 |
{ |
189 |
error(); |
190 |
} |
191 |
} |
192 |
|
193 |
size = MAX_PATH; |
194 |
|
195 |
if (RegQueryValueEx(key, "SaveDir", NULL, &type, LPBYTE(data), &size) |
196 |
== ERROR_SUCCESS) |
197 |
{ |
198 |
data[size - 1] = '\0'; |
199 |
} |
200 |
else |
201 |
{ |
202 |
data[0] = '\0'; |
203 |
} |
204 |
|
205 |
if (saveDir != data || type != REG_SZ) |
206 |
{ |
207 |
if (RegSetValueEx(key, "SaveDir", 0, REG_SZ, |
208 |
LPBYTE(saveDir.c_str()), saveDir.length() + 1) != |
209 |
ERROR_SUCCESS) |
210 |
{ |
211 |
error(); |
212 |
} |
213 |
} |
214 |
|
215 |
RegCloseKey(key); |
216 |
} |
217 |
else |
218 |
{ |
219 |
error(); |
220 |
} |
221 |
} |
222 |
|
223 |
void ScanUtility::setScanDir(HWND parent) |
224 |
{ |
225 |
BROWSEINFO info; |
226 |
|
227 |
info.hwndOwner = parent; |
228 |
info.pidlRoot = NULL; |
229 |
info.pszDisplayName = NULL; |
230 |
info.lpszTitle = "Select the Scan Directory"; |
231 |
info.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT; |
232 |
info.lpfn = browse; |
233 |
info.lParam = MAKELPARAM(number, true); |
234 |
info.iImage = 0; |
235 |
|
236 |
do |
237 |
{ |
238 |
LPITEMIDLIST id = SHBrowseForFolder(&info); |
239 |
|
240 |
if (id != NULL) |
241 |
{ |
242 |
char folder[MAX_PATH]; |
243 |
|
244 |
if (SHGetPathFromIDList(id, folder)) |
245 |
{ |
246 |
scanDir = folder; |
247 |
} |
248 |
} |
249 |
|
250 |
LPMALLOC destruct; |
251 |
|
252 |
SHGetMalloc(&destruct); |
253 |
destruct->Free(id); |
254 |
destruct->Release(); |
255 |
|
256 |
if (scanDir == "") |
257 |
{ |
258 |
switch (MessageBox(parent, "Scan Directory needs to be selected.", |
259 |
programName.c_str(), MB_ABORTRETRYIGNORE | MB_ICONERROR)) |
260 |
{ |
261 |
case IDABORT: |
262 |
exit(1); |
263 |
break; |
264 |
case IDRETRY: |
265 |
break; |
266 |
case IDIGNORE: |
267 |
Beep(2200, 250); |
268 |
Beep(1100, 500); |
269 |
Beep(3300, 250); |
270 |
exit(2); |
271 |
break; |
272 |
} |
273 |
} |
274 |
} |
275 |
while (scanDir == ""); |
276 |
|
277 |
if (debug) cerr << "scanDir = " << scanDir << "\n"; |
278 |
} |
279 |
|
280 |
void ScanUtility::setSaveDir(HWND parent) |
281 |
{ |
282 |
BROWSEINFO info; |
283 |
|
284 |
info.hwndOwner = parent; |
285 |
info.pidlRoot = NULL; |
286 |
info.pszDisplayName = NULL; |
287 |
info.lpszTitle = "Select the Save Directory"; |
288 |
info.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT; |
289 |
info.lpfn = browse; |
290 |
info.lParam = MAKELPARAM(number, false); |
291 |
info.iImage = 0; |
292 |
|
293 |
do |
294 |
{ |
295 |
LPITEMIDLIST id = SHBrowseForFolder(&info); |
296 |
|
297 |
if (id != NULL) |
298 |
{ |
299 |
char folder[MAX_PATH]; |
300 |
|
301 |
if (SHGetPathFromIDList(id, folder)) |
302 |
{ |
303 |
saveDir = folder; |
304 |
} |
305 |
} |
306 |
|
307 |
LPMALLOC destruct; |
308 |
|
309 |
SHGetMalloc(&destruct); |
310 |
destruct->Free(id); |
311 |
destruct->Release(); |
312 |
|
313 |
if (saveDir == "") |
314 |
{ |
315 |
switch (MessageBox(parent, "Scan Directory needs to be selected.", |
316 |
programName.c_str(), MB_ABORTRETRYIGNORE | MB_ICONERROR)) |
317 |
{ |
318 |
case IDABORT: |
319 |
exit(1); |
320 |
break; |
321 |
case IDRETRY: |
322 |
break; |
323 |
case IDIGNORE: |
324 |
Beep(2200, 250); |
325 |
Beep(1100, 500); |
326 |
Beep(3300, 250); |
327 |
exit(2); |
328 |
break; |
329 |
} |
330 |
} |
331 |
} |
332 |
while (saveDir == ""); |
333 |
|
334 |
if (debug) cerr << "saveDir = " << saveDir << "\n"; |
335 |
} |
336 |
|
337 |
void ScanUtility::populate(HWND parent) |
338 |
{ |
339 |
set<string> scans; |
340 |
|
341 |
do |
342 |
{ |
343 |
string scan = scanDir + "\\SCAN????_000." + |
344 |
IndividualClient::getExtension(); |
345 |
WIN32_FIND_DATA found; |
346 |
HANDLE finder = FindFirstFile(scan.c_str(), &found); |
347 |
|
348 |
if (finder != INVALID_HANDLE_VALUE) |
349 |
{ |
350 |
do |
351 |
{ |
352 |
scans.insert(found.cFileName); |
353 |
} |
354 |
while (FindNextFile(finder, &found)); |
355 |
|
356 |
FindClose(finder); |
357 |
} |
358 |
|
359 |
if (scans.empty()) |
360 |
{ |
361 |
if (MessageBox(parent, "No scanned documents found.", |
362 |
programName.c_str(), MB_RETRYCANCEL | MB_ICONQUESTION) == |
363 |
IDCANCEL) break; |
364 |
} |
365 |
} |
366 |
while (scans.empty()); |
367 |
|
368 |
if (scans.empty()) |
369 |
{ |
370 |
PropSheet_PressButton(GetParent(parent), PSBTN_BACK); |
371 |
} |
372 |
else |
373 |
{ |
374 |
SHFILEINFO info; |
375 |
HIMAGELIST icons = HIMAGELIST(SHGetFileInfo((*scans.begin()).c_str(), |
376 |
0, &info, sizeof(info), SHGFI_SMALLICON | SHGFI_SYSICONINDEX)); |
377 |
|
378 |
ListView_SetImageList(GetDlgItem(parent, IDC_SELECT_SCANS), icons, |
379 |
LVSIL_SMALL); |
380 |
|
381 |
int index = 0; |
382 |
|
383 |
if (debug) cerr << "scans = {\n"; |
384 |
|
385 |
for (set<string>::iterator itor = scans.begin(); itor != scans.end(); |
386 |
itor++) |
387 |
{ |
388 |
if (debug) cerr << " " << *itor << "\n"; |
389 |
|
390 |
char scan[MAX_PATH]; |
391 |
|
392 |
sprintf(scan, "%s", (*itor).c_str()); |
393 |
|
394 |
LVITEM item; |
395 |
|
396 |
item.mask = LVIF_IMAGE | LVIF_TEXT; |
397 |
item.iItem = index++; |
398 |
item.iSubItem = 0; |
399 |
item.state = LVIS_SELECTED; |
400 |
item.stateMask = LVIS_SELECTED; |
401 |
item.pszText = scan; |
402 |
item.iImage = info.iIcon; |
403 |
|
404 |
ListView_InsertItem(GetDlgItem(parent, IDC_SELECT_SCANS), &item); |
405 |
} |
406 |
|
407 |
if (debug) cerr << "}\n"; |
408 |
} |
409 |
} |
410 |
|
411 |
int ScanUtility::browse(HWND dialog, UINT msg, LPARAM l, LPARAM d) |
412 |
{ |
413 |
map<unsigned, ScanUtility*>::iterator itor = utilities.find(LOWORD(d)); |
414 |
ScanUtility* data = itor->second; |
415 |
|
416 |
switch (msg) |
417 |
{ |
418 |
case BFFM_INITIALIZED: |
419 |
center(dialog); |
420 |
SendMessage(dialog, BFFM_SETOKTEXT, 0, |
421 |
LPARAM(toWide("&Select").c_str())); |
422 |
SendMessage(dialog, BFFM_SETEXPANDED, FALSE, CSIDL_DRIVES); |
423 |
SendMessage(dialog, BFFM_SETSELECTION, TRUE, LPARAM(HIWORD(d) ? |
424 |
data->scanDir.c_str() : data->saveDir.c_str())); |
425 |
break; |
426 |
case BFFM_SELCHANGED: |
427 |
{ |
428 |
IShellFolder* object; |
429 |
|
430 |
SHGetDesktopFolder(&object); |
431 |
|
432 |
STRRET thing; |
433 |
char* folder; |
434 |
|
435 |
object->GetDisplayNameOf(LPCITEMIDLIST(l), SHGDN_FORPARSING, |
436 |
&thing); |
437 |
StrRetToStr(&thing, LPCITEMIDLIST(l), &folder); |
438 |
SendMessage(dialog, BFFM_SETSTATUSTEXT, 0, LPARAM(folder)); |
439 |
|
440 |
if (PathIsUNCServer(folder)) |
441 |
{ |
442 |
SendMessage(dialog, BFFM_ENABLEOK, 0, 0); |
443 |
} |
444 |
|
445 |
CoTaskMemFree(folder); |
446 |
object->Release(); |
447 |
} |
448 |
break; |
449 |
} |
450 |
|
451 |
return 0; |
452 |
} |
453 |
|
454 |
INT_PTR ScanUtility::start(HWND dialog, UINT msg, WPARAM w, LPARAM l) |
455 |
{ |
456 |
map<HWND, ScanUtility*>::iterator itor = windows.find(dialog); |
457 |
ScanUtility* data = itor->second; |
458 |
|
459 |
switch (msg) |
460 |
{ |
461 |
case WM_INITDIALOG: |
462 |
center(GetParent(dialog)); |
463 |
{ |
464 |
LPPROPSHEETPAGE page = LPPROPSHEETPAGE(l); |
465 |
map<unsigned, ScanUtility*>::iterator itor = |
466 |
utilities.find(page->lParam); |
467 |
|
468 |
windows.insert(pair<HWND, ScanUtility*>(dialog, itor->second)); |
469 |
} |
470 |
{ |
471 |
ostringstream instructions; |
472 |
|
473 |
instructions << "1.\tIf you need instructions, you should not be r" |
474 |
<< "unning this program\n\tin Scan Utility mode.\n" |
475 |
<< "2.\tOtherwise, go forth and scan.\n" |
476 |
<< "3.\tThen come back and click Next.\n"; |
477 |
|
478 |
SetDlgItemText(dialog, IDC_START_INSTRUCTIONS, |
479 |
instructions.str().c_str()); |
480 |
} |
481 |
break; |
482 |
case WM_NOTIFY: |
483 |
{ |
484 |
LPNMHDR nm = LPNMHDR(l); |
485 |
|
486 |
switch (nm->code) |
487 |
{ |
488 |
case PSN_SETACTIVE: |
489 |
PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_NEXT); |
490 |
{ |
491 |
char folder[44]; |
492 |
|
493 |
PathCompactPathEx(folder, data->scanDir.c_str(), 44, 0); |
494 |
SetDlgItemText(dialog, IDC_START_SCAN_TEXT, folder); |
495 |
PathCompactPathEx(folder, data->saveDir.c_str(), 44, 0); |
496 |
SetDlgItemText(dialog, IDC_START_SAVE_TEXT, folder); |
497 |
} |
498 |
break; |
499 |
case PSN_WIZNEXT: |
500 |
SetCurrentDirectory(data->scanDir.c_str()); |
501 |
break; |
502 |
} |
503 |
} |
504 |
break; |
505 |
case WM_COMMAND: |
506 |
switch (LOWORD(w)) |
507 |
{ |
508 |
case IDC_START_SCAN_BROWSE: |
509 |
data->setScanDir(dialog); |
510 |
{ |
511 |
char folder[44]; |
512 |
|
513 |
PathCompactPathEx(folder, data->scanDir.c_str(), 44, 0); |
514 |
SetDlgItemText(dialog, IDC_START_SCAN_TEXT, folder); |
515 |
} |
516 |
break; |
517 |
case IDC_START_SAVE_BROWSE: |
518 |
data->setSaveDir(dialog); |
519 |
{ |
520 |
char folder[44]; |
521 |
|
522 |
PathCompactPathEx(folder, data->saveDir.c_str(), 44, 0); |
523 |
SetDlgItemText(dialog, IDC_START_SAVE_TEXT, folder); |
524 |
} |
525 |
break; |
526 |
} |
527 |
break; |
528 |
} |
529 |
|
530 |
return FALSE; |
531 |
} |
532 |
|
533 |
INT_PTR ScanUtility::select(HWND dialog, UINT msg, WPARAM w, LPARAM l) |
534 |
{ |
535 |
map<HWND, ScanUtility*>::iterator itor = windows.find(dialog); |
536 |
ScanUtility* data = itor->second; |
537 |
|
538 |
switch (msg) |
539 |
{ |
540 |
case WM_INITDIALOG: |
541 |
{ |
542 |
LPPROPSHEETPAGE page = LPPROPSHEETPAGE(l); |
543 |
map<unsigned, ScanUtility*>::iterator itor = |
544 |
utilities.find(page->lParam); |
545 |
|
546 |
windows.insert(pair<HWND, ScanUtility*>(dialog, itor->second)); |
547 |
} |
548 |
{ |
549 |
LVCOLUMN column; |
550 |
|
551 |
column.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH; |
552 |
column.fmt = LVCFMT_IMAGE; |
553 |
column.cx = 120; |
554 |
column.pszText = "Scanned Documents"; |
555 |
|
556 |
ListView_InsertColumn(GetDlgItem(dialog, IDC_SELECT_SCANS), 0, |
557 |
&column); |
558 |
} |
559 |
break; |
560 |
case WM_NOTIFY: |
561 |
if (w == IDC_SELECT_SCANS) |
562 |
{ |
563 |
switch (ListView_GetSelectedCount(GetDlgItem(dialog, |
564 |
IDC_SELECT_SCANS))) |
565 |
{ |
566 |
case 1: |
567 |
PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK | |
568 |
PSWIZB_NEXT); |
569 |
break; |
570 |
default: |
571 |
PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK); |
572 |
break; |
573 |
} |
574 |
} |
575 |
else |
576 |
{ |
577 |
LPNMHDR nm = LPNMHDR(l); |
578 |
|
579 |
switch (nm->code) |
580 |
{ |
581 |
case PSN_SETACTIVE: |
582 |
PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK); |
583 |
data->populate(dialog); |
584 |
break; |
585 |
case PSN_WIZBACK: |
586 |
ListView_DeleteAllItems(GetDlgItem(dialog, IDC_SELECT_SCANS)); |
587 |
break; |
588 |
case PSN_WIZNEXT: |
589 |
// |
590 |
ListView_DeleteAllItems(GetDlgItem(dialog, IDC_SELECT_SCANS)); |
591 |
break; |
592 |
} |
593 |
} |
594 |
break; |
595 |
case WM_COMMAND: |
596 |
break; |
597 |
} |
598 |
|
599 |
return FALSE; |
600 |
} |