Second model converted to be less boilerplated.

Second model converted to be less boilerplated,
there are right now 7 models that will need to
be ported. each model can remove around 30 to 40
LOC, so this is not such a bigger improvement,
but it's an 'keeping an eye on the future'.

Also fixed a bug where the returned value was dummy.

Signed-off-by: Tomaz Canabrava <tcanabrava@kde.org>

1 files changed, 2 insertions, 4 deletions

diff --git a/qt-ui/models.h b/qt-ui/models.h
index 24f829add..5e3d07121 100644
--- a/qt-ui/models.h
+++ b/qt-ui/models.h
@@ -109,14 +109,12 @@ private:
 
 /* Encapsulation of the Weight Model, that represents
  * the current weights on a dive. */
-class WeightModel : public QAbstractTableModel {
+class WeightModel : public CleanerTableModel {
 Q_OBJECT
 public:
-	enum Column {REMOVE, TYPE, WEIGHT, COLUMNS};
+	enum Column {REMOVE, TYPE, WEIGHT};
 
 	explicit WeightModel(QObject *parent = 0);
-	/*reimp*/ QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
-	/*reimp*/ int columnCount(const QModelIndex& parent = QModelIndex()) const;
 	/*reimp*/ QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
 	/*reimp*/ int rowCount(const QModelIndex& parent = QModelIndex()) const;
 	/*reimp*/ Qt::ItemFlags flags(const QModelIndex& index) const;