diff options
Diffstat (limited to 'leptonica/prog/dewarptest5.c')
-rw-r--r-- | leptonica/prog/dewarptest5.c | 140 |
1 files changed, 140 insertions, 0 deletions
diff --git a/leptonica/prog/dewarptest5.c b/leptonica/prog/dewarptest5.c new file mode 100644 index 00000000..ec88a7b8 --- /dev/null +++ b/leptonica/prog/dewarptest5.c @@ -0,0 +1,140 @@ +/*====================================================================* + - Copyright (C) 2001 Leptonica. All rights reserved. + - + - Redistribution and use in source and binary forms, with or without + - modification, are permitted provided that the following conditions + - are met: + - 1. Redistributions of source code must retain the above copyright + - notice, this list of conditions and the following disclaimer. + - 2. Redistributions in binary form must reproduce the above + - copyright notice, this list of conditions and the following + - disclaimer in the documentation and/or other materials + - provided with the distribution. + - + - THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + - ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + - LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + - A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ANY + - CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + - EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + - PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR + - PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY + - OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING + - NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + - SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + *====================================================================*/ + +/* + * dewarptest5.c + * + * Tests dewarping model applied to word bounding boxes. + */ + +#ifdef HAVE_CONFIG_H +#include <config_auto.h> +#endif /* HAVE_CONFIG_H */ + +#include "allheaders.h" + +static l_int32 pageno = 35; +static l_int32 build_output = 0; +static l_int32 apply_output = 0; +static l_int32 map_output = 1; + +l_int32 main(int argc, + char **argv) +{ +char buf[64]; +BOXA *boxa1, *boxa2, *boxa3, *boxa4; +L_DEWARP *dew; +L_DEWARPA *dewa; +PIX *pixs, *pixn, *pixg, *pixb, *pix2, *pix3, *pix4, *pix5, *pix6; + + setLeptDebugOK(1); + lept_mkdir("lept"); + + snprintf(buf, sizeof(buf), "cat.%03d.jpg", pageno); + pixs = pixRead(buf); + dewa = dewarpaCreate(40, 30, 1, 15, 10); + dewarpaUseBothArrays(dewa, 1); + + /* Normalize for varying background and binarize */ + pixn = pixBackgroundNormSimple(pixs, NULL, NULL); + pixg = pixConvertRGBToGray(pixn, 0.5, 0.3, 0.2); + pixb = pixThresholdToBinary(pixg, 130); + pixDisplay(pixb, 0, 100); + + /* Build the model */ + dew = dewarpCreate(pixb, pageno); + dewarpaInsertDewarp(dewa, dew); + if (build_output) { + snprintf(buf, sizeof(buf), "/tmp/lept/dewarp_build_%d.pdf", pageno); + dewarpBuildPageModel(dew, buf); + } else { + dewarpBuildPageModel(dew, NULL); + } + + /* Apply the model */ + dewarpPopulateFullRes(dew, pixg, 0, 0); + if (apply_output) { + snprintf(buf, sizeof(buf), "/tmp/lept/dewarp_apply_%d.pdf", pageno); + dewarpaApplyDisparity(dewa, pageno, pixb, 200, 0, 0, &pix2, buf); + } else { + dewarpaApplyDisparity(dewa, pageno, pixb, 200, 0, 0, &pix2, NULL); + } + pixDisplay(pix2, 200, 100); + + /* Reverse direction: get the word boxes for the dewarped pix ... */ + pixGetWordBoxesInTextlines(pix2, 5, 5, 500, 100, &boxa1, NULL); + pix3 = pixConvertTo32(pix2); + pixRenderBoxaArb(pix3, boxa1, 2, 255, 0, 0); + pixDisplay(pix3, 400, 100); + + /* ... and map to the word boxes for the input image */ + if (map_output) { + snprintf(buf, sizeof(buf), "/tmp/lept/dewarp_map1_%d.pdf", pageno); + dewarpaApplyDisparityBoxa(dewa, pageno, pix2, boxa1, 0, 0, 0, &boxa2, + buf); + } else { + dewarpaApplyDisparityBoxa(dewa, pageno, pix2, boxa1, 0, 0, 0, &boxa2, + NULL); + } + pix4 = pixConvertTo32(pixb); + pixRenderBoxaArb(pix4, boxa2, 2, 0, 255, 0); + pixDisplay(pix4, 600, 100); + + /* Forward direction: get the word boxes for the input pix ... */ + pixGetWordBoxesInTextlines(pixb, 5, 5, 500, 100, &boxa3, NULL); + pix5 = pixConvertTo32(pixb); + pixRenderBoxaArb(pix5, boxa3, 2, 255, 0, 0); + pixDisplay(pix5, 800, 100); + + /* ... and map to the word boxes for the dewarped image */ + if (map_output) { + snprintf(buf, sizeof(buf), "/tmp/lept/dewarp_map2_%d.pdf", pageno); + dewarpaApplyDisparityBoxa(dewa, pageno, pixb, boxa3, 1, 0, 0, &boxa4, + buf); + } else { + dewarpaApplyDisparityBoxa(dewa, pageno, pixb, boxa3, 1, 0, 0, &boxa4, + NULL); + } + pix6 = pixConvertTo32(pix2); + pixRenderBoxaArb(pix6, boxa4, 2, 0, 255, 0); + pixDisplay(pix6, 1000, 100); + + dewarpaDestroy(&dewa); + pixDestroy(&pixs); + pixDestroy(&pixn); + pixDestroy(&pixg); + pixDestroy(&pixb); + pixDestroy(&pix2); + pixDestroy(&pix3); + pixDestroy(&pix4); + pixDestroy(&pix5); + pixDestroy(&pix6); + boxaDestroy(&boxa1); + boxaDestroy(&boxa2); + boxaDestroy(&boxa3); + boxaDestroy(&boxa4); + return 0; +} |