avoid NPE

kermitt2 · Dec 30, 2024 · 433a065 · 433a065
1 parent 3e006a6
commit 433a065
Showing 1 changed file with 5 additions and 3 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
@@ -336,7 +336,8 @@ else if (config.getConsolidateCitations() == 2)
 				annexResults = label(annexFeatures);
 				//System.out.println(rese);
 
-				annexFigures = processFigures(annexResults, annexTokenization, doc, bodyFigures.size());
+                int startFigureID = bodyFigures != null ? bodyFigures.size() : 0;
+				annexFigures = processFigures(annexResults, annexTokenization, doc, startFigureID);
 
                 long numberFiguresInAnnex = Arrays.stream(annexResults.split("\n"))
                     .filter(r -> r.endsWith("I-" + TaggingLabels.FIGURE_LABEL))
@@ -355,7 +356,8 @@ else if (config.getConsolidateCitations() == 2)
                     .collect(Collectors.toList());
 				postProcessFigureCaptions(annexFigures, doc);
 
-				annexTables = processTables(annexResults, annexTokenization, doc, bodyTables.size());
+                int startTableID = bodyTables != null ? bodyTables.size() : 0;
+				annexTables = processTables(annexResults, annexTokenization, doc, startTableID);
 
                 long numberTablesInAnnex = Arrays.stream(bodyResults.split("\n"))
                     .filter(r -> r.endsWith("I-" + TaggingLabels.TABLE_LABEL))
@@ -2399,7 +2401,7 @@ protected List<Table> processTables(String rese,
     				}
     			}
     			results.add(result);
-    			result.setId("" + tableId);
+    			result.setId(String.valueOf(tableId));
                 tableId ++;
             }
 		}