private void fitTemplate() {
int startX = 0, startY = 0;
int endX = img.getWidth() - bestfit.getTemplate().getWidth();
int endY = img.getHeight() - bestfit.getTemplate().getHeight();
centerTemplate(startX, startY, endX, endY, 3);
templateXOR(img, bestfit.getX(), bestfit.getY(), bestfit.getTemplate(), true);
sizeTemplate();
aspectTemplate();
shiftTemplate();
sizeTemplate();
aspectTemplate();
shiftTemplate();
templateXOR(img, bestfit.getX(), bestfit.getY(), bestfit.getTemplate(), true);
}
private void shiftTemplate() {
double maxsim =
1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), bestfit.getTemplate(), false);
// System.out.println("maxsim = " + maxsim + ":" + bestfit.getSim());
int oldX = bestfit.getX();
int oldY = bestfit.getY();
for (int newX = oldX - 2; newX <= oldX + 2; newX++) {
for (int newY = oldY - 2; newY <= oldY + 2; newY++) {
double currsim = 1.0 - templateXOR(img, newX, newY, bestfit.getTemplate(), false);
if (currsim > maxsim) {
// System.out.println("--newX = " + newX
// + ": newY = " + newY + ":" + currsim);
bestfit.setX(newX);
bestfit.setY(newY);
bestfit.setSim(currsim);
maxsim = currsim;
}
}
}
}
private void centerTemplate(int startX, int startY, int endX, int endY, int granularity) {
int stepX = bestfit.getTemplate().getWidth() / granularity;
int stepY = bestfit.getTemplate().getHeight() / granularity;
// System.out.println("stepX = " + stepX + ": stepY = " + stepY);
double maxsim = -1;
int simi = -1, simj = -1;
for (int iCount = startX; iCount <= endX; iCount += stepX) {
for (int jCount = startY; jCount <= endY; jCount += stepY) {
double currsim = 1.0 - templateXOR(img, iCount, jCount, bestfit.getTemplate(), false);
// System.out.println(i + ":" + j + ":" + currsim);
if (maxsim == -1 || maxsim < currsim) {
maxsim = currsim;
simi = iCount;
simj = jCount;
}
}
}
// System.out.println("--- maxsim = " + maxsim + ":"
// + simi + ":" + simj);
if (maxsim > 0.5) {
if (stepX >= 4) { // up to an accuracy of 2 pixels
centerTemplate(
Math.max(simi - stepX / 2, 0),
Math.max(simj - stepY / 2, 0),
Math.min(simi + stepX / 2, img.getWidth()),
Math.min(simj + stepY / 2, img.getHeight()),
granularity * 2);
} else {
bestfit.setX(simi);
bestfit.setY(simj);
bestfit.setSim(maxsim);
}
}
}
private void aspectTemplate() {
Gray8Image template = (Gray8Image) (bestfit.getTemplate().createCopy());
double maxsim = 1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), template, false);
// System.out.println("maxsim = " + maxsim + ":" + bestfit.getSim());
double oldaspectscale = bestfit.getAspectScale();
for (double aspectscale = oldaspectscale - 0.05;
aspectscale <= oldaspectscale + 0.05;
aspectscale += 0.0025) {
fillTemplate(
template, bestfit.getApproxCircleOuterX(), bestfit.getApproxCircleInnerX(), aspectscale);
double currsim = 1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), template, false);
if (currsim > maxsim) {
// System.out.println("--aspectscale = " + aspectscale
// + ":" + currsim);
bestfit.setTemplate(template);
bestfit.setAspectScale(aspectscale);
bestfit.setSim(currsim);
template = (Gray8Image) (bestfit.getTemplate().createCopy());
maxsim = currsim;
}
}
}
private void sizeTemplate() {
Gray8Image template = (Gray8Image) (bestfit.getTemplate().createCopy());
double maxsim = 1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), template, false);
for (double outerdiam = bestfit.getApproxCircleOuterX() - 1; outerdiam > 0; outerdiam--) {
fillTemplate(template, outerdiam, bestfit.getApproxCircleInnerX(), bestfit.getAspectScale());
double currsim = 1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), template, false);
if (currsim < maxsim) {
break;
} else {
// System.out.println("--outerdiam = " + outerdiam
// + ":" + currsim);
bestfit.setTemplate(template);
bestfit.setApproxCircleOuterX(outerdiam);
bestfit.setSim(currsim);
template = (Gray8Image) (bestfit.getTemplate().createCopy());
maxsim = currsim;
}
}
for (double innerdiam = bestfit.approxCircleInnerX - 1; innerdiam > 0; innerdiam--) {
fillTemplate(template, bestfit.getApproxCircleOuterX(), innerdiam, bestfit.getAspectScale());
double currsim = 1.0 - templateXOR(img, bestfit.getX(), bestfit.getY(), template, false);
if (currsim < maxsim) {
break;
} else {
// System.out.println("--innerdiam = " + innerdiam
// + ":" + currsim);
bestfit.setTemplate(template);
bestfit.setApproxCircleInnerX(innerdiam);
bestfit.setSim(currsim);
template = (Gray8Image) (bestfit.getTemplate().createCopy());
maxsim = currsim;
}
}
}