aboutsummaryrefslogtreecommitdiffstats
path: root/src/modules/language/libkvilanguage.cpp
blob: 6d5ea426dac5cdf5f5c319685812d13aa0eae782 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
//=============================================================================
//
//   File : libkvilanguage.cpp
//   Creation date : Fri 25 Dec 20:16:49 2004 GMT by Szymon Stefanek
//
//   This file is part of the KVIrc IRC client distribution
//   Copyright (C) 2004-2010 Szymon Stefanek (pragma at kvirc dot net)
//
//   This program is FREE software. You can redistribute it and/or
//   modify it under the terms of the GNU General Public License
//   as published by the Free Software Foundation; either version 2
//   of the License, or (at your option) any later version.
//
//   This program is distributed in the HOPE that it will be USEFUL,
//   but WITHOUT ANY WARRANTY; without even the implied warranty of
//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
//   See the GNU General Public License for more details.
//
//   You should have received a copy of the GNU General Public License
//   along with this program. If not, write to the Free Software Foundation,
//   Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
//
//=============================================================================

#include "KviModule.h"
#include "KviApplication.h"
#include "KviLocale.h"

#include <QFileInfo>

#include "detector.h"

/*
	@doc: language.detect
	@type:
		function
	@title:
		$language.detect
	@short:
		Tries to detect the language and encoding of a string
	@syntax:
		<hash> language.detect(<text: string>)
	@description:
		Tries to detect the language and encoding of the string specified
		as [text]. The accuracy of the detection mainly depends on the length
		of the supplied text. Good results can be achieved with some thousands
		characters.[br]
		This command is exported by the [i]language[/i] module.
	@examples:
		[example]
			%dati = $language.detect("I'm a lord and I speak perfect English.");
			if(%dati{"error"} != "")echo "Language detection failed: " %dati{"error"};
				%count = %dati{"matchcount"};
				for(%i=0;%i<%count;%i++)
			{
				echo "LANGUAGE " %i " : " %dati{"matches"}[%i]{"language"};
				echo "ENCODING " %i " : " %dati{"matches"}[%i]{"encoding"};
				echo "SCORE " %i " : "%dati{"matches"}[%i]{"score"};
			}
		echo "ACCURACY: " %dati{"accuracy"};
		[/example]
*/

static bool language_kvs_cmd_detect(KviKvsModuleFunctionCall * c)
{
	QString text, error;
	KviKvsArray * matches;
	KviKvsHash *match, *ret;
	LanguageAndEncodingResult r;
	int matchcount = DLE_NUM_BEST_MATCHES;

	KVSM_PARAMETERS_BEGIN(c)
	KVSM_PARAMETER("text", KVS_PT_STRING, KVS_PF_OPTIONAL, text)
	KVSM_PARAMETERS_END(c)

	//on error we return an empty array
	matches = new KviKvsArray();

	if(text.isEmpty())
	{
		error = "err_notext";
		matchcount = 0;
	}
	else
	{
		detect_language_and_encoding(text.toUtf8().data(), &r, 0);

		for(int i = 0; i < matchcount; i++)
		{
			match = new KviKvsHash();
			match->set("language", new KviKvsVariant(QString(r.match[i].szLanguage)));
			match->set("encoding", new KviKvsVariant(r.match[i].szEncoding));
			match->set("score", new KviKvsVariant((kvs_real_t)r.match[i].dScore));
			matches->set(i, new KviKvsVariant(match));
		}
	}

	ret = new KviKvsHash();
	ret->set("matches", new KviKvsVariant(matches));
	ret->set("matchcount", new KviKvsVariant((kvs_int_t)matchcount));
	ret->set("error", new KviKvsVariant(error));
	ret->set("accuracy", new KviKvsVariant((kvs_real_t)r.dAccuracy));

	c->returnValue()->setHash(ret);

	return true;
}

static bool language_module_init(KviModule * m)
{
	KVSM_REGISTER_FUNCTION(m, "detect", language_kvs_cmd_detect);
	return true;
}

static bool language_module_cleanup(KviModule *)
{
	return true;
}

KVIRC_MODULE(
    "Language",                                                     // module name
    "4.0.0",                                                        // module version
    "Copyright (C) 2004 Szymon Stefanek (pragma at kvirc dot net)", // author & (C)
    "A simple statistical language/encoding detector",
    language_module_init,
    0,
    0,
    language_module_cleanup,
    0)