Add support for paletted cache cost

SmilyOrg · Sep 18, 2022 · a8679f4 · a8679f4
1 parent 53a6fe5
commit a8679f4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/internal/image/cache.go b/internal/image/cache.go
@@ -1,8 +1,8 @@
 package image
 
 import (
-	"fmt"
 	"image"
+	"log"
 	"photofield/internal/metrics"
 	"reflect"
 	"time"
@@ -93,11 +93,18 @@ func newImageCache(caches Caches) ImageCache {
 				return int64(unsafe.Sizeof(*img)) +
 					int64(cap(img.Pix))*int64(unsafe.Sizeof(img.Pix[0]))
 
+			case *image.Paletted:
+				return int64(unsafe.Sizeof(*img)) +
+					int64(cap(img.Pix))*int64(unsafe.Sizeof(img.Pix[0])) +
+					int64(cap(img.Palette))*int64(unsafe.Sizeof(img.Pix[0]))
+
 			case nil:
 				return 1
 
 			default:
-				panic(fmt.Sprintf("Unable to compute cost, unsupported image format %v", reflect.TypeOf(img)))
+				log.Printf("Unable to compute cost, unsupported image format %v", reflect.TypeOf(img))
+				// Fallback image size (10MB)
+				return 10000000
 			}
 		},
 	})