ViewVC Help
View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing
root/proj/trunk/VTBFileUtil2/ScanUtility.cxx
Revision: 295
Committed: 2003-09-11T21:26:54-07:00 (21 years, 9 months ago) by douglas
File size: 24102 byte(s)
Log Message:
Using strsafe.h! Fancy!

File Contents

# Content
1 // Vance Thrift and Biller File Utility 2
2 //
3 // Douglas Thrift
4 //
5 // $Id: ScanUtility.cxx,v 1.18 2003/09/12 04:26:54 douglas Exp $
6
7 #include "ScanUtility.h"
8
9 ScanUtility::ScanUtility() : DiscBrowse()
10 {
11 number = count++;
12
13 utilities.insert(pair<unsigned, ScanUtility*>(number, this));
14
15 title = programName + " - Scan Utility";
16
17 // start
18 wizard[0].dwSize = sizeof(wizard[0]);
19 wizard[0].dwFlags = PSP_DEFAULT | PSP_USETITLE | PSP_USEHEADERTITLE |
20 PSP_USEHEADERSUBTITLE;
21 wizard[0].hInstance = gui.instance;
22 wizard[0].pszTemplate = MAKEINTRESOURCE(IDD_START);
23 wizard[0].pszTitle = title.c_str();
24 wizard[0].pfnDlgProc = start;
25 wizard[0].lParam = number;
26 wizard[0].pszHeaderTitle = "Start";
27 wizard[0].pszHeaderSubTitle = "Change any settings before scanning.";
28
29 // select
30 wizard[1].dwSize = sizeof(wizard[1]);
31 wizard[1].dwFlags = PSP_DEFAULT | PSP_USETITLE | PSP_USEHEADERTITLE |
32 PSP_USEHEADERSUBTITLE;
33 wizard[1].hInstance = gui.instance;
34 wizard[1].pszTemplate = MAKEINTRESOURCE(IDD_SELECT);
35 wizard[1].pszTitle = title.c_str();
36 wizard[1].pfnDlgProc = select;
37 wizard[1].lParam = number;
38 wizard[1].pszHeaderTitle = "Select";
39 wizard[1].pszHeaderSubTitle = "Choose the scanned document to save.";
40
41 // enter
42 wizard[2].dwSize = sizeof(wizard[2]);
43 wizard[2].dwFlags = PSP_DEFAULT | PSP_USETITLE | PSP_USEHEADERTITLE |
44 PSP_USEHEADERSUBTITLE;
45 wizard[2].hInstance = gui.instance;
46 wizard[2].pszTemplate = MAKEINTRESOURCE(IDD_ENTER);
47 wizard[2].pszTitle = title.c_str();
48 wizard[2].pfnDlgProc = enter;
49 wizard[2].lParam = number;
50 wizard[2].pszHeaderTitle = "Enter";
51 wizard[2].pszHeaderSubTitle = "Input the client information.";
52
53 // confirm
54 wizard[3].dwSize = sizeof(wizard[3]);
55 wizard[3].dwFlags = PSP_DEFAULT | PSP_USETITLE | PSP_USEHEADERTITLE |
56 PSP_USEHEADERSUBTITLE;
57 wizard[3].hInstance = gui.instance;
58 wizard[3].pszTemplate = MAKEINTRESOURCE(IDD_CONFIRM);
59 wizard[3].pszTitle = title.c_str();
60 wizard[3].pfnDlgProc = confirm;
61 wizard[3].lParam = number;
62 wizard[3].pszHeaderTitle = "Confirm";
63 wizard[3].pszHeaderSubTitle = "Make sure the file information is correct.";
64
65 // complete
66 wizard[4].dwSize = sizeof(wizard[3]);
67 wizard[4].dwFlags = PSP_DEFAULT | PSP_USETITLE | PSP_USEHEADERTITLE |
68 PSP_USEHEADERSUBTITLE;
69 wizard[4].hInstance = gui.instance;
70 wizard[4].pszTemplate = MAKEINTRESOURCE(IDD_DONE);
71 wizard[4].pszTitle = title.c_str();
72 wizard[4].pfnDlgProc = done;
73 wizard[4].lParam = number;
74 wizard[4].pszHeaderTitle = "Done";
75 wizard[4].pszHeaderSubTitle = "Exit or start over for another scan.";
76 }
77
78 ScanUtility::~ScanUtility()
79 {
80 utilities.erase(number);
81 }
82
83 void ScanUtility::run(void)
84 {
85 loadDirs();
86
87 PROPSHEETHEADER header;
88
89 // header
90 header.dwSize = sizeof(header);
91 header.dwFlags = PSH_DEFAULT | PSH_HEADER | PSH_PROPSHEETPAGE |
92 PSH_USEICONID | PSH_WIZARD97 | PSH_WIZARDHASFINISH;
93 header.hwndParent = NULL;
94 header.hInstance = gui.instance;
95 header.pszIcon = MAKEINTRESOURCE(IDI_VTB_ICON);
96 header.nPages = 5;
97 header.nStartPage = 0;
98 header.ppsp = wizard;
99 header.pszbmHeader = MAKEINTRESOURCE(IDB_VTB_BMP);
100
101 PropertySheet(&header);
102 saveDirs();
103 }
104
105 map<unsigned, ScanUtility*> ScanUtility::utilities;
106 map<HWND, ScanUtility*> ScanUtility::windows;
107
108 void ScanUtility::loadDirs(void)
109 {
110 HKEY key;
111
112 if (RegOpenKeyEx(HKEY_CURRENT_USER,
113 "Software\\DouglasThrift\\VTBFileUtil2", 0, KEY_QUERY_VALUE, &key) ==
114 ERROR_SUCCESS)
115 {
116 DWORD type;
117 char data[MAX_PATH];
118 DWORD size = MAX_PATH;
119
120 if (RegQueryValueEx(key, "ScanDir", NULL, &type, LPBYTE(data), &size)
121 == ERROR_SUCCESS)
122 {
123 data[size - 1] = '\0';
124
125 switch (type)
126 {
127 case REG_EXPAND_SZ:
128 {
129 char folder[MAX_PATH];
130
131 ExpandEnvironmentStrings(data, folder, MAX_PATH);
132
133 scanDir = tail(folder);
134 }
135 break;
136 case REG_SZ:
137 scanDir = tail(data);
138 break;
139 default:
140 setScanDir();
141 break;
142 }
143 }
144 else
145 {
146 setScanDir();
147 }
148
149 size = MAX_PATH;
150
151 if (RegQueryValueEx(key, "SaveDir", NULL, &type, LPBYTE(data), &size)
152 == ERROR_SUCCESS)
153 {
154 data[size - 1] = '\0';
155
156 switch (type)
157 {
158 case REG_EXPAND_SZ:
159 {
160 char folder[MAX_PATH];
161
162 ExpandEnvironmentStrings(data, folder, MAX_PATH);
163
164 saveDir = tail(folder);
165 }
166 break;
167 case REG_SZ:
168 saveDir = tail(data);
169 break;
170 default:
171 setSaveDir();
172 break;
173 }
174 }
175 else
176 {
177 setSaveDir();
178 }
179
180 RegCloseKey(key);
181 }
182 else
183 {
184 setScanDir();
185 setSaveDir();
186 }
187
188 if (debug) cerr << "scanDir = " << scanDir << "\n"
189 << "saveDir = " << saveDir << "\n";
190 }
191
192 void ScanUtility::saveDirs(void)
193 {
194 HKEY key;
195
196 if (LONG code = RegCreateKeyEx(HKEY_CURRENT_USER,
197 "Software\\DouglasThrift\\VTBFileUtil2", 0, NULL,
198 REG_OPTION_NON_VOLATILE, KEY_QUERY_VALUE | KEY_SET_VALUE, NULL, &key,
199 NULL) == ERROR_SUCCESS)
200 {
201 DWORD type;
202 char data[MAX_PATH];
203 DWORD size = MAX_PATH;
204
205 if (RegQueryValueEx(key, "ScanDir", NULL, &type, LPBYTE(data), &size)
206 == ERROR_SUCCESS)
207 {
208 data[size - 1] = '\0';
209 }
210 else
211 {
212 data[0] = '\0';
213 }
214
215 if (scanDir != data || type != REG_SZ)
216 {
217 if (RegSetValueEx(key, "ScanDir", 0, REG_SZ,
218 LPBYTE(scanDir.c_str()), scanDir.length() + 1) !=
219 ERROR_SUCCESS)
220 {
221 error(NULL, code);
222 }
223 }
224
225 size = MAX_PATH;
226
227 if (RegQueryValueEx(key, "SaveDir", NULL, &type, LPBYTE(data), &size)
228 == ERROR_SUCCESS)
229 {
230 data[size - 1] = '\0';
231 }
232 else
233 {
234 data[0] = '\0';
235 }
236
237 if (saveDir != data || type != REG_SZ)
238 {
239 if (RegSetValueEx(key, "SaveDir", 0, REG_SZ,
240 LPBYTE(saveDir.c_str()), saveDir.length() + 1) !=
241 ERROR_SUCCESS)
242 {
243 error(NULL, code);
244 }
245 }
246
247 RegCloseKey(key);
248 }
249 else
250 {
251 error(NULL, code);
252 }
253 }
254
255 void ScanUtility::setScanDir(HWND parent)
256 {
257 char buffer[MAX_PATH];
258
259 SHGetFolderPath(parent, CSIDL_FLAG_CREATE | CSIDL_DESKTOPDIRECTORY, NULL,
260 0, buffer);
261
262 BROWSEINFO info;
263
264 info.hwndOwner = parent;
265 info.pidlRoot = NULL;
266 info.pszDisplayName = NULL;
267 info.lpszTitle = "Select the Scan Directory";
268 info.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT;
269 info.lpfn = browse;
270 info.lParam = MAKELPARAM(number, true);
271 info.iImage = 0;
272
273 do
274 {
275 LPITEMIDLIST id = SHBrowseForFolder(&info);
276
277 if (id != NULL)
278 {
279 char folder[MAX_PATH];
280
281 if (SHGetPathFromIDList(id, folder)) scanDir = tail(folder);
282 }
283
284 LPMALLOC destruct;
285
286 SHGetMalloc(&destruct);
287 destruct->Free(id);
288 destruct->Release();
289
290 if (scanDir == "")
291 {
292 switch (MessageBox(parent, "Scan Directory needs to be selected.",
293 title.c_str(), MB_ABORTRETRYIGNORE | MB_ICONERROR))
294 {
295 case IDABORT:
296 exit(1);
297 break;
298 case IDRETRY:
299 break;
300 case IDIGNORE:
301 Beep(2200, 250);
302 Beep(1100, 500);
303 Beep(3300, 250);
304 exit(2);
305 break;
306 }
307 }
308 }
309 while (scanDir == "");
310
311 if (debug) cerr << "scanDir = " << scanDir << "\n";
312 }
313
314 void ScanUtility::setSaveDir(HWND parent)
315 {
316 char buffer[MAX_PATH];
317
318 SHGetFolderPath(parent, CSIDL_FLAG_CREATE | CSIDL_DESKTOPDIRECTORY, NULL,
319 0, buffer);
320
321 BROWSEINFO info;
322
323 info.hwndOwner = parent;
324 info.pidlRoot = NULL;
325 info.pszDisplayName = NULL;
326 info.lpszTitle = "Select the Save Directory";
327 info.ulFlags = BIF_RETURNONLYFSDIRS | BIF_STATUSTEXT;
328 info.lpfn = browse;
329 info.lParam = MAKELPARAM(number, false);
330 info.iImage = 0;
331
332 do
333 {
334 LPITEMIDLIST id = SHBrowseForFolder(&info);
335
336 if (id != NULL)
337 {
338 char folder[MAX_PATH];
339
340 if (SHGetPathFromIDList(id, folder)) saveDir = tail(folder);
341 }
342
343 LPMALLOC destruct;
344
345 SHGetMalloc(&destruct);
346 destruct->Free(id);
347 destruct->Release();
348
349 if (saveDir == "")
350 {
351 switch (MessageBox(parent, "Scan Directory needs to be selected.",
352 title.c_str(), MB_ABORTRETRYIGNORE | MB_ICONERROR))
353 {
354 case IDABORT:
355 exit(1);
356 break;
357 case IDRETRY:
358 break;
359 case IDIGNORE:
360 Beep(2200, 250);
361 Beep(1100, 500);
362 Beep(3300, 250);
363 exit(2);
364 break;
365 }
366 }
367 }
368 while (saveDir == "");
369
370 if (debug) cerr << "saveDir = " << saveDir << "\n";
371 }
372
373 void ScanUtility::populate(HWND parent)
374 {
375 set<string> scans;
376
377 do
378 {
379 string scan = scanDir + "SCAN????_000." +
380 IndividualClient::getExtension();
381 WIN32_FIND_DATA found;
382 HANDLE finder = FindFirstFile(scan.c_str(), &found);
383
384 if (finder != INVALID_HANDLE_VALUE)
385 {
386 do
387 {
388 scans.insert(found.cFileName);
389 }
390 while (FindNextFile(finder, &found));
391
392 FindClose(finder);
393 }
394
395 if (scans.empty())
396 {
397 if (MessageBox(parent, "No scanned documents found.",
398 title.c_str(), MB_RETRYCANCEL | MB_ICONQUESTION) == IDCANCEL)
399 break;
400 }
401 }
402 while (scans.empty());
403
404 if (scans.empty())
405 {
406 PropSheet_PressButton(GetParent(parent), PSBTN_BACK);
407 }
408 else
409 {
410 SHFILEINFO info;
411 HIMAGELIST icons = HIMAGELIST(SHGetFileInfo((*scans.begin()).c_str(),
412 0, &info, sizeof(info), SHGFI_SMALLICON | SHGFI_SYSICONINDEX));
413
414 ListView_SetImageList(GetDlgItem(parent, IDC_SELECT_SCANS), icons,
415 LVSIL_SMALL);
416
417 if (debug) cerr << "scans = {\n";
418
419 for (set<string>::iterator itor = scans.begin(); itor != scans.end();
420 itor++)
421 {
422 if (debug) cerr << " " << *itor << "\n";
423
424 char scan[MAX_PATH];
425
426 StringCchPrintf(scan, MAX_PATH, "%s", (*itor).c_str());
427
428 LVITEM item;
429
430 item.mask = LVIF_IMAGE | LVIF_TEXT;
431 item.iItem = 0;
432 item.iSubItem = 0;
433 item.pszText = scan;
434 item.iImage = info.iIcon;
435
436 ListView_InsertItem(GetDlgItem(parent, IDC_SELECT_SCANS), &item);
437 }
438
439 if (debug) cerr << "}\n";
440
441 ListView_SetItemState(GetDlgItem(parent, IDC_SELECT_SCANS), 0,
442 LVIS_SELECTED, LVIS_SELECTED);
443 }
444 }
445
446 ScanUtility* ScanUtility::which(HWND window)
447 {
448 map<HWND, ScanUtility*>::iterator itor = windows.find(window);
449
450 return itor->second;
451 }
452
453 ScanUtility* ScanUtility::which(HWND window, LPARAM l)
454 {
455 LPPROPSHEETPAGE page = LPPROPSHEETPAGE(l);
456 map<unsigned, ScanUtility*>::iterator itor = utilities.find(page->lParam);
457
458 windows.insert(pair<HWND, ScanUtility*>(window, itor->second));
459
460 return itor->second;
461 }
462
463 int ScanUtility::browse(HWND dialog, UINT msg, LPARAM l, LPARAM d)
464 {
465 map<unsigned, ScanUtility*>::iterator itor = utilities.find(LOWORD(d));
466 ScanUtility* data = itor->second;
467
468 switch (msg)
469 {
470 case BFFM_INITIALIZED:
471 center(dialog);
472 SendMessage(dialog, BFFM_SETOKTEXT, 0,
473 LPARAM(toWide("&Select").c_str()));
474 SendMessage(dialog, BFFM_SETEXPANDED, FALSE, CSIDL_DRIVES);
475
476 if ((HIWORD(d) ? data->scanDir : data->saveDir) != "")
477 {
478 IShellFolder* desktop;
479 LPWSTR path = new WCHAR[(HIWORD(d) ? data->scanDir.length() :
480 data->saveDir.length()) + 1];
481 LPITEMIDLIST id;
482
483 StringCchPrintfW(path, (HIWORD(d) ? data->scanDir.length() :
484 data->saveDir.length()) + 1, toWide(HIWORD(d) ? data->scanDir :
485 data->saveDir).c_str());
486
487 if (debug) cerr << "path = " << toAnsi(path) << "\n";
488
489 SHGetDesktopFolder(&desktop);
490 desktop->ParseDisplayName(dialog, NULL, path, NULL, &id, NULL);
491
492 if (id != NULL)
493 {
494 SendMessage(dialog, BFFM_SETSELECTION, FALSE, LPARAM(id));
495
496 LPMALLOC destruct;
497
498 SHGetMalloc(&destruct);
499 destruct->Free(id);
500 destruct->Release();
501 }
502
503 desktop->Release();
504 }
505 break;
506 case BFFM_SELCHANGED:
507 {
508 SHFILEINFO info;
509
510 SHGetFileInfo(LPCSTR(l), 0, &info, sizeof(info), SHGFI_DISPLAYNAME
511 | SHGFI_PIDL);
512 SendMessage(dialog, BFFM_SETSTATUSTEXT, 0,
513 LPARAM(info.szDisplayName));
514
515 char folder[MAX_PATH];
516
517 if (!SHGetPathFromIDList(LPCITEMIDLIST(l), folder))
518 {
519 SendMessage(dialog, BFFM_ENABLEOK, 0, 0);
520 }
521 else
522 {
523 SendMessage(dialog, BFFM_SETSTATUSTEXT, 0, LPARAM(folder));
524 }
525 }
526 break;
527 }
528
529 return 0;
530 }
531
532 INT_PTR ScanUtility::start(HWND dialog, UINT msg, WPARAM w, LPARAM l)
533 {
534 ScanUtility* data = which(dialog);
535
536 switch (msg)
537 {
538 case WM_INITDIALOG:
539 center(GetParent(dialog));
540 SendMessage(GetParent(dialog), WM_SETICON, ICON_BIG, LPARAM(gui.icon));
541
542 data = which(dialog, l);
543
544 {
545 ostringstream instructions;
546
547 instructions << "1. If you need instructions, you should not be ru"
548 << "nning this program in Scan Utility mode.\n"
549 << "2. Otherwise, go forth and scan.\n"
550 << "3. Then come back and click Next.\n";
551
552 SetDlgItemText(dialog, IDC_START_INSTRUCTIONS,
553 instructions.str().c_str());
554 }
555 break;
556 case WM_NOTIFY:
557 {
558 LPNMHDR nm = LPNMHDR(l);
559
560 switch (nm->code)
561 {
562 case PSN_SETACTIVE:
563 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_FINISH |
564 PSWIZB_NEXT);
565
566 {
567 SHFILEINFO info;
568
569 SHGetFileInfo(data->scanDir.c_str(), 0, &info,
570 sizeof(info), SHGFI_ICONLOCATION);
571
572 HICON icon = ExtractIcon(gui.instance, info.szDisplayName,
573 info.iIcon);
574
575 SendDlgItemMessage(dialog, IDC_START_SCAN_ICON,
576 STM_SETIMAGE, IMAGE_ICON, LPARAM(icon));
577 SHGetFileInfo(data->saveDir.c_str(), 0, &info,
578 sizeof(info), SHGFI_ICONLOCATION);
579
580 icon = ExtractIcon(gui.instance, info.szDisplayName,
581 info.iIcon);
582
583 SendDlgItemMessage(dialog, IDC_START_SAVE_ICON,
584 STM_SETIMAGE, IMAGE_ICON, LPARAM(icon));
585 }
586
587 {
588 char folder[38];
589
590 PathCompactPathEx(folder, data->scanDir.substr(0,
591 data->scanDir.length() - 1).c_str(), 38, 0);
592 SetDlgItemText(dialog, IDC_START_SCAN_TEXT, folder);
593 PathCompactPathEx(folder, data->saveDir.substr(0,
594 data->saveDir.length() - 1).c_str(), 38, 0);
595 SetDlgItemText(dialog, IDC_START_SAVE_TEXT, folder);
596 }
597 break;
598 case PSN_WIZNEXT:
599 SetCurrentDirectory(data->scanDir.c_str());
600 break;
601 }
602 }
603 break;
604 case WM_COMMAND:
605 switch (LOWORD(w))
606 {
607 case IDC_START_SCAN_BROWSE:
608 data->setScanDir(dialog);
609
610 {
611 SHFILEINFO info;
612
613 SHGetFileInfo(data->scanDir.c_str(), 0, &info, sizeof(info),
614 SHGFI_ICONLOCATION);
615
616 HICON icon = ExtractIcon(gui.instance, info.szDisplayName,
617 info.iIcon);
618
619 SendDlgItemMessage(dialog, IDC_START_SCAN_ICON, STM_SETIMAGE,
620 IMAGE_ICON, LPARAM(icon));
621 }
622
623 {
624 char folder[38];
625
626 PathCompactPathEx(folder, data->scanDir.substr(0,
627 data->scanDir.length() - 1).c_str(), 38, 0);
628 SetDlgItemText(dialog, IDC_START_SCAN_TEXT, folder);
629 }
630 break;
631 case IDC_START_SAVE_BROWSE:
632 data->setSaveDir(dialog);
633
634 {
635 SHFILEINFO info;
636
637 SHGetFileInfo(data->saveDir.c_str(), 0, &info, sizeof(info),
638 SHGFI_ICONLOCATION);
639
640 HICON icon = ExtractIcon(gui.instance, info.szDisplayName,
641 info.iIcon);
642
643 SendDlgItemMessage(dialog, IDC_START_SAVE_ICON, STM_SETIMAGE,
644 IMAGE_ICON, LPARAM(icon));
645 }
646
647 {
648 char folder[38];
649
650 PathCompactPathEx(folder, data->saveDir.substr(0,
651 data->saveDir.length() - 1).c_str(), 38, 0);
652 SetDlgItemText(dialog, IDC_START_SAVE_TEXT, folder);
653 }
654 break;
655 }
656 break;
657 }
658
659 return FALSE;
660 }
661
662 INT_PTR ScanUtility::select(HWND dialog, UINT msg, WPARAM w, LPARAM l)
663 {
664 ScanUtility* data = which(dialog);
665
666 switch (msg)
667 {
668 case WM_INITDIALOG:
669 data = which(dialog, l);
670
671 {
672 ostringstream select;
673
674 select << "Select the scanned document that you need to save.";
675
676 SetDlgItemText(dialog, IDC_SELECT_TEXT, select.str().c_str());
677 }
678
679 {
680 LVCOLUMN column;
681
682 column.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
683 column.fmt = LVCFMT_IMAGE;
684 column.cx = 120;
685 column.pszText = "Scanned Documents";
686
687 ListView_InsertColumn(GetDlgItem(dialog, IDC_SELECT_SCANS), 0,
688 &column);
689 }
690 break;
691 case WM_NOTIFY:
692 if (w == IDC_SELECT_SCANS)
693 {
694 LPNMITEMACTIVATE ni = LPNMITEMACTIVATE(l);
695
696 switch (ni->hdr.code)
697 {
698 case NM_DBLCLK:
699 if (ni->iItem != -1)
700 {
701 char scan[MAX_PATH];
702
703 ListView_GetItemText(GetDlgItem(dialog, IDC_SELECT_SCANS),
704 ni->iItem, 0, scan, MAX_PATH);
705 ShellExecute(dialog, NULL, scan, NULL, NULL,
706 SW_SHOWDEFAULT);
707 }
708 break;
709 }
710 }
711 else
712 {
713 LPNMHDR nm = LPNMHDR(l);
714
715 switch (nm->code)
716 {
717 case PSN_SETACTIVE:
718 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
719 PSWIZB_DISABLEDFINISH | PSWIZB_NEXT);
720 data->populate(dialog);
721 break;
722 case PSN_WIZBACK:
723 ListView_DeleteAllItems(GetDlgItem(dialog, IDC_SELECT_SCANS));
724 data->client.setFile("");
725 break;
726 case PSN_WIZNEXT:
727 {
728 int index = ListView_GetNextItem(GetDlgItem(dialog,
729 IDC_SELECT_SCANS), -1, LVNI_SELECTED);
730 char scan[MAX_PATH];
731
732 if (index != -1)
733 {
734 ListView_GetItemText(GetDlgItem(dialog,
735 IDC_SELECT_SCANS), index, 0, scan, MAX_PATH);
736 }
737 else
738 {
739 ListView_GetItemText(GetDlgItem(dialog,
740 IDC_SELECT_SCANS), 0, 0, scan, MAX_PATH);
741 }
742
743 data->scan = scan;
744 }
745
746 if (debug) cerr << "scan = " << data->scan << "\n";
747
748 ListView_DeleteAllItems(GetDlgItem(dialog, IDC_SELECT_SCANS));
749 break;
750 }
751 }
752 break;
753 case WM_CONTEXTMENU:
754 if (ListView_GetNextItem(GetDlgItem(dialog, IDC_SELECT_SCANS), -1,
755 LVNI_SELECTED) != -1)
756 {
757 char scan[MAX_PATH];
758 POINT spot;
759 LVHITTESTINFO test;
760
761 test.pt.x = GET_X_LPARAM(l);
762 test.pt.y = GET_Y_LPARAM(l);
763
764 ScreenToClient(GetDlgItem(dialog, IDC_SELECT_SCANS), &test.pt);
765 ListView_HitTest(GetDlgItem(dialog, IDC_SELECT_SCANS), &test);
766
767 if (test.iItem != -1)
768 {
769 ListView_GetItemText(GetDlgItem(dialog, IDC_SELECT_SCANS),
770 test.iItem, 0, scan, MAX_PATH);
771
772 spot.x = test.pt.x;
773 spot.y = test.pt.y;
774 }
775 else if (GET_X_LPARAM(l) == -1 && GET_Y_LPARAM(l) == -1)
776 {
777 int index = ListView_GetNextItem(GetDlgItem(dialog,
778 IDC_SELECT_SCANS), -1, LVNI_SELECTED);
779 RECT rect;
780
781 ListView_GetItemText(GetDlgItem(dialog, IDC_SELECT_SCANS),
782 index, 0, scan, MAX_PATH);
783 ListView_EnsureVisible(GetDlgItem(dialog, IDC_SELECT_SCANS),
784 index, FALSE);
785 ListView_GetItemRect(GetDlgItem(dialog, IDC_SELECT_SCANS),
786 index, &rect, LVIR_SELECTBOUNDS);
787
788 spot.x = rect.left;
789 spot.y = rect.top;
790 }
791
792 ClientToScreen(GetDlgItem(dialog, IDC_SELECT_SCANS), &spot);
793
794 int code = TrackPopupMenuEx(data->popup, TPM_LEFTALIGN |
795 TPM_TOPALIGN | TPM_NONOTIFY | TPM_RETURNCMD | TPM_RIGHTBUTTON,
796 spot.x, spot.y, GetDlgItem(dialog, IDC_SELECT_SCANS), NULL);
797
798 switch (code)
799 {
800 case 1:
801 ShellExecute(dialog, NULL, scan, NULL, NULL, SW_SHOWDEFAULT);
802 break;
803 case 2:
804 {
805 SHELLEXECUTEINFO info;
806
807 info.cbSize = sizeof(info);
808 info.fMask = SEE_MASK_INVOKEIDLIST;
809 info.hwnd = dialog;
810 info.lpVerb = "properties";
811 info.lpFile = scan;
812 info.lpParameters = NULL;
813 info.lpDirectory = NULL;
814 info.nShow = SW_SHOWDEFAULT;
815 info.lpIDList = NULL;
816
817 ShellExecuteEx(&info);
818 }
819 break;
820 }
821 }
822 break;
823 }
824
825 return FALSE;
826 }
827
828 INT_PTR ScanUtility::enter(HWND dialog, UINT msg, WPARAM w, LPARAM l)
829 {
830 ScanUtility* data = which(dialog);
831
832 switch (msg)
833 {
834 case WM_INITDIALOG:
835 data = which(dialog, l);
836
837 {
838 ostringstream enter;
839
840 enter << "Enter the client name and number.";
841
842 SetDlgItemText(dialog, IDC_ENTER_TEXT, enter.str().c_str());
843 }
844 break;
845 case WM_NOTIFY:
846 {
847 LPNMHDR nm = LPNMHDR(l);
848
849 switch (nm->code)
850 {
851 case PSN_SETACTIVE:
852 if (data->client.getName() != "" && data->client.getNumber() !=
853 0)
854 {
855 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
856 PSWIZB_DISABLEDFINISH | PSWIZB_NEXT);
857 }
858 else
859 {
860 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
861 PSWIZB_DISABLEDFINISH);
862 }
863
864 SetDlgItemText(dialog, IDC_ENTER_NAME,
865 data->client.getName().c_str());
866
867 if (data->client.getNumber() != 0)
868 {
869 SetDlgItemInt(dialog, IDC_ENTER_NUM,
870 data->client.getNumber(), FALSE);
871 }
872 else
873 {
874 SetDlgItemText(dialog, IDC_ENTER_NUM, "");
875 }
876 break;
877 case PSN_WIZBACK:
878 break;
879 case PSN_WIZNEXT:
880 if (debug)
881 {
882 cerr << "client = {\n"
883 << " name = " << data->client.getName() << "\n"
884 << " number = " << data->client.getNumber() << "\n"
885 << " file = " << data->client.getFile() << "\n"
886 << "}\n";
887 }
888
889 data->save = data->saveDir + data->client.getFile();
890
891 if (debug) cerr << "save = " << data->save << "\n";
892
893 break;
894 }
895 }
896 break;
897 case WM_COMMAND:
898 switch (LOWORD(w))
899 {
900 case IDC_ENTER_NAME:
901 {
902 char name[BUFSIZ];
903
904 GetDlgItemText(dialog, IDC_ENTER_NAME, name, BUFSIZ);
905
906 if (name != data->client.getName())
907 {
908 data->client.setName(name);
909 }
910 }
911
912 if (data->client.getName() != "" && data->client.getNumber() != 0)
913 {
914 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
915 PSWIZB_DISABLEDFINISH | PSWIZB_NEXT);
916 }
917 else
918 {
919 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
920 PSWIZB_DISABLEDFINISH);
921 }
922 break;
923 case IDC_ENTER_NUM:
924 {
925 unsigned number = GetDlgItemInt(dialog, IDC_ENTER_NUM, NULL,
926 FALSE);
927
928 if (number != data->client.getNumber())
929 {
930 data->client.setNumber(number);
931 }
932 }
933
934 if (data->client.getName() != "" && data->client.getNumber() != 0)
935 {
936 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
937 PSWIZB_DISABLEDFINISH | PSWIZB_NEXT);
938 }
939 else
940 {
941 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
942 PSWIZB_DISABLEDFINISH);
943 }
944 break;
945 }
946 break;
947 }
948
949 return FALSE;
950 }
951
952 INT_PTR ScanUtility::confirm(HWND dialog, UINT msg, WPARAM w, LPARAM l)
953 {
954 ScanUtility* data = which(dialog);
955
956 switch (msg)
957 {
958 case WM_INITDIALOG:
959 data = which(dialog, l);
960
961 {
962 ostringstream confirm;
963
964 confirm << "Confirm the client file\'s name and size.";
965
966 SetDlgItemText(dialog, IDC_CONFIRM_TEXT, confirm.str().c_str());
967 }
968 break;
969 case WM_NOTIFY:
970 {
971 LPNMHDR nm = LPNMHDR(l);
972
973 switch (nm->code)
974 {
975 case PSN_SETACTIVE:
976 CheckDlgButton(dialog, IDC_CONFIRM_GOOD, BST_UNCHECKED);
977 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
978 PSWIZB_DISABLEDFINISH);
979 SetDlgItemText(dialog, IDC_CONFIRM_FILE,
980 data->client.getFile().c_str());
981
982 {
983 HANDLE scan = CreateFile(data->scan.c_str(), GENERIC_READ,
984 FILE_SHARE_READ, NULL, OPEN_EXISTING,
985 FILE_ATTRIBUTE_NORMAL, NULL);
986 DWORD bytes = GetFileSize(scan, NULL);
987
988 CloseHandle(scan);
989
990 ostringstream size;
991
992 size << format(bytes);
993
994 size.setf(ios_base::fixed, ios_base::floatfield);
995 size.precision(2);
996
997 FLOAT megabytes = FLOAT(bytes) / FLOAT(1024 * 1024);
998
999 size << " bytes (" << megabytes << " MB)";
1000
1001 SetDlgItemText(dialog, IDC_CONFIRM_SIZE,
1002 size.str().c_str());
1003 }
1004 break;
1005 case PSN_WIZBACK:
1006 break;
1007 case PSN_WIZNEXT:
1008 if (MoveFile(data->scan.c_str(), data->save.c_str()) == 0)
1009 {
1010 do
1011 {
1012 LPVOID message;
1013
1014 FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
1015 FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(),
1016 0, LPTSTR(&message), 0, NULL);
1017
1018 int code = MessageBox(dialog, LPCTSTR(message),
1019 data->title.c_str(), MB_RETRYCANCEL |
1020 MB_ICONEXCLAMATION);
1021
1022 LocalFree(message);
1023
1024 if (code == IDCANCEL)
1025 {
1026 PropSheet_PressButton(GetParent(dialog),
1027 PSBTN_BACK);
1028 break;
1029 }
1030 }
1031 while (MoveFileEx(data->scan.c_str(), data->save.c_str(),
1032 MOVEFILE_REPLACE_EXISTING) == 0);
1033 }
1034 break;
1035 }
1036 }
1037 break;
1038 case WM_COMMAND:
1039 switch (LOWORD(w))
1040 {
1041 case IDC_CONFIRM_GOOD:
1042 if (IsDlgButtonChecked(dialog, IDC_CONFIRM_GOOD) == BST_CHECKED)
1043 {
1044 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
1045 PSWIZB_DISABLEDFINISH | PSWIZB_NEXT);
1046 }
1047 else
1048 {
1049 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
1050 PSWIZB_DISABLEDFINISH);
1051 }
1052 break;
1053 }
1054 break;
1055 }
1056
1057 return FALSE;
1058 }
1059
1060 INT_PTR ScanUtility::done(HWND dialog, UINT msg, WPARAM w, LPARAM l)
1061 {
1062 ScanUtility* data = which(dialog);
1063
1064 switch (msg)
1065 {
1066 case WM_INITDIALOG:
1067 data = which(dialog, l);
1068
1069 {
1070 ostringstream done;
1071
1072 done << "You are done saving the scanned document. Click Finish to"
1073 << " exit or click Back to return to the beginning.";
1074
1075 SetDlgItemText(dialog, IDC_DONE_TEXT, done.str().c_str());
1076 }
1077 break;
1078 case WM_NOTIFY:
1079 {
1080 LPNMHDR nm = LPNMHDR(l);
1081
1082 switch (nm->code)
1083 {
1084 case PSN_SETACTIVE:
1085 PropSheet_SetWizButtons(GetParent(dialog), PSWIZB_BACK |
1086 PSWIZB_FINISH);
1087 break;
1088 case PSN_WIZBACK:
1089 data->client.setFile("");
1090 PropSheet_SetCurSelByID(GetParent(dialog), IDD_START);
1091 break;
1092 case PSN_WIZFINISH:
1093 break;
1094 }
1095 }
1096 break;
1097 }
1098
1099 return FALSE;
1100 }