kms++util/src/resourcemanager.cpp

   1 #include <kms++util/resourcemanager.h>
   2 #include <algorithm>
   3 #include <kms++util/strhelpers.h>
   4
   5 using namespace kms;
   6 using namespace std;
   7
   8 template<class C, class T>
   9 auto contains(const C& v, const T& x)
  10 -> decltype(end(v), true)
  11 {
  12         return end(v) != std::find(begin(v), end(v), x);
  13 }
  14
  15 ResourceManager::ResourceManager(Card& card)
  16         : m_card(card)
  17 {
  18 }
  19
  20 void ResourceManager::reset()
  21 {
  22         m_reserved_connectors.clear();
  23         m_reserved_crtcs.clear();
  24         m_reserved_planes.clear();
  25 }
  26
  27 static Connector* find_connector(Card& card, const vector<Connector*> reserved)
  28 {
  29         for (Connector* conn : card.get_connectors()) {
  30                 if (!conn->connected())
  31                         continue;
  32
  33                 if (contains(reserved, conn))
  34                         continue;
  35
  36                 return conn;
  37         }
  38
  39         return nullptr;
  40 }
  41
  42 static Connector* resolve_connector(Card& card, const string& name, const vector<Connector*> reserved)
  43 {
  44         auto connectors = card.get_connectors();
  45
  46         if (name[0] == '@') {
  47                 char* endptr;
  48                 unsigned id = strtoul(name.c_str() + 1, &endptr, 10);
  49                 if (*endptr == 0) {
  50                         Connector* c = card.get_connector(id);
  51
  52                         if (!c || contains(reserved, c))
  53                                 return nullptr;
  54
  55                         return c;
  56                 }
  57         } else {
  58                 char* endptr;
  59                 unsigned idx = strtoul(name.c_str(), &endptr, 10);
  60                 if (*endptr == 0) {
  61                         if (idx >= connectors.size())
  62                                 return nullptr;
  63
  64                         Connector* c = connectors[idx];
  65
  66                         if (contains(reserved, c))
  67                                 return nullptr;
  68
  69                         return c;
  70                 }
  71         }
  72
  73         for (Connector* conn : connectors) {
  74                 if (to_lower(conn->fullname()).find(to_lower(name)) == string::npos)
  75                         continue;
  76
  77                 if (contains(reserved, conn))
  78                         continue;
  79
  80                 return conn;
  81         }
  82
  83         return nullptr;
  84 }
  85
  86 Connector* ResourceManager::reserve_connector(const string& name)
  87 {
  88         Connector* conn;
  89
  90         if (name.empty())
  91                 conn = find_connector(m_card, m_reserved_connectors);
  92         else
  93                 conn = resolve_connector(m_card, name, m_reserved_connectors);
  94
  95         if (!conn)
  96                 return nullptr;
  97
  98         m_reserved_connectors.push_back(conn);
  99         return conn;
 100 }
 101
 102 Connector* ResourceManager::reserve_connector(Connector* conn)
 103 {
 104         if (!conn)
 105                 return nullptr;
 106
 107         if (contains(m_reserved_connectors, conn))
 108                 return nullptr;
 109
 110         m_reserved_connectors.push_back(conn);
 111         return conn;
 112 }
 113
 114 Crtc* ResourceManager::reserve_crtc(Connector* conn)
 115 {
 116         if (!conn)
 117                 return nullptr;
 118
 119         if (Crtc* crtc = conn->get_current_crtc()) {
 120                 m_reserved_crtcs.push_back(crtc);
 121                 return crtc;
 122         }
 123
 124         for (Crtc* crtc : conn->get_possible_crtcs()) {
 125                 if (contains(m_reserved_crtcs, crtc))
 126                         continue;
 127
 128                 m_reserved_crtcs.push_back(crtc);
 129                 return crtc;
 130         }
 131
 132         return nullptr;
 133 }
 134
 135 Plane* ResourceManager::reserve_plane(Crtc* crtc, PlaneType type, PixelFormat format)
 136 {
 137         if (!crtc)
 138                 return nullptr;
 139
 140         for (Plane* plane : crtc->get_possible_planes()) {
 141                 if (plane->plane_type() != type)
 142                         continue;
 143
 144                 if (format != PixelFormat::Undefined && !plane->supports_format(format))
 145                         continue;
 146
 147                 if (contains(m_reserved_planes, plane))
 148                         continue;
 149
 150                 m_reserved_planes.push_back(plane);
 151                 return plane;
 152         }
 153
 154         return nullptr;
 155 }
 156
 157 Plane* ResourceManager::reserve_generic_plane(Crtc* crtc, PixelFormat format)
 158 {
 159         if (!crtc)
 160                 return nullptr;
 161
 162         for (Plane* plane : crtc->get_possible_planes()) {
 163                 if (plane->plane_type() == PlaneType::Cursor)
 164                         continue;
 165
 166                 if (format != PixelFormat::Undefined && !plane->supports_format(format))
 167                         continue;
 168
 169                 if (contains(m_reserved_planes, plane))
 170                         continue;
 171
 172                 m_reserved_planes.push_back(plane);
 173                 return plane;
 174         }
 175
 176         return nullptr;
 177 }
 178
 179 Plane* ResourceManager::reserve_primary_plane(Crtc* crtc, PixelFormat format)
 180 {
 181         return reserve_plane(crtc, PlaneType::Primary, format);
 182 }
 183
 184 Plane* ResourceManager::reserve_overlay_plane(Crtc* crtc, PixelFormat format)
 185 {
 186         return reserve_plane(crtc, PlaneType::Overlay, format);
 187 }