3aaa8a6ee0ac691c884b9e3c11d842251fc9bcd3
[android/external-libkmsxx.git] / libkms++ / card.cpp
1 #include <stdio.h>
2 #include <unistd.h>
3 #include <fcntl.h>
4 #include <utility>
5 #include <stdexcept>
6 #include <string.h>
7 #include <algorithm>
9 #include <xf86drm.h>
10 #include <xf86drmMode.h>
12 #include "kms++.h"
14 #ifndef DRM_CLIENT_CAP_ATOMIC
15 #define DRM_CLIENT_CAP_ATOMIC 3
16 #endif
18 using namespace std;
20 namespace kms
21 {
23 Card::Card()
24 {
25         const char *card = "/dev/dri/card0";
27         int fd = open(card, O_RDWR | O_CLOEXEC);
28         if (fd < 0)
29                 throw invalid_argument(string(strerror(errno)) + " opening " +
30                                        card);
31         m_fd = fd;
33         int r;
35         r = drmSetMaster(fd);
36         m_master = r == 0;
38         if (getenv("LIBKMSXX_DISABLE_UNIVERSAL_PLANES") == 0) {
39                 r = drmSetClientCap(m_fd, DRM_CLIENT_CAP_UNIVERSAL_PLANES, 1);
40                 m_has_universal_planes = r == 0;
41         } else {
42                 m_has_universal_planes = false;
43         }
45         if (getenv("LIBKMSXX_DISABLE_ATOMIC") == 0) {
46                 r = drmSetClientCap(m_fd, DRM_CLIENT_CAP_ATOMIC, 1);
47                 m_has_atomic = r == 0;
48         } else {
49                 m_has_atomic = false;
50         }
52         uint64_t has_dumb;
53         r = drmGetCap(fd, DRM_CAP_DUMB_BUFFER, &has_dumb);
54         if (r || !has_dumb)
55                 throw invalid_argument("Dumb buffers not available");
57         auto res = drmModeGetResources(m_fd);
58         if (!res)
59                 throw invalid_argument("Can't get card resources");
61         for (int i = 0; i < res->count_connectors; ++i) {
62                 uint32_t id = res->connectors[i];
63                 auto ob = new Connector(*this, id, i);
64                 m_obmap[id] = ob;
65                 m_connectors.push_back(ob);
66         }
68         for (int i = 0; i < res->count_crtcs; ++i) {
69                 uint32_t id = res->crtcs[i];
70                 auto ob = new Crtc(*this, id, i);
71                 m_obmap[id] = ob;
72                 m_crtcs.push_back(ob);
73         }
75         for (int i = 0; i < res->count_encoders; ++i) {
76                 uint32_t id = res->encoders[i];
77                 auto ob = new Encoder(*this, id);
78                 m_obmap[id] = ob;
79                 m_encoders.push_back(ob);
80         }
82         drmModeFreeResources(res);
84         auto planeRes = drmModeGetPlaneResources(m_fd);
86         for (uint i = 0; i < planeRes->count_planes; ++i) {
87                 uint32_t id = planeRes->planes[i];
88                 auto ob = new Plane(*this, id);
89                 m_obmap[id] = ob;
90                 m_planes.push_back(ob);
91         }
93         drmModeFreePlaneResources(planeRes);
95         // collect all possible props
96         for (auto ob : get_objects()) {
97                 auto props = drmModeObjectGetProperties(m_fd, ob->id(), ob->object_type());
99                 if (props == nullptr)
100                         continue;
102                 for (unsigned i = 0; i < props->count_props; ++i) {
103                         uint32_t prop_id = props->props[i];
105                         if (m_obmap.find(prop_id) == m_obmap.end()) {
106                                 auto ob = new Property(*this, prop_id);
107                                 m_obmap[prop_id] = ob;
108                                 m_properties.push_back(ob);
109                         }
110                 }
112                 drmModeFreeObjectProperties(props);
113         }
115         for (auto pair : m_obmap)
116                 pair.second->setup();
119 Card::~Card()
121         restore_modes();
123         for (auto pair : m_obmap)
124                 delete pair.second;
126         close(m_fd);
129 void Card::restore_modes()
131         for (auto conn : get_connectors())
132                 conn->restore_mode();
135 template <class T> static void print_obs(const map<uint32_t, DrmObject*>& obmap)
137         for (auto pair : obmap) {
138                 auto ob = pair.second;
139                 if (dynamic_cast<T*>(ob)) {
140                         ob->print_short();
141                         //ob->print_props();
142                 }
143         }
146 void Card::print_short() const
148         print_obs<Connector>(m_obmap);
149         print_obs<Encoder>(m_obmap);
150         print_obs<Crtc>(m_obmap);
151         print_obs<Plane>(m_obmap);
154 Property* Card::get_prop(const string& name) const
156         for (auto pair : m_obmap) {
157                 auto prop = dynamic_cast<Property*>(pair.second);
158                 if (!prop)
159                         continue;
161                 if (name == prop->name())
162                         return prop;
163         }
165         throw invalid_argument(string("Card property ") + name + " not found");
168 Connector* Card::get_first_connected_connector() const
170         for(auto pair : m_obmap) {
171                 auto c = dynamic_cast<Connector*>(pair.second);
173                 if (c && c->connected())
174                         return c;
175         }
177         throw invalid_argument("no connected connectors");
180 DrmObject* Card::get_object(uint32_t id) const
182         return m_obmap.at(id);
185 vector<Connector*> Card::get_connectors() const
187         vector<Connector*> v;
188         for(auto pair : m_obmap) {
189                 auto p = dynamic_cast<Connector*>(pair.second);
190                 if (p)
191                         v.push_back(p);
192         }
193         return v;
196 vector<Crtc*> Card::get_crtcs() const
198         vector<Crtc*> v;
199         for(auto pair : m_obmap) {
200                 auto p = dynamic_cast<Crtc*>(pair.second);
201                 if (p)
202                         v.push_back(p);
203         }
204         return v;
207 vector<Plane*> Card::get_planes() const
209         vector<Plane*> v;
210         for(auto pair : m_obmap) {
211                 auto p = dynamic_cast<Plane*>(pair.second);
212                 if (p)
213                         v.push_back(p);
214         }
215         return v;
218 vector<DrmObject*> Card::get_objects() const
220         vector<DrmObject*> v;
221         for(auto pair : m_obmap)
222                 v.push_back(pair.second);
223         return v;
226 Crtc* Card::get_crtc_by_index(uint32_t idx) const
228         for(auto pair : m_obmap) {
229                 auto crtc = dynamic_cast<Crtc*>(pair.second);
230                 if (crtc && crtc->idx() == idx)
231                         return crtc;
232         }
233         throw invalid_argument(string("Crtc #") + to_string(idx) + " not found");
236 Connector* Card::get_connector(uint32_t id) const { return dynamic_cast<Connector*>(get_object(id)); }
237 Crtc* Card::get_crtc(uint32_t id) const { return dynamic_cast<Crtc*>(get_object(id)); }
238 Encoder* Card::get_encoder(uint32_t id) const { return dynamic_cast<Encoder*>(get_object(id)); }
239 Property* Card::get_prop(uint32_t id) const { return dynamic_cast<Property*>(get_object(id)); }
240 Plane* Card::get_plane(uint32_t id) const { return dynamic_cast<Plane*>(get_object(id)); }
242 std::vector<kms::Pipeline> Card::get_connected_pipelines()
244         vector<Pipeline> outputs;
246         for (auto conn : get_connectors())
247         {
248                 if (conn->connected() == false)
249                         continue;
251                 Crtc* crtc = conn->get_current_crtc();
253                 if (!crtc) {
254                         for (auto possible : conn->get_possible_crtcs()) {
255                                 if (find_if(outputs.begin(), outputs.end(), [possible](Pipeline out) { return out.crtc == possible; }) == outputs.end()) {
256                                         crtc = possible;
257                                         break;
258                                 }
259                         }
260                 }
262                 if (!crtc)
263                         throw invalid_argument(string("Connector #") +
264                                                to_string(conn->idx()) +
265                                                " has no possible crtcs");
267                 outputs.push_back(Pipeline { crtc, conn });
268         }
270         return outputs;
273 static void page_flip_handler(int fd, unsigned int frame,
274                               unsigned int sec, unsigned int usec,
275                               void *data)
277         auto handler = (PageFlipHandlerBase*)data;
278         double time = sec + usec / 1000000.0;
279         handler->handle_page_flip(frame, time);
282 void Card::call_page_flip_handlers()
284         drmEventContext ev = {
285                 .version = DRM_EVENT_CONTEXT_VERSION,
286                 .vblank_handler = 0,
287                 .page_flip_handler = page_flip_handler,
288         };
290         drmHandleEvent(fd(), &ev);