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 4b5069c0b5556b16e4e823255263b54d5527cc94..2e2c100d73ed4efdb9f69a491794688d58f018f2 100644 --- a/src/ReplayUnit.cpp +++ b/src/ReplayUnit.cpp @@ -48,4 +48,9 @@ ReplayUnit ReplayUnit::getTarget() const return *this; } +int ReplayUnit::getPlayer() const +{ + return m_unit->owner; +} + diff --git a/src/ReplayUnit.h b/src/ReplayUnit.h index 08051ca0eab0f2e83fa70a936b39f045155c80e4..23fc50c1e7fc37de672ae04f9e39a8767f559e70 100644 --- a/src/ReplayUnit.h +++ b/src/ReplayUnit.h @@ -16,5 +16,7 @@ public: const UnitType & getType() const; bool hasTarget() const; ReplayUnit getTarget() const; + int getPlayer() const; + }; \ No newline at end of file