From dd262d147e621a6e0973f8a7a8b39df177090520 Mon Sep 17 00:00:00 2001
From: Davide Ramo <davide.ramo@finantix.com>
Date: Tue, 26 Jan 2016 11:44:28 +0100
Subject: [PATCH] Fix #30 Created another map to link Forms Widget with proper
 page starting from Catalogue -> Pages -> Annots. If no link to parent is
 present in the Form Widget, the new map is used

---
 ILPDFKit/Model/PDFFormContainer.m | 42 +++++++++++++++++++++++--------
 1 file changed, 31 insertions(+), 11 deletions(-)

diff --git a/ILPDFKit/Model/PDFFormContainer.m b/ILPDFKit/Model/PDFFormContainer.m
index 3d8c34d..b5f6822 100644
--- a/ILPDFKit/Model/PDFFormContainer.m
+++ b/ILPDFKit/Model/PDFFormContainer.m
@@ -26,8 +26,11 @@
 @interface PDFFormContainer(Private)
 - (void)populateNameTreeNode:(NSMutableDictionary *)node withComponents:(NSArray *)components final:(PDFForm *)final;
 - (NSArray *)formsDescendingFromTreeNode:(NSDictionary *)node;
-- (void)applyAnnotationTypeLeafToForms:(PDFDictionary *)leaf parent:(PDFDictionary *)parent pageMap:(NSDictionary *)pmap;
-- (void)enumerateFields:(PDFDictionary *)fieldDict pageMap:(NSDictionary *)pmap;
+
+- (void)applyAnnotationTypeLeafToForms:(PDFDictionary *)leaf parent:(PDFDictionary *)parent pageMap:(NSDictionary *)pmap annotationsMap:(NSDictionary *)amap;
+
+- (void)enumerateFields:(PDFDictionary *)fieldDict pageMap:(NSDictionary *)pmap annotationsMap:(NSDictionary *)amap;
+
 - (NSArray *)allForms;
 - (NSString *)formXMLForFormsWithRootNode:(NSDictionary *)node;
 - (void)addForm:(PDFForm *)form;
@@ -49,11 +52,21 @@ - (instancetype)initWithParentDocument:(PDFDocument *)parent {
         _nameTree = [[NSMutableDictionary alloc] init];
         _document = parent;
         NSMutableDictionary *pmap = [NSMutableDictionary dictionary];
-        for (PDFPage *page in _document.pages) {
+        NSMutableDictionary *amap = [NSMutableDictionary dictionary];
+        for (PDFPage *page in _document.pages)
+        {
+            PDFArray* annotations = page.dictionary[@"Annots"];
+            for (PDFDictionary* annotation in annotations)
+            {
+                amap[@((NSUInteger)(annotation.dict))] = @(page.pageNumber);
+            }
+            
             pmap[@((NSUInteger)(page.dictionary.dict))] = @(page.pageNumber);
         }
-        for (PDFDictionary *field in _document.catalog[@"AcroForm"][@"Fields"]) {
-            [self enumerateFields:field pageMap:pmap];
+        
+        for (PDFDictionary *field in _document.catalog[@"AcroForm"][@"Fields"])
+        {
+            [self enumerateFields:field pageMap:pmap annotationsMap:amap];
         }
     }
     return self;
@@ -113,23 +126,30 @@ - (void)removeForm:(PDFForm *)form {
 
 #pragma mark - Private
 
-- (void)enumerateFields:(PDFDictionary *)fieldDict pageMap:(NSDictionary *)pmap {
+- (void)enumerateFields:(PDFDictionary *)fieldDict pageMap:(NSDictionary *)pmap annotationsMap:(NSDictionary *)amap{
     if (fieldDict[@"Subtype"]) {
         PDFDictionary *parent = fieldDict.parent;
-        [self applyAnnotationTypeLeafToForms:fieldDict parent:parent pageMap:pmap];
+        [self applyAnnotationTypeLeafToForms:fieldDict parent:parent pageMap:pmap annotationsMap:amap];
     } else {
         for (PDFDictionary *innerFieldDictionary in fieldDict[@"Kids"]) {
             PDFDictionary *parent = innerFieldDictionary.parent;
-            if (parent != nil) [self enumerateFields:innerFieldDictionary pageMap:pmap];
-            else [self applyAnnotationTypeLeafToForms:innerFieldDictionary parent:fieldDict pageMap:pmap];
+            if (parent != nil) [self enumerateFields:innerFieldDictionary pageMap:pmap annotationsMap:amap];
+            else [self applyAnnotationTypeLeafToForms:innerFieldDictionary parent:fieldDict pageMap:pmap annotationsMap:amap];
         }
     }
 }
 
-- (void)applyAnnotationTypeLeafToForms:(PDFDictionary *)leaf parent:(PDFDictionary *)parent pageMap:(NSDictionary *)pmap {
+- (void)applyAnnotationTypeLeafToForms:(PDFDictionary *)leaf
+                                parent:(PDFDictionary *)parent
+                               pageMap:(NSDictionary *)pmap
+                        annotationsMap:(NSDictionary *)amap
+{
     NSUInteger targ = (NSUInteger)(((PDFDictionary *)(leaf[@"P"])).dict);
+    NSUInteger aarg = (NSUInteger)(PDFDictionary *)leaf.dict;
     leaf.parent = parent;
-    NSUInteger index = targ ? ([pmap[@(targ)] unsignedIntegerValue] - 1):0;
+    
+
+    NSUInteger index = targ ? ([pmap[@(targ)] unsignedIntegerValue] - 1):([amap[@(aarg)] unsignedIntegerValue] - 1);
     PDFForm *form = [[PDFForm alloc] initWithFieldDictionary:leaf page:_document.pages[index] parent:self];
     [self addForm:form];
 }