mirror of
https://github.com/DrPerkyLegit/LCEServerTest.git
synced 2026-05-22 11:34:33 +00:00
removed git history
This commit is contained in:
208
Minecraft.World/SavedDataStorage.cpp
Normal file
208
Minecraft.World/SavedDataStorage.cpp
Normal file
@@ -0,0 +1,208 @@
|
||||
#include "stdafx.h"
|
||||
#include "net.minecraft.world.level.saveddata.h"
|
||||
#include "net.minecraft.world.level.storage.h"
|
||||
#include "net.minecraft.world.entity.ai.village.h"
|
||||
#include "com.mojang.nbt.h"
|
||||
#include "File.h"
|
||||
#include "SavedDataStorage.h"
|
||||
|
||||
#include "ConsoleSaveFileIO.h"
|
||||
|
||||
SavedDataStorage::SavedDataStorage(LevelStorage *levelStorage)
|
||||
{
|
||||
/*
|
||||
cache = new unordered_map<wstring, shared_ptr<SavedData> >;
|
||||
savedDatas = new vector<shared_ptr<SavedData> >;
|
||||
usedAuxIds = new unordered_map<wstring, short*>;
|
||||
*/
|
||||
|
||||
this->levelStorage = levelStorage;
|
||||
loadAuxValues();
|
||||
}
|
||||
|
||||
shared_ptr<SavedData> SavedDataStorage::get(const type_info& clazz, const wstring& id)
|
||||
{
|
||||
auto it = cache.find(id);
|
||||
if (it != cache.end()) return (*it).second;
|
||||
|
||||
shared_ptr<SavedData> data = nullptr;
|
||||
if (levelStorage != nullptr)
|
||||
{
|
||||
//File file = levelStorage->getDataFile(id);
|
||||
ConsoleSavePath file = levelStorage->getDataFile(id);
|
||||
if (!file.getName().empty() && levelStorage->getSaveFile()->doesFileExist( file ) )
|
||||
{
|
||||
// mob = dynamic_pointer_cast<Mob>(Mob::_class->newInstance( level ));
|
||||
//data = clazz.getConstructor(String.class).newInstance(id);
|
||||
|
||||
if( clazz == typeid(MapItemSavedData) )
|
||||
{
|
||||
data = dynamic_pointer_cast<SavedData>(std::make_shared<MapItemSavedData>(id));
|
||||
}
|
||||
else if( clazz == typeid(Villages) )
|
||||
{
|
||||
data = dynamic_pointer_cast<SavedData>(std::make_shared<Villages>(id));
|
||||
}
|
||||
else if( clazz == typeid(StructureFeatureSavedData) )
|
||||
{
|
||||
data = dynamic_pointer_cast<SavedData>(std::make_shared<StructureFeatureSavedData>(id));
|
||||
}
|
||||
else
|
||||
{
|
||||
// Handling of new SavedData class required
|
||||
__debugbreak();
|
||||
}
|
||||
|
||||
ConsoleSaveFileInputStream fis = ConsoleSaveFileInputStream(levelStorage->getSaveFile(), file);
|
||||
CompoundTag *root = NbtIo::readCompressed(&fis);
|
||||
fis.close();
|
||||
|
||||
data->load(root->getCompound(L"data"));
|
||||
}
|
||||
}
|
||||
|
||||
if (data != nullptr)
|
||||
{
|
||||
cache.insert( unordered_map<wstring, shared_ptr<SavedData> >::value_type( id , data ) );
|
||||
savedDatas.push_back(data);
|
||||
}
|
||||
return data;
|
||||
}
|
||||
|
||||
void SavedDataStorage::set(const wstring& id, shared_ptr<SavedData> data)
|
||||
{
|
||||
if (data == nullptr)
|
||||
{
|
||||
// TODO 4J Stu - throw new RuntimeException("Can't set null data");
|
||||
assert( false );
|
||||
}
|
||||
auto it = cache.find(id);
|
||||
if ( it != cache.end() )
|
||||
{
|
||||
auto it2 = find(savedDatas.begin(), savedDatas.end(), it->second);
|
||||
if( it2 != savedDatas.end() )
|
||||
{
|
||||
savedDatas.erase( it2 );
|
||||
}
|
||||
cache.erase( it );
|
||||
}
|
||||
cache.insert( cacheMapType::value_type(id, data) );
|
||||
savedDatas.push_back(data);
|
||||
}
|
||||
|
||||
void SavedDataStorage::save()
|
||||
{
|
||||
for (auto& data : savedDatas)
|
||||
{
|
||||
if (data->isDirty())
|
||||
{
|
||||
save(data);
|
||||
data->setDirty(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void SavedDataStorage::save(shared_ptr<SavedData> data)
|
||||
{
|
||||
if (levelStorage == nullptr) return;
|
||||
//File file = levelStorage->getDataFile(data->id);
|
||||
ConsoleSavePath file = levelStorage->getDataFile(data->id);
|
||||
if (!file.getName().empty())
|
||||
{
|
||||
CompoundTag *dataTag = new CompoundTag();
|
||||
data->save(dataTag);
|
||||
|
||||
CompoundTag *tag = new CompoundTag();
|
||||
tag->putCompound(L"data", dataTag);
|
||||
|
||||
ConsoleSaveFileOutputStream fos = ConsoleSaveFileOutputStream(levelStorage->getSaveFile(), file);
|
||||
NbtIo::writeCompressed(tag, &fos);
|
||||
fos.close();
|
||||
|
||||
delete tag;
|
||||
}
|
||||
}
|
||||
|
||||
void SavedDataStorage::loadAuxValues()
|
||||
{
|
||||
usedAuxIds.clear();
|
||||
|
||||
if (levelStorage == nullptr) return;
|
||||
//File file = levelStorage->getDataFile(L"idcounts");
|
||||
ConsoleSavePath file = levelStorage->getDataFile(L"idcounts");
|
||||
if (!file.getName().empty() && levelStorage->getSaveFile()->doesFileExist( file ) )
|
||||
{
|
||||
ConsoleSaveFileInputStream fis = ConsoleSaveFileInputStream(levelStorage->getSaveFile(), file);
|
||||
DataInputStream dis = DataInputStream(&fis);
|
||||
CompoundTag *tags = NbtIo::read(&dis);
|
||||
dis.close();
|
||||
|
||||
vector<Tag *> *allTags = tags->getAllTags();
|
||||
for ( Tag *tag : *allTags )
|
||||
{
|
||||
if (dynamic_cast<ShortTag *>(tag))
|
||||
{
|
||||
ShortTag *sTag = static_cast<ShortTag *>(tag);
|
||||
wstring id = sTag->getName();
|
||||
short val = sTag->data;
|
||||
usedAuxIds.insert( uaiMapType::value_type( id, val ) );
|
||||
}
|
||||
}
|
||||
delete allTags;
|
||||
}
|
||||
}
|
||||
|
||||
int SavedDataStorage::getFreeAuxValueFor(const wstring& id)
|
||||
{
|
||||
auto it = usedAuxIds.find(id);
|
||||
short val = 0;
|
||||
if ( it != usedAuxIds.end() )
|
||||
{
|
||||
val = (*it).second;
|
||||
val++;
|
||||
}
|
||||
|
||||
usedAuxIds[id] = val;
|
||||
if (levelStorage == nullptr) return val;
|
||||
//File file = levelStorage->getDataFile(L"idcounts");
|
||||
ConsoleSavePath file = levelStorage->getDataFile(L"idcounts");
|
||||
if (!file.getName().empty())
|
||||
{
|
||||
CompoundTag *tag = new CompoundTag();
|
||||
|
||||
// TODO 4J Stu - This was iterating over the keySet in Java, so potentially we are looking at more items?
|
||||
for(auto& it : usedAuxIds)
|
||||
{
|
||||
short value = it.second;
|
||||
tag->putShort(it.first.c_str(), value);
|
||||
}
|
||||
|
||||
ConsoleSaveFileOutputStream fos = ConsoleSaveFileOutputStream(levelStorage->getSaveFile(), file);
|
||||
DataOutputStream dos = DataOutputStream(&fos);
|
||||
NbtIo::write(tag, &dos);
|
||||
dos.close();
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
// 4J Added
|
||||
int SavedDataStorage::getAuxValueForMap(PlayerUID xuid, int dimension, int centreXC, int centreZC, int scale)
|
||||
{
|
||||
if( levelStorage == nullptr )
|
||||
{
|
||||
switch(dimension)
|
||||
{
|
||||
case -1:
|
||||
return MAP_NETHER_DEFAULT_INDEX;
|
||||
case 1:
|
||||
return MAP_END_DEFAULT_INDEX;
|
||||
case 0:
|
||||
default:
|
||||
return MAP_OVERWORLD_DEFAULT_INDEX;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
return levelStorage->getAuxValueForMap(xuid, dimension, centreXC, centreZC, scale);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user