File: ocr.pm

package info (click to toggle)
os-autoinst 4.5.1527308405.8b586d5-4.1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid
  • size: 22,688 kB
  • sloc: perl: 10,424; cpp: 1,527; python: 217; makefile: 211; sh: 71; xml: 11
file content (81 lines) | stat: -rw-r--r-- 2,347 bytes parent folder | download
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
# Copyright © 2009-2013 Bernhard M. Wiedemann
# Copyright © 2012-2015 SUSE LLC
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, see <http://www.gnu.org/licenses/>.

package ocr;
use strict;
use warnings;
require IPC::System::Simple;
use autodie ':all';

our $gocrbin = "/usr/bin/gocr";
if (!-x $gocrbin) { $gocrbin = undef }

# input: image ref
# TODO get_ocr is nowhere called, also not in opensuse tests or openQA, delete?
sub get_ocr {
    my $ppm        = shift;
    my $gocrparams = shift || "";
    my @ocrrect    = @{$_[0]};
    if (!$gocrbin || !@ocrrect) { return "" }
    if (@ocrrect != 4) { return " ocr: bad rect" }
    return unless $ppm;
    my $ppm2 = $ppm->copyrect(@ocrrect);
    if (!$ppm2) { return "" }
    my $tempname = "ocr.$$-" . time . rand(10000) . ".ppm";
    $ppm2->write($tempname) or return " ocr error writing $tempname";

    # init DB file:
    if (!-e "db/db.lst") {
        mkdir "db";
        open(my $fd, ">", "db/db.lst");
        close $fd;
    }

    open(my $pipe, '-|', "$gocrbin -l 128 -d 0 $gocrparams $tempname");
    local $/;
    my $ocr = <$pipe>;
    close($pipe);
    unlink $tempname;
    return $ocr;
}

# input: image ref, area
# FIXME: pass options
# FIXME: write C library bindings instead of system()
sub tesseract {
    my ($img, $area) = @_;
    my $imgfn = 'ocr.png';
    my $txtfn = 'ocr';       # tesseract appends .txt automatically o_O
    my $txt;

    if ($area) {
        $img = $img->copyrect($area->{xpos}, $area->{ypos}, $area->{width}, $area->{height});
    }

    $img->write($imgfn);
    system('tesseract', $imgfn, $txtfn);
    $txtfn .= '.txt';
    open(my $fh, '<:encoding(UTF-8)', $txtfn);
    local $/;
    $txt = <$fh>;
    close $fh;
    unlink $imgfn;
    unlink $txtfn;
    return $txt;
}

1;
# vim: set sw=4 et: