LucidCube/Tools/AnvilStats/ChunkExtract.cpp
Rebekah Rowe 6c4b2e9186
Implement GPL3+ and Apache2.0 Dual License.
Commit is being made to allow additions of GPL3+ code previously
un-addable. With these changes, contributions back to cuberite are
possible with the backporting exemtion, as well as adding stuff in
minetest with minetest code properly being read through and implimented
to upgrade it to GPL3 from GPL2.

project still has Apache2.0 license and credits to all its contributers, but now has the freedom of GPL3+ and all the code that can be implimented and shared with it.
2023-03-20 11:49:56 -04:00

121 lines
3.1 KiB
C++

/*
* Copyright 2011-2022 Cuberite Contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
// ChunkExtract.cpp
// Implements the cChunkExtract class representing a cCallback descendant that extracts raw chunk data into separate .chunk files
#include "Globals.h"
#include "ChunkExtract.h"
#include "../../src/OSSupport/GZipFile.h"
cChunkExtract::cChunkExtract(const AString & iWorldFolder) :
mWorldFolder(iWorldFolder)
{
}
bool cChunkExtract::OnNewChunk(int a_ChunkX, int a_ChunkZ)
{
int AnvilX = (a_ChunkX - ((a_ChunkX > 0) ? 0 : 31)) / 32;
int AnvilZ = (a_ChunkZ - ((a_ChunkZ > 0) ? 0 : 31)) / 32;
if ((AnvilX != mCurAnvilX) || (AnvilZ != mCurAnvilZ))
{
OpenAnvilFile(AnvilX, AnvilZ);
}
mCurChunkX = a_ChunkX;
mCurChunkZ = a_ChunkZ;
return false;
}
bool cChunkExtract::OnCompressedDataSizePos(int a_CompressedDataSize, int a_DataOffset, char a_CompressionMethod)
{
if (!mAnvilFile.IsOpen())
{
return true;
}
cFile ChunkFile;
AString ChunkPath = Printf("%d.%d.zchunk", mCurChunkX, mCurChunkZ);
if (!ChunkFile.Open(ChunkPath, cFile::fmWrite))
{
LOG("Cannot open zchunk file \"%s\" for writing. Chunk [%d, %d] skipped.", ChunkPath.c_str(), mCurChunkX, mCurChunkZ);
return false;
}
// Copy data from mAnvilFile to ChunkFile:
mAnvilFile.Seek(a_DataOffset);
for (int BytesToCopy = a_CompressedDataSize; BytesToCopy > 0;)
{
char Buffer[64000];
int NumBytes = std::min(BytesToCopy, (int)sizeof(Buffer));
int BytesRead = mAnvilFile.Read(Buffer, NumBytes);
if (BytesRead != NumBytes)
{
LOG("Cannot copy chunk data, chunk [%d, %d] is probably corrupted. Skipping chunk.", mCurChunkX, mCurChunkZ);
return false;
}
ChunkFile.Write(Buffer, BytesRead);
BytesToCopy -= BytesRead;
} // for BytesToCopy
return false;
}
bool cChunkExtract::OnDecompressedData(const char * a_DecompressedNBT, int a_DataSize)
{
ASSERT(mAnvilFile.IsOpen()); // If it weren't, the OnCompressedDataSizePos would've prevented this from running
AString FileName = Printf("%d.%d.gzchunk", mCurChunkX, mCurChunkZ);
cGZipFile GZipChunk;
if (!GZipChunk.Open(FileName, cGZipFile::fmWrite))
{
LOG("Cannot open gzchunk file \"%s\" for writing. Chunk [%d, %d] skipped.", FileName.c_str(), mCurChunkX, mCurChunkZ);
return true;
}
GZipChunk.Write(a_DecompressedNBT, a_DataSize);
return true;
}
void cChunkExtract::OpenAnvilFile(int a_AnvilX, int a_AnvilZ)
{
mAnvilFile.Close();
AString FileName = Printf("%s/r.%d.%d.mca", mWorldFolder.c_str(), a_AnvilX, a_AnvilZ);
if (!mAnvilFile.Open(FileName, cFile::fmRead))
{
LOG("Cannot open Anvil file \"%s\" for reading", FileName.c_str());
}
mCurAnvilX = a_AnvilX;
mCurAnvilZ = a_AnvilZ;
}