mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-12-19 20:08:16 +08:00
b60c6065e3
git-svn-id: https://tesseract-ocr.googlecode.com/svn/trunk@110 d0cd1f9f-072b-0410-8dd7-cf729c803f20
359 lines
12 KiB
C++
359 lines
12 KiB
C++
/**********************************************************************
|
|
* File: tessedit.cpp (Formerly tessedit.c)
|
|
* Description: Main program for merge of tess and editor.
|
|
* Author: Ray Smith
|
|
* Created: Tue Jan 07 15:21:46 GMT 1992
|
|
*
|
|
* (C) Copyright 1992, Hewlett-Packard Ltd.
|
|
** Licensed under the Apache License, Version 2.0 (the "License");
|
|
** you may not use this file except in compliance with the License.
|
|
** You may obtain a copy of the License at
|
|
** http://www.apache.org/licenses/LICENSE-2.0
|
|
** Unless required by applicable law or agreed to in writing, software
|
|
** distributed under the License is distributed on an "AS IS" BASIS,
|
|
** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
** See the License for the specific language governing permissions and
|
|
** limitations under the License.
|
|
*
|
|
**********************************************************************/
|
|
|
|
#include "mfcpch.h"
|
|
//#include <osfcn.h>
|
|
//#include <signal.h>
|
|
//#include <time.h>
|
|
//#include <unistd.h>
|
|
#include "tfacep.h" //must be before main.h
|
|
//#include "fileerr.h"
|
|
#include "stderr.h"
|
|
#include "basedir.h"
|
|
#include "tessvars.h"
|
|
//#include "debgwin.h"
|
|
//#include "epapdest.h"
|
|
#include "control.h"
|
|
#include "imgs.h"
|
|
#include "reject.h"
|
|
#include "pageres.h"
|
|
//#include "gpapdest.h"
|
|
#include "mainblk.h"
|
|
#include "nwmain.h"
|
|
#include "pgedit.h"
|
|
#include "ocrshell.h"
|
|
#include "tprintf.h"
|
|
//#include "ipeerr.h"
|
|
//#include "restart.h"
|
|
#include "tessedit.h"
|
|
//#include "fontfind.h"
|
|
#include "permute.h"
|
|
#include "permdawg.h"
|
|
#include "permnum.h"
|
|
#include "stopper.h"
|
|
#include "adaptmatch.h"
|
|
#include "intmatcher.h"
|
|
#include "chop.h"
|
|
#include "efio.h"
|
|
#include "danerror.h"
|
|
#include "globals.h"
|
|
|
|
/*
|
|
** Include automatically generated configuration file if running autoconf
|
|
*/
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config_auto.h"
|
|
// Includes libtiff if HAVE_LIBTIFF is defined
|
|
#ifdef HAVE_LIBTIFF
|
|
#include "tiffio.h"
|
|
#endif
|
|
#endif
|
|
|
|
#ifdef GOOGLE3
|
|
#include "third_party/tiff/tiffio.h"
|
|
#endif
|
|
|
|
//extern "C" {
|
|
#include "callnet.h" //phils nn stuff
|
|
//}
|
|
#include "notdll.h" //phils nn stuff
|
|
|
|
#define VARDIR "configs/" /*variables files */
|
|
//config under api
|
|
#define API_CONFIG "configs/api_config"
|
|
#define EXTERN
|
|
|
|
EXTERN BOOL_EVAR (tessedit_write_vars, FALSE, "Write all vars to file");
|
|
EXTERN BOOL_VAR (tessedit_tweaking_tess_vars, FALSE,
|
|
"Fiddle tess config values");
|
|
|
|
EXTERN double_VAR (tweak_garbage, 1.5, "Tess VAR");
|
|
EXTERN double_VAR (tweak_ok_word, 1.25, "Tess VAR");
|
|
EXTERN double_VAR (tweak_good_word, 1.1, "Tess VAR");
|
|
EXTERN double_VAR (tweak_freq_word, 1.0, "Tess VAR");
|
|
EXTERN double_VAR (tweak_ok_number, 1.4, "Tess VAR");
|
|
EXTERN double_VAR (tweak_good_number, 1.1, "Tess VAR");
|
|
EXTERN double_VAR (tweak_non_word, 1.25, "Tess VAR");
|
|
EXTERN double_VAR (tweak_CertaintyPerChar, -0.5, "Tess VAR");
|
|
EXTERN double_VAR (tweak_NonDictCertainty, -2.5, "Tess VAR");
|
|
EXTERN double_VAR (tweak_RejectCertaintyOffset, 1.0, "Tess VAR");
|
|
EXTERN double_VAR (tweak_GoodAdaptiveMatch, 0.125, "Tess VAR");
|
|
EXTERN double_VAR (tweak_GreatAdaptiveMatch, 0.10, "Tess VAR");
|
|
EXTERN INT_VAR (tweak_ReliableConfigThreshold, 2, "Tess VAR");
|
|
EXTERN INT_VAR (tweak_AdaptProtoThresh, 230, "Tess VAR");
|
|
EXTERN INT_VAR (tweak_AdaptFeatureThresh, 230, "Tess VAR");
|
|
EXTERN INT_VAR (tweak_min_outline_points, 6, "Tess VAR");
|
|
EXTERN INT_VAR (tweak_min_outline_area, 2000, "Tess VAR");
|
|
EXTERN double_VAR (tweak_good_split, 50.0, "Tess VAR");
|
|
EXTERN double_VAR (tweak_ok_split, 100.0, "Tess VAR");
|
|
|
|
extern INT16 XOFFSET;
|
|
extern INT16 YOFFSET;
|
|
extern int NO_BLOCK;
|
|
|
|
//progress monitor
|
|
ETEXT_DESC *global_monitor = NULL;
|
|
|
|
int init_tesseract(const char *arg0,
|
|
const char *textbase,
|
|
const char *language,
|
|
const char *configfile,
|
|
int configc,
|
|
const char *const *configv) {
|
|
FILE *var_file;
|
|
static char c_path[MAX_PATH]; //path for c code
|
|
|
|
// Set the basename, compute the data directory and read C++ configs.
|
|
main_setup(arg0, textbase, configc, configv);
|
|
debug_window_on.set_value (FALSE);
|
|
|
|
if (tessedit_write_vars) {
|
|
var_file = fopen ("edited.cfg", "w");
|
|
if (var_file != NULL) {
|
|
print_variables(var_file);
|
|
fclose(var_file);
|
|
}
|
|
}
|
|
strcpy (c_path, datadir.string ());
|
|
c_path[strlen (c_path) - strlen (m_data_sub_dir.string ())] = '\0';
|
|
demodir = c_path;
|
|
|
|
// Set the language data path prefix
|
|
language_data_path_prefix = datadir;
|
|
if (language != NULL)
|
|
language_data_path_prefix += language;
|
|
else
|
|
language_data_path_prefix += "eng";
|
|
language_data_path_prefix += ".";
|
|
|
|
// Load the unichar set
|
|
STRING unicharpath = language_data_path_prefix;
|
|
unicharpath += "unicharset";
|
|
if (!unicharset.load_from_file(unicharpath.string())) {
|
|
cprintf("Unable to load unicharset file %s\n", unicharpath.string());
|
|
exit(1);
|
|
}
|
|
if (unicharset.size() > MAX_NUM_CLASSES) {
|
|
cprintf("Error: Size of unicharset is greater than MAX_NUM_CLASSES\n");
|
|
exit(1);
|
|
}
|
|
|
|
start_recog(configfile, textbase);
|
|
|
|
set_tess_tweak_vars();
|
|
|
|
if (tessedit_use_nn) //phils nn stuff
|
|
init_net();
|
|
return 0; //Normal exit
|
|
}
|
|
|
|
void end_tesseract() {
|
|
end_recog();
|
|
}
|
|
|
|
#ifdef _TIFFIO_
|
|
void read_tiff_image(TIFF* tif, IMAGE* image) {
|
|
tdata_t buf;
|
|
uint32 image_width, image_height;
|
|
uint16 photometric;
|
|
short bpp;
|
|
TIFFGetField(tif, TIFFTAG_IMAGEWIDTH, &image_width);
|
|
TIFFGetField(tif, TIFFTAG_IMAGELENGTH, &image_height);
|
|
TIFFGetField(tif, TIFFTAG_BITSPERSAMPLE, &bpp);
|
|
TIFFGetField(tif, TIFFTAG_PHOTOMETRIC, &photometric);
|
|
// Tesseract's internal representation is 0-is-black,
|
|
// so if the photometric is 1 (min is black) then high-valued pixels
|
|
// are 1 (white), otherwise they are 0 (black).
|
|
UINT8 high_value = photometric == 1;
|
|
image->create(image_width, image_height, bpp);
|
|
IMAGELINE line;
|
|
line.init(image_width);
|
|
|
|
buf = _TIFFmalloc(TIFFScanlineSize(tif));
|
|
int bytes_per_line = (image_width*bpp + 7)/8;
|
|
UINT8* dest_buf = image->get_buffer();
|
|
// This will go badly wrong with one of the more exotic tiff formats,
|
|
// but the majority will work OK.
|
|
for (int y = 0; y < image_height; ++y) {
|
|
TIFFReadScanline(tif, buf, y);
|
|
memcpy(dest_buf, buf, bytes_per_line);
|
|
dest_buf += bytes_per_line;
|
|
}
|
|
if (high_value == 0)
|
|
invert_image(image);
|
|
_TIFFfree(buf);
|
|
}
|
|
#endif
|
|
|
|
/* Define command type identifiers */
|
|
|
|
enum CMD_EVENTS
|
|
{
|
|
ACTION_1_CMD_EVENT,
|
|
RECOG_WERDS,
|
|
RECOG_PSEUDO,
|
|
ACTION_2_CMD_EVENT
|
|
};
|
|
|
|
/**********************************************************************
|
|
* extend_menu()
|
|
*
|
|
* Function called by pgeditor to let you extend the command menu.
|
|
* Items can be added to the "MODES" and "OTHER" menus. The modes_id_base
|
|
* and other_id_base parameters are required to offset your command event ids
|
|
* from those of pgeditor, and to let the pgeditor which commands are mode
|
|
* changes and which are unmoded commands. (Sorry if you think these offsets
|
|
* are a bit kludgy, the alternative would be to duplicate all the menu
|
|
* constructor modes within pgeditor so that the offsets could be hidden.)
|
|
*
|
|
* Items for the "MODES" menu may only be simple menu items (just a name and
|
|
* id). Items for the "OTHER" menu can be editable parameters or boolean
|
|
* toggles. Refer to menu.h to see how to build different types.
|
|
**********************************************************************/
|
|
|
|
void extend_menu( //handle for "MODES"
|
|
RADIO_MENU *modes_menu,
|
|
INT16 modes_id_base, //mode cmd ids offset
|
|
NON_RADIO_MENU *other_menu, //handle for "OTHER"
|
|
INT16 other_id_base //mode cmd ids offset
|
|
) {
|
|
/* Example new mode */
|
|
|
|
modes_menu->add_child (new RADIO_MENU_LEAF ("Recog Words",
|
|
modes_id_base + RECOG_WERDS));
|
|
modes_menu->add_child (new RADIO_MENU_LEAF ("Recog Blobs",
|
|
modes_id_base + RECOG_PSEUDO));
|
|
|
|
/* Example toggle
|
|
|
|
other_menu->add_child(
|
|
new TOGGLE_MENU_LEAF( "Action 2", //Display string
|
|
other_id_base + ACTION_2_CMD_EVENT, //offset command id
|
|
FALSE ) ); //Initial value
|
|
|
|
Example text parm (commented out)
|
|
|
|
other_menu->add_child(
|
|
new VARIABLE_MENU_LEAF( "Parm change", //Display string
|
|
other_id_base + ACTION_3_CMD_EVENT, //offset command id
|
|
"default value" ) ); //default value string
|
|
*/
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* extend_moded_commands()
|
|
*
|
|
* Function called by pgeditor when the user is in one of the extended modes
|
|
* defined by extend_menu() and the user has selected an area in the image
|
|
* window.
|
|
**********************************************************************/
|
|
|
|
void extend_moded_commands( //current mode
|
|
INT32 mode,
|
|
BOX selection_box //area selected
|
|
) {
|
|
char msg[MAX_CHARS + 1];
|
|
|
|
switch (mode) {
|
|
case RECOG_WERDS:
|
|
command_window->msg ("Recogging selected words");
|
|
|
|
/* This is how to apply a "word processor" function to each selected word */
|
|
|
|
process_selected_words(current_block_list,
|
|
selection_box,
|
|
&recog_interactive);
|
|
break;
|
|
case RECOG_PSEUDO:
|
|
command_window->msg ("Recogging selected blobs");
|
|
|
|
/* This is how to apply a "word processor" function to each selected word */
|
|
|
|
recog_pseudo_word(current_block_list, selection_box);
|
|
break;
|
|
default:
|
|
sprintf (msg, "Unexpected extended mode " INT32FORMAT, mode);
|
|
command_window->msg (msg);
|
|
}
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* extend_unmoded_commands()
|
|
*
|
|
* Function called by pgeditor when the user has selected one of the unmoded
|
|
* extended menu options.
|
|
**********************************************************************/
|
|
|
|
void extend_unmoded_commands( //current mode
|
|
INT32 cmd_event,
|
|
char *new_value //changed value if any
|
|
) {
|
|
char msg[MAX_CHARS + 1];
|
|
|
|
switch (cmd_event) {
|
|
case ACTION_2_CMD_EVENT: //a toggle event
|
|
if (new_value[0] == 'T')
|
|
//Display message
|
|
command_window->msg ("Extended Action 2 ON!!");
|
|
else
|
|
command_window->msg ("Extended Action 2 OFF!!");
|
|
break;
|
|
default:
|
|
sprintf (msg, "Unrecognised extended command " INT32FORMAT " (%s)",
|
|
cmd_event, new_value);
|
|
command_window->msg (msg);
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* set_tess_tweak_vars()
|
|
* Set TESS vars from the tweek value - This is only really of use during search
|
|
* of the space of tess configs - othertimes the default values are set
|
|
*
|
|
*************************************************************************/
|
|
void set_tess_tweak_vars() {
|
|
if (tessedit_tweaking_tess_vars) {
|
|
garbage = tweak_garbage;
|
|
ok_word = tweak_ok_word;
|
|
good_word = tweak_good_word;
|
|
freq_word = tweak_freq_word;
|
|
ok_number = tweak_ok_number;
|
|
good_number = tweak_good_number;
|
|
non_word = tweak_non_word;
|
|
CertaintyPerChar = tweak_CertaintyPerChar;
|
|
NonDictCertainty = tweak_NonDictCertainty;
|
|
RejectCertaintyOffset = tweak_RejectCertaintyOffset;
|
|
GoodAdaptiveMatch = tweak_GoodAdaptiveMatch;
|
|
GreatAdaptiveMatch = tweak_GreatAdaptiveMatch;
|
|
ReliableConfigThreshold = tweak_ReliableConfigThreshold;
|
|
AdaptProtoThresh = tweak_AdaptProtoThresh;
|
|
AdaptFeatureThresh = tweak_AdaptFeatureThresh;
|
|
min_outline_points = tweak_min_outline_points;
|
|
min_outline_area = tweak_min_outline_area;
|
|
good_split = tweak_good_split;
|
|
ok_split = tweak_ok_split;
|
|
}
|
|
// if (expiry_day * 24 * 60 * 60 < time(NULL))
|
|
// err_exit();
|
|
}
|