You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
2.7 KiB
119 lines
2.7 KiB
var Tesseract = (function(){
|
|
|
|
var Tesseract = {}
|
|
|
|
var blob = new Blob(["importScripts('http://localhost:1234/master/worker/worker.js');"]);
|
|
console.log('localhost')
|
|
var worker = new Worker(window.URL.createObjectURL(blob));
|
|
|
|
var index = 0
|
|
var handlers = []
|
|
|
|
worker.onmessage = function(e){
|
|
// console.log(handlers, e)
|
|
var handler = handlers[e.data.index]
|
|
if(e.data.progress){
|
|
handler.progress(e.data.progress)
|
|
}
|
|
else if(e.data.err){
|
|
handler.reject(e.data.err)
|
|
handler.callback(e.data.err)
|
|
}
|
|
else {
|
|
handler.resolve(e.data.result)
|
|
handler.callback(null,e.data.result)
|
|
}
|
|
}
|
|
|
|
function convertToImageData(image){
|
|
if(image.getContext){
|
|
image = image.getContext('2d');
|
|
}else if(image.tagName == "IMG" || image.tagName == "VIDEO"){
|
|
var c = document.createElement('canvas');
|
|
if(image.tagName == "IMG"){
|
|
c.width = image.naturalWidth;
|
|
c.height = image.naturalHeight;
|
|
}else if(image.tagName == "VIDEO"){
|
|
c.width = image.videoWidth;
|
|
c.height = image.videoHeight;
|
|
}
|
|
var ctx = c.getContext('2d');
|
|
ctx.drawImage(image, 0, 0);
|
|
image = ctx;
|
|
}
|
|
if(image.getImageData) image = image.getImageData(0, 0, image.canvas.width, image.canvas.height);
|
|
return image
|
|
}
|
|
|
|
Tesseract.detect = function(image, progress, callback){
|
|
image = convertToImageData(image)
|
|
|
|
if(typeof progress === "undefined"){
|
|
progress = callback = new Function()
|
|
}
|
|
|
|
if (typeof callback === "undefined"){
|
|
callback = progress
|
|
progress = new Function()
|
|
}
|
|
|
|
var i = index++
|
|
|
|
handlers[i] = {
|
|
resolve: new Function(),
|
|
reject: new Function()
|
|
}
|
|
handlers[i].callback = callback
|
|
handlers[i].progress = progress
|
|
|
|
return new Promise(function(resolve, reject){
|
|
handlers[i].resolve = resolve
|
|
handlers[i].reject = reject
|
|
worker.postMessage({index: i, fun: 'detect', image: image})
|
|
})
|
|
|
|
}
|
|
|
|
Tesseract.recognize = function(image, options, callback){
|
|
var lang = options.lang
|
|
if(typeof lang === "undefined"){
|
|
lang = 'eng'
|
|
}
|
|
|
|
if (typeof options === 'string') {
|
|
lang = options
|
|
options = {}
|
|
}
|
|
|
|
if (typeof options === "function") {
|
|
callback = options
|
|
options = {}
|
|
}
|
|
|
|
image = convertToImageData(image)
|
|
|
|
var i = index++
|
|
|
|
handlers[i] = {
|
|
resolve: new Function(),
|
|
reject: new Function()
|
|
}
|
|
handlers[i].callback = callback || new Function()
|
|
handlers[i].progress = (function(){
|
|
if(typeof options.progress === 'function'){
|
|
var p = options.progress
|
|
delete options.progress
|
|
return p
|
|
}
|
|
return function(){}
|
|
})()
|
|
|
|
return new Promise(function(resolve, reject){
|
|
handlers[i].resolve = resolve
|
|
handlers[i].reject = reject
|
|
worker.postMessage({index: i, fun: 'recognize', image: image, lang: lang, options: options})
|
|
})
|
|
|
|
}
|
|
return Tesseract
|
|
})() |