diff --git a/python-api-src/lib_replay_unit.cpp b/python-api-src/lib_replay_unit.cpp index 63197f20a344ae22df680e8ffee70d7396aeba29..a7e1f3a63bf4afc75b08d3351a5ad673062711d0 100644 --- a/python-api-src/lib_replay_unit.cpp +++ b/python-api-src/lib_replay_unit.cpp @@ -38,6 +38,6 @@ void define_replay_unit(py::module & m) .def_property_readonly("is_carrying_minerals", &ReplayUnit::isCarryingMinerals) .def("__hash__", [](const ReplayUnit & unit) { return std::hash<const sc2::Unit *>{}(unit.getUnitPtr()); }) .def(py::self == py::self) - .def("__repr__", [](const ReplayUnit & unit) { return "<Unit of type: '" + unit.getTypeName() + "'>"; }) + .def("__repr__", [](const ReplayUnit & unit) { return "<Unit of type: '" + unit.getTypeName() +" player: " + std::to_string(unit.getPlayer()) +">"; }) ; } diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp index 023e554c8cb770a4629c15787693b08d5a6f40a9..874e065dccec81649ead2b80d25f18fa7aa746d7 100644 --- a/python-api-src/library.cpp +++ b/python-api-src/library.cpp @@ -29,7 +29,9 @@ PYBIND11_MODULE(library, m) .def("update", &sc2::Coordinator::Update) .def("set_real_time", &sc2::Coordinator::SetRealtime) .def("load_replay_list",&sc2::Coordinator::SetReplayPath, "replay_path"_a) - .def("add_replay_observer",&sc2::Coordinator::AddReplayObserver, "replay_observer"_a); + .def("add_replay_observer",&sc2::Coordinator::AddReplayObserver, "replay_observer"_a) + .def("set_replay_perspective",&sc2::Coordinator::SetReplayPerspective, "perspective"_a) + ; py::enum_<sc2::Race>(m, "Race") .value("Terran", sc2::Race::Terran) diff --git a/src/ReplayUnit.cpp b/src/ReplayUnit.cpp index 5b6298054ac1b692f2a505336980b4f0d00146c1..91c102d561773e4134fa3a06d40bdaf6b4c54777 100644 --- a/src/ReplayUnit.cpp +++ b/src/ReplayUnit.cpp @@ -55,4 +55,9 @@ ReplayUnit ReplayUnit::getTarget() const return this_unit; } +int ReplayUnit::getPlayer() const +{ + return m_unit->owner; +} + diff --git a/src/ReplayUnit.h b/src/ReplayUnit.h index 171dfc6823739b7180ee2a07d119f9a2ed5788f2..542c7870fedbb93d3ca0ad7c442941f7999b80a3 100644 --- a/src/ReplayUnit.h +++ b/src/ReplayUnit.h @@ -16,5 +16,7 @@ public: std::string getTypeName() const; bool hasTarget() const; ReplayUnit getTarget() const; + int getPlayer() const; + }; \ No newline at end of file