295 lines
10 KiB
JavaScript
295 lines
10 KiB
JavaScript
class GuardLine {
|
|
|
|
// ardConf — reference to ArDetector that has current GuardLine instance
|
|
constructor(ardConf){
|
|
this.blackbar = {top: undefined, bottom: undefined};
|
|
this.imageBar = {top: undefined, bottom: undefined};
|
|
|
|
this.conf = ardConf;
|
|
}
|
|
|
|
reset() {
|
|
this.blackbar = {top: undefined, bottom: undefined};
|
|
this.imageBar = {top: undefined, bottom: undefined};
|
|
}
|
|
|
|
setBlackbar(bbconf){
|
|
var bbTop = bbconf.top - ExtensionConf.arDetect.guardLine.edgeTolerancePx;
|
|
var bbBottom = bbconf.bottom + ExtensionConf.arDetect.guardLine.edgeTolerancePx;
|
|
|
|
// to odstrani vse neveljavne nastavitve in vse možnosti, ki niso smiselne
|
|
// this removes any configs with invalid values or values that dont make sense
|
|
if (bbTop < 0 || bbBottom >= this.conf.canvas.height ){
|
|
this.reset();
|
|
return;
|
|
}
|
|
|
|
this.blackbar = {
|
|
top: bbTop,
|
|
bottom: bbBottom
|
|
}
|
|
|
|
this.imageBar = {
|
|
top: bbconf.top + 1 + ExtensionConf.arDetect.guardLine.edgeTolerancePx,
|
|
bottom: bbconf.bottom - 1 - ExtensionConf.arDetect.guardLine.edgeTolerancePx
|
|
}
|
|
}
|
|
|
|
check(image, fallbackMode){
|
|
// izračunaj enkrat in shrani na objekt
|
|
// calculate once and save object-instance-wide
|
|
this.blackbarTreshold = this.conf.blackLevel + ExtensionConf.arDetect.blackbarTreshold;
|
|
|
|
// dejansko testiranje
|
|
// actual checks
|
|
var guardLineResult = this.guardLineCheck(image, fallbackMode);
|
|
|
|
// Zaznali smo kršitev črnega dela, zato nam ni treba preveriti, ali je slika
|
|
// prisotna. Vemo namreč, da se je razmerje stranic zmanjšalo.
|
|
//
|
|
// blackbar violation detected. We don't even need to check for presence of image
|
|
// as aspect ratio just decreased
|
|
if(! guardLineResult.success) {
|
|
return {
|
|
blackbarFail: true,
|
|
offenders: guardLineResult.offenders,
|
|
imageFail: false
|
|
}
|
|
}
|
|
|
|
var imageCheckResult = this.imageCheck(image, fallbackMode);
|
|
|
|
return {
|
|
blackbarFail: false,
|
|
imageFail: ! imageCheckResult.success
|
|
}
|
|
}
|
|
|
|
|
|
// don't use methods below this line outside this class
|
|
guardLineCheck(image, fallbackMode){
|
|
// this test tests for whether we crop too aggressively
|
|
|
|
// if this test is passed, then aspect ratio probably didn't change from wider to narrower. However, further
|
|
// checks are needed to determine whether aspect ratio got wider.
|
|
// if this test fails, it returns a list of offending points.
|
|
|
|
// if the upper edge is null, then edge hasn't been detected before. This test is pointless, therefore it
|
|
// should succeed by default. Also need to check bottom, for cases where only one edge is known
|
|
|
|
if(! fallbackMode && (! this.blackbar.top || ! this.blackbar.bottom)) {
|
|
console.log("NO EDGE WAS DETECTED. THIS TEST IS POINTLESS. btw guardline")
|
|
return { success: true };
|
|
}
|
|
|
|
var offset = parseInt(this.conf.canvas.width * ExtensionConf.arDetect.guardLine.ignoreEdgeMargin) << 2;
|
|
|
|
var offenders = [];
|
|
var firstOffender = -1;
|
|
var offenderCount = -1; // doing it this way means first offender has offenderCount==0. Ez index.
|
|
|
|
// TODO: implement logo check.
|
|
|
|
// preglejmo obe vrstici
|
|
// check both rows
|
|
|
|
if(! fallbackMode){
|
|
var edge_upper = this.blackbar.top;
|
|
var edge_lower = this.blackbar.bottom;
|
|
}
|
|
else{
|
|
// fallback mode is a bit different
|
|
edge_upper = 0;
|
|
edge_lower = this.conf.canvas.height - 1;
|
|
}
|
|
|
|
var rowStart, rowEnd;
|
|
|
|
// <<<=======| checking upper row |========>>>
|
|
|
|
rowStart = ((edge_upper * this.conf.canvas.width) << 2) + offset;
|
|
rowEnd = rowStart + ( this.conf.canvas.width << 2 ) - (offset * 2);
|
|
|
|
if (Debug.debugCanvas.enabled && Debug.debugCanvas.guardLine) {
|
|
offenderCount = this._gl_debugRowCheck(image, rowStart, rowEnd, offenders, offenderCount);
|
|
} else {
|
|
offenderCount = this._gl_rowCheck(image, rowStart, rowEnd, offenders, offenderCount);
|
|
}
|
|
// <<<=======| checking lower row |========>>>
|
|
|
|
rowStart = ((edge_lower * this.conf.canvas.width) << 2) + offset;
|
|
rowEnd = rowStart + ( this.conf.canvas.width << 2 ) - (offset * 2);
|
|
|
|
if (Debug.debugCanvas.enabled && Debug.debugCanvas.guardLine) {
|
|
offenderCount = this._gl_debugRowCheck(image, rowStart, rowEnd, offenders, offenderCount);
|
|
} else {
|
|
offenderCount = this._gl_rowCheck(image, rowStart, rowEnd, offenders, offenderCount);
|
|
}
|
|
|
|
// če nismo našli nobenih prekrškarjev, vrnemo uspeh. Drugače vrnemo seznam prekrškarjev
|
|
// vrnemo tabelo, ki vsebuje sredinsko točko vsakega prekrškarja (x + width*0.5)
|
|
//
|
|
// if we haven't found any offenders, we return success. Else we return list of offenders
|
|
// we return array of middle points of offenders (x + (width * 0.5) for every offender)
|
|
|
|
if(offenderCount == -1){
|
|
return {success: true};
|
|
}
|
|
|
|
var ret = new Array(offenders.length);
|
|
for(var o in offenders){
|
|
ret[o] = offenders[o].x + (offenders[o].width * 0.25);
|
|
}
|
|
|
|
return {success: false, offenders: ret};
|
|
}
|
|
|
|
imageCheck(image){
|
|
if(!this.imageBar.top || !this.imageBar.bottom)
|
|
return { success: false };
|
|
|
|
|
|
var edges = GlobalVars.arDetect.guardLine;
|
|
|
|
var offset = parseInt(this.conf.canvas.width * ExtensionConf.arDetect.guardLine.ignoreEdgeMargin) << 2;
|
|
|
|
// TODO: implement logo check.
|
|
|
|
|
|
// preglejmo obe vrstici - tukaj po pravilih ne bi smeli iti prek mej platna. ne rabimo preverjati
|
|
// check both rows - by the rules and definitions, we shouldn't go out of bounds here. no need to check, then
|
|
|
|
// if(fallbackMode){
|
|
// var edge_upper = ExtensionConf.arDetect.fallbackMode.noTriggerZonePx;
|
|
// var edge_lower = this.conf.canvas.height - ExtensionConf.arDetect.fallbackMode.noTriggerZonePx - 1;
|
|
// }
|
|
// else{
|
|
var edge_upper = this.imageBar.top;
|
|
var edge_lower = this.imageBar.bottom;
|
|
// }
|
|
|
|
// koliko pikslov rabimo zaznati, da je ta funkcija uspe. Tu dovoljujemo tudi, da so vsi piksli na enem
|
|
// robu (eden izmed robov je lahko v celoti črn)
|
|
// how many non-black pixels we need to consider this check a success. We only need to detect enough pixels
|
|
// on one edge (one of the edges can be black as long as both aren't)
|
|
var successTreshold = parseInt(this.conf.canvas.width * ExtensionConf.arDetect.guardLine.imageTestTreshold);
|
|
var rowStart, rowEnd;
|
|
|
|
|
|
// <<<=======| checking upper row |========>>>
|
|
|
|
rowStart = ((edge_upper * this.conf.canvas.width) << 2) + offset;
|
|
rowEnd = rowStart + ( this.conf.canvas.width << 2 ) - (offset * 2);
|
|
|
|
var res = false;
|
|
|
|
if(Debug.debugCanvas.enabled && Debug.debugCanvas.guardLine){
|
|
res = this._ti_debugCheckRow(image, rowStart, rowEnd, successTreshold);
|
|
} else {
|
|
res = this._ti_checkRow(image, rowStart, rowEnd,successTreshold);
|
|
}
|
|
|
|
if(res)
|
|
return {success: true};
|
|
|
|
|
|
// <<<=======| checking lower row |========>>>
|
|
|
|
rowStart = ((edge_lower * this.conf.canvas.width) << 2) + offset;
|
|
// rowEnd = rowStart + ( this.conf.canvas.width << 2 ) - (offset * 2);
|
|
|
|
|
|
if(Debug.debugCanvas.enabled && Debug.debugCanvas.guardLine){
|
|
res = this._ti_debugCheckRow(image, rowStart, rowEnd, successTreshold);
|
|
} else {
|
|
res = this._ti_checkRow(image, rowStart, rowEnd,successTreshold);
|
|
}
|
|
|
|
return {success: res};
|
|
}
|
|
|
|
// pomožne metode
|
|
// helper methods
|
|
|
|
|
|
_gl_rowCheck(image, rowStart, rowEnd, offenders, offenderCount){
|
|
var firstOffender = -1;
|
|
for(var i = rowStart; i < rowEnd; i+=4){
|
|
|
|
// we track sections that go over what's supposed to be a black line, so we can suggest more
|
|
// columns to sample
|
|
if(image[i] > this.blackbarTreshold || image[i+1] > this.blackbarTreshold || image[i+2] > this.blackbarTreshold){
|
|
if(firstOffender < 0){
|
|
firstOffender = (i - rowStart) >> 2;
|
|
offenderCount++;
|
|
offenders.push({x: firstOffender, width: 1});
|
|
}
|
|
else{
|
|
offenders[offenderCount].width++
|
|
}
|
|
}
|
|
else{
|
|
// is that a black pixel again? Let's reset the 'first offender'
|
|
firstOffender = -1;
|
|
}
|
|
}
|
|
|
|
return offenderCount;
|
|
}
|
|
|
|
_gl_debugRowCheck(image, rowStart, rowEnd, offenders, offenderCount){
|
|
var firstOffender = -1;
|
|
for(var i = rowStart; i < rowEnd; i+=4){
|
|
|
|
// we track sections that go over what's supposed to be a black line, so we can suggest more
|
|
// columns to sample
|
|
if(image[i] > this.blackbarTreshold || image[i+1] > this.blackbarTreshold || image[i+2] > this.blackbarTreshold){
|
|
this.conf.debugCanvas.trace(i, DebugCanvasClasses.VIOLATION);
|
|
if(firstOffender < 0){
|
|
firstOffender = (i - rowStart) >> 2;
|
|
offenderCount++;
|
|
offenders.push({x: firstOffender, width: 1});
|
|
}
|
|
else{
|
|
offenders[offenderCount].width++
|
|
}
|
|
}
|
|
else{
|
|
this.conf.debugCanvas.trace(i, DebugCanvasClasses.GUARDLINE_BLACKBAR);
|
|
// is that a black pixel again? Let's reset the 'first offender'
|
|
firstOffender = -1;
|
|
}
|
|
|
|
}
|
|
|
|
return offenderCount;
|
|
}
|
|
|
|
_ti_checkRow(image, rowStart, rowEnd, successTreshold) {
|
|
for(var i = rowStart; i < rowEnd; i+=4){
|
|
if(image[i] > this.blackbarTreshold || image[i+1] > this.blackbarTreshold || image[i+2] > this.blackbarTreshold){
|
|
if(successTreshold --<= 0){
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
_ti_debugCheckRow(image, rowStart, rowEnd, successTreshold) {
|
|
for(var i = rowStart; i < rowEnd; i+=4){
|
|
if(image[i] > this.blackbarTreshold || image[i+1] > this.blackbarTreshold || image[i+2] > this.blackbarTreshold){
|
|
this.conf.debugCanvas.trace(i, DebugCanvasClasses.GUARDLINE_IMAGE);
|
|
if(successTreshold --<= 0){
|
|
return true;
|
|
}
|
|
} else {
|
|
this.conf.debugCanvas.trace(i, DebugCanvasClasses.WARN);
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|