1 files changed, 6 insertions, 4 deletions
diff --git a/qt-ui/profile/diveplotdatamodel.h b/qt-ui/profile/diveplotdatamodel.h
index 6f4929bb2..e84645b40 100644
--- a/qt-ui/profile/diveplotdatamodel.h
+++ b/qt-ui/profile/diveplotdatamodel.h
@@ -3,6 +3,8 @@
 
 #include <QAbstractTableModel>
 
+#include "display.h"
+
 struct dive;
 struct plot_data;
 struct plot_info;
@@ -19,16 +21,16 @@ public:
 	virtual QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
 	virtual int rowCount(const QModelIndex& parent = QModelIndex()) const;
 	void clear();
-	void setDive(struct dive *d, const plot_info& pInfo);
-	plot_data* data();
+	void setDive(struct dive *d,  const plot_info& pInfo);
+	const plot_info& data() const;
 	int id() const;
 	double pheMax();
 	double pn2Max();
 	double po2Max();
 	void emitDataChanged();
+	void calculateDecompression();
 private:
-	int sampleCount;
-	plot_data *plotData;
+	plot_info pInfo;
 	int diveId;
 };