1 files changed, 10 insertions, 1 deletions
diff --git a/src/factories/sm64/GeoLayoutFactory.h b/src/factories/sm64/GeoLayoutFactory.h
index e769b64..2cc7120 100644
--- a/src/factories/sm64/GeoLayoutFactory.h
+++ b/src/factories/sm64/GeoLayoutFactory.h
@@ -5,7 +5,11 @@
 #include <vector>
 #include <variant>
 
-typedef std::variant<uint8_t, int8_t, uint16_t, int16_t, uint32_t, int32_t, uint64_t, Vec2f, Vec3f, Vec3s, Vec3i, Vec4f, Vec4s> GeoArgument;
+typedef std::variant<uint8_t, int8_t, uint16_t, int16_t, uint32_t, int32_t, uint64_t, Vec2f, Vec3f, Vec3s, Vec3i, Vec4f, Vec4s, std::string> GeoArgument;
+
+enum class GeoArgumentType {
+    U8, S8, U16, S16, U32, S32, U64, VEC2F, VEC3F, VEC3S, VEC3I, VEC4F, VEC4S, STRING
+};
 
 namespace SM64 {
 
@@ -21,6 +25,10 @@ public:
     explicit GeoLayout(std::vector<GeoCommand> commands) : commands(std::move(commands)) {}
 };
 
+class GeoCodeExporter : public BaseExporter {
+    ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
+};
+
 class GeoHeaderExporter : public BaseExporter {
     ExportResult Export(std::ostream& write, std::shared_ptr<IParsedData> data, std::string& entryName, YAML::Node& node, std::string* replacement) override;
 };
@@ -36,6 +44,7 @@ public:
         return {
             REGISTER(Header, GeoHeaderExporter)
             REGISTER(Binary, GeoBinaryExporter)
+            REGISTER(Code, GeoCodeExporter)
         };
     }
 };