World: add support for negative Y coordinates and MC 1.18 chunks

MC 1.18 biomes are not handled yet.
This commit is contained in:
Matthias Schiffer 2021-12-04 00:16:16 +01:00
parent 9ee1007ade
commit aacea26716
Signed by: neocturne
GPG key ID: 16EF3F64CB201D9C
5 changed files with 92 additions and 45 deletions

View file

@ -18,27 +18,35 @@ namespace MinedMap {
namespace World {
Chunk::Chunk(const ChunkData *data) {
std::shared_ptr<const NBT::CompoundTag> level =
assertValue(data->getRoot()->get<NBT::CompoundTag>("Level"));
std::shared_ptr<const NBT::ListTag> sectionsTag;
std::shared_ptr<const NBT::ListTag> sectionsTag = level->get<NBT::ListTag>("Sections");
if (!sectionsTag || sectionsTag->empty())
return;
auto level = data->getRoot()->get<NBT::CompoundTag>("Level");
if (level) {
sectionsTag = level->get<NBT::ListTag>("Sections");
if (!sectionsTag || sectionsTag->empty())
return;
auto biomesIntArray = level->get<NBT::IntArrayTag>("Biomes");
auto biomesByteArray = level->get<NBT::ByteArrayTag>("Biomes");
auto biomesIntArray = level->get<NBT::IntArrayTag>("Biomes");
auto biomesByteArray = level->get<NBT::ByteArrayTag>("Biomes");
if (biomesIntArray && biomesIntArray->getLength() == BSIZE*BSIZE*BMAXY) {
biomeInts = std::move(biomesIntArray);
biomeFormat = INT_ARRAY;
} else if (biomesIntArray && biomesIntArray->getLength() == SIZE*SIZE) {
biomeInts = std::move(biomesIntArray);
biomeFormat = INT_ARRAY_PRE1_15;
} else if (biomesByteArray && biomesByteArray->getLength() == SIZE*SIZE) {
biomeBytes = std::move(biomesByteArray);
biomeFormat = BYTE_ARRAY;
if (biomesIntArray && biomesIntArray->getLength() == BSIZE*BSIZE*BMAXY) {
biomeInts = std::move(biomesIntArray);
biomeFormat = INT_ARRAY;
} else if (biomesIntArray && biomesIntArray->getLength() == SIZE*SIZE) {
biomeInts = std::move(biomesIntArray);
biomeFormat = INT_ARRAY_PRE1_15;
} else if (biomesByteArray && biomesByteArray->getLength() == SIZE*SIZE) {
biomeBytes = std::move(biomesByteArray);
biomeFormat = BYTE_ARRAY;
} else {
throw std::invalid_argument("corrupt biome data");
}
} else {
throw std::invalid_argument("corrupt biome data");
sectionsTag = data->getRoot()->get<NBT::ListTag>("sections");
if (!sectionsTag || sectionsTag->empty())
return;
biomeFormat = SECTION;
}
auto dataVersionTag = data->getRoot()->get<NBT::IntTag>("DataVersion");
@ -48,25 +56,34 @@ Chunk::Chunk(const ChunkData *data) {
auto s = std::dynamic_pointer_cast<const NBT::CompoundTag>(sTag);
std::unique_ptr<Section> section = Section::makeSection(s, dataVersion);
section_idx_t Y = section->getY();
if (sections.empty())
sectionOffset = Y;
Y -= sectionOffset;
assertValue(Y >= 0 && size_t(Y) >= sections.size());
sections.resize(Y);
sections.push_back(std::move(section));
}
}
uint8_t Chunk::getBiome(block_idx_t x, y_idx_t y, block_idx_t z) const {
if (x > SIZE || y > MAXY || z > SIZE)
throw std::invalid_argument("corrupt chunk data");
if (x >= SIZE || z >= SIZE)
throw std::invalid_argument("getBiome: invalid block coordinate");
switch (biomeFormat) {
case INT_ARRAY:
if (y < 0 || y >= MAXY)
break;
return biomeInts->getValue((y>>BSHIFT)*BSIZE*BSIZE + (z>>BSHIFT)*BSIZE + (x>>BSHIFT));
case INT_ARRAY_PRE1_15:
return biomeInts->getValue(z*SIZE + x);
case BYTE_ARRAY:
return biomeBytes->getValue(z*SIZE + x);
default:
return 0xff;
break;
}
return 0xff;
}
Block Chunk::getBlock(block_idx_t x, Chunk::Height height, block_idx_t z) const {
@ -74,16 +91,19 @@ Block Chunk::getBlock(block_idx_t x, Chunk::Height height, block_idx_t z) const
block.depth = height.depth;
section_idx_t Y = height.y >> HSHIFT;
if (height.y == y_idx_min)
return block;
section_idx_t Y = (height.y >> HSHIFT) - sectionOffset;
block_idx_t y = height.y & HMASK;
if (Y < sections.size() && sections[Y])
if (Y >= 0 && size_t(Y) < sections.size() && sections[Y])
block.type = sections[Y]->getBlockStateAt(x, y, z);
section_idx_t Yt = (height.y + 1) >> HSHIFT;
section_idx_t Yt = ((height.y + 1) >> HSHIFT) - sectionOffset;
block_idx_t yt = (height.y + 1) & HMASK;
if (Yt < sections.size() && sections[Yt])
if (Yt >= 0 && size_t(Yt) < sections.size() && sections[Yt])
block.blockLight = sections[Yt]->getBlockLightAt(x, yt, z);
return block;
@ -93,17 +113,17 @@ bool Chunk::getHeight(
Chunk::Height *height, const Section *section,
block_idx_t x, block_idx_t y, block_idx_t z, int flags
) const {
if (height->depth > 0)
if (height->depth > y_idx_min)
return false;
if (!(flags & WITH_DEPTH) && height->y > 0)
if (!(flags & WITH_DEPTH) && height->y > y_idx_min)
return false;
const Resource::BlockType *type = section->getBlockStateAt(x, y, z);
if (!type || !(type->flags & BLOCK_OPAQUE))
return false;
if (height->y == 0)
if (height->y == y_idx_min)
height->y = (section->getY() << HSHIFT) + y;
if (!(flags & WITH_DEPTH))
@ -119,9 +139,14 @@ bool Chunk::getHeight(
Chunk::Heightmap Chunk::getTopLayer(int flags) const {
uint32_t done = 0;
Heightmap ret = {};
Heightmap ret;
for (int16_t Y = sections.size() - 1; Y >= 0; Y--) {
for (block_idx_t z = 0; z < SIZE; z++) {
for (block_idx_t x = 0; x < SIZE; x++)
ret.v[x][z] = Height { .y = y_idx_min, .depth = y_idx_min };
}
for (section_idx_t Y = sections.size() - 1; Y >= 0; Y--) {
if (done == SIZE*SIZE)
break;