Allow remapping resources via addons

Author: Doridian

core/src/main/java/de/bluecolored/bluemap/core/resources/pack/ResourcePool.java

@@ -39,6 +39,15 @@ public class ResourcePool<T> {
 
     private final Map<Key, T> pool = new HashMap<>();
     private final Map<Key, ResourcePath<T>> paths = new HashMap<>();
+    private final ResourcePoolMapper mapper;
+
+    public ResourcePool() {
+        this.mapper = null;
+    }
+
+    public ResourcePool(ResourcePoolMapper mapper) {
+        this.mapper = mapper;
+    }
 
     public void put(Key path, T value) {
         put(new ResourcePath<>(path), value);
@@ -70,6 +79,9 @@ public Collection<T> values() {
     }
 
     public boolean contains(Key path) {
+        if (mapper != null) {
+            path = mapper.remapResource(path);
+        }
         return paths.containsKey(path);
     }
 
@@ -78,7 +90,7 @@ public boolean contains(Key path) {
     }
 
     public @Nullable T get(Key path) {
-        ResourcePath<T> rp = paths.get(path);
+        ResourcePath<T> rp = getPath(path);
         return rp == null ? null : rp.getResource(this::get);
     }
 
@@ -88,6 +100,9 @@ public synchronized void remove(Key path) {
     }
 
     public @Nullable ResourcePath<T> getPath(Key path) {
+        if (mapper != null) {
+            path = mapper.remapResource(path);
+        }
         return paths.get(path);
     }
 

core/src/main/java/de/bluecolored/bluemap/core/resources/pack/ResourcePoolMapper.java

@@ -0,0 +1,7 @@
+package de.bluecolored.bluemap.core.resources.pack;
+
+import de.bluecolored.bluemap.core.util.Key;
+
+public interface ResourcePoolMapper {
+    Key remapResource(Key src);
+}

core/src/main/java/de/bluecolored/bluemap/core/resources/pack/resourcepack/ResourcePack.java

@@ -35,6 +35,7 @@
 import de.bluecolored.bluemap.core.resources.pack.Pack;
 import de.bluecolored.bluemap.core.resources.pack.PackVersion;
 import de.bluecolored.bluemap.core.resources.pack.ResourcePool;
+import de.bluecolored.bluemap.core.resources.pack.ResourcePoolMapper;
 import de.bluecolored.bluemap.core.resources.pack.resourcepack.atlas.Atlas;
 import de.bluecolored.bluemap.core.resources.pack.resourcepack.blockstate.BlockState;
 import de.bluecolored.bluemap.core.resources.pack.resourcepack.entitystate.EntityState;
@@ -63,7 +64,7 @@
 import java.util.concurrent.TimeUnit;
 import java.util.function.Predicate;
 
-public class ResourcePack extends Pack {
+public class ResourcePack extends Pack implements ResourcePoolMapper {
 
     public interface Extension<T extends ResourcePackExtension> extends Keyed {
         Registry<Extension<?>> REGISTRY = new Registry<>();
@@ -94,12 +95,12 @@ public interface Extension<T extends ResourcePackExtension> extends Keyed {
     public ResourcePack(PackVersion packVersion) {
         super(packVersion);
 
-        this.atlases = new ResourcePool<>();
-        this.blockStates = new ResourcePool<>();
-        this.entityStates = new ResourcePool<>();
-        this.models = new ResourcePool<>();
-        this.textures = new ResourcePool<>();
-        this.colormaps = new ResourcePool<>();
+        this.atlases = new ResourcePool<>(this);
+        this.blockStates = new ResourcePool<>(this);
+        this.entityStates = new ResourcePool<>(this);
+        this.models = new ResourcePool<>(this);
+        this.textures = new ResourcePool<>(this);
+        this.colormaps = new ResourcePool<>(this);
 
         this.colorCalculatorFactory = new BlockColorCalculatorFactory();
         this.blockPropertiesConfig = new BlockPropertiesConfig();
@@ -115,6 +116,14 @@ public ResourcePack(PackVersion packVersion) {
             extensions.put(extensionType, extensionType.create(this));
     }
 
+    @Override
+    public Key remapResource(Key src) {
+        for (ResourcePackExtension extension : extensions.values()) {
+            src = extension.remapResource(src);
+        }
+        return src;
+    }
+
     @Override
     public synchronized void loadResources(Iterable<Path> roots) throws IOException, InterruptedException {
         Logger.global.logInfo("Loading resources...");

core/src/main/java/de/bluecolored/bluemap/core/resources/pack/resourcepack/ResourcePackExtension.java

@@ -35,4 +35,7 @@ default Set<Key> collectUsedTextureKeys() {
         return Set.of();
     }
 
+    default Key remapResource(Key src) {
+        return src;
+    }
 }