update BMap design, use more safe design

This commit is contained in:
2023-10-03 21:33:00 +08:00
parent 4c8500e444
commit 9cb57adf6e
4 changed files with 293 additions and 168 deletions

View File

@ -1,16 +1,47 @@
#include "BMExports.hpp"
#include <IronPad.hpp>
#include <set>
#include <type_traits>
#include <memory>
#pragma region Help & Save Functions
static bool g_IsInited = false;
static std::set<BMap::BMFile*> g_AllBMFiles = std::set<BMap::BMFile*>();
static std::set<BMap::BMMeshTransition*> g_AllBMMeshTrans = std::set<BMap::BMMeshTransition*>();
bool CheckInited() {
return g_IsInited;
}
bool CheckBMFile(BMap::BMFile* possible_bmfile) {
return (g_IsInited && possible_bmfile != nullptr && g_AllBMFiles.contains(possible_bmfile));
}
bool CheckBMMeshTrans(BMap::BMMeshTransition* possible_trans) {
return (g_IsInited && possible_trans != nullptr && g_AllBMMeshTrans.contains(possible_trans));
}
template<class _Ty, std::enable_if_t<std::is_pointer_v<_Ty>, int> = 0>
_Ty CheckCKObject(BMap::BMFile* possible_bmfile, LibCmo::CK2::CK_ID possible_id, LibCmo::CK2::CK_CLASSID expected_cid) {
// check bm file self.
if (!CheckBMFile(possible_bmfile)) return nullptr;
// check id
LibCmo::CK2::ObjImpls::CKObject* obj = possible_bmfile->GetObjectPtr(possible_id);
// check id validation and class id
if (obj == nullptr || LibCmo::CK2::CKIsChildClassOf(obj->GetClassID(), expected_cid)) return nullptr;
return static_cast<_Ty>(obj);
}
#pragma endregion
void BMInit() {
// register IronPad
IronPad::IronPadRegister();
// and startup CK environment
LibCmo::CK2::CKStartUp();
// set init
g_IsInited = true;
}
@ -29,6 +60,7 @@ void BMDispose() {
// disable init
g_IsInited = false;
// shutdown CK environment
LibCmo::CK2::CKShutdown();
// unregister iron pad
@ -38,141 +70,180 @@ void BMDispose() {
#pragma region BMFile
BMap::BMFile* BMFile_Load(LibCmo::CKSTRING file_name, LibCmo::CKSTRING temp_folder, LibCmo::CKSTRING texture_folder, LibCmo::CKDWORD encoding_count, LibCmo::CKSTRING encodings[]) {
if (!g_IsInited) return nullptr;
auto file = new BMap::BMFile(temp_folder, texture_folder, encoding_count, encodings, false);
if (file->IsFailed()) {
delete file;
file = nullptr;
}
if (!file->Load(file_name)) {
delete file;
file = nullptr;
}
if (!CheckInited()) return nullptr;
g_AllBMFiles.emplace(file);
return file;
// create a now one and try to load data.
std::unique_ptr<BMap::BMFile> file(new BMap::BMFile(temp_folder, texture_folder, encoding_count, encodings, false));
if (file->IsFreezed()) return nullptr;
if (!file->Load(file_name)) return nullptr;
// add into list and return
g_AllBMFiles.emplace(file.get());
return file.release();
}
BMap::BMFile* BMFile_Create(LibCmo::CKSTRING temp_folder, LibCmo::CKSTRING texture_folder, LibCmo::CKDWORD encoding_count, LibCmo::CKSTRING encodings[]) {
if (!g_IsInited) return nullptr;
auto file = new BMap::BMFile(temp_folder, texture_folder, encoding_count, encodings, false);
if (file->IsFailed()) {
delete file;
file = nullptr;
}
if (!CheckInited()) return nullptr;
g_AllBMFiles.emplace(file);
return file;
// create a now one
std::unique_ptr<BMap::BMFile> file(new BMap::BMFile(temp_folder, texture_folder, encoding_count, encodings, false));
if (file->IsFreezed()) return nullptr;
// add into list and return if success
g_AllBMFiles.emplace(file.get());
return file.release();
}
bool BMFile_Save(BMap::BMFile* map_file, LibCmo::CKSTRING file_name, LibCmo::CKINT compreess_level) {
if (!g_AllBMFiles.contains(map_file)) return false;
if (!CheckBMFile(map_file)) return false;
return map_file->Save(file_name, compreess_level);
}
void BMFile_Free(BMap::BMFile* map_file) {
// only free correct pointer
if (map_file == nullptr) return;
if (g_AllBMFiles.erase(map_file) != 0) {
delete map_file;
}
bool BMFile_Free(BMap::BMFile* map_file) {
if (!CheckBMFile(map_file)) return false;
g_AllBMFiles.erase(map_file);
delete map_file;
return true;
}
#define VISITOR_IMPL(namepart, cidpart) \
LibCmo::CKDWORD BMFile_Get ## namepart ## Count(BMap::BMFile* map_file) { \
if (!g_AllBMFiles.contains(map_file)) return 0; \
return map_file->Get ## namepart ## Count(); \
} \
LibCmo::CK2::ObjImpls::CK ## namepart * BMFile_Get ## namepart (BMap::BMFile* map_file, LibCmo::CKDWORD idx) { \
if (!g_AllBMFiles.contains(map_file)) return nullptr; \
return map_file->Get ## namepart (idx); \
} \
LibCmo::CK2::ObjImpls::CK ## namepart * BMFile_Create ## namepart (BMap::BMFile* map_file, LibCmo::CKSTRING name) { \
if (!g_AllBMFiles.contains(map_file)) return nullptr; \
return map_file->Create ## namepart (name); \
LibCmo::CKDWORD BMFile_GetGroupCount(BMap::BMFile* bmfile) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetGroupCount();
}
LibCmo::CK2::CK_ID BMFile_GetGroup(BMap::BMFile* bmfile, LibCmo::CKDWORD idx) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetGroup(idx);
}
LibCmo::CK2::CK_ID BMFile_CreateGroup(BMap::BMFile* bmfile, LibCmo::CKSTRING name) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->CreateGroup(name);
}
LibCmo::CKDWORD BMFile_Get3dObjectCount(BMap::BMFile* bmfile) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->Get3dObjectCount();
}
LibCmo::CK2::CK_ID BMFile_Get3dObject(BMap::BMFile* bmfile, LibCmo::CKDWORD idx) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->Get3dObject(idx);
}
LibCmo::CK2::CK_ID BMFile_Create3dObject(BMap::BMFile* bmfile, LibCmo::CKSTRING name) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->Create3dObject(name);
}
LibCmo::CKDWORD BMFile_GetMeshCount(BMap::BMFile* bmfile) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetMeshCount();
}
LibCmo::CK2::CK_ID BMFile_GetMesh(BMap::BMFile* bmfile, LibCmo::CKDWORD idx) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetMesh(idx);
}
LibCmo::CK2::CK_ID BMFile_CreateMesh(BMap::BMFile* bmfile, LibCmo::CKSTRING name) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->CreateMesh(name);
}
LibCmo::CKDWORD BMFile_GetMaterialCount(BMap::BMFile* bmfile) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetMaterialCount();
}
LibCmo::CK2::CK_ID BMFile_GetMaterial(BMap::BMFile* bmfile, LibCmo::CKDWORD idx) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetMaterial(idx);
}
LibCmo::CK2::CK_ID BMFile_CreateMaterial(BMap::BMFile* bmfile, LibCmo::CKSTRING name) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->CreateMaterial(name);
}
LibCmo::CKDWORD BMFile_GetTextureCount(BMap::BMFile* bmfile) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetTextureCount();
}
LibCmo::CK2::CK_ID BMFile_GetTexture(BMap::BMFile* bmfile, LibCmo::CKDWORD idx) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->GetTexture(idx);
}
LibCmo::CK2::CK_ID BMFile_CreateTexture(BMap::BMFile* bmfile, LibCmo::CKSTRING name) {
if (!CheckBMFile(bmfile)) return 0;
return bmfile->CreateTexture(name);
}
VISITOR_IMPL(Group, GROUP)
VISITOR_IMPL(3dObject, 3DOBJECT)
VISITOR_IMPL(Mesh, MESH)
VISITOR_IMPL(Material, MATERIAL)
VISITOR_IMPL(Texture, TEXTURE)
#undef VISITOR_IMPL
#pragma endregion
#pragma region BMMeshTransition
BMap::BMMeshTransition* BMMeshTrans_New() {
if (!g_IsInited) return nullptr;
if (!CheckInited()) return nullptr;
// create new one, insert and return.
auto meshtrans = new BMap::BMMeshTransition();
g_AllBMMeshTrans.emplace(meshtrans);
return meshtrans;
}
void BMMeshTrans_Delete(BMap::BMMeshTransition* trans) {
// only free correct pointer
if (trans == nullptr) return;
if (g_AllBMMeshTrans.erase(trans) != 0) {
delete trans;
}
bool BMMeshTrans_Delete(BMap::BMMeshTransition* trans) {
if (!CheckBMMeshTrans(trans)) return false;
g_AllBMMeshTrans.erase(trans);
delete trans;
return true;
}
void BMMeshTrans_PrepareVertexCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!g_AllBMMeshTrans.contains(trans)) return;
trans->PrepareVertexCount(count);
bool BMMeshTrans_PrepareVertexCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!CheckBMMeshTrans(trans)) return false;
return trans->PrepareVertexCount(count);
}
LibCmo::VxMath::VxVector3* BMMeshTrans_PrepareVertex(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareVertex();
}
void BMMeshTrans_PrepareNormalCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!g_AllBMMeshTrans.contains(trans)) return;
trans->PrepareNormalCount(count);
bool BMMeshTrans_PrepareNormalCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!CheckBMMeshTrans(trans)) return false;
return trans->PrepareNormalCount(count);
}
LibCmo::VxMath::VxVector3* BMMeshTrans_PrepareNormal(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareNormal();
}
void BMMeshTrans_PrepareUVCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!g_AllBMMeshTrans.contains(trans)) return;
trans->PrepareUVCount(count);
bool BMMeshTrans_PrepareUVCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!CheckBMMeshTrans(trans)) return false;
return trans->PrepareUVCount(count);
}
LibCmo::VxMath::VxVector2* BMMeshTrans_PrepareUV(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareUV();
}
void BMMeshTrans_PrepareMtlSlotCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!g_AllBMMeshTrans.contains(trans)) return;
trans->PrepareMtlSlotCount(count);
bool BMMeshTrans_PrepareMtlSlotCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!CheckBMMeshTrans(trans)) return false;
return trans->PrepareMtlSlotCount(count);
}
LibCmo::CK2::ObjImpls::CKMaterial** BMMeshTrans_PrepareMtlSlot(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareMtlSlot();
}
void BMMeshTrans_PrepareFaceCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!g_AllBMMeshTrans.contains(trans)) return;
trans->PrepareFaceCount(count);
bool BMMeshTrans_PrepareFaceCount(BMap::BMMeshTransition* trans, LibCmo::CKDWORD count) {
if (!CheckBMMeshTrans(trans)) return false;
return trans->PrepareFaceCount(count);
}
LibCmo::CKDWORD* BMMeshTrans_PrepareFaceVertexIndices(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareFaceVertexIndices();
}
LibCmo::CKDWORD* BMMeshTrans_PrepareFaceNormalIndices(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareFaceNormalIndices();
}
LibCmo::CKDWORD* BMMeshTrans_PrepareFaceUVIndices(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareFaceUVIndices();
}
LibCmo::CKDWORD* BMMeshTrans_PrepareFaceMtlSlot(BMap::BMMeshTransition* trans) {
if (!g_AllBMMeshTrans.contains(trans)) return nullptr;
if (!CheckBMMeshTrans(trans)) return nullptr;
return trans->PrepareFaceMtlSlot();
}
bool BMMeshTrans_Parse(BMap::BMMeshTransition* trans, LibCmo::CK2::ObjImpls::CKMesh* write_into_mesh) {
if (!g_AllBMMeshTrans.contains(trans)) return false;
if (!CheckBMMeshTrans(trans)) return false;
return trans->Parse(write_into_mesh);
}