598 lines
17 KiB
C++
598 lines
17 KiB
C++
#include "WarpGraphModel.h"
|
|
|
|
#include <QColor>
|
|
#include <QJsonObject>
|
|
#include <QVariant>
|
|
|
|
#include <QtNodes/NodeStyle>
|
|
#include <QtNodes/StyleCollection>
|
|
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
|
|
WarpGraphModel::WarpGraphModel(warppipe::Client *client, QObject *parent)
|
|
: QtNodes::AbstractGraphModel(), m_client(client) {
|
|
if (parent) {
|
|
setParent(parent);
|
|
}
|
|
}
|
|
|
|
QtNodes::NodeId WarpGraphModel::newNodeId() { return m_nextNodeId++; }
|
|
|
|
std::unordered_set<QtNodes::NodeId> WarpGraphModel::allNodeIds() const {
|
|
std::unordered_set<QtNodes::NodeId> ids;
|
|
ids.reserve(m_nodes.size());
|
|
for (const auto &entry : m_nodes) {
|
|
ids.insert(entry.first);
|
|
}
|
|
return ids;
|
|
}
|
|
|
|
std::unordered_set<QtNodes::ConnectionId>
|
|
WarpGraphModel::allConnectionIds(QtNodes::NodeId const nodeId) const {
|
|
std::unordered_set<QtNodes::ConnectionId> result;
|
|
for (const auto &conn : m_connections) {
|
|
if (conn.outNodeId == nodeId || conn.inNodeId == nodeId) {
|
|
result.insert(conn);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::unordered_set<QtNodes::ConnectionId>
|
|
WarpGraphModel::connections(QtNodes::NodeId nodeId,
|
|
QtNodes::PortType portType,
|
|
QtNodes::PortIndex portIndex) const {
|
|
std::unordered_set<QtNodes::ConnectionId> result;
|
|
for (const auto &conn : m_connections) {
|
|
if (portType == QtNodes::PortType::Out) {
|
|
if (conn.outNodeId == nodeId && conn.outPortIndex == portIndex) {
|
|
result.insert(conn);
|
|
}
|
|
} else if (portType == QtNodes::PortType::In) {
|
|
if (conn.inNodeId == nodeId && conn.inPortIndex == portIndex) {
|
|
result.insert(conn);
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool WarpGraphModel::connectionExists(
|
|
QtNodes::ConnectionId const connectionId) const {
|
|
return m_connections.find(connectionId) != m_connections.end();
|
|
}
|
|
|
|
QtNodes::NodeId WarpGraphModel::addNode(QString const) {
|
|
return newNodeId();
|
|
}
|
|
|
|
bool WarpGraphModel::connectionPossible(
|
|
QtNodes::ConnectionId const connectionId) const {
|
|
if (!nodeExists(connectionId.outNodeId) ||
|
|
!nodeExists(connectionId.inNodeId)) {
|
|
return false;
|
|
}
|
|
if (connectionExists(connectionId)) {
|
|
return false;
|
|
}
|
|
|
|
auto outIt = m_nodes.find(connectionId.outNodeId);
|
|
auto inIt = m_nodes.find(connectionId.inNodeId);
|
|
if (outIt == m_nodes.end() || inIt == m_nodes.end()) {
|
|
return false;
|
|
}
|
|
|
|
auto outIdx = static_cast<size_t>(connectionId.outPortIndex);
|
|
auto inIdx = static_cast<size_t>(connectionId.inPortIndex);
|
|
if (outIdx >= outIt->second.outputPorts.size()) {
|
|
return false;
|
|
}
|
|
if (inIdx >= inIt->second.inputPorts.size()) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void WarpGraphModel::addConnection(
|
|
QtNodes::ConnectionId const connectionId) {
|
|
if (!connectionPossible(connectionId)) {
|
|
return;
|
|
}
|
|
m_connections.insert(connectionId);
|
|
Q_EMIT connectionCreated(connectionId);
|
|
}
|
|
|
|
bool WarpGraphModel::nodeExists(QtNodes::NodeId const nodeId) const {
|
|
return m_nodes.find(nodeId) != m_nodes.end();
|
|
}
|
|
|
|
QVariant WarpGraphModel::nodeData(QtNodes::NodeId nodeId,
|
|
QtNodes::NodeRole role) const {
|
|
auto it = m_nodes.find(nodeId);
|
|
if (it == m_nodes.end()) {
|
|
return QVariant();
|
|
}
|
|
|
|
const auto &data = it->second;
|
|
|
|
switch (role) {
|
|
case QtNodes::NodeRole::Caption:
|
|
return captionForNode(data.info);
|
|
case QtNodes::NodeRole::CaptionVisible:
|
|
return true;
|
|
case QtNodes::NodeRole::Position: {
|
|
auto posIt = m_positions.find(nodeId);
|
|
if (posIt != m_positions.end()) {
|
|
return posIt->second;
|
|
}
|
|
return QPointF(0, 0);
|
|
}
|
|
case QtNodes::NodeRole::Size: {
|
|
auto sizeIt = m_sizes.find(nodeId);
|
|
if (sizeIt != m_sizes.end()) {
|
|
return sizeIt->second;
|
|
}
|
|
return estimateNodeSize(data);
|
|
}
|
|
case QtNodes::NodeRole::InPortCount:
|
|
return static_cast<unsigned int>(data.inputPorts.size());
|
|
case QtNodes::NodeRole::OutPortCount:
|
|
return static_cast<unsigned int>(data.outputPorts.size());
|
|
case QtNodes::NodeRole::Type:
|
|
return QString("PipeWire");
|
|
case QtNodes::NodeRole::Style: {
|
|
bool ghost = m_ghostNodes.find(nodeId) != m_ghostNodes.end();
|
|
WarpNodeType type = classifyNode(data.info);
|
|
return styleForNode(type, ghost);
|
|
}
|
|
default:
|
|
return QVariant();
|
|
}
|
|
}
|
|
|
|
bool WarpGraphModel::setNodeData(QtNodes::NodeId nodeId,
|
|
QtNodes::NodeRole role, QVariant value) {
|
|
if (!nodeExists(nodeId)) {
|
|
return false;
|
|
}
|
|
|
|
if (role == QtNodes::NodeRole::Position) {
|
|
m_positions[nodeId] = value.toPointF();
|
|
Q_EMIT nodePositionUpdated(nodeId);
|
|
return true;
|
|
}
|
|
|
|
if (role == QtNodes::NodeRole::Size) {
|
|
m_sizes[nodeId] = value.toSize();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
QVariant WarpGraphModel::portData(QtNodes::NodeId nodeId,
|
|
QtNodes::PortType portType,
|
|
QtNodes::PortIndex portIndex,
|
|
QtNodes::PortRole role) const {
|
|
auto it = m_nodes.find(nodeId);
|
|
if (it == m_nodes.end()) {
|
|
return QVariant();
|
|
}
|
|
|
|
const auto &data = it->second;
|
|
|
|
if (role == QtNodes::PortRole::DataType) {
|
|
return QString("audio");
|
|
}
|
|
|
|
if (role == QtNodes::PortRole::CaptionVisible) {
|
|
return true;
|
|
}
|
|
|
|
if (role == QtNodes::PortRole::Caption) {
|
|
if (portType == QtNodes::PortType::In) {
|
|
auto idx = static_cast<size_t>(portIndex);
|
|
if (idx < data.inputPorts.size()) {
|
|
return QString::fromStdString(data.inputPorts[idx].name);
|
|
}
|
|
} else if (portType == QtNodes::PortType::Out) {
|
|
auto idx = static_cast<size_t>(portIndex);
|
|
if (idx < data.outputPorts.size()) {
|
|
return QString::fromStdString(data.outputPorts[idx].name);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (role == QtNodes::PortRole::ConnectionPolicyRole) {
|
|
return QVariant::fromValue(QtNodes::ConnectionPolicy::Many);
|
|
}
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
bool WarpGraphModel::setPortData(QtNodes::NodeId, QtNodes::PortType,
|
|
QtNodes::PortIndex, QVariant const &,
|
|
QtNodes::PortRole) {
|
|
return false;
|
|
}
|
|
|
|
bool WarpGraphModel::deleteConnection(
|
|
QtNodes::ConnectionId const connectionId) {
|
|
auto it = m_connections.find(connectionId);
|
|
if (it == m_connections.end()) {
|
|
return false;
|
|
}
|
|
m_connections.erase(it);
|
|
Q_EMIT connectionDeleted(connectionId);
|
|
return true;
|
|
}
|
|
|
|
bool WarpGraphModel::deleteNode(QtNodes::NodeId const nodeId) {
|
|
if (!nodeExists(nodeId)) {
|
|
return false;
|
|
}
|
|
|
|
std::vector<QtNodes::ConnectionId> toRemove;
|
|
for (const auto &conn : m_connections) {
|
|
if (conn.outNodeId == nodeId || conn.inNodeId == nodeId) {
|
|
toRemove.push_back(conn);
|
|
}
|
|
}
|
|
for (const auto &conn : toRemove) {
|
|
deleteConnection(conn);
|
|
}
|
|
|
|
m_nodes.erase(nodeId);
|
|
m_positions.erase(nodeId);
|
|
m_sizes.erase(nodeId);
|
|
Q_EMIT nodeDeleted(nodeId);
|
|
return true;
|
|
}
|
|
|
|
QJsonObject WarpGraphModel::saveNode(QtNodes::NodeId const nodeId) const {
|
|
QJsonObject obj;
|
|
obj["id"] = static_cast<qint64>(nodeId);
|
|
QPointF pos = nodeData(nodeId, QtNodes::NodeRole::Position).toPointF();
|
|
QJsonObject posObj;
|
|
posObj["x"] = pos.x();
|
|
posObj["y"] = pos.y();
|
|
obj["position"] = posObj;
|
|
return obj;
|
|
}
|
|
|
|
void WarpGraphModel::loadNode(QJsonObject const &) {}
|
|
|
|
void WarpGraphModel::refreshFromClient() {
|
|
if (!m_client) {
|
|
return;
|
|
}
|
|
|
|
auto nodesResult = m_client->ListNodes();
|
|
if (!nodesResult.ok()) {
|
|
return;
|
|
}
|
|
|
|
std::unordered_set<uint32_t> seenPwIds;
|
|
|
|
for (const auto &nodeInfo : nodesResult.value) {
|
|
seenPwIds.insert(nodeInfo.id.value);
|
|
|
|
auto existing = m_pwToQt.find(nodeInfo.id.value);
|
|
if (existing != m_pwToQt.end()) {
|
|
QtNodes::NodeId qtId = existing->second;
|
|
auto &data = m_nodes[qtId];
|
|
data.info = nodeInfo;
|
|
if (m_ghostNodes.erase(qtId)) {
|
|
Q_EMIT nodeUpdated(qtId);
|
|
}
|
|
continue;
|
|
}
|
|
|
|
QtNodes::NodeId ghostMatch = 0;
|
|
std::string nodeName = nodeInfo.name;
|
|
for (const auto &ghostId : m_ghostNodes) {
|
|
auto ghostIt = m_nodes.find(ghostId);
|
|
if (ghostIt != m_nodes.end() &&
|
|
ghostIt->second.info.name == nodeName) {
|
|
ghostMatch = ghostId;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ghostMatch != 0) {
|
|
m_ghostNodes.erase(ghostMatch);
|
|
m_pwToQt.emplace(nodeInfo.id.value, ghostMatch);
|
|
auto &data = m_nodes[ghostMatch];
|
|
data.info = nodeInfo;
|
|
|
|
auto portsResult = m_client->ListPorts(nodeInfo.id);
|
|
if (portsResult.ok()) {
|
|
data.inputPorts.clear();
|
|
data.outputPorts.clear();
|
|
for (const auto &port : portsResult.value) {
|
|
if (port.is_input) {
|
|
data.inputPorts.push_back(port);
|
|
} else {
|
|
data.outputPorts.push_back(port);
|
|
}
|
|
}
|
|
std::sort(data.inputPorts.begin(), data.inputPorts.end(),
|
|
[](const auto &a, const auto &b) { return a.name < b.name; });
|
|
std::sort(data.outputPorts.begin(), data.outputPorts.end(),
|
|
[](const auto &a, const auto &b) { return a.name < b.name; });
|
|
}
|
|
|
|
Q_EMIT nodeUpdated(ghostMatch);
|
|
continue;
|
|
}
|
|
|
|
auto portsResult = m_client->ListPorts(nodeInfo.id);
|
|
std::vector<warppipe::PortInfo> inputs;
|
|
std::vector<warppipe::PortInfo> outputs;
|
|
if (portsResult.ok()) {
|
|
for (const auto &port : portsResult.value) {
|
|
if (port.is_input) {
|
|
inputs.push_back(port);
|
|
} else {
|
|
outputs.push_back(port);
|
|
}
|
|
}
|
|
std::sort(inputs.begin(), inputs.end(),
|
|
[](const auto &a, const auto &b) { return a.name < b.name; });
|
|
std::sort(outputs.begin(), outputs.end(),
|
|
[](const auto &a, const auto &b) { return a.name < b.name; });
|
|
}
|
|
|
|
QtNodes::NodeId qtId = newNodeId();
|
|
WarpNodeData data;
|
|
data.info = nodeInfo;
|
|
data.inputPorts = std::move(inputs);
|
|
data.outputPorts = std::move(outputs);
|
|
|
|
auto [nodeIt, _] = m_nodes.emplace(qtId, std::move(data));
|
|
m_pwToQt.emplace(nodeInfo.id.value, qtId);
|
|
m_positions.emplace(qtId, nextPosition(nodeIt->second));
|
|
|
|
Q_EMIT nodeCreated(qtId);
|
|
}
|
|
|
|
std::vector<uint32_t> disappearedPwIds;
|
|
for (const auto &[pwId, qtId] : m_pwToQt) {
|
|
if (seenPwIds.find(pwId) == seenPwIds.end()) {
|
|
disappearedPwIds.push_back(pwId);
|
|
}
|
|
}
|
|
for (uint32_t pwId : disappearedPwIds) {
|
|
auto it = m_pwToQt.find(pwId);
|
|
if (it == m_pwToQt.end()) {
|
|
continue;
|
|
}
|
|
QtNodes::NodeId qtId = it->second;
|
|
auto nodeIt = m_nodes.find(qtId);
|
|
if (nodeIt == m_nodes.end()) {
|
|
continue;
|
|
}
|
|
WarpNodeType type = classifyNode(nodeIt->second.info);
|
|
if (type == WarpNodeType::kApplication) {
|
|
m_ghostNodes.insert(qtId);
|
|
m_pwToQt.erase(it);
|
|
Q_EMIT nodeUpdated(qtId);
|
|
} else {
|
|
m_pwToQt.erase(it);
|
|
deleteNode(qtId);
|
|
}
|
|
}
|
|
|
|
auto linksResult = m_client->ListLinks();
|
|
if (linksResult.ok()) {
|
|
std::unordered_set<uint32_t> seenLinkIds;
|
|
for (const auto &link : linksResult.value) {
|
|
seenLinkIds.insert(link.id.value);
|
|
|
|
if (m_linkIdToConn.find(link.id.value) != m_linkIdToConn.end()) {
|
|
continue;
|
|
}
|
|
|
|
auto outNodeIt = m_pwToQt.end();
|
|
auto inNodeIt = m_pwToQt.end();
|
|
QtNodes::PortIndex outPortIdx = 0;
|
|
QtNodes::PortIndex inPortIdx = 0;
|
|
bool found = false;
|
|
|
|
for (const auto &[qtId, nodeData] : m_nodes) {
|
|
for (size_t i = 0; i < nodeData.outputPorts.size(); ++i) {
|
|
if (nodeData.outputPorts[i].id.value == link.output_port.value) {
|
|
outNodeIt = m_pwToQt.find(nodeData.info.id.value);
|
|
outPortIdx = static_cast<QtNodes::PortIndex>(i);
|
|
}
|
|
}
|
|
for (size_t i = 0; i < nodeData.inputPorts.size(); ++i) {
|
|
if (nodeData.inputPorts[i].id.value == link.input_port.value) {
|
|
inNodeIt = m_pwToQt.find(nodeData.info.id.value);
|
|
inPortIdx = static_cast<QtNodes::PortIndex>(i);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (outNodeIt != m_pwToQt.end() && inNodeIt != m_pwToQt.end()) {
|
|
found = true;
|
|
}
|
|
|
|
if (found) {
|
|
QtNodes::ConnectionId connId{outNodeIt->second, outPortIdx,
|
|
inNodeIt->second, inPortIdx};
|
|
if (m_connections.find(connId) == m_connections.end()) {
|
|
m_connections.insert(connId);
|
|
m_linkIdToConn.emplace(link.id.value, connId);
|
|
Q_EMIT connectionCreated(connId);
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<uint32_t> staleLinkIds;
|
|
for (const auto &[linkId, connId] : m_linkIdToConn) {
|
|
if (seenLinkIds.find(linkId) == seenLinkIds.end()) {
|
|
staleLinkIds.push_back(linkId);
|
|
}
|
|
}
|
|
for (uint32_t linkId : staleLinkIds) {
|
|
auto it = m_linkIdToConn.find(linkId);
|
|
if (it != m_linkIdToConn.end()) {
|
|
auto connIt = m_connections.find(it->second);
|
|
if (connIt != m_connections.end()) {
|
|
QtNodes::ConnectionId connId = it->second;
|
|
m_connections.erase(connIt);
|
|
Q_EMIT connectionDeleted(connId);
|
|
}
|
|
m_linkIdToConn.erase(it);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
const WarpNodeData *
|
|
WarpGraphModel::warpNodeData(QtNodes::NodeId nodeId) const {
|
|
auto it = m_nodes.find(nodeId);
|
|
if (it != m_nodes.end()) {
|
|
return &it->second;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
QtNodes::NodeId WarpGraphModel::qtNodeIdForPw(uint32_t pwNodeId) const {
|
|
auto it = m_pwToQt.find(pwNodeId);
|
|
if (it != m_pwToQt.end()) {
|
|
return it->second;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
QString WarpGraphModel::captionForNode(const warppipe::NodeInfo &info) {
|
|
if (!info.description.empty()) {
|
|
return QString::fromStdString(info.description);
|
|
}
|
|
if (!info.application_name.empty() && info.application_name != info.name) {
|
|
return QString::fromStdString(info.application_name);
|
|
}
|
|
return QString::fromStdString(info.name);
|
|
}
|
|
|
|
QSize WarpGraphModel::estimateNodeSize(const WarpNodeData &data) {
|
|
int maxPorts = static_cast<int>(
|
|
std::max(data.inputPorts.size(), data.outputPorts.size()));
|
|
int height = std::max(80, 50 + maxPorts * 28);
|
|
|
|
QString caption = captionForNode(data.info);
|
|
int captionWidth = caption.length() * 8 + 40;
|
|
|
|
int maxInputLen = 0;
|
|
int maxOutputLen = 0;
|
|
for (const auto &p : data.inputPorts)
|
|
maxInputLen = std::max(maxInputLen, static_cast<int>(p.name.length()));
|
|
for (const auto &p : data.outputPorts)
|
|
maxOutputLen = std::max(maxOutputLen, static_cast<int>(p.name.length()));
|
|
int portWidth = (maxInputLen + maxOutputLen) * 7 + 60;
|
|
|
|
int width = std::max(180, std::max(captionWidth, portWidth));
|
|
return QSize(width, height);
|
|
}
|
|
|
|
QPointF WarpGraphModel::nextPosition(const WarpNodeData &data) {
|
|
QSize size = estimateNodeSize(data);
|
|
double nodeW = size.width();
|
|
double nodeH = size.height();
|
|
|
|
if (m_nextX + nodeW > kMaxRowWidth && m_nextX > 0) {
|
|
m_nextX = 0.0;
|
|
m_nextY += m_rowMaxHeight + kVerticalGap;
|
|
m_rowMaxHeight = 0.0;
|
|
}
|
|
|
|
QPointF pos(m_nextX, m_nextY);
|
|
m_nextX += nodeW + kHorizontalGap;
|
|
m_rowMaxHeight = std::max(m_rowMaxHeight, nodeH);
|
|
return pos;
|
|
}
|
|
|
|
bool WarpGraphModel::isGhost(QtNodes::NodeId nodeId) const {
|
|
return m_ghostNodes.find(nodeId) != m_ghostNodes.end();
|
|
}
|
|
|
|
WarpNodeType
|
|
WarpGraphModel::classifyNode(const warppipe::NodeInfo &info) {
|
|
const std::string &mc = info.media_class;
|
|
|
|
bool isVirtual = (info.name.find("warppipe") != std::string::npos);
|
|
|
|
if (mc == "Audio/Sink" || mc == "Audio/Duplex") {
|
|
return isVirtual ? WarpNodeType::kVirtualSink : WarpNodeType::kHardwareSink;
|
|
}
|
|
if (mc == "Audio/Source") {
|
|
return isVirtual ? WarpNodeType::kVirtualSource
|
|
: WarpNodeType::kHardwareSource;
|
|
}
|
|
if (mc == "Stream/Output/Audio" || mc == "Stream/Input/Audio") {
|
|
return WarpNodeType::kApplication;
|
|
}
|
|
|
|
return WarpNodeType::kUnknown;
|
|
}
|
|
|
|
QVariant WarpGraphModel::styleForNode(WarpNodeType type, bool ghost) {
|
|
QtNodes::NodeStyle style = QtNodes::StyleCollection::nodeStyle();
|
|
|
|
QColor base;
|
|
switch (type) {
|
|
case WarpNodeType::kHardwareSink:
|
|
base = QColor(72, 94, 118);
|
|
break;
|
|
case WarpNodeType::kHardwareSource:
|
|
base = QColor(94, 72, 118);
|
|
break;
|
|
case WarpNodeType::kVirtualSink:
|
|
base = QColor(62, 122, 104);
|
|
break;
|
|
case WarpNodeType::kVirtualSource:
|
|
base = QColor(62, 104, 122);
|
|
break;
|
|
case WarpNodeType::kApplication:
|
|
base = QColor(138, 104, 72);
|
|
break;
|
|
default:
|
|
base = QColor(86, 94, 108);
|
|
break;
|
|
}
|
|
|
|
if (ghost) {
|
|
style.GradientColor0 = base.darker(150);
|
|
style.GradientColor1 = base.darker(160);
|
|
style.GradientColor2 = base.darker(170);
|
|
style.GradientColor3 = base.darker(180);
|
|
style.NormalBoundaryColor = base.darker(130);
|
|
style.FontColor = QColor(160, 168, 182);
|
|
style.FontColorFaded = QColor(120, 128, 142);
|
|
style.ConnectionPointColor = QColor(140, 148, 160);
|
|
style.FilledConnectionPointColor = QColor(180, 140, 80);
|
|
style.Opacity = 0.6f;
|
|
} else {
|
|
style.GradientColor0 = base.lighter(120);
|
|
style.GradientColor1 = base.lighter(108);
|
|
style.GradientColor2 = base.darker(105);
|
|
style.GradientColor3 = base.darker(120);
|
|
style.NormalBoundaryColor = base.lighter(135);
|
|
style.FontColor = QColor(236, 240, 246);
|
|
style.FontColorFaded = QColor(160, 168, 182);
|
|
style.ConnectionPointColor = QColor(200, 208, 220);
|
|
style.FilledConnectionPointColor = QColor(255, 165, 0);
|
|
style.Opacity = 1.0f;
|
|
}
|
|
|
|
style.SelectedBoundaryColor = QColor(255, 165, 0);
|
|
style.PenWidth = 1.3f;
|
|
style.HoveredPenWidth = 2.4f;
|
|
style.ConnectionPointDiameter = 10.0f;
|
|
|
|
return style.toJson().toVariantMap();
|
|
}
|