improve logic of image cache

bit-bots · Mar 30, 2021 · 7f7163f · 7f7163f
1 parent 75a9b39
commit 7f7163f
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/imagetagger/imagetagger/annotations/static/annotations/js/annotations.js b/imagetagger/imagetagger/annotations/static/annotations/js/annotations.js
@@ -987,11 +987,14 @@ function calculateImageScale() {
   async function preloadImages() {
     let keepImages = [];
     let cacheLoadings = [];
-    for (let imageId = gImageId - PRELOAD_BACKWARD;
-         imageId <= gImageId + PRELOAD_FORWARD;
-         imageId++) {
-      keepImages.push(imageId);
-      cacheLoadings.push(loadImageToCache(imageId));
+    let currentImageIndex = gImageList.indexOf(gImageId);
+    for (let index = currentImageIndex - PRELOAD_BACKWARD;
+         index <= currentImageIndex + PRELOAD_FORWARD;
+         index++) {
+      if (gImageList[index] !== undefined) {
+        keepImages.push(gImageList[index]);
+        cacheLoadings.push(loadImageToCache(gImageList[index]));
+      }
     }
     await Promise.all(cacheLoadings);
     pruneImageCache(keepImages);