tree_completer.h

Go to the documentation of this file.

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */
 
#ifndef TREE_COMPLETER_H
#define TREE_COMPLETER_H
 
#include <QStandardItem>
#include <QStandardItemModel>
#include <QString>
#include <QVariant>
#include <QFont>
#include <QFontDatabase>
#include <map>
#include "PlotJuggler/alphanum.hpp"
/*/
class TreeItem {
 public:
  explicit TreeItem(QStandardItem* name_item, QStandardItem* value_item)
      : _name_item(name_item), _value_item(value_item) {}
 
  TreeItem* appendChild(const QString& name)
  {
    auto child_name = new QStandardItem(name);
    auto child_value = new QStandardItem("-");
 
    child_value->setSelectable(false);
    child_value->setTextAlignment(Qt::AlignRight);
    child_value->setFlags( Qt::NoItemFlags | Qt::ItemIsEnabled );
    auto font = QFontDatabase::systemFont(QFontDatabase::FixedFont);
    child_value->setFont( font );
    child_value->setFlags(Qt::NoItemFlags);
 
    QList<QStandardItem*> columns;
    columns << child_name << child_value;
    _name_item->appendRow(columns);
 
    auto res =
        _child_items_map.insert(std::make_pair(name, TreeItem(child_name, child_value)));
    return &(res.first->second);
  }
 
  TreeItem* findChild(const QString& name) {
    auto it = _child_items_map.find(name);
    if (it == _child_items_map.end()) {
      return nullptr;
    }
    return &(it->second);
  }
 
  QStandardItem* nameItem() { return _name_item; }
  QStandardItem* valueItem() { return _value_item; }
 
 private:
  std::map<QString, TreeItem> _child_items_map;
  QStandardItem* _name_item;
  QStandardItem* _value_item;
};
 
class TreeModel : public QAbstractItemModel {
 public:
  TreeModel(QStandardItemModel* parent_model)
      : QStandardItemModel(0, 2, parent_model),
        _root_tree_item(invisibleRootItem(), nullptr),
        _parent_model(parent_model) {}
 
  void clear() {
    QStandardItemModel::clear();
    _root_tree_item = TreeItem(invisibleRootItem(), nullptr);
  }
 
  void addToTree(const QString& name, int reference_row) {
    auto parts = name.split('/', PJ::SkipEmptyParts);
    if (parts.size() == 0) {
      return;
    }
 
    TreeItem* tree_parent = &_root_tree_item;
 
    for (int i = 0; i < parts.size(); i++) {
      bool is_leaf = (i == parts.size() - 1);
      const auto& part = parts[i];
 
      TreeItem* matching_child = tree_parent->findChild(part);
      if (matching_child) {
        tree_parent = matching_child;
      } else {
 
        tree_parent = tree_parent->appendChild(part);
        tree_parent->nameItem()->setSelectable(is_leaf);
      }
    }
  }
 
 private:
  TreeItem _root_tree_item;
  QStandardItemModel* _parent_model;
};*/
 
#endif  // TREE_COMPLETER_H