#ifndef DELEGATES_H
#define DELEGATES_H

#include <QStyledItemDelegate>
#include <QDoubleValidator>
#include <QComboBox>
#include <QLineEdit>

class ComboBoxItemDelegate : public QStyledItemDelegate {
  Q_OBJECT

public:
  QWidget* createEditor(QWidget* parent,
                        const QStyleOptionViewItem& option,
                        const QModelIndex& index) const override {
    QComboBox* editor = new QComboBox(parent);
    editor->setEditable(true);
    editor->setValidator(new QDoubleValidator(-100,100,4));

    QStringList word_list;
    word_list << "0";
    word_list << "1";
    editor->addItems(word_list);

    return editor;
  }
};

class LineEditItemDelegate : public QStyledItemDelegate {
  Q_OBJECT

public:
  QWidget* createEditor(QWidget* parent,
                        const QStyleOptionViewItem& option,
                        const QModelIndex& index) const override {
    QLineEdit* editor = new QLineEdit(parent);
    editor->setValidator(new QDoubleValidator(-100,100,4));

    return editor;
  }
};

#endif // DELEGATES_H