1 files changed, 52 insertions, 0 deletions
diff --git a/src/widgets/unitcombobox.h b/src/widgets/unitcombobox.h
new file mode 100644
index 0000000..dd75107
--- /dev/null
+++ b/src/widgets/unitcombobox.h
@@ -0,0 +1,52 @@
+/***************************************************************************
+*   Copyright (C) 2006 by                                                 *
+*   Jason Kivlighn ([email protected])                                  *
+*                                                                         *
+*   This program is free software; you can redistribute it and/or modify  *
+*   it under the terms of the GNU General Public License as published by  *
+*   the Free Software Foundation; either version 2 of the License, or     *
+*   (at your option) any later version.                                   *
+***************************************************************************/
+
+#ifndef UNITCOMBOBOX_H
+#define UNITCOMBOBOX_H
+
+#include <kcombobox.h>
+
+#include <tqmap.h>
+
+#include "datablocks/unit.h"
+
+class RecipeDB;
+
+class UnitComboBox : public KComboBox
+{
+	TQ_OBJECT
+
+public:
+	UnitComboBox( TQWidget *parent, RecipeDB *db, Unit::Type type = Unit::All );
+
+	void reload();
+	int id( int row ) const;
+	void setSelected( int unitID );
+	Unit unit() const;
+
+protected:
+	virtual void popup();
+
+private slots:
+	void createUnit( const Unit & );
+	void removeUnit( int id );
+
+	int findInsertionPoint( const TQString &name );
+
+private:
+	void loadUnits( const UnitList &unitList );
+
+	RecipeDB *database;
+	TQMap<int, int> unitComboRows; // Contains the unit id for every given row in the unit combobox
+	Unit::Type m_type;
+};
+
+#endif //UNITCOMBOBOX_H
+