From 16fd1439faa4f2c04275c17ca84272328abb843f Mon Sep 17 00:00:00 2001
From: Stefan Weil <sw@weilnetz.de>
Date: Sat, 7 Aug 2021 22:13:16 +0200
Subject: [PATCH] Write image filename in ALTO output

Signed-off-by: Stefan Weil <sw@weilnetz.de>
---
 include/tesseract/renderer.h |  3 +++
 src/api/altorenderer.cpp     | 32 +++++++++++++++++++-------------
 2 files changed, 22 insertions(+), 13 deletions(-)

diff --git a/include/tesseract/renderer.h b/include/tesseract/renderer.h
index 2cfa06c5..ffc27d41 100644
--- a/include/tesseract/renderer.h
+++ b/include/tesseract/renderer.h
@@ -189,6 +189,9 @@ protected:
   bool BeginDocumentHandler() override;
   bool AddImageHandler(TessBaseAPI *api) override;
   bool EndDocumentHandler() override;
+
+private:
+  bool begin_document;
 };
 
 /**
diff --git a/src/api/altorenderer.cpp b/src/api/altorenderer.cpp
index c0afcb07..089189ef 100644
--- a/src/api/altorenderer.cpp
+++ b/src/api/altorenderer.cpp
@@ -55,7 +55,17 @@ static void AddBoxToAlto(const ResultIterator *it, PageIteratorLevel level,
 /// Append the ALTO XML for the beginning of the document
 ///
 bool TessAltoRenderer::BeginDocumentHandler() {
-  AppendString(
+  // Delay the XML output because we need the name of the image file.
+  begin_document = true;
+  return true;
+}
+
+///
+/// Append the ALTO XML for the layout of the image
+///
+bool TessAltoRenderer::AddImageHandler(TessBaseAPI *api) {
+  if (begin_document) {
+    AppendString(
       "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
       "<alto xmlns=\"http://www.loc.gov/standards/alto/ns-v3#\" "
       "xmlns:xlink=\"http://www.w3.org/1999/xlink\" "
@@ -67,31 +77,26 @@ bool TessAltoRenderer::BeginDocumentHandler() {
       "\t\t<sourceImageInformation>\n"
       "\t\t\t<fileName>");
 
-  AppendString(title());
+    AppendString(api->GetInputName());
 
-  AppendString(
+    AppendString(
       "</fileName>\n"
       "\t\t</sourceImageInformation>\n"
       "\t\t<OCRProcessing ID=\"OCR_0\">\n"
       "\t\t\t<ocrProcessingStep>\n"
       "\t\t\t\t<processingSoftware>\n"
       "\t\t\t\t\t<softwareName>tesseract ");
-  AppendString(TessBaseAPI::Version());
-  AppendString(
+    AppendString(TessBaseAPI::Version());
+    AppendString(
       "</softwareName>\n"
       "\t\t\t\t</processingSoftware>\n"
       "\t\t\t</ocrProcessingStep>\n"
       "\t\t</OCRProcessing>\n"
       "\t</Description>\n"
       "\t<Layout>\n");
+    begin_document = false;
+  }
 
-  return true;
-}
-
-///
-/// Append the ALTO XML for the layout of the image
-///
-bool TessAltoRenderer::AddImageHandler(TessBaseAPI *api) {
   const std::unique_ptr<const char[]> text(api->GetAltoText(imagenum()));
   if (text == nullptr) {
     return false;
@@ -112,7 +117,8 @@ bool TessAltoRenderer::EndDocumentHandler() {
 }
 
 TessAltoRenderer::TessAltoRenderer(const char *outputbase)
-    : TessResultRenderer(outputbase, "xml") {}
+    : TessResultRenderer(outputbase, "xml"),
+      begin_document(false) {}
 
 ///
 /// Make an XML-formatted string with ALTO markup from the internal
-- 
GitLab