-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
195 additions
and
11 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
185 changes: 185 additions & 0 deletions
185
src/main/java/io/github/meeples10/mcresourceanalyzer/RegionAnalyzerAnvil118.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,185 @@ | ||
| package io.github.meeples10.mcresourceanalyzer; | ||
|
|
||
| import java.io.DataInputStream; | ||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.util.Comparator; | ||
| import java.util.Date; | ||
| import java.util.HashMap; | ||
| import java.util.LinkedHashMap; | ||
| import java.util.Map; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| import net.minecraft.nbt.CompressedStreamTools; | ||
| import net.minecraft.nbt.NBTTagCompound; | ||
| import net.minecraft.nbt.NBTTagList; | ||
| import net.minecraft.nbt.NBTTagLongArray; | ||
|
|
||
| public class RegionAnalyzerAnvil118 extends RegionAnalyzer { | ||
|
|
||
| @Override | ||
| public void analyze(File regionDir) { | ||
| if(!regionDir.exists()) { | ||
| System.out.println("Error: No region directory found at " + regionDir.getAbsolutePath()); | ||
| System.exit(1); | ||
| } | ||
| int totalRegions = regionDir.listFiles().length; | ||
| if(totalRegions == 0) { | ||
| System.out.println("Error: Region directory is empty"); | ||
| System.exit(1); | ||
| } | ||
| System.out.println(totalRegions + " regions found"); | ||
| int rnum = 1; | ||
| for(File f : regionDir.listFiles()) { | ||
| long startTime = System.currentTimeMillis(); | ||
| String name = Main.formatRegionName(regionDir, f); | ||
| RegionFile r = new RegionFile(f); | ||
| System.out.print("Scanning region " + name + " [" + rnum + "/" + totalRegions + "] (modified " | ||
| + Main.DATE_FORMAT.format(new Date(r.lastModified())) + ")... "); | ||
| for(int x = 0; x < 32; x++) { | ||
| for(int z = 0; z < 32; z++) { | ||
| if(r.hasChunk(x, z)) { | ||
| try { | ||
| processRegion(r, name, x, z); | ||
| } catch(Exception e) { | ||
| e.printStackTrace(); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| System.out.println( | ||
| "Done (" + String.format("%.2f", (double) (System.currentTimeMillis() - startTime) / 1000) + "s)"); | ||
| rnum++; | ||
| } | ||
| long duration = System.currentTimeMillis() - getStartTime(); | ||
| System.out.println(("Completed analysis in " + Main.millisToHMS(duration) + " (" + chunkCount + " chunks)")); | ||
| long totalBlocks = 0L; | ||
| for(String key : blockCounter.keySet()) { | ||
| totalBlocks += blockCounter.get(key); | ||
| } | ||
| System.out.println("--------------------------------\n" + blockCounter.size() + " unique blocks\n" + totalBlocks | ||
| + " blocks total\n--------------------------------"); | ||
|
|
||
| System.out.print("Sorting data... "); | ||
| heightCounter = heightCounter.entrySet().stream().sorted(Map.Entry.comparingByKey(new Comparator<String>() { | ||
| @Override | ||
| public int compare(String arg0, String arg1) { | ||
| return Long.compare(blockCounter.get(arg1), blockCounter.get(arg0)); | ||
| } | ||
| })).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (e1, e2) -> e1, LinkedHashMap::new)); | ||
| System.out.println("Done"); | ||
|
|
||
| double totalExcludingAir = (double) (totalBlocks | ||
| - (blockCounter.containsKey("minecraft:air") ? blockCounter.get("minecraft:air") : 0) | ||
| - (blockCounter.containsKey("minecraft:cave_air") ? blockCounter.get("minecraft:cave_air") : 0)); | ||
| System.out.print("Generating CSV... "); | ||
| String data = ""; | ||
| if(Main.saveStatistics) { | ||
| data += "chunk-count=" + chunkCount + ",unique-blocks=" + blockCounter.size() + ",total-blocks=" | ||
| + totalBlocks + ",duration-millis=" + duration + ",duration-readable=" + Main.millisToHMS(duration) | ||
| + "\n"; | ||
| } | ||
| data += "id,"; | ||
| int minY = getMinimumY(); | ||
| int maxY = getMaximumY(); | ||
| for(int i = minY; i <= maxY; i++) { | ||
| data += i + ","; | ||
| } | ||
| data += "total,percent_of_total,percent_excluding_air\n"; | ||
| int digits = String.valueOf(blockCounter.size()).length(); | ||
| String completionFormat = "[%0" + digits + "d/%0" + digits + "d]"; | ||
| int keyIndex = 0; | ||
| for(String key : heightCounter.keySet()) { | ||
| keyIndex += 1; | ||
| System.out.print("\rGenerating CSV... " + String.format(completionFormat, keyIndex, blockCounter.size())); | ||
| data += key + ","; | ||
| for(int i = minY; i <= maxY; i++) { | ||
| if(!heightCounter.get(key).containsKey(i)) { | ||
| data += "0,"; | ||
| } else { | ||
| data += heightCounter.get(key).get(i) + ","; | ||
| } | ||
| } | ||
| data += blockCounter.get(key) + "," | ||
| + Main.DECIMAL_FORMAT.format(((double) blockCounter.get(key) / (double) totalBlocks) * 100.0d); | ||
| if(key.equals("minecraft:air") || key.equals("minecraft:cave_air")) { | ||
| data += ",N/A"; | ||
| } else { | ||
| data += "," + Main.DECIMAL_FORMAT.format(((double) blockCounter.get(key) / totalExcludingAir) * 100.0d); | ||
| } | ||
| data += "\n"; | ||
| } | ||
| try { | ||
| File out = new File(Main.getOutputPrefix() + ".csv"); | ||
| Main.writeStringToFile(out, data); | ||
| System.out.println("\nData written to " + out.getAbsolutePath()); | ||
| } catch(IOException e) { | ||
| e.printStackTrace(); | ||
| System.exit(1); | ||
| } | ||
| if(Main.generateTable) { | ||
| try { | ||
| File out = new File(Main.getOutputPrefix() + "_table.html"); | ||
| Main.writeStringToFile(out, generateTable((double) totalBlocks, totalExcludingAir)); | ||
| System.out.println("\nTable written to " + out.getAbsolutePath()); | ||
| } catch(IOException e) { | ||
| e.printStackTrace(); | ||
| System.exit(1); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private void processRegion(RegionFile r, String name, int x, int z) throws Exception { | ||
| DataInputStream chunkDataInputStream = r.getChunkDataInputStream(x, z); | ||
| if(chunkDataInputStream == null) { | ||
| // Skip malformed chunks | ||
| return; | ||
| } | ||
| NBTTagList sections = CompressedStreamTools.read(r.getChunkDataInputStream(x, z)).getTagList("sections", 10); | ||
| analyzeChunk(sections); | ||
| chunkCount++; | ||
| } | ||
|
|
||
| private void analyzeChunk(NBTTagList sections) { | ||
| int i = 0; | ||
| for(; i < sections.tagCount(); i++) { | ||
| NBTTagCompound tag = sections.getCompoundTagAt(i); | ||
|
|
||
| NBTTagLongArray blockStatesTag = ((NBTTagLongArray) tag.getCompoundTag("block_states").getTag("data")); | ||
| long[] blockStates = blockStatesTag == null ? new long[0] : blockStatesTag.get(); | ||
| NBTTagList palette = tag.getCompoundTag("block_states").getTagList("palette", 10); | ||
| if(palette.hasNoTags()) continue; | ||
| int bitLength = Main.bitLength(palette.tagCount() - 1); | ||
| bitLength = bitLength < 4 ? 4 : bitLength; | ||
| int[] blocks = Main.unstream(bitLength, 64, true, blockStates); | ||
| if(blocks.length == 0) continue; // skip empty sections | ||
| int sectionY = tag.getByte("Y"); | ||
|
|
||
| for(int y = 0; y < 16; y++) { | ||
| for(int x = 0; x < 16; x++) { | ||
| for(int z = 0; z < 16; z++) { | ||
| int actualY = sectionY * 16 + y; | ||
| int blockIndex = y * 16 * 16 + z * 16 + x; | ||
| String block = palette.getStringTagAt(blocks[blockIndex]); | ||
| int startIndex = block.indexOf("Name:\""); | ||
| String blockName = block.substring(startIndex + 6, block.indexOf('"', startIndex + 6)); | ||
| if(blockCounter.containsKey(blockName)) { | ||
| blockCounter.put(blockName, blockCounter.get(blockName) + 1L); | ||
| } else { | ||
| blockCounter.put(blockName, 1L); | ||
| } | ||
| if(!heightCounter.containsKey(blockName)) { | ||
| heightCounter.put(blockName, new HashMap<Integer, Long>()); | ||
| } | ||
| if(heightCounter.get(blockName).containsKey(actualY)) { | ||
| heightCounter.get(blockName).put(actualY, heightCounter.get(blockName).get(actualY) + 1L); | ||
| } else { | ||
| heightCounter.get(blockName).put(actualY, 1L); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
| airHack(i, "minecraft:air"); | ||
| } | ||
| } |