GUIでコミットし忘れていたファイルをコミット

waifu2x-caffe-gui/LangStringList.h

@@ -0,0 +1,248 @@
+#pragma once
+
+#define WIN32_LEAN_AND_MEAN
+#include <Windows.h>
+#include <vector>
+#include <string>
+#include <boost/tokenizer.hpp>
+#include <unordered_map>
+#include <rapidjson/document.h>
+#include <stdio.h>
+#include "tstring.h"
+
+
+class LangStringList
+{
+public:
+	struct stLangSetting
+	{
+		std::wstring LangName;
+		WORD LangID;
+		WORD SubLangID;
+		std::wstring FileName;
+
+		stLangSetting() : LangID(0), SubLangID(0)
+		{}
+	};
+
+private:
+	tstring LangBaseDir;
+	std::vector<stLangSetting> LangList;
+	stLangSetting NowLang;
+	LANGID NowLangID = GetUserDefaultUILanguage();
+	std::unordered_map<std::wstring, std::wstring> LangStringMap;
+	std::unordered_map<std::wstring, std::wstring> DefaultLangStringMap; // 現在の言語になかった文字列のフォールバック用。リストの一番最初に書かれている言語がデフォルト
+
+private:
+	static std::wstring Utf8ToUtf16(const char *src, int src_size = -1)
+	{
+		int ret;
+
+		ret = MultiByteToWideChar(CP_UTF8, 0, src, src_size, NULL, 0);
+		if (ret == 0)
+			return std::wstring();
+
+		std::vector<wchar_t> buf(ret);
+
+		ret = MultiByteToWideChar(CP_UTF8, 0, src, src_size, buf.data(), ret);
+		if (ret == 0)
+			return std::wstring();
+
+		if (buf.back() != L'\0')
+			buf.push_back(L'\0');
+
+		return std::wstring(buf.data());
+	}
+
+	static int getNum(const std::string &str)
+	{
+		if (str.length() >= 3 && str.substr(0, 2) == "0x")
+			return strtol(str.c_str(), NULL, 16);
+		else
+			return strtol(str.c_str(), NULL, 10);
+	}
+
+	const stLangSetting& GetLang(const LANGID id) const
+	{
+		const auto Primarylang = PRIMARYLANGID(id);
+		const auto Sublang = SUBLANGID(id);
+
+		int FindPrimary = -1;
+		int FindSub = -1;
+
+		for (size_t i = 0; i < LangList.size(); i++)
+		{
+			const auto &l = LangList[i];
+
+			if (Primarylang == l.LangID)
+			{
+				FindPrimary = (int)i;
+				if (Primarylang == l.SubLangID)
+				{
+					FindSub = (int)i;
+					break;
+				}
+			}
+		}
+
+		if (FindPrimary >= 0 && FindSub >= 0) // 現在の言語にピッタリ合うやつが見つかった
+			return LangList[FindSub];
+		else if (FindPrimary >= 0) // 現在の言語に属するものが見つかった
+			return LangList[FindPrimary];
+
+		// 見つからなかったから一番最初に書かれているやつにする
+		if (LangList.size() > 0)
+			return LangList[0];
+
+		return stLangSetting();
+	}
+
+	void ReadLangFile(const stLangSetting &lang, std::unordered_map<std::wstring, std::wstring> &langStringMap) const
+	{
+		langStringMap.clear();
+
+		if (lang.FileName.length() == 0)
+			return;
+
+		const auto LangFilePath = LangBaseDir + lang.FileName;
+
+		rapidjson::Document d;
+		std::vector<char> jsonBuf;
+
+		FILE *fp = nullptr;
+		try
+		{
+			fp = _wfopen(LangFilePath.c_str(), L"rb");
+			if (!fp)
+				return;
+
+			fseek(fp, 0, SEEK_END);
+			const auto size = ftell(fp);
+			fseek(fp, 0, SEEK_SET);
+
+			jsonBuf.resize(size + 1);
+			fread(jsonBuf.data(), 1, jsonBuf.size(), fp);
+
+			jsonBuf[jsonBuf.size() - 1] = '\0';
+
+			d.Parse(jsonBuf.data());
+
+			if (d.HasParseError())
+				throw 0;
+
+			for (auto it = d.MemberBegin(); it != d.MemberEnd(); ++it)
+			{
+				auto name = Utf8ToUtf16(it->name.GetString(), it->name.GetStringLength());
+				auto val = Utf8ToUtf16(it->value.GetString(), it->value.GetStringLength());
+
+				langStringMap.emplace(name, val);
+			}
+		}
+		catch (...)
+		{
+		}
+
+		if (fp)
+			fclose(fp);
+	}
+		
+
+public:
+	void SetLangBaseDir(const tstring &LangBaseDir)
+	{
+		this->LangBaseDir = LangBaseDir;
+		if (LangBaseDir.length() > 0 && (LangBaseDir.back() != TEXT('\\') && LangBaseDir.back() != TEXT('/')))
+			this->LangBaseDir += TEXT('/');
+	}
+
+	bool ReadLangList(const tstring &LangListPath)
+	{
+		std::ifstream ifs(LangListPath);
+		if (ifs.fail())
+			return false;
+
+		LangList.clear();
+
+		std::string str;
+		while (getline(ifs, str))
+		{
+			if (str.length() > 0 && str.front() == ';')
+				continue;
+
+			boost::char_separator<char> sep("\t");
+			boost::tokenizer<boost::char_separator<char>> tokens(str, sep);
+
+			std::vector<std::string> list;
+			for (const auto& t : tokens)
+				list.emplace_back(t);
+
+			if (list.size() != 4)
+				continue;
+
+			stLangSetting ls;
+			ls.LangName = Utf8ToUtf16(list[0].c_str(), list[0].length());
+			ls.LangID = getNum(list[1]);
+			ls.SubLangID = getNum(list[2]);
+			ls.FileName = Utf8ToUtf16(list[3].c_str(), list[3].length());
+
+			LangList.push_back(ls);
+		}
+
+		if (LangList.size() == 0)
+			return false;
+
+		if (NowLangID != 0)
+			SetLang(NowLangID);
+
+		// デフォルト言語を読みだす
+		ReadLangFile(LangList[0], DefaultLangStringMap);
+
+		return true;
+	}
+
+	void SetLang(const LANGID id)
+	{
+		NowLang = GetLang(id);
+		NowLangID = id;
+
+		ReadLangFile(NowLang, LangStringMap);
+	}
+
+	void SetLang(const stLangSetting &lang)
+	{
+		NowLang = lang;
+		NowLangID = MAKELANGID(lang.LangID, lang.SubLangID);
+
+		ReadLangFile(NowLang, LangStringMap);
+	}
+
+	void SetLang()
+	{
+		SetLang(GetUserDefaultUILanguage());
+	}
+
+	const stLangSetting& GetLang() const
+	{
+		return NowLang;
+	}
+
+	const std::vector<stLangSetting>& GetLangList() const
+	{
+		return LangList;
+	}
+
+	const std::wstring& GetString(const wchar_t *key) const
+	{
+		static std::wstring none;
+
+		auto it = LangStringMap.find(key);
+		if (it != LangStringMap.end())
+			return it->second;
+
+		it = DefaultLangStringMap.find(key);
+		if (it != DefaultLangStringMap.end())
+			return it->second;
+
+		return none;
+	}
+};

waifu2x-caffe-gui/tstring.h

@@ -0,0 +1,19 @@
+#pragma once
+
+#include <string>
+#include <boost/filesystem.hpp>
+
+
+#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