summaryrefslogtreecommitdiffstats
path: root/kooka/ocrresedit.cpp
blob: a289a0798d1795a85ee2f70bc0485532ef316821 (plain)
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
/***************************************************************************
                   ocrresedit.cpp - ocr result editor widget
                             -------------------
    begin                : Tue 12 Feb 2003
    copyright            : (C) 2003 by Klaas Freitag
    email                : freitag@suse.de
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *  This file may be distributed and/or modified under the terms of the    *
 *  GNU General Public License version 2 as published by the Free Software *
 *  Foundation and appearing in the file COPYING included in the           *
 *  packaging of this file.                                                *
 *
 *  As a special exception, permission is given to link this program       *
 *  with any version of the KADMOS ocr/icr engine of reRecognition GmbH,   *
 *  Kreuzlingen and distribute the resulting executable without            *
 *  including the source code for KADMOS in the source distribution.       *
 *
 *  As a special exception, permission is given to link this program       *
 *  with any edition of Qt, and distribute the resulting executable,       *
 *  without including the source code for Qt in the source distribution.   *
 *                                                                         *
 ***************************************************************************/
#include <qcolor.h>

#include "ocrresedit.h"
#include "ocrword.h"
#include <kdebug.h>
#include <kfiledialog.h>
#include <klocale.h>

#include <qfile.h>
#include <qtextstream.h>

/* -------------------- ocrResEdit -------------------- */

ocrResEdit::ocrResEdit( QWidget *parent )
    : QTextEdit(parent)
{
    m_updateColor.setNamedColor( "SeaGreen");
    m_ignoreColor.setNamedColor( "CadetBlue4" );
    m_wrnColor.setNamedColor( "firebrick2" );
}


void ocrResEdit::slMarkWordWrong( int line, const ocrWord& word )
{
    // m_textEdit->setSelection( line,
    slReplaceWord( line, word, word, m_wrnColor );
}


void ocrResEdit::slUpdateOCRResult( int line, const QString& wordFrom,
                                   const QString& wordTo )
{
    /* the index is quite useless here, since  the text could have had been
     * changed by corrections before. Thus better search the word and update
     * it.
     */
    slReplaceWord( line, wordFrom, wordTo, m_updateColor );

}


void ocrResEdit::slIgnoreWrongWord( int line, const ocrWord& word )
{
    slReplaceWord( line, word, word, m_ignoreColor );
}


void ocrResEdit::slSelectWord( int line, const ocrWord& word )
{
   if( line < paragraphs() )
   {
      QString editLine = text(line);
      int cnt = editLine.contains( word);

      if( cnt > 0 )
      {
	 int pos = editLine.find(word);
	 setCursorPosition( line, pos );
	 setSelection( line, pos, line, pos + word.length());
      }
   }
}

void ocrResEdit::slReplaceWord( int line, const QString& wordFrom,
                               const QString& wordTo, const QColor& color )
{
    kdDebug(28000) << "Updating word " << wordFrom << " in line " << line << endl;

    bool isRO = isReadOnly();

    if( line < paragraphs() )
    {
        QString editLine = text(line);
        int cnt = editLine.contains( wordFrom );

        if( cnt > 0 )
        {
            int pos = editLine.find(wordFrom);
            setSelection( line, pos, line, pos+wordFrom.length());

            QColor saveCol = this->color();
            setColor( color );
            if( isRO ) {
                setReadOnly(false);
            }
            insert( wordTo, unsigned (4) );
            if( isRO ) {
                setReadOnly( true );
            }
            setColor(saveCol);
        }
        else
        {
            kdDebug(28000) << "WRN: Paragraph does not contain word " << wordFrom << endl;
        }

    }
    else
    {
        kdDebug(28000) << "WRN: editor does not have line " << line << endl;
    }
}


void ocrResEdit::slSaveText()
{
    QString fileName = KFileDialog::getSaveFileName( (QDir::home()).path(),
                                                 "*.txt",
                                                 this,
                                                 i18n("Save OCR Result Text") );
    if( fileName.isEmpty() )
      return;
    QFile file( fileName );
    if ( file.open( IO_WriteOnly ) )
    {
        QTextStream stream( &file );
        stream << text();
        file.close();
    }
}

#include "ocrresedit.moc"
/*   */