Guillermo
10 years ago
3 changed files with 131 additions and 1 deletions
@ -0,0 +1,130 @@
@@ -0,0 +1,130 @@
|
||||
var Tesseract = (function(){ |
||||
|
||||
var Tesseract = {} |
||||
|
||||
//https://cdn.rawgit.com/naptha/tesseract.js/master/worker/worker.js
|
||||
//https://rawgit.com/naptha/tesseract.js/master/worker/worker.js for testing
|
||||
//https://cdn.rawgit.com/naptha/tesseract.js/master/worker/worker.js
|
||||
|
||||
var blob = new Blob(["importScripts('http://localhost:1234/master/lib/worker.2015.07.26.js');"]); // changed on build
|
||||
// console.log('localhost')
|
||||
var worker = new Worker(window.URL.createObjectURL(blob)); |
||||
worker.postMessage({init: {mem: 16777216*6}}) |
||||
var bigworker = false |
||||
|
||||
var index = 0 |
||||
var handlers = [] |
||||
|
||||
worker.onmessage = function(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 (!bigworker && ['chi_sim', 'chi_tra', 'jpn'].indexOf(lang) != -1){ |
||||
worker.postMessage({init: {mem: 16777216*10}}) |
||||
bigworker = true |
||||
console.log('started big worker') |
||||
} |
||||
|
||||
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 |
||||
})() |
Loading…
Reference in new issue