org.pepsoft.worldpainter.tools.UpdateMaterials Maven / Gradle / Ivy
package org.pepsoft.worldpainter.tools;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import org.pepsoft.minecraft.Material;
import org.pepsoft.util.CSVDataSource;
import org.pepsoft.worldpainter.ColourScheme;
import org.pepsoft.worldpainter.Configuration;
import org.pepsoft.worldpainter.biomeschemes.BiomeSchemeManager;
import org.pepsoft.worldpainter.dynmap.DynmapColourScheme;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import static com.google.common.base.Strings.isNullOrEmpty;
import static java.lang.Math.max;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.Objects.requireNonNull;
import static java.util.zip.ZipFile.OPEN_READ;
import static org.pepsoft.minecraft.Constants.*;
import static org.pepsoft.minecraft.Material.DIRT_PATH;
import static org.pepsoft.minecraft.MaterialImporter.*;
import static org.pepsoft.worldpainter.Constants.UNKNOWN_MATERIAL_COLOUR;
@SuppressWarnings("ConstantConditions")
public class UpdateMaterials {
public static void main(String[] args) throws IOException, ClassNotFoundException {
enrichMaterialsFromPluginDump();
}
private static void enrichMaterialsWithTransmissiveness() throws IOException, ClassNotFoundException {
Configuration config = Configuration.load();
if (config == null) {
config = new Configuration();
}
Configuration.setInstance(config);
try (Reader in = new InputStreamReader(Material.class.getResourceAsStream("mc-materials.csv"), UTF_8); Writer out = new OutputStreamWriter(System.out, UTF_8)) {
CSVDataSource csvIn = new CSVDataSource();
csvIn.openForReading(in);
CSVDataSource csvOut = new CSVDataSource();
csvOut.openForWriting(out, "name", "discriminator", "properties", "opacity", "receivesLight", "terrain", "insubstantial", "veryInsubstantial", "resource", "tileEntity", "tileEntityId", "treeRelated", "vegetation", "blockLight", "natural", "watery", "colour", "colourOrigin");
do {
String name = csvIn.getString("name");
csvOut.setString("name", name);
csvOut.setString("discriminator", csvIn.getString("discriminator"));
csvOut.setString("properties", csvIn.getString("properties"));
csvOut.setInt("opacity", csvIn.getInt("opacity"));
csvOut.setBoolean("receivesLight", guessReceivesLight(name));
csvOut.setBoolean("terrain", csvIn.getBoolean("terrain"));
csvOut.setBoolean("insubstantial", csvIn.getBoolean("insubstantial"));
csvOut.setBoolean("veryInsubstantial", csvIn.getBoolean("veryInsubstantial"));
csvOut.setBoolean("resource", csvIn.getBoolean("resource"));
csvOut.setBoolean("tileEntity", csvIn.getBoolean("tileEntity"));
csvOut.setString("tileEntityId", csvIn.getString("tileEntityId"));
csvOut.setBoolean("treeRelated", csvIn.getBoolean("treeRelated"));
csvOut.setBoolean("vegetation", csvIn.getBoolean("vegetation"));
csvOut.setInt("blockLight", csvIn.getInt("blockLight"));
csvOut.setBoolean("natural", csvIn.getBoolean("natural"));
csvOut.setBoolean("watery", csvIn.getBoolean("watery"));
csvOut.setString("colour", csvIn.getString("colour"));
csvOut.setString("colourOrigin", csvIn.getString("colourOrigin"));
csvIn.next();
csvOut.next();
} while (! csvIn.isEndOfFile());
}
}
private static void enrichMaterialsWithColours() throws IOException, ClassNotFoundException {
Configuration config = Configuration.load();
if (config == null) {
config = new Configuration();
}
Configuration.setInstance(config);
try (JarFile jarFile = new JarFile(BiomeSchemeManager.getLatestMinecraftJar(), true, OPEN_READ); Reader in = new InputStreamReader(Material.class.getResourceAsStream("mc-materials.csv"), UTF_8); Writer out = new OutputStreamWriter(System.out, UTF_8)) {
System.out.printf("Using jar: %s%n%n", jarFile.getName());
CSVDataSource csvIn = new CSVDataSource();
csvIn.openForReading(in);
CSVDataSource csvOut = new CSVDataSource();
csvOut.openForWriting(out, "name", "discriminator", "properties", "opacity", "terrain", "insubstantial", "veryInsubstantial", "resource", "tileEntity", "treeRelated", "vegetation", "blockLight", "natural", "watery", "colour", "colourOrigin");
do {
String name = csvIn.getString("name");
csvOut.setString("name", name);
String str = csvIn.getString("discriminator");
if (! isNullOrEmpty(str)) {
csvOut.setString("discriminator", str);
}
str = csvIn.getString("properties");
if (! isNullOrEmpty(str)) {
csvOut.setString("properties", str);
}
csvOut.setInt("opacity", csvIn.getInt("opacity"));
csvOut.setBoolean("terrain", csvIn.getBoolean("terrain"));
csvOut.setBoolean("insubstantial", csvIn.getBoolean("insubstantial"));
csvOut.setBoolean("veryInsubstantial", csvIn.getBoolean("veryInsubstantial"));
csvOut.setBoolean("resource", csvIn.getBoolean("resource"));
csvOut.setBoolean("tileEntity", csvIn.getBoolean("tileEntity"));
csvOut.setBoolean("treeRelated", csvIn.getBoolean("treeRelated"));
csvOut.setBoolean("vegetation", csvIn.getBoolean("vegetation"));
csvOut.setInt("blockLight", csvIn.getInt("blockLight"));
csvOut.setBoolean("natural", csvIn.getBoolean("natural"));
csvOut.setBoolean("watery", csvIn.getBoolean("watery"));
if (! isNullOrEmpty(csvIn.getString("colour"))) {
csvOut.setString("colour", csvIn.getString("colour"));
if (csvIn.getString("colourOrigin") != null) {
csvOut.setString("colourOrigin", csvIn.getString("colourOrigin"));
}
} else {
Material material = Material.getPrototype(name);
if (material == null) {
material = Material.get(name);
}
final ColourAndOrigin colourAndOrigin = determineColour(material, jarFile);
if (colourAndOrigin != null) {
csvOut.setString("colour", String.format("%8x", colourAndOrigin.colour));
csvOut.setString("colourOrigin", colourAndOrigin.origin);
}
}
csvIn.next();
csvOut.next();
} while (! csvIn.isEndOfFile());
}
}
private static void enrichMaterialsFromPluginDump() throws IOException, ClassNotFoundException {
Configuration config = Configuration.load();
if (config == null) {
config = new Configuration();
}
Configuration.setInstance(config);
try (JarFile jarFile = new JarFile(BiomeSchemeManager.getLatestMinecraftJar(), true, OPEN_READ); Reader in = new InputStreamReader(Material.class.getResourceAsStream("materials.csv"), UTF_8); Writer out = new OutputStreamWriter(System.out, UTF_8)) {
final CSVDataSource csvIn = new CSVDataSource();
final CSVDataSource csvOut = new CSVDataSource();
csvIn.openForReading(in);
csvOut.openForWriting(out, "name", "discriminator", "properties", "opacity", "receivesLight", "terrain", "insubstantial", "veryInsubstantial", "resource", "tileEntity", "tileEntityId", "treeRelated", "vegetation", "blockLight", "natural", "watery", "colour", "colourOrigin");
do {
final String name = csvIn.getString("name");
final String properties = csvIn.getString("properties");
final boolean tileEntity = csvIn.getBoolean("tileEntity");
final String tileEntityId = csvIn.getString("tileEntityId");
final int blockLight = csvIn.getInt("blockLight");
if (! MATERIAL_SPECS.containsKey(name)) {
csvOut.setString("name", name);
if (! isNullOrEmpty(properties)) {
csvOut.setString("properties", properties);
}
csvOut.setInt("opacity", guessOpacity(name));
csvOut.setBoolean("receivesLight", guessReceivesLight(name));
csvOut.setBoolean("terrain", false);
csvOut.setBoolean("insubstantial", guessInsubstantial(name));
csvOut.setBoolean("veryInsubstantial", guessVeryInsubstantial(name));
csvOut.setBoolean("resource", guessResource(name));
csvOut.setBoolean("tileEntity", tileEntity);
if (! isNullOrEmpty(tileEntityId)) {
csvOut.setString("tileEntityId", tileEntityId);
}
csvOut.setBoolean("treeRelated", guessTreeRelated(name));
csvOut.setBoolean("vegetation", guessVegetation(name));
csvOut.setInt("blockLight", blockLight);
csvOut.setBoolean("natural", guessNatural(name));
csvOut.setBoolean("watery", false);
final ColourAndOrigin colourAndOrigin = findAverageColour(jarFile, name.substring(name.indexOf(':') + 1));
if (colourAndOrigin != null) {
csvOut.setString("colour", String.format("%8x", colourAndOrigin.colour));
csvOut.setString("colourOrigin", colourAndOrigin.origin);
}
csvOut.next();
}
csvIn.next();
} while (! csvIn.isEndOfFile());
} catch (IOException e) {
throw new RuntimeException(e);
}
}
public static ColourAndOrigin determineColour(Material material, JarFile jarFile) {
material = PATCHES.getOrDefault(material.name, material);
// Preserve the colours people are used to. TODO: some of these were temporary and are probably wrong!
if (DYNMAP_COLOUR_SCHEME.getColour(material) != UNKNOWN_MATERIAL_COLOUR) {
return new ColourAndOrigin(DYNMAP_COLOUR_SCHEME.getColour(material), "old default");
} else if (HARDCODED_COLOURS.containsKey(material.name)) {
return new ColourAndOrigin(HARDCODED_COLOURS.get(material.name), "hardcoded");
} else {
return findAverageColour(jarFile, material.simpleName);
}
}
private static ColourAndOrigin findAverageColour(JarFile jarFile, String simpleName) {
final JarEntry entry = findJarEntry(jarFile, simpleName);
if (entry != null) {
int red = 0, green = 0, blue = 0;
int pixelCount = 0;
try (InputStream in = jarFile.getInputStream(entry)) {
BufferedImage image = ImageIO.read(in);
for (int x = 0; x < image.getWidth(); x++) {
for (int y = 0; y < image.getHeight(); y++) {
final int colour = image.getRGB(x, y);
if ((colour & 0xff000000) == 0) {
// Ignore transparent pixels
continue;
}
red += (colour & 0xff0000) >> 16;
green += (colour & 0x00ff00) >> 8;
blue += (colour & 0x0000ff);
pixelCount++;
}
}
} catch (IOException e) {
throw new RuntimeException(e);
}
// TODO apply water, leaf, etc. colours!
return new ColourAndOrigin(0xff000000 | ((red / pixelCount) << 16) | ((green / pixelCount) << 8) | (blue / pixelCount), entry.getName().substring(entry.getName().lastIndexOf('/') + 1));
} else {
return null;
}
}
private static JarEntry findJarEntry(JarFile jarFile, String name) {
do {
for (String prefix: PREFIXES) {
String subName = name.startsWith(prefix) ? name.substring(prefix.length()) : name;
for (String extension: EXTENSIONS) {
JarEntry entry = jarFile.getJarEntry("assets/minecraft/textures/block/" + subName + extension + ".png");
if (entry != null) {
return entry;
}
}
}
name = name.substring(0, max(name.lastIndexOf('_'), 0));
} while (! name.isEmpty());
return null;
}
private static final String[] EXTENSIONS = {"", "s", "_block", "_top", "_block_top", "_front", "_planks", "_stage7", "_stage3", "_stage2", "_0"}; // Try adding these
private static final String[] PREFIXES = {"", "infested_", "smooth_"}; // Try removing these
private static final Map HARDCODED_COLOURS = ImmutableMap.builder()
.put(MC_WATER, 0xff3f76e4)
.put(MC_FLOWING_WATER, 0xff3f76e4)
.put(MC_LAVA, 0xfffe432a) // TODO: check
.put(MC_FLOWING_LAVA, 0xfffe432a) // TODO: check
.put(MC_BIRCH_LEAVES, 0xff80a755)
.put(MC_SPRUCE_LEAVES, 0xff619961)
.put(MC_LILY_PAD, 0xff208030)
.put(MC_GRASS_BLOCK, 0xff91bd59)
.put(MC_DIRT_PATH, 0xff91bd59)
.put(MC_GRASS_PATH, 0xff91bd59)
.put(MC_GRASS, 0xff91bd59)
.put(MC_TALL_GRASS, 0xff91bd59)
.put(MC_FERN, 0xff91bd59)
.put(MC_LARGE_FERN, 0xff91bd59)
.put(MC_POTTED_FERN, 0xff91bd59)
.put(MC_SUGAR_CANE, 0xff91bd59)
.put(MC_OAK_LEAVES, 0xff77ab2f)
.put(MC_JUNGLE_LEAVES, 0xff77ab2f)
.put(MC_ACACIA_LEAVES, 0xff77ab2f)
.put(MC_DARK_OAK_LEAVES, 0xff77ab2f)
.put(MC_VINE, 0xff77ab2f)
.put(MC_MELON_STEM, 0xff77ab2f) // TODO
.put(MC_PUMPKIN_STEM, 0xff77ab2f) // TODO
.build();
private static final Map PATCHES = ImmutableMap.of(MC_GRASS_PATH, DIRT_PATH);
private static final ColourScheme DYNMAP_COLOUR_SCHEME = DynmapColourScheme.loadDynMapColourScheme("default", 0);
private static final Map>> MATERIAL_SPECS = new HashMap<>();
static {
// Read MC materials database
try (Reader in = new InputStreamReader(requireNonNull(Material.class.getResourceAsStream("mc-materials.csv")), UTF_8)) {
CSVDataSource csvDataSource = new CSVDataSource();
csvDataSource.openForReading(in);
do {
Map materialSpecs = new HashMap<>();
String name = csvDataSource.getString("name");
materialSpecs.put("name", name);
String str = csvDataSource.getString("discriminator");
if (! isNullOrEmpty(str)) {
materialSpecs.put("discriminator", ImmutableSet.copyOf(str.split(",")));
}
str = csvDataSource.getString("properties");
if (! isNullOrEmpty(str)) {
materialSpecs.put("properties", ImmutableSet.copyOf(str.split(",")));
}
materialSpecs.put("opacity", csvDataSource.getInt("opacity"));
materialSpecs.put("terrain", csvDataSource.getBoolean("terrain"));
materialSpecs.put("insubstantial", csvDataSource.getBoolean("insubstantial"));
materialSpecs.put("veryInsubstantial", csvDataSource.getBoolean("veryInsubstantial"));
materialSpecs.put("resource", csvDataSource.getBoolean("resource"));
materialSpecs.put("tileEntity", csvDataSource.getBoolean("tileEntity"));
materialSpecs.put("treeRelated", csvDataSource.getBoolean("treeRelated"));
materialSpecs.put("vegetation", csvDataSource.getBoolean("vegetation"));
materialSpecs.put("blockLight", csvDataSource.getInt("blockLight"));
materialSpecs.put("natural", csvDataSource.getBoolean("natural"));
materialSpecs.put("watery", csvDataSource.getBoolean("watery"));
str = csvDataSource.getString("colour");
if (! isNullOrEmpty(str)) {
materialSpecs.put("colour", Integer.parseUnsignedInt(str, 16));
}
str = csvDataSource.getString("colourOrigin");
if (! isNullOrEmpty(str)) {
materialSpecs.put("colourOrigin", str);
}
MATERIAL_SPECS.computeIfAbsent(name, s -> new HashSet<>()).add(materialSpecs);
csvDataSource.next();
} while (! csvDataSource.isEndOfFile());
} catch (IOException e) {
throw new RuntimeException("I/O error while reading Minecraft materials database materials.csv from classpath", e);
}
}
public static class ColourAndOrigin {
ColourAndOrigin(int colour, String origin) {
this.colour = 0xff000000 | colour;
this.origin = origin;
}
public final int colour;
public final String origin;
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy