From bae085dd8713f1423257c936dd9f6c231853e84f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Bergstr=C3=B6m?= <davbe125@student.liu.se>
Date: Thu, 19 Jul 2018 10:54:48 +0200
Subject: [PATCH] Add TechTreeImproved to Python library
---
python-api-src/library.cpp | 14 ++++++++++++++
1 file changed, 14 insertions(+)
diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 2f3a8e3..7f54cb2 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -101,4 +101,18 @@ PYBIND11_MODULE(library, m)
m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a);
m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a);
+
+ py::class_<BuildDescription>(m, "BuildDescription")
+ .def(py::init())
+ .def_readwrite("producer_type", &BuildDescription::producer_type)
+ .def_readwrite("result_type", &BuildDescription::result_type)
+ .def_readwrite("ability_used", &BuildDescription::ability_used)
+ .def_readwrite("time", &BuildDescription::time)
+ .def_readwrite("buildings_needed", &BuildDescription::buildings_needed)
+ .def_readwrite("addons_needed", &BuildDescription::addons_needed);
+
+ py::class_<TechTreeImproved>(m, "TechTreeImproved")
+ .def(py::init())
+ .def("load_data", &TechTreeImproved::LoadData)
+ .def("how_to_build", &TechTreeImproved::HowToBuild);
}
--
GitLab