b38dea0597
Share Gaia assets and adapters, keep private profiles and development documents local, and update both public READMEs.
122 lines
2.6 KiB
JavaScript
Executable file
122 lines
2.6 KiB
JavaScript
Executable file
'use strict';
|
|
|
|
var MAX_STROKE_NUM = 1000;
|
|
|
|
self.onmessage = function(e) {
|
|
var data = e.data;
|
|
|
|
switch(data.id) {
|
|
case 'setLang':
|
|
var ret = Recognition.setLang(data.param);
|
|
post('setLang', ret);
|
|
break;
|
|
case 'recognize':
|
|
var words = Recognition.recognize(JSON.parse(data.param));
|
|
post('recognize', JSON.stringify(words));
|
|
break;
|
|
case 'getPrediction':
|
|
var words = Recognition.getPrediction(data.param);
|
|
post('getPrediction', words);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
};
|
|
|
|
var Recognition = {
|
|
_inited: false,
|
|
|
|
setLang: function(lang) {
|
|
if (!this._lang) {
|
|
this._lang = lang;
|
|
return this._init(lang);
|
|
}
|
|
if (this._lang != lang) {
|
|
this._lang = lang;
|
|
this._uinit();
|
|
return this._init(lang);
|
|
}
|
|
return true;
|
|
},
|
|
|
|
recognize: function(points) {
|
|
if (!this._inited) {
|
|
return [];
|
|
}
|
|
|
|
var strokes = points.slice(0, 2 * MAX_STROKE_NUM);
|
|
// Append [-1, -1] to strokes.
|
|
strokes.push.apply(strokes, [-1, -1]);
|
|
this._strokeBuffer.set(strokes);
|
|
return this._CharacterRecognize(this._strokeBuffer.byteOffset);
|
|
},
|
|
|
|
getPrediction: function(words) {
|
|
if (this._getPrediction) {
|
|
return this._getPrediction(words);
|
|
}
|
|
},
|
|
|
|
_init: function(lang) {
|
|
this._inited = false;
|
|
var model = '';
|
|
|
|
if (lang == 'zh-Hans') {
|
|
model = 'zh_Hans.model';
|
|
}
|
|
if (lang == 'zh-Hant') {
|
|
model = 'zh_Hant.model';
|
|
}
|
|
|
|
if (!Module.ccall('InitRecog',
|
|
'number',
|
|
['string', 'number', 'number'],
|
|
[model, 140, 140])) {
|
|
return false;
|
|
}
|
|
|
|
this._CharacterRecognize =
|
|
Module.cwrap('CharacterRecognize', 'string', ['number']);
|
|
|
|
if (Module.ccall('InitPrediction',
|
|
'number',
|
|
['string'],
|
|
['dict_pinyin.data'])) {
|
|
this._getPrediction =
|
|
Module.cwrap('GetPrediction', 'string', ['string']);
|
|
}
|
|
|
|
var buf = Module._malloc((MAX_STROKE_NUM + 1) * 4);
|
|
this._strokeBuffer =
|
|
new Int16Array(Module.HEAPU8.buffer, buf, (MAX_STROKE_NUM + 1) * 2);
|
|
|
|
this._inited = true;
|
|
return true;
|
|
},
|
|
|
|
_uinit: function() {
|
|
Module._free(this._strokeBuffer.byteOffset);
|
|
Module.ccall('ExitRecog', '', [], []);
|
|
},
|
|
|
|
_strokeBuffer: null,
|
|
_CharacterRecognize: null,
|
|
_lang: '',
|
|
_getPrediction: null
|
|
};
|
|
|
|
var Module = {
|
|
filePackagePrefixURL: './',
|
|
canvas: {},
|
|
stdout: null,
|
|
_main: function() {
|
|
post('init', true);
|
|
}
|
|
};
|
|
|
|
function post(id, value) {
|
|
postMessage({
|
|
id: id,
|
|
value: value
|
|
});
|
|
}
|