RetroArch/translation/drivers_ocr/ocr_tesseract.c

#include <libretro.h>

#include "../ocr_driver.h"
#include "tesseract/wrapper/tess_get_text.h"

static void* ocr_tesseract_init(int game_character_set)
{
	const char* tess_char_set = NULL;
	switch (game_character_set)
	{
		case RETRO_LANGUAGE_JAPANESE:
		tess_char_set = "jpn";
		break;
		
		case RETRO_LANGUAGE_ENGLISH:
		tess_char_set = "eng";
		break;
		
		case RETRO_LANGUAGE_SPANISH:
		tess_char_set = "spa";
		break;
	}
	
	if (!tess_char_set)
		return NULL;
	
	char* lang_data_dir = NULL;
	
	/* TODO: get lang data from system dir */
	
	bool pass = tess_init(lang_data_dir, tess_char_set);
	if (pass)
	{
		/* data is unused by tesseract */
		return (void*)32;
	}
	
	return NULL;
}
	
static void ocr_tesseract_free(void* data)
{
}

char* ocr_tesseract_get_text(void* data, struct ocr_image_info image)
{
	tess_image temp_image;
	
	temp_image.width  = image.width;
	temp_image.height = image.height;
	temp_image.data   = image.data;
	
	switch (image.pixel_format)
	{
		case RETRO_PIXEL_FORMAT_0RGB1555:
		case RETRO_PIXEL_FORMAT_RGB565:
			temp_image.bytes_per_pixel = 2;
			break
		
		case RETRO_PIXEL_FORMAT_XRGB8888:
			temp_image.bytes_per_pixel = 4;
			break;
			
		default:
			/* unsupported format */
			return "";
	}
	
	return tess_get_text(temp_image);
}
	
const ocr_driver_t ocr_tesseract = {
   ocr_tesseract_init,
   ocr_tesseract_free,
   ocr_tesseract_get_text,
   "tesseract"
};