/****************************************************************** * File: control.cpp (Formerly control.c) * Description: Module-independent matcher controller. * Author: Ray Smith * Created: Thu Apr 23 11:09:58 BST 1992 * ReHacked: Tue Sep 22 08:42:49 BST 1992 Phil Cheatle * * (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 "mainblk.h" #include #include #ifdef __UNIX__ #include #include #include #endif #include #include "ocrclass.h" #include "werdit.h" #include "drawfx.h" #include "tfacep.h" #include "tessbox.h" #include "tessvars.h" //#include "fxtop.h" #include "pgedit.h" #include "reject.h" #include "adaptions.h" #include "charcut.h" #include "fixxht.h" #include "fixspace.h" #include "genblob.h" #include "docqual.h" #include "control.h" #include "secname.h" #include "output.h" #include "callcpp.h" #include "notdll.h" #include "tordvars.h" #include "adaptmatch.h" #include "globals.h" #define MIN_FONT_ROW_COUNT 8 #define MAX_XHEIGHT_DIFF 3 #define EXTERN //extern "C" { //EXTERN BOOL_VAR(tessedit_small_match,FALSE,"Use small matrix matcher"); //extern FILE* matcher_fp; //extern FILE* correct_fp; //}; BOOL_VAR (tessedit_small_match, FALSE, "Use small matrix matcher"); EXTERN BOOL_VAR (tessedit_print_text, FALSE, "Write text to stdout"); EXTERN BOOL_VAR (tessedit_draw_words, FALSE, "Draw source words"); EXTERN BOOL_VAR (tessedit_draw_outwords, FALSE, "Draw output words"); EXTERN BOOL_VAR (tessedit_training_wiseowl, FALSE, "Call WO to learn blobs"); EXTERN BOOL_VAR (tessedit_training_tess, FALSE, "Call Tess to learn blobs"); EXTERN BOOL_VAR (tessedit_matcher_is_wiseowl, FALSE, "Call WO to classify"); EXTERN BOOL_VAR (tessedit_dump_choices, FALSE, "Dump char choices"); EXTERN BOOL_VAR (tessedit_fix_fuzzy_spaces, TRUE, "Try to improve fuzzy spaces"); EXTERN BOOL_VAR (tessedit_unrej_any_wd, FALSE, "Dont bother with word plausibility"); EXTERN BOOL_VAR (tessedit_fix_hyphens, TRUE, "Crunch double hyphens?"); EXTERN BOOL_VAR (tessedit_reject_fullstops, FALSE, "Reject all fullstops"); EXTERN BOOL_VAR (tessedit_reject_suspect_fullstops, FALSE, "Reject suspect fullstops"); EXTERN BOOL_VAR (tessedit_redo_xheight, TRUE, "Check/Correct x-height"); EXTERN BOOL_VAR (tessedit_cluster_adaption_on, TRUE, "Do our own adaption - ems only"); EXTERN BOOL_VAR (tessedit_enable_doc_dict, TRUE, "Add words to the document dictionary"); EXTERN BOOL_VAR (word_occ_first, FALSE, "Do word occ before re-est xht"); EXTERN BOOL_VAR (tessedit_debug_fonts, FALSE, "Output font info per char"); EXTERN BOOL_VAR (tessedit_xht_fiddles_on_done_wds, TRUE, "Apply xht fix up even if done"); EXTERN BOOL_VAR (tessedit_xht_fiddles_on_no_rej_wds, TRUE, "Apply xht fix up even in no rejects"); EXTERN INT_VAR (x_ht_check_word_occ, 2, "Check Char Block occupancy"); EXTERN INT_VAR (x_ht_stringency, 1, "How many confirmed a/n to accept?"); EXTERN BOOL_VAR (x_ht_quality_check, TRUE, "Dont allow worse quality"); EXTERN BOOL_VAR (tessedit_debug_block_rejection, FALSE, "Block and Row stats"); EXTERN INT_VAR (debug_x_ht_level, 0, "Reestimate debug"); EXTERN BOOL_VAR (rej_use_xht, TRUE, "Individual rejection control"); EXTERN BOOL_VAR (debug_acceptable_wds, FALSE, "Dump word pass/fail chk"); EXTERN STRING_VAR (chs_leading_punct, "('`\"", "Leading punctuation"); EXTERN STRING_VAR (chs_trailing_punct1, ").,;:?!", "1st Trailing punctuation"); EXTERN STRING_VAR (chs_trailing_punct2, ")'`\"", "2nd Trailing punctuation"); EXTERN double_VAR (quality_rej_pc, 0.08, "good_quality_doc lte rejection limit"); EXTERN double_VAR (quality_blob_pc, 0.0, "good_quality_doc gte good blobs limit"); EXTERN double_VAR (quality_outline_pc, 1.0, "good_quality_doc lte outline error limit"); EXTERN double_VAR (quality_char_pc, 0.95, "good_quality_doc gte good char limit"); EXTERN INT_VAR (quality_min_initial_alphas_reqd, 2, "alphas in a good word"); EXTERN BOOL_VAR (tessedit_tess_adapt_to_rejmap, FALSE, "Use reject map to control Tesseract adaption"); EXTERN INT_VAR (tessedit_tess_adaption_mode, 0x27, "Adaptation decision algorithm for tess"); EXTERN INT_VAR (tessedit_em_adaption_mode, 0, "Adaptation decision algorithm for ems matrix matcher"); EXTERN BOOL_VAR (tessedit_cluster_adapt_after_pass1, FALSE, "Adapt using clusterer after pass 1"); EXTERN BOOL_VAR (tessedit_cluster_adapt_after_pass2, FALSE, "Adapt using clusterer after pass 1"); EXTERN BOOL_VAR (tessedit_cluster_adapt_after_pass3, FALSE, "Adapt using clusterer after pass 1"); EXTERN BOOL_VAR (tessedit_cluster_adapt_before_pass1, FALSE, "Adapt using clusterer before Tess adaping during pass 1"); EXTERN INT_VAR (tessedit_cluster_adaption_mode, 0, "Adaptation decision algorithm for matrix matcher"); EXTERN BOOL_VAR (tessedit_adaption_debug, FALSE, "Generate and print debug information for adaption"); EXTERN BOOL_VAR (tessedit_minimal_rej_pass1, FALSE, "Do minimal rejection on pass 1 output"); EXTERN BOOL_VAR (tessedit_test_adaption, FALSE, "Test adaption criteria"); EXTERN BOOL_VAR (tessedit_global_adaption, FALSE, "Adapt to all docs over time"); EXTERN BOOL_VAR (tessedit_matcher_log, FALSE, "Log matcher activity"); EXTERN INT_VAR (tessedit_test_adaption_mode, 3, "Adaptation decision algorithm for tess"); EXTERN BOOL_VAR (test_pt, FALSE, "Test for point"); EXTERN double_VAR (test_pt_x, 99999.99, "xcoord"); EXTERN double_VAR (test_pt_y, 99999.99, "ycoord"); extern int MatcherDebugLevel; extern int display_ratings; extern int number_debug; extern int adjust_debug; FILE *choice_file = NULL; //Choice file ptr CLISTIZEH (PBLOB) CLISTIZE (PBLOB) /* DEBUGGING */ INT16 blob_count(WERD *w) { return w->blob_list ()->length (); } /********************************************************************** * recog_pseudo_word * * Make a word from the selected blobs and run Tess on them. **********************************************************************/ void recog_pseudo_word( //recognize blobs BLOCK_LIST *block_list, //blocks to check BOX &selection_box) { WERD *word; ROW *pseudo_row; //row of word BLOCK *pseudo_block; //block of word word = make_pseudo_word (block_list, selection_box, pseudo_block, pseudo_row); if (word != NULL) { recog_interactive(pseudo_block, pseudo_row, word); delete word; } } /********************************************************************** * recog_interactive * * Recognize a single word in interactive mode. **********************************************************************/ BOOL8 recog_interactive( //recognize blobs BLOCK *, //block ROW *row, //row of word WERD *word //word to recognize ) { WERD_RES word_res(word); INT16 char_qual; INT16 good_char_qual; classify_word_pass2(&word_res, row); #ifndef SECURE_NAMES if (tessedit_debug_quality_metrics) { word_char_quality(&word_res, row, &char_qual, &good_char_qual); tprintf ("\n%d chars; word_blob_quality: %d; outline_errs: %d; char_quality: %d; good_char_quality: %d\n", word_res.reject_map.length (), word_blob_quality (&word_res, row), word_outline_errs (&word_res), char_qual, good_char_qual); } #endif return TRUE; } /********************************************************************** * recog_all_words() * * Walk the current block list applying the specified word processor function * to all words **********************************************************************/ void recog_all_words( //process words PAGE_RES *page_res, //page structure volatile ETEXT_DESC *monitor, //progress monitor BOX *target_word_box,//specifies just to extract a retangle INT16 dopasses //0 - all, 1 just pass 1, 2 passes 2 and higher ) { //reset page iterator static PAGE_RES_IT page_res_it; INT16 chars_in_word; INT16 rejects_in_word; static CHAR_SAMPLES_LIST em_clusters; static CHAR_SAMPLE_LIST ems_waiting; static CHAR_SAMPLES_LIST char_clusters; static CHAR_SAMPLE_LIST chars_waiting; INT16 blob_quality = 0; INT16 outline_errs = 0; static INT16 doc_blob_quality = 0; static INT16 doc_outline_errs = 0; static INT16 doc_char_quality = 0; INT16 all_char_quality; INT16 accepted_all_char_quality; static INT16 good_char_count = 0; static INT16 doc_good_char_quality = 0; int i; INT32 tess_adapt_mode = 0; static INT32 word_count; //count of words in doc INT32 word_index; //current word static int dict_words; if (tessedit_minimal_rej_pass1) { tessedit_test_adaption.set_value (TRUE); tessedit_minimal_rejection.set_value (TRUE); } if (tessedit_cluster_adapt_before_pass1) { tess_adapt_mode = tessedit_tess_adaption_mode; tessedit_tess_adaption_mode.set_value (0); tessedit_tess_adapt_to_rejmap.set_value (TRUE); } if (dopasses==0 || dopasses==1) { page_res_it.page_res=page_res; page_res_it.restart_page(); /* Pass 1 */ word_count = 0; if (monitor != NULL) { monitor->ocr_alive = TRUE; while (page_res_it.word () != NULL) { word_count++; page_res_it.forward (); } page_res_it.restart_page (); } else word_count = 1; word_index = 0; em_clusters.clear(); ems_waiting.clear(); char_clusters.clear(); chars_waiting.clear(); dict_words = 0; doc_blob_quality = 0; doc_outline_errs = 0; doc_char_quality = 0; good_char_count = 0; doc_good_char_quality = 0; while (page_res_it.word () != NULL) { set_global_loc_code(LOC_PASS1); word_index++; if (monitor != NULL) { monitor->ocr_alive = TRUE; monitor->progress = 30 + 50 * word_index / word_count; if ((monitor->end_time != 0 && clock() > monitor->end_time) || (monitor->cancel != NULL && (*monitor->cancel)(monitor->cancel_this, dict_words))) return; } classify_word_pass1 (page_res_it.word (), page_res_it.row ()->row, FALSE, NULL, NULL); if (tessedit_test_adaption && !tessedit_minimal_rejection) { if (!word_adaptable (page_res_it.word (), tessedit_test_adaption_mode)) page_res_it.word ()->reject_map.rej_word_tess_failure (); //FAKE PERM REJ else { const STRING* wordstr = &(page_res_it.word ()->best_choice->string ()); /* Override rejection mechanisms for this word */ const char* text = wordstr->string (); for (i = 0; text[i] != '\0'; i++) { if ((text[i] != ' ') && page_res_it.word ()->reject_map[i].rejected ()) page_res_it.word ()->reject_map[i]. setrej_minimal_rej_accept(); } } } if ((tessedit_cluster_adapt_after_pass1 || tessedit_cluster_adapt_after_pass3 || tessedit_cluster_adapt_before_pass1) && tessedit_cluster_adaption_mode != 0) { collect_characters_for_adaption (page_res_it.word (), &char_clusters, &chars_waiting); } // Count dict words. if (page_res_it.word()->best_choice->permuter() == USER_DAWG_PERM) ++dict_words; page_res_it.forward (); } if (tessedit_cluster_adapt_before_pass1) tessedit_tess_adaption_mode.set_value (tess_adapt_mode); page_res_it.restart_page (); while ((tessedit_cluster_adapt_after_pass1 || tessedit_cluster_adapt_before_pass1) && page_res_it.word () != NULL) { if (monitor != NULL) monitor->ocr_alive = TRUE; if (tessedit_cluster_adapt_after_pass1) adapt_to_good_samples (page_res_it.word (), &char_clusters, &chars_waiting); else classify_word_pass1 (page_res_it.word (), page_res_it.row ()->row, TRUE, &char_clusters, &chars_waiting); page_res_it.forward (); } // } if (dopasses==1) return; /* Pass 2 */ page_res_it.restart_page (); word_index = 0; while (!tessedit_test_adaption && page_res_it.word () != NULL) { set_global_loc_code(LOC_PASS2); word_index++; if (monitor != NULL) { monitor->ocr_alive = TRUE; monitor->progress = 80 + 10 * word_index / word_count; if ((monitor->end_time != 0 && clock() > monitor->end_time) || (monitor->cancel != NULL && (*monitor->cancel)(monitor->cancel_this, dict_words))) return; } //changed by jetsoft //specific to its needs to extract one word when need if (target_word_box) { BOX current_word_box=page_res_it.word ()->word->bounding_box(); FCOORD center_pt((current_word_box.right()+current_word_box.left())/2,(current_word_box.bottom()+current_word_box.top())/2); if (!target_word_box->contains(center_pt)) { page_res_it.forward (); continue; } } //end jetsoft classify_word_pass2 (page_res_it.word (), page_res_it.row ()->row); if (tessedit_em_adaption_mode > 0) collect_ems_for_adaption (page_res_it.word (), &em_clusters, &ems_waiting); if (tessedit_cluster_adapt_after_pass2 && tessedit_cluster_adaption_mode != 0) collect_characters_for_adaption (page_res_it.word (), &char_clusters, &chars_waiting); page_res_it.forward (); } /* Another pass */ set_global_loc_code(LOC_FUZZY_SPACE); if (!tessedit_test_adaption && tessedit_fix_fuzzy_spaces && !tessedit_word_for_word) fix_fuzzy_spaces(monitor, word_count, page_res); if (!tessedit_test_adaption && tessedit_em_adaption_mode != 0) // Initially ems only print_em_stats(&em_clusters, &ems_waiting); /* Pass 3 - used for checking confusion sets */ page_res_it.restart_page (); word_index = 0; while (!tessedit_test_adaption && page_res_it.word () != NULL) { set_global_loc_code(LOC_MM_ADAPT); word_index++; if (monitor != NULL) { monitor->ocr_alive = TRUE; monitor->progress = 95 + 5 * word_index / word_count; } check_debug_pt (page_res_it.word (), 70); /* Use good matches to sort out confusions */ //changed by jetsoft //specific to its needs to extract one word when need if (target_word_box) { BOX current_word_box=page_res_it.word ()->word->bounding_box(); FCOORD center_pt((current_word_box.right()+current_word_box.left())/2,(current_word_box.bottom()+current_word_box.top())/2); if (!target_word_box->contains(center_pt)) { page_res_it.forward (); continue; } } // end jetsoft if (tessedit_em_adaption_mode != 0) adapt_to_good_ems (page_res_it.word (), &em_clusters, &ems_waiting); if (tessedit_cluster_adapt_after_pass2 && tessedit_cluster_adaption_mode != 0) adapt_to_good_samples (page_res_it.word (), &char_clusters, &chars_waiting); if (tessedit_reject_fullstops && strchr (page_res_it.word ()->best_choice->string ().string (), '.') != NULL) reject_all_fullstops (page_res_it.word ()); else if (tessedit_reject_suspect_fullstops && strchr (page_res_it.word ()->best_choice->string (). string (), '.') != NULL) reject_suspect_fullstops (page_res_it.word ()); page_res_it.rej_stat_word (); chars_in_word = page_res_it.word ()->reject_map.length (); rejects_in_word = page_res_it.word ()->reject_map.reject_count (); blob_quality = word_blob_quality (page_res_it.word (), page_res_it.row ()->row); doc_blob_quality += blob_quality; outline_errs = word_outline_errs (page_res_it.word ()); doc_outline_errs += outline_errs; word_char_quality (page_res_it.word (), page_res_it.row ()->row, &all_char_quality, &accepted_all_char_quality); doc_char_quality += all_char_quality; UINT8 permuter_type = page_res_it.word ()->best_choice->permuter (); if ((permuter_type == SYSTEM_DAWG_PERM) || (permuter_type == FREQ_DAWG_PERM) || (permuter_type == USER_DAWG_PERM)) { good_char_count += chars_in_word - rejects_in_word; doc_good_char_quality += accepted_all_char_quality; } check_debug_pt (page_res_it.word (), 80); if (tessedit_reject_bad_qual_wds && (blob_quality == 0) && (outline_errs >= chars_in_word)) page_res_it.word ()->reject_map.rej_word_bad_quality (); check_debug_pt (page_res_it.word (), 90); page_res_it.forward (); } page_res_it.restart_page (); while (!tessedit_test_adaption && tessedit_cluster_adapt_after_pass3 && page_res_it.word () != NULL) { if (monitor != NULL) monitor->ocr_alive = TRUE; //changed by jetsoft //specific to its needs to extract one word when need if (target_word_box) { BOX current_word_box=page_res_it.word ()->word->bounding_box(); FCOORD center_pt((current_word_box.right()+current_word_box.left())/2,(current_word_box.bottom()+current_word_box.top())/2); if (!target_word_box->contains(center_pt)) { page_res_it.forward (); continue; } } //end jetsoft if (tessedit_cluster_adaption_mode != 0) adapt_to_good_samples (page_res_it.word (), &char_clusters, &chars_waiting); page_res_it.forward (); } #ifndef SECURE_NAMES if (tessedit_debug_quality_metrics) { tprintf ("QUALITY: num_chs= %d num_rejs= %d %5.3f blob_qual= %d %5.3f outline_errs= %d %5.3f char_qual= %d %5.3f good_ch_qual= %d %5.3f\n", page_res->char_count, page_res->rej_count, page_res->rej_count / (float) page_res->char_count, doc_blob_quality, doc_blob_quality / (float) page_res->char_count, doc_outline_errs, doc_outline_errs / (float) page_res->char_count, doc_char_quality, doc_char_quality / (float) page_res->char_count, doc_good_char_quality, good_char_count > 0 ? doc_good_char_quality / (float) good_char_count : 0.0); } #endif BOOL8 good_quality_doc = (page_res->rej_count / (float) page_res->char_count <= quality_rej_pc) && (doc_blob_quality / (float) page_res->char_count >= quality_blob_pc) && (doc_outline_errs / (float) page_res->char_count <= quality_outline_pc) && (doc_char_quality / (float) page_res->char_count >= quality_char_pc); /* Do whole document or whole block rejection pass*/ if (!tessedit_test_adaption) { set_global_loc_code(LOC_DOC_BLK_REJ); quality_based_rejection(page_res_it, good_quality_doc); } font_recognition_pass(page_res_it); /* Write results pass */ set_global_loc_code(LOC_WRITE_RESULTS); // This is now redundant, but retained commented so show how to obtain // bounding boxes and style information. ////changed by jetsoft //needed for dll to output memory structure if ((dopasses == 0 || dopasses == 2) && (monitor || tessedit_write_unlv)) output_pass(page_res_it, ocr_char_space() > 0, target_word_box); // end jetsoft } /********************************************************************** * classify_word_pass1 * * Baseline normalize the word and pass it to Tess. **********************************************************************/ void classify_word_pass1( //recog one word WERD_RES *word, //word to do ROW *row, BOOL8 cluster_adapt, CHAR_SAMPLES_LIST *char_clusters, CHAR_SAMPLE_LIST *chars_waiting) { WERD *bln_word; //baseline norm copy //detailed results BLOB_CHOICE_LIST_CLIST blob_choices; BOOL8 adapt_ok; const char *rejmap; INT16 index; STRING mapstr = ""; char *match_string; char word_string[1024]; if (matcher_fp != NULL) { fgets (word_string, 1023, correct_fp); if ((match_string = strchr (word_string, '\r')) != NULL) *match_string = '\0'; if ((match_string = strchr (word_string, '\n')) != NULL) *match_string = '\0'; if (word_string[0] != '\0') { word->word->set_text (word_string); word_answer = (char *) word->word->text (); } else word_answer = NULL; } check_debug_pt (word, 0); matcher_pass = 0; bln_word = make_bln_copy (word->word, row, word->x_height, &word->denorm); word->best_choice = tess_segment_pass1 (bln_word, &word->denorm, tess_default_matcher, word->raw_choice, &blob_choices, word->outword); /* Test for TESS screw up on word. Recog_word has already ensured that the choice list, outword blob lists and best_choice string are the same length. A TESS screw up is indicated by a blank filled or 0 length string. */ if ((word->best_choice->lengths ().length () == 0) || (strspn (word->best_choice->string ().string (), " ") == word->best_choice->string ().length ())) { word->done = FALSE; //Try again on pass2 - adaption may help word->tess_failed = TRUE; word->reject_map.initialise (word->best_choice->lengths ().length ()); word->reject_map.rej_word_tess_failure (); } else { word->tess_failed = FALSE; if ((word->best_choice->lengths ().length () != word->outword->blob_list ()->length ()) || (word->best_choice->lengths ().length () != blob_choices.length ())) { tprintf ("ASSERT FAIL String:\"%s\"; Strlen=%d; #Blobs=%d; #Choices=%d\n", word->best_choice->string ().string (), word->best_choice->lengths ().length (), word->outword->blob_list ()->length (), blob_choices.length ()); } ASSERT_HOST (word->best_choice->lengths ().length () == word->outword->blob_list ()->length ()); ASSERT_HOST (word->best_choice->lengths ().length () == blob_choices.length ()); /* The adaption step used to be here. It has been moved to after make_reject_map so that we know whether the word will be accepted in the first pass or not. This move will PREVENT adaption to words containing double quotes because the word will not be identical to what tess thinks its best choice is. (See CurrentBestChoiceIs in danj/microfeatures/stopper.c which is used by AdaptableWord in danj/microfeatures/adaptmatch.c) */ if (word->word->flag (W_REP_CHAR)) { fix_rep_char(word); } else { fix_quotes (word->best_choice, //turn to double word->outword, &blob_choices); if (tessedit_fix_hyphens) //turn 2 to 1 fix_hyphens (word->best_choice, word->outword, &blob_choices); record_certainty (word->best_choice->certainty (), 1); //accounting word->tess_accepted = tess_acceptable_word (word->best_choice, word->raw_choice); word->tess_would_adapt = tess_adaptable_word (word->outword, word->best_choice, word->raw_choice); // Also sets word->done flag make_reject_map (word, &blob_choices, row, 1); adapt_ok = word_adaptable (word, tessedit_tess_adaption_mode); if (cluster_adapt) adapt_to_good_samples(word, char_clusters, chars_waiting); if (adapt_ok || tessedit_tess_adapt_to_rejmap) { if (!tessedit_tess_adapt_to_rejmap) rejmap = NULL; else { ASSERT_HOST (word->reject_map.length () == word->best_choice->lengths ().length ()); for (index = 0; index < word->reject_map.length (); index++) { if (adapt_ok || word->reject_map[index].accepted ()) mapstr += '1'; else mapstr += '0'; } rejmap = mapstr.string (); } //adapt to it tess_adapter (word->outword, &word->denorm, *word->best_choice, *word->raw_choice, rejmap); } if (tessedit_enable_doc_dict) tess_add_doc_word (word->best_choice); set_word_fonts(word, &blob_choices); } } #if 0 if (tessedit_print_text) { write_cooked_text (bln_word, word->best_choice->string (), word->done, FALSE, stdout); } #endif delete bln_word; blob_choices.deep_clear (); } /********************************************************************** * classify_word_pass2 * * Control what to do with the word in pass 2 **********************************************************************/ void classify_word_pass2( //word to do WERD_RES *word, ROW *row) { BOOL8 done_this_pass = FALSE; WERD_RES new_x_ht_word (word->word); float new_x_ht = 0.0; INT16 old_xht_reject_count; INT16 new_xht_reject_count; INT16 old_xht_accept_count; INT16 new_xht_accept_count; BOOL8 accept_new_x_ht = FALSE; INT16 old_chs_in_wd; INT16 new_chs_in_wd; INT16 old_word_quality; INT16 new_word_quality; INT16 dummy; set_global_subloc_code(SUBLOC_NORM); check_debug_pt (word, 30); if (!word->done || tessedit_training_tess || tessedit_training_wiseowl || tessedit_dump_choices) { word->caps_height = 0.0; if (word->x_height == 0.0f) word->x_height = row->x_height(); if (word->outword != NULL) { delete word->outword; //get rid of junk delete word->best_choice; delete word->raw_choice; } match_word_pass2 (word, row, word->x_height); done_this_pass = TRUE; check_debug_pt (word, 40); } if (!word->tess_failed && !word->word->flag (W_REP_CHAR)) { set_global_subloc_code(SUBLOC_FIX_XHT); if ((tessedit_xht_fiddles_on_done_wds || !word->done) && (tessedit_xht_fiddles_on_no_rej_wds || (word->reject_map.reject_count () > 0))) { if ((x_ht_check_word_occ >= 2) && word_occ_first) check_block_occ(word); if (tessedit_redo_xheight) re_estimate_x_ht(word, &new_x_ht); if (((x_ht_check_word_occ >= 2) && !word_occ_first) || ((x_ht_check_word_occ >= 1) && (new_x_ht > 0))) check_block_occ(word); } if (new_x_ht > 0) { old_chs_in_wd = word->reject_map.length (); /* Re-estimated x_ht error suggests a rematch is worthwhile. */ new_x_ht_word.x_height = new_x_ht; new_x_ht_word.caps_height = 0.0; match_word_pass2 (&new_x_ht_word, row, new_x_ht_word.x_height); if (!new_x_ht_word.tess_failed) { if ((x_ht_check_word_occ >= 1) && word_occ_first) check_block_occ(&new_x_ht_word); re_estimate_x_ht(&new_x_ht_word, &new_x_ht); if ((x_ht_check_word_occ >= 1) && !word_occ_first) check_block_occ(&new_x_ht_word); old_xht_reject_count = word->reject_map.reject_count (); old_xht_accept_count = old_chs_in_wd - old_xht_reject_count; new_xht_reject_count = new_x_ht_word.reject_map.reject_count (); new_chs_in_wd = new_x_ht_word.reject_map.length (); new_xht_accept_count = new_chs_in_wd - new_xht_reject_count; accept_new_x_ht = ((new_xht_accept_count > old_xht_accept_count) || ((new_xht_accept_count == old_xht_accept_count) && (new_xht_accept_count > 0))) && (!new_x_ht_word.guessed_x_ht || !new_x_ht_word.guessed_caps_ht); if (accept_new_x_ht && x_ht_quality_check) { word_char_quality(word, row, &old_word_quality, &dummy); word_char_quality(&new_x_ht_word, row, &new_word_quality, &dummy); if (old_word_quality > new_word_quality) accept_new_x_ht = FALSE; } if (accept_new_x_ht && (x_ht_stringency > 0)) { accept_new_x_ht = (count_alphanums (&new_x_ht_word) > x_ht_stringency); if (!accept_new_x_ht && rej_use_xht) { if (debug_x_ht_level >= 1) tprintf ("Failed stringency test so reject original word\n"); word->reject_map.rej_word_xht_fixup (); } } #ifndef SECURE_NAMES if (debug_x_ht_level >= 1) { tprintf ("New XHT Match:: %s ", word->best_choice->string ().string ()); word->reject_map.print (debug_fp); tprintf (" -> %s ", new_x_ht_word.best_choice->string ().string ()); new_x_ht_word.reject_map.print (debug_fp); tprintf (" %s->%s %s %s\n", word->guessed_x_ht ? "GUESS" : "CERT", new_x_ht_word.guessed_x_ht ? "GUESS" : "CERT", new_x_ht > 0.1 ? "STILL DOUBT" : "OK", accept_new_x_ht ? "ACCEPTED" : ""); } #endif } if (accept_new_x_ht) { /* The new x_ht is deemed superior so put the final results in the real word and destroy the old results */ delete word->outword; //get rid of junk word->outword = new_x_ht_word.outword; word->denorm = new_x_ht_word.denorm; delete word->best_choice; word->best_choice = new_x_ht_word.best_choice; delete word->raw_choice; word->raw_choice = new_x_ht_word.raw_choice; word->reject_map = new_x_ht_word.reject_map; word->done = new_x_ht_word.done; done_this_pass = TRUE; } else { /* The new x_ht is no better, so destroy the copy word and put any uncertain x or cap ht estimate back to default. (I.e. dont blame me if its bad!) Conditionally, use any ammended block occ chars. */ //get rid of junk delete new_x_ht_word.outword; delete new_x_ht_word.best_choice; delete new_x_ht_word.raw_choice; } //to keep new destructor happy new_x_ht_word.outword = NULL; //to keep new destructor happy new_x_ht_word.best_choice = NULL; //to keep new destructor happy new_x_ht_word.raw_choice = NULL; if (rej_mostly_reject_mode == 2) { reject_mostly_rejects(word); tprintf ("Rejecting mostly rejects on %s ", word->best_choice->string ().string ()); } } set_global_subloc_code(SUBLOC_NORM); if (done_this_pass && !word->done && tessedit_save_stats) SaveBadWord (word->best_choice->string ().string (), word->best_choice->certainty ()); record_certainty (word->best_choice->certainty (), 2); //accounting } #ifndef GRAPHICS_DISABLED if (tessedit_draw_outwords) { if (fx_win == NO_WINDOW) create_fx_win(); clear_fx_win(); word->outword->plot (fx_win); make_picture_current(fx_win); } #endif set_global_subloc_code(SUBLOC_NORM); #if 0 if (tessedit_print_text) { write_cooked_text (word->outword, word->best_choice->string (), word->done, done_this_pass, stdout); } #endif check_debug_pt (word, 50); } /********************************************************************** * match_word_pass2 * * Baseline normalize the word and pass it to Tess. **********************************************************************/ void match_word_pass2( //recog one word WERD_RES *word, //word to do ROW *row, float x_height) { WERD *bln_word; //baseline norm copy //detailed results BLOB_CHOICE_LIST_CLIST blob_choices; set_global_subsubloc_code(SUBSUBLOC_OTHER); if (matcher_fp != NULL) { word_answer = (char *) word->word->text (); if (word_answer != NULL && word_answer[0] == '\0') word_answer = NULL; } matcher_pass = 0; bln_word = make_bln_copy (word->word, row, x_height, &word->denorm); set_global_subsubloc_code(SUBSUBLOC_TESS); if (tessedit_training_tess) word->best_choice = correct_segment_pass2 (bln_word, &word->denorm, tess_default_matcher, tess_training_tester, word->raw_choice, &blob_choices, word->outword); else if (tessedit_dump_choices) word->best_choice = test_segment_pass2 (bln_word, &word->denorm, tess_default_matcher, choice_dump_tester, word->raw_choice, &blob_choices, word->outword); // else if (tessedit_training_wiseowl) // best_choice=correct_segment_pass2( word, &denorm, // tess_default_matcher,wo_learn, // raw_choice,&blob_choices,outword); // else if (tessedit_matcher_is_wiseowl) // best_choice=tess_segment_pass2( word, &denorm, wo_classify, // raw_choice, &blob_choices, outword); else { word->best_choice = tess_segment_pass2 (bln_word, &word->denorm, tess_default_matcher, word->raw_choice, &blob_choices, word->outword); } set_global_subsubloc_code(SUBSUBLOC_OTHER); /* Test for TESS screw up on word. Recog_word has already ensured that the choice list, outword blob lists and best_choice string are the same length. A TESS screw up is indicated by a blank filled or 0 length string. */ if ((word->best_choice->string ().length () == 0) || (strspn (word->best_choice->string ().string (), " ") == word->best_choice->string ().length ())) { word->tess_failed = TRUE; word->reject_map.initialise (word->best_choice->string ().length ()); word->reject_map.rej_word_tess_failure (); // tprintf("Empty word produced\n"); } else { if ((word->best_choice->lengths ().length () != word->outword->blob_list ()->length ()) || (word->best_choice->lengths ().length () != blob_choices.length ())) { tprintf ("ASSERT FAIL String:\"%s\"; Strlen=%d; #Blobs=%d; #Choices=%d\n", word->best_choice->string ().string (), word->best_choice->lengths ().length (), word->outword->blob_list ()->length (), blob_choices.length ()); } ASSERT_HOST (word->best_choice->lengths ().length () == word->outword->blob_list ()->length ()); ASSERT_HOST (word->best_choice->lengths ().length () == blob_choices.length ()); word->tess_failed = FALSE; if (word->word->flag (W_REP_CHAR)) { fix_rep_char(word); } else { fix_quotes (word->best_choice, word->outword, &blob_choices); if (tessedit_fix_hyphens) fix_hyphens (word->best_choice, word->outword, &blob_choices); /* Dont trust fix_quotes! - though I think I've fixed the bug */ if ((word->best_choice->lengths ().length () != word->outword->blob_list ()->length ()) || (word->best_choice->lengths ().length () != blob_choices.length ())) { #ifndef SECURE_NAMES tprintf ("POST FIX_QUOTES FAIL String:\"%s\"; Strlen=%d; #Blobs=%d; #Choices=%d\n", word->best_choice->string ().string (), word->best_choice->lengths ().length (), word->outword->blob_list ()->length (), blob_choices.length ()); #endif } ASSERT_HOST (word->best_choice->lengths ().length () == word->outword->blob_list ()->length ()); ASSERT_HOST (word->best_choice->lengths ().length () == blob_choices.length ()); word->tess_accepted = tess_acceptable_word (word->best_choice, word->raw_choice); make_reject_map (word, &blob_choices, row, 2); } } blob_choices.deep_clear (); delete bln_word; assert (word->raw_choice != NULL); } /************************************************************************* * fix_rep_char() * The word is a repeated char. Find the repeated char character. Make a reject * string which rejects any char other than the voted char. Set the word to done * to stop rematching it. * *************************************************************************/ void fix_rep_char( //Repeated char word WERD_RES *word //word to do ) { struct REP_CH { char ch[UNICHAR_LEN + 1]; int count; }; REP_CH *rep_ch; //array of char counts int word_len; int rep_ch_count = 0; //how many unique chs const char *word_str; //the repeated chs int i, j; int offset; int total = 0; int max = 0; char *maxch = NULL; //Most common char word_str = word->best_choice->string ().string (); word_len = word->best_choice->lengths ().length ();; rep_ch = (REP_CH *) alloc_mem (word_len * sizeof (REP_CH)); for (i = 0, offset = 0; i < word_len; offset += word->best_choice->lengths()[i++]) { for (j = 0; j < rep_ch_count && strncmp(rep_ch[j].ch, word_str + offset, word->best_choice->lengths()[i]) != 0; j++); if (j < rep_ch_count) rep_ch[j].count++; else { strncpy(rep_ch[rep_ch_count].ch, word_str + offset, word->best_choice->lengths()[i]); rep_ch[rep_ch_count].ch[word->best_choice->lengths()[i]] = '\0'; rep_ch[rep_ch_count].count = 1; rep_ch_count++; } } for (j = 0; j < rep_ch_count; j++) { total += rep_ch[j].count; if ((rep_ch[j].count > max) && (*rep_ch[j].ch != ' ')) { max = rep_ch[j].count; maxch = rep_ch[j].ch; } } // tprintf( "REPEATED CHAR %s len=%d total=%d choice=%c\n", // word_str, word_len, total, maxch ); free_mem(rep_ch); word->reject_map.initialise (word_len); for (i = 0, offset = 0; i < word_len; offset += word->best_choice->lengths()[i++]) { if (strncmp(word_str + offset, maxch, word->best_choice->lengths()[i]) != 0) //rej unrecognised blobs word->reject_map[i].setrej_bad_repetition (); } word->done = TRUE; } // TODO(tkielbus) Decide between keeping this behavior here or modifying the // training data. // Utility function for fix_quotes // Return true if the next character in the string (given the UTF8 length in // bytes) is a quote character. static int is_simple_quote(const char* signed_str, int length) { const unsigned char* str = reinterpret_cast(signed_str); //standard 1 byte quotes return (length == 1 && (*str == '\'' || *str == '`')) || //utf8 3 bytes curved quotes (length == 3 && ((*str == 0xe2 && *(str + 1) == 0x80 && *(str + 2) == 0x98) || (*str == 0xe2 && *(str + 1) == 0x80 && *(str + 2) == 0x99))); } /********************************************************************** * fix_quotes * * Change pairs of quotes to double quotes. **********************************************************************/ void fix_quotes( //make double quotes WERD_CHOICE *choice, //choice to fix WERD *word, //word to do //char choices BLOB_CHOICE_LIST_CLIST *blob_choices) { char *str = (char *) choice->string().string();//string ptr int i; int offset; //blobs PBLOB_IT blob_it = word->blob_list (); //choices BLOB_CHOICE_LIST_C_IT choice_it = blob_choices; BLOB_CHOICE_IT it1; //first choices BLOB_CHOICE_IT it2; //second choices for (i = 0, offset = 0; str[offset] != '\0'; offset += choice->lengths()[i++], blob_it.forward (), choice_it.forward ()) { if (str[offset + choice->lengths()[i]] != '\0' && is_simple_quote(str + offset, choice->lengths()[i]) && is_simple_quote(str + offset + choice->lengths()[i], choice->lengths()[i + 1])) { str[offset] = '"'; //turn to double strcpy (str + offset + 1, str + offset + choice->lengths()[i] + choice->lengths()[i + 1]); //shuffle up choice->lengths()[i] = 1; strcpy ((char*) choice->lengths().string() + i + 1, choice->lengths().string() + i + 2); merge_blobs (blob_it.data (), blob_it.data_relative (1)); blob_it.forward (); delete blob_it.extract (); //get rid of spare it1.set_to_list (choice_it.data ()); it2.set_to_list (choice_it.data_relative (1)); if (it1.data ()->certainty () < it2.data ()->certainty ()) { choice_it.forward (); //get rid of spare delete choice_it.extract (); } else { //get rid of spare delete choice_it.extract (); choice_it.forward (); } } } } /********************************************************************** * fix_hyphens * * Change pairs of hyphens to a single hyphen if the bounding boxes touch * Typically a long dash which has been segmented. **********************************************************************/ void fix_hyphens( //crunch double hyphens WERD_CHOICE *choice, //choice to fix WERD *word, //word to do //char choices BLOB_CHOICE_LIST_CLIST *blob_choices) { char *str = (char *) choice->string().string();//string ptr int i; int offset; //blobs PBLOB_IT blob_it = word->blob_list (); //choices BLOB_CHOICE_LIST_C_IT choice_it = blob_choices; BLOB_CHOICE_IT it1; //first choices BLOB_CHOICE_IT it2; //second choices for (i = 0, offset = 0; str[offset] != '\0'; offset += choice->lengths()[i++], blob_it.forward (), choice_it.forward ()) { if ((str[offset] == '-' || str[offset] == '~') && (str[offset + choice->lengths()[i]] == '-' || str[offset + choice->lengths()[i]] == '~') && (blob_it.data ()->bounding_box ().right () >= blob_it.data_relative (1)->bounding_box ().left ())) { str[offset] = '-'; //turn to single hyphen strcpy (str + offset + choice->lengths()[i], str + offset + choice->lengths()[i] + choice->lengths()[i + 1]); //shuffle up strcpy ((char*) choice->lengths().string() + i + 1, choice->lengths().string() + i + 2); merge_blobs (blob_it.data (), blob_it.data_relative (1)); blob_it.forward (); delete blob_it.extract (); //get rid of spare it1.set_to_list (choice_it.data ()); it2.set_to_list (choice_it.data_relative (1)); if (it1.data ()->certainty () < it2.data ()->certainty ()) { choice_it.forward (); //get rid of spare delete choice_it.extract (); } else { //get rid of spare delete choice_it.extract (); choice_it.forward (); } } } } /********************************************************************** * merge_blobs * * Add the outlines from blob2 to blob1. Blob2 is emptied but not deleted. **********************************************************************/ void merge_blobs( //combine 2 blobs PBLOB *blob1, //dest blob PBLOB *blob2 //source blob ) { OUTLINE_IT outline_it = blob1->out_list (); //iterator outline_it.move_to_last (); //go to end //do it outline_it.add_list_after (blob2->out_list ()); } /********************************************************************** * choice_dump_tester * * Matcher tester function which generates .chc file entries. * Called via test_segment_pass2 for every blob tested by tess in a word. * (But only for words for which a correct segmentation could be found.) **********************************************************************/ void choice_dump_tester( //dump chars in word PBLOB *, //blob DENORM *, //de-normaliser BOOL8 correct, //ly segmented char *text, //correct text INT32 count, //chars in text BLOB_CHOICE_LIST *ratings //list of results ) { STRING choice_file_name; BLOB_CHOICE *blob_choice; BLOB_CHOICE_IT it; char source_chars[20]; char correct_char[3]; if (choice_file == NULL) { choice_file_name = imagebasename + ".chc"; if (!(choice_file = fopen (choice_file_name.string (), "w"))) { CANTOPENFILE.error ("choice_dump_tester", EXIT, "%s %d", choice_file_name.string (), errno); } } if ((count == 0) || (text == NULL) || (text[0] == '\0')) { strcpy (source_chars, "$$"); strcpy (correct_char, "$$"); } else { strncpy(source_chars, text, count); source_chars[count] = '\0'; if (correct) { correct_char[0] = text[0]; correct_char[1] = '\0'; } else { strcpy (correct_char, "$$"); } } fprintf (choice_file, "%s\t%s", source_chars, correct_char); it.set_to_list (ratings); for (it.mark_cycle_pt (); !it.cycled_list (); it.forward ()) { blob_choice = it.data (); fprintf (choice_file, "\t%s\t%f\t%f", blob_choice->unichar (), blob_choice->rating (), blob_choice->certainty ()); } fprintf (choice_file, "\n"); } /************************************************************************* * make_bln_copy() * * Generate a baseline normalised copy of the source word. The copy is done so * that whatever format the original word is in, a polygonal bln version is * generated as output. *************************************************************************/ WERD *make_bln_copy(WERD *src_word, ROW *row, float x_height, DENORM *denorm) { WERD *result; // if (wordit_linearc && !src_word->flag(W_POLYGON)) // { // larc_word = src_word->larc_copy( row->x_height() ); // result = larc_word->poly_copy( row->x_height() ); // delete larc_word; // } // else result = src_word->poly_copy (row->x_height ()); // if (tessedit_draw_words) // { // if ( la_win == NO_WINDOW ) // create_la_win(); // result->plot( la_win ); // } result->baseline_normalise_x (row, x_height, denorm); return result; } ACCEPTABLE_WERD_TYPE acceptable_word_string(const char *s, const char *lengths) { int i = 0; int offset = 0; int leading_punct_count; int upper_count = 0; int hyphen_pos = -1; ACCEPTABLE_WERD_TYPE word_type = AC_UNACCEPTABLE; if (strlen (lengths) > 20) return word_type; /* Single Leading punctuation char*/ if ((s[offset] != '\0') && (STRING (chs_leading_punct).contains (s[offset]))) offset += lengths[i++]; leading_punct_count = i; /* Initial cap */ while ((s[offset] != '\0') && unicharset.get_isupper(s + offset, lengths[i])) { offset += lengths[i++]; upper_count++; } if (upper_count > 1) word_type = AC_UPPER_CASE; else { /* Lower case word, possibly with an initial cap */ while ((s[offset] != '\0') && unicharset.get_islower (s + offset, lengths[i])) { offset += lengths[i++]; } if (i - leading_punct_count < quality_min_initial_alphas_reqd) goto not_a_word; /* Allow a single hyphen in a lower case word - dont trust upper case - I've seen several cases of "H" -> "I-I" */ if (lengths[i] == 1 && s[offset] == '-') { hyphen_pos = i; offset += lengths[i++]; if (s[offset] != '\0') { while ((s[offset] != '\0') && unicharset.get_islower(s + offset, lengths[i])) { offset += lengths[i++]; } if (i < hyphen_pos + 3) goto not_a_word; } } else { /* Allow "'s" in NON hyphenated lower case words */ if (lengths[i] == 1 && (s[offset] == '\'') && lengths[i + 1] == 1 && (s[offset + lengths[i]] == 's')) { offset += lengths[i++]; offset += lengths[i++]; } } if (upper_count > 0) word_type = AC_INITIAL_CAP; else word_type = AC_LOWER_CASE; } /* Up to two different, constrained trailing punctuation chars */ if (lengths[i] == 1 && (s[offset] != '\0') && (STRING (chs_trailing_punct1).contains (s[offset]))) offset += lengths[i++]; if (lengths[i] == 1 && (s[offset] != '\0') && i > 0 && (s[offset - lengths[i - 1]] != s[offset]) && (STRING (chs_trailing_punct2).contains (s[offset]))) offset += lengths[i++]; if (s[offset] != '\0') word_type = AC_UNACCEPTABLE; not_a_word: if (word_type == AC_UNACCEPTABLE) { /* Look for abbreviation string */ i = 0; offset = 0; if (s[0] != '\0' && unicharset.get_isupper (s, lengths[0])) { word_type = AC_UC_ABBREV; while ((s[offset] != '\0') && unicharset.get_isupper(s + offset, lengths[i]) && (lengths[i + 1] == 1 && s[offset + lengths[i]] == '.')) { offset += lengths[i++]; offset += lengths[i++]; } } else if (s[0] != '\0' && unicharset.get_islower (s, lengths[0])) { word_type = AC_LC_ABBREV; while ((s[offset] != '\0') && unicharset.get_islower(s + offset, lengths[i]) && (lengths[i + 1] == 1 && s[offset + lengths[i]] == '.')) { offset += lengths[i++]; offset += lengths[i++]; } } if (s[offset] != '\0') word_type = AC_UNACCEPTABLE; } return word_type; } /* DEBUGGING ROUTINE */ BOOL8 check_debug_pt(WERD_RES *word, int location) { BOOL8 show_map_detail = FALSE; INT16 i; #ifndef SECURE_NAMES if (!test_pt) return FALSE; tessedit_rejection_debug.set_value (FALSE); debug_x_ht_level.set_value (0); tessedit_cluster_debug.set_value (FALSE); nn_debug.set_value (FALSE); nn_reject_debug.set_value (FALSE); if (word->word->bounding_box ().contains (FCOORD (test_pt_x, test_pt_y))) { if (location < 0) return TRUE; //For breakpoint use tessedit_rejection_debug.set_value (TRUE); debug_x_ht_level.set_value (20); tessedit_cluster_debug.set_value (TRUE); nn_debug.set_value (TRUE); nn_reject_debug.set_value (TRUE); tprintf ("\n\nTESTWD::"); switch (location) { case 0: tprintf ("classify_word_pass1 start\n"); word->word->print (debug_fp); break; case 10: tprintf ("make_reject_map: initial map"); break; case 20: tprintf ("make_reject_map: after NN"); break; case 30: tprintf ("classify_word_pass2 - START"); break; case 40: tprintf ("classify_word_pass2 - Pre Xht"); break; case 50: tprintf ("classify_word_pass2 - END"); show_map_detail = TRUE; break; case 60: tprintf ("fixspace"); break; case 70: tprintf ("MM pass START"); break; case 80: tprintf ("MM pass END"); break; case 90: tprintf ("After Poor quality rejection"); break; case 100: tprintf ("unrej_good_quality_words - START"); break; case 110: tprintf ("unrej_good_quality_words - END"); break; case 120: tprintf ("Write results pass"); show_map_detail = TRUE; break; } tprintf (" \"%s\" ", word->best_choice->string ().string ()); word->reject_map.print (debug_fp); tprintf ("\n"); if (show_map_detail) { tprintf ("\"%s\"\n", word->best_choice->string ().string ()); for (i = 0; word->best_choice->string ()[i] != '\0'; i++) { tprintf ("**** \"%c\" ****\n", word->best_choice->string ()[i]); word->reject_map[i].full_print (debug_fp); } } tprintf ("Tess Accepted: %s\n", word->tess_accepted ? "TRUE" : "FALSE"); tprintf ("Done flag: %s\n\n", word->done ? "TRUE" : "FALSE"); return TRUE; } else #endif return FALSE; } /********************************************************************** * set_word_fonts * * Get the fonts for the word. **********************************************************************/ void set_word_fonts( //good chars in word WERD_RES *word, //word to adapt to //detailed results BLOB_CHOICE_LIST_CLIST *blob_choices) { INT32 index; //char index INT32 offset; //char offset char choice_char[UNICHAR_LEN + 1]; //char from word INT8 config; //font of char //character iterator BLOB_CHOICE_LIST_C_IT char_it = blob_choices; BLOB_CHOICE_IT choice_it; //choice iterator STATS fonts (0, 32); //font counters static INT8 italic_table[32] = { 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1, 1, -1 }; static INT8 bold_table[32] = { 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1, 1, 1, -1, -1 }; static INT8 font_table[32] = { 2, 2, 2, 2, -1, -1, -1, -1, 0, 0, 0, 0, 1, 1, 1, 1, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 2, 2, 2, 2 }; word->italic = 0; word->bold = 0; for (char_it.mark_cycle_pt (), index = 0, offset = 0; !char_it.cycled_list (); char_it.forward (), offset += word->best_choice->lengths()[index++]) { strncpy(choice_char, word->best_choice->string ().string() + offset, word->best_choice->lengths()[index]); choice_char[word->best_choice->lengths()[index]] = '\0'; choice_it.set_to_list (char_it.data ()); for (choice_it.mark_cycle_pt (); !choice_it.cycled_list (); choice_it.forward ()) { if (strcmp(choice_it.data ()->unichar (), choice_char) == 0) { config = choice_it.data ()->config (); if (tessedit_debug_fonts) tprintf ("%s(%d=%d%c%c)", choice_char, config, (config & 31) >> 2, config & 2 ? 'N' : 'B', config & 1 ? 'N' : 'I'); if (config != -1) { config &= 31; word->italic += italic_table[config]; word->bold += bold_table[config]; if (font_table[config] != -1) fonts.add (font_table[config], 1); } break; } } } find_modal_font (&fonts, &word->font1, &word->font1_count); find_modal_font (&fonts, &word->font2, &word->font2_count); if (tessedit_debug_fonts) tprintf ("\n"); /* if (word->font1_count>0) { for (char_it.mark_cycle_pt(),index=0; !char_it.cycled_list();char_it.forward(),index++) { choice_char=word->best_choice->string()[index]; choice_it.set_to_list(char_it.data()); for (choice_it.mark_cycle_pt();!choice_it.cycled_list();choice_it.forward()) { if (choice_it.data()->char_class()==choice_char) { config=choice_it.data()->config(); if (config!=-1 && font_table[config&31]==word->font1) { word->italic+=italic_table[config]; word->bold+=bold_table[config]; } break; } } } }*/ } /********************************************************************** * font_recognition_pass * * Smooth the fonts for the document. **********************************************************************/ void font_recognition_pass( //good chars in word PAGE_RES_IT &page_res_it) { INT32 length; //of word INT32 count; //of a feature INT8 doc_font; //modal font INT8 doc_font_count; //modal font INT32 doc_italic; //total italics INT32 doc_bold; //total bolds ROW_RES *row = NULL; //current row WERD_RES *word; //current word STATS fonts (0, 32); //font counters STATS doc_fonts (0, 32); //font counters doc_italic = 0; doc_bold = 0; page_res_it.restart_page (); while (page_res_it.word () != NULL) { if (row != page_res_it.row ()) { if (row != NULL) { find_modal_font (&fonts, &row->font1, &row->font1_count); find_modal_font (&fonts, &row->font2, &row->font2_count); } row = page_res_it.row (); //current row fonts.clear (); //clear counters row->italic = 0; row->bold = 0; } word = page_res_it.word (); row->italic += word->italic; row->bold += word->bold; fonts.add (word->font1, word->font1_count); fonts.add (word->font2, word->font2_count); doc_italic += word->italic; doc_bold += word->bold; doc_fonts.add (word->font1, word->font1_count); doc_fonts.add (word->font2, word->font2_count); page_res_it.forward (); } if (row != NULL) { find_modal_font (&fonts, &row->font1, &row->font1_count); find_modal_font (&fonts, &row->font2, &row->font2_count); } find_modal_font(&doc_fonts, &doc_font, &doc_font_count); /* row=NULL; page_res_it.restart_page(); while (page_res_it.word() != NULL) { if (row!=page_res_it.row()) { row2=row; row=page_res_it.row(); if (row->font1_countrow->x_height()-row2->row->x_height(); if (hdiff<0) hdiff=-hdiff; if (hdiffrow->x_height()-row2->row->x_height(); if (hdiff<0) hdiff=-hdiff; if (hdiffitalic=italic; row->bold=bold; find_modal_font(&fonts,&row->font1,&row->font1_count); find_modal_font(&fonts,&row->font2,&row->font2_count); } else page_res_it.forward(); } else page_res_it.forward(); }*/ page_res_it.restart_page (); while (page_res_it.word () != NULL) { row = page_res_it.row (); //current row word = page_res_it.word (); length = word->best_choice->string ().length (); count = word->italic; if (count < 0) count = -count; if (!(count == length || length > 3 && count >= length * 3 / 4)) word->italic = doc_italic > 0 ? 1 : -1; count = word->bold; if (count < 0) count = -count; if (!(count == length || length > 3 && count >= length * 3 / 4)) word->bold = doc_bold > 0 ? 1 : -1; count = word->font1_count; if (!(count == length || length > 3 && count >= length * 3 / 4)) { word->font1 = doc_font; word->font1_count = doc_font_count; } page_res_it.forward (); } } /********************************************************************** * add_in_one_row * * Add into the stats for one row. **********************************************************************/ void add_in_one_row( //good chars in word ROW_RES *row, //current row STATS *fonts, //font stats INT8 *italic, //output count INT8 *bold //output count ) { WERD_RES *word; //current word WERD_RES_IT word_it = &row->word_res_list; for (word_it.mark_cycle_pt (); !word_it.cycled_list (); word_it.forward ()) { word = word_it.data (); *italic += word->italic; *bold += word->bold; if (word->font1_count > 0) fonts->add (word->font1, word->font1_count); if (word->font2_count > 0) fonts->add (word->font2, word->font2_count); } } /********************************************************************** * find_modal_font * * Find the modal font and remove from the stats. **********************************************************************/ void find_modal_font( //good chars in word STATS *fonts, //font stats INT8 *font_out, //output font INT8 *font_count //output count ) { INT8 font; //font index INT32 count; //pile couat if (fonts->get_total () > 0) { font = (INT8) fonts->mode (); *font_out = font; count = fonts->pile_count (font); *font_count = count < MAX_INT8 ? count : MAX_INT8; fonts->add (font, -*font_count); } else { *font_out = -1; *font_count = 0; } }