diff --git a/src/tools/pcb_reader/main.cpp b/src/tools/pcb_reader/main.cpp index 412695ce..7125efd6 100644 --- a/src/tools/pcb_reader/main.cpp +++ b/src/tools/pcb_reader/main.cpp @@ -151,7 +151,7 @@ int main( int argc, char* argv[] ) auto test_file1 = data1.get_file( "bg/ffxiv/" + zonePath + "/collision/list.pcb" ); auto section1 = test_file1->access_data_sections().at( 0 ); std::string path = "bg/ffxiv/" + zonePath + "/collision/"; - int offset1 = 0x20; + uint32_t offset1 = 0x20; for( ; ; ) { @@ -170,14 +170,7 @@ int main( int argc, char* argv[] ) } LGB_FILE bgLgb( §ion[0] ); - - int max_index = 0; - - std::vector vertices; - std::vector indices; - - char *data; - int counter = 0; + uint32_t max_index = 0; // dont bother if we cant write to a file auto fp_out = fopen( ( zoneName + ".obj" ).c_str(), "w" ); @@ -248,27 +241,30 @@ int main( int argc, char* argv[] ) auto loadSgbFile = [&]( const std::string& fileName ) -> bool { + SGB_FILE sgbFile; try { auto file = data1.get_file( fileName ); auto sections = file->get_data_sections(); auto dataSection = §ions.at( 0 )[0]; - SGB_FILE sgbFile = SGB_FILE( &dataSection[0] ); + sgbFile = SGB_FILE( &dataSection[0] ); sgbFiles.insert( std::make_pair( fileName, sgbFile ) ); return true; } catch( std::exception& e ) { std::cout << "Unable to load SGB " << fileName << "\n\tError:\n\t" << e.what() << "\n"; - return false; + sgbFiles.insert( std::make_pair( fileName, sgbFile ) ); } + return false; }; auto pushVerts = [&]( const PCB_FILE& pcb_file, const std::string& name, const vec3* scale = nullptr, const vec3* rotation = nullptr, const vec3* translation = nullptr ) { - char name2[0x7F]; - memset( name2, 0, 0x7F ); - sprintf(&name2[0], "%s_%u", &name[0], objCount[name]++ ); - fprintf( fp_out, "o %s\n", &name2[0] ); + char name2[0x100]; + memset( name2, 0, 0x100 ); + sprintf( &name2[0], "%s_%u", &name[0], objCount[name]++ ); + //fprintf( fp_out, "o %s\n", &name2[0] ); + uint32_t groupCount = 0; for( const auto &entry : pcb_file.entries ) { @@ -317,13 +313,10 @@ int main( int argc, char* argv[] ) //fprintf( fp_out, "g %s_", (name2 + "_" + std::to_string( groupCount++ )).c_str() ); for( const auto &index : entry.data.indices ) { - //if( index.index[0] != 0 || index.index[1] != 0 || index.index[2] != 0 ) - { - fprintf( fp_out, "f %i %i %i\n", - index.index[0] + max_index + 1, - index.index[1] + max_index + 1, - index.index[2] + max_index + 1 ); - } + fprintf( fp_out, "f %i %i %i\n", + index.index[0] + max_index + 1, + index.index[1] + max_index + 1, + index.index[2] + max_index + 1 ); // std::cout << std::to_string( index.unknown[0] )<< " " << std::to_string( index.unknown[1] )<< " " << std::to_string( index.unknown[2]) << std::endl; } max_index += entry.data.vertices.size() + entry.data.vertices_i16.size(); @@ -343,59 +336,63 @@ int main( int argc, char* argv[] ) { //std::cout << "\t" << group.name << " Size " << group.header.entryCount << "\n"; totalGroups++; - for( const auto pEntry : group.entries ) + for( const auto& pEntry : group.entries ) { - LGB_GIMMICK_ENTRY* pGimmick = nullptr; + LGB_GIMMICK_ENTRY* pGimmick = dynamic_cast( pEntry.get() ); auto pBgParts = dynamic_cast( pEntry.get() ); - if( pBgParts && pBgParts->header.type != LgbEntryType::BgParts ) - pBgParts = nullptr; - auto fileName = pBgParts ? pBgParts->collisionFileName : std::string(""); - - if( pBgParts && fileName.empty() ) - continue; + std::string fileName( "" ); + fileName.resize( 256 ); + totalGroupEntries++; // write files - auto writeOutput = [&]() + auto writeOutput = [&]( const std::string& fileName, const vec3* scale, const vec3* rotation, const vec3* translation ) -> bool { { const auto& it = pcbFiles.find( fileName ); if( it == pcbFiles.end() ) { - if( !fileName.empty() ) - loadPcbFile( fileName ); + if( fileName.empty() || !loadPcbFile( fileName ) ) + return false; //std::cout << "\t\tLoaded PCB File " << pBgParts->collisionFileName << "\n"; } } const auto& it = pcbFiles.find( fileName ); if( it != pcbFiles.end() ) { - totalGroupEntries++; - - const auto* scale = pBgParts ? &pBgParts->header.scale : &pGimmick->header.scale; - const auto* rotation = pBgParts ? &pBgParts->header.rotation : &pGimmick->header.rotation; - const auto* translation = pBgParts ? &pBgParts->header.translation : &pGimmick->header.translation; - const auto& pcb_file = it->second; pushVerts( pcb_file, fileName, scale, rotation, translation ); } + return true; }; - // gimmick entry - if( !pBgParts ) + if( pBgParts ) + { + fileName = pBgParts->collisionFileName; + + if( !writeOutput( fileName, &pBgParts->header.scale, &pBgParts->header.rotation, &pBgParts->header.translation ) ) + { + fileName = pBgParts->modelFileName; + boost::replace_all( fileName, "bgparts", "collision" ); + boost::replace_all( fileName, ".mdl", ".pcb" ); + writeOutput( fileName, &pBgParts->header.scale, &pBgParts->header.rotation, &pBgParts->header.translation ); + } + } + + // gimmick entry + if( pGimmick ) { - pGimmick = dynamic_cast( pEntry.get() ); { const auto& it = sgbFiles.find( pGimmick->gimmickFileName ); if( it == sgbFiles.end() ) { + //std::cout << "\tGIMMICK:\n\t\t" << pGimmick->name << " " << pGimmick->gimmickFileName << "\n"; loadSgbFile( pGimmick->gimmickFileName ); } } const auto& it = sgbFiles.find( pGimmick->gimmickFileName ); if( it != sgbFiles.end() ) { - totalGroupEntries++; const auto& sgbFile = it->second; for( const auto& group : sgbFile.entries ) @@ -403,18 +400,18 @@ int main( int argc, char* argv[] ) for( const auto& pEntry : group.entries ) { auto pModel = dynamic_cast( pEntry.get() ); - if( !pModel->collisionFileName.empty() ) + fileName = pModel->collisionFileName; + if( !writeOutput( fileName, &pGimmick->header.scale, &pGimmick->header.rotation, &pGimmick->header.translation ) ) { - fileName = pModel->collisionFileName; - writeOutput(); + fileName = pModel->modelFileName; + boost::replace_all( fileName, "bgparts", "collision" ); + boost::replace_all( fileName, ".mdl", ".pcb" ); + writeOutput( fileName, &pGimmick->header.scale, &pGimmick->header.rotation, &pGimmick->header.translation ); } } } } - continue; } - // bgparts - writeOutput(); } } std::cout << "\n\nLoaded " << pcbFiles.size() << " PCB Files \n"; diff --git a/src/tools/pcb_reader/sgb.h b/src/tools/pcb_reader/sgb.h index 5ca9e3a4..8c597d0e 100644 --- a/src/tools/pcb_reader/sgb.h +++ b/src/tools/pcb_reader/sgb.h @@ -114,17 +114,19 @@ struct SGB_GROUP SGB_FILE* parent; std::vector> entries; - SGB_GROUP( char* buf, SGB_FILE* file, uint32_t offset ) + SGB_GROUP( char* buf, SGB_FILE* file, uint32_t fileSize, uint32_t offset ) { parent = file; header = *reinterpret_cast( buf + offset ); name = std::string( buf + offset + header.nameOffset ); auto entriesOffset = offset + sizeof( header ); + for( auto i = 0; i < header.entryCount; ++i ) { - auto entryOffset = entriesOffset + *reinterpret_cast( buf + ( entriesOffset + i * 4 ) ); - + auto entryOffset = entriesOffset + *reinterpret_cast( buf + ( entriesOffset + (i * 4) ) ); + if( entryOffset > fileSize ) + throw std::runtime_error( "SGB_GROUP entry offset was larger than SGB file size!" ); auto type = *reinterpret_cast( buf + entryOffset ); if( type == SgbGroupEntryType::Model ) { @@ -170,6 +172,10 @@ struct SGB_FILE SGB_HEADER header; std::vector entries; + SGB_FILE() + { + memset( &header, 0, sizeof( header ) ); + } SGB_FILE( char* buf ) { constexpr int baseOffset = 0x14; @@ -178,10 +184,17 @@ struct SGB_FILE if( strncmp( &header.magic[0], "SGB1", 4 ) != 0 || strncmp( &header.magic2[0], "SCN1", 4 ) != 0 ) throw std::runtime_error( "Unable to load SGB File!" ); - auto group = SGB_GROUP( buf, this, baseOffset + header.sharedOffset ); - //auto group2 = SGB_GROUP( buf, this, baseOffset + header.offset1C ); - entries.push_back( group ); - //entries.push_back( group2 ); + try + { + auto group = SGB_GROUP( buf, this, header.fileSize, baseOffset + header.sharedOffset ); + entries.push_back( group ); + auto group2 = SGB_GROUP( buf, this, header.fileSize, baseOffset + header.offset1C ); + entries.push_back( group2 ); + } + catch( std::exception& e ) + { + std::cout << e.what() << "\n"; + } }; };