tesseract::TessdataManager Class Reference

#include <tessdatamanager.h>

Public Member Functions
	TessdataManager ()
	TessdataManager (FileReader reader)
	~TessdataManager ()=default
bool	swap () const
bool	is_loaded () const
void	LoadFileLater (const char *data_file_name)
bool	Init (const char *data_file_name)
bool	LoadMemBuffer (const char *name, const char *data, int size)
void	OverwriteEntry (TessdataType type, const char *data, int size)
bool	SaveFile (const STRING &filename, FileWriter writer) const
void	Serialize (GenericVector< char > *data) const
void	Clear ()
void	Directory () const
bool	IsComponentAvailable (TessdataType type) const
bool	GetComponent (TessdataType type, TFile *fp)
bool	GetComponent (TessdataType type, TFile *fp) const
std::string	VersionString () const
void	SetVersionString (const std::string &v_str)
bool	IsBaseAvailable () const
bool	IsLSTMAvailable () const
const STRING &	GetDataFileName () const
bool	CombineDataFiles (const char *language_data_path_prefix, const char *output_filename)
bool	OverwriteComponents (const char *new_traineddata_filename, char **component_filenames, int num_new_components)
bool	ExtractToFile (const char *filename)

Detailed Description

Definition at line 126 of file tessdatamanager.h.

Constructor & Destructor Documentation

tesseract::TessdataManager::TessdataManager

(

)

Definition at line 42 of file tessdatamanager.cpp.

                                 : reader_(nullptr), is_loaded_(false), swap_(false) {
  SetVersionString(PACKAGE_VERSION);
}

tesseract::TessdataManager::TessdataManager ( FileReader  reader )

explicit

Definition at line 46 of file tessdatamanager.cpp.

  : reader_(reader),
    is_loaded_(false),
    swap_(false) {
  SetVersionString(PACKAGE_VERSION);
}

tesseract::TessdataManager::~TessdataManager ( )

default

Member Function Documentation

void tesseract::TessdataManager::Clear

(

)

Definition at line 194 of file tessdatamanager.cpp.

                            {
  for (auto& entry : entries_) {
    entry.clear();
  }
  is_loaded_ = false;
}

bool tesseract::TessdataManager::CombineDataFiles	(	const char *	language_data_path_prefix,
		const char *	output_filename
	)

Reads all the standard tesseract config and data files for a language at the given path and bundles them up into one binary data file. Returns true if the combined traineddata file was successfully written.

Definition at line 244 of file tessdatamanager.cpp.

                                 {
  // Load individual tessdata components from files.
  for (auto filesuffix : kTessdataFileSuffixes) {
    TessdataType type;
    ASSERT_HOST(TessdataTypeFromFileSuffix(filesuffix, &type));
    STRING filename = language_data_path_prefix;
    filename += filesuffix;
    FILE *fp = fopen(filename.string(), "rb");
    if (fp != nullptr) {
      fclose(fp);
      if (!LoadDataFromFile(filename, &entries_[type])) {
        tprintf("Load of file %s failed!\n", filename.string());
        return false;
      }
    }
  }
  is_loaded_ = true;

  // Make sure that the required components are present.
  if (!IsBaseAvailable() && !IsLSTMAvailable()) {
    tprintf(
        "Error: traineddata file must contain at least (a unicharset file"
        "and inttemp) OR an lstm file.\n");
    return false;
  }
  // Write updated data to the output traineddata file.
  return SaveFile(output_filename, nullptr);
}

void tesseract::TessdataManager::Directory

(

)

const

Definition at line 202 of file tessdatamanager.cpp.

                                      {
  tprintf("Version string:%s\n", VersionString().c_str());
  int offset = TESSDATA_NUM_ENTRIES * sizeof(int64_t);
  for (unsigned i = 0; i < TESSDATA_NUM_ENTRIES; ++i) {
    if (!entries_[i].empty()) {
      tprintf("%d:%s:size=%d, offset=%d\n", i, kTessdataFileSuffixes[i],
              entries_[i].size(), offset);
      offset += entries_[i].size();
    }
  }
}

bool tesseract::TessdataManager::ExtractToFile

(

const char *

filename

)

Extracts tessdata component implied by the name of the input file from the combined traineddata loaded into TessdataManager. Writes the extracted component to the file indicated by the file name. E.g. if the filename given is somepath/somelang.unicharset, unicharset will be extracted from the data loaded into the TessdataManager and will be written to somepath/somelang.unicharset.

Returns

true if the component was successfully extracted, false if the component was not present in the traineddata loaded into TessdataManager.

Definition at line 295 of file tessdatamanager.cpp.

                                                        {
  TessdataType type = TESSDATA_NUM_ENTRIES;
  ASSERT_HOST(
      tesseract::TessdataManager::TessdataTypeFromFileName(filename, &type));
  if (entries_[type].empty()) return false;
  return SaveDataToFile(entries_[type], filename);
}

bool tesseract::TessdataManager::GetComponent	(	TessdataType	type,
		TFile *	fp
	)

Definition at line 216 of file tessdatamanager.cpp.

                                                               {
  if (!is_loaded_ && !Init(data_file_name_.string())) return false;
  const TessdataManager *const_this = this;
  return const_this->GetComponent(type, fp);
}

bool tesseract::TessdataManager::GetComponent	(	TessdataType	type,
		TFile *	fp
	)		const

Definition at line 224 of file tessdatamanager.cpp.

                                                                     {
  ASSERT_HOST(is_loaded_);
  if (entries_[type].empty()) return false;
  fp->Open(&entries_[type][0], entries_[type].size());
  fp->set_swap(swap_);
  return true;
}

const STRING& tesseract::TessdataManager::GetDataFileName ( ) const

inline

Definition at line 186 of file tessdatamanager.h.

{ return data_file_name_; }

bool tesseract::TessdataManager::Init

(

const char *

data_file_name

)

Opens and reads the given data file right now.

Returns

true on success.

Definition at line 97 of file tessdatamanager.cpp.

                                                     {
  GenericVector<char> data;
  if (reader_ == nullptr) {
#if defined(HAVE_LIBARCHIVE)
    if (LoadArchiveFile(data_file_name)) return true;
#endif
    if (!LoadDataFromFile(data_file_name, &data)) return false;
  } else {
    if (!(*reader_)(data_file_name, &data)) return false;
  }
  return LoadMemBuffer(data_file_name, &data[0], data.size());
}

bool tesseract::TessdataManager::is_loaded ( ) const

inline

Definition at line 134 of file tessdatamanager.h.

{ return is_loaded_; }

bool tesseract::TessdataManager::IsBaseAvailable ( ) const

inline

Definition at line 177 of file tessdatamanager.h.

                               {
    return !entries_[TESSDATA_UNICHARSET].empty() &&
           !entries_[TESSDATA_INTTEMP].empty();
  }

bool tesseract::TessdataManager::IsComponentAvailable ( TessdataType  type ) const

inline

Definition at line 161 of file tessdatamanager.h.

                                                     {
    return !entries_[type].empty();
  }

bool tesseract::TessdataManager::IsLSTMAvailable ( ) const

inline

Definition at line 183 of file tessdatamanager.h.

{ return !entries_[TESSDATA_LSTM].empty(); }

void tesseract::TessdataManager::LoadFileLater

(

const char *

data_file_name

)

Definition at line 55 of file tessdatamanager.cpp.

                                                              {
  Clear();
  data_file_name_ = data_file_name;
}

bool tesseract::TessdataManager::LoadMemBuffer	(	const char *	name,
		const char *	data,
		int	size
	)

Definition at line 111 of file tessdatamanager.cpp.

                                              {
  // TODO: This method supports only the proprietary file format.
  Clear();
  data_file_name_ = name;
  TFile fp;
  fp.Open(data, size);
  uint32_t num_entries;
  if (!fp.DeSerialize(&num_entries)) return false;
  swap_ = num_entries > kMaxNumTessdataEntries;
  fp.set_swap(swap_);
  if (swap_) ReverseN(&num_entries, sizeof(num_entries));
  if (num_entries > kMaxNumTessdataEntries) return false;
  GenericVector<int64_t> offset_table;
  offset_table.resize_no_init(num_entries);
  if (!fp.DeSerialize(&offset_table[0], num_entries)) return false;
  for (unsigned i = 0; i < num_entries && i < TESSDATA_NUM_ENTRIES; ++i) {
    if (offset_table[i] >= 0) {
      int64_t entry_size = size - offset_table[i];
      unsigned j = i + 1;
      while (j < num_entries && offset_table[j] == -1) ++j;
      if (j < num_entries) entry_size = offset_table[j] - offset_table[i];
      entries_[i].resize_no_init(entry_size);
      if (!fp.DeSerialize(&entries_[i][0], entry_size)) return false;
    }
  }
  if (entries_[TESSDATA_VERSION].empty()) {
    SetVersionString("Pre-4.0.0");
  }
  is_loaded_ = true;
  return true;
}

bool tesseract::TessdataManager::OverwriteComponents	(	const char *	new_traineddata_filename,
		char **	component_filenames,
		int	num_new_components
	)

Gets the individual components from the data_file_ with which the class was initialized. Overwrites the components specified by component_filenames. Writes the updated traineddata file to new_traineddata_filename.

Definition at line 275 of file tessdatamanager.cpp.

                            {
  // Open the files with the new components.
  // TODO: This method supports only the proprietary file format.
  for (int i = 0; i < num_new_components; ++i) {
    TessdataType type;
    if (TessdataTypeFromFileName(component_filenames[i], &type)) {
      if (!LoadDataFromFile(component_filenames[i], &entries_[type])) {
        tprintf("Failed to read component file:%s\n", component_filenames[i]);
        return false;
      }
    }
  }

  // Write updated data to the output traineddata file.
  return SaveFile(new_traineddata_filename, nullptr);
}

void tesseract::TessdataManager::OverwriteEntry	(	TessdataType	type,
		const char *	data,
		int	size
	)

Definition at line 145 of file tessdatamanager.cpp.

                                               {
  is_loaded_ = true;
  entries_[type].resize_no_init(size);
  memcpy(&entries_[type][0], data, size);
}

bool tesseract::TessdataManager::SaveFile	(	const STRING &	filename,
		FileWriter	writer
	)		const

Definition at line 153 of file tessdatamanager.cpp.

                                                        {
  // TODO: This method supports only the proprietary file format.
  ASSERT_HOST(is_loaded_);
  GenericVector<char> data;
  Serialize(&data);
  if (writer == nullptr)
    return SaveDataToFile(data, filename);
  else
    return (*writer)(data, filename);
}

void tesseract::TessdataManager::Serialize

(

GenericVector< char > *

data

)

const

Definition at line 166 of file tessdatamanager.cpp.

                                                               {
  // TODO: This method supports only the proprietary file format.
  ASSERT_HOST(is_loaded_);
  // Compute the offset_table and total size.
  int64_t offset_table[TESSDATA_NUM_ENTRIES];
  int64_t offset = sizeof(int32_t) + sizeof(offset_table);
  for (unsigned i = 0; i < TESSDATA_NUM_ENTRIES; ++i) {
    if (entries_[i].empty()) {
      offset_table[i] = -1;
    } else {
      offset_table[i] = offset;
      offset += entries_[i].size();
    }
  }
  data->init_to_size(offset, 0);
  int32_t num_entries = TESSDATA_NUM_ENTRIES;
  TFile fp;
  fp.OpenWrite(data);
  fp.Serialize(&num_entries);
  fp.Serialize(&offset_table[0], countof(offset_table));
  for (const auto& entry : entries_) {
    if (!entry.empty()) {
      fp.Serialize(&entry[0], entry.size());
    }
  }
}

void tesseract::TessdataManager::SetVersionString

(

const std::string &

v_str

)

Definition at line 239 of file tessdatamanager.cpp.

                                                             {
  entries_[TESSDATA_VERSION].resize_no_init(v_str.size());
  memcpy(&entries_[TESSDATA_VERSION][0], v_str.data(), v_str.size());
}

bool tesseract::TessdataManager::swap ( ) const

inline

Definition at line 133 of file tessdatamanager.h.

{ return swap_; }

std::string tesseract::TessdataManager::VersionString

(

)

const

Definition at line 233 of file tessdatamanager.cpp.

                                               {
  return std::string(&entries_[TESSDATA_VERSION][0],
                     entries_[TESSDATA_VERSION].size());
}

---

The documentation for this class was generated from the following files:

• src/ccutil/tessdatamanager.h
• src/ccutil/tessdatamanager.cpp