RecipePicker/RecipePicker.cpp at main · ligazetom/RecipePicker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include "RecipePicker.h"

#include "GeneralTypes.h"
#include "AddRecipeDialog.h"

#include <qmessagebox.h>


RecipePicker::RecipePicker(QWidget *parent)
    : QMainWindow(parent)
{
    ui.setupUi(this);
	SetupTableWidget();
}

RecipePicker::~RecipePicker()
{}

void RecipePicker::SetupTableWidget()
{
    //ui.tableWidgetRecipes->setEditTriggers(QAbstractItemView::NoEditTriggers);
    ui.tableWidgetRecipes->setColumnCount(TableColumns::NumberOfRecipeColumns);

    QStringList headerLabels;
    headerLabels.reserve(TableColumns::NumberOfRecipeColumns);

    for (int i = 0; i < TableColumns::NumberOfRecipeColumns; ++i)
    {
        headerLabels.push_back(TableColumns::RecipesColumnNames[i]);
    }

    ui.tableWidgetRecipes->setHorizontalHeaderLabels(headerLabels);
    ui.tableWidgetRecipes->resizeColumnsToContents();
}

void RecipePicker::AddNewRow(const std::vector<QTableWidgetItem*>& rowItems)
{
    const int numOfRows = ui.tableWidgetRecipes->rowCount();
    ui.tableWidgetRecipes->insertRow(numOfRows);

    int colCounter = 0;
    for (const auto& rowItem : rowItems)
    {
        ui.tableWidgetRecipes->setItem(numOfRows, colCounter, rowItem);
        ++colCounter;
    }
}

std::vector<QTableWidgetItem*> RecipePicker::CreateRow(const TableWidgetTypes::RecipeRow& row)
{
    std::vector<QTableWidgetItem*> items(TableColumns::NumberOfRecipeColumns);

    items[TableColumns::RecipeID] = new QTableWidgetItem();
    items[TableColumns::RecipeID]->setData(Qt::DisplayRole, row.RecipeID);
    items[TableColumns::RecipeName] = new QTableWidgetItem(row.RecipeName);
    items[TableColumns::RecipeMealTime] = new QTableWidgetItem(Recipe::RecipeMealtimeNames[row.RecipeMealtime]);
    items[TableColumns::RecipeMealTime]->setData(Qt::UserRole, row.RecipeMealtime);
    items[TableColumns::RecipeCategory] = new QTableWidgetItem(Recipe::RecipeCategoryNames[row.RecipeCategory]);
    items[TableColumns::RecipeCategory]->setData(Qt::UserRole, row.RecipeCategory);
    items[TableColumns::RecipeDifficulty] = new QTableWidgetItem(Recipe::RecipeDifficultyNames[row.RecipeDifficulty]);
    items[TableColumns::RecipeDifficulty]->setData(Qt::UserRole, row.RecipeDifficulty);

    return items;
}

TableWidgetTypes::RecipeRow RecipePicker::ExtractRow(int rowIndex) const
{
    TableWidgetTypes::RecipeRow row;

    row.RecipeID = ui.tableWidgetRecipes->item(rowIndex, TableColumns::RecipeID)->data(Qt::DisplayRole).toInt();
    row.RecipeName = ui.tableWidgetRecipes->item(rowIndex, TableColumns::RecipeName)->text();
    row.RecipeMealtime = ui.tableWidgetRecipes->item(rowIndex, TableColumns::RecipeMealTime)->data(Qt::UserRole).toInt();
    row.RecipeCategory = ui.tableWidgetRecipes->item(rowIndex, TableColumns::RecipeCategory)->data(Qt::UserRole).toInt();
    row.RecipeDifficulty = ui.tableWidgetRecipes->item(rowIndex, TableColumns::RecipeDifficulty)->data(Qt::UserRole).toInt();

    return row;
}

void RecipePicker::on_actionAddNewRecipe_triggered()
{
    AddRecipeDialog dlg(this);

    if (dlg.exec() == QDialog::Accepted)
    {
        TableWidgetTypes::RecipeRow row;

        row.RecipeID = 0;
        row.RecipeName = dlg.GetRecipeName();
        row.RecipeMealtime = dlg.GetRecipeMealtime();
        row.RecipeCategory = dlg.GetRecipeCategory();
        row.RecipeDifficulty = dlg.GetRecipeDifficulty();

        AddNewRow(CreateRow(row));
        ui.tableWidgetRecipes->resizeColumnsToContents();
    }
}