mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-06-26 13:42:48 +00:00
1438 lines
42 KiB
C++
1438 lines
42 KiB
C++
#define _CRT_SECURE_NO_WARNINGS
|
||
#include <glog/logging.h>
|
||
#include <windows.h>
|
||
#include <Commctrl.h>
|
||
#include <tchar.h>
|
||
#include <stdio.h>
|
||
#include <string>
|
||
#include <thread>
|
||
#include <atomic>
|
||
#include <chrono>
|
||
#include <iostream>
|
||
#include <fstream>
|
||
#include <algorithm>
|
||
#include <boost/filesystem.hpp>
|
||
#include <boost/tokenizer.hpp>
|
||
#include <boost/foreach.hpp>
|
||
#include <boost/math/common_factor_rt.hpp>
|
||
#include <opencv2/opencv.hpp>
|
||
#include <cblas.h>
|
||
#include <dlgs.h>
|
||
#include "resource.h"
|
||
#include "../common/waifu2x.h"
|
||
|
||
#include "CDialog.h"
|
||
#include "CControl.h"
|
||
|
||
#undef ERROR
|
||
|
||
#define WM_FAILD_CREATE_DIR (WM_APP + 5)
|
||
#define WM_ON_WAIFU2X_ERROR (WM_APP + 6)
|
||
#define WM_END_THREAD (WM_APP + 7)
|
||
|
||
const size_t AR_PATH_MAX(1024);
|
||
|
||
const int MinCommonDivisor = 50;
|
||
const int DefaultCommonDivisor = 128;
|
||
const std::pair<int, int> DefaultCommonDivisorRange = {90, 140};
|
||
|
||
const TCHAR * const CropSizeListName = TEXT("crop_size_list.txt");
|
||
const TCHAR * const SettingFileName = TEXT("setting.ini");
|
||
|
||
|
||
#ifdef UNICODE
|
||
typedef std::wstring tstring;
|
||
inline tstring getTString(const boost::filesystem::path& p)
|
||
{
|
||
return p.wstring();
|
||
}
|
||
#else
|
||
typedef std::string tstring;
|
||
inline tstring getTString(const boost::filesystem::path& p)
|
||
{
|
||
return p.string();
|
||
}
|
||
#endif
|
||
|
||
// http://stackoverflow.com/questions/10167382/boostfilesystem-get-relative-path
|
||
boost::filesystem::path relativePath(const boost::filesystem::path &path, const boost::filesystem::path &relative_to)
|
||
{
|
||
// create absolute paths
|
||
boost::filesystem::path p = boost::filesystem::absolute(path);
|
||
boost::filesystem::path r = boost::filesystem::absolute(relative_to);
|
||
|
||
// if root paths are different, return absolute path
|
||
if (p.root_path() != r.root_path())
|
||
return p;
|
||
|
||
// initialize relative path
|
||
boost::filesystem::path result;
|
||
|
||
// find out where the two paths diverge
|
||
boost::filesystem::path::const_iterator itr_path = p.begin();
|
||
boost::filesystem::path::const_iterator itr_relative_to = r.begin();
|
||
while (*itr_path == *itr_relative_to && itr_path != p.end() && itr_relative_to != r.end()) {
|
||
++itr_path;
|
||
++itr_relative_to;
|
||
}
|
||
|
||
// add "../" for each remaining token in relative_to
|
||
if (itr_relative_to != r.end()) {
|
||
++itr_relative_to;
|
||
while (itr_relative_to != r.end()) {
|
||
result /= "..";
|
||
++itr_relative_to;
|
||
}
|
||
}
|
||
|
||
// add remaining path
|
||
while (itr_path != p.end()) {
|
||
result /= *itr_path;
|
||
++itr_path;
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
std::vector<int> CommonDivisorList(const int N)
|
||
{
|
||
std::vector<int> list;
|
||
|
||
const int sq = sqrt(N);
|
||
for (int i = 1; i <= sq; i++)
|
||
{
|
||
if (N % i == 0)
|
||
list.push_back(i);
|
||
}
|
||
|
||
const int sqs = list.size();
|
||
for (int i = 0; i < sqs; i++)
|
||
list.push_back(N / list[i]);
|
||
|
||
std::sort(list.begin(), list.end());
|
||
|
||
return list;
|
||
}
|
||
|
||
// <20>_<EFBFBD>C<EFBFBD>A<EFBFBD><41><EFBFBD>O<EFBFBD>p
|
||
class DialogEvent
|
||
{
|
||
private:
|
||
HWND dh;
|
||
|
||
boost::filesystem::path exeDir;
|
||
std::vector<int> CropSizeList;
|
||
|
||
tstring input_str;
|
||
tstring output_str;
|
||
std::string mode;
|
||
int noise_level;
|
||
double scale_ratio;
|
||
tstring model_dir;
|
||
std::string process;
|
||
tstring outputExt;
|
||
tstring inputFileExt;
|
||
|
||
bool use_tta;
|
||
|
||
int crop_size;
|
||
int batch_size;
|
||
|
||
std::vector<tstring> extList;
|
||
|
||
std::thread processThread;
|
||
std::atomic_bool cancelFlag;
|
||
|
||
tstring autoSetAddName;
|
||
bool isLastError;
|
||
|
||
tstring logMessage;
|
||
|
||
std::string usedProcess;
|
||
std::chrono::system_clock::duration cuDNNCheckTime;
|
||
std::chrono::system_clock::duration InitTime;
|
||
std::chrono::system_clock::duration ProcessTime;
|
||
|
||
enum eModelType
|
||
{
|
||
eModelTypeRGB,
|
||
eModelTypePhoto,
|
||
eModelTypeY,
|
||
eModelTypeEnd,
|
||
};
|
||
|
||
eModelType modelType;
|
||
|
||
private:
|
||
template<typename T>
|
||
static tstring to_tstring(T val)
|
||
{
|
||
#ifdef UNICODE
|
||
return std::to_wstring(val);
|
||
#else
|
||
return std::to_string(val);
|
||
#endif
|
||
}
|
||
|
||
tstring AddName() const
|
||
{
|
||
tstring addstr;
|
||
|
||
addstr += TEXT("(");
|
||
switch (modelType)
|
||
{
|
||
case eModelTypeRGB:
|
||
addstr += TEXT("RGB");
|
||
break;
|
||
|
||
case eModelTypePhoto:
|
||
addstr += TEXT("Photo");
|
||
break;
|
||
|
||
case eModelTypeY:
|
||
addstr += TEXT("Y");
|
||
break;
|
||
}
|
||
addstr += TEXT(")");
|
||
|
||
addstr += TEXT("(");
|
||
if (mode == "noise")
|
||
addstr += TEXT("noise");
|
||
else if (mode == "scale")
|
||
addstr += TEXT("scale");
|
||
else if (mode == "noise_scale")
|
||
addstr += TEXT("noise_scale");
|
||
else if (mode == "auto_scale")
|
||
addstr += TEXT("auto_scale");
|
||
addstr += TEXT(")");
|
||
|
||
if (mode.find("noise") != mode.npos || mode.find("auto_scale") != mode.npos)
|
||
addstr += TEXT("(Level") + to_tstring(noise_level) + TEXT(")");
|
||
if (use_tta)
|
||
addstr += TEXT("(tta)");
|
||
if (mode.find("scale") != mode.npos)
|
||
addstr += TEXT("(x") + to_tstring(scale_ratio) + TEXT(")");
|
||
|
||
return addstr;
|
||
}
|
||
|
||
bool SyncMember(const bool NotSyncCropSize)
|
||
{
|
||
bool ret = true;
|
||
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_EDIT_INPUT), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
input_str = buf;
|
||
}
|
||
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_EDIT_OUTPUT), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
output_str = buf;
|
||
}
|
||
|
||
if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODE_NOISE), BM_GETCHECK, 0, 0))
|
||
mode = "noise";
|
||
else if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODE_SCALE), BM_GETCHECK, 0, 0))
|
||
mode = "scale";
|
||
else if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODE_NOISE_SCALE), BM_GETCHECK, 0, 0))
|
||
mode = "noise_scale";
|
||
else
|
||
mode = "auto_scale";
|
||
|
||
if (SendMessage(GetDlgItem(dh, IDC_RADIONOISE_LEVEL1), BM_GETCHECK, 0, 0))
|
||
noise_level = 1;
|
||
else
|
||
noise_level = 2;
|
||
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_EDIT_SCALE_RATIO), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
TCHAR *ptr = nullptr;
|
||
scale_ratio = _tcstod(buf, &ptr);
|
||
if (!ptr || *ptr != TEXT('\0') || scale_ratio <= 0.0)
|
||
{
|
||
scale_ratio = 2.0;
|
||
ret = false;
|
||
|
||
MessageBox(dh, TEXT("<EFBFBD>g<EFBFBD>嗦<EFBFBD><EFBFBD>0.0<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>傫<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ł<EFBFBD><EFBFBD><EFBFBD><EFBFBD>K<EFBFBD>v<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
}
|
||
}
|
||
|
||
if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODEL_RGB), BM_GETCHECK, 0, 0))
|
||
{
|
||
model_dir = TEXT("models/anime_style_art_rgb");
|
||
modelType = eModelTypeRGB;
|
||
}
|
||
else if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODEL_Y), BM_GETCHECK, 0, 0))
|
||
{
|
||
model_dir = TEXT("models/anime_style_art");
|
||
modelType = eModelTypeY;
|
||
}
|
||
else
|
||
{
|
||
model_dir = TEXT("models/photo");
|
||
modelType = eModelTypePhoto;
|
||
}
|
||
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_EDIT_OUT_EXT), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
outputExt = buf;
|
||
if (outputExt.length() > 0 && outputExt[0] != TEXT('.'))
|
||
outputExt = TEXT(".") + outputExt;
|
||
}
|
||
|
||
if (SendMessage(GetDlgItem(dh, IDC_RADIO_MODE_CPU), BM_GETCHECK, 0, 0))
|
||
process = "cpu";
|
||
else
|
||
process = "gpu";
|
||
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_EDIT_INPUT_EXT_LIST), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
inputFileExt = buf;
|
||
|
||
// input_extention_list<73><EFBFBD><F095B68E><EFBFBD><EFBFBD>̔z<CC94><7A><EFBFBD>ɂ<EFBFBD><C982><EFBFBD>
|
||
|
||
typedef boost::char_separator<TCHAR> char_separator;
|
||
typedef boost::tokenizer<char_separator, tstring::const_iterator, tstring> tokenizer;
|
||
|
||
char_separator sep(TEXT(":"), TEXT(""), boost::drop_empty_tokens);
|
||
tokenizer tokens(inputFileExt, sep);
|
||
|
||
for (auto tok_iter = tokens.begin(); tok_iter != tokens.end(); ++tok_iter)
|
||
{
|
||
tstring ext(*tok_iter);
|
||
std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
|
||
extList.push_back(TEXT(".") + ext);
|
||
}
|
||
}
|
||
|
||
if (!NotSyncCropSize)
|
||
{
|
||
TCHAR buf[AR_PATH_MAX] = TEXT("");
|
||
GetWindowText(GetDlgItem(dh, IDC_COMBO_CROP_SIZE), buf, _countof(buf));
|
||
buf[_countof(buf) - 1] = TEXT('\0');
|
||
|
||
TCHAR *ptr = nullptr;
|
||
crop_size = _tcstol(buf, &ptr, 10);
|
||
if (!ptr || *ptr != '\0' || crop_size <= 0)
|
||
{
|
||
crop_size = 128;
|
||
ret = false;
|
||
|
||
MessageBox(dh, TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>T<EFBFBD>C<EFBFBD>Y<EFBFBD><EFBFBD>0<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>傫<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ł<EFBFBD><EFBFBD><EFBFBD><EFBFBD>K<EFBFBD>v<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
}
|
||
}
|
||
|
||
use_tta = SendMessage(GetDlgItem(dh, IDC_CHECK_TTA), BM_GETCHECK, 0, 0) == BST_CHECKED;
|
||
|
||
return ret;
|
||
}
|
||
|
||
void SetCropSizeList(const boost::filesystem::path &input_path)
|
||
{
|
||
HWND hcrop = GetDlgItem(dh, IDC_COMBO_CROP_SIZE);
|
||
|
||
int gcd = 1;
|
||
if (boost::filesystem::is_directory(input_path))
|
||
{
|
||
BOOST_FOREACH(const boost::filesystem::path& p, std::make_pair(boost::filesystem::recursive_directory_iterator(input_path),
|
||
boost::filesystem::recursive_directory_iterator()))
|
||
{
|
||
|
||
if (!boost::filesystem::is_directory(p))
|
||
{
|
||
tstring ext(getTString(p.extension()));
|
||
#ifdef UNICODE
|
||
std::transform(ext.begin(), ext.end(), ext.begin(), ::towlower);
|
||
#else
|
||
std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
|
||
#endif
|
||
if (std::find(extList.begin(), extList.end(), ext) != extList.end())
|
||
{
|
||
auto mat = Waifu2x::LoadMat(p.string());
|
||
if (mat.empty())
|
||
continue;
|
||
|
||
auto size = mat.size();
|
||
mat.release();
|
||
|
||
gcd = boost::math::gcd(size.width, size.height);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
auto mat = Waifu2x::LoadMat(input_path.string());
|
||
if (mat.empty())
|
||
return;
|
||
|
||
auto size = mat.size();
|
||
mat.release();
|
||
|
||
gcd = boost::math::gcd(size.width, size.height);
|
||
}
|
||
|
||
while (SendMessage(hcrop, CB_GETCOUNT, 0, 0) != 0)
|
||
SendMessage(hcrop, CB_DELETESTRING, 0, 0);
|
||
|
||
// <20>ő<EFBFBD><C591><EFBFBD><EFBFBD>̖̃<F1909482><CC83>X<EFBFBD>g<EFBFBD>擾
|
||
std::vector<int> list(CommonDivisorList(gcd));
|
||
|
||
// MinCommonDivisor<6F><72><EFBFBD><EFBFBD><EFBFBD>̖폜
|
||
list.erase(std::remove_if(list.begin(), list.end(), [](const int v)
|
||
{
|
||
return v < MinCommonDivisor;
|
||
}
|
||
), list.end());
|
||
|
||
int mindiff = INT_MAX;
|
||
int defaultIndex = -1;
|
||
for (int i = 0; i < list.size(); i++)
|
||
{
|
||
const int n = list[i];
|
||
|
||
tstring str(to_tstring(n));
|
||
SendMessage(hcrop, CB_ADDSTRING, 0, (LPARAM)str.c_str());
|
||
|
||
const int diff = abs(DefaultCommonDivisor - n);
|
||
if (DefaultCommonDivisorRange.first <= n && n <= DefaultCommonDivisorRange.second && diff < mindiff)
|
||
{
|
||
mindiff = diff;
|
||
defaultIndex = i;
|
||
}
|
||
}
|
||
|
||
SendMessage(hcrop, CB_ADDSTRING, 0, (LPARAM)TEXT("-----------------------"));
|
||
|
||
// CropSizeList<73>̒l<CC92><6C><EFBFBD>lj<EFBFBD><C789><EFBFBD><EFBFBD>Ă<EFBFBD><C482><EFBFBD>
|
||
mindiff = INT_MAX;
|
||
int defaultListIndex = -1;
|
||
for (const auto n : CropSizeList)
|
||
{
|
||
tstring str(to_tstring(n));
|
||
const int index = SendMessage(hcrop, CB_ADDSTRING, 0, (LPARAM)str.c_str());
|
||
|
||
const int diff = abs(DefaultCommonDivisor - n);
|
||
if (DefaultCommonDivisorRange.first <= n && n <= DefaultCommonDivisorRange.second && diff < mindiff)
|
||
{
|
||
mindiff = diff;
|
||
defaultListIndex = index;
|
||
}
|
||
}
|
||
|
||
if (defaultIndex == -1)
|
||
defaultIndex = defaultListIndex;
|
||
|
||
if (GetWindowTextLength(hcrop) == 0)
|
||
SendMessage(hcrop, CB_SETCURSEL, defaultIndex, 0);
|
||
}
|
||
|
||
void ProcessWaifu2x()
|
||
{
|
||
const boost::filesystem::path input_path(boost::filesystem::absolute(input_str));
|
||
|
||
std::vector<std::pair<tstring, tstring>> file_paths;
|
||
if (boost::filesystem::is_directory(input_path)) // input_path<74><68><EFBFBD>t<EFBFBD>H<EFBFBD><48><EFBFBD>_<EFBFBD>Ȃ炻<C882>̃f<CC83>B<EFBFBD><42><EFBFBD>N<EFBFBD>g<EFBFBD><67><EFBFBD>ȉ<EFBFBD><C889>̉摜<CC89>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD>ꊇ<EFBFBD>ϊ<EFBFBD>
|
||
{
|
||
boost::filesystem::path output_path(output_str);
|
||
|
||
output_path = boost::filesystem::absolute(output_path);
|
||
|
||
if (!boost::filesystem::exists(output_path))
|
||
{
|
||
if (!boost::filesystem::create_directory(output_path))
|
||
{
|
||
SendMessage(dh, WM_FAILD_CREATE_DIR, (WPARAM)&output_path, 0);
|
||
PostMessage(dh, WM_END_THREAD, 0, 0);
|
||
// printf("<22>o<EFBFBD>̓t<CD83>H<EFBFBD><48><EFBFBD>_<EFBFBD>u%s<>v<EFBFBD>̍쐬<CC8D>Ɏ<EFBFBD><C98E>s<EFBFBD><73><EFBFBD>܂<EFBFBD><DC82><EFBFBD>\n", output_path.string().c_str());
|
||
return;
|
||
}
|
||
}
|
||
|
||
// <20>ϊ<EFBFBD><CF8A><EFBFBD><EFBFBD><EFBFBD><EFBFBD>摜<EFBFBD>̓<EFBFBD><CC93>́A<CD81>o<EFBFBD>̓p<CD83>X<EFBFBD><58><EFBFBD>擾
|
||
const auto func = [this, &input_path, &output_path, &file_paths](const boost::filesystem::path &path)
|
||
{
|
||
BOOST_FOREACH(const boost::filesystem::path& p, std::make_pair(boost::filesystem::recursive_directory_iterator(path),
|
||
boost::filesystem::recursive_directory_iterator()))
|
||
{
|
||
if (!boost::filesystem::is_directory(p))
|
||
{
|
||
tstring ext(getTString(p.extension()));
|
||
#ifdef UNICODE
|
||
std::transform(ext.begin(), ext.end(), ext.begin(), ::towlower);
|
||
#else
|
||
std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
|
||
#endif
|
||
|
||
if (std::find(extList.begin(), extList.end(), ext) != extList.end())
|
||
{
|
||
const auto out_relative = relativePath(p, input_path);
|
||
const auto out_absolute = output_path / out_relative;
|
||
|
||
const auto out = getTString(out_absolute.branch_path() / out_absolute.stem()) + outputExt;
|
||
|
||
file_paths.emplace_back(getTString(p), out);
|
||
}
|
||
}
|
||
}
|
||
|
||
return true;
|
||
};
|
||
|
||
if (!func(input_path))
|
||
return;
|
||
|
||
for (const auto &p : file_paths)
|
||
{
|
||
const boost::filesystem::path out_path(p.second);
|
||
const boost::filesystem::path out_dir(out_path.parent_path());
|
||
|
||
if (!boost::filesystem::exists(out_dir))
|
||
{
|
||
if (!boost::filesystem::create_directories(out_dir))
|
||
{
|
||
SendMessage(dh, WM_FAILD_CREATE_DIR, (WPARAM)&out_dir, 0);
|
||
PostMessage(dh, WM_END_THREAD, 0, 0);
|
||
//printf("<22>o<EFBFBD>̓t<CD83>H<EFBFBD><48><EFBFBD>_<EFBFBD>u%s<>v<EFBFBD>̍쐬<CC8D>Ɏ<EFBFBD><C98E>s<EFBFBD><73><EFBFBD>܂<EFBFBD><DC82><EFBFBD>\n", out_absolute.string().c_str());
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
else
|
||
file_paths.emplace_back(input_str, output_str);
|
||
|
||
bool isFirst = true;
|
||
|
||
const auto ProgessFunc = [this, &isFirst](const int ProgressFileMax, const int ProgressFileNow)
|
||
{
|
||
if (isFirst)
|
||
{
|
||
isFirst = true;
|
||
|
||
SendMessage(GetDlgItem(dh, IDC_PROGRESS), PBM_SETRANGE32, 0, ProgressFileMax);
|
||
}
|
||
|
||
SendMessage(GetDlgItem(dh, IDC_PROGRESS), PBM_SETPOS, ProgressFileNow, 0);
|
||
};
|
||
|
||
const auto cuDNNCheckStartTime = std::chrono::system_clock::now();
|
||
|
||
if (process == "gpu")
|
||
Waifu2x::can_use_cuDNN();
|
||
|
||
const auto cuDNNCheckEndTime = std::chrono::system_clock::now();
|
||
|
||
Waifu2x::eWaifu2xError ret;
|
||
|
||
Waifu2x w;
|
||
ret = w.init(__argc, __argv, mode, noise_level, scale_ratio, model_dir, process, use_tta, crop_size, batch_size);
|
||
if(ret != Waifu2x::eWaifu2xError_OK)
|
||
SendMessage(dh, WM_ON_WAIFU2X_ERROR, (WPARAM)&ret, 0);
|
||
else
|
||
{
|
||
const auto InitEndTime = std::chrono::system_clock::now();
|
||
|
||
for (const auto &p : file_paths)
|
||
{
|
||
ret = w.waifu2x(p.first, p.second, [this]()
|
||
{
|
||
return cancelFlag;
|
||
});
|
||
|
||
if (ret != Waifu2x::eWaifu2xError_OK)
|
||
{
|
||
SendMessage(dh, WM_ON_WAIFU2X_ERROR, (WPARAM)&ret, (LPARAM)&p);
|
||
|
||
if (ret == Waifu2x::eWaifu2xError_Cancel)
|
||
break;
|
||
}
|
||
}
|
||
|
||
const auto ProcessEndTime = std::chrono::system_clock::now();
|
||
|
||
cuDNNCheckTime = cuDNNCheckEndTime - cuDNNCheckStartTime;
|
||
InitTime = InitEndTime - cuDNNCheckEndTime;
|
||
ProcessTime = ProcessEndTime - InitEndTime;
|
||
usedProcess = w.used_process();
|
||
}
|
||
|
||
PostMessage(dh, WM_END_THREAD, 0, 0);
|
||
}
|
||
|
||
void ReplaceAddString() // <20>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD>̎<EFBFBD><CC8E><EFBFBD><EFBFBD>ݒ蕔<DD92><E89594><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
{
|
||
SyncMember(true);
|
||
|
||
const boost::filesystem::path output_path(output_str);
|
||
tstring stem;
|
||
|
||
if (!boost::filesystem::is_directory(input_str))
|
||
stem = getTString(output_path.stem());
|
||
else
|
||
stem = getTString(output_path.filename());
|
||
|
||
if (stem.length() > 0 && stem.length() >= autoSetAddName.length())
|
||
{
|
||
const auto pos = stem.find(autoSetAddName);
|
||
if (pos != tstring::npos)
|
||
{
|
||
const tstring addstr(AddName());
|
||
|
||
auto new_name = stem;
|
||
new_name.replace(pos, autoSetAddName.length(), addstr);
|
||
|
||
autoSetAddName = addstr;
|
||
|
||
boost::filesystem::path new_out_path;
|
||
if (!boost::filesystem::is_directory(input_str))
|
||
new_out_path = output_path.branch_path() / (new_name + outputExt);
|
||
else
|
||
new_out_path = output_path.branch_path() / (new_name);
|
||
|
||
SetWindowText(GetDlgItem(dh, IDC_EDIT_OUTPUT), getTString(new_out_path).c_str());
|
||
}
|
||
}
|
||
}
|
||
|
||
void AddLogMessage(const TCHAR *msg)
|
||
{
|
||
if (logMessage.length() == 0)
|
||
logMessage += msg;
|
||
else
|
||
logMessage += tstring(TEXT("\r\n")) + msg;
|
||
|
||
SetWindowText(GetDlgItem(dh, IDC_EDIT_LOG), logMessage.c_str());
|
||
}
|
||
|
||
void Waifu2xTime()
|
||
{
|
||
TCHAR msg[1024 * 2];
|
||
TCHAR *ptr = msg;
|
||
|
||
{
|
||
tstring p;
|
||
if (usedProcess == "cpu")
|
||
p = TEXT("CPU");
|
||
else if (usedProcess == "gpu")
|
||
p = TEXT("CUDA");
|
||
else // if (p == "cudnn")
|
||
p = TEXT("cuDNN");
|
||
|
||
ptr += _stprintf(ptr, TEXT("<EFBFBD>g<EFBFBD>p<EFBFBD>v<EFBFBD><EFBFBD><EFBFBD>Z<EFBFBD>b<EFBFBD>T<EFBFBD>[<5B><><EFBFBD>[<5B>h: %s\r\n"), p.c_str());
|
||
}
|
||
|
||
{
|
||
uint64_t t = std::chrono::duration_cast<std::chrono::milliseconds>(ProcessTime).count();
|
||
const int msec = t % 1000; t /= 1000;
|
||
const int sec = t % 60; t /= 60;
|
||
const int min = t % 60; t /= 60;
|
||
const int hour = (int)t;
|
||
ptr += _stprintf(ptr, TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>: %02d:%02d:%02d.%03d\r\n"), hour, min, sec, msec);
|
||
}
|
||
|
||
{
|
||
uint64_t t = std::chrono::duration_cast<std::chrono::milliseconds>(InitTime).count();
|
||
const int msec = t % 1000; t /= 1000;
|
||
const int sec = t % 60; t /= 60;
|
||
const int min = t % 60; t /= 60;
|
||
const int hour = (int)t;
|
||
ptr += _stprintf(ptr, TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>: %02d:%02d:%02d.%03d\r\n"), hour, min, sec, msec);
|
||
}
|
||
|
||
if (process == "gpu" || process == "cudnn")
|
||
{
|
||
uint64_t t = std::chrono::duration_cast<std::chrono::milliseconds>(cuDNNCheckTime).count();
|
||
const int msec = t % 1000; t /= 1000;
|
||
const int sec = t % 60; t /= 60;
|
||
const int min = t % 60; t /= 60;
|
||
const int hour = (int)t;
|
||
ptr += _stprintf(ptr, TEXT("cuDNN<EFBFBD>`<60>F<EFBFBD>b<EFBFBD>N<EFBFBD><4E><EFBFBD><EFBFBD>: %02d:%02d:%02d.%03d"), hour, min, sec, msec);
|
||
}
|
||
|
||
AddLogMessage(msg);
|
||
}
|
||
|
||
void SaveIni(const bool isSyncMember = true)
|
||
{
|
||
if (isSyncMember)
|
||
SyncMember(true);
|
||
|
||
const boost::filesystem::path SettingFilePath(exeDir / SettingFileName);
|
||
|
||
tstring tScale;
|
||
tstring tmode;
|
||
tstring tprcess;
|
||
tstring toutputExt;
|
||
|
||
tScale = to_tstring(scale_ratio);
|
||
|
||
if (mode == ("noise"))
|
||
tmode = TEXT("noise");
|
||
else if (mode == ("scale"))
|
||
tmode = TEXT("scale");
|
||
else if (mode == ("auto_scale"))
|
||
tmode = TEXT("auto_scale");
|
||
else // noise_scale
|
||
tmode = TEXT("noise_scale");
|
||
|
||
if (process == "gpu")
|
||
tprcess = TEXT("gpu");
|
||
else
|
||
tprcess = TEXT("cpu");
|
||
|
||
toutputExt = outputExt;
|
||
if (toutputExt.length() > 0 && toutputExt[0] == TEXT('.'))
|
||
toutputExt.erase(0, 1);
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastScale"), tScale.c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastOutputExt"), toutputExt.c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastInputFileExt"), inputFileExt.c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastMode"), tmode.c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastNoiseLevel"), to_tstring(noise_level).c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastProcess"), tprcess.c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastModel"), to_tstring(modelType).c_str(), getTString(SettingFilePath).c_str());
|
||
|
||
WritePrivateProfileString(TEXT("Setting"), TEXT("LastUseTTA"), to_tstring(use_tta ? 1 : 0).c_str(), getTString(SettingFilePath).c_str());
|
||
}
|
||
|
||
// <20>o<EFBFBD>̓p<CD83>X<EFBFBD><58><EFBFBD>I<EFBFBD><49><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
static UINT_PTR CALLBACK OFNHookProcIn(
|
||
_In_ HWND hdlg,
|
||
_In_ UINT uiMsg,
|
||
_In_ WPARAM wParam,
|
||
_In_ LPARAM lParam
|
||
)
|
||
{
|
||
switch (uiMsg)
|
||
{
|
||
case WM_INITDIALOG:
|
||
{
|
||
// <20>_<EFBFBD>C<EFBFBD>A<EFBFBD><41><EFBFBD>O<EFBFBD>𒆉<EFBFBD><F0928689>ɕ\<5C><>
|
||
|
||
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:
|
||
{
|
||
// <20>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD>ƃt<C683>H<EFBFBD><48><EFBFBD>_<EFBFBD><5F><EFBFBD>I<EFBFBD><49><EFBFBD>ł<EFBFBD><C582><EFBFBD><EFBFBD>悤<EFBFBD>ɂ<EFBFBD><C982><EFBFBD>
|
||
|
||
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;
|
||
}
|
||
|
||
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")),
|
||
use_tta(false), crop_size(128), batch_size(1), isLastError(false)
|
||
{
|
||
}
|
||
|
||
void Exec(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
if (processThread.joinable())
|
||
return;
|
||
|
||
if (!SyncMember(false))
|
||
return;
|
||
|
||
if (input_str.length() == 0)
|
||
{
|
||
MessageBox(dh, TEXT("<EFBFBD><EFBFBD><EFBFBD>̓p<EFBFBD>X<EFBFBD><EFBFBD><EFBFBD>w<EFBFBD>肵<EFBFBD>ĉ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
return;
|
||
}
|
||
|
||
if (output_str.length() == 0)
|
||
{
|
||
MessageBox(dh, TEXT("<EFBFBD>o<EFBFBD>̓p<EFBFBD>X<EFBFBD><EFBFBD><EFBFBD>w<EFBFBD>肵<EFBFBD>ĉ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
return;
|
||
}
|
||
|
||
if (outputExt.length() == 0)
|
||
{
|
||
MessageBox(dh, TEXT("<EFBFBD>o<EFBFBD>͊g<EFBFBD><EFBFBD><EFBFBD>q<EFBFBD><EFBFBD><EFBFBD>w<EFBFBD>肵<EFBFBD>ĉ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
return;
|
||
}
|
||
|
||
if (process == "gpu" || process == "cudnn")
|
||
{
|
||
const auto flag = Waifu2x::can_use_CUDA();
|
||
switch (flag)
|
||
{
|
||
case Waifu2x::eWaifu2xCudaError_NotFind:
|
||
MessageBox(dh, TEXT("GPU<EFBFBD>ŕϊ<EFBFBD><EFBFBD>o<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>C<EFBFBD><43><EFBFBD>X<EFBFBD>g<EFBFBD>[<5B><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482>Ȃ<EFBFBD><C882>\<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>C<EFBFBD><43><EFBFBD>X<EFBFBD>g<EFBFBD>[<5B><><EFBFBD><EFBFBD><EFBFBD>ĉ<EFBFBD><C489><EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
return;
|
||
case Waifu2x::eWaifu2xCudaError_OldVersion:
|
||
MessageBox(dh, TEXT("GPU<EFBFBD>ŕϊ<EFBFBD><EFBFBD>o<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B>̃o<CC83>[<5B>W<EFBFBD><57><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Â<EFBFBD><C382>\<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>X<EFBFBD>V<EFBFBD><56><EFBFBD>ĉ<EFBFBD><C489><EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
return;
|
||
}
|
||
}
|
||
|
||
SaveIni(true); // <20><><EFBFBD><EFBFBD><EFBFBD>I<EFBFBD><49><EFBFBD>̉\<5C><><EFBFBD><EFBFBD><EFBFBD>l<EFBFBD><6C><EFBFBD>Ď<EFBFBD><C48E>s<EFBFBD><73><EFBFBD>ɐݒ<C990><DD92>ۑ<EFBFBD>
|
||
|
||
SendMessage(GetDlgItem(dh, IDC_PROGRESS), PBM_SETPOS, 0, 0);
|
||
cancelFlag = false;
|
||
isLastError = false;
|
||
|
||
processThread = std::thread(std::bind(&DialogEvent::ProcessWaifu2x, this));
|
||
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), TRUE);
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_EXEC), FALSE);
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CHECK_CUDNN), FALSE);
|
||
|
||
SetWindowText(GetDlgItem(hWnd, IDC_EDIT_LOG), TEXT(""));
|
||
logMessage.clear();
|
||
}
|
||
|
||
void WaitThreadExit(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
processThread.join();
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), FALSE);
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_EXEC), TRUE);
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CHECK_CUDNN), TRUE);
|
||
|
||
if (!isLastError)
|
||
{
|
||
if (!cancelFlag)
|
||
AddLogMessage(TEXT("<EFBFBD>ϊ<EFBFBD><EFBFBD>ɐ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"));
|
||
|
||
Waifu2xTime();
|
||
MessageBeep(MB_ICONASTERISK);
|
||
}
|
||
else
|
||
MessageBox(dh, TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>[<5B><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><DC82><EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
}
|
||
|
||
void OnDialogEnd(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
SaveIni();
|
||
|
||
if (!processThread.joinable())
|
||
PostQuitMessage(0);
|
||
else
|
||
MessageBeep(MB_ICONEXCLAMATION);
|
||
}
|
||
|
||
void OnFaildCreateDir(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
const boost::filesystem::path *p = (const boost::filesystem::path *)wParam;
|
||
|
||
// <20>o<EFBFBD>̓t<CD83>H<EFBFBD><48><EFBFBD>_<EFBFBD>u%s<>v<EFBFBD>̍쐬<CC8D>Ɏ<EFBFBD><C98E>s<EFBFBD><73><EFBFBD>܂<EFBFBD><DC82><EFBFBD>\n", out_absolute.string().c_str());
|
||
tstring msg(TEXT("<EFBFBD>o<EFBFBD>̓t<EFBFBD>H<EFBFBD><EFBFBD><EFBFBD>_\r\n<EFBFBD>u"));
|
||
msg += getTString(*p);
|
||
msg += TEXT("<EFBFBD>v\r\n<EFBFBD>̍쐬<EFBFBD>Ɏ<EFBFBD><EFBFBD>s<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>");
|
||
|
||
MessageBox(dh, msg.c_str(), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), MB_OK | MB_ICONERROR);
|
||
|
||
isLastError = true;
|
||
}
|
||
|
||
void OnWaifu2xError(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
const Waifu2x::eWaifu2xError ret = *(const Waifu2x::eWaifu2xError *)wParam;
|
||
|
||
if (ret != Waifu2x::eWaifu2xError_OK)
|
||
{
|
||
TCHAR msg[1024] = TEXT("");
|
||
|
||
if (lParam == 0)
|
||
{
|
||
switch (ret)
|
||
{
|
||
case Waifu2x::eWaifu2xError_Cancel:
|
||
_stprintf(msg, TEXT("<EFBFBD>L<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Z<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_InvalidParameter:
|
||
_stprintf(msg, TEXT("<EFBFBD>p<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>[<5B>^<5E><><EFBFBD>s<EFBFBD><73><EFBFBD>ł<EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedOpenModelFile:
|
||
_stprintf(msg, TEXT("<EFBFBD><EFBFBD><EFBFBD>f<EFBFBD><EFBFBD><EFBFBD>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD>J<EFBFBD><4A><EFBFBD>܂<EFBFBD><DC82><EFBFBD><EFBFBD>ł<EFBFBD><C582><EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedParseModelFile:
|
||
_stprintf(msg, TEXT("<EFBFBD><EFBFBD><EFBFBD>f<EFBFBD><EFBFBD><EFBFBD>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482>܂<EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedConstructModel:
|
||
_stprintf(msg, TEXT("<EFBFBD>l<EFBFBD>b<EFBFBD>g<EFBFBD><EFBFBD><EFBFBD>[<5B>N<EFBFBD>̍\<EFBFBD>z<EFBFBD>Ɏ<EFBFBD><EFBFBD>s<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"));
|
||
break;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
const auto &fp = *(const std::pair<tstring, tstring> *)lParam;
|
||
|
||
switch (ret)
|
||
{
|
||
case Waifu2x::eWaifu2xError_Cancel:
|
||
_stprintf(msg, TEXT("<EFBFBD>L<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Z<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_InvalidParameter:
|
||
_stprintf(msg, TEXT("<EFBFBD>p<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>[<5B>^<5E><><EFBFBD>s<EFBFBD><73><EFBFBD>ł<EFBFBD>"));
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedOpenInputFile:
|
||
_stprintf(msg, TEXT("<EFBFBD><EFBFBD><EFBFBD>͉摜<EFBFBD>u%s<>v<EFBFBD><76><EFBFBD>J<EFBFBD><4A><EFBFBD>܂<EFBFBD><DC82><EFBFBD><EFBFBD>ł<EFBFBD><C582><EFBFBD>"), fp.first.c_str());
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedOpenOutputFile:
|
||
_stprintf(msg, TEXT("<EFBFBD>o<EFBFBD>͉摜<EFBFBD><EFBFBD><EFBFBD>u%s<>v<EFBFBD>ɏ<EFBFBD><C98F><EFBFBD><EFBFBD><EFBFBD><EFBFBD>߂܂<DF82><DC82><EFBFBD><EFBFBD>ł<EFBFBD><C582><EFBFBD>"), fp.second.c_str());
|
||
break;
|
||
case Waifu2x::eWaifu2xError_FailedProcessCaffe:
|
||
_stprintf(msg, TEXT("<EFBFBD><EFBFBD><EFBFBD>ԏ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ɏ<EFBFBD><EFBFBD>s<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"));
|
||
break;
|
||
}
|
||
}
|
||
|
||
AddLogMessage(msg);
|
||
|
||
if (ret != Waifu2x::eWaifu2xError_Cancel)
|
||
isLastError = true;
|
||
}
|
||
}
|
||
|
||
void Create(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
dh = hWnd;
|
||
|
||
{
|
||
TCHAR texepath[1024 * 3] = TEXT("");
|
||
GetModuleFileName(NULL, texepath, _countof(texepath));
|
||
texepath[_countof(texepath) - 1] = TEXT('\0');
|
||
|
||
const boost::filesystem::path exePath(texepath);
|
||
exeDir = exePath.branch_path();
|
||
}
|
||
|
||
const boost::filesystem::path CropSizeListPath(exeDir / CropSizeListName);
|
||
std::ifstream ifs(CropSizeListPath.wstring());
|
||
if (ifs)
|
||
{
|
||
std::string str;
|
||
while (getline(ifs, str))
|
||
{
|
||
char *ptr = nullptr;
|
||
const long n = strtol(str.c_str(), &ptr, 10);
|
||
if (ptr && *ptr == '\0')
|
||
CropSizeList.push_back(n);
|
||
}
|
||
}
|
||
|
||
{
|
||
HWND hcrop = GetDlgItem(dh, IDC_COMBO_CROP_SIZE);
|
||
|
||
SendMessage(hcrop, CB_ADDSTRING, 0, (LPARAM)TEXT("-----------------------"));
|
||
|
||
// CropSizeList<73>̒l<CC92><6C><EFBFBD>lj<EFBFBD><C789><EFBFBD><EFBFBD>Ă<EFBFBD><C482><EFBFBD>
|
||
int mindiff = INT_MAX;
|
||
int defaultListIndex = -1;
|
||
for (const auto n : CropSizeList)
|
||
{
|
||
tstring str(to_tstring(n));
|
||
const int index = SendMessage(hcrop, CB_ADDSTRING, 0, (LPARAM)str.c_str());
|
||
|
||
const int diff = abs(DefaultCommonDivisor - n);
|
||
if (DefaultCommonDivisorRange.first <= n && n <= DefaultCommonDivisorRange.second && diff < mindiff)
|
||
{
|
||
mindiff = diff;
|
||
defaultListIndex = index;
|
||
}
|
||
}
|
||
|
||
if (GetWindowTextLength(hcrop) == 0)
|
||
SendMessage(hcrop, CB_SETCURSEL, defaultListIndex, 0);
|
||
}
|
||
|
||
const boost::filesystem::path SettingFilePath(exeDir / SettingFileName);
|
||
|
||
tstring tScale;
|
||
tstring tmode;
|
||
tstring tprcess;
|
||
{
|
||
TCHAR tmp[1000];
|
||
|
||
GetPrivateProfileString(TEXT("Setting"), TEXT("LastScale"), TEXT("2.00"), tmp, _countof(tmp), getTString(SettingFilePath).c_str());
|
||
tmp[_countof(tmp) - 1] = TEXT('\0');
|
||
tScale = tmp;
|
||
|
||
GetPrivateProfileString(TEXT("Setting"), TEXT("LastOutputExt"), TEXT("png"), tmp, _countof(tmp), getTString(SettingFilePath).c_str());
|
||
tmp[_countof(tmp) - 1] = TEXT('\0');
|
||
outputExt = tmp;
|
||
|
||
GetPrivateProfileString(TEXT("Setting"), TEXT("LastInputFileExt"), TEXT("png:jpg:jpeg:tif:tiff:bmp:tga"), tmp, _countof(tmp), getTString(SettingFilePath).c_str());
|
||
tmp[_countof(tmp) - 1] = TEXT('\0');
|
||
inputFileExt = tmp;
|
||
|
||
GetPrivateProfileString(TEXT("Setting"), TEXT("LastMode"), TEXT("noise_scale"), tmp, _countof(tmp), getTString(SettingFilePath).c_str());
|
||
tmp[_countof(tmp) - 1] = TEXT('\0');
|
||
tmode = tmp;
|
||
|
||
noise_level = GetPrivateProfileInt(TEXT("Setting"), TEXT("LastNoiseLevel"), 1, getTString(SettingFilePath).c_str());
|
||
|
||
GetPrivateProfileString(TEXT("Setting"), TEXT("LastProcess"), TEXT("gpu"), tmp, _countof(tmp), getTString(SettingFilePath).c_str());
|
||
tmp[_countof(tmp) - 1] = TEXT('\0');
|
||
tprcess = tmp;
|
||
|
||
modelType = (eModelType)GetPrivateProfileInt(TEXT("Setting"), TEXT("LastModel"), 0, getTString(SettingFilePath).c_str());
|
||
|
||
use_tta = GetPrivateProfileInt(TEXT("Setting"), TEXT("LastUseTTA"), 0, getTString(SettingFilePath).c_str()) != 0;
|
||
}
|
||
|
||
TCHAR *ptr = nullptr;
|
||
const double tempScale = _tcstod(tScale.c_str(), &ptr);
|
||
if (!ptr || *ptr != TEXT('\0') || tempScale <= 0.0)
|
||
tScale = TEXT("2.00");
|
||
|
||
if (outputExt.length() > 0 && outputExt[0] == TEXT('.'))
|
||
outputExt.erase(0, 1);
|
||
|
||
if (!(1 <= noise_level && noise_level <= 2))
|
||
noise_level = 1;
|
||
|
||
if (tprcess == TEXT("gpu"))
|
||
process = "gpu";
|
||
else
|
||
process = "cpu";
|
||
|
||
if (!((eModelType)0 <= modelType && modelType < eModelTypeEnd))
|
||
modelType = eModelTypeRGB;
|
||
|
||
if (tmode == TEXT("noise"))
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_AUTO_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else if (tmode == TEXT("scale"))
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_SCALE), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_AUTO_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else if (tmode == TEXT("auto_scale"))
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_AUTO_SCALE), BM_SETCHECK, BST_CHECKED, 0);
|
||
}
|
||
else // noise_scale
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE_SCALE), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_NOISE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_AUTO_SCALE), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
|
||
if (noise_level == 1)
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL1), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL2), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL1), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL2), BM_SETCHECK, BST_CHECKED, 0);
|
||
}
|
||
|
||
if (process == "gpu")
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_GPU), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_CPU), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_GPU), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_CPU), BM_SETCHECK, BST_CHECKED, 0);
|
||
}
|
||
|
||
if (modelType == eModelTypeRGB)
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_RGB), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_PHOTO), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_Y), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else if (modelType == eModelTypePhoto)
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_RGB), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_PHOTO), BM_SETCHECK, BST_CHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_Y), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
}
|
||
else
|
||
{
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_RGB), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_PHOTO), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODEL_Y), BM_SETCHECK, BST_CHECKED, 0);
|
||
}
|
||
|
||
if (use_tta)
|
||
SendMessage(GetDlgItem(hWnd, IDC_CHECK_TTA), BM_SETCHECK, BST_CHECKED, 0);
|
||
else
|
||
SendMessage(GetDlgItem(hWnd, IDC_CHECK_TTA), BM_SETCHECK, BST_UNCHECKED, 0);
|
||
|
||
SetWindowText(GetDlgItem(hWnd, IDC_EDIT_SCALE_RATIO), tScale.c_str());
|
||
SetWindowText(GetDlgItem(hWnd, IDC_EDIT_OUT_EXT), outputExt.c_str());
|
||
SetWindowText(GetDlgItem(hWnd, IDC_EDIT_INPUT_EXT_LIST), inputFileExt.c_str());
|
||
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), FALSE);
|
||
}
|
||
|
||
void Cancel(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
cancelFlag = true;
|
||
EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), FALSE);
|
||
}
|
||
|
||
void RadioButtom(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
ReplaceAddString();
|
||
}
|
||
|
||
void CheckCUDNN(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
const auto flag = Waifu2x::can_use_CUDA();
|
||
switch (flag)
|
||
{
|
||
case Waifu2x::eWaifu2xCudaError_NotFind:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>C<EFBFBD><43><EFBFBD>X<EFBFBD>g<EFBFBD>[<5B><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482>Ȃ<EFBFBD><C882>\<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>C<EFBFBD><43><EFBFBD>X<EFBFBD>g<EFBFBD>[<5B><><EFBFBD><EFBFBD><EFBFBD>ĉ<EFBFBD><C489><EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
return;
|
||
case Waifu2x::eWaifu2xCudaError_OldVersion:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B>̃o<CC83>[<5B>W<EFBFBD><57><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Â<EFBFBD><C382>\<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD>B\r\nCUDA<EFBFBD>h<EFBFBD><EFBFBD><EFBFBD>C<EFBFBD>o<EFBFBD>[<5B><><EFBFBD>X<EFBFBD>V<EFBFBD><56><EFBFBD>ĉ<EFBFBD><C489><EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
return;
|
||
}
|
||
|
||
switch (Waifu2x::can_use_cuDNN())
|
||
{
|
||
case Waifu2x::eWaifu2xcuDNNError_OK:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD><EFBFBD><EFBFBD>g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONINFORMATION);
|
||
break;
|
||
case Waifu2x::eWaifu2xcuDNNError_NotFind:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\n<EFBFBD>u") TEXT(CUDNN_DLL_NAME) TEXT("<EFBFBD>v<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
break;
|
||
case Waifu2x::eWaifu2xcuDNNError_OldVersion:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\n<EFBFBD>u") TEXT(CUDNN_DLL_NAME) TEXT("<EFBFBD>v<EFBFBD>̃o<EFBFBD>[<5B>W<EFBFBD><57><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Â<EFBFBD><C382>ł<EFBFBD><C582>Bv2<76><32><EFBFBD>g<EFBFBD><67><EFBFBD>ĉ<EFBFBD><C489><EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
break;
|
||
case Waifu2x::eWaifu2xcuDNNError_CannotCreate:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B\r\ncuDNN<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>o<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>B"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
break;
|
||
default:
|
||
MessageBox(dh, TEXT("cuDNN<EFBFBD>͎g<EFBFBD><EFBFBD><EFBFBD>܂<EFBFBD><EFBFBD><EFBFBD>"), TEXT("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
|
||
}
|
||
}
|
||
|
||
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("<EFBFBD><EFBFBD><EFBFBD>̓t<EFBFBD>@<40>C<EFBFBD><43>/<2F>t<EFBFBD>H<EFBFBD><48><EFBFBD>_<EFBFBD><5F><EFBFBD><EFBFBD><EFBFBD>݂<EFBFBD><DD82>܂<EFBFBD><DC82><EFBFBD>"), TEXT("<EFBFBD>G<EFBFBD><EFBFBD><EFBFBD>["), 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;
|
||
}
|
||
|
||
// <20><><EFBFBD><EFBFBD><EFBFBD>œn<C593><6E><EFBFBD><EFBFBD><EFBFBD><EFBFBD>hWnd<6E><64>IDC_EDIT<49><54>HWND(<28>R<EFBFBD><52><EFBFBD>g<EFBFBD><67><EFBFBD>[<5B><><EFBFBD>̃C<CC83>x<EFBFBD><78><EFBFBD>g<EFBFBD><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD>)
|
||
LRESULT DropInput(HWND hWnd, WPARAM wParam, LPARAM lParam, WNDPROC OrgSubWnd, LPVOID lpData)
|
||
{
|
||
TCHAR szTmp[AR_PATH_MAX];
|
||
|
||
// <20>h<EFBFBD><68><EFBFBD>b<EFBFBD>v<EFBFBD><76><EFBFBD>ꂽ<EFBFBD>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>擾
|
||
UINT FileNum = DragQueryFile((HDROP)wParam, 0xFFFFFFFF, szTmp, _countof(szTmp));
|
||
if (FileNum >= 1)
|
||
{
|
||
DragQueryFile((HDROP)wParam, 0, szTmp, _countof(szTmp));
|
||
szTmp[_countof(szTmp) - 1] = TEXT('\0');
|
||
|
||
OnSetInputFilePath(szTmp);
|
||
}
|
||
|
||
return 0L;
|
||
}
|
||
|
||
// <20><><EFBFBD><EFBFBD><EFBFBD>œn<C593><6E><EFBFBD><EFBFBD><EFBFBD><EFBFBD>hWnd<6E><64>IDC_EDIT<49><54>HWND(<28>R<EFBFBD><52><EFBFBD>g<EFBFBD><67><EFBFBD>[<5B><><EFBFBD>̃C<CC83>x<EFBFBD><78><EFBFBD>g<EFBFBD><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD>)
|
||
LRESULT DropOutput(HWND hWnd, WPARAM wParam, LPARAM lParam, WNDPROC OrgSubWnd, LPVOID lpData)
|
||
{
|
||
TCHAR szTmp[AR_PATH_MAX];
|
||
|
||
// <20>h<EFBFBD><68><EFBFBD>b<EFBFBD>v<EFBFBD><76><EFBFBD>ꂽ<EFBFBD>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>擾
|
||
UINT FileNum = DragQueryFile((HDROP)wParam, 0xFFFFFFFF, szTmp, AR_PATH_MAX);
|
||
if (FileNum >= 1)
|
||
{
|
||
DragQueryFile((HDROP)wParam, 0, szTmp, AR_PATH_MAX);
|
||
SetWindowText(hWnd, szTmp);
|
||
}
|
||
|
||
return 0L;
|
||
}
|
||
|
||
LRESULT TextInput(HWND hWnd, WPARAM wParam, LPARAM lParam, WNDPROC OrgSubWnd, LPVOID lpData)
|
||
{
|
||
const auto ret = CallWindowProc(OrgSubWnd, hWnd, WM_CHAR, wParam, lParam);
|
||
ReplaceAddString();
|
||
return ret;
|
||
}
|
||
|
||
void InputRef(HWND hWnd, WPARAM wParam, LPARAM lParam, LPVOID lpData)
|
||
{
|
||
SyncMember(false);
|
||
|
||
OPENFILENAME ofn;
|
||
TCHAR szPath[AR_PATH_MAX] = TEXT("");
|
||
TCHAR szFile[AR_PATH_MAX] = TEXT("");
|
||
|
||
GetCurrentDirectory(_countof(szPath), szPath);
|
||
szPath[_countof(szPath) - 1] = TEXT('\0');
|
||
|
||
tstring extStr;
|
||
for (const auto &ext : extList)
|
||
{
|
||
if (extStr.length() != 0)
|
||
extStr += TEXT(";*") + ext;
|
||
else
|
||
extStr = TEXT("*") + ext;
|
||
}
|
||
|
||
TCHAR szFilter[AR_PATH_MAX] = TEXT("");
|
||
TCHAR *tfp = szFilter;
|
||
|
||
if (extStr.length() > 0)
|
||
{
|
||
tfp += _stprintf(tfp, TEXT("<EFBFBD>w<EFBFBD>肳<EFBFBD>ꂽ<EFBFBD>ϊ<EFBFBD><EFBFBD>Ώۊg<EFBFBD><EFBFBD><EFBFBD>q(%s)"), extStr.c_str(), extStr.c_str());
|
||
tfp++;
|
||
|
||
memcpy(tfp, extStr.c_str(), extStr.length() * sizeof(TCHAR));
|
||
tfp += extStr.length();
|
||
|
||
*tfp = TEXT('\0');
|
||
tfp++;
|
||
}
|
||
|
||
const tstring allFilesTitle(TEXT("<EFBFBD><EFBFBD><EFBFBD>ׂẴt<EFBFBD>@<40>C<EFBFBD><43><EFBFBD>A<EFBFBD>t<EFBFBD>H<EFBFBD><48><EFBFBD>_(*.*)"));
|
||
memcpy(tfp, allFilesTitle.c_str(), allFilesTitle.length() * sizeof(TCHAR));
|
||
tfp += allFilesTitle.length();
|
||
*tfp = TEXT('\0');
|
||
tfp++;
|
||
|
||
const tstring allFilesExt(TEXT("*.*"));
|
||
memcpy(tfp, allFilesExt.c_str(), allFilesExt.length() * sizeof(TCHAR));
|
||
tfp += allFilesExt.length();
|
||
|
||
*tfp = TEXT('\0');
|
||
tfp++;
|
||
*tfp = TEXT('\0');
|
||
tfp++;
|
||
|
||
ofn.lStructSize = sizeof(ofn);
|
||
ofn.hwndOwner = NULL;
|
||
ofn.lpstrFile = szFile;
|
||
ofn.nMaxFile = _countof(szFile);
|
||
ofn.lpstrFilter = szFilter;
|
||
ofn.nFilterIndex = 1;
|
||
ofn.lpstrTitle = TEXT("<EFBFBD><EFBFBD><EFBFBD>͂<EFBFBD><EFBFBD><EFBFBD><EFBFBD>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD><EFBFBD><EFBFBD>t<EFBFBD>H<EFBFBD><48><EFBFBD>_<EFBFBD><5F><EFBFBD>I<EFBFBD><49><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>");
|
||
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);
|
||
}
|
||
}
|
||
};
|
||
|
||
|
||
int WINAPI WinMain(HINSTANCE hInstance,
|
||
HINSTANCE hPrevInstance,
|
||
LPSTR lpCmdLine,
|
||
int nCmdShow)
|
||
{
|
||
Waifu2x::init_liblary();
|
||
|
||
// <20>Ǘ<EFBFBD><C797>Ҍ<EFBFBD><D28C><EFBFBD><EFBFBD>ŋN<C58B><4E><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482>t<EFBFBD>@<40>C<EFBFBD><43><EFBFBD>̃h<CC83><68><EFBFBD>b<EFBFBD>v<EFBFBD><76><EFBFBD>t<F382AF95><74><EFBFBD><EFBFBD><EFBFBD>悤<EFBFBD>ɂ<EFBFBD><C982><EFBFBD>
|
||
ChangeWindowMessageFilter(WM_DROPFILES, MSGFLT_ADD);
|
||
ChangeWindowMessageFilter(WM_COPYDATA, MSGFLT_ADD);
|
||
ChangeWindowMessageFilter(0x0049, MSGFLT_ADD);
|
||
|
||
// Caffe<66>̃G<CC83><47><EFBFBD>[<5B>łȂ<C582><C882><EFBFBD><EFBFBD>O<EFBFBD><4F><EFBFBD>ۑ<EFBFBD><DB91><EFBFBD><EFBFBD>Ȃ<EFBFBD><C882>悤<EFBFBD>ɂ<EFBFBD><C982><EFBFBD>
|
||
google::SetLogDestination(google::INFO, "");
|
||
google::SetLogDestination(google::WARNING, "");
|
||
|
||
// Caffe<66>̃G<CC83><47><EFBFBD>[<5B><><EFBFBD>O<EFBFBD><4F><EFBFBD>uerror_log_<67>`<60>v<EFBFBD>ɏo<C98F><6F>
|
||
google::SetLogDestination(google::ERROR, "error_log_");
|
||
google::SetLogDestination(google::FATAL, "error_log_");
|
||
|
||
// CDialog<6F>N<EFBFBD><4E><EFBFBD>X<EFBFBD>Ń_<C583>C<EFBFBD>A<EFBFBD><41><EFBFBD>O<EFBFBD><4F><EFBFBD>쐬<EFBFBD><EC90AC><EFBFBD><EFBFBD>
|
||
CDialog cDialog;
|
||
CDialog cDialog2;
|
||
// IDC_EDIT<49>̃T<CC83>u<EFBFBD>N<EFBFBD><4E><EFBFBD>X
|
||
CControl cControlInput(IDC_EDIT_INPUT);
|
||
CControl cControlOutput(IDC_EDIT_OUTPUT);
|
||
CControl cControlScale(IDC_EDIT_SCALE_RATIO);
|
||
CControl cControlOutExt(IDC_EDIT_OUT_EXT);
|
||
|
||
// <20>o<EFBFBD>^<5E><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><D690><EFBFBD><EFBFBD>܂Ƃ߂<C682><DF82>ꂽ<EFBFBD>N<EFBFBD><4E><EFBFBD>X
|
||
// <20>O<EFBFBD><4F><EFBFBD>[<5B>o<EFBFBD><6F><EFBFBD><EFBFBD><D690><EFBFBD><EFBFBD>g<EFBFBD><67><EFBFBD>N<CE83><4E><EFBFBD>X<EFBFBD>ɂ܂Ƃ߂<C682><DF82>K<EFBFBD>v<EFBFBD>͂Ȃ<CD82><C882><EFBFBD><EFBFBD><EFBFBD><EFBFBD>̕<EFBFBD><CC95>@<40><><EFBFBD>𗧂<F097A782><C282>Ƃ<EFBFBD><C682><EFBFBD><EFBFBD><EFBFBD><EFBFBD>͂<EFBFBD>
|
||
DialogEvent cDialogEvent;
|
||
|
||
// <20>N<EFBFBD><4E><EFBFBD>X<EFBFBD>̊<CC8A><D690><EFBFBD><EFBFBD>o<EFBFBD>^<5E><><EFBFBD><EFBFBD><EFBFBD>ꍇ
|
||
|
||
// IDC_EDIT<49><54>WM_DROPFILES<45><53><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ă<EFBFBD><C482><EFBFBD><EFBFBD>Ƃ<EFBFBD><C682>Ɏ<EFBFBD><C98E>s<EFBFBD><73><EFBFBD><EFBFBD><EFBFBD><EFBFBD><D690>̓o<CC93>^
|
||
cControlInput.SetEventCallBack(SetClassCustomFunc(DialogEvent::DropInput, &cDialogEvent), NULL, WM_DROPFILES);
|
||
cControlOutput.SetEventCallBack(SetClassCustomFunc(DialogEvent::DropOutput, &cDialogEvent), NULL, WM_DROPFILES);
|
||
cControlScale.SetEventCallBack(SetClassCustomFunc(DialogEvent::TextInput, &cDialogEvent), NULL, WM_CHAR);
|
||
cControlOutExt.SetEventCallBack(SetClassCustomFunc(DialogEvent::TextInput, &cDialogEvent), NULL, WM_CHAR);
|
||
|
||
// <20>R<EFBFBD><52><EFBFBD>g<EFBFBD><67><EFBFBD>[<5B><><EFBFBD>̃T<CC83>u<EFBFBD>N<EFBFBD><4E><EFBFBD>X<EFBFBD><58><EFBFBD>o<EFBFBD>^
|
||
cDialog.AddControl(&cControlInput);
|
||
cDialog.AddControl(&cControlOutput);
|
||
cDialog.AddControl(&cControlScale);
|
||
cDialog.AddControl(&cControlOutExt);
|
||
|
||
// <20>e<EFBFBD>R<EFBFBD><52><EFBFBD>g<EFBFBD><67><EFBFBD>[<5B><><EFBFBD>̃C<CC83>x<EFBFBD><78><EFBFBD>g<EFBFBD>Ŏ<EFBFBD><C58E>s<EFBFBD><73><EFBFBD><EFBFBD><EFBFBD><EFBFBD><D690>̓o<CC93>^
|
||
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::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_NOISE);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_SCALE);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_NOISE_SCALE);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_AUTO_SCALE);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIONOISE_LEVEL1);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIONOISE_LEVEL2);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_CPU);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODE_GPU);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODEL_RGB);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODEL_PHOTO);
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_RADIO_MODEL_Y);
|
||
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::RadioButtom, &cDialogEvent), NULL, IDC_CHECK_TTA);
|
||
|
||
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::CheckCUDNN, &cDialogEvent), NULL, IDC_BUTTON_CHECK_CUDNN);
|
||
|
||
// <20>_<EFBFBD>C<EFBFBD>A<EFBFBD><41><EFBFBD>O<EFBFBD>̃C<CC83>x<EFBFBD><78><EFBFBD>g<EFBFBD>Ŏ<EFBFBD><C58E>s<EFBFBD><73><EFBFBD><EFBFBD><EFBFBD><EFBFBD><D690>̓o<CC93>^
|
||
cDialog.SetEventCallBack(SetClassFunc(DialogEvent::Create, &cDialogEvent), NULL, WM_INITDIALOG);
|
||
cDialog.SetEventCallBack(SetClassFunc(DialogEvent::OnDialogEnd, &cDialogEvent), NULL, WM_CLOSE);
|
||
cDialog.SetEventCallBack(SetClassFunc(DialogEvent::OnFaildCreateDir, &cDialogEvent), NULL, WM_FAILD_CREATE_DIR);
|
||
cDialog.SetEventCallBack(SetClassFunc(DialogEvent::OnWaifu2xError, &cDialogEvent), NULL, WM_ON_WAIFU2X_ERROR);
|
||
cDialog.SetEventCallBack(SetClassFunc(DialogEvent::WaitThreadExit, &cDialogEvent), NULL, WM_END_THREAD);
|
||
|
||
// <20>_<EFBFBD>C<EFBFBD>A<EFBFBD><41><EFBFBD>O<EFBFBD><4F><EFBFBD>\<5C><>
|
||
cDialog.DoModal(hInstance, IDD_DIALOG);
|
||
|
||
Waifu2x::quit_liblary();
|
||
|
||
return 0;
|
||
}
|