1 files changed, 19 insertions, 14 deletions

diff --git a/qt-ui/profile/diveprofileitem.cpp b/qt-ui/profile/diveprofileitem.cpp
index 398d71220..1a96d7009 100644
--- a/qt-ui/profile/diveprofileitem.cpp
+++ b/qt-ui/profile/diveprofileitem.cpp
@@ -7,43 +7,43 @@
 #include <QPainter>
 #include <QLinearGradient>
 
-DiveProfileItem::DiveProfileItem(): QObject(), QGraphicsPolygonItem(),
+AbstractProfilePolygonItem::AbstractProfilePolygonItem(): QObject(), QGraphicsPolygonItem(),
 	hAxis(NULL), vAxis(NULL), dataModel(NULL), hDataColumn(-1), vDataColumn(-1)
 {
 
 }
 
-void DiveProfileItem::setHorizontalAxis(DiveCartesianAxis* horizontal)
+void AbstractProfilePolygonItem::setHorizontalAxis(DiveCartesianAxis* horizontal)
 {
 	hAxis = horizontal;
 	modelDataChanged();
 }
 
-void DiveProfileItem::setHorizontalDataColumn(int column)
+void AbstractProfilePolygonItem::setHorizontalDataColumn(int column)
 {
 	hDataColumn = column;
 	modelDataChanged();
 }
 
-void DiveProfileItem::setModel(QAbstractTableModel* model)
+void AbstractProfilePolygonItem::setModel(QAbstractTableModel* model)
 {
 	dataModel = model;
 	modelDataChanged();
 }
 
-void DiveProfileItem::setVerticalAxis(DiveCartesianAxis* vertical)
+void AbstractProfilePolygonItem::setVerticalAxis(DiveCartesianAxis* vertical)
 {
 	vAxis = vertical;
 	modelDataChanged();
 }
 
-void DiveProfileItem::setVerticalDataColumn(int column)
+void AbstractProfilePolygonItem::setVerticalDataColumn(int column)
 {
 	vDataColumn = column;
 	modelDataChanged();
 }
 
-void DiveProfileItem::modelDataChanged()
+void AbstractProfilePolygonItem::modelDataChanged()
 {
 	// We don't have enougth data to calculate things, quit.
 	if (!hAxis || !vAxis || !dataModel || hDataColumn == -1 || vDataColumn == -1)
@@ -62,13 +62,6 @@ void DiveProfileItem::modelDataChanged()
 		poly.append(point);
 	}
 	setPolygon(poly);
-
-	// This is the blueish gradient that the Depth Profile should have.
-	// It's a simple QLinearGradient with 2 stops, starting from top to bottom.
-	QLinearGradient pat(0, poly.boundingRect().top(), 0, poly.boundingRect().bottom());
-	pat.setColorAt(1, getColor(DEPTH_BOTTOM));
-	pat.setColorAt(0, getColor(DEPTH_TOP));
-	setBrush(QBrush(pat));
 }
 
 void DiveProfileItem::paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget* widget) {
@@ -92,3 +85,15 @@ void DiveProfileItem::paint(QPainter* painter, const QStyleOptionGraphicsItem* o
 		painter->drawLine(polygon()[i-1],polygon()[i]);
 	}
 }
+
+void DiveProfileItem::modelDataChanged(){
+	AbstractProfilePolygonItem::modelDataChanged();
+	if(polygon().isEmpty())
+		return;
+		// This is the blueish gradient that the Depth Profile should have.
+	// It's a simple QLinearGradient with 2 stops, starting from top to bottom.
+	QLinearGradient pat(0, polygon().boundingRect().top(), 0, polygon().boundingRect().bottom());
+	pat.setColorAt(1, getColor(DEPTH_BOTTOM));
+	pat.setColorAt(0, getColor(DEPTH_TOP));
+	setBrush(QBrush(pat));
+}
\ No newline at end of file