1 files changed, 181 insertions, 173 deletions

diff --git a/qt-ui/maintab.cpp b/qt-ui/maintab.cpp
index f9d36342f..42250b42c 100644
--- a/qt-ui/maintab.cpp
+++ b/qt-ui/maintab.cpp
@@ -12,17 +12,20 @@
 #include "divelistview.h"
 #include "modeldelegates.h"
 #include "globe.h"
+#include "completionmodels.h"
 
 #include <QLabel>
+#include <QCompleter>
 #include <QDebug>
 #include <QSet>
+#include <QTableView>
 #include <QSettings>
+#include <QPalette>
 
 MainTab::MainTab(QWidget *parent) : QTabWidget(parent),
 				    ui(new Ui::MainTab()),
 				    weightModel(new WeightModel()),
 				    cylindersModel(new CylindersModel()),
-				    currentDive(0),
 				    editMode(NONE)
 {
 	ui->setupUi(this);
@@ -52,7 +55,7 @@ MainTab::MainTab(QWidget *parent) : QTabWidget(parent),
 	ui->divemaster->installEventFilter(this);
 	ui->buddy->installEventFilter(this);
 	ui->suit->installEventFilter(this);
-	ui->notes->installEventFilter(this);
+	ui->notes->viewport()->installEventFilter(this);
 	ui->rating->installEventFilter(this);
 	ui->visibility->installEventFilter(this);
 
@@ -62,82 +65,51 @@ MainTab::MainTab(QWidget *parent) : QTabWidget(parent),
 		if (label)
 			label->setAlignment(Qt::AlignHCenter);
 	}
-
-	/*Thid couldn't be done on the ui file because element
-	is floating, instead of being fixed on the layout. */
-	QIcon plusIcon(":plus");
-	addCylinder = new QPushButton(plusIcon, QString(), ui->cylindersGroup);
-	addCylinder->setFlat(true);
-	addCylinder->setToolTip(tr("Add Cylinder"));
-	connect(addCylinder, SIGNAL(clicked(bool)), this, SLOT(addCylinder_clicked()));
-	addCylinder->setEnabled(false);
-	addWeight = new QPushButton(plusIcon, QString(), ui->weightGroup);
-	addWeight->setFlat(true);
-	addWeight->setToolTip(tr("Add Weight System"));
-	connect(addWeight, SIGNAL(clicked(bool)), this, SLOT(addWeight_clicked()));
-	addWeight->setEnabled(false);
-
-	connect(ui->cylinders, SIGNAL(clicked(QModelIndex)), ui->cylinders->model(), SLOT(remove(QModelIndex)));
-	connect(ui->cylinders, SIGNAL(clicked(QModelIndex)), this, SLOT(editCylinderWidget(QModelIndex)));
-	connect(ui->weights, SIGNAL(clicked(QModelIndex)), ui->weights->model(), SLOT(remove(QModelIndex)));
-	connect(ui->weights, SIGNAL(clicked(QModelIndex)), this, SLOT(editWeigthWidget(QModelIndex)));
-
-	QFontMetrics metrics(defaultModelFont());
-	QFontMetrics metrics2(font());
-
-	ui->cylinders->horizontalHeader()->setResizeMode(CylindersModel::REMOVE, QHeaderView::Fixed);
-	ui->cylinders->verticalHeader()->setDefaultSectionSize( metrics.height() +8 );
-	ui->cylinders->setItemDelegateForColumn(CylindersModel::TYPE, new TankInfoDelegate());
-
-	ui->weights->horizontalHeader()->setResizeMode (WeightModel::REMOVE , QHeaderView::Fixed);
-	ui->weights->verticalHeader()->setDefaultSectionSize( metrics.height() +8 );
-	ui->weights->setItemDelegateForColumn(WeightModel::TYPE, new WSInfoDelegate());
-
-	connect(this, SIGNAL(currentChanged(int)), this, SLOT(tabChanged(int)));
-	initialUiSetup();
-}
-
-// We need to manually position the 'plus' on cylinder and weight.
-void MainTab::resizeEvent(QResizeEvent* event)
-{
-	equipmentPlusUpdate();
-	QTabWidget::resizeEvent(event);
-}
-
-void MainTab::showEvent(QShowEvent* event)
-{
-	QTabWidget::showEvent(event);
-	equipmentPlusUpdate();
+	ui->cylinders->setTitle(tr("Cylinders"));
+	ui->cylinders->setBtnToolTip(tr("Add Cylinder"));
+	connect(ui->cylinders, SIGNAL(addButtonClicked()), this, SLOT(addCylinder_clicked()));
+
+	ui->weights->setTitle(tr("Weights"));
+	ui->weights->setBtnToolTip(tr("Add Weight System"));
+	connect(ui->weights, SIGNAL(addButtonClicked()), this, SLOT(addWeight_clicked()));
+
+	connect(ui->cylinders->view(), SIGNAL(clicked(QModelIndex)), this, SLOT(editCylinderWidget(QModelIndex)));
+	connect(ui->weights->view(), SIGNAL(clicked(QModelIndex)), this, SLOT(editWeigthWidget(QModelIndex)));
+
+	ui->cylinders->view()->setItemDelegateForColumn(CylindersModel::TYPE, new TankInfoDelegate());
+	ui->weights->view()->setItemDelegateForColumn(WeightModel::TYPE, new WSInfoDelegate());
+
+	completers.buddy = new QCompleter(BuddyCompletionModel::instance(), ui->buddy);
+	completers.divemaster = new QCompleter(DiveMasterCompletionModel::instance(), ui->divemaster);
+	completers.location = new QCompleter(LocationCompletionModel::instance(), ui->location);
+	completers.suit = new QCompleter(SuitCompletionModel::instance(), ui->suit);
+	ui->buddy->setCompleter(completers.buddy);
+	ui->divemaster->setCompleter(completers.divemaster);
+	ui->location->setCompleter(completers.location);
+	ui->suit->setCompleter(completers.suit);
 }
 
-void MainTab::tabChanged(int idx)
+void MainTab::enableEdition()
 {
-	/* if the current tab has become of index 1 (i.e. the equipment tab) call update
-	 * for the plus signs */
-	if (idx == 1)
-		equipmentPlusUpdate();
-}
+	if (ui->editAccept->isVisible() || !selected_dive)
+		return;
 
-void MainTab::equipmentPlusUpdate()
-{
-	if (ui->cylindersGroup->isVisible())
-		addCylinder->setGeometry(ui->cylindersGroup->contentsRect().width() - 30, 2, 24,24);
-	if (ui->weightGroup->isVisible())
-		addWeight->setGeometry(ui->weightGroup->contentsRect().width() - 30, 2, 24,24);
+	ui->editAccept->setChecked(true);
+	ui->editAccept->show();
+	ui->editReset->show();
+	on_editAccept_clicked(true);
 }
 
 bool MainTab::eventFilter(QObject* object, QEvent* event)
 {
-	if (event->type() == QEvent::FocusIn || event->type() == QEvent::MouseButtonPress) {
-		if (ui->editAccept->isVisible() || !currentDive)
-			return false;
-
-		ui->editAccept->setChecked(true);
-		ui->editAccept->show();
-		ui->editReset->show();
-		on_editAccept_clicked(true);
+	if (event->type() == QEvent::FocusIn && (object == ui->rating || object == ui->visibility)){
+		enableEdition();
 	}
-	return false;
+
+	if (event->type() == QEvent::MouseButtonPress) {
+		enableEdition();
+	}
+	return false; // don't "eat" the event.
 }
 
 void MainTab::clearEquipment()
@@ -170,15 +142,15 @@ void MainTab::clearStats()
 	ui->timeLimits->clear();
 }
 
-#define UPDATE_TEXT(d, field)				\
+#define UPDATE_TEXT(d, field)			\
 	if (!d || !d->field)				\
 		ui->field->setText("");			\
-	else						\
+	else								\
 		ui->field->setText(d->field)
 
-
 void MainTab::updateDiveInfo(int dive)
 {
+	editMode = NONE;
 	// This method updates ALL tabs whenever a new dive or trip is
 	// selected.
 	// If exactly one trip has been selected, we show the location / notes
@@ -191,7 +163,7 @@ void MainTab::updateDiveInfo(int dive)
 
 	process_selected_dives();
 	process_all_dives(d, &prevd);
-	currentDive = d;
+
 	UPDATE_TEXT(d, notes);
 	UPDATE_TEXT(d, location);
 	UPDATE_TEXT(d, suit);
@@ -225,6 +197,8 @@ void MainTab::updateDiveInfo(int dive)
 			ui->suit->setVisible(true);
 			ui->rating->setVisible(true);
 			ui->visibility->setVisible(true);
+			ui->BuddyLabel->setVisible(true);
+			ui->DivemasterLabel->setVisible(true);
 			ui->divemaster->setReadOnly(false);
 			ui->buddy->setReadOnly(false);
 			ui->suit->setReadOnly(false);
@@ -284,8 +258,6 @@ void MainTab::updateDiveInfo(int dive)
 		ui->timeLimits->setMinimum(get_time_string(stats_selection.shortest_time.seconds, 0));
 		cylindersModel->setDive(d);
 		weightModel->setDive(d);
-		addCylinder->setEnabled(true);
-		addWeight->setEnabled(true);
 	} else {
 		/* make the fields read-only */
 		ui->location->setReadOnly(true);
@@ -312,8 +284,6 @@ void MainTab::updateDiveInfo(int dive)
 		ui->airPressureText->clear();
 		cylindersModel->clear();
 		weightModel->clear();
-		addCylinder->setEnabled(false);
-		addWeight->setEnabled(false);
 		ui->depthLimits->clear();
 		ui->sacLimits->clear();
 		ui->divesAllText->clear();
@@ -337,6 +307,10 @@ void MainTab::addWeight_clicked()
 
 void MainTab::reload()
 {
+	SuitCompletionModel::instance()->updateModel();
+	BuddyCompletionModel::instance()->updateModel();
+	LocationCompletionModel::instance()->updateModel();
+	DiveMasterCompletionModel::instance()->updateModel();
 }
 
 void MainTab::on_editAccept_clicked(bool edit)
@@ -352,24 +326,39 @@ void MainTab::on_editAccept_clicked(bool edit)
 	mainWindow()->dive_list()->setEnabled(!edit);
 
 	if (edit) {
+
+		// We may be editing one or more dives here. backup everything.
+		notesBackup.clear();
+
 		if (mainWindow() && mainWindow()->dive_list()->selectedTrips.count() == 1) {
 			// we are editing trip location and notes
 			ui->diveNotesMessage->setText(tr("This trip is being edited. Select Save or Undo when ready."));
 			ui->diveNotesMessage->animatedShow();
-			notesBackup.notes = ui->notes->toPlainText();
-			notesBackup.location = ui->location->text();
+			notesBackup[NULL].notes = ui->notes->toPlainText();
+			notesBackup[NULL].location = ui->location->text();
 			editMode = TRIP;
 		} else {
 			ui->diveNotesMessage->setText(tr("This dive is being edited. Select Save or Undo when ready."));
 			ui->diveNotesMessage->animatedShow();
-			notesBackup.buddy = ui->buddy->text();
-			notesBackup.suit = ui->suit->text();
-			notesBackup.notes = ui->notes->toPlainText();
-			notesBackup.divemaster = ui->divemaster->text();
-			notesBackup.location = ui->location->text();
-			notesBackup.rating = ui->rating->currentStars();
-			notesBackup.visibility = ui->visibility->currentStars();
-			editMode = DIVE;
+
+			// We may be editing one or more dives here. backup everything.
+			struct dive *mydive;
+			for (int i = 0; i < dive_table.nr; i++) {
+				mydive = get_dive(i);
+				if (!mydive)
+					continue;
+				if (!mydive->selected)
+					continue;
+
+				notesBackup[mydive].buddy = QString(mydive->buddy);
+				notesBackup[mydive].suit = QString(mydive->suit);
+				notesBackup[mydive].notes = QString(mydive->notes);
+				notesBackup[mydive].divemaster = QString(mydive->divemaster);
+				notesBackup[mydive].location = QString(mydive->location);
+				notesBackup[mydive].rating = mydive->rating;
+				notesBackup[mydive].visibility = mydive->visibility;
+			}
+		editMode = DIVE;
 		}
 	} else {
 		ui->diveNotesMessage->animatedHide();
@@ -377,38 +366,78 @@ void MainTab::on_editAccept_clicked(bool edit)
 		ui->editReset->hide();
 		/* now figure out if things have changed */
 		if (mainWindow() && mainWindow()->dive_list()->selectedTrips.count() == 1) {
-			if (notesBackup.notes != ui->notes->toPlainText() ||
-			    notesBackup.location != ui->location->text())
+			if (notesBackup[NULL].notes != ui->notes->toPlainText() ||
+			    notesBackup[NULL].location != ui->location->text())
 				mark_divelist_changed(TRUE);
 		} else {
-			if (notesBackup.buddy != ui->buddy->text() ||
-			    notesBackup.suit != ui->suit->text() ||
-			    notesBackup.notes != ui->notes->toPlainText() ||
-			    notesBackup.divemaster != ui->divemaster->text() ||
-			    notesBackup.location != ui->location->text() ||
-			    notesBackup.visibility != ui->visibility->currentStars() ||
-			    notesBackup.rating != ui->rating->currentStars())
+			struct dive *curr = current_dive;
+			if (notesBackup[curr].buddy != ui->buddy->text() ||
+			    notesBackup[curr].suit != ui->suit->text() ||
+			    notesBackup[curr].notes != ui->notes->toPlainText() ||
+			    notesBackup[curr].divemaster != ui->divemaster->text() ||
+			    notesBackup[curr].location  != ui->location->text() ||
+			    notesBackup[curr].rating 	!= ui->visibility->currentStars() ||
+			    notesBackup[curr].visibility != ui->rating->currentStars())
+
 				mark_divelist_changed(TRUE);
-			if (notesBackup.location != ui->location->text())
+			if (notesBackup[curr].location != ui->location->text())
 				mainWindow()->globe()->reload();
 		}
 		editMode = NONE;
 	}
+	QPalette p;
+	ui->buddy->setPalette(p);
+	ui->notes->setPalette(p);
+	ui->location->setPalette(p);
+	ui->divemaster->setPalette(p);
+	ui->suit->setPalette(p);
 }
 
+#define EDIT_TEXT2(what, text) \
+	textByteArray = text.toLocal8Bit(); \
+	free(what);\
+	what = strdup(textByteArray.data());
+
+#define EDIT_TEXT(what, text) \
+	QByteArray textByteArray = text.toLocal8Bit(); \
+	free(what);\
+	what = strdup(textByteArray.data());
+
 void MainTab::on_editReset_clicked()
 {
 	if (!ui->editAccept->isChecked())
 		return;
 
-	ui->notes->setText(notesBackup.notes);
-	ui->location->setText(notesBackup.location);
-	if (mainWindow() && mainWindow()->dive_list()->selectedTrips.count() != 1) {
-		ui->buddy->setText(notesBackup.buddy);
-		ui->suit->setText(notesBackup.suit);
-		ui->divemaster->setText(notesBackup.divemaster);
-		ui->rating->setCurrentStars(notesBackup.rating);
-		ui->visibility->setCurrentStars(notesBackup.visibility);
+	if (mainWindow() && mainWindow()->dive_list()->selectedTrips.count() == 1){
+		ui->notes->setText(notesBackup[NULL].notes );
+		ui->location->setText(notesBackup[NULL].location);
+	}else{
+		struct dive *curr = current_dive;
+		ui->notes->setText(notesBackup[curr].notes );
+		ui->location->setText(notesBackup[curr].location);
+		ui->buddy->setText(notesBackup[curr].buddy);
+		ui->suit->setText(notesBackup[curr].suit);
+		ui->divemaster->setText(notesBackup[curr].divemaster);
+		ui->rating->setCurrentStars(notesBackup[curr].rating);
+		ui->visibility->setCurrentStars(notesBackup[curr].visibility);
+
+		struct dive *mydive;
+		for (int i = 0; i < dive_table.nr; i++) {
+			mydive = get_dive(i);
+			if (!mydive)
+				continue;
+			if (!mydive->selected)
+				continue;
+
+			QByteArray textByteArray;
+			EDIT_TEXT2(mydive->buddy, notesBackup[mydive].buddy);
+			EDIT_TEXT2(mydive->suit, notesBackup[mydive].suit);
+			EDIT_TEXT2(mydive->notes, notesBackup[mydive].notes);
+			EDIT_TEXT2(mydive->divemaster, notesBackup[mydive].divemaster);
+			EDIT_TEXT2(mydive->location, notesBackup[mydive].location);
+			mydive->rating = notesBackup[mydive].rating;
+			mydive->visibility = notesBackup[mydive].visibility;
+		}
 	}
 	ui->editAccept->setChecked(false);
 	ui->diveNotesMessage->animatedHide();
@@ -424,126 +453,105 @@ void MainTab::on_editReset_clicked()
 
 	ui->editAccept->hide();
 	ui->editReset->hide();
+	notesBackup.clear();
+	QPalette p;
+	ui->buddy->setPalette(p);
+	ui->notes->setPalette(p);
+	ui->location->setPalette(p);
+	ui->divemaster->setPalette(p);
+	ui->suit->setPalette(p);
 	editMode = NONE;
 }
+#undef EDIT_TEXT2
+
+#define EDIT_SELECTED_DIVES( WHAT ) \
+	if (editMode == NONE) \
+		return; \
+	struct dive *mydive; \
+\
+	for (int i = 0; i < dive_table.nr; i++) { \
+		mydive = get_dive(i); \
+		if (!mydive) \
+			continue; \
+		if (!mydive->selected) \
+			continue; \
+\
+		WHAT; \
+	}
 
-#define EDIT_TEXT(what, text) \
-	QByteArray textByteArray = text.toLocal8Bit(); \
-	free(what);\
-	what = strdup(textByteArray.data());
+void markChangedWidget(QWidget *w){
+	QPalette p;
+	p.setBrush(QPalette::Base, QColor(Qt::yellow).lighter());
+	w->setPalette(p);
+}
 
 void MainTab::on_buddy_textChanged(const QString& text)
 {
-	if (!currentDive)
-		return;
-	EDIT_TEXT(currentDive->buddy, text);
+	EDIT_SELECTED_DIVES( EDIT_TEXT(mydive->buddy, text) );
+	markChangedWidget(ui->buddy);
 }
 
 void MainTab::on_divemaster_textChanged(const QString& text)
 {
-	if (!currentDive)
-		return;
-	EDIT_TEXT(currentDive->divemaster, text);
+	EDIT_SELECTED_DIVES( EDIT_TEXT(mydive->divemaster, text) );
+	markChangedWidget(ui->divemaster);
 }
 
 void MainTab::on_location_textChanged(const QString& text)
 {
+	if (editMode == NONE)
+		return;
 	if (editMode == TRIP && mainWindow() && mainWindow()->dive_list()->selectedTrips.count() == 1) {
 		// we are editing a trip
 		dive_trip_t *currentTrip = *mainWindow()->dive_list()->selectedTrips.begin();
 		EDIT_TEXT(currentTrip->location, text);
 	} else if (editMode == DIVE){
-		if (!currentDive)
-			return;
-		EDIT_TEXT(currentDive->location, text);
+		EDIT_SELECTED_DIVES( EDIT_TEXT(mydive->location, text) )
 	}
+
+	markChangedWidget(ui->location);
 }
 
 void MainTab::on_suit_textChanged(const QString& text)
 {
-	if (!currentDive)
-		return;
-	EDIT_TEXT(currentDive->suit, text);
+	EDIT_SELECTED_DIVES( EDIT_TEXT(mydive->suit, text) );
+	markChangedWidget(ui->suit);
 }
 
 void MainTab::on_notes_textChanged()
 {
+	if (editMode == NONE)
+		return;
 	if (editMode == TRIP && mainWindow() && mainWindow()->dive_list()->selectedTrips.count() == 1) {
 		// we are editing a trip
 		dive_trip_t *currentTrip = *mainWindow()->dive_list()->selectedTrips.begin();
 		EDIT_TEXT(currentTrip->notes, ui->notes->toPlainText());
 	} else if (editMode == DIVE) {
-		if (!currentDive)
-			return;
-		EDIT_TEXT(currentDive->notes, ui->notes->toPlainText());
+		EDIT_SELECTED_DIVES( EDIT_TEXT(mydive->notes, ui->notes->toPlainText()) );
 	}
+	markChangedWidget(ui->notes);
 }
 
 #undef EDIT_TEXT
 
 void MainTab::on_rating_valueChanged(int value)
 {
-	if (!currentDive)
-		return;
-	currentDive->rating = value;
+	EDIT_SELECTED_DIVES(mydive->rating  = value );
 }
 
 void MainTab::on_visibility_valueChanged(int value)
 {
-	if (!currentDive)
-		return;
-	currentDive->visibility = value;
-}
-
-void MainTab::hideEvent(QHideEvent* event)
-{
-	QSettings s;
-	s.beginGroup("MainTab");
-	s.beginGroup("Cylinders");
-	for (int i = 0; i < CylindersModel::COLUMNS; i++) {
-		s.setValue(QString("colwidth%1").arg(i), ui->cylinders->columnWidth(i));
-	}
-	s.endGroup();
-	s.beginGroup("Weights");
-	for (int i = 0; i < WeightModel::COLUMNS; i++) {
-		s.setValue(QString("colwidth%1").arg(i), ui->weights->columnWidth(i));
-	}
-	s.endGroup();
-	s.sync();
-}
-
-void MainTab::initialUiSetup()
-{
-	QSettings s;
-	s.beginGroup("MainTab");
-	s.beginGroup("Cylinders");
-	for (int i = 0; i < CylindersModel::COLUMNS; i++) {
-		QVariant width = s.value(QString("colwidth%1").arg(i));
-		if (width.isValid())
-			ui->cylinders->setColumnWidth(i, width.toInt());
-		else
-			ui->cylinders->resizeColumnToContents(i);
-	}
-	s.endGroup();
-	s.beginGroup("Weights");
-	for (int i = 0; i < WeightModel::COLUMNS; i++) {
-		QVariant width = s.value(QString("colwidth%1").arg(i));
-		if (width.isValid())
-			ui->weights->setColumnWidth(i, width.toInt());
-		else
-			ui->weights->resizeColumnToContents(i);
-	}
-	s.endGroup();
+	EDIT_SELECTED_DIVES( mydive->visibility = value );
 }
 
 void MainTab::editCylinderWidget(const QModelIndex& index)
 {
-	if (index.column() != CylindersModel::REMOVE)
+	if (index.isValid() && index.column() != CylindersModel::REMOVE)
 		ui->cylinders->edit(index);
 }
 
 void MainTab::editWeigthWidget(const QModelIndex& index)
 {
-	if (index.column() != WeightModel::REMOVE)
+	if (index.isValid() && index.column() != WeightModel::REMOVE)
 		ui->weights->edit(index);
 }