agent_controls.cpp

Go to the documentation of this file.

#include "gui/agent_controls.h"
 
#include <QColorDialog>
#include <QLabel>
 
namespace Gui
{
    AgentControls::AgentControls(QWidget* parent)
        : QFrame(parent), m_Layout{ new QGridLayout(this) },
          m_ActiveAgentComboBox{ new QComboBox(this) },
          m_ActiveAgentColorBox{ new ColorBox(this) },
          m_NewAgentButton{ new QPushButton(this) }, m_ActiveAgentIndex{}
    {
        setObjectName("AgentControls");
 
        setFrameStyle(QFrame::Panel | QFrame::Raised);
 
        QLabel* heading{ new QLabel(this) };
        heading->setText("Agent Controls");
        m_Layout->addWidget(heading, 0, 0, 1, 3, Qt::AlignHCenter);
 
        QFrame* hLine{ new QFrame(this) };
        hLine->setFrameStyle(QFrame::HLine | QFrame::Sunken);
        m_Layout->addWidget(hLine, 1, 0, 1, 3);
 
        QFrame* activeAgentFrame{ new QFrame(this) };
        activeAgentFrame->setFrameStyle(QFrame::Panel | QFrame::Raised);
        QGridLayout* activeAgentFrameLayout{ new QGridLayout(
            activeAgentFrame) };
        QLabel* activeAgentHeading{ new QLabel(activeAgentFrame) };
        activeAgentHeading->setText("Active agent");
        activeAgentFrameLayout->addWidget(activeAgentHeading, 0, 0, 1, 3,
                                          Qt::AlignLeft);
        m_ActiveAgentComboBox->setCursor(Qt::PointingHandCursor);
        activeAgentFrameLayout->addWidget(m_ActiveAgentComboBox, 1, 0, 1, 2);
        activeAgentFrameLayout->addWidget(m_ActiveAgentColorBox, 1, 2, 1, 1);
        m_Layout->addWidget(activeAgentFrame, 2, 0, 1, 3);
 
        m_NewAgentButton->setText("New agent");
        m_NewAgentButton->setCursor(Qt::PointingHandCursor);
        m_Layout->addWidget(m_NewAgentButton, 3, 0, 1, 3);
 
        connect(m_ActiveAgentComboBox, SIGNAL(currentIndexChanged(int)), this,
                SLOT(SetActiveAgentIndex(int)));
 
        connect(m_NewAgentButton, &QPushButton::clicked,
                [this]() { emit AddAgent(); });
 
        connect(m_ActiveAgentColorBox, SIGNAL(ColorUpdated(QColor)), this,
                SLOT(SetAgentColor(QColor)));
    }
 
    void
    AgentControls::SetAgentColor(QColor color)
    {
        auto agent =
            std::find_if(m_Agents.first, m_Agents.second,
                         [&](const Core::Agent& agent) { return agent.Id == m_ActiveAgentIndex; });
        if (agent != m_Agents.second) {
            agent->Color = color.name().toStdString();
        }
 
        emit AgentChanged(m_Agents);
    }
 
    void
    AgentControls::UpdateAgents(
        std::pair<std::vector<Core::Agent>::iterator, std::vector<Core::Agent>::iterator>
            agents)
    {
        m_Agents = agents;
        m_ActiveAgentComboBox->clear();
        for (auto iter{ agents.first }; iter != agents.second; ++iter) {
            QString newAgentText = "Agent " + QString::number(iter->Id);
            m_ActiveAgentComboBox->blockSignals(true);
            m_ActiveAgentComboBox->insertItem(iter->Id, newAgentText);
            m_ActiveAgentComboBox->setCurrentIndex(iter->Id);
            m_ActiveAgentComboBox->blockSignals(false);
        }
        m_ActiveAgentIndex = m_ActiveAgentComboBox->currentIndex();
 
        m_ActiveAgentColorBox->update();
        emit AgentChanged(m_Agents);
        emit ActiveAgentChanged(m_ActiveAgentIndex);
    }
 
    void
    AgentControls::SetActiveAgentIndex(int index)
    {
        if (index == -1) {
            return;
        }
 
        m_ActiveAgentIndex = index;
        auto agent =
            std::find_if(m_Agents.first, m_Agents.second,
                         [&](const Core::Agent& agent) { return agent.Id == index; });
        if (agent != m_Agents.second) {
            m_ActiveAgentColorBox->UpdateColor(
                QColor(QString::fromStdString(agent->Color)));
        }
 
        m_ActiveAgentColorBox->update();
        emit ActiveAgentChanged(m_ActiveAgentIndex);
    }
 
    void
    AgentControls::SyncColor()
    {
        auto agent =
            std::find_if(m_Agents.first, m_Agents.second,
                         [&](const Core::Agent& agent) { return agent.Id == m_ActiveAgentIndex; });
        if (agent != m_Agents.second) {
            m_ActiveAgentColorBox->UpdateColor(
                QColor(QString::fromStdString(agent->Color)));
        }
 
        m_ActiveAgentColorBox->update();
    }
 
} // namespace Gui