Adds TessHOcrTsvRenderer class for rendering HOCR info in tsv format.

sundarcf · tfmorris · commit 4d13892f5b70 · 2016-03-01T12:13:42.000-05:00
diff --git a/api/renderer.cpp b/api/renderer.cpp
@@ -179,6 +179,61 @@ bool TessHOcrRenderer::AddImageHandler(TessBaseAPI* api) {
   return true;
 }
 
+/**********************************************************************
+ * HOcr Text Renderer interface implementation
+ **********************************************************************/
+TessHOcrTsvRenderer::TessHOcrTsvRenderer(const char *outputbase)
+    : TessResultRenderer(outputbase, "hocr.tsv") {
+    font_info_ = false;
+}
+
+TessHOcrTsvRenderer::TessHOcrTsvRenderer(const char *outputbase, bool font_info)
+    : TessResultRenderer(outputbase, "hocr.tsv") {
+    font_info_ = font_info;
+}
+
+bool TessHOcrTsvRenderer::BeginDocumentHandler() {
+  AppendString(
+        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
+        "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Transitional//EN\"\n"
+        "    \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd\">\n"
+        "<html xmlns=\"http://www.w3.org/1999/xhtml\" xml:lang=\"en\" "
+        "lang=\"en\">\n <head>\n  <title>\n");
+  AppendString(title());
+  AppendString(
+      "</title>\n"
+      "<meta http-equiv=\"Content-Type\" content=\"text/html;"
+      "charset=utf-8\" />\n"
+      "  <meta name='ocr-system' content='tesseract " TESSERACT_VERSION_STR
+              "' />\n"
+      "  <meta name='ocr-capabilities' content='ocr_page ocr_carea ocr_par"
+      " ocr_line ocrx_word");
+  if (font_info_)
+    AppendString(
+      " ocrp_lang ocrp_dir ocrp_font ocrp_fsize ocrp_wconf");
+  AppendString(
+      "'/>\n"
+      "</head>\n<body>\n");
+
+  return true;
+}
+
+bool TessHOcrTsvRenderer::EndDocumentHandler() {
+  AppendString(" </body>\n</html>\n");
+
+  return true;
+}
+
+bool TessHOcrTsvRenderer::AddImageHandler(TessBaseAPI* api) {
+  char* hocr = api->GetHOCRText(imagenum());
+  if (hocr == NULL) return false;
+
+  AppendString(hocr);
+  delete[] hocr;
+
+  return true;
+}
+
 /**********************************************************************
  * UNLV Text Renderer interface implementation
  **********************************************************************/
diff --git a/api/renderer.h b/api/renderer.h
@@ -162,6 +162,23 @@ class TESS_API TessHOcrRenderer : public TessResultRenderer {
   bool font_info_;              // whether to print font information
 };
 
+/**
+ * Renders tesseract output into an hocr tsv string
+ */
+class TESS_API TessHOcrTsvRenderer : public TessResultRenderer {
+ public:
+  explicit TessHOcrTsvRenderer(const char *outputbase, bool font_info);
+  explicit TessHOcrTsvRenderer(const char *outputbase);
+
+protected:
+  virtual bool BeginDocumentHandler();
+  virtual bool AddImageHandler(TessBaseAPI* api);
+  virtual bool EndDocumentHandler();
+
+private:
+  bool font_info_;              // whether to print font information
+};
+
 /**
  * Renders tesseract output into searchable PDF
  */
