/*!
* quadtreeGetChildren()
*
* Input: fpixa (mean, variance or root variance)
* level, x, y (of current pixel)
* &val00, val01, val10, val11 (<return> child pixel values)
* Return: 0 if OK, 1 on error
*
* Notes:
* (1) Check return value for error. On error, all return vals are 0.0.
* (2) The returned child pixels are located at:
* level + 1
* (2x, 2y), (2x+1, 2y), (2x, 2y+1), (2x+1, 2y+1)
*/
l_int32
quadtreeGetChildren(FPIXA *fpixa,
l_int32 level,
l_int32 x,
l_int32 y,
l_float32 *pval00,
l_float32 *pval10,
l_float32 *pval01,
l_float32 *pval11) {
l_int32 n;
PROCNAME("quadtreeGetChildren");
if (!pval00 || !pval01 || !pval10 || !pval11)
return ERROR_INT("&val* not all defined", procName, 1);
*pval00 = *pval10 = *pval01 = *pval11 = 0.0;
if (!fpixa)
return ERROR_INT("fpixa not defined", procName, 1);
n = fpixaGetCount(fpixa);
if (level < 0 || level >= n - 1)
return ERROR_INT("invalid level", procName, 1);
if (fpixaGetPixel(fpixa, level + 1, 2 * x, 2 * y, pval00) != 0)
return ERROR_INT("invalid coordinates", procName, 1);
fpixaGetPixel(fpixa, level + 1, 2 * x + 1, 2 * y, pval10);
fpixaGetPixel(fpixa, level + 1, 2 * x, 2 * y + 1, pval01);
fpixaGetPixel(fpixa, level + 1, 2 * x + 1, 2 * y + 1, pval11);
return 0;
}
/*!
* quadtreeGetParent()
*
* Input: fpixa (mean, variance or root variance)
* level, x, y (of current pixel)
* &val (<return> parent pixel value), or 0.0 on error.
* Return: 0 if OK, 1 on error
*
* Notes:
* (1) Check return value for error. On error, val is returned as 0.0.
* (2) The parent is located at:
* level - 1
* (x/2, y/2)
*/
l_int32
quadtreeGetParent(FPIXA *fpixa,
l_int32 level,
l_int32 x,
l_int32 y,
l_float32 *pval) {
l_int32 n;
PROCNAME("quadtreeGetParent");
if (!pval)
return ERROR_INT("&val not defined", procName, 1);
*pval = 0.0;
if (!fpixa)
return ERROR_INT("fpixa not defined", procName, 1);
n = fpixaGetCount(fpixa);
if (level < 1 || level >= n)
return ERROR_INT("invalid level", procName, 1);
if (fpixaGetPixel(fpixa, level - 1, x / 2, y / 2, pval) != 0)
return ERROR_INT("invalid coordinates", procName, 1);
return 0;
}
main(int argc,
char **argv)
{
l_int32 i, j, w, h, error;
l_float32 val1, val2;
l_float32 val00, val10, val01, val11, valc00, valc10, valc01, valc11;
PIX *pixs, *pixg, *pixt1, *pixt2, *pixt3, *pixt4, *pixt5;
FPIXA *fpixam, *fpixav, *fpixarv;
BOXAA *baa;
static char mainName[] = "quadtreetest";
if (argc != 1)
return ERROR_INT(" Syntax: quadtreetest", mainName, 1);
/* Test generation of quadtree regions. */
baa = boxaaQuadtreeRegions(1000, 500, 3);
boxaaWriteStream(stderr, baa);
boxaaDestroy(&baa);
baa = boxaaQuadtreeRegions(1001, 501, 3);
boxaaWriteStream(stderr, baa);
boxaaDestroy(&baa);
/* Test quadtree stats generation */
#if 1
pixs = pixRead("rabi.png");
pixg = pixScaleToGray4(pixs);
#else
pixs = pixRead("test24.jpg");
pixg = pixConvertTo8(pixs, 0);
#endif
pixQuadtreeMean(pixg, 8, NULL, &fpixam);
pixt1 = fpixaDisplayQuadtree(fpixam, 4);
pixDisplay(pixt1, 100, 0);
pixWrite("/tmp/quadtree1.png", pixt1, IFF_PNG);
pixQuadtreeVariance(pixg, 8, NULL, NULL, &fpixav, &fpixarv);
pixt2 = fpixaDisplayQuadtree(fpixav, 4);
pixDisplay(pixt2, 100, 200);
pixWrite("/tmp/quadtree2.png", pixt2, IFF_PNG);
pixt3 = fpixaDisplayQuadtree(fpixarv, 4);
pixDisplay(pixt3, 100, 400);
pixWrite("/tmp/quadtree3.png", pixt3, IFF_PNG);
/* Compare with fixed-size tiling at a resolution corresponding
* to the deepest level of the quadtree above */
pixt4 = pixGetAverageTiled(pixg, 5, 6, L_MEAN_ABSVAL);
pixt5 = pixExpandReplicate(pixt4, 4);
pixWrite("/tmp/quadtree4.png", pixt5, IFF_PNG);
pixDisplay(pixt5, 800, 0);
pixDestroy(&pixt4);
pixDestroy(&pixt5);
pixt4 = pixGetAverageTiled(pixg, 5, 6, L_STANDARD_DEVIATION);
pixt5 = pixExpandReplicate(pixt4, 4);
pixWrite("/tmp/quadtree5.png", pixt5, IFF_PNG);
pixDisplay(pixt5, 800, 400);
/* Test quadtree parent/child access */
error = FALSE;
fpixaGetFPixDimensions(fpixam, 4, &w, &h);
for (i = 0; i < w; i += 2) {
for (j = 0; j < h; j += 2) {
quadtreeGetParent(fpixam, 4, j, i, &val1);
fpixaGetPixel(fpixam, 3, j / 2, i / 2, &val2);
if (val1 != val2) error = TRUE;
}
}
if (error)
fprintf(stderr, "\n======================\nError: parent access\n");
else
fprintf(stderr, "\n======================\nSuccess: parent access\n");
error = FALSE;
for (i = 0; i < w; i++) {
for (j = 0; j < h; j++) {
quadtreeGetChildren(fpixam, 4, j, i,
&val00, &val10, &val01, &val11);
fpixaGetPixel(fpixam, 5, 2 * j, 2 * i, &valc00);
fpixaGetPixel(fpixam, 5, 2 * j + 1, 2 * i, &valc10);
fpixaGetPixel(fpixam, 5, 2 * j, 2 * i + 1, &valc01);
fpixaGetPixel(fpixam, 5, 2 * j + 1, 2 * i + 1, &valc11);
if ((val00 != valc00) || (val10 != valc10) ||
(val01 != valc01) || (val11 != valc11))
error = TRUE;
}
}
if (error)
fprintf(stderr, "Error: child access\n======================\n");
else
fprintf(stderr, "Success: child access\n======================\n");
pixDestroy(&pixs);
pixDestroy(&pixg);
pixDestroy(&pixt1);
pixDestroy(&pixt2);
pixDestroy(&pixt3);
pixDestroy(&pixt4);
pixDestroy(&pixt5);
fpixaDestroy(&fpixam);
fpixaDestroy(&fpixav);
fpixaDestroy(&fpixarv);
return 0;
}
