libkms++/connector.cpp

   1 #include <stdio.h>
   2 #include <iostream>
   3 #include <unistd.h>
   4 #include <fcntl.h>
   5 #include <cassert>
   6
   7 #include "kms++.h"
   8 #include "helpers.h"
   9
  10 using namespace std;
  11
  12 namespace kms
  13 {
  14
  15
  16 static const map<int, string> connector_names = {
  17 #define DEF_CONN(c) { DRM_MODE_CONNECTOR_##c, #c }
  18         DEF_CONN(Unknown),
  19         DEF_CONN(VGA),
  20         DEF_CONN(DVII),
  21         DEF_CONN(DVID),
  22         DEF_CONN(DVIA),
  23         DEF_CONN(Composite),
  24         DEF_CONN(SVIDEO),
  25         DEF_CONN(LVDS),
  26         DEF_CONN(Component),
  27         DEF_CONN(9PinDIN),
  28         DEF_CONN(DisplayPort),
  29         DEF_CONN(HDMIA),
  30         DEF_CONN(HDMIB),
  31         DEF_CONN(TV),
  32         DEF_CONN(eDP),
  33         DEF_CONN(VIRTUAL),
  34         DEF_CONN(DSI),
  35 #undef DEF_CONN
  36 };
  37
  38 static const map<int, string> connection_str = {
  39         { 0, "<unknown>" },
  40         { DRM_MODE_CONNECTED, "Connected" },
  41         { DRM_MODE_DISCONNECTED, "Disconnected" },
  42         { DRM_MODE_UNKNOWNCONNECTION, "Unknown" },
  43 };
  44
  45 static const map<int, string> subpix_str = {
  46 #define DEF_SUBPIX(c) { DRM_MODE_SUBPIXEL_##c, #c }
  47         DEF_SUBPIX(UNKNOWN),
  48         DEF_SUBPIX(HORIZONTAL_RGB),
  49         DEF_SUBPIX(HORIZONTAL_BGR),
  50         DEF_SUBPIX(VERTICAL_RGB),
  51         DEF_SUBPIX(VERTICAL_BGR),
  52         DEF_SUBPIX(NONE),
  53 #undef DEF_SUBPIX
  54 };
  55
  56 struct ConnectorPriv
  57 {
  58         drmModeConnectorPtr drm_connector;
  59 };
  60
  61 Connector::Connector(Card &card, uint32_t id, uint32_t idx)
  62         :DrmObject(card, id, DRM_MODE_OBJECT_CONNECTOR, idx)
  63 {
  64         m_priv = new ConnectorPriv();
  65
  66         m_priv->drm_connector = drmModeGetConnector(this->card().fd(), this->id());
  67         assert(m_priv->drm_connector);
  68
  69         const auto& name = connector_names.at(m_priv->drm_connector->connector_type);
  70         m_fullname = name + to_string(m_priv->drm_connector->connector_type_id);
  71 }
  72
  73
  74 Connector::~Connector()
  75 {
  76         drmModeFreeConnector(m_priv->drm_connector);
  77         delete m_priv;
  78 }
  79
  80 void Connector::setup()
  81 {
  82         if (m_priv->drm_connector->encoder_id != 0)
  83                 m_current_encoder = card().get_encoder(m_priv->drm_connector->encoder_id);
  84         else
  85                 m_current_encoder = 0;
  86
  87         if (m_current_encoder)
  88                 m_saved_crtc = m_current_encoder->get_crtc();
  89         else
  90                 m_saved_crtc = 0;
  91 }
  92
  93 void Connector::restore_mode()
  94 {
  95         if (m_saved_crtc)
  96                 m_saved_crtc->restore_mode(this);
  97 }
  98
  99 void Connector::print_short() const
 100 {
 101         auto c = m_priv->drm_connector;
 102
 103         printf("Connector %d, %s, %dx%dmm, %s\n", id(), m_fullname.c_str(),
 104                c->mmWidth, c->mmHeight, connection_str.at(c->connection).c_str());
 105 }
 106
 107 Videomode Connector::get_default_mode() const
 108 {
 109         drmModeModeInfo drmmode = m_priv->drm_connector->modes[0];
 110
 111         return drm_mode_to_video_mode(drmmode);
 112 }
 113
 114 Videomode Connector::get_mode(const string& mode) const
 115 {
 116         auto c = m_priv->drm_connector;
 117
 118         for (int i = 0; i < c->count_modes; i++)
 119                 if (mode == c->modes[i].name)
 120                         return drm_mode_to_video_mode(c->modes[i]);
 121
 122         throw invalid_argument(mode + ": mode not found");
 123 }
 124
 125 bool Connector::connected() const
 126 {
 127         return m_priv->drm_connector->connection == DRM_MODE_CONNECTED;
 128 }
 129
 130 vector<Crtc*> Connector::get_possible_crtcs() const
 131 {
 132         vector<Crtc*> crtcs;
 133
 134         for (int i = 0; i < m_priv->drm_connector->count_encoders; ++i) {
 135                 auto enc = card().get_encoder(m_priv->drm_connector->encoders[i]);
 136
 137                 auto l = enc->get_possible_crtcs();
 138
 139                 crtcs.insert(crtcs.end(), l.begin(), l.end());
 140         }
 141
 142         return crtcs;
 143 }
 144
 145 Crtc* Connector::get_current_crtc() const
 146 {
 147         if (m_current_encoder)
 148                 return m_current_encoder->get_crtc();
 149         else
 150                 return 0;
 151 }
 152
 153 uint32_t Connector::connector_type() const
 154 {
 155         return m_priv->drm_connector->connector_type;
 156 }
 157
 158 uint32_t Connector::connector_type_id() const
 159 {
 160         return m_priv->drm_connector->connector_type_id;
 161 }
 162
 163 uint32_t Connector::mmWidth() const
 164 {
 165         return m_priv->drm_connector->mmWidth;
 166 }
 167
 168 uint32_t Connector::mmHeight() const
 169 {
 170         return m_priv->drm_connector->mmHeight;
 171 }
 172
 173 uint32_t Connector::subpixel() const
 174 {
 175         return m_priv->drm_connector->subpixel;
 176 }
 177
 178 const string& Connector::subpixel_str() const
 179 {
 180         return subpix_str.at(subpixel());
 181 }
 182
 183 std::vector<Videomode> Connector::get_modes() const
 184 {
 185         vector<Videomode> modes;
 186
 187         for (int i = 0; i < m_priv->drm_connector->count_modes; i++)
 188                 modes.push_back(drm_mode_to_video_mode(
 189                                         m_priv->drm_connector->modes[i]));
 190
 191         return modes;
 192 }
 193
 194 std::vector<Encoder*> Connector::get_encoders() const
 195 {
 196         vector<Encoder*> encoders;
 197
 198         for (int i = 0; i < m_priv->drm_connector->count_encoders; i++) {
 199                 auto enc = card().get_encoder(m_priv->drm_connector->encoders[i]);
 200                 encoders.push_back(enc);
 201         }
 202         return encoders;
 203 }
 204
 205 }