Added perspectives and fixed get_player

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()) +">"; })
 		;
 }

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)

src/ReplayUnit.cpp

@@ -55,4 +55,9 @@ ReplayUnit ReplayUnit::getTarget() const
 	return this_unit;
 }
 
+int ReplayUnit::getPlayer() const
+{
+	return m_unit->owner;
+}
+
 

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