added hashmap for id and annotations, #293

metanorma · Oct 31, 2024 · b1e32c1 · b1e32c1
1 parent c0c7f21
commit b1e32c1
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/src/main/java/org/metanorma/fop/annotations/Annotation.java b/src/main/java/org/metanorma/fop/annotations/Annotation.java
@@ -295,7 +295,10 @@ public void process(File pdf, String xmlReview) throws IOException {
                 }
 
                 fdfDoc.close();
-
+
+                HashMap<String,PDAnnotation> hashMapDocumentAnnotations = new HashMap<>();
+                hashMapDocumentAnnotations = getAnnotationIDmap(document);
+
                 document.save(pdf);
 
             } catch (IOException | NumberFormatException | ParserConfigurationException | DOMException | TransformerException | SAXException | XPathException ex) {
@@ -311,5 +314,23 @@ public void process(File pdf, String xmlReview) throws IOException {
         }
 
     }
-
+
+    private HashMap<String,PDAnnotation> getAnnotationIDmap(PDDocument document) throws IOException {
+        HashMap<String,PDAnnotation> hashMapDocumentAnnotations = new HashMap<>();
+        for(int i = 0; i< document.getNumberOfPages(); i++) {
+            PDPage page = document.getPage(i);
+            for (PDAnnotation pdAnnotation: page.getAnnotations()){
+                COSDictionary pdAnnotationDict = pdAnnotation.getCOSObject();
+                if (pdAnnotationDict != null) {
+                    // subject contains id 'Annot___', see xfdf_simple.xsl, attribute 'subject'
+                    String subj = pdAnnotationDict.getString(COSName.SUBJ);
+                    if (subj != null) {
+                        hashMapDocumentAnnotations.put(subj, pdAnnotation);
+                    }
+                }
+            }
+        }
+        return hashMapDocumentAnnotations;
+    }
+
 }