add_display_dialog.h

Go to the documentation of this file.

/*
 * Copyright (c) 2008, 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 RVIZ_ADD_DISPLAY_DIALOG_H
#define RVIZ_ADD_DISPLAY_DIALOG_H

#include <QDialog>
#include <QTreeWidget>
#include <QComboBox>

#include <boost/shared_ptr.hpp>

#include "rviz/factory.h"

class QTextBrowser;
class QLineEdit;
class QDialogButtonBox;
class QLabel;
class QCheckBox;

namespace rviz
{

class DisplayFactory;
class Display;

struct SelectionData {
  QString whats_this;
  QString lookup_name;
  QString display_name;
  QString topic;
  QString datatype;
};

class AddDisplayDialog : public QDialog
{
Q_OBJECT
public:
  AddDisplayDialog( DisplayFactory* factory,
                    const QString& object_type,
                    const QStringList& disallowed_display_names,
                    const QStringList& disallowed_class_lookup_names,
                    QString* lookup_name_output,
                    QString* display_name_output = 0,
                    QString* topic_output = 0,
                    QString* datatype_output = 0,
                    QWidget* parent = 0 );

  virtual QSize sizeHint () const;

public Q_SLOTS:
  virtual void accept();

private Q_SLOTS:
  void onDisplaySelected( SelectionData *data);
  void onTopicSelected( SelectionData *data );
  void onTabChanged( int index );
  void onNameChanged();

private:
  void fillTree( QTreeWidget* tree );

  bool isValid();

  void setError( const QString& error_text );

  void updateDisplay();

  Factory* factory_;
  const QStringList& disallowed_display_names_;
  const QStringList& disallowed_class_lookup_names_;

  QString* lookup_name_output_;
  QString* display_name_output_;
  QString* topic_output_;
  QString* datatype_output_;

  QTabWidget *tab_widget_;
  int topic_tab_;
  int display_tab_;
  SelectionData display_data_;
  SelectionData topic_data_;

  QTextBrowser* description_;

  QLineEdit* name_editor_;

  QDialogButtonBox* button_box_;

  QString lookup_name_;
};


class DisplayTypeTree : public QTreeWidget {
Q_OBJECT
public:
  DisplayTypeTree();

  void fillTree(Factory *factory);

Q_SIGNALS:
  void itemChanged( SelectionData *selection );

private Q_SLOTS:
  void onCurrentItemChanged(QTreeWidgetItem *curr, QTreeWidgetItem *prev);
};

class TopicDisplayWidget : public QWidget {
Q_OBJECT
public:
  TopicDisplayWidget();
  void fill(DisplayFactory *factory);

Q_SIGNALS:
  void itemChanged( SelectionData *selection );
  void itemActivated( QTreeWidgetItem *item, int column );

private Q_SLOTS:
  void stateChanged(int state);
  void onCurrentItemChanged( QTreeWidgetItem *curr );
  void onComboBoxClicked( QTreeWidgetItem *curr );

private:
  void findPlugins( DisplayFactory* );

  QTreeWidgetItem* insertItem ( const QString &topic, bool disabled );

  QTreeWidget *tree_;
  QCheckBox *enable_hidden_box_;

  // Map from ROS topic type to all displays that can visualize it.
  // One key may have multiple values.
  QMap<QString, QString > datatype_plugins_;
};

class EmbeddableComboBox : public QComboBox
{
  Q_OBJECT
public:
  EmbeddableComboBox(QTreeWidgetItem *parent, int col)
    : parent_( parent ), column_(col)
  {
    connect(this, SIGNAL( activated( int )), this, SLOT( onActivated( int )));
  }

Q_SIGNALS:
  void itemClicked( QTreeWidgetItem *item, int column );

private Q_SLOTS:
  void onActivated( int index )
  {
    Q_EMIT itemClicked( parent_, column_ );
  }

private:
  QTreeWidgetItem *parent_;
  int column_;
};

} //namespace rviz

#endif // RVIZ_ADD_DISPLAY_DIALOG_H