mirror of
https://github.com/tesseract-ocr/tesseract.git
synced 2024-12-11 23:19:04 +08:00
9c25eda469
git-svn-id: https://tesseract-ocr.googlecode.com/svn/trunk@936 d0cd1f9f-072b-0410-8dd7-cf729c803f20
309 lines
10 KiB
C++
309 lines
10 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 automatically generated configuration file if running autoconf
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config_auto.h"
|
|
#endif
|
|
|
|
#include "allheaders.h"
|
|
#include "baseapi.h"
|
|
#include "renderer.h"
|
|
#include "strngs.h"
|
|
#include "tprintf.h"
|
|
#include "openclwrapper.h"
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#ifdef _WIN32
|
|
#include <fcntl.h>
|
|
#include <io.h>
|
|
#endif // _WIN32
|
|
|
|
/**********************************************************************
|
|
* main()
|
|
*
|
|
**********************************************************************/
|
|
|
|
int main(int argc, char **argv) {
|
|
if ((argc == 2 && strcmp(argv[1], "-v") == 0) ||
|
|
(argc == 2 && strcmp(argv[1], "--version") == 0)) {
|
|
char *versionStrP;
|
|
|
|
fprintf(stderr, "tesseract %s\n", tesseract::TessBaseAPI::Version());
|
|
|
|
versionStrP = getLeptonicaVersion();
|
|
fprintf(stderr, " %s\n", versionStrP);
|
|
lept_free(versionStrP);
|
|
|
|
versionStrP = getImagelibVersions();
|
|
fprintf(stderr, " %s\n", versionStrP);
|
|
lept_free(versionStrP);
|
|
|
|
#ifdef USE_OPENCL
|
|
cl_platform_id platform;
|
|
cl_uint num_platforms;
|
|
cl_device_id devices[2];
|
|
cl_uint num_devices;
|
|
cl_int err;
|
|
char info[256];
|
|
int i;
|
|
|
|
fprintf(stderr, " OpenCL info:\n");
|
|
clGetPlatformIDs(1, &platform, &num_platforms);
|
|
fprintf(stderr, " Found %d platforms.\n", num_platforms);
|
|
clGetPlatformInfo(platform, CL_PLATFORM_NAME, 256, info, 0);
|
|
fprintf(stderr, " Platform name: %s.\n", info);
|
|
clGetPlatformInfo(platform, CL_PLATFORM_VERSION, 256, info, 0);
|
|
fprintf(stderr, " Version: %s.\n", info);
|
|
clGetDeviceIDs(platform, CL_DEVICE_TYPE_ALL, 2, devices, &num_devices);
|
|
fprintf(stderr, " Found %d devices.\n", num_devices);
|
|
for (i = 0; i < num_devices; ++i) {
|
|
clGetDeviceInfo(devices[i], CL_DEVICE_NAME, 256, info, 0);
|
|
fprintf(stderr, " Device %d name: %s.\n", i+1, info);
|
|
}
|
|
#endif
|
|
exit(0);
|
|
}
|
|
|
|
// Make the order of args a bit more forgiving than it used to be.
|
|
const char* lang = "eng";
|
|
const char* image = NULL;
|
|
const char* output = NULL;
|
|
const char* datapath = NULL;
|
|
bool noocr = false;
|
|
bool list_langs = false;
|
|
bool print_parameters = false;
|
|
|
|
tesseract::PageSegMode pagesegmode = tesseract::PSM_AUTO;
|
|
int arg = 1;
|
|
while (arg < argc && (output == NULL || argv[arg][0] == '-')) {
|
|
if (strcmp(argv[arg], "-l") == 0 && arg + 1 < argc) {
|
|
lang = argv[arg + 1];
|
|
++arg;
|
|
} else if (strcmp(argv[arg], "--tessdata-dir") == 0 && arg + 1 < argc) {
|
|
datapath = argv[arg + 1];
|
|
++arg;
|
|
} else if (strcmp(argv[arg], "--list-langs") == 0) {
|
|
noocr = true;
|
|
list_langs = true;
|
|
} else if (strcmp(argv[arg], "-psm") == 0 && arg + 1 < argc) {
|
|
pagesegmode = static_cast<tesseract::PageSegMode>(atoi(argv[arg + 1]));
|
|
++arg;
|
|
} else if (strcmp(argv[arg], "--print-parameters") == 0) {
|
|
noocr = true;
|
|
print_parameters = true;
|
|
} else if (strcmp(argv[arg], "-c") == 0 && arg + 1 < argc) {
|
|
// handled properly after api init
|
|
++arg;
|
|
} else if (image == NULL) {
|
|
image = argv[arg];
|
|
} else if (output == NULL) {
|
|
output = argv[arg];
|
|
}
|
|
++arg;
|
|
}
|
|
|
|
if (argc == 2 && strcmp(argv[1], "--list-langs") == 0) {
|
|
list_langs = true;
|
|
noocr = true;
|
|
}
|
|
|
|
if (output == NULL && noocr == false) {
|
|
fprintf(stderr, "Usage:\n %s imagename|stdin outputbase|stdout [options...] "
|
|
"[configfile...]\n\n", argv[0]);
|
|
|
|
fprintf(stderr, "OCR options:\n");
|
|
fprintf(stderr, " --tessdata-dir /path\tspecify location of tessdata"
|
|
" path\n");
|
|
fprintf(stderr, " -l lang[+lang]\tspecify language(s) used for OCR\n");
|
|
fprintf(stderr, " -c configvar=value\tset value for control parameter.\n"
|
|
"\t\t\tMultiple -c arguments are allowed.\n");
|
|
fprintf(stderr, " -psm pagesegmode\tspecify page segmentation mode.\n");
|
|
fprintf(stderr, "These options must occur before any configfile.\n\n");
|
|
fprintf(stderr,
|
|
"pagesegmode values are:\n"
|
|
" 0 = Orientation and script detection (OSD) only.\n"
|
|
" 1 = Automatic page segmentation with OSD.\n"
|
|
" 2 = Automatic page segmentation, but no OSD, or OCR\n"
|
|
" 3 = Fully automatic page segmentation, but no OSD. (Default)\n"
|
|
" 4 = Assume a single column of text of variable sizes.\n"
|
|
" 5 = Assume a single uniform block of vertically aligned text.\n"
|
|
" 6 = Assume a single uniform block of text.\n"
|
|
" 7 = Treat the image as a single text line.\n"
|
|
" 8 = Treat the image as a single word.\n"
|
|
" 9 = Treat the image as a single word in a circle.\n"
|
|
" 10 = Treat the image as a single character.\n\n");
|
|
fprintf(stderr, "Single options:\n");
|
|
fprintf(stderr, " -v --version: version info\n");
|
|
fprintf(stderr, " --list-langs: list available languages for tesseract "
|
|
"engine. Can be used with --tessdata-dir.\n");
|
|
fprintf(stderr, " --print-parameters: print tesseract parameters to the "
|
|
"stdout.\n");
|
|
exit(1);
|
|
}
|
|
|
|
tprintf("Tesseract Open Source OCR Engine v%s with Leptonica\n",
|
|
tesseract::TessBaseAPI::Version());
|
|
PERF_COUNT_START("Tesseract:main")
|
|
tesseract::TessBaseAPI api;
|
|
|
|
api.SetOutputName(output);
|
|
int rc = api.Init(datapath, lang, tesseract::OEM_DEFAULT,
|
|
&(argv[arg]), argc - arg, NULL, NULL, false);
|
|
|
|
if (rc) {
|
|
fprintf(stderr, "Could not initialize tesseract.\n");
|
|
exit(1);
|
|
}
|
|
|
|
char opt1[255], opt2[255];
|
|
for (arg = 0; arg < argc; arg++) {
|
|
if (strcmp(argv[arg], "-c") == 0 && arg + 1 < argc) {
|
|
strncpy(opt1, argv[arg + 1], 255);
|
|
*(strchr(opt1, '=')) = 0;
|
|
strncpy(opt2, strchr(argv[arg + 1], '=') + 1, 255);
|
|
opt2[254] = 0;
|
|
++arg;
|
|
|
|
if (!api.SetVariable(opt1, opt2)) {
|
|
fprintf(stderr, "Could not set option: %s=%s\n", opt1, opt2);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (list_langs) {
|
|
GenericVector<STRING> languages;
|
|
api.GetAvailableLanguagesAsVector(&languages);
|
|
fprintf(stderr, "List of available languages (%d):\n",
|
|
languages.size());
|
|
for (int index = 0; index < languages.size(); ++index) {
|
|
STRING& string = languages[index];
|
|
fprintf(stderr, "%s\n", string.string());
|
|
}
|
|
api.End();
|
|
exit(0);
|
|
}
|
|
|
|
if (print_parameters) {
|
|
FILE* fout = stdout;
|
|
fprintf(stdout, "Tesseract parameters:\n");
|
|
api.PrintVariables(fout);
|
|
api.End();
|
|
exit(0);
|
|
}
|
|
|
|
// We have 2 possible sources of pagesegmode: a config file and
|
|
// the command line. For backwards compatability reasons, the
|
|
// default in tesseract is tesseract::PSM_SINGLE_BLOCK, but the
|
|
// default for this program is tesseract::PSM_AUTO. We will let
|
|
// the config file take priority, so the command-line default
|
|
// can take priority over the tesseract default, so we use the
|
|
// value from the command line only if the retrieved mode
|
|
// is still tesseract::PSM_SINGLE_BLOCK, indicating no change
|
|
// in any config file. Therefore the only way to force
|
|
// tesseract::PSM_SINGLE_BLOCK is from the command line.
|
|
// It would be simpler if we could set the value before Init,
|
|
// but that doesn't work.
|
|
if (api.GetPageSegMode() == tesseract::PSM_SINGLE_BLOCK)
|
|
api.SetPageSegMode(pagesegmode);
|
|
|
|
tesseract::TessResultRenderer* renderer = NULL;
|
|
bool b;
|
|
api.GetBoolVariable("tessedit_create_hocr", &b);
|
|
if (b) renderer = new tesseract::TessHOcrRenderer();
|
|
|
|
api.GetBoolVariable("tessedit_create_boxfile", &b);
|
|
if (b) renderer = new tesseract::TessBoxTextRenderer();
|
|
|
|
if (renderer == NULL) renderer = new tesseract::TessTextRenderer();
|
|
|
|
bool stdInput = false;
|
|
if (!strcmp(image, "stdin") || !strcmp(image, "-"))
|
|
stdInput = true;
|
|
|
|
if (stdInput) {
|
|
char byt;
|
|
PIX *pixd = NULL;
|
|
std::vector<char> ch_data;
|
|
std::istream file(std::cin.rdbuf());
|
|
|
|
#ifdef WIN32
|
|
if (_setmode(_fileno(stdin), _O_BINARY) == -1)
|
|
tprintf("ERROR: cin to binary: %s", strerror(errno));
|
|
#endif // WIN32
|
|
|
|
while (file.get(byt)) {
|
|
ch_data.push_back(byt);
|
|
}
|
|
std::cin.ignore(std::cin.rdbuf()->in_avail() + 1);
|
|
|
|
size_t size = ch_data.size();
|
|
l_uint8 *data;
|
|
if ( (data = (l_uint8 *) malloc( size )) != NULL ) {
|
|
memcpy(data, &(ch_data)[0], size);
|
|
} else {
|
|
tprintf("Memory allocation error\n");
|
|
exit(1);
|
|
}
|
|
|
|
pixd = pixReadMem(data, size);
|
|
api.ProcessPage(pixd, 0, NULL, NULL, 0, renderer);
|
|
} else {
|
|
FILE* fin = fopen(image, "rb");
|
|
if (fin == NULL) {
|
|
fprintf(stderr, "Cannot open input file: %s\n", image);
|
|
exit(2);
|
|
}
|
|
fclose(fin);
|
|
}
|
|
|
|
if (!stdInput && !api.ProcessPages(image, NULL, 0, renderer)) {
|
|
fprintf(stderr, "Error during processing.\n");
|
|
} else {
|
|
for (tesseract::TessResultRenderer* r = renderer; r != NULL;
|
|
r = r->next()) {
|
|
FILE* fout = stdout;
|
|
if (strcmp(output, "-") && strcmp(output, "stdout")) {
|
|
STRING outfile = STRING(output)
|
|
+ STRING(".")
|
|
+ STRING(r->file_extension());
|
|
fout = fopen(outfile.string(), "wb");
|
|
if (fout == NULL) {
|
|
fprintf(stderr, "Cannot create output file %s\n", outfile.string());
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
const char* data;
|
|
inT32 data_len;
|
|
if (r->GetOutput(&data, &data_len)) {
|
|
fwrite(data, 1, data_len, fout);
|
|
if (fout != stdout)
|
|
fclose(fout);
|
|
else
|
|
clearerr(fout);
|
|
}
|
|
}
|
|
}
|
|
PERF_COUNT_END
|
|
return 0; // Normal exit
|
|
}
|