uint8_t ADMVideoForcedPP::configure( AVDMGenericVideoStream *instream)
{
_in=instream;
uint32_t postproc, strength,uv;
int forced;
postproc=_param->postProcType;
strength=_param->postProcStrength;
uv=0;
if( DIA_getMPParams( &postproc,&strength,&uv))
{
_param->postProcType=postproc;
_param->postProcStrength=strength;
forced=_param->forcedQuant;
if( DIA_GetIntegerValue(&forced,2,31, "Quant Value","Enter forced Q:"))
{
_param->forcedQuant=forced;
}
updatePostProc(&_postproc );
}
return 1;
}
uint8_t ADMVideoFields::configure( AVDMGenericVideoStream *instream)
{
int i,j;
_in=instream;
i=(int)_param->motion_trigger;
j=(int)_param->blend_trigger;
if(DIA_GetIntegerValue(&i,0,255, ("Motion Threshold"),""))
{
if(DIA_GetIntegerValue(&j,0,255, ("Blend Threshold"),""))
{
_param->motion_trigger=(uint8_t)i;
_param->blend_trigger=(uint8_t)j;
return 1;
}
}
return 0;
}
//*************************************
uint8_t vidHardPDRemoval::configure (AVDMGenericVideoStream * in)
{
int v,w;
_param->show=GUI_YesNo(QT_TR_NOOP("Metrics"),QT_TR_NOOP("Do you want to print metrics on screen ?" ));
v=_param->threshold;
w=_param->noise;
if(DIA_GetIntegerValue(&v, 2, 99,"Treshold","Treshold value (smaller = harder to match)"))
{
if(DIA_GetIntegerValue(&w, 2, 99,"Noise","Noise threshold"))
{
_param->threshold=v;
_param->noise=w;
_lastRemoved=0xFFFF;
return 1;
}
}
return 0;
}
uint8_t GUI_getIntegerValue(int *valye, int min, int max, const char *title)
{
return DIA_GetIntegerValue(valye,min,max,title,title);
}
/* +++++++++++++++++++++++++++++++++++++++++++++++++++++++
Main
+++++++++++++++++++++++++++++++++++++++++++++++++++++++ */
uint8_t ADM_ocr_engine( void)
{
//
uint32_t nbSub=0;
FILE *out=NULL;
head.next=NULL;
ADMVideoVobSub *vobsub=NULL;
uint32_t startTime,endTime;
uint32_t w,h,oldw=0,oldh=0;
uint32_t oldbitmapw=0;
uint32_t oldbitmaph=0;
uint32_t first,last;
uint32_t seqNum;
char text[1024];
lang_index=0;
nbGlyphs=0;
ReplyType reply;
// Create UI && prepare callback
dialog=DIA_ocr();
gtk_register_dialog(dialog);
#define ASSOCIATE(x,y) gtk_dialog_add_action_widget (GTK_DIALOG (dialog), WID(x),y)
ASSOCIATE(buttonStart,actionGo);
ASSOCIATE(buttonOk, actionAccept);
ASSOCIATE(buttonSkip, actionSkip);
ASSOCIATE(buttonSkipAll, actionSkipAll);
ASSOCIATE(buttonIgnore, actionIgnore);
ASSOCIATE(buttonCalibrate, actionCalibrate);
ASSOCIATE(buttonGlyphLoad, actionLoadGlyph);
ASSOCIATE(buttonGlyphSave, actionSaveGlyph);
ASSOCIATE(buttonVobsub, actionLoadVob);
ASSOCIATE(buttonSrt, actionSaveSub);
gtk_widget_show(dialog);
// disable
mainDisplay=WID(drawingareaBitmap);
smallDisplay=WID(drawingareaSmall);
CONNECT(drawingareaBitmap,expose_event,gui_draw);
CONNECT(drawingareaSmall,expose_event,gui_draw_small);
CONNECT(entry,activate,cb_accept);
_again:
reply=setup();
if(reply==ReplyClose) goto endIt;
printf("Go go go\n");
// Everything ready go go go
redraw_x=redraw_y=0;
GTK_PURGE;
// Time to go
// Inactivate frame1=glyph frame2=in/out buttonStart
gtk_widget_set_sensitive(WID(buttonStart),0);
gtk_widget_set_sensitive(WID(frameGlyph),0);
gtk_widget_set_sensitive(WID(frameLoad),0);
gtk_widget_set_sensitive(WID(frameBitmap),1);
gtk_widget_set_sensitive(WID(buttonStart),0);
char *fileout;
fileout=(char *)gtk_label_get_text(GTK_LABEL(WID(labelSrt)));
if(!fileout)
{
GUI_Error_HIG(_("Incorrect output file"), NULL);
goto _again;
}
out=fopen(fileout,"wb");
if(!out)
{
GUI_Error_HIG(_("Output file error"), _("Could not open \"%s\" for writing."), fileout);
goto _again;
}
vobsub=new ADMVideoVobSub(subparam.subname,subparam.index);
nbSub=vobsub->getNbImage();
if(!nbSub)
{
GUI_Error_HIG(_("Problem loading sub"), NULL);
delete vobsub;
vobsub=NULL;
goto _again;
}
seqNum=1; // Sub number in srt file
oldw=oldh=0;
//******************
// Load all bitmaps
//******************
for(uint32_t i=0;i<nbSub;i++)
{
first=last=0;
bitmap=vobsub->getBitmap(i,&startTime, &endTime,&first,&last);
ADM_assert(last>=first);
// something ?
if(!bitmap) continue;
if(first==last) continue;
// If the bitmap size changed or does not exist yet...
if(!workArea || oldbitmapw!=bitmap->_width || oldbitmaph!=bitmap->_height)
{
if(workArea)
{
delete [] workArea;
workArea=NULL;
}
// Workarea is actually bigger than what we use
workArea=new uint8_t[bitmap->_width*(bitmap->_height)];
memset(workArea,0,bitmap->_width*(bitmap->_height));
}
oldbitmaph=bitmap->_height;
oldbitmapw=bitmap->_width;
//
w=bitmap->_width;
h=last-first+1;
redraw_x=w;
redraw_y=h;
//**
// Build
againPlease:
mergeBitmap(bitmap->_bitmap+first*w, workArea, bitmap->_alphaMask+first*w, w, h);
if(oldw!=w || oldh !=h)
{
GTK_PURGE; // Force redaw
}
// Merge
GTK_PURGE;
gui_draw();
GTK_PURGE;
// OCR
reply=ocrBitmap(workArea,w,h);
if(reply==ReplyClose) goto endIt;
if(reply==ReplyCalibrate)
{
//
//printf("TADA!!!!\n");
int val;
#if 0
val=minAlpha;
if(DIA_GetIntegerValue(&val, 3, 7, "Minimum alpha value", "Enter new minimum alpha"))
{
minAlpha=val;
}
#endif
val=minThreshold;
if(DIA_GetIntegerValue(&val, 0x30, 0x80, "Minimum pixel value", "Enter new minimum pixel"))
{
minThreshold=val;
}
goto againPlease;
}
//
gtk_label_set_text(GTK_LABEL(WID(labelText)),decodedString);
fprintf(out,"%d\n",seqNum++);
uint16_t hh,mm,ss,ms;
ms2time(startTime, &hh, &mm, &ss, &ms);
fprintf(out,"%02d:%02d:%02d,%03d --> ",hh,mm,ss,ms);
ms2time(endTime, &hh, &mm, &ss, &ms);
fprintf(out,"%02d:%02d:%02d,%03d\n",hh,mm,ss,ms);
fprintf(out,"%s\n\n",decodedString);
//
oldw=w;
oldh=h;
// Update infos
sprintf(text,"%03d/%03d",i+1,nbSub);
gtk_label_set_text(GTK_LABEL(WID(labelNbLines)),text);
sprintf(text,"%03d",nbGlyphs);
gtk_label_set_text(GTK_LABEL(WID(labelNbGlyphs)),text);
}
endIt:
// Final round
gtk_widget_set_sensitive(WID(frameGlyph),1);
gtk_widget_set_sensitive(WID(frameLoad),0);
gtk_widget_set_sensitive(WID(buttonStart),0);
gtk_widget_set_sensitive(WID(frameBitmap),0);
// gtk_widget_set_sensitive(WID(Current_Glyph),0);
if(nbGlyphs && actionSaveGlyph==gtk_dialog_run(GTK_DIALOG(dialog)))
saveGlyph();
if(vobsub)
delete vobsub;
vobsub=NULL;
if(out)
fclose(out);
out=NULL;
gtk_unregister_dialog(dialog);
gtk_widget_destroy(dialog);
if(head.next)
destroyGlyphTree(&head);
head.next=NULL;
return 1;
}
