RecipeDB/model/recipe/ingredients/ingredientlistmodel.cpp

#include "model/recipe/ingredients/ingredientlistmodel.h"

IngredientListModel::IngredientListModel(){
	this->ingredients = vector<Ingredient>();
}

int IngredientListModel::rowCount(const QModelIndex &parent) const{
	Q_UNUSED(parent);
    return this->ingredients.size();
}

QVariant IngredientListModel::data(const QModelIndex &index, int role) const{
    int row = index.row();
	Ingredient i = this->ingredients[row];

	string displayStr = i.getContent();

    switch(role){
    case Qt::DisplayRole:
		return QString::fromStdString(displayStr);
    }

    return QVariant();
}

void IngredientListModel::setIngredients(vector<Ingredient> ingredients){
    this->ingredients = ingredients;
    QModelIndex index = createIndex(0, 0);
    QModelIndex bottomIndex = createIndex(ingredients.size()-1, 0);
	emit dataChanged(index, bottomIndex);
}

bool IngredientListModel::addIngredient(Ingredient i){
	//Add only if it doesn't exist already.
	for (Ingredient ing : this->ingredients){
		if (!ing.getContent().compare(i.getContent())){
			return false;
		}
	}
	//The ingredient doesn't exist already, so we'll add it.
	this->ingredients.push_back(i);
	QModelIndex index = createIndex(this->ingredients.size()-1, 0);
	QModelIndex bottomIndex = createIndex(this->ingredients.size()-1, 0);
	emit dataChanged(index, bottomIndex);
	return true;
}

void IngredientListModel::deleteIngredient(int index){
	this->ingredients.erase(this->ingredients.begin() + index);
	emit dataChanged(createIndex(0, 0), createIndex(this->ingredients.size()-1, 0));
}

vector<Ingredient> IngredientListModel::getIngredients(){
	return this->ingredients;
}