Zleub/texture provider

build.gradle

@@ -134,6 +134,9 @@ neoForge {
 // Include resources generated by data generators.
 sourceSets.main.resources { srcDir 'src/generated/resources' }
 
+processResources {
+    duplicatesStrategy = DuplicatesStrategy.EXCLUDE
+}
 
 dependencies {
     implementation "software.bernie.geckolib:geckolib-neoforge-${minecraft_version}:${geckolib_version}"

src/generated/resources/assets/wotr/textures/block/water_flow_1.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_10.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_11.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_12.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_13.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_14.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_2.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_3.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_4.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_5.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_6.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_7.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_8.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_flow_9.mcmeta

@@ -0,0 +1 @@
+{}

src/generated/resources/assets/wotr/textures/block/water_still_1.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_10.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_11.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_12.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_13.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_14.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_15.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_2.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_3.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_4.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_5.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_6.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_7.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_8.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/generated/resources/assets/wotr/textures/block/water_still_9.mcmeta

@@ -0,0 +1,3 @@
+{
+  "frametime": 2
+}

src/main/java/com/wanderersoftherift/wotr/datagen/DataGenerators.java

@@ -6,6 +6,7 @@
 import net.minecraft.core.RegistrySetBuilder;
 import net.minecraft.core.registries.Registries;
 import net.minecraft.data.loot.LootTableProvider;
+import net.minecraft.server.packs.PackType;
 import net.minecraft.world.damagesource.DamageScaling;
 import net.minecraft.world.damagesource.DamageType;
 import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
@@ -40,6 +41,9 @@ public static void gatherData(GatherDataEvent.Client event) {
                         .add(WotrRegistries.Keys.RUNEGEM_DATA, WotrRuneGemDataProvider::bootstrapRuneGems)
 
         );
+
+        event.createProvider(
+                (output) -> new WotrTextureProvider(event.getResourceManager(PackType.CLIENT_RESOURCES), output));
         event.createProvider(WotrModelProvider::new);
 
         event.createProvider(WotrDataMapProvider::new);

src/main/java/com/wanderersoftherift/wotr/datagen/WotrTextureProvider.java

@@ -0,0 +1,115 @@
+package com.wanderersoftherift.wotr.datagen;
+
+import com.wanderersoftherift.wotr.WanderersOfTheRift;
+import com.wanderersoftherift.wotr.datagen.textures.TextureGenerator;
+import com.wanderersoftherift.wotr.datagen.textures.TextureGeneratorCollector;
+import net.minecraft.data.CachedOutput;
+import net.minecraft.data.DataProvider;
+import net.minecraft.data.PackOutput;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.server.packs.resources.ResourceManager;
+import net.neoforged.neoforge.common.extensions.IModelProviderExtension;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.CompletableFuture;
+
+public class WotrTextureProvider implements DataProvider, IModelProviderExtension {
+    static Map<Integer, Integer> processorBlockColorMap = new HashMap<>();
+    private final ResourceManager resourceManager;
+    private final PackOutput.PathProvider texturePathProvider;
+
+    static {
+        processorBlockColorMap.put(15, 0x5EDFB2);
+        processorBlockColorMap.put(14, 0xC96C7E);
+        processorBlockColorMap.put(13, 0xA2DD7A);
+        processorBlockColorMap.put(12, 0xFFFFFF);
+        processorBlockColorMap.put(11, 0xF66FED);
+        processorBlockColorMap.put(10, 0xC773F2);
+        processorBlockColorMap.put(9, 0x609CF9);
+        processorBlockColorMap.put(8, 0x64D9F5);
+        processorBlockColorMap.put(7, 0x636C7E);
+        processorBlockColorMap.put(6, 0xC9D97E);
+        processorBlockColorMap.put(5, 0x646FF5);
+        processorBlockColorMap.put(4, 0x9360F9);
+        processorBlockColorMap.put(3, 0x63DA7E);
+        processorBlockColorMap.put(2, 0xCE7D67);
+        processorBlockColorMap.put(1, 0x888F9C);
+    }
+
+    enum TextureVariant {
+        Block(""),
+        DirectionPillar("directional_pillar"),
+        DirectionPillarTop("directional_pillar_top"),
+        Glass("glass"),
+        GlassPaneTop("glass_pane_top"),
+        Trapdoor("trapdoor");
+
+        private final String suffix;
+
+        TextureVariant(String suffix) {
+            this.suffix = suffix;
+        }
+
+        String getSuffix() {
+            return this.suffix;
+        }
+    }
+
+    WotrTextureProvider(ResourceManager resourceManager, PackOutput output) {
+        this.texturePathProvider = output.createPathProvider(PackOutput.Target.RESOURCE_PACK, "textures");
+        this.resourceManager = resourceManager;
+    }
+
+    @Override
+    public @NotNull CompletableFuture<?> run(@NotNull CachedOutput cachedOutput) {
+        TextureGeneratorCollector textureCollector = new TextureGeneratorCollector(resourceManager, cachedOutput,
+                texturePathProvider);
+
+        this.registerTextures(new TextureGenerator(textureCollector));
+
+        return textureCollector.save();
+    }
+
+    public void registerTextures(TextureGenerator textureGenerator) {
+        processorBlockColorMap.forEach((number, color) -> {
+            // Make all processor blocks textures
+            Arrays.stream(TextureVariant.values()).toList().forEach((variant) -> {
+                String sourcePath;
+                if (!Objects.equals(variant.getSuffix(), "")) {
+                    sourcePath = "block/processor_block/processor_block_template_" + variant.getSuffix();
+                } else {
+                    sourcePath = "block/processor_block/processor_block_template";
+                }
+
+                textureGenerator.tintGenerator(
+                        ResourceLocation.fromNamespaceAndPath(WanderersOfTheRift.MODID, sourcePath),
+                        ResourceLocation.fromNamespaceAndPath(WanderersOfTheRift.MODID,
+                                sourcePath.replace("template", String.valueOf(number)).replace("processor_block/", "")),
+                        color
+                );
+            });
+
+            // Make all fluid block textures
+            textureGenerator.tintGenerator(
+                    ResourceLocation.withDefaultNamespace("block/water_flow"),
+                    ResourceLocation.fromNamespaceAndPath(WanderersOfTheRift.MODID, "block/water_flow_" + number), color
+            );
+
+            textureGenerator.tintGenerator(
+                    ResourceLocation.withDefaultNamespace("block/water_still"),
+                    ResourceLocation.fromNamespaceAndPath(WanderersOfTheRift.MODID, "block/water_still_" + number),
+                    color
+            );
+
+        });
+    }
+
+    @Override
+    public @NotNull String getName() {
+        return "Wotr Texture Definition";
+    }
+}

src/main/java/com/wanderersoftherift/wotr/datagen/textures/TextureGenerator.java

@@ -0,0 +1,22 @@
+package com.wanderersoftherift.wotr.datagen.textures;
+
+import net.minecraft.resources.ResourceLocation;
+
+import java.awt.Color;
+import java.util.function.Consumer;
+
+public class TextureGenerator {
+    public final Consumer<TextureTransform> textureOutput;
+
+    public TextureGenerator(Consumer<TextureTransform> textureOutput) {
+        this.textureOutput = textureOutput;
+    }
+
+    public void tintGenerator(ResourceLocation sourcePath, ResourceLocation destinationPath, int color) {
+        this.textureOutput.accept(new TintTransform(sourcePath, destinationPath, color));
+    }
+
+    public void tintGenerator(ResourceLocation sourcePath, ResourceLocation destinationPath, Color color) {
+        this.textureOutput.accept(new TintTransform(sourcePath, destinationPath, color));
+    }
+}

src/main/java/com/wanderersoftherift/wotr/datagen/textures/TextureGeneratorCollector.java

@@ -0,0 +1,105 @@
+package com.wanderersoftherift.wotr.datagen.textures;
+
+import com.google.common.hash.Hashing;
+import com.google.common.hash.HashingOutputStream;
+import com.google.gson.JsonElement;
+import com.mojang.serialization.JsonOps;
+import com.wanderersoftherift.wotr.WanderersOfTheRift;
+import net.minecraft.Util;
+import net.minecraft.client.resources.metadata.animation.AnimationMetadataSection;
+import net.minecraft.data.CachedOutput;
+import net.minecraft.data.DataProvider;
+import net.minecraft.data.PackOutput;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.server.packs.resources.Resource;
+import net.minecraft.server.packs.resources.ResourceManager;
+import net.minecraft.server.packs.resources.ResourceMetadata;
+
+import javax.imageio.ImageIO;
+import javax.imageio.stream.ImageInputStream;
+import javax.imageio.stream.ImageOutputStream;
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.concurrent.CompletableFuture;
+import java.util.function.Consumer;
+
+public class TextureGeneratorCollector implements Consumer<TextureTransform> {
+    private final ResourceManager resourceManager;
+    private final CachedOutput output;
+    private final PackOutput.PathProvider pathProvider;
+    private final ArrayList<TextureTransform> transforms;
+
+    public TextureGeneratorCollector(ResourceManager resourceManager, CachedOutput output,
+            PackOutput.PathProvider pathProvider) {
+        this.resourceManager = resourceManager;
+        this.transforms = new ArrayList<>();
+        this.output = output;
+        this.pathProvider = pathProvider;
+    }
+
+    @Override
+    public void accept(TextureTransform textureTransform) {
+        transforms.add(textureTransform);
+    }
+
+    public CompletableFuture<?> applyTransform(TextureTransform transform) {
+        ResourceLocation resourceLocation = transform.sourcePath.withPrefix("textures/").withSuffix(".png");
+        Path pathOut = pathProvider.file(transform.destinationPath, "png");
+
+        ByteArrayOutputStream bytearrayoutputstream = new ByteArrayOutputStream();
+        HashingOutputStream hashingoutputstream = new HashingOutputStream(Hashing.sha1(), bytearrayoutputstream);
+
+        try {
+            Resource input = resourceManager.getResource(resourceLocation).orElseThrow();
+            ResourceMetadata resourceMeta = input.metadata();
+            AnimationMetadataSection textureMeta = resourceMeta.getSection(AnimationMetadataSection.TYPE).orElse(null);
+
+            InputStream r = input.open();
+
+            ImageInputStream imageInputStream = ImageIO.createImageInputStream(r);
+            BufferedImage image = ImageIO.read(imageInputStream);
+
+            int width = image.getWidth();
+            int height = image.getHeight();
+
+            BufferedImage out = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
+            for (int x = 0; x < width; x++) {
+                for (int y = 0; y < height; y++) {
+                    int transform1 = transform.transform(x, y, image.getRGB(x, y));
+                    out.setRGB(x, y, transform1);
+                }
+            }
+
+            ImageOutputStream imageOutputStream = ImageIO.createImageOutputStream(hashingoutputstream);
+            ImageIO.write(out, "png", imageOutputStream);
+
+            if (textureMeta != null) {
+                WanderersOfTheRift.LOGGER.info("{} has metadata", resourceLocation);
+                JsonElement json = AnimationMetadataSection.CODEC.encodeStart(JsonOps.INSTANCE, textureMeta)
+                        .getOrThrow();
+                DataProvider.saveStable(output, json, pathProvider.file(transform.destinationPath, "mcmeta"));
+            }
+
+            output.writeIfNeeded(pathOut, bytearrayoutputstream.toByteArray(), hashingoutputstream.hash());
+
+        } catch (Exception e) {
+            WanderersOfTheRift.LOGGER.warn("WotrTextureProvider - Error with resource {}", resourceLocation);
+        }
+
+        return CompletableFuture.completedFuture(transform.destinationPath);
+    }
+
+    public CompletableFuture<?> applyTransformAsync(TextureTransform transform) {
+        return CompletableFuture.runAsync(() -> applyTransform(transform),
+                Util.backgroundExecutor().forName("saveStable"));
+    }
+
+    public CompletableFuture<?> save() {
+        WanderersOfTheRift.LOGGER.info("TextureGeneratorCollector save {}", transforms.size());
+        return CompletableFuture
+                .allOf(transforms.stream().map(this::applyTransformAsync).toArray(CompletableFuture[]::new));
+    }
+}