Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 21:52:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

入力パスと出力パスをダイアログで設定するボタンをつけた

Browse Source
This commit is contained in:
lltcggie 2015-12-17 16:28:53 +09:00
parent 927726889a
commit 7f84627e04
3 changed files with 284 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
waifu2x-caffe-gui/Resource.rc
View File

Binary file not shown.

328
waifu2x-caffe-gui/Source.cpp
View File

@ -17,6 +17,7 @@
#include <boost/math/common_factor_rt.hpp>
#include <opencv2/opencv.hpp>
#include <cblas.h>
#include <dlgs.h>
#include "resource.h"
#include "../common/waifu2x.h"
@ -716,6 +717,136 @@ private:
WritePrivateProfileString(TEXT("Setting"), TEXT("LastUseTTA"), to_tstring(use_tta ? 1 : 0).c_str(), getTString(SettingFilePath).c_str());
}
// 出力パスを選択する
static UINT_PTR CALLBACK OFNHookProcIn(
_In_ HWND hdlg,
_In_ UINT uiMsg,
_In_ WPARAM wParam,
_In_ LPARAM lParam
)
{
switch (uiMsg)
{
case WM_INITDIALOG:
{
// ダイアログを中央に表示
HWND hParent = GetParent(hdlg);
HWND hwndScreen;
RECT rectScreen;
hwndScreen = GetDesktopWindow();
GetWindowRect(hwndScreen, &rectScreen);
RECT rDialog;
GetWindowRect(hParent, &rDialog);
const int Width = rDialog.left = rDialog.right;
const int Height = rDialog.bottom - rDialog.top;
int DialogPosX;
int DialogPosY;
DialogPosX = ((rectScreen.right - rectScreen.left) / 2 - Width / 2);
DialogPosY = ((rectScreen.bottom - rectScreen.top) / 2 - Height / 2);
SetWindowPos(hParent, NULL, DialogPosX, DialogPosY, 0, 0, SWP_NOSIZE | SWP_NOZORDER);
}
break;
case WM_NOTIFY:
{
// ファイルとフォルダを選択できるようにする
NMHDR *pnmh;
OFNOTIFY *pnot;
pnot = (OFNOTIFY *)lParam;
pnmh = &pnot->hdr;
switch (pnmh->code)
{
case CDN_SELCHANGE:
{
TCHAR szPath[AR_PATH_MAX] = TEXT("");
HWND hParent = GetParent(hdlg);
if (CommDlg_OpenSave_GetFilePath(hParent, szPath, _countof(szPath)) > 0)
{
szPath[_countof(szPath) - 1] = TEXT('\0');
boost::filesystem::path p(szPath);
const auto filename = getTString(p.filename());
CommDlg_OpenSave_SetControlText(hParent, edt1, filename.c_str());
}
}
break;
}
}
break;
}
return 0L;
}
// 出力フォルダを選択する
static UINT_PTR CALLBACK OFNHookProcOut(
_In_ HWND hdlg,
_In_ UINT uiMsg,
_In_ WPARAM wParam,
_In_ LPARAM lParam
)
{
switch (uiMsg)
{
case WM_INITDIALOG:
{
// ダイアログを中央に表示
HWND hParent = GetParent(hdlg);
HWND hwndScreen;
RECT rectScreen;
hwndScreen = GetDesktopWindow();
GetWindowRect(hwndScreen, &rectScreen);
RECT rDialog;
GetWindowRect(hParent, &rDialog);
const int Width = rDialog.left = rDialog.right;
const int Height = rDialog.bottom - rDialog.top;
int DialogPosX;
int DialogPosY;
DialogPosX = ((rectScreen.right - rectScreen.left) / 2 - Width / 2);
DialogPosY = ((rectScreen.bottom - rectScreen.top) / 2 - Height / 2);
SetWindowPos(hParent, NULL, DialogPosX, DialogPosY, 0, 0, SWP_NOSIZE | SWP_NOZORDER);
}
break;
case WM_NOTIFY:
{
// 常に出力ファイル名を固定する
NMHDR *pnmh;
OFNOTIFY *pnot;
pnot = (OFNOTIFY *)lParam;
pnmh = &pnot->hdr;
tstring &nowName = *(tstring *)pnot->lpOFN->lCustData;
switch (pnmh->code)
{
case CDN_SELCHANGE:
{
HWND hParent = GetParent(hdlg);
CommDlg_OpenSave_SetControlText(hParent, edt1, nowName.c_str());
}
break;
}
}
break;
}
return 0L;
}
public:
DialogEvent() : dh(nullptr), mode("noise_scale"), noise_level(1), scale_ratio(2.0), model_dir(TEXT("models/anime_style_art_rgb")),
process("gpu"), outputExt(TEXT("png")), inputFileExt(TEXT("png:jpg:jpeg:tif:tiff:bmp:tga")),
@ -1113,6 +1244,58 @@ public:
}
}
LRESULT OnSetInputFilePath(const TCHAR *tPath)
{
HWND hWnd = GetDlgItem(dh, IDC_EDIT_INPUT);
boost::filesystem::path path(tPath);
if (!boost::filesystem::exists(path))
{
MessageBox(dh, TEXT("入力ファイル/フォルダが存在しません"), TEXT("エラー"), MB_OK | MB_ICONERROR);
return 0L;
}
if (!SyncMember(true))
return 0L;
if (boost::filesystem::is_directory(path))
{
HWND ho = GetDlgItem(dh, IDC_EDIT_OUTPUT);
const tstring addstr(AddName());
autoSetAddName = AddName();
const auto str = getTString(path.branch_path() / (path.stem().wstring() + addstr));
SetWindowText(ho, str.c_str());
SetWindowText(hWnd, tPath);
}
else
{
HWND ho = GetDlgItem(dh, IDC_EDIT_OUTPUT);
tstring outputFileName = tPath;
const auto tailDot = outputFileName.find_last_of('.');
if (tailDot != outputFileName.npos)
outputFileName.erase(tailDot, outputFileName.length());
const tstring addstr(AddName());
autoSetAddName = addstr;
outputFileName += addstr + outputExt;
SetWindowText(ho, outputFileName.c_str());
SetWindowText(hWnd, tPath);
}
SetCropSizeList(path);
return 0L;
}
// ここで渡されるhWndはIDC_EDITのHWND(コントロールのイベントだから)
LRESULT DropInput(HWND hWnd, WPARAM wParam, LPARAM lParam, WNDPROC OrgSubWnd, LPVOID lpData)
{
@ -1123,51 +1306,9 @@ public:
if (FileNum >= 1)
{
DragQueryFile((HDROP)wParam, 0, szTmp, _countof(szTmp));
szTmp[_countof(szTmp) - 1] = TEXT('\0');
boost::filesystem::path path(szTmp);
if (!boost::filesystem::exists(path))
{
MessageBox(dh, TEXT("入力ファイル/フォルダが存在しません"), TEXT("エラー"), MB_OK | MB_ICONERROR);
return 0L;
}
if (!SyncMember(true))
return 0L;
if (boost::filesystem::is_directory(path))
{
HWND ho = GetDlgItem(dh, IDC_EDIT_OUTPUT);
const tstring addstr(AddName());
autoSetAddName = AddName();
const auto str = getTString(path.branch_path() / (path.stem().wstring() + addstr));
SetWindowText(ho, str.c_str());
SetWindowText(hWnd, szTmp);
}
else
{
HWND ho = GetDlgItem(dh, IDC_EDIT_OUTPUT);
tstring outputFileName = szTmp;
const auto tailDot = outputFileName.find_last_of('.');
if (tailDot != outputFileName.npos)
outputFileName.erase(tailDot, outputFileName.length());
const tstring addstr(AddName());
autoSetAddName = addstr;
outputFileName += addstr + outputExt;
SetWindowText(ho, outputFileName.c_str());
SetWindowText(hWnd, szTmp);
}
SetCropSizeList(path);
OnSetInputFilePath(szTmp);
}
return 0L;
@ -1195,6 +1336,103 @@ public:
ReplaceAddString();
return ret;
}
void InputRef(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
{
OPENFILENAME ofn;
TCHAR szPath[AR_PATH_MAX] = TEXT("");
TCHAR szFile[AR_PATH_MAX] = TEXT("");
GetCurrentDirectory(_countof(szPath), szPath);
szPath[_countof(szPath) - 1] = TEXT('\0');
ofn.lStructSize = sizeof(ofn);
ofn.hwndOwner = NULL;
ofn.lpstrFile = szFile;
ofn.nMaxFile = _countof(szFile);
ofn.lpstrFilter = TEXT("すべてのファイル、フォルダ(*.*)\0*.*\0");
ofn.nFilterIndex = 1;
ofn.lpstrTitle = TEXT("入力するファイルかフォルダを選択してください");
ofn.lpstrInitialDir = szPath;
ofn.lpstrCustomFilter = NULL;
ofn.nMaxCustFilter = 0;
ofn.lpstrFileTitle = NULL;
ofn.nMaxFileTitle = 0;
ofn.nFileOffset = 0;
ofn.nFileExtension = 0;
ofn.lpstrDefExt = NULL;
ofn.lCustData = 0;
ofn.lpfnHook = OFNHookProcIn;
ofn.lpTemplateName = 0;
ofn.Flags = OFN_HIDEREADONLY | OFN_NOVALIDATE | OFN_PATHMUSTEXIST | OFN_READONLY | OFN_EXPLORER | OFN_ENABLEHOOK;
if (GetOpenFileName(&ofn))
{
szFile[_countof(szFile) - 1] = TEXT('\0');
OnSetInputFilePath(szFile);
}
}
void OutputRef(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
{
OPENFILENAME ofn;
TCHAR szPath[AR_PATH_MAX] = TEXT("");
TCHAR szFile[AR_PATH_MAX] = TEXT("");
tstring nowName;
bool isDir = false;
GetWindowText(GetDlgItem(dh, IDC_EDIT_INPUT), szFile, _countof(szFile));
szFile[_countof(szPath) - 1] = TEXT('\0');
if (*szFile != TEXT('\0'))
{
boost::filesystem::path p(szFile);
isDir = boost::filesystem::is_directory(p);
}
GetWindowText(GetDlgItem(dh, IDC_EDIT_OUTPUT), szFile, _countof(szFile));
szFile[_countof(szPath) - 1] = TEXT('\0');
if (*szFile == TEXT('\0'))
{
nowName = TEXT("");
GetCurrentDirectory(_countof(szPath), szPath);
szPath[_countof(szPath) - 1] = TEXT('\0');
}
else
{
boost::filesystem::path p(szFile);
const auto filename = getTString(p.filename());
const auto dir = getTString(p.branch_path());
nowName = filename;
_tcscpy(szFile, filename.c_str());
_tcscpy(szPath, dir.c_str());
}
ofn.lStructSize = sizeof(ofn);
ofn.hwndOwner = NULL;
ofn.lpstrFile = szFile;
ofn.nMaxFile = _countof(szFile);
ofn.lpstrFilter = TEXT("すべてのフォルダ\0*.folder\0");
ofn.nFilterIndex = 1;
ofn.lpstrTitle = TEXT("出力するフォルダを選択してください");
ofn.lpstrInitialDir = szPath;
ofn.lpstrCustomFilter = NULL;
ofn.nMaxCustFilter = 0;
ofn.lpstrFileTitle = NULL;
ofn.nMaxFileTitle = 0;
ofn.nFileOffset = 0;
ofn.nFileExtension = 0;
ofn.lpstrDefExt = NULL;
ofn.lCustData = (LPARAM)&nowName;
ofn.lpfnHook = OFNHookProcOut;
ofn.lpTemplateName = 0;
ofn.Flags = OFN_HIDEREADONLY | OFN_NOVALIDATE | OFN_PATHMUSTEXIST | OFN_READONLY | OFN_EXPLORER | OFN_ENABLEHOOK | OFN_OVERWRITEPROMPT;
if (GetSaveFileName(&ofn))
{
szFile[_countof(szFile) - 1] = TEXT('\0');
SetWindowText(GetDlgItem(dh, IDC_EDIT_OUTPUT), szFile);
}
}
};
@ -1243,6 +1481,8 @@ int WINAPI WinMain(HINSTANCE hInstance,
// 各コントロールのイベントで実行する関数の登録
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::Exec, &cDialogEvent), NULL, IDC_BUTTON_EXEC);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::Cancel, &cDialogEvent), NULL, IDC_BUTTON_CANCEL);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::InputRef, &cDialogEvent), NULL, IDC_BUTTON_INPUT_REF);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::OutputRef, &cDialogEvent), NULL, IDC_BUTTON_OUTPUT_REF);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_NOISE);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_SCALE);

BIN
waifu2x-caffe-gui/resource.h
View File

Binary file not shown.