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 = 0 );

  virtual ~PropertyTreeModel();

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

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

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

  QModelIndex parentIndex( const Property* child ) const;

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

  virtual int columnCount( const QModelIndex &parent = QModelIndex() ) const { return 2; }

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

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

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

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

  virtual QStringList mimeTypes () const;

  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