trunk/VTBFileUtil2/Chooser.cxx

// Vance Thrift and Biller File Utility 2
//
// Douglas Thrift
//
// $Id: Chooser.cxx,v 1.11 2003/09/05 08:57:00 douglas Exp $

#include "Chooser.h"

Chooser::Chooser()
{
        shortcut = false;
        choice = none;
        number = count++;

        choosers.insert(pair<unsigned, Chooser*>(number, this));
}

Chooser::~Chooser()
{
        choosers.erase(number);
}

Mode Chooser::choose(HWND parent)
{
        DialogBoxParam(gui.instance, MAKEINTRESOURCE(IDD_CHOICE), parent, window,
                number);

        desktop();

        return choice;
}

unsigned Chooser::count = 0;
map<unsigned, Chooser*> Chooser::choosers;
map<HWND, Chooser*> Chooser::windows;

INT_PTR Chooser::window(HWND dialog, UINT msg, WPARAM w, LPARAM l)
{
        map<HWND, Chooser*>::iterator itor = windows.find(dialog);
        Chooser* data = itor->second;

        switch (msg)
        {
        case WM_INITDIALOG:
                center(dialog);
                SendMessage(dialog, WM_SETICON, ICON_BIG, LPARAM(gui.icon));
                SendMessage(dialog, WM_SETTEXT, 0, LPARAM(programName.c_str()));

                {
                        map<unsigned, Chooser*>::iterator itor = choosers.find(l);
                        
                        windows.insert(pair<HWND, Chooser*>(dialog, itor->second));

                        data = itor->second;
                }

                {
                        HMENU menu = GetSystemMenu(dialog, FALSE);
                        MENUITEMINFO separator, usage, version, license;
                        int count = GetMenuItemCount(menu);

                        separator.cbSize = sizeof(separator);
                        separator.fMask = MIIM_TYPE;
                        separator.fType = MFT_SEPARATOR;
                        usage.cbSize = sizeof(usage);
                        usage.fMask = MIIM_ID | MIIM_TYPE;
                        usage.fType = MFT_STRING;
                        usage.wID = 1;
                        usage.dwTypeData = "&Usage...";
                        usage.cch = 9;
                        version.cbSize = sizeof(version);
                        version.fMask = MIIM_ID | MIIM_TYPE;
                        version.fType = MFT_STRING;
                        version.wID = 2;
                        version.dwTypeData = "&Version...";
                        version.cch = 11;
                        license.cbSize = sizeof(license);
                        license.fMask = MIIM_ID | MIIM_TYPE;
                        license.fType = MFT_STRING;
                        license.wID = 3;
                        license.dwTypeData = "&License...";
                        license.cch = 11;

                        InsertMenuItem(menu, count++, TRUE, &separator);
                        InsertMenuItem(menu, count++, TRUE, &usage);
                        InsertMenuItem(menu, count++, TRUE, &version);
                        InsertMenuItem(menu, count++, TRUE, &license);
                }
                break;
        case WM_SYSCOMMAND:
                switch (w)
                {
                case 1:
                        usage(dialog);
                        break;
                case 2:
                        version(dialog);
                        break;
                case 3:
                        {
                                License license;

                                license.display(dialog);
                        }
                        break;
                }
                break;
        case WM_COMMAND:
                switch (LOWORD(w))
                {
                case IDC_CHOICE_DISC:
                        data->choice = disc;

                        if (debug) cerr << "choice = disc\n";

                        EnableWindow(GetDlgItem(dialog, IDC_CHOICE_ALWAYS), TRUE);
                        EnableWindow(GetDlgItem(dialog, IDOK), TRUE);
                        break;
                case IDC_CHOICE_SCAN:
                        data->choice = scan;

                        if (debug) cerr << "choice = scan\n";

                        EnableWindow(GetDlgItem(dialog, IDC_CHOICE_ALWAYS), TRUE);
                        EnableWindow(GetDlgItem(dialog, IDOK), TRUE);
                        break;
                case IDC_CHOICE_ALWAYS:
                        data->shortcut = !data->shortcut;

                        if (debug) cerr << "shortcut = " << data->shortcut << "\n";
                        break;
                case IDCANCEL:
                        data->choice = none;
                case IDOK:
                        EndDialog(dialog, w);
                        return TRUE;
                        break;
                }
                break;
        }

        return FALSE;
}

void Chooser::desktop(void)
{
        if (!shortcut || choice == none) return;

        IShellLink* link;

        CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
                IID_IShellLink, (LPVOID*)(&link));

        string name;

        link->SetPath(program.c_str());
        link->SetIconLocation(program.c_str(), 0);
        link->SetDescription(programName.c_str());
        
        switch (choice)
        {
        case disc:
                name = "Disc Browse";
                link->SetArguments("-disc");
                break;
        case scan:
                name = "Scan Utility";
                link->SetArguments("-scan");
                break;
        }

        IPersistFile* file;

        link->QueryInterface(IID_IPersistFile, (LPVOID*)(&file));

        char* desktop = new char[MAX_PATH];

        SHGetFolderPath(NULL, CSIDL_FLAG_CREATE | CSIDL_DESKTOPDIRECTORY, NULL,
                SHGFP_TYPE_CURRENT, desktop);

        wstring path = toWide(string(desktop) + "\\" + name + ".lnk");

        if (debug) cerr << "path = " << toAnsi(path) << "\n";

        file->Save(path.c_str(), TRUE);
        file->Release();
        link->Release();

        delete [] desktop;
}
Revision:	291
Committed:	2003-09-05T01:57:00-07:00 (21 years, 10 months ago) by douglas
File size:	4028 byte(s)
Log Message:	Editing is rich.
#	User	Rev	Content
1	douglas	263	// Vance Thrift and Biller File Utility 2
2			//
3			// Douglas Thrift
4			//
5	douglas	291	// $Id: Chooser.cxx,v 1.11 2003/09/05 08:57:00 douglas Exp $
6	douglas	263
7			#include "Chooser.h"
8
9	douglas	266	Chooser::Chooser()
10			{
11			shortcut = false;
12			choice = none;
13			number = count++;
14	douglas	291
15	douglas	266	choosers.insert(pair<unsigned, Chooser*>(number, this));
16			}
17	douglas	265
18	douglas	266	Chooser::~Chooser()
19	douglas	263	{
20	douglas	266	choosers.erase(number);
21	douglas	263	}
22
23			Mode Chooser::choose(HWND parent)
24			{
25	douglas	291	DialogBoxParam(gui.instance, MAKEINTRESOURCE(IDD_CHOICE), parent, window,
26			number);
27	douglas	263
28			desktop();
29
30			return choice;
31			}
32
33	douglas	268	unsigned Chooser::count = 0;
34			map<unsigned, Chooser*> Chooser::choosers;
35			map<HWND, Chooser*> Chooser::windows;
36
37	douglas	273	INT_PTR Chooser::window(HWND dialog, UINT msg, WPARAM w, LPARAM l)
38	douglas	266	{
39			map<HWND, Chooser*>::iterator itor = windows.find(dialog);
40			Chooser* data = itor->second;
41
42			switch (msg)
43			{
44			case WM_INITDIALOG:
45			center(dialog);
46			SendMessage(dialog, WM_SETICON, ICON_BIG, LPARAM(gui.icon));
47			SendMessage(dialog, WM_SETTEXT, 0, LPARAM(programName.c_str()));
48	douglas	291
49	douglas	266	{
50			map<unsigned, Chooser*>::iterator itor = choosers.find(l);
51
52			windows.insert(pair<HWND, Chooser*>(dialog, itor->second));
53	douglas	291
54			data = itor->second;
55	douglas	266	}
56	douglas	291
57			{
58			HMENU menu = GetSystemMenu(dialog, FALSE);
59			MENUITEMINFO separator, usage, version, license;
60			int count = GetMenuItemCount(menu);
61
62			separator.cbSize = sizeof(separator);
63			separator.fMask = MIIM_TYPE;
64			separator.fType = MFT_SEPARATOR;
65			usage.cbSize = sizeof(usage);
66			usage.fMask = MIIM_ID \| MIIM_TYPE;
67			usage.fType = MFT_STRING;
68			usage.wID = 1;
69			usage.dwTypeData = "&Usage...";
70			usage.cch = 9;
71			version.cbSize = sizeof(version);
72			version.fMask = MIIM_ID \| MIIM_TYPE;
73			version.fType = MFT_STRING;
74			version.wID = 2;
75			version.dwTypeData = "&Version...";
76			version.cch = 11;
77			license.cbSize = sizeof(license);
78			license.fMask = MIIM_ID \| MIIM_TYPE;
79			license.fType = MFT_STRING;
80			license.wID = 3;
81			license.dwTypeData = "&License...";
82			license.cch = 11;
83
84			InsertMenuItem(menu, count++, TRUE, &separator);
85			InsertMenuItem(menu, count++, TRUE, &usage);
86			InsertMenuItem(menu, count++, TRUE, &version);
87			InsertMenuItem(menu, count++, TRUE, &license);
88			}
89	douglas	266	break;
90	douglas	291	case WM_SYSCOMMAND:
91			switch (w)
92			{
93			case 1:
94			usage(dialog);
95			break;
96			case 2:
97			version(dialog);
98			break;
99			case 3:
100			{
101			License license;
102
103			license.display(dialog);
104			}
105			break;
106			}
107			break;
108	douglas	266	case WM_COMMAND:
109			switch (LOWORD(w))
110			{
111			case IDC_CHOICE_DISC:
112	douglas	268	data->choice = disc;
113	douglas	291
114	douglas	266	if (debug) cerr << "choice = disc\n";
115	douglas	291
116	douglas	266	EnableWindow(GetDlgItem(dialog, IDC_CHOICE_ALWAYS), TRUE);
117			EnableWindow(GetDlgItem(dialog, IDOK), TRUE);
118			break;
119			case IDC_CHOICE_SCAN:
120	douglas	268	data->choice = scan;
121	douglas	291
122	douglas	266	if (debug) cerr << "choice = scan\n";
123	douglas	291
124	douglas	266	EnableWindow(GetDlgItem(dialog, IDC_CHOICE_ALWAYS), TRUE);
125			EnableWindow(GetDlgItem(dialog, IDOK), TRUE);
126			break;
127			case IDC_CHOICE_ALWAYS:
128	douglas	268	data->shortcut = !data->shortcut;
129	douglas	291
130	douglas	268	if (debug) cerr << "shortcut = " << data->shortcut << "\n";
131	douglas	266	break;
132			case IDCANCEL:
133	douglas	268	data->choice = none;
134	douglas	266	case IDOK:
135			EndDialog(dialog, w);
136			return TRUE;
137			break;
138			}
139			break;
140			}
141
142			return FALSE;
143			}
144
145	douglas	263	void Chooser::desktop(void)
146			{
147	douglas	264	if (!shortcut \|\| choice == none) return;
148	douglas	263
149	douglas	264	IShellLink* link;
150
151			CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
152			IID_IShellLink, (LPVOID*)(&link));
153
154			string name;
155
156			link->SetPath(program.c_str());
157			link->SetIconLocation(program.c_str(), 0);
158			link->SetDescription(programName.c_str());
159
160			switch (choice)
161			{
162			case disc:
163			name = "Disc Browse";
164			link->SetArguments("-disc");
165			break;
166			case scan:
167			name = "Scan Utility";
168			link->SetArguments("-scan");
169			break;
170			}
171
172			IPersistFile* file;
173
174			link->QueryInterface(IID_IPersistFile, (LPVOID*)(&file));
175
176			char* desktop = new char[MAX_PATH];
177
178	douglas	290	SHGetFolderPath(NULL, CSIDL_FLAG_CREATE \| CSIDL_DESKTOPDIRECTORY, NULL,
179			SHGFP_TYPE_CURRENT, desktop);
180	douglas	264
181			wstring path = toWide(string(desktop) + "\\" + name + ".lnk");
182
183			if (debug) cerr << "path = " << toAnsi(path) << "\n";
184
185			file->Save(path.c_str(), TRUE);
186			file->Release();
187			link->Release();
188
189	douglas	267	delete [] desktop;
190	douglas	263	}