diff --git a/desktop-widgets/tab-widgets/TabDiveEquipment.cpp b/desktop-widgets/tab-widgets/TabDiveEquipment.cpp index 0e0c1fee1..ef739c131 100644 --- a/desktop-widgets/tab-widgets/TabDiveEquipment.cpp +++ b/desktop-widgets/tab-widgets/TabDiveEquipment.cpp @@ -50,10 +50,10 @@ TabDiveEquipment::TabDiveEquipment(MainTab *parent) : TabBase(parent), connect(cylindersModel, &CylindersModel::divesEdited, this, &TabDiveEquipment::divesEdited); connect(weightModel, &WeightModel::divesEdited, this, &TabDiveEquipment::divesEdited); - ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::TYPE, new TankInfoDelegate(this)); - ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::USE, new TankUseDelegate(this)); - ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::SENSORS, new SensorDelegate(this)); - ui.weights->view()->setItemDelegateForColumn(WeightModel::TYPE, new WSInfoDelegate(this)); + ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::TYPE, &tankInfoDelegate); + ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::USE, &tankUseDelegate); + ui.cylinders->view()->setItemDelegateForColumn(CylindersModel::SENSORS, &sensorDelegate); + ui.weights->view()->setItemDelegateForColumn(WeightModel::TYPE, &wsInfoDelegate); ui.cylinders->view()->setColumnHidden(CylindersModel::DEPTH, true); ui.cylinders->view()->setColumnHidden(CylindersModel::WORKINGPRESS_INT, true); ui.cylinders->view()->setColumnHidden(CylindersModel::SIZE_INT, true); diff --git a/desktop-widgets/tab-widgets/TabDiveEquipment.h b/desktop-widgets/tab-widgets/TabDiveEquipment.h index 934ec9126..d01786aa9 100644 --- a/desktop-widgets/tab-widgets/TabDiveEquipment.h +++ b/desktop-widgets/tab-widgets/TabDiveEquipment.h @@ -6,6 +6,7 @@ #include "ui_TabDiveEquipment.h" #include "qt-models/completionmodels.h" #include "desktop-widgets/divelistview.h" +#include "desktop-widgets/modeldelegates.h" namespace Ui { class TabDiveEquipment; @@ -38,6 +39,11 @@ private: SuitCompletionModel suitModel; CylindersModel *cylindersModel; WeightModel *weightModel; + + TankInfoDelegate tankInfoDelegate; + TankUseDelegate tankUseDelegate; + SensorDelegate sensorDelegate; + WSInfoDelegate wsInfoDelegate; }; #endif // TAB_DIVE_EQUIPMENT_H