1 files changed, 126 insertions, 0 deletions
diff --git a/qt-ui/models.h b/qt-ui/models.h
new file mode 100644
index 000000000..ac533fd71
--- /dev/null
+++ b/qt-ui/models.h
@@ -0,0 +1,126 @@
+/*
+ * models.h
+ *
+ * header file for the equipment models of Subsurface
+ *
+ */
+#ifndef MODELS_H
+#define MODELS_H
+
+#include <QAbstractTableModel>
+#include <QCoreApplication>
+
+#include "../dive.h"
+#include "../divelist.h"
+
+/* Encapsulates the tank_info global variable
+ * to show on Qt's Model View System.*/
+class TankInfoModel : public QAbstractTableModel {
+Q_OBJECT
+public:
+	enum Column { DESCRIPTION, ML, BAR};
+	TankInfoModel();
+
+	/*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;
+
+	void add(const QString& description);
+	void clear();
+	void update();
+private:
+	int rows;
+};
+
+/* Encapsulation of the Cylinder Model, that presents the
+ * Current cylinders that are used on a dive. */
+class CylindersModel : public QAbstractTableModel {
+Q_OBJECT
+public:
+	enum Column {TYPE, SIZE, MAXPRESS, START, END, O2, HE};
+
+	explicit CylindersModel(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;
+
+	void add(cylinder_t *cyl);
+	void clear();
+	void update();
+private:
+	/* Since the dive doesn't stores the number of cylinders that
+	 * it has (max 8) and since I don't want to make a
+	 * model-for-each-dive, let's hack this here instead. */
+	QMap<struct dive *, int> usedRows;
+};
+
+/* Encapsulation of the Weight Model, that represents
+ * the current weights on a dive. */
+class WeightModel : public QAbstractTableModel {
+Q_OBJECT
+public:
+	enum Column {TYPE, WEIGHT};
+	/*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;
+
+	void add(weightsystem_t *weight);
+	void clear();
+	void update();
+private:
+	/* Remember the number of rows in a dive */
+	QMap<struct dive *, int> usedRows;
+};
+
+/*! An AbstractItemModel for recording dive trip information such as a list of dives.
+*
+*/
+
+struct TreeItemDT {
+	Q_DECLARE_TR_FUNCTIONS ( TreeItemDT );
+public:
+	enum Column {NR, DATE, RATING, DEPTH, DURATION, TEMPERATURE, TOTALWEIGHT,
+				SUIT, CYLINDER, NITROX, SAC, OTU, MAXCNS, LOCATION, DIVE, COLUMNS };
+
+	enum ExtraRoles{STAR_ROLE = Qt::UserRole + 1, DIVE_ROLE};
+
+	virtual ~TreeItemDT();
+	int columnCount() const {
+		return COLUMNS;
+	};
+
+	virtual QVariant data ( int column, int role ) const;
+	int row() const;
+	QList<TreeItemDT *> children;
+	TreeItemDT *parent;
+};
+
+struct TripItem;
+
+class DiveTripModel : public QAbstractItemModel
+{
+	Q_OBJECT
+
+public:
+	DiveTripModel(QObject *parent = 0);
+	~DiveTripModel();
+
+	/*reimp*/ Qt::ItemFlags flags(const QModelIndex &index) const;
+	/*reimp*/ QVariant data(const QModelIndex &index, int role) const;
+	/*reimp*/ QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
+	/*reimp*/ int rowCount(const QModelIndex &parent = QModelIndex()) const;
+	/*reimp*/ int columnCount(const QModelIndex &parent = QModelIndex()) const;
+	/*reimp*/ QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
+	/*reimp*/ QModelIndex parent(const QModelIndex &child) const;
+
+private:
+	void setupModelData();
+
+	TreeItemDT *rootItem;
+	QMap<dive_trip_t*, TripItem*> trips;
+};
+
+#endif