[api-minor][Editor] When switching to editing mode, redraw pages cont…

…aining editable annotations

Right now, editable annotations are using their own canvas when they're drawn, but
it induces several issues:
 - if the annotation has to be composed with the page then the canvas must be correctly
   composed with its parent. That means we should move the canvas under canvasWrapper
   and we should extract composing info from the drawing instructions...
   Currently it's the case with highlight annotations.
 - we use some extra memory for those canvas even if the user will never edit them, which
   the case for example when opening a pdf in Fenix.

So with this patch, all the editable annotations are drawn on the canvas. When the
user switches to editing mode, then the pages with some editable annotations are redrawn but
without them: they'll be replaced by their counterpart in the annotation editor layer.

src/core/annotation.js

@@ -680,6 +680,7 @@ class Annotation {
       hasOwnCanvas: false,
       noRotate: !!(this.flags & AnnotationFlag.NOROTATE),
       noHTML: isLocked && isContentLocked,
+      isEditable: false,
     };
 
     if (params.collectFields) {
@@ -776,6 +777,10 @@ class Annotation {
     return this.printable;
   }
 
+  mustBeViewedWhenEditing() {
+    return !this.data.isEditable;
+  }
+
   /**
    * @type {boolean}
    */
@@ -3794,7 +3799,8 @@ class FreeTextAnnotation extends MarkupAnnotation {
     // It uses its own canvas in order to be hidden if edited.
     // But if it has the noHTML flag, it means that we don't want to be able
     // to modify it so we can just draw it on the main canvas.
-    this.data.hasOwnCanvas = !this.data.noHTML;
+    this.data.hasOwnCanvas = this.data.noRotate;
+    this.data.isEditable = !this.data.noHTML;
     // We want to be able to add mouse listeners to the annotation.
     this.data.noHTML = false;
 

src/core/document.js

@@ -411,6 +411,8 @@ class Page {
     intent,
     cacheKey,
     annotationStorage = null,
+    isEditing = false,
+    modifiedIds = null,
   }) {
     const contentStreamPromise = this.getContentStream();
     const resourcesPromise = this.loadResources([
@@ -579,7 +581,9 @@ class Page {
         if (
           intentAny ||
           (intentDisplay &&
-            annotation.mustBeViewed(annotationStorage, renderForms)) ||
+            annotation.mustBeViewed(annotationStorage, renderForms) &&
+            ((isEditing && annotation.mustBeViewedWhenEditing()) ||
+              (!isEditing && !modifiedIds?.has(annotation.data.id)))) ||
           (intentPrint && annotation.mustBePrinted(annotationStorage))
         ) {
           opListPromises.push(

src/core/worker.js

@@ -752,6 +752,8 @@ class WorkerMessageHandler {
             intent: data.intent,
             cacheKey: data.cacheKey,
             annotationStorage: data.annotationStorage,
+            isEditing: data.isEditing,
+            modifiedIds: data.modifiedIds,
           })
           .then(
             function (operatorListInfo) {

src/display/annotation_layer.js

@@ -198,6 +198,10 @@ class AnnotationElement {
     return !!(titleObj?.str || contentsObj?.str || richText?.str);
   }
 
+  get _isEditable() {
+    return this.data.isEditable;
+  }
+
   get hasPopupData() {
     return AnnotationElement._hasPopupData(this.data);
   }
@@ -734,10 +738,6 @@ class AnnotationElement {
     }
   }
 
-  get _isEditable() {
-    return false;
-  }
-
   _editOnDoubleClick() {
     if (!this._isEditable) {
       return;
@@ -2530,10 +2530,6 @@ class FreeTextAnnotationElement extends AnnotationElement {
 
     return this.container;
   }
-
-  get _isEditable() {
-    return this.data.hasOwnCanvas;
-  }
 }
 
 class LineAnnotationElement extends AnnotationElement {
@@ -3107,6 +3103,10 @@ class AnnotationLayer {
     }
   }
 
+  hasEditableAnnotations() {
+    return this.#editableAnnotations.size > 0;
+  }
+
   #appendElement(element, id) {
     const contentElement = element.firstChild || element;
     contentElement.id = `${AnnotationPrefix}${id}`;
@@ -3188,7 +3188,7 @@ class AnnotationLayer {
       }
       this.#appendElement(rendered, data.id);
 
-      if (element.annotationEditorType > 0) {
+      if (element._isEditable) {
         this.#editableAnnotations.set(element.data.id, element);
         this._annotationEditorUIManager?.renderAnnotationElement(element);
       }

src/display/annotation_storage.js

@@ -29,6 +29,8 @@ const SerializableEmpty = Object.freeze({
 class AnnotationStorage {
   #modified = false;
 
+  #modifiedIds = null;
+
   #storage = new Map();
 
   constructor() {
@@ -248,6 +250,28 @@ class AnnotationStorage {
     }
     return stats;
   }
+
+  resetModifiedIds() {
+    this.#modifiedIds = null;
+  }
+
+  get modifiedIds() {
+    if (this.#modifiedIds) {
+      return this.#modifiedIds;
+    }
+    const ids = (this.#modifiedIds = new Set());
+    for (const value of this.#storage.values()) {
+      if (
+        !(value instanceof AnnotationEditor) ||
+        !value.annotationElementId ||
+        !value.serialize()
+      ) {
+        continue;
+      }
+      ids.add(value.annotationElementId);
+    }
+    return ids;
+  }
 }
 
 /**

src/display/api.js

@@ -1229,6 +1229,7 @@ class PDFDocumentProxy {
  * @property {Map<string, HTMLCanvasElement>} [annotationCanvasMap] - Map some
  *   annotation ids with canvases used to render them.
  * @property {PrintAnnotationStorage} [printAnnotationStorage]
+ * @property {boolean} [noCache] - Don't use the cached page if any.
  */
 
 /**
@@ -1422,13 +1423,17 @@ class PDFPageProxy {
     annotationCanvasMap = null,
     pageColors = null,
     printAnnotationStorage = null,
+    noCache = false,
+    isEditing = false,
   }) {
     this._stats?.time("Overall");
 
     const intentArgs = this._transport.getRenderingIntent(
       intent,
       annotationMode,
-      printAnnotationStorage
+      printAnnotationStorage,
+      /* isOpList = */ false,
+      isEditing
     );
     const { renderingIntent, cacheKey } = intentArgs;
     // If there was a pending destroy, cancel it so no cleanup happens during
@@ -1440,7 +1445,7 @@ class PDFPageProxy {
     optionalContentConfigPromise ||=
       this._transport.getOptionalContentConfig(renderingIntent);
 
-    let intentState = this._intentStates.get(cacheKey);
+    let intentState = noCache ? null : this._intentStates.get(cacheKey);
     if (!intentState) {
       intentState = Object.create(null);
       this._intentStates.set(cacheKey, intentState);
@@ -1814,6 +1819,8 @@ class PDFPageProxy {
     renderingIntent,
     cacheKey,
     annotationStorageSerializable,
+    modifiedIds,
+    isEditing,
   }) {
     if (typeof PDFJSDev === "undefined" || PDFJSDev.test("TESTING")) {
       assert(
@@ -1830,6 +1837,8 @@ class PDFPageProxy {
         intent: renderingIntent,
         cacheKey,
         annotationStorage: map,
+        modifiedIds,
+        isEditing,
       },
       transfer
     );
@@ -2427,10 +2436,12 @@ class WorkerTransport {
     intent,
     annotationMode = AnnotationMode.ENABLE,
     printAnnotationStorage = null,
-    isOpList = false
+    isOpList = false,
+    isEditing = false
   ) {
     let renderingIntent = RenderingIntentFlag.DISPLAY; // Default value.
     let annotationStorageSerializable = SerializableEmpty;
+    let modifiedIds = null;
 
     switch (intent) {
       case "any":
@@ -2473,10 +2484,16 @@ class WorkerTransport {
       renderingIntent += RenderingIntentFlag.OPLIST;
     }
 
+    if (!isEditing) {
+      modifiedIds = this.annotationStorage.modifiedIds;
+    }
+
     return {
       renderingIntent,
       cacheKey: `${renderingIntent}_${annotationStorageSerializable.hash}`,
       annotationStorageSerializable,
+      modifiedIds,
+      isEditing,
     };
   }