1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
package dev.isxander.yacl3.gui.image;
import com.mojang.blaze3d.systems.RenderSystem;
import dev.isxander.yacl3.gui.image.impl.AnimatedDynamicTextureImage;
import dev.isxander.yacl3.impl.utils.YACLConstants;
import net.minecraft.client.Minecraft;
import net.minecraft.resources.ResourceLocation;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.concurrent.*;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.function.Supplier;
public class ImageRendererManager {
private static final ExecutorService SINGLE_THREAD_EXECUTOR = Executors.newSingleThreadExecutor(task -> new Thread(task, "YACL Image Prep"));
private static final Map<ResourceLocation, CompletableFuture<ImageRenderer>> IMAGE_CACHE = new ConcurrentHashMap<>();
static final Map<ResourceLocation, ImageRenderer> PRELOADED_IMAGE_CACHE = new ConcurrentHashMap<>();
static final List<PreloadedImageFactory> PRELOADED_IMAGE_FACTORIES = List.of(
new PreloadedImageFactory(
location -> location.getPath().endsWith(".webp"),
AnimatedDynamicTextureImage::createWEBPFromTexture
),
new PreloadedImageFactory(
location -> location.getPath().endsWith(".gif"),
AnimatedDynamicTextureImage::createGIFFromTexture
)
);
public static <T extends ImageRenderer> Optional<T> getImage(ResourceLocation id) {
if (PRELOADED_IMAGE_CACHE.containsKey(id)) {
return Optional.of((T) PRELOADED_IMAGE_CACHE.get(id));
}
if (IMAGE_CACHE.containsKey(id)) {
return Optional.ofNullable((T) IMAGE_CACHE.get(id).getNow(null));
}
return Optional.empty();
}
@SuppressWarnings("unchecked")
public static <T extends ImageRenderer> CompletableFuture<T> registerOrGetImage(ResourceLocation id, Supplier<ImageRendererFactory> factorySupplier) {
if (PRELOADED_IMAGE_CACHE.containsKey(id)) {
return CompletableFuture.completedFuture((T) PRELOADED_IMAGE_CACHE.get(id));
}
if (IMAGE_CACHE.containsKey(id)) {
return (CompletableFuture<T>) IMAGE_CACHE.get(id);
}
var future = new CompletableFuture<ImageRenderer>();
IMAGE_CACHE.put(id, future);
ImageRendererFactory factory = factorySupplier.get();
SINGLE_THREAD_EXECUTOR.submit(() -> {
Supplier<Optional<ImageRendererFactory.ImageSupplier>> supplier =
factory.requiresOffThreadPreparation()
? new CompletedSupplier<>(safelyPrepareFactory(id, factory))
: () -> safelyPrepareFactory(id, factory);
Minecraft.getInstance().execute(() -> completeImageFactory(id, supplier, future));
});
return (CompletableFuture<T>) future;
}
@Deprecated
public static <T extends ImageRenderer> CompletableFuture<T> registerImage(ResourceLocation id, ImageRendererFactory factory) {
return registerOrGetImage(id, () -> factory);
}
private static <T extends ImageRenderer> void completeImageFactory(ResourceLocation id, Supplier<Optional<ImageRendererFactory.ImageSupplier>> supplier, CompletableFuture<ImageRenderer> future) {
RenderSystem.assertOnRenderThread();
ImageRendererFactory.ImageSupplier completableImage = supplier.get().orElse(null);
if (completableImage == null) {
return;
}
// sanity check - this should never happen
if (future.isDone()) {
YACLConstants.LOGGER.error("Image '{}' was already completed", id);
return;
}
ImageRenderer image;
try {
image = completableImage.completeImage();
} catch (Exception e) {
YACLConstants.LOGGER.error("Failed to create image '{}'", id, e);
return;
}
future.complete(image);
}
public static void closeAll() {
SINGLE_THREAD_EXECUTOR.shutdownNow();
IMAGE_CACHE.values().removeIf(future -> {
if (future.isDone()) {
future.join().close();
}
return true;
});
}
static Optional<ImageRendererFactory.ImageSupplier> safelyPrepareFactory(ResourceLocation id, ImageRendererFactory factory) {
try {
return Optional.of(factory.prepareImage());
} catch (Exception e) {
YACLConstants.LOGGER.error("Failed to prepare image '{}'", id, e);
IMAGE_CACHE.remove(id);
return Optional.empty();
}
}
public record PreloadedImageFactory(Predicate<ResourceLocation> predicate, Function<ResourceLocation, ImageRendererFactory> factory) {
}
private record CompletedSupplier<T>(T get) implements Supplier<T> {
}
}
|
