Temp - add cache info WIP

Author: NicoKiaru

src/main/java/sc/fiji/bdvpg/cache/AbstractGlobalCache.java

@@ -149,4 +149,34 @@ else if (data instanceof IntAccess) {
 
 	public abstract <V> void touch(GlobalCacheKey key, V value);
 
+	/**
+	 * Statistics for cached data for a specific source and timepoint.
+	 */
+	public static class CacheStats {
+		public final long numberOfCells;
+		public final long sizeInBytes;
+
+		public CacheStats(long numberOfCells, long sizeInBytes) {
+			this.numberOfCells = numberOfCells;
+			this.sizeInBytes = sizeInBytes;
+		}
+
+		public String getSizeInMB() {
+			return String.format("%.2f", sizeInBytes / (1024.0 * 1024.0));
+		}
+
+		public String getSizeInKB() {
+			return String.format("%.2f", sizeInBytes / 1024.0);
+		}
+	}
+
+	/**
+	 * Get cache statistics for a specific source and timepoint.
+	 *
+	 * @param source the source object
+	 * @param timepoint the timepoint (-1 for all timepoints)
+	 * @return cache statistics
+	 */
+	abstract public CacheStats getCacheStats(Object source, int setupId, int timepoint);
+
 }

src/main/java/sc/fiji/bdvpg/cache/BoundedLinkedHashMapGlobalCache.java

@@ -29,10 +29,12 @@
 
 package sc.fiji.bdvpg.cache;
 
+import bdv.img.cache.VolatileGlobalCellCache;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.lang.ref.SoftReference;
+import java.lang.reflect.Field;
 import java.util.HashMap;
 import java.util.LinkedHashMap;
 import java.util.Map;
@@ -193,4 +195,56 @@ public String toString() {
 			" %)";
 	}
 
+	@Override
+	public CacheStats getCacheStats(Object source, int setupid, int timepoint) {
+		long totalSize = 0;
+		long cellCount = 0;
+        try {
+
+            Field tpField = VolatileGlobalCellCache.Key.class.getDeclaredField("timepoint");
+            tpField.setAccessible(true);
+            Field setupField = VolatileGlobalCellCache.Key.class.getDeclaredField("setup");
+            setupField.setAccessible(true);
+
+            synchronized (cache) {
+                for (Map.Entry<GlobalCacheKey, SoftReference<Object>> entry : cache.entrySet()) {
+                    GlobalCacheKey key = entry.getKey();
+                    VolatileGlobalCellCache.Key innerKey = null;
+                    if (key.key.get() instanceof VolatileGlobalCellCache.Key) {
+                        innerKey = (VolatileGlobalCellCache.Key) key.key.get();
+                    }
+                    if (innerKey == null) continue;
+                    int tp = (int) tpField.get(innerKey);
+                    int setupId = (int) setupField.get(innerKey);
+
+                    if (timepoint == -1) {
+                        // Match source for any timepoint
+                        if ((key.getSource() == source) && (setupid == setupId)) {
+                            Long cost = cache.cost.get(key);
+                            if (cost != null) {
+                                totalSize += cost;
+                                cellCount++;
+                            }
+                        }
+                    } else {
+                        // Match source and specific timepoint (any level)
+                        if ((key.getSource() == source) && (tp == timepoint) && (setupid == setupId)) {
+                            Long cost = cache.cost.get(key);
+                            if (cost != null) {
+                                totalSize += cost;
+                                cellCount++;
+                            }
+                        }
+                    }
+
+                }
+                return new CacheStats(cellCount, totalSize);
+            }
+
+        } catch (Exception e) {
+            e.printStackTrace();
+            return new CacheStats(0,0);
+        }
+	}
+
 }

src/main/java/sc/fiji/bdvpg/cache/CaffeineGlobalCache.java

@@ -129,4 +129,33 @@ public String toString() {
 			(int) (100.0 * (double) totalBytes / (double) maxCacheSize) + " %)";
 	}
 
+	@Override
+	public CacheStats getCacheStats(Object source, int setupid, int timepoint) {
+		long totalSize = 0;
+		long cellCount = 0;
+
+		for (GlobalCacheKey key : cache.asMap().keySet()) {
+			if (timepoint == -1) {
+				// Match source for any timepoint
+				if (key.source.get() == source) {
+					Object value = cache.getIfPresent(key);
+					if (value != null) {
+						totalSize += getWeight(value);
+						cellCount++;
+					}
+				}
+			} else {				// Match source and specific timepoint (any level)
+                if (key.getSource() == source && key.getTimepoint() == timepoint) {
+                    Object value = cache.getIfPresent(key);
+                    if (value != null) {
+                        totalSize += getWeight(value);
+                        cellCount++;
+                    }
+                }
+            }
+		}
+
+		return new CacheStats(cellCount, totalSize);
+	}
+
 }

src/main/java/sc/fiji/bdvpg/cache/GlobalCacheKey.java

@@ -33,13 +33,13 @@
 
 public class GlobalCacheKey {
 
-	private final WeakReference<Object> source;
+	protected final WeakReference<Object> source;
 
 	private final int timepoint;
 
 	private final int level;
 
-	public final WeakReference<Object> key;
+	protected final WeakReference<Object> key;
 
 	public GlobalCacheKey(final Object source, final int timepoint,
 		final int level, final Object key)
@@ -86,4 +86,16 @@ public boolean equals(final Object other) {
 	public int hashCode() {
 		return hashcode;
 	}
+
+	public int getTimepoint() {
+		return timepoint;
+	}
+
+	public int getLevel() {
+		return level;
+	}
+
+	public Object getSource() {
+		return source.get();
+	}
 }

src/main/java/sc/fiji/bdvpg/scijava/services/ui/SourceAndConverterInspector.java

@@ -49,6 +49,7 @@
 import net.imglib2.realtransform.RealTransform;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import sc.fiji.bdvpg.cache.AbstractGlobalCache;
 import sc.fiji.bdvpg.scijava.services.SourceAndConverterService;
 import sc.fiji.bdvpg.services.ISourceAndConverterService;
 import sc.fiji.bdvpg.services.SourceAndConverterServices;
@@ -586,7 +587,7 @@ private static void appendAttributesInfo(DefaultMutableTreeNode parent,
 	private static void appendViewRegistrationsInfo(DefaultMutableTreeNode parent, AbstractSpimData<?> asd, int setupId, SourceAndConverter<?> sac) {
 
         ViewRegistrations vrs = asd.getViewRegistrations();
-        DefaultMutableTreeNode registrationsNode = new DefaultMutableTreeNode("View Transforms");
+        DefaultMutableTreeNode registrationsNode = new DefaultMutableTreeNode("Data");
         parent.add(registrationsNode);
 
         // Get the source to query dimensions
@@ -670,6 +671,24 @@ public String toString() {
                 } catch (Exception e) {
                     logger.debug("Could not get dimension info for timepoint " + tp.getId() + ": " + e.getMessage());
                 }
+
+                // Add cache statistics for this timepoint
+                try {
+                    AbstractGlobalCache globalCache = SourceAndConverterServices
+                        .getSourceAndConverterService().getCache();
+                    if (globalCache != null) {
+                        // Get the root source object for cache lookup
+                        //Source<?> rootSource = SourceAndConverterHelper.getRootSource(sac.getSpimSource());
+                        AbstractGlobalCache.CacheStats stats = globalCache.getCacheStats(asd, setupId, tp.getId());
+                        //if (stats.numberOfCells > 0) {
+                            String cacheInfo = "Cache: " + stats.numberOfCells + " cells, " +
+                                stats.getSizeInMB() + " MB";
+                            tpNode.add(new DefaultMutableTreeNode(cacheInfo));
+                        //}
+                    }
+                } catch (Exception e) {
+                    logger.debug("Could not get cache stats for timepoint " + tp.getId() + ": " + e.getMessage());
+                }
             }
         }
 	}