Added sizes on more factories

1 files changed, 4 insertions, 1 deletions

diff --git a/src/factories/VtxFactory.cpp b/src/factories/VtxFactory.cpp
index c0b06ea..1aabaf0 100644
--- a/src/factories/VtxFactory.cpp
+++ b/src/factories/VtxFactory.cpp
@@ -28,6 +28,9 @@ void VtxCodeExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> r
     }
 
     if (Companion::Instance->IsDebug()) {
+        if (IS_SEGMENTED(offset)) {
+            offset = SEGMENT_OFFSET(offset);
+        }
         write << "// 0x" << std::hex << std::uppercase << offset << "\n";
     }
 
@@ -61,7 +64,7 @@ void VtxCodeExporter::Export(std::ostream &write, std::shared_ptr<IParsedData> r
     write << "};\n";
 
     if (Companion::Instance->IsDebug()) {
-        write << "// size: " << vtx.size() << "\n";
+        write << "// count: " << vtx.size() << "\n";
         write << "// 0x" << std::hex << std::uppercase << (offset + (sizeof(VtxRaw) * vtx.size())) << "\n";
     }