dialog.cpp Example File

sql/masterdetail/dialog.cpp
 /****************************************************************************
 **
 ** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies).
 ** All rights reserved.
 ** Contact: Nokia Corporation (qt-info@nokia.com)
 **
 ** This file is part of the examples of the Qt Toolkit.
 **
 ** $QT_BEGIN_LICENSE:BSD$
 ** You may use this file under the terms of the BSD license as follows:
 **
 ** "Redistribution and use in source and binary forms, with or without
 ** modification, are permitted provided that the following conditions are
 ** met:
 **   * Redistributions of source code must retain the above copyright
 **     notice, this list of conditions and the following disclaimer.
 **   * Redistributions in binary form must reproduce the above copyright
 **     notice, this list of conditions and the following disclaimer in
 **     the documentation and/or other materials provided with the
 **     distribution.
 **   * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
 **     the names of its contributors may be used to endorse or promote
 **     products derived from this software without specific prior written
 **     permission.
 **
 ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
 ** $QT_END_LICENSE$
 **
 ****************************************************************************/

 #include "dialog.h"

 int uniqueAlbumId;
 int uniqueArtistId;

 Dialog::Dialog(QSqlRelationalTableModel *albums, QDomDocument details,
                QFile *output, QWidget *parent)
      : QDialog(parent)
 {
     model = albums;
     albumDetails = details;
     outputFile = output;

     QGroupBox *inputWidgetBox = createInputWidgets();
     QDialogButtonBox *buttonBox = createButtons();

     QVBoxLayout *layout = new QVBoxLayout;
     layout->addWidget(inputWidgetBox);
     layout->addWidget(buttonBox);
     setLayout(layout);

     setWindowTitle(tr("Add Album"));
 }

 void Dialog::submit()
 {
     QString artist = artistEditor->text();
     QString title = titleEditor->text();

     if (artist.isEmpty() || title.isEmpty()) {
         QString message(tr("Please provide both the name of the artist "
                            "and the title of the album."));
         QMessageBox::information(this, tr("Add Album"), message);
     } else {
         int artistId = findArtistId(artist);
         int albumId = addNewAlbum(title, artistId);

         QStringList tracks;
         tracks = tracksEditor->text().split(',', QString::SkipEmptyParts);
         addTracks(albumId, tracks);

         increaseAlbumCount(indexOfArtist(artist));
         accept();
     }
 }

 int Dialog::findArtistId(const QString &artist)
 {
     QSqlTableModel *artistModel = model->relationModel(2);
     int row = 0;

     while (row < artistModel->rowCount()) {
         QSqlRecord record = artistModel->record(row);
         if (record.value("artist") == artist)
             return record.value("id").toInt();
         else
             row++;
     }
     return addNewArtist(artist);
 }

 int Dialog::addNewArtist(const QString &name)
 {
     QSqlTableModel *artistModel = model->relationModel(2);
     QSqlRecord record;

     int id = generateArtistId();

     QSqlField f1("id", QVariant::Int);
     QSqlField f2("artist", QVariant::String);
     QSqlField f3("albumcount", QVariant::Int);

     f1.setValue(QVariant(id));
     f2.setValue(QVariant(name));
     f3.setValue(QVariant(0));
     record.append(f1);
     record.append(f2);
     record.append(f3);

     artistModel->insertRecord(-1, record);
     return id;
 }

 int Dialog::addNewAlbum(const QString &title, int artistId)
 {
     int id = generateAlbumId();
     QSqlRecord record;

     QSqlField f1("albumid", QVariant::Int);
     QSqlField f2("title", QVariant::String);
     QSqlField f3("artistid", QVariant::Int);
     QSqlField f4("year", QVariant::Int);

     f1.setValue(QVariant(id));
     f2.setValue(QVariant(title));
     f3.setValue(QVariant(artistId));
     f4.setValue(QVariant(yearEditor->value()));
     record.append(f1);
     record.append(f2);
     record.append(f3);
     record.append(f4);

     model->insertRecord(-1, record);
     return id;
 }

 void Dialog::addTracks(int albumId, QStringList tracks)
 {
     QDomElement albumNode = albumDetails.createElement("album");
     albumNode.setAttribute("id", albumId);

     for (int i = 0; i < tracks.count(); i++) {
         QString trackNumber = QString::number(i);
         if (i < 10)
             trackNumber.prepend("0");

         QDomText textNode = albumDetails.createTextNode(tracks.at(i));

         QDomElement trackNode = albumDetails.createElement("track");
         trackNode.setAttribute("number", trackNumber);
         trackNode.appendChild(textNode);

         albumNode.appendChild(trackNode);
     }

     QDomNodeList archive = albumDetails.elementsByTagName("archive");
     archive.item(0).appendChild(albumNode);

 /*
     The following code is commented out since the example uses an in
     memory database, i.e., altering the XML file will bring the data
     out of sync.

     if (!outputFile->open(QIODevice::WriteOnly)) {
         return;
     } else {
         QTextStream stream(outputFile);
         archive.item(0).save(stream, 4);
         outputFile->close();
     }
 */
 }

 void Dialog::increaseAlbumCount(QModelIndex artistIndex)
 {
     QSqlTableModel *artistModel = model->relationModel(2);

     QModelIndex albumCountIndex;
     albumCountIndex = artistIndex.sibling(artistIndex.row(), 2);

     int albumCount = albumCountIndex.data().toInt();
     artistModel->setData(albumCountIndex, QVariant(albumCount + 1));
 }

 void Dialog::revert()
 {
     artistEditor->clear();
     titleEditor->clear();
     yearEditor->setValue(QDate::currentDate().year());
     tracksEditor->clear();
 }

 QGroupBox *Dialog::createInputWidgets()
 {
     QGroupBox *box = new QGroupBox(tr("Add Album"));

     QLabel *artistLabel = new QLabel(tr("Artist:"));
     QLabel *titleLabel = new QLabel(tr("Title:"));
     QLabel *yearLabel = new QLabel(tr("Year:"));
     QLabel *tracksLabel = new QLabel(tr("Tracks (separated by comma):"));

     artistEditor = new QLineEdit;
     titleEditor = new QLineEdit;

     yearEditor = new QSpinBox;
     yearEditor->setMinimum(1900);
     yearEditor->setMaximum(QDate::currentDate().year());
     yearEditor->setValue(yearEditor->maximum());
     yearEditor->setReadOnly(false);

     tracksEditor = new QLineEdit;

     QGridLayout *layout = new QGridLayout;
     layout->addWidget(artistLabel, 0, 0);
     layout->addWidget(artistEditor, 0, 1);
     layout->addWidget(titleLabel, 1, 0);
     layout->addWidget(titleEditor, 1, 1);
     layout->addWidget(yearLabel, 2, 0);
     layout->addWidget(yearEditor, 2, 1);
     layout->addWidget(tracksLabel, 3, 0, 1, 2);
     layout->addWidget(tracksEditor, 4, 0, 1, 2);
     box->setLayout(layout);

     return box;
 }

 QDialogButtonBox *Dialog::createButtons()
 {
     QPushButton *closeButton = new QPushButton(tr("&Close"));
     QPushButton *revertButton = new QPushButton(tr("&Revert"));
     QPushButton *submitButton = new QPushButton(tr("&Submit"));

     closeButton->setDefault(true);

     connect(closeButton, SIGNAL(clicked()), this, SLOT(close()));
     connect(revertButton, SIGNAL(clicked()), this, SLOT(revert()));
     connect(submitButton, SIGNAL(clicked()), this, SLOT(submit()));

     QDialogButtonBox *buttonBox = new QDialogButtonBox;
     buttonBox->addButton(submitButton, QDialogButtonBox::ResetRole);
     buttonBox->addButton(revertButton, QDialogButtonBox::ResetRole);
     buttonBox->addButton(closeButton, QDialogButtonBox::RejectRole);

     return buttonBox;
 }

 QModelIndex Dialog::indexOfArtist(const QString &artist)
 {
     QSqlTableModel *artistModel = model->relationModel(2);

     for (int i = 0; i < artistModel->rowCount(); i++) {
         QSqlRecord record =  artistModel->record(i);
         if (record.value("artist") == artist)
             return artistModel->index(i, 1);
     }

     return QModelIndex();
 }

 int Dialog::generateArtistId()
 {
     uniqueArtistId += 1;
     return uniqueArtistId;
 }

 int Dialog::generateAlbumId()
 {
     uniqueAlbumId += 1;
     return uniqueAlbumId;
 }