calc4gwidget.cpp

#include "calc4gwidget.h"
#include "ui_calc4gwidget.h"
#include "carrierrow4g.h"
#include <QVBoxLayout>
#include <QOpenGLWidget>
#include <QScroller>

Calc4gWidget::Calc4gWidget(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::Calc4gWidget)
{
    ui->setupUi(this);

    // Fix scrolling and gesture recognition
    QScroller::grabGesture(ui->scrollArea, QScroller::TouchGesture);

    if (isOpenGLSupported()) {
        // Safely "extract" the content panel created by the .ui file (which contains the carriersLayout)
        // takeWidget() will remove the parent-child relationship but will not delete the object.
        QWidget* contentWidget = ui->scrollArea->takeWidget();
        // Now safe to set a new OpenGL viewport
        ui->scrollArea->setViewport(new QOpenGLWidget());
        // Put the removed content panel "back" to the QScrollArea.
        // setWidget() will automatically set it as a child of the new viewport and manage its size.
        ui->scrollArea->setWidget(contentWidget);
    }

    connect(ui->addCarrierButton, &QPushButton::clicked, this, &Calc4gWidget::addCarrierRow);
    addCarrierRow(); // Add the primary carrier on init
}

Calc4gWidget::~Calc4gWidget()
{
    delete ui;
}

void Calc4gWidget::addCarrierRow()
{
    int newId = m_carrierRows.count();
    bool isPrimary = (newId == m_primaryCarrierId);
    CarrierRow4G *row = new CarrierRow4G(newId, isPrimary, this);
    ui->carriersLayout->addWidget(row);
    m_carrierRows.append(row);

    connect(row, &CarrierRow4G::removeClicked, this, &Calc4gWidget::removeCarrierRow);
    connect(row, &CarrierRow4G::uplinkAggClicked, this, &Calc4gWidget::toggleUplinkAggregation);
    connect(row, &CarrierRow4G::dataChanged, this, &Calc4gWidget::calculateAll, Qt::QueuedConnection);

    calculateAll();
}

void Calc4gWidget::removeCarrierRow(int carrierId)
{
    if (carrierId == m_primaryCarrierId) return;

    CarrierRow4G *rowToRemove = nullptr;
    int indexToRemove = -1;
    for (int i = 0; i < m_carrierRows.size(); ++i) {
        if (m_carrierRows[i]->id() == carrierId) {
            rowToRemove = m_carrierRows[i];
            indexToRemove = i;
            break;
        }
    }

    if (rowToRemove) {
        // Remove from data model
        m_carrierRows.removeAt(indexToRemove);

        // Remove and delete objects from the UI layout
        ui->carriersLayout->removeWidget(rowToRemove);
        rowToRemove->deleteLater();

        // Update uplink aggregated carrier ID (important)
        QSet<int> updatedUplinkCarriers;
        for (int id : m_uplinkCarriers) {
            if (id > carrierId) {
                updatedUplinkCarriers.insert(id - 1);
            } else if (id < carrierId) {
                updatedUplinkCarriers.insert(id);
            }
        }
        m_uplinkCarriers = updatedUplinkCarriers;

        // Reindex all remaining carriers (important)
        reindexCarriers();

        // In the end recalculate all
        calculateAll();
    }
}

void Calc4gWidget::reindexCarriers()
{
    for (int i = 0; i < m_carrierRows.count(); ++i) {
        CarrierRow4G *row = m_carrierRows.at(i);
        row->setId(i);
    }
}

void Calc4gWidget::toggleUplinkAggregation(int carrierId)
{
    if (m_uplinkCarriers.contains(carrierId)) {
        m_uplinkCarriers.remove(carrierId);
    } else {
        m_uplinkCarriers.insert(carrierId);
    }
    calculateAll();
}

void Calc4gWidget::calculateAll()
{
    double totalDownlink = 0.0;
    double totalUplink = 0.0;
    QString errorMessage;

    for (int i = 0; i < m_carrierRows.size(); ++i) {
        CarrierRow4G* row = m_carrierRows[i];
        Calc4GResult result = row->calculate();

        if (result.hasUnselectedBands) {
            errorMessage = QString("Please select a band for Carrier #%1").arg(i + 1);
            break;
        }

        totalDownlink += result.downlink;
        if (m_uplinkCarriers.contains(row->id()) || row->id() == m_primaryCarrierId || row->isSUL()) {
            totalUplink += result.uplink;
        }
    }

    if (!errorMessage.isEmpty()) {
        ui->totalSpeedLabel->setText(errorMessage);
    } else {
        QString dlTxt = totalDownlink > 0 ? QString("%1Mbps ↓").arg(round2(totalDownlink)) : "";
        QString ulTxt = totalUplink > 0 ? QString("%1Mbps ↑").arg(round2(totalUplink)) : "";
        QStringList parts;
        if (!dlTxt.isEmpty()) parts << dlTxt;
        if (!ulTxt.isEmpty()) parts << ulTxt;

        QString displayText = parts.join(" & "); // Label does not require '&&' escape
        if (displayText.isEmpty()) {
            displayText = "Please select carrier options";
        }

        ui->totalSpeedLabel->setText(displayText);
    }
}