feat(bsppp): add getter/setter for lump versions

craftablescience · Jul 18, 2024 · ba3b4c3 · ba3b4c3
1 parent 6575c73
commit ba3b4c3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/include/bsppp/bsppp.h b/include/bsppp/bsppp.h
@@ -139,6 +139,10 @@ class BSP {
 
 	[[nodiscard]] bool hasLump(BSPLump lumpIndex) const;
 
+	[[nodiscard]] int32_t getLumpVersion(BSPLump lumpIndex) const;
+
+	void setLumpVersion(BSPLump lumpIndex, int32_t version);
+
 	[[nodiscard]] std::optional<std::vector<std::byte>> readLump(BSPLump lumpIndex) const;
 
 	void writeLump(BSPLump lumpIndex, const std::vector<std::byte>& data);

diff --git a/src/bsppp/bsppp.cpp b/src/bsppp/bsppp.cpp
@@ -94,6 +94,18 @@ bool BSP::hasLump(BSPLump lumpIndex) const {
 	return this->header.lumps[lump].length != 0 && this->header.lumps[lump].offset != 0;
 }
 
+int32_t BSP::getLumpVersion(BSPLump lumpIndex) const {
+	if (this->path.empty()) {
+		return 0;
+	}
+	return this->header.lumps[static_cast<int32_t>(lumpIndex)].version;
+}
+
+void BSP::setLumpVersion(BSPLump lumpIndex, int32_t version) {
+	this->header.lumps[static_cast<int32_t>(lumpIndex)].version = version;
+	this->writeHeader();
+}
+
 std::optional<std::vector<std::byte>> BSP::readLump(BSPLump lump) const {
 	if (this->path.empty() || !this->hasLump(lump)) {
 		return std::nullopt;