mirror of
https://github.com/Cockatrice/Cockatrice.git
synced 2026-06-09 15:54:47 -07:00
ace4063371
* Helper to query deckList for DecklistCardNodes. Took 30 minutes Took 6 minutes Took 2 minutes * Fix unused. Took 3 minutes Took 1 minute * Convert string to string list. Took 2 minutes * Adjust to rebase. Took 2 minutes --------- Co-authored-by: Lukas Brübach <Bruebach.Lukas@bdosecurity.de>
94 lines
2.7 KiB
C++
94 lines
2.7 KiB
C++
#include "mana_curve_widget.h"
|
|
|
|
#include "../../../main.h"
|
|
#include "../../deck_loader/deck_loader.h"
|
|
#include "../general/display/banner_widget.h"
|
|
#include "../general/display/bar_widget.h"
|
|
|
|
#include <libcockatrice/card/database/card_database.h>
|
|
#include <libcockatrice/card/database/card_database_manager.h>
|
|
#include <libcockatrice/deck_list/deck_list.h>
|
|
#include <unordered_map>
|
|
|
|
ManaCurveWidget::ManaCurveWidget(QWidget *parent, DeckListModel *_deckListModel)
|
|
: QWidget(parent), deckListModel(_deckListModel)
|
|
{
|
|
layout = new QVBoxLayout(this);
|
|
setLayout(layout);
|
|
|
|
bannerWidget = new BannerWidget(this, tr("Mana Curve"), Qt::Vertical, 100);
|
|
bannerWidget->setMaximumHeight(100);
|
|
layout->addWidget(bannerWidget);
|
|
|
|
barContainer = new QWidget(this);
|
|
barLayout = new QHBoxLayout(barContainer);
|
|
layout->addWidget(barContainer);
|
|
|
|
connect(deckListModel, &DeckListModel::dataChanged, this, &ManaCurveWidget::analyzeManaCurve);
|
|
|
|
retranslateUi();
|
|
}
|
|
|
|
void ManaCurveWidget::retranslateUi()
|
|
{
|
|
bannerWidget->setText(tr("Mana Curve"));
|
|
}
|
|
|
|
void ManaCurveWidget::setDeckModel(DeckListModel *deckModel)
|
|
{
|
|
deckListModel = deckModel;
|
|
connect(deckListModel, &DeckListModel::dataChanged, this, &ManaCurveWidget::analyzeManaCurve);
|
|
analyzeManaCurve();
|
|
}
|
|
|
|
std::unordered_map<int, int> ManaCurveWidget::analyzeManaCurve()
|
|
{
|
|
manaCurveMap.clear();
|
|
|
|
QList<DecklistCardNode *> cardsInDeck = deckListModel->getDeckList()->getCardNodes();
|
|
|
|
for (auto currentCard : cardsInDeck) {
|
|
for (int k = 0; k < currentCard->getNumber(); ++k) {
|
|
CardInfoPtr info = CardDatabaseManager::query()->getCardInfo(currentCard->getName());
|
|
if (info) {
|
|
int cmc = info->getCmc().toInt();
|
|
manaCurveMap[cmc]++;
|
|
}
|
|
}
|
|
}
|
|
|
|
updateDisplay();
|
|
|
|
return manaCurveMap;
|
|
}
|
|
|
|
void ManaCurveWidget::updateDisplay()
|
|
{
|
|
// Clear the layout first
|
|
if (barLayout != nullptr) {
|
|
QLayoutItem *item;
|
|
while ((item = barLayout->takeAt(0)) != nullptr) {
|
|
item->widget()->deleteLater();
|
|
delete item;
|
|
}
|
|
}
|
|
|
|
int highestEntry = 0;
|
|
for (const auto &entry : manaCurveMap) {
|
|
if (entry.second > highestEntry) {
|
|
highestEntry = entry.second;
|
|
}
|
|
}
|
|
|
|
// Convert unordered_map to ordered map to ensure sorting by CMC
|
|
std::map<int, int> sortedManaCurve(manaCurveMap.begin(), manaCurveMap.end());
|
|
|
|
// Add new widgets to the layout in sorted order
|
|
for (const auto &entry : sortedManaCurve) {
|
|
BarWidget *barWidget =
|
|
new BarWidget(QString::number(entry.first), entry.second, highestEntry, QColor(122, 122, 122), this);
|
|
barLayout->addWidget(barWidget);
|
|
}
|
|
|
|
update(); // Update the widget display
|
|
}
|