py: add convert_vector helper

Signed-off-by: Tomi Valkeinen <tomi.valkeinen@ti.com>

1 files changed, 15 insertions, 20 deletions

diff --git a/py/pykms/pykmsbase.cpp b/py/pykms/pykmsbase.cpp
index dd09fae..2c97bd7 100644
--- a/py/pykms/pykmsbase.cpp
+++ b/py/pykms/pykmsbase.cpp
@@ -7,6 +7,16 @@ namespace py = pybind11;
 using namespace kms;
 using namespace std;
 
+// Helper to convert vector<T*> to vector<unique_ptr<T, py::nodelete>>
+template<typename T>
+static vector<unique_ptr<T, py::nodelete>> convert_vector(const vector<T*>& source)
+{
+        vector<unique_ptr<T, py::nodelete>> v;
+        for (T* p : source)
+                v.push_back(unique_ptr<T, py::nodelete>(p));
+        return v;
+}
+
 void init_pykmsbase(py::module &m)
 {
         py::class_<Card>(m, "Card")
@@ -18,31 +28,19 @@ void init_pykmsbase(py::module &m)
                         // RuntimeError: return_value_policy = move, but the object is neither movable nor copyable!
                         // So we do this manually.
                         .def_property_readonly("connectors", [](Card* self) {
-                                vector<unique_ptr<Connector, py::nodelete>> v;
-                                for (Connector* p : self->get_connectors())
-                                        v.push_back(unique_ptr<Connector, py::nodelete>(p));
-                                return v;
+                                return convert_vector(self->get_connectors());
                         })
 
                         .def_property_readonly("crtcs", [](Card* self) {
-                                vector<unique_ptr<Crtc, py::nodelete>> v;
-                                for (Crtc* p : self->get_crtcs())
-                                        v.push_back(unique_ptr<Crtc, py::nodelete>(p));
-                                return v;
+                                return convert_vector(self->get_crtcs());
                         })
 
                         .def_property_readonly("encoders", [](Card* self) {
-                                vector<unique_ptr<Encoder, py::nodelete>> v;
-                                for (Encoder* p : self->get_encoders())
-                                        v.push_back(unique_ptr<Encoder, py::nodelete>(p));
-                                return v;
+                                return convert_vector(self->get_encoders());
                         })
 
                         .def_property_readonly("planes", [](Card* self) {
-                                vector<unique_ptr<Plane, py::nodelete>> v;
-                                for (Plane* p : self->get_planes())
-                                        v.push_back(unique_ptr<Plane, py::nodelete>(p));
-                                return v;
+                                return convert_vector(self->get_planes());
                         })
 
                         .def_property_readonly("has_atomic", &Card::has_atomic)
@@ -69,10 +67,7 @@ void init_pykmsbase(py::module &m)
                         .def("get_default_mode", &Connector::get_default_mode)
                         .def("get_current_crtc", &Connector::get_current_crtc)
                         .def("get_possible_crtcs", [](Connector* self) {
-                                vector<unique_ptr<Crtc, py::nodelete>> v;
-                                for (Crtc* p : self->get_possible_crtcs())
-                                        v.push_back(unique_ptr<Crtc, py::nodelete>(p));
-                                return v;
+                                return convert_vector(self->get_possible_crtcs());
                         })
                         .def("get_modes", &Connector::get_modes)
                         .def("get_mode", (Videomode (Connector::*)(const string& mode) const)&Connector::get_mode)