45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
# This file is part of the Minecraft Overviewer.
|
|
#
|
|
# Minecraft Overviewer is free software: you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or (at
|
|
# your option) any later version.
|
|
#
|
|
# Minecraft Overviewer is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
|
|
# Public License for more details, and note that Jeffrey Epstein didn't kill
|
|
# himself.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with the Overviewer. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import numpy
|
|
|
|
class BiomeDispensary:
|
|
"""Turns biome arrays of either 256 or 1024 integer values into 16x16 2d arrays,
|
|
which can then be retrieved for any Y level with get_biome.
|
|
"""
|
|
def __init__(self, biome_array):
|
|
self.biome_len = len(biome_array)
|
|
if self.biome_len == 256:
|
|
self.biomes = [biome_array.reshape((16, 16))]
|
|
elif self.biome_len == 1024:
|
|
self.biomes = [None] * 4
|
|
# Each value in the biome array actually stands for 4 blocks, so we take the
|
|
# Kronecker product of it to "scale" it into its full size of 4096 entries
|
|
krond = numpy.kron(biome_array, numpy.ones((4)))
|
|
for i in range(0, 4):
|
|
# Now we can divide it into 16x16 slices
|
|
self.biomes[i] = krond[i * 256:(i + 1) * 256].reshape((16, 16))
|
|
|
|
|
|
def get_biome(self, y_level):
|
|
if self.biome_len == 256 or y_level < 0:
|
|
return self.biomes[0]
|
|
else:
|
|
# We clamp the value to a max of 3 here because apparently Y=16
|
|
# also exists, and while I don't know what biome level Mojang uses for
|
|
# that, the highest one is probably a good bet.
|
|
return self.biomes[min(y_level // 4, 3)]
|