1 files changed, 135 insertions, 0 deletions
diff --git a/leptonica/prog/baseline_reg.c b/leptonica/prog/baseline_reg.c
new file mode 100644
index 00000000..a02de42e
--- /dev/null
+++ b/leptonica/prog/baseline_reg.c
@@ -0,0 +1,135 @@
+/*====================================================================*
+ -  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.
+ *====================================================================*/
+
+/*
+ * baselinetest.c
+ *
+ *   This tests two things:
+ *   (1) The ability to find a projective transform that will deskew
+ *       textlines in an image with keystoning.
+ *   (2) The ability to find baselines in a text image.
+ *   (3) The ability to clean background to white in a dark and
+ *       mottled text image.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config_auto.h>
+#endif  /* HAVE_CONFIG_H */
+
+#include "allheaders.h"
+
+int main(int    argc,
+         char **argv)
+{
+NUMA         *na;
+PIX          *pixs, *pix1, *pix2, *pix3, *pix4, *pix5;
+PIXA         *pixadb;
+PTA          *pta;
+L_REGPARAMS  *rp;
+
+#if !defined(HAVE_LIBPNG)
+    L_ERROR("This test requires libpng to run.\n", "baseline_reg");
+    exit(77);
+#endif
+
+    if (regTestSetup(argc, argv, &rp))
+        return 1;
+
+    pixs = pixRead("keystone.png");
+
+        /* Test function for deskewing using projective transform
+	 * on linear approximation for local skew angle */
+    pix1 = pixDeskewLocal(pixs, 10, 0, 0, 0.0, 0.0, 0.0);
+    regTestWritePixAndCheck(rp, pix1, IFF_PNG);  /* 0 */
+
+        /* Test function for finding local skew angles */
+    na = pixGetLocalSkewAngles(pixs, 10, 0, 0, 0.0, 0.0, 0.0, NULL, NULL, 1);
+    gplotSimple1(na, GPLOT_PNG, "/tmp/lept/baseline/ang", "Angles in degrees");
+    pix2 = pixRead("/tmp/lept/baseline/ang.png");
+    pix3 = pixRead("/tmp/lept/baseline/skew.png");
+    regTestWritePixAndCheck(rp, pix2, IFF_PNG);  /* 1 */
+    regTestWritePixAndCheck(rp, pix3, IFF_PNG);  /* 2 */
+    pixDisplayWithTitle(pix2, 0, 550, NULL, rp->display);
+    pixDisplayWithTitle(pix3, 700, 550, NULL, rp->display);
+    numaDestroy(&na);
+    pixDestroy(&pix2);
+    pixDestroy(&pix3);
+    pixDestroy(&pixs);
+
+        /* Test baseline finder */
+    pixadb = pixaCreate(6);
+    na = pixFindBaselines(pix1, &pta, pixadb);
+    pix2 = pixRead("/tmp/lept/baseline/diff.png");
+    pix3 = pixRead("/tmp/lept/baseline/loc.png");
+    pix4 = pixRead("/tmp/lept/baseline/baselines.png");
+    regTestWritePixAndCheck(rp, pix2, IFF_PNG);  /* 3 */
+    regTestWritePixAndCheck(rp, pix3, IFF_PNG);  /* 4 */
+    regTestWritePixAndCheck(rp, pix4, IFF_PNG);  /* 5 */
+    pixDisplayWithTitle(pix2, 0, 0, NULL, rp->display);
+    pixDisplayWithTitle(pix3, 700, 0, NULL, rp->display);
+    pixDisplayWithTitle(pix4, 1350, 0, NULL, rp->display);
+    pix5 = pixaDisplayTiledInRows(pixadb, 32, 1500, 1.0, 0, 30, 2);
+    pixDisplayWithTitle(pix5, 0, 500, NULL, rp->display);
+    regTestWritePixAndCheck(rp, pix5, IFF_PNG);  /* 6 */
+    pixDestroy(&pix1);
+    pixDestroy(&pix2);
+    pixDestroy(&pix3);
+    pixDestroy(&pix4);
+    pixDestroy(&pix5);
+    pixaDestroy(&pixadb);
+    numaDestroy(&na);
+    ptaDestroy(&pta);
+
+        /* Another test for baselines, with dark image */
+    pixadb = pixaCreate(6);
+    pixs = pixRead("pedante.079.jpg");  /* 75 ppi */
+    pix1 = pixRemoveBorder(pixs, 30);
+    pixaAddPix(pixadb, pix1, L_COPY);
+    pix2 = pixConvertRGBToGray(pix1, 0.33, 0.34, 0.33);
+    pix3 = pixScale(pix2, 4.0, 4.0);   /* scale up to 300 ppi */
+    pix4 = pixCleanBackgroundToWhite(pix3, NULL, NULL, 1.0, 70, 170);
+    pix5 = pixThresholdToBinary(pix4, 170);
+    regTestWritePixAndCheck(rp, pix5, IFF_PNG);  /* 7 */
+    pixaAddPix(pixadb, pixScale(pix5, 0.25, 0.25), L_INSERT);
+    pixDestroy(&pix1);
+    pixDestroy(&pix2);
+    pixDestroy(&pix3);
+    pixDestroy(&pix4);
+    pix1 = pixDeskew(pix5, 2);
+    na = pixFindBaselines(pix1, &pta, pixadb);
+    pix2 = pixaDisplayTiledInRows(pixadb, 32, 1500, 1.0, 0, 30, 2);
+    regTestWritePixAndCheck(rp, pix2, IFF_PNG);  /* 8 */
+    pixDisplayWithTitle(pix2, 800, 500, NULL, rp->display);
+    pixaDestroy(&pixadb);
+    pixDestroy(&pixs);
+    pixDestroy(&pix1);
+    pixDestroy(&pix2);
+    pixDestroy(&pix5);
+    numaDestroy(&na);
+    ptaDestroy(&pta);
+
+    return regTestCleanup(rp);
+}