property_tree_model.h

Go to the documentation of this file.

/*
 * Copyright (c) 2012, Willow Garage, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the Willow Garage, Inc. nor the names of its
 *       contributors may be used to endorse or promote products derived from
 *       this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
#ifndef PROPERTY_MODEL_H
#define PROPERTY_MODEL_H

#include <QAbstractItemModel>

namespace rviz
{
class Property;

class PropertyTreeModel : public QAbstractItemModel
{
  Q_OBJECT
public:
  PropertyTreeModel(Property* root_property, QObject* parent = nullptr);

  ~PropertyTreeModel() override;

  void setDragDropClass(const QString& drag_drop_class)
  {
    drag_drop_class_ = drag_drop_class;
  }

  // Read-only model functions:
  QVariant data(const QModelIndex& index, int role) const override;
  QVariant
  headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;

  QModelIndex index(int row, int column, const QModelIndex& parent = QModelIndex()) const override;
  QModelIndex parent(const QModelIndex& index) const override;

  QModelIndex parentIndex(const Property* child) const;

  int rowCount(const QModelIndex& parent = QModelIndex()) const override;

  int columnCount(const QModelIndex& /*parent*/ = QModelIndex()) const override
  {
    return 2;
  }

  // Editable model functions:
  Qt::ItemFlags flags(const QModelIndex& index) const override;
  bool setData(const QModelIndex& index, const QVariant& value, int role = Qt::EditRole) override;

  Qt::DropActions supportedDropActions() const override
  {
    return Qt::MoveAction;
  }

  QMimeData* mimeData(const QModelIndexList& indexes) const override;

  bool dropMimeData(const QMimeData* data,
                    Qt::DropAction action,
                    int destination_row,
                    int destination_column,
                    const QModelIndex& destination_parent) override;

  QStringList mimeTypes() const override;

  Property* getRoot() const
  {
    return root_property_;
  }

  QModelIndex indexOf(Property* property) const;

  void emitDataChanged(Property* property);

  void beginInsert(Property* parent_property, int row_within_parent, int count = 1);
  void endInsert();

  void beginRemove(Property* parent_property, int row_within_parent, int count = 1);
  void endRemove();

  Property* getProp(const QModelIndex& index) const;

  void emitPropertyHiddenChanged(const Property* property)
  {
    Q_EMIT propertyHiddenChanged(property);
  }

  void expandProperty(Property* property);

  void collapseProperty(Property* property);

  void printPersistentIndices();

Q_SIGNALS:
  void propertyHiddenChanged(const Property* property);

  void configChanged();

  void expand(const QModelIndex& index);

  void collapse(const QModelIndex& index);

private:
  Property* root_property_;
  QString drag_drop_class_; 
};

} // end namespace rviz

#endif // PROPERTY_MODEL_H