forked from otiai10/gosseract
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tesseract0303.go
60 lines (54 loc) · 1.16 KB
/
tesseract0303.go
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
package gosseract
import "fmt"
import "os"
import "os/exec"
import "bytes"
import "io/ioutil"
type tesseract0303 struct {
version string
resultFilePath string
commandPath string
}
func (t tesseract0303) Version() string {
return t.version
}
func (t tesseract0303) Execute(params []string) (res string, e error) {
// command args
var args []string
// Register source file
args = append(args, params[0])
// generate result file path
t.resultFilePath, e = generateTmpFile()
if e != nil {
return
}
// Register result file
args = append(args, t.resultFilePath)
// Register digest file
if len(params) > 1 {
args = append(args, params[1])
}
// prepare command
cmd := exec.Command(TESSERACT, args...)
// execute
var stderr bytes.Buffer
cmd.Stderr = &stderr
if e = cmd.Run(); e != nil {
e = fmt.Errorf(stderr.String())
return
}
// read result
res, e = t.readResult()
return
}
func (t tesseract0303) readResult() (res string, e error) {
fpath := t.resultFilePath + outFILEEXTENSION
file, e := os.OpenFile(fpath, 1, 1)
if e != nil {
return
}
buffer, _ := ioutil.ReadFile(file.Name())
res = string(buffer)
os.Remove(file.Name())
return
}