output funding statement in the back of the TEI output

kermitt2 · Oct 13, 2022 · b1c0cfd · b1c0cfd
1 parent 06a526f
commit b1c0cfd
Showing 1 changed file with 33 additions and 2 deletions.
diff --git a/grobid-core/src/main/java/org/grobid/core/engines/FullTextParser.java b/grobid-core/src/main/java/org/grobid/core/engines/FullTextParser.java
@@ -2475,7 +2475,7 @@ private void toTEI(Document doc,
                 List<LayoutToken> headerAvailabilityStatementTokens = resHeader.getLayoutTokens(TaggingLabels.HEADER_AVAILABILITY);
                 Pair<String, List<LayoutToken>> headerAvailabilityProcessed = processShort(headerAvailabilityStatementTokens, doc);
                 if (headerAvailabilityProcessed != null) {
-                    availabilityStmt = teiFormatter.processTEIDivSection("availability", 
+                    availabilityStmt = teiFormatter.processTEIDivSection("availability",
                         "\t\t\t", 
                         headerAvailabilityProcessed.getLeft(), 
                         headerAvailabilityProcessed.getRight(), 
@@ -2488,7 +2488,7 @@ private void toTEI(Document doc,
             }
 
             // availability statements in non-header part
-            availabilityStmt = getSectionAsTEI("availability", 
+            availabilityStmt = getSectionAsTEI("availability",
                 "\t\t\t", 
                 doc, 
                 SegmentationLabels.AVAILABILITY, 
@@ -2499,6 +2499,37 @@ private void toTEI(Document doc,
                 tei.append(availabilityStmt.toString());
             }
 
+            // funding in header
+            StringBuilder fundingStmt = new StringBuilder();
+            if (StringUtils.isNotBlank(resHeader.getFunding())) {
+                List<LayoutToken> headerFundingTokens = resHeader.getLayoutTokens(TaggingLabels.HEADER_FUNDING);
+                Pair<String, List<LayoutToken>> headerFundingProcessed = processShort(headerFundingTokens, doc);
+                if (headerFundingProcessed != null) {
+                    fundingStmt = teiFormatter.processTEIDivSection("funding",
+                        "\t\t\t",
+                        headerFundingProcessed.getLeft(),
+                        headerFundingProcessed.getRight(),
+                        resCitations,
+                        config);
+                }
+                if (fundingStmt.length() > 0) {
+                    tei.append(fundingStmt.toString());
+                }
+            }
+
+            // funding statements in non-header part
+            fundingStmt = getSectionAsTEI("funding",
+                "\t\t\t",
+                doc,
+                SegmentationLabels.FUNDING,
+                teiFormatter,
+                resCitations,
+                config);
+
+            if (fundingStmt.length() > 0) {
+                tei.append(fundingStmt);
+            }
+
 			tei = teiFormatter.toTEIAnnex(tei, reseAnnex, resHeader, resCitations,
 				tokenizationsAnnex, markerTypes, doc, config);