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

GUIでフォルダの場合にファイル名の末尾の自動設定機能が動いていなかったのを修正

Browse Source
This commit is contained in:
lltcggie 2015-11-19 02:46:23 +09:00
parent 4b21788fcd
commit 08555f0864
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -493,7 +493,13 @@ private:
SyncMember(true);
const boost::filesystem::path output_path(output_str);
std::string stem = output_path.stem().string();
std::string stem;
if (!boost::filesystem::is_directory(input_str))
stem = output_path.stem().string();
else
stem = output_path.filename().string();
if (stem.length() > 0 && stem.length() >= autoSetAddName.length())
{
const std::string base = stem.substr(0, stem.length() - autoSetAddName.length());
@ -503,7 +509,11 @@ private:
const std::string addstr(AddName());
autoSetAddName = addstr;
boost::filesystem::path new_out_path = output_path.branch_path() / (base + addstr + outputExt);
boost::filesystem::path new_out_path;
if (!boost::filesystem::is_directory(input_str))
new_out_path = output_path.branch_path() / (base + addstr + outputExt);
else
new_out_path = output_path.branch_path() / (base + addstr);
SetWindowTextA(GetDlgItem(dh, IDC_EDIT_OUTPUT), new_out_path.string().c_str());
}