Source/dicts/StructDictLoader/Main.cpp - master - bear/aot

Source/dicts/StructDictLoader/Main.cpp
89 строк2 KB
sokirko74
distribute over folders
01 дек 2021, 21:21
01 дек 2021, 21:219bdfb1a
#include "morph_dict/common/utilit.h"
#include "dicts/StructDictLib/Ross.h"
#include "dicts/StructDictLib/TempArticle.h"

void PrintUsage()
{
	std::cout << "StructDict Loader\n";
	std::cout << "This program converts text version of structural dictionary to binary\n";
	std::cout << "Usage: StructDictLoader <ToTxt|FromTxt> <Inputfile> <OutputCatalog>\n";
	std::cout << "Example: StructDictLoader ToTxt ross.txt c:/rml/dicts/ross \n";
	exit(-1);
};

void export_dict(std::string fileName, std::string folder) {
	CDictionary Dict;
	if (!Dict.Load(folder.c_str()) || !Dict.ReadUnitComments())
	{
		throw CExpc("Cannot load dictionary from %s", folder.c_str());
	};
	std::ofstream outf (fileName, std::ios::binary);
	if (!outf.is_open())
	{
		throw CExpc("Cannot write to %s", fileName.c_str());
	};

	CTempArticle A;
	A.m_pRoss = &Dict;

	for (uint16_t i = 0; i < Dict.m_Units.size(); i++)
	{
		A.ReadFromDictionary(i, true, true);
		outf << "============\n" << Dict.GetUnitTextHeader(i);
		outf << A.GetArticleStrUtf8(true);
	};
}

void import_dict(std::string fileName, std::string folder) {
	CDictionary Dict;
	if (!FileExists(fileName.c_str()))
	{
		throw CExpc("Cannot read %s", fileName.c_str());
	};

	if (!Dict.LoadOnlyConstants(folder.c_str()))
	{
		throw CExpc("Cannot load an empty dictionary from %s", folder.c_str());
	};
	Dict.m_bShouldSaveComments = true;

	std::string Messages;
	if (!Dict.ImportFromText(fileName, 1, Messages)) {
		throw CExpc(Messages);
	}
	Dict.Save();
}

int main(int argc, char** argv)
{
	if (argc != 4) {
		PrintUsage();
	}

	std::string Action = argv[1];
	if ((Action != "ToTxt") && (Action != "FromTxt")) {
		PrintUsage();
	}

	try {
		if (Action == "FromTxt")
		{
			import_dict(argv[2], argv[3]);
		}
		else
		{
			export_dict(argv[2], argv[3]);
		};
	}
	catch (CExpc c)
	{
		std::cerr << c.m_strCause << std::endl;
	}
	catch (const std::exception& ex)
	{
		std::cerr << ex.what() << std::endl;
	}



}