git.maemo.org Git - scorecard/blob - src/main-window.cpp

   1 #include <QtGui>
   2 #include <QDirModel>
   3 #include <QTreeView>
   4 #include <QListView>
   5 #include <QStandardItemModel>
   6
   7 #include "main-window.h"
   8 #include "score-dialog.h"
   9 #include "course-dialog.h"
  10 #include "tree-widget.h"
  11 #include "xml-parser.h"
  12 #include "xml-dom-parser.h"
  13
  14 QString appName("scorecard");
  15 QString topDir("/opt/scorecard");
  16 QString mmcDir("/media/mmc1/scorecard");
  17 QString dataDirName("/data/");
  18 QString dataDir;
  19 QString imgDir(topDir + "/pixmaps");
  20 QString scoreFileName("score.xml");
  21 QString scoreFile;
  22 QString clubFileName("club.xml");
  23 QString clubFile;
  24 QString logFile("/tmp/scorecard.log");
  25
  26 bool dateLessThan(const Score *s1, const Score *s2)
  27 {
  28   return (*s1) < (*s2);
  29 }
  30
  31 MainWindow::MainWindow(QMainWindow *parent): QMainWindow(parent)
  32 {
  33   resize(800, 480);
  34
  35   loadSettings();
  36
  37   QWidget *centralWidget = new QWidget(this);
  38
  39   setCentralWidget(centralWidget);
  40
  41   loadScoreFile(scoreFile, scoreList);
  42   loadClubFile(clubFile, clubList);
  43
  44   // Sort the scores based on dates
  45   qSort(scoreList.begin(), scoreList.end(), dateLessThan);
  46
  47   createTableView(scoreList, clubList);
  48   //createTreeView(scoreList, parent);
  49   createStatusBar();
  50
  51   createLayout(centralWidget);
  52
  53   createActions();
  54   createMenus();
  55 }
  56
  57 void MainWindow::loadSettings(void)
  58 {
  59   bool external = false;
  60
  61 #ifndef Q_WS_HILDON
  62   topDir = ".";
  63 #endif
  64
  65   QDir mmc(mmcDir);
  66   if (mmc.exists())
  67     external = true;
  68
  69   if (external) {
  70     dataDir = mmcDir + dataDirName;
  71   }
  72   else {
  73     dataDir = topDir + dataDirName;
  74   }
  75   scoreFile = dataDir + scoreFileName;
  76   clubFile = dataDir + clubFileName;
  77
  78   QDir dir(dataDir);
  79   if (!dir.exists())
  80     if (!dir.mkdir(dataDir)) {
  81       // TODO: mkdir does not work...
  82       qDebug() << "Unable to create: " + dataDir;
  83       return;
  84     }
  85   qDebug() << "Data is at: " + dataDir;
  86 }
  87
  88 void MainWindow::createLayout(QWidget *parent)
  89 {
  90   buttonLayout = new QVBoxLayout;
  91   //labelLayout->addStretch();
  92   buttonLayout->addWidget(nextButton);
  93   buttonLayout->addWidget(prevButton);
  94   buttonLayout->addWidget(lastButton);
  95   buttonLayout->addWidget(firstButton);
  96
  97   tableLayout = new QVBoxLayout;
  98   tableLayout->addWidget(table);
  99
 100   QHBoxLayout *mainLayout = new QHBoxLayout(parent);
 101   mainLayout->addLayout(tableLayout);
 102   mainLayout->addLayout(buttonLayout);
 103   setLayout(mainLayout);
 104 }
 105
 106 // Setup 'score' tab view
 107 void MainWindow::createTableView(QList<Score *> &scoreList, QList <Club *> &clubList)
 108 {
 109   table = new QTableView;
 110
 111   nextButton = new QPushButton(tr(">"));
 112   prevButton = new QPushButton(tr("<"));
 113   firstButton = new QPushButton(tr("<<"));
 114   lastButton = new QPushButton(tr(">>"));
 115
 116   connect(nextButton, SIGNAL(clicked()), this, SLOT(nextButtonClicked()));
 117   connect(prevButton, SIGNAL(clicked()), this, SLOT(prevButtonClicked()));
 118   connect(firstButton, SIGNAL(clicked()), this, SLOT(firstButtonClicked()));
 119   connect(lastButton, SIGNAL(clicked()), this, SLOT(lastButtonClicked()));
 120
 121   scoreTableModel = new ScoreTableModel();
 122   scoreTableModel->setScore(scoreList);
 123   scoreTableModel->setClub(clubList);
 124
 125   table->showGrid();
 126
 127   table->setModel(scoreTableModel);
 128
 129   // Fill out all the space with the tables
 130   table->horizontalHeader()->setResizeMode(QHeaderView::Stretch);
 131   table->verticalHeader()->setResizeMode(QHeaderView::Stretch);
 132   table->horizontalHeader()->hide();
 133 }
 134
 135 // When selection down in 'stat' view, this is called.
 136 void MainWindow::updateTreeView(const QModelIndex & index)
 137 {
 138   QString scope("Scope");
 139   QString count("Rounds");
 140   QString scoreAvg("Score (avg.)");
 141   QString scoreBest("Score (best)");
 142   QString score("Total");
 143   QString scoreIn("Total in");
 144   QString scoreOut("Total out");
 145
 146   QVariant str = scoreTreeModel->data(index, Qt::DisplayRole);
 147   QVariant type = scoreTreeModel->data(index, ScoreTreeModel::Type);
 148
 149   qDebug() << "update(" << index.row() << "/" << index.column() << "):" << str << type;
 150
 151   tableModel->setData(tableModel->index(0, 0, QModelIndex()), scope);
 152   tableModel->setData(tableModel->index(0, 1, QModelIndex()), str);
 153
 154   if (type == TreeItem::TypeDate) {
 155     tableModel->setData(tableModel->index(1, 0, QModelIndex()), count);
 156     tableModel->setData(tableModel->index(2, 0, QModelIndex()), scoreAvg);
 157     tableModel->setData(tableModel->index(3, 0, QModelIndex()), scoreBest);
 158   }
 159   else if (type == TreeItem::TypeScore) {
 160     QVariant value = scoreTreeModel->data(index, ScoreTreeModel::Total);
 161     tableModel->setData(tableModel->index(1, 0, QModelIndex()), score);
 162     tableModel->setData(tableModel->index(1, 1, QModelIndex()), value);
 163
 164     value = scoreTreeModel->data(index, ScoreTreeModel::TotalOut);
 165     tableModel->setData(tableModel->index(2, 0, QModelIndex()), scoreOut);
 166     tableModel->setData(tableModel->index(2, 1, QModelIndex()), value);
 167
 168     value = scoreTreeModel->data(index, ScoreTreeModel::TotalIn);
 169     tableModel->setData(tableModel->index(3, 0, QModelIndex()), scoreIn);
 170     tableModel->setData(tableModel->index(3, 1, QModelIndex()), value);
 171   }
 172 }
 173
 174 void MainWindow::createStatusBar()
 175 {
 176   updateStatusBar();
 177 }
 178
 179 void MainWindow::createActions()
 180 {
 181   newScoreAct = new QAction(tr("New Score"), this);
 182   connect(newScoreAct, SIGNAL(triggered()), this, SLOT(newScore()));
 183
 184   newCourseAct = new QAction(tr("New Course"), this);
 185   connect(newCourseAct, SIGNAL(triggered()), this, SLOT(newCourse()));
 186
 187   editScoreAct = new QAction(tr("Edit Score"), this);
 188   connect(editScoreAct, SIGNAL(triggered()), this, SLOT(editScore()));
 189
 190 #if 0
 191   viewScoreAct = new QAction(tr("&View Scores"), this);
 192   connect(viewScoreAct, SIGNAL(triggered()), this, SLOT(viewScore()));
 193
 194   viewCourseAct = new QAction(tr("&View Courses"), this);
 195   connect(viewCourseAct, SIGNAL(triggered()), this, SLOT(viewCourse()));
 196
 197   viewStatisticAct = new QAction(tr("&View Statistics"), this);
 198   connect(viewStatisticAct, SIGNAL(triggered()), this, SLOT(viewStatistic()));
 199 #endif
 200 }
 201
 202 void MainWindow::createMenus()
 203 {
 204   menu = menuBar()->addMenu(tr("fremantle"));
 205 #if 0
 206   menu->addAction(viewScoreAct);
 207   menu->addAction(viewCourseAct);
 208   menu->addAction(viewStatisticAct);
 209 #endif
 210   menu->addAction(newScoreAct);
 211   menu->addAction(newCourseAct);
 212   menu->addAction(editScoreAct);
 213 }
 214
 215 void MainWindow::updateStatusBar()
 216 {
 217   setWindowTitle(scoreTableModel->getInfoText());
 218 }
 219
 220 void MainWindow::firstButtonClicked()
 221 {
 222   scoreTableModel->first();
 223   updateStatusBar();
 224 }
 225
 226 void MainWindow::lastButtonClicked()
 227 {
 228   scoreTableModel->last();
 229   updateStatusBar();
 230 }
 231
 232 void MainWindow::nextButtonClicked()
 233 {
 234   scoreTableModel->next();
 235   updateStatusBar();
 236 }
 237
 238 void MainWindow::prevButtonClicked()
 239 {
 240   scoreTableModel->prev();
 241   updateStatusBar();
 242 }
 243
 244 // FIXME: dup code from table-model.cpp
 245 Club *MainWindow::findClub(QString &name)
 246 {
 247   QListIterator<Club *> i(clubList);
 248   Club *c;
 249
 250   while (i.hasNext()) {
 251     c = i.next();
 252     if (c->getName() == name)
 253       return c;
 254   }
 255   return 0;
 256 }
 257
 258 void MainWindow::newCourse()
 259 {
 260   CourseSelectDialog *selectDialog = new CourseSelectDialog(this);
 261
 262   int result = selectDialog->exec();
 263   if (result) {
 264     QString clubName;
 265     QString courseName;
 266     QString date;
 267
 268     selectDialog->results(clubName, courseName);
 269
 270     CourseDialog *courseDialog = new CourseDialog(this);
 271
 272     QString title = "New Course: " + clubName + "," + courseName;
 273     courseDialog->setWindowTitle(title);
 274
 275     int result = courseDialog->exec();
 276     if (result) {
 277       QVector<QString> par(18);
 278       QVector<QString> hcp(18);
 279       QVector<QString> len(18);
 280
 281       courseDialog->results(par, hcp, len);
 282
 283       Course *course = 0;
 284       Club *club = findClub(clubName);
 285       if (club) {
 286         course = club->getCourse(courseName);
 287         if (course) {
 288           qDebug() << "Error: club/course already in the database";
 289           return;
 290         }
 291         else {
 292           course = new Course(courseName, par, hcp);
 293           club->addCourse(course);
 294         }
 295       }
 296       else {
 297         club = new Club(clubName);
 298         course = new Course(courseName, par, hcp);
 299         club->addCourse(course);
 300         clubList << club;
 301       }
 302       saveClubFile(clubFile, clubList);
 303
 304       // TODO: does this really work? No mem leaks?
 305       scoreTableModel->setClub(clubList);
 306
 307     }
 308   }
 309 }
 310
 311 void MainWindow::newScore()
 312 {
 313   SelectDialog *selectDialog = new SelectDialog(this);
 314
 315   selectDialog->init(clubList);
 316
 317   int result = selectDialog->exec();
 318   if (result) {
 319     QString clubName;
 320     QString courseName;
 321     QString date;
 322
 323     selectDialog->results(clubName, courseName, date);
 324
 325     ScoreDialog *scoreDialog = new ScoreDialog(this);
 326     QString title = "New Score: " + courseName + ", " + date;
 327     scoreDialog->setWindowTitle(title);
 328
 329     Club *club = findClub(clubName);
 330     if (!club) {
 331       qDebug() << "Error: no such club: " << clubName;
 332       return;
 333     }
 334     Course *course = club->getCourse(courseName);
 335     if (!course) {
 336       qDebug() << "Error: no such course: " << courseName;
 337       return;
 338     }
 339
 340     scoreDialog->init(course);
 341
 342     result = scoreDialog->exec();
 343
 344     if (result) {
 345       QVector<QString> scores(18);
 346
 347       scoreDialog->results(scores);
 348       Score *score = new Score(scores, clubName, courseName, date);
 349       scoreList << score;
 350
 351       // Sort the scores based on dates
 352       qSort(scoreList.begin(), scoreList.end(), dateLessThan);
 353       // Save it
 354       saveScoreFile(scoreFile, scoreList);
 355
 356       // TODO: does this really work? No mem leaks?
 357       scoreTableModel->setScore(scoreList, score);
 358       updateStatusBar();
 359     }
 360   }
 361 }
 362
 363 void MainWindow::editScore()
 364 {
 365   Course *course = scoreTableModel->getCourse();
 366   Score *score = scoreTableModel->getScore();
 367   QString date = score->getDate();
 368
 369   ScoreDialog *scoreDialog = new ScoreDialog(this);
 370
 371   QString title = "Edit Score: " + course->getName() + ", " + date;
 372   scoreDialog->setWindowTitle(title);
 373
 374   scoreDialog->init(course, score);
 375
 376   int result = scoreDialog->exec();
 377
 378   if (result) {
 379     QVector<QString> scores(18);
 380
 381     scoreDialog->results(scores);
 382
 383     score->update(scores);
 384
 385     // Sort the scores based on dates
 386     qSort(scoreList.begin(), scoreList.end(), dateLessThan);
 387     // Save it
 388     saveScoreFile(scoreFile, scoreList);
 389
 390     // TODO: does this really work? No mem leaks?
 391     scoreTableModel->setScore(scoreList, score);
 392     updateStatusBar();
 393   }
 394 }
 395
 396 void MainWindow::loadScoreFile(QString &fileName, QList<Score *> &list)
 397 {
 398   ScoreXmlHandler handler(list);
 399
 400   if (handler.parse(fileName))
 401     qDebug() << "File loaded: " << fileName << " entries: " << list.size();
 402 }
 403
 404 void MainWindow::saveScoreFile(QString &fileName, QList<Score *> &list)
 405 {
 406   ScoreXmlHandler handler(list);
 407
 408   if (handler.save(fileName))
 409     // TODO: banner
 410     qDebug() << "File saved: " << fileName << " entries: " << list.size();
 411   else
 412     qDebug() << "Unable to save: " << fileName;
 413 }
 414
 415 void MainWindow::loadClubFile(QString &fileName, QList<Club *> &list)
 416 {
 417   ClubXmlHandler handler(list);
 418
 419   if (handler.parse(fileName))
 420     qDebug() << "File loaded: " << fileName << " entries: " << list.size();
 421 }
 422
 423 void MainWindow::saveClubFile(QString &fileName, QList<Club *> &list)
 424 {
 425   ClubXmlHandler handler(list);
 426
 427   if (handler.save(fileName))
 428     // TODO: banner
 429     qDebug() << "File saved: " << fileName << " entries: " << list.size();
 430   else
 431     qDebug() << "Unable to save: " << fileName;
 432
 433 }