Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2026-01-04 07:48:49 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

🎨 桌面端支持搜索图片 OCR 文本 https://github.com/siyuan-note/siyuan/issues/3470

Browse source
This commit is contained in:
Liang Ding 2023-01-16 22:49:34 +08:00
parent 3d20b5c834
commit ca6e82a549
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
1 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

24
kernel/util/tesseract.go
View file

@ -38,7 +38,7 @@ var (
AssetsTextsLock = sync.Mutex{}
AssetsTextsChanged = false
tesseractLangs []string
TesseractLangs []string
)
func GetAssetText(asset string) string {
@ -75,7 +75,7 @@ func Tesseract(imgAbsPath string) string {
defer cancel()
now := time.Now()
cmd := exec.CommandContext(ctx, "tesseract", "-c", "debug_file=/dev/null", imgAbsPath, "stdout", "-l", strings.Join(tesseractLangs, "+"))
cmd := exec.CommandContext(ctx, "tesseract", "-c", "debug_file=/dev/null", imgAbsPath, "stdout", "-l", strings.Join(TesseractLangs, "+"))
gulu.CmdAttr(cmd)
output, err := cmd.CombinedOutput()
if ctx.Err() == context.DeadlineExceeded {
@ -106,13 +106,27 @@ func initTesseract() {
return
}
tesseractLangs = getTesseractLangs()
if 1 > len(tesseractLangs) {
langs := getTesseractLangs()
if 1 > len(langs) {
logging.LogWarnf("no tesseract langs found")
TesseractEnabled = false
return
}
logging.LogInfof("tesseract-ocr enabled [ver=%s, langs=%s]", ver, strings.Join(tesseractLangs, "+"))
if !gulu.Str.Contains("eng", langs) {
logging.LogWarnf("no eng tesseract lang found")
return
}
if !gulu.Str.Contains("chi_sim", langs) {
logging.LogWarnf("no chi_sim tesseract lang found")
return
}
for _, lang := range langs {
if "eng" == lang || "chi_sim" == lang {
TesseractLangs = append(TesseractLangs, lang)
}
}
logging.LogInfof("tesseract-ocr enabled [ver=%s, langs=%s]", ver, strings.Join(TesseractLangs, "+"))
}
func getTesseractVer() (ret string) {