src/ResearchTree/ResearchTree.cpp

@@ -401,6 +401,52 @@ void ResearchTree::readEngineUnlocks(Containers::ArrayView<Int> engines, Contain
     }
 }
 
+void ResearchTree::readOSUnlocks(Containers::ArrayView<Int> os, Containers::ArrayView<Int> modules) {
+    if(os == nullptr || os.size() == 0) {
+        Utility::Error{} << "OSes can't be empty";
+        return;
+    }
+
+    for(Int os_id : os) {
+        if(_osNodes.find(os_id) != _osNodes.end()) {
+            _osNodes.at(os_id).setState(Node::State::Unlocked);
+        }
+    }
+
+    if(modules == nullptr || modules.size() == 0) {
+        return;
+    }
+
+    for(Int module : modules) {
+        if(_osNodes.find(module) != _osNodes.end()) {
+            _osNodes.at(module).setState(Node::State::Unlocked);
+        }
+    }
+}
+
+void ResearchTree::readArchUnlocks(Containers::ArrayView<Int> archs, Containers::ArrayView<Int> techs) {
+    if(archs == nullptr || archs.size() == 0) {
+        Utility::Error{} << "Archs can't be empty";
+        return;
+    }
+
+    for(Int arch : archs) {
+        if(_archNodes.find(arch) != _archNodes.end()) {
+            _archNodes.at(arch).setState(Node::State::Unlocked);
+        }
+    }
+
+    if(techs == nullptr || techs.size() == 0) {
+        return;
+    }
+
+    for(Int tech : techs) {
+        if(_archNodes.find(tech) != _archNodes.end()) {
+            _archNodes.at(tech).setState(Node::State::Unlocked);
+        }
+    }
+}
+
 auto ResearchTree::getEngineRootNode() -> Node& {
     return _engineNodes.at(VerseEngine);
 }

src/ResearchTree/ResearchTree.h

@@ -99,6 +99,8 @@ class ResearchTree {
         void generateArchTree();
 
         void readEngineUnlocks(Containers::ArrayView<Int> engines, Containers::ArrayView<Int> gears = nullptr);
+        void readOSUnlocks(Containers::ArrayView<Int> os, Containers::ArrayView<Int> modules = nullptr);
+        void readArchUnlocks(Containers::ArrayView<Int> archs, Containers::ArrayView<Int> techs = nullptr);
 
         auto getEngineRootNode() -> Node&;
         auto getOSRootNode() -> Node&;