LXR qemu/tests/avocado/tesseract

   1# ...
   2#
   3# Copyright (c) 2019 Philippe Mathieu-Daudé <f4bug@amsat.org>
   4#
   5# This work is licensed under the terms of the GNU GPL, version 2 or
   6# later. See the COPYING file in the top-level directory.
   7
   8import re
   9import logging
  10
  11from avocado.utils import process
  12from avocado.utils.path import find_command, CmdNotFoundError
  13
  14def tesseract_available(expected_version):
  15    try:
  16        find_command('tesseract')
  17    except CmdNotFoundError:
  18        return False
  19    res = process.run('tesseract --version')
  20    try:
  21        version = res.stdout_text.split()[1]
  22    except IndexError:
  23        version = res.stderr_text.split()[1]
  24    return int(version.split('.')[0]) == expected_version
  25
  26    match = re.match(r'tesseract\s(\d)', res)
  27    if match is None:
  28        return False
  29    # now this is guaranteed to be a digit
  30    return int(match.groups()[0]) == expected_version
  31
  32
  33def tesseract_ocr(image_path, tesseract_args='', tesseract_version=3):
  34    console_logger = logging.getLogger('tesseract')
  35    console_logger.debug(image_path)
  36    if tesseract_version == 4:
  37        tesseract_args += ' --oem 1'
  38    proc = process.run("tesseract {} {} stdout".format(tesseract_args,
  39                                                       image_path))
  40    lines = []
  41    for line in proc.stdout_text.split('\n'):
  42        sline = line.strip()
  43        if len(sline):
  44            console_logger.debug(sline)
  45            lines += [sline]
  46    return lines
  47