char *get_ocr_text_simple_threshold(struct lib_hardsubx_ctx *ctx, PIX *image, float threshold)
{
char *text_out=NULL;
TessBaseAPISetImage2(ctx->tess_handle, image);
if(TessBaseAPIRecognize(ctx->tess_handle, NULL) != 0)
{
mprint("Error in Tesseract recognition, skipping frame\n");
return NULL;
}
if((text_out = TessBaseAPIGetUTF8Text(ctx->tess_handle)) == NULL)
{
mprint("Error getting text, skipping frame\n");
}
int conf = TessBaseAPIMeanTextConf(ctx->tess_handle);
if(conf < threshold)
return NULL;
ctx->cur_conf = (float)conf;
return text_out;
}
char* ocr_bitmap(void* arg, png_color *palette,png_byte *alpha, unsigned char* indata,int w, int h)
{
PIX *pix = NULL;
PIX *cpix = NULL;
char*text_out= NULL;
int i,j,index;
unsigned int wpl;
unsigned int *data,*ppixel;
BOOL tess_ret = FALSE;
struct ocrCtx* ctx = arg;
pix = pixCreate(w, h, 32);
if(pix == NULL)
{
return NULL;
}
wpl = pixGetWpl(pix);
data = pixGetData(pix);
#if LEPTONICA_VERSION > 69
pixSetSpp(pix, 4);
#endif
for (i = 0; i < h; i++)
{
ppixel = data + i * wpl;
for (j = 0; j < w; j++)
{
index = indata[i * w + (j)];
composeRGBPixel(palette[index].red, palette[index].green,palette[index].blue, ppixel);
SET_DATA_BYTE(ppixel, L_ALPHA_CHANNEL,alpha[index]);
ppixel++;
}
}
ignore_alpha_at_edge(alpha, indata, w, h, pix, &cpix);
#ifdef OCR_DEBUG
{
char str[128] = "";
static int i = 0;
sprintf(str,"temp/file_c_%d.png",i);
pixWrite(str, cpix, IFF_PNG);
i++;
}
#endif
TessBaseAPISetImage2(ctx->api, cpix);
tess_ret = TessBaseAPIRecognize(ctx->api, NULL);
if( tess_ret != 0)
printf("\nsomething messy\n");
text_out = TessBaseAPIGetUTF8Text(ctx->api);
pixDestroy(&pix);
pixDestroy(&cpix);
return text_out;
}
char *get_ocr_text_letterwise_threshold(struct lib_hardsubx_ctx *ctx, PIX *image, float threshold)
{
char *text_out=NULL;
TessBaseAPISetImage2(ctx->tess_handle, image);
if(TessBaseAPIRecognize(ctx->tess_handle, NULL) != 0)
{
mprint("Error in Tesseract recognition, skipping symbol\n");
return NULL;
}
TessResultIterator *it = TessBaseAPIGetIterator(ctx->tess_handle);
TessPageIteratorLevel level = RIL_SYMBOL;
float total_conf = 0.0;
int num_words = 0;
if(it!=0)
{
do
{
char* letter = TessResultIteratorGetUTF8Text(it, level);
if(letter==NULL || strlen(letter)==0)
continue;
float conf = TessResultIteratorConfidence(it,level);
if(conf < threshold)
continue;
total_conf+=conf;
num_words++;
if(text_out==NULL)
{
text_out = strdup(letter);
continue;
}
text_out = realloc(text_out, strlen(text_out) + strlen(letter) + 1);
strcat(text_out, letter);
free(letter);
} while(TessPageIteratorNext((TessPageIterator *)it, level));
}
if(num_words>0)
total_conf = total_conf/num_words;
ctx->cur_conf = total_conf;
TessResultIteratorDelete(it);
return text_out;
}
char *get_ocr_text_simple(struct lib_hardsubx_ctx *ctx, PIX *image)
{
char *text_out=NULL;
TessBaseAPISetImage2(ctx->tess_handle, image);
if(TessBaseAPIRecognize(ctx->tess_handle, NULL) != 0)
{
mprint("Error in Tesseract recognition, skipping frame\n");
return NULL;
}
if((text_out = TessBaseAPIGetUTF8Text(ctx->tess_handle)) == NULL)
{
mprint("Error getting text, skipping frame\n");
}
return text_out;
}
char *get_ocr_text_letterwise(struct lib_hardsubx_ctx *ctx, PIX *image)
{
char *text_out=NULL;
TessBaseAPISetImage2(ctx->tess_handle, image);
if(TessBaseAPIRecognize(ctx->tess_handle, NULL) != 0)
{
mprint("Error in Tesseract recognition, skipping symbol\n");
return NULL;
}
TessResultIterator *it = TessBaseAPIGetIterator(ctx->tess_handle);
TessPageIteratorLevel level = RIL_SYMBOL;
if(it!=0)
{
do
{
char* letter = TessResultIteratorGetUTF8Text(it, level);
if(letter==NULL || strlen(letter)==0)
continue;
if(text_out==NULL)
{
text_out = strdup(letter);
continue;
}
text_out = realloc(text_out, strlen(text_out) + strlen(letter) + 1);
strcat(text_out, letter);
free(letter);
} while(TessPageIteratorNext((TessPageIterator *)it, level));
}
TessResultIteratorDelete(it);
return text_out;
}
char *get_ocr_text_wordwise(struct lib_hardsubx_ctx *ctx, PIX *image)
{
char *text_out=NULL;
TessBaseAPISetImage2(ctx->tess_handle, image);
if(TessBaseAPIRecognize(ctx->tess_handle, NULL) != 0)
{
mprint("Error in Tesseract recognition, skipping word\n");
return NULL;
}
TessResultIterator *it = TessBaseAPIGetIterator(ctx->tess_handle);
TessPageIteratorLevel level = RIL_WORD;
int prev_ital = 0;
if(it!=0)
{
do
{
char* word = TessResultIteratorGetUTF8Text(it, level);
if(word==NULL || strlen(word)==0)
continue;
if(text_out == NULL)
{
if(ctx->detect_italics)
{
int italic=0;
int dummy=0;
TessResultIteratorWordFontAttributes(it, &dummy, &italic,&dummy, &dummy, &dummy,&dummy, &dummy, &dummy);
if(italic==1 && prev_ital==0)
{
char *word_copy = strdup(word);
word = realloc(word, strlen(word)+strlen("<i>")+2);
strcpy(word,"<i>");
strcat(word, word_copy);
free(word_copy);
prev_ital = 1;
}
else if(italic == 0 && prev_ital == 1)
{
word = realloc(word, strlen(word)+strlen("</i>")+2);
strcat(word, "</i>");
prev_ital = 0;
}
}
text_out = strdup(word);
text_out = realloc(text_out, strlen(text_out)+2);
strcat(text_out, " ");
continue;
}
if(ctx->detect_italics)
{
int italic=0;
int dummy=0;
TessResultIteratorWordFontAttributes(it, &dummy, &italic,&dummy, &dummy, &dummy,&dummy, &dummy, &dummy);
if(italic==1 && prev_ital==0)
{
char *word_copy = strdup(word);
word = realloc(word, strlen(word)+strlen("<i>")+2);
strcpy(word,"<i>");
strcat(word, word_copy);
free(word_copy);
prev_ital = 1;
}
else if(italic == 0 && prev_ital == 1)
{
word = realloc(word, strlen(word)+strlen("</i>")+2);
strcat(word, "</i>");
prev_ital = 0;
}
}
text_out = realloc(text_out, strlen(text_out)+strlen(word)+2);
strcat(text_out, word);
strcat(text_out, " ");
free(word);
} while(TessPageIteratorNext((TessPageIterator *)it, level));
}
if(ctx->detect_italics && prev_ital == 1)
{
text_out = realloc(text_out, strlen(text_out)+strlen("</i>")+2);
strcat(text_out, "</i>");
}
TessResultIteratorDelete(it);
return text_out;
}
