combine_tessdata.cpp File Reference

#include "commontraining.h"
#include "lstmrecognizer.h"
#include "tessdatamanager.h"
#include <cerrno>
#include <iostream>

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 117 of file combine_tessdata.cpp.

                                {
  tesseract::CheckSharedLibraryVersion();
 
  int i;
  tesseract::TessdataManager tm;
  if (argc > 1 && (!strcmp(argv[1], "-v") || !strcmp(argv[1], "--version"))) {
    printf("%s\n", tesseract::TessBaseAPI::Version());
    return EXIT_SUCCESS;
  } else if (argc == 2) {
    printf("Combining tessdata files\n");
    std::string lang = argv[1];
    char *last = &argv[1][strlen(argv[1]) - 1];
    if (*last != '.') {
      lang += '.';
    }
    std::string output_file = lang;
    output_file += kTrainedDataSuffix;
    if (!tm.CombineDataFiles(lang.c_str(), output_file.c_str())) {
      printf("Error combining tessdata files into %s\n", output_file.c_str());
    } else {
      printf("Output %s created successfully.\n", output_file.c_str());
    }
  } else if (argc >= 4 &&
             (strcmp(argv[1], "-e") == 0 || strcmp(argv[1], "-u") == 0)) {
    // Initialize TessdataManager with the data in the given traineddata file.
    if (!tm.Init(argv[2])) {
      tprintf("Failed to read %s\n", argv[2]);
      return EXIT_FAILURE;
    }
    printf("Extracting tessdata components from %s\n", argv[2]);
    if (strcmp(argv[1], "-e") == 0) {
      for (i = 3; i < argc; ++i) {
        errno = 0;
        if (tm.ExtractToFile(argv[i])) {
          printf("Wrote %s\n", argv[i]);
        } else if (errno == 0) {
          printf(
              "Not extracting %s, since this component"
              " is not present\n",
              argv[i]);
          return EXIT_FAILURE;
        } else {
          printf("Error, could not extract %s: %s\n", argv[i], strerror(errno));
          return EXIT_FAILURE;
        }
      }
    } else { // extract all the components
      for (i = 0; i < tesseract::TESSDATA_NUM_ENTRIES; ++i) {
        std::string filename = argv[3];
        char *last = &argv[3][strlen(argv[3]) - 1];
        if (*last != '.') {
          filename += '.';
        }
        filename += tesseract::kTessdataFileSuffixes[i];
        errno = 0;
        if (tm.ExtractToFile(filename.c_str())) {
          printf("Wrote %s\n", filename.c_str());
        } else if (errno != 0) {
          printf("Error, could not extract %s: %s\n", filename.c_str(),
                 strerror(errno));
          return EXIT_FAILURE;
        }
      }
    }
  } else if (argc >= 4 && strcmp(argv[1], "-o") == 0) {
    // Rename the current traineddata file to a temporary name.
    const char *new_traineddata_filename = argv[2];
    std::string traineddata_filename = new_traineddata_filename;
    traineddata_filename += ".__tmp__";
    if (rename(new_traineddata_filename, traineddata_filename.c_str()) != 0) {
      tprintf("Failed to create a temporary file %s\n",
              traineddata_filename.c_str());
      return EXIT_FAILURE;
    }
 
    // Initialize TessdataManager with the data in the given traineddata file.
    tm.Init(traineddata_filename.c_str());
 
    // Write the updated traineddata file.
    tm.OverwriteComponents(new_traineddata_filename, argv + 3, argc - 3);
  } else if (argc == 3 && strcmp(argv[1], "-c") == 0) {
    if (!tm.Init(argv[2])) {
      tprintf("Failed to read %s\n", argv[2]);
      return EXIT_FAILURE;
    }
    tesseract::TFile fp;
    if (!tm.GetComponent(tesseract::TESSDATA_LSTM, &fp)) {
      tprintf("No LSTM Component found in %s!\n", argv[2]);
      return EXIT_FAILURE;
    }
    tesseract::LSTMRecognizer recognizer;
    if (!recognizer.DeSerialize(&tm, &fp)) {
      tprintf("Failed to deserialize LSTM in %s!\n", argv[2]);
      return EXIT_FAILURE;
    }
    recognizer.ConvertToInt();
    std::vector<char> lstm_data;
    fp.OpenWrite(&lstm_data);
    ASSERT_HOST(recognizer.Serialize(&tm, &fp));
    tm.OverwriteEntry(tesseract::TESSDATA_LSTM, &lstm_data[0],
                      lstm_data.size());
    if (!tm.SaveFile(argv[2], nullptr)) {
      tprintf("Failed to write modified traineddata:%s!\n", argv[2]);
      return EXIT_FAILURE;
    }
  } else if (argc == 3 && strcmp(argv[1], "-d") == 0) {
    return list_components(tm, argv[2]);
  } else if (argc == 3 && strcmp(argv[1], "-l") == 0) {
    return list_network(tm, argv[2]);
  } else if (argc == 3 && strcmp(argv[1], "-dl") == 0) {
    int result = list_components(tm, argv[2]);
    if (result == EXIT_SUCCESS) {
      result = list_network(tm, nullptr);
    }
    return result;
  } else if (argc == 3 && strcmp(argv[1], "-ld") == 0) {
    int result = list_network(tm, argv[2]);
    if (result == EXIT_SUCCESS) {
      result = list_components(tm, nullptr);
    }
    return result;
  } else {
    printf(
        "Usage for combining tessdata components:\n"
        "  %s language_data_path_prefix\n"
        "  (e.g. %s tessdata/eng.)\n\n",
        argv[0], argv[0]);
    printf(
        "Usage for extracting tessdata components:\n"
        "  %s -e traineddata_file [output_component_file...]\n"
        "  (e.g. %s -e eng.traineddata eng.unicharset)\n\n",
        argv[0], argv[0]);
    printf(
        "Usage for overwriting tessdata components:\n"
        "  %s -o traineddata_file [input_component_file...]\n"
        "  (e.g. %s -o eng.traineddata eng.unicharset)\n\n",
        argv[0], argv[0]);
    printf(
        "Usage for unpacking all tessdata components:\n"
        "  %s -u traineddata_file output_path_prefix\n"
        "  (e.g. %s -u eng.traineddata tmp/eng.)\n\n",
        argv[0], argv[0]);
    printf(
        "Usage for listing the network information\n"
        "  %s -l traineddata_file\n"
        "  (e.g. %s -l eng.traineddata)\n\n",
        argv[0], argv[0]);
    printf(
        "Usage for listing directory of components:\n"
        "  %s -d traineddata_file\n\n",
        argv[0]);
    printf(
        "Usage for compacting LSTM component to int:\n"
        "  %s -c traineddata_file\n",
        argv[0]);
    return EXIT_FAILURE;
  }
  tm.Directory();
  return EXIT_SUCCESS;
}