src/InventoryFermentables.cpp

/**
 * InventoryFermentables.cpp is part of bmsapp.
 *
 * bmsapp is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * bmsapp is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "InventoryFermentables.h"
#include "EditFermentable.h"
#include "MainWindow.h"
#include "config.h"
#include "global.h"
#include "Utils.h"


InventoryFermentables::InventoryFermentables(QWidget *parent) : QDialog(parent)
{
    qDebug() << "InventoryFermentables start";

    gridLayout = new QGridLayout(this);
    gridLayout->setObjectName(QString::fromUtf8("gridLayout"));
    tableFermentables = new QTableWidget(this);
    tableFermentables->setObjectName(QString::fromUtf8("tableFermentables"));
    tableFermentables->setEnabled(true);
    QSizePolicy sizePolicy(QSizePolicy::Fixed, QSizePolicy::Expanding);
    sizePolicy.setHorizontalStretch(0);
    sizePolicy.setVerticalStretch(0);
    //sizePolicy.setHeightForWidth(tableFermentables->sizePolicy().hasHeightForWidth());
    tableFermentables->setSizePolicy(sizePolicy);
    tableFermentables->setMinimumSize(QSize(1074, 0));
    gridLayout->addWidget(tableFermentables, 0, 0, 1, 1);

    groupBox = new QGroupBox(this);
    groupBox->setObjectName(QString::fromUtf8("groupBox"));
    groupBox->setEnabled(true);
    groupBox->setFlat(false);
    horizontalLayout = new QHBoxLayout(groupBox);
    horizontalLayout->setSpacing(6);
    horizontalLayout->setObjectName(QString::fromUtf8("horizontalLayout"));
    horizontalLayout->setContentsMargins(0, 0, 0, 0);

    quitButton = new QPushButton(groupBox);
    quitButton->setObjectName(QString::fromUtf8("quitButton"));
    quitButton->setMinimumSize(QSize(80, 24));
    quitButton->setText(tr("Quit"));
    QIcon icon;
    icon.addFile(QString::fromUtf8(":icons/silk/door_out.png"), QSize(), QIcon::Normal, QIcon::Off);
    quitButton->setIcon(icon);
    horizontalLayout->addWidget(quitButton, 0, Qt::AlignLeft);

    exportButton = new QPushButton(groupBox);
    exportButton->setObjectName(QString::fromUtf8("exportButton"));
    exportButton->setMinimumSize(QSize(80, 24));
    exportButton->setText(tr("Export"));
    QIcon icon1;
    icon1.addFile(QString::fromUtf8(":/icons/silk/database_save.png"), QSize(), QIcon::Normal, QIcon::Off);
    exportButton->setIcon(icon1);
    horizontalLayout->addWidget(exportButton, 0, Qt::AlignRight);

    importButton = new QPushButton(groupBox);
    importButton->setObjectName(QString::fromUtf8("importButton"));
    importButton->setMinimumSize(QSize(80, 24));
    importButton->setText(tr("Import"));
    QIcon icon2;
    icon2.addFile(QString::fromUtf8(":/icons/silk/database_add.png"), QSize(), QIcon::Normal, QIcon::Off);
    importButton->setIcon(icon2);
    horizontalLayout->addWidget(importButton, 0, Qt::AlignRight);

    insertButton = new QPushButton(groupBox);
    insertButton->setObjectName(QString::fromUtf8("insertButton"));
    insertButton->setMinimumSize(QSize(80, 24));
    insertButton->setText(tr("New"));
    QIcon icon3;
    icon3.addFile(QString::fromUtf8(":icons/silk/table_row_insert.png"), QSize(), QIcon::Normal, QIcon::Off);
    insertButton->setIcon(icon3);
    horizontalLayout->addWidget(insertButton, 0, Qt::AlignRight);
    gridLayout->addWidget(groupBox, 1, 0, 1, 1);

    connect(quitButton, SIGNAL(clicked()), parent, SLOT(fromInventoryFermentables()));
    connect(insertButton, SIGNAL(clicked()), this, SLOT(on_insertButton_clicked()));
    connect(exportButton, SIGNAL(clicked()), this, SLOT(on_exportButton_clicked()));
//    connect(importButton, SIGNAL(clicked()), this, SLOT(on_importButton_clicked()));
    connect(this, SIGNAL(setStatus(QString)), parent, SLOT(statusMsg(QString)));
    emit refreshTable();
}


void InventoryFermentables::refreshTable()
{
    QString w;

    qDebug() << "InventoryFermentables reload";

    QSqlQuery query("SELECT * FROM inventory_fermentables ORDER BY supplier,name");
    const QStringList labels({tr("Origin"), tr("Supplier"), tr("Name"), tr("Type"), tr("Grain"), tr("Color"), tr("Yield"), tr("Stock"), tr("Edit")});

    /* origin supplier name type graintype color yield inventory Edit */
    this->tableFermentables->setColumnCount(9);
    this->tableFermentables->setColumnWidth(0, 130);	/* Origin	*/
    this->tableFermentables->setColumnWidth(1, 170);	/* Supplier	*/
    this->tableFermentables->setColumnWidth(2, 250);	/* Name		*/
    this->tableFermentables->setColumnWidth(3,  90);	/* Type		*/
    this->tableFermentables->setColumnWidth(4,  90);	/* Graintype	*/
    this->tableFermentables->setColumnWidth(5,  80);	/* Color	*/
    this->tableFermentables->setColumnWidth(6,  80);	/* Yield	*/
    this->tableFermentables->setColumnWidth(7,  80);	/* Stock	*/
    this->tableFermentables->setColumnWidth(8,  80);	/* Edit button	*/
    this->tableFermentables->setRowCount(query.size());
    this->tableFermentables->setHorizontalHeaderLabels(labels);
    this->tableFermentables->verticalHeader()->hide();

    QTableWidgetItem *rightitem = new QTableWidgetItem();
    rightitem->setTextAlignment(Qt::AlignRight);

    query.first();
    for (int ridx = 0 ; ridx < query.size() ; ridx++ ) {
	this->tableFermentables->setItem(ridx, 0, new QTableWidgetItem(query.value(6).toString()));
	this->tableFermentables->setItem(ridx, 1, new QTableWidgetItem(query.value(7).toString()));
	this->tableFermentables->setItem(ridx, 2, new QTableWidgetItem(query.value(1).toString()));
	this->tableFermentables->setItem(ridx, 3, new QTableWidgetItem(fermentable_types[query.value(2).toInt()]));
	this->tableFermentables->setItem(ridx, 4, new QTableWidgetItem(fermentable_graintypes[query.value(20).toInt()]));
	w = QString("%1 EBC").arg(query.value(4).toDouble(), 1, 'f', 0, '0' );
	QTableWidgetItem *item = new QTableWidgetItem(w);
	item->setTextAlignment(Qt::AlignRight|Qt::AlignVCenter);
        this->tableFermentables->setItem(ridx, 5, item);
	w = QString("%1 %").arg(query.value(3).toFloat(), 3, 'f', 1, '0' );
	item = new QTableWidgetItem(w);
	item->setTextAlignment(Qt::AlignRight|Qt::AlignVCenter);
	this->tableFermentables->setItem(ridx, 6, item);
	w = QString("");
	if (query.value(21).toDouble() > 0) {
	  if (query.value(21).toDouble() < 1.000) {
	    w = QString("%1 gr").arg(query.value(21).toDouble() * 1000.0, 1, 'f', 0, '0' );
	  } else {
            w = QString("%1 kg").arg(query.value(21).toDouble(), 4, 'f', 3, '0' );
	  }
	}
	item = new QTableWidgetItem(w);
	item->setTextAlignment(Qt::AlignRight|Qt::AlignVCenter);
	this->tableFermentables->setItem(ridx, 7, item);

	/* Add the Edit button */
	QWidget* pWidget = new QWidget();
	QPushButton* btn_edit = new QPushButton();
	btn_edit->setObjectName(QString("%1").arg(query.value(0).toString()));	/* Send record with the button */
	btn_edit->setText(tr("Edit"));
	connect(btn_edit, SIGNAL(clicked()), this, SLOT(on_editButton_clicked()));
	QHBoxLayout* pLayout = new QHBoxLayout(pWidget);
	pLayout->addWidget(btn_edit);
	pLayout->setContentsMargins(5, 0, 5, 0);
	pWidget->setLayout(pLayout);
	this->tableFermentables->setCellWidget(ridx, 8, pWidget);
	query.next();
    }
    emit setStatus(QString(tr("Total items: %1")).arg(query.size()));
}


InventoryFermentables::~InventoryFermentables()
{
    qDebug() << "InventoryFermentables done";
}


void InventoryFermentables::edit(int recno)
{
    qDebug() << "InventoryFermentables edit:" << recno;

    EditFermentable dialog(recno, this);
    /* Signal from editor if a refresh is needed */
    connect(&dialog, SIGNAL(entry_changed()), this, SLOT(refreshTable()));
    dialog.setModal(true);
    dialog.exec();
}


void InventoryFermentables::on_editButton_clicked()
{
    QPushButton *pb = qobject_cast<QPushButton *>(QObject::sender());
    int recno = pb->objectName().toInt();
    edit(recno);
}


void InventoryFermentables::on_insertButton_clicked()
{
    edit(-1);
}


void InventoryFermentables::on_exportButton_clicked()
{
    qDebug() << Q_FUNC_INFO;

    QSqlQuery query("SELECT * FROM inventory_fermentables ORDER BY supplier,name");
    const QStringList types({"Grain", "Sugar", "Extract", "Dry extract", "Adjunct"});
    const QStringList graintypes({"Base", "Roast", "Crystal", "Kilned", "Sour Malt", "Special", "No malt"});
    const QStringList added({"Mash", "Boil", "Fermentation", "Lagering", "Bottle"});

    QString fileName = QFileDialog::getSaveFileName(this, tr("Save File"), QDir::homePath() + "/fermentables.xml", tr("Files (*.xml)"));
    if (fileName == 0) {
	QMessageBox::warning(this, tr("Save File"), tr("No XML file selected."));
        return;
    }

    QFile file(fileName);
    file.open(QIODevice::WriteOnly);

    QXmlStreamWriter *xmlWriter = new QXmlStreamWriter(&file);
    xmlWriter->writeStartDocument();
    xmlWriter->setAutoFormatting(true);
    xmlWriter->setAutoFormattingIndent(1);
    xmlWriter->writeStartElement("FERMENTABLES");

    query.first();
    for (int i = 0 ; i < query.size() ; i++ ) {
	xmlWriter->writeStartElement("FERMENTABLE");
	xmlWriter->writeTextElement("VERSION", "1");
	xmlWriter->writeTextElement("NAME", query.value(1).toString());
	xmlWriter->writeTextElement("NOTES", query.value(8).toString());
	xmlWriter->writeTextElement("TYPE", types[query.value(2).toInt()]);
	xmlWriter->writeTextElement("YIELD", QString::number(query.value(3).toDouble(), 'f', 5));
	xmlWriter->writeTextElement("COLOR", QString::number(Utils::ebc_to_srm(query.value(4).toDouble()), 'f', 3) );
	xmlWriter->writeTextElement("ADD_AFTER_BOIL", query.value(5).toInt() ? "TRUE":"FALSE");
	xmlWriter->writeTextElement("ORIGIN", query.value(6).toString());
	xmlWriter->writeTextElement("SUPPLIER", query.value(7).toString());
	if (query.value(9).toDouble() > 0)
	    xmlWriter->writeTextElement("COARSE_FINE_DIFF", QString::number(query.value(9).toDouble(), 'f', 4));
	if (query.value(10).toDouble() > 0)
	    xmlWriter->writeTextElement("MOISTURE", QString::number(query.value(10).toDouble(), 'f', 4));
	if (query.value(11).toDouble() > 0)
	    xmlWriter->writeTextElement("DIASTATIC_POWER", QString::number(query.value(11).toDouble(), 'f', 4));
	if (query.value(12).toDouble() > 0)
	    xmlWriter->writeTextElement("PROTEIN", QString::number(query.value(12).toDouble(), 'f', 4));
	if (query.value(14).toDouble() > 0)
	    xmlWriter->writeTextElement("MAX_IN_BATCH", QString::number(query.value(14).toDouble(), 'f', 3));
	xmlWriter->writeTextElement("RECOMMEND_MASH", query.value(15).toInt() ? "TRUE":"FALSE");
	xmlWriter->writeTextElement("DISPLAY_COLOR", QString("%1 EBC").arg(query.value(4).toDouble(), 1, 'f', 0));
	xmlWriter->writeTextElement("GRAINTYPE", graintypes[query.value(20).toInt()]);
	xmlWriter->writeTextElement("ADDED", added[query.value(17).toInt()]);
	if (query.value(18).toDouble() > 0)
	    xmlWriter->writeTextElement("DI_pH", QString::number(query.value(18).toDouble(), 'f', 5));
	if (query.value(19).toDouble() > 0)
	    xmlWriter->writeTextElement("ACID_TO_pH_5.7", QString::number(query.value(19).toDouble(), 'f', 5));

	xmlWriter->writeEndElement();
	query.next();
    }

    xmlWriter->writeEndElement();
    xmlWriter->writeEndDocument();
    QMessageBox::information(this, tr("Save File"), tr("XML export ready"));

    file.close();
}