From 0bc644d781426189203d75d686ab963c8fb92f05 Mon Sep 17 00:00:00 2001
From: SteBry <stefan.brynielsson@gmail.com>
Date: Thu, 21 Nov 2019 12:37:49 +0100
Subject: [PATCH] Problem might be fixed, TODO: test

---
 src/TechTree.cpp         | 1 +
 src/TechTreeImproved.cpp | 1 +
 src/TechTreeImproved.h   | 1 +
 3 files changed, 3 insertions(+)

diff --git a/src/TechTree.cpp b/src/TechTree.cpp
index fdd3f3f5c..f9b5aba5d 100644
--- a/src/TechTree.cpp
+++ b/src/TechTree.cpp
@@ -48,6 +48,7 @@ void TechTree::onStart()
             data.requiredUnits.clear();
             data.requiredUpgrades.clear();
             updated.insert(description.result_type);
+			data.buildTime = description.buildTime;
 
             data.buildAbility = sc2::ABILITY_ID::INVALID;
         }
diff --git a/src/TechTreeImproved.cpp b/src/TechTreeImproved.cpp
index 8b07889ae..a6594be4f 100644
--- a/src/TechTreeImproved.cpp
+++ b/src/TechTreeImproved.cpp
@@ -116,6 +116,7 @@ void parse_build_description(BuildDescription & description, json & build_item)
 {
     description.result_type = static_cast<sc2::UNIT_TYPEID>(build_item["unit"]);
     description.build_ability = static_cast<sc2::ABILITY_ID>(build_item["ability"]);
+	description.buildTime = static_cast<int>(build_item["time"]);
 
     if (build_item.find("requires") != build_item.end())
     {
diff --git a/src/TechTreeImproved.h b/src/TechTreeImproved.h
index 0948a328c..4dd5a253b 100644
--- a/src/TechTreeImproved.h
+++ b/src/TechTreeImproved.h
@@ -18,6 +18,7 @@ struct BuildDescription
     std::vector<sc2::UNIT_TYPEID> buildings_needed;
     std::vector<sc2::UNIT_TYPEID> addons_needed;
     std::vector<sc2::UPGRADE_ID> upgrades_needed;
+	int buildTime;
 };
 
 struct ResearchDescription
-- 
GitLab