File: ocr.cpp

package info (click to toggle)
musescore-snapshot 3.2.s20190704%2Bdfsg1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 218,116 kB
  • sloc: cpp: 290,563; xml: 200,238; sh: 3,706; ansic: 1,447; python: 393; makefile: 222; perl: 82; pascal: 79
file content (79 lines) | stat: -rw-r--r-- 2,333 bytes parent folder | download | duplicates (4)
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
//=============================================================================
//  MusE Reader
//  Music Score Reader
//
//  Copyright (C) 2010 Werner Schweer
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License version 2.
//
//  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, write to the Free Software
//  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//=============================================================================

#include "ocr.h"

#include <tesseract/baseapi.h>
#include <locale.h>

namespace Ms {

//---------------------------------------------------------
//   Ocr
//---------------------------------------------------------

Ocr::Ocr()
      {
      tess = 0;
      }

//---------------------------------------------------------
//   init
//---------------------------------------------------------

void Ocr::init()
      {
      if (tess == 0)
            tess = new tesseract::TessBaseAPI;
      tess->Init("/usr/local/share/tessdata", 0, 0, 0, false);
      }

//---------------------------------------------------------
//   readLine
//---------------------------------------------------------

QString Ocr::readLine(const OcrImage& img)
      {
      int w = img.r.width();
      int h = img.r.height();

      int bw = (w + 7) / 8;
      uchar* d = new uchar[bw * h];
      memset(d, 0, bw * h);
      uchar* p = d;
      int yo = img.r.y();
      int xo = img.r.x();

      for (int y = 0; y < h; ++y) {
            for (int x = 0; x < bw; ++x) {
                  int mask = 0x80;
                  uchar dst = 0;
                  for (int xx = 0; xx < 8; ++xx) {
                        if (img.dot(x * 8 + xo + xx, y + yo))
                              dst |= mask;
                        mask >>= 1;
                        }
                  *p++ = ~dst;
                  }
            }
      char* txt = tess->TesseractRect(d, 0, bw, 0, 0, w, h);
      return QString(txt);
      }
}