properties_widget.h

// Copyright 2018 The Darwin Neuroevolution Framework Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once

#include <core/properties.h>

#include <QString>
#include <QStyledItemDelegate>
#include <QTreeWidget>
#include <QTreeWidgetItem>

#include <string>
#include <unordered_map>
using namespace std;

namespace core_ui {

class PropertyItem;
class BoundPropertyItem;
class PropertiesWidget;

class PropertyItemBase : public QTreeWidgetItem {
 public:
  PropertiesWidget* parentWidget() const;

  PropertyItemBase* parent() const;

  virtual bool isModified() const;

  virtual void valueChanged();

 protected:
  explicit PropertyItemBase(PropertyItemBase* parent) : QTreeWidgetItem(parent) {}
};

class PropertiesSectionItem : public PropertyItemBase {
  friend class PropertiesWidget;

 public:
  PropertyItem* addProperty(const string& name);

  BoundPropertyItem* addProperty(core::Property* property);

 private:
  explicit PropertiesSectionItem(const string& name);
};

class PropertyItem : public PropertyItemBase {
  friend class PropertiesSectionItem;

 public:
  template <class T, class... EXTRA>
  void setValue(const T& value, EXTRA&&... extra) {
    setText(1, QString("%1").arg(value, std::forward<EXTRA>(extra)...));
  }

  void setValue(const string& value) { setValue(QString::fromStdString(value)); }

 private:
  PropertyItem(PropertyItemBase* section, const string& name);
};

class BoundPropertyItem : public PropertyItemBase {
  friend class PropertiesSectionItem;

 public:
  core::Property* property() const { return property_; }

  bool isModified() const override;

  void valueChanged() override;

 private:
  BoundPropertyItem(PropertyItemBase* section, core::Property* property);

  void updateChildProperties();

 private:
  core::Property* property_ = nullptr;
  string original_value_;
  unordered_map<core::Property*, BoundPropertyItem*> child_properties_;
};

class PropertyItemDelegate : public QStyledItemDelegate {
  Q_OBJECT

 public:
  using QStyledItemDelegate::QStyledItemDelegate;

 private slots:
  void onComboBoxUpdate(int);

 private:
  QWidget* createEditor(QWidget* parent,
                        const QStyleOptionViewItem& option,
                        const QModelIndex& index) const override;

  void setModelData(QWidget* editor,
                    QAbstractItemModel* model,
                    const QModelIndex& index) const override;
};

class PropertiesWidget : public QTreeWidget {
  Q_OBJECT

  friend PropertyItemDelegate;

 public:
  explicit PropertiesWidget(QWidget* parent);

  PropertiesSectionItem* addSection(const string& name);

  PropertiesSectionItem* addPropertiesSection(const string& name,
                                              core::PropertySet* property_set);

  void autoSizeColumns();

  bool isModified() const;

 private:
  BoundPropertyItem* boundItemFromIndex(const QModelIndex& index) const;
};

}  // namespace core_ui