1 files changed, 221 insertions, 0 deletions
diff --git a/kms++util/src/resourcemanager.cpp b/kms++util/src/resourcemanager.cpp
new file mode 100644
index 0000000..424c8f0
--- /dev/null
+++ b/kms++util/src/resourcemanager.cpp
@@ -0,0 +1,221 @@
+#include <kms++util/resourcemanager.h>
+#include <algorithm>
+#include <kms++util/strhelpers.h>
+
+using namespace kms;
+using namespace std;
+
+ResourceManager::ResourceManager(Card& card)
+	: m_card(card)
+{
+}
+
+void ResourceManager::reset()
+{
+	m_reserved_connectors.clear();
+	m_reserved_crtcs.clear();
+	m_reserved_planes.clear();
+}
+
+static Connector* find_connector(Card& card, const set<Connector*>& reserved)
+{
+	for (Connector* conn : card.get_connectors()) {
+		if (!conn->connected())
+			continue;
+
+		if (reserved.count(conn))
+			continue;
+
+		return conn;
+	}
+
+	return nullptr;
+}
+
+static Connector* resolve_connector(Card& card, const string& name, const set<Connector*>& reserved)
+{
+	auto connectors = card.get_connectors();
+
+	if (name[0] == '@') {
+		char* endptr;
+		unsigned id = strtoul(name.c_str() + 1, &endptr, 10);
+		if (*endptr == 0) {
+			Connector* c = card.get_connector(id);
+
+			if (!c || reserved.count(c))
+				return nullptr;
+
+			return c;
+		}
+	} else {
+		char* endptr;
+		unsigned idx = strtoul(name.c_str(), &endptr, 10);
+		if (*endptr == 0) {
+			if (idx >= connectors.size())
+				return nullptr;
+
+			Connector* c = connectors[idx];
+
+			if (reserved.count(c))
+				return nullptr;
+
+			return c;
+		}
+	}
+
+	for (Connector* conn : connectors) {
+		if (to_lower(conn->fullname()).find(to_lower(name)) == string::npos)
+			continue;
+
+		if (reserved.count(conn))
+			continue;
+
+		return conn;
+	}
+
+	return nullptr;
+}
+
+Connector* ResourceManager::reserve_connector(const string& name)
+{
+	Connector* conn;
+
+	if (name.empty())
+		conn = find_connector(m_card, m_reserved_connectors);
+	else
+		conn = resolve_connector(m_card, name, m_reserved_connectors);
+
+	if (!conn)
+		return nullptr;
+
+	m_reserved_connectors.insert(conn);
+	return conn;
+}
+
+Connector* ResourceManager::reserve_connector(Connector* conn)
+{
+	if (!conn)
+		return nullptr;
+
+	if (m_reserved_connectors.count(conn))
+		return nullptr;
+
+	m_reserved_connectors.insert(conn);
+	return conn;
+}
+
+void ResourceManager::release_connector(Connector* conn)
+{
+	m_reserved_connectors.erase(conn);
+}
+
+Crtc* ResourceManager::reserve_crtc(Connector* conn)
+{
+	if (!conn)
+		return nullptr;
+
+	if (Crtc* crtc = conn->get_current_crtc()) {
+		m_reserved_crtcs.insert(crtc);
+		return crtc;
+	}
+
+	for (Crtc* crtc : conn->get_possible_crtcs()) {
+		if (m_reserved_crtcs.count(crtc))
+			continue;
+
+		m_reserved_crtcs.insert(crtc);
+		return crtc;
+	}
+
+	return nullptr;
+}
+
+Crtc* ResourceManager::reserve_crtc(Crtc* crtc)
+{
+	if (!crtc)
+		return nullptr;
+
+	if (m_reserved_crtcs.count(crtc))
+		return nullptr;
+
+	m_reserved_crtcs.insert(crtc);
+
+	return crtc;
+}
+
+void ResourceManager::release_crtc(Crtc* crtc)
+{
+	m_reserved_crtcs.erase(crtc);
+}
+
+Plane* ResourceManager::reserve_plane(Crtc* crtc, PlaneType type, PixelFormat format)
+{
+	if (!crtc)
+		return nullptr;
+
+	for (Plane* plane : crtc->get_possible_planes()) {
+		if (plane->plane_type() != type)
+			continue;
+
+		if (format != PixelFormat::Undefined && !plane->supports_format(format))
+			continue;
+
+		if (m_reserved_planes.count(plane))
+			continue;
+
+		m_reserved_planes.insert(plane);
+		return plane;
+	}
+
+	return nullptr;
+}
+
+Plane* ResourceManager::reserve_plane(Plane* plane)
+{
+	if (!plane)
+		return nullptr;
+
+	if (m_reserved_planes.count(plane))
+		return nullptr;
+
+	m_reserved_planes.insert(plane);
+
+	return plane;
+}
+
+Plane* ResourceManager::reserve_generic_plane(Crtc* crtc, PixelFormat format)
+{
+	if (!crtc)
+		return nullptr;
+
+	for (Plane* plane : crtc->get_possible_planes()) {
+		if (plane->plane_type() == PlaneType::Cursor)
+			continue;
+
+		if (format != PixelFormat::Undefined && !plane->supports_format(format))
+			continue;
+
+		if (m_reserved_planes.count(plane))
+			continue;
+
+		m_reserved_planes.insert(plane);
+		return plane;
+	}
+
+	return nullptr;
+}
+
+Plane* ResourceManager::reserve_primary_plane(Crtc* crtc, PixelFormat format)
+{
+	return reserve_plane(crtc, PlaneType::Primary, format);
+}
+
+Plane* ResourceManager::reserve_overlay_plane(Crtc* crtc, PixelFormat format)
+{
+	return reserve_plane(crtc, PlaneType::Overlay, format);
+}
+
+void ResourceManager::release_plane(Plane* plane)
+{
+	m_reserved_planes.erase(plane);
+}