/*=========================================================================

  Library:   CTK

  Copyright (c) Kitware Inc.

  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.txt

  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.

=========================================================================*/

#ifndef __ctkVTKDataSetArrayComboBox_h
#define __ctkVTKDataSetArrayComboBox_h

// Qt includes
#include <QComboBox>

// CTK includes
#include "ctkVisualizationVTKWidgetsExport.h"
#include "ctkVTKDataSetModel.h"
class ctkVTKDataSetArrayComboBoxPrivate;

class vtkAbstractArray;
class vtkDataSet;

/// \ingroup Visualization_VTK_Widgets
///
/// QComboBox linked to vtkDataSet field arrays
class CTK_VISUALIZATION_VTK_WIDGETS_EXPORT ctkVTKDataSetArrayComboBox
  : public QComboBox
{
  Q_OBJECT
  Q_PROPERTY(ctkVTKDataSetModel::AttributeTypes attributeTypes READ attributeTypes WRITE setAttributeTypes)

public:
  /// Superclass typedef
  typedef QComboBox Superclass;

  /// Constructors
  explicit ctkVTKDataSetArrayComboBox(QWidget* parent = 0);
  virtual ~ctkVTKDataSetArrayComboBox();

  vtkAbstractArray* currentArray()const;
  QString currentArrayName()const;
  vtkDataSet* dataSet()const;

  /// Return the current array location, i.e. whether it's a point data
  /// array or a cell data array.
  /// \sa currentArrayChanged(int), ctkVTKDataSetModel::locationFromItem()
  int currentArrayLocation()const;

  ctkVTKDataSetModel::AttributeTypes attributeTypes()const;
  void setAttributeTypes(const ctkVTKDataSetModel::AttributeTypes& attributeTypes);

  /// Return a pointer to the model used to populate the combobox.
  /// \sa dataSet()
  ctkVTKDataSetModel* dataSetModel()const;

public Q_SLOTS:
  /// Set the dataset to observe. It is used to as the "model" for the combobox.
  /// \sa dataSet(), dataSetModel()
  void setDataSet(vtkDataSet* dataSet);
  /// The array must exist in the dataset
  void setCurrentArray(vtkAbstractArray* dataArray);
  /// the array must exist in the dataset
  void setCurrentArray(const QString& name);

Q_SIGNALS:
  void currentArrayChanged(vtkAbstractArray*);
  void currentArrayChanged(const QString& name);

protected Q_SLOTS:
  void onCurrentIndexChanged(int);
protected:
  QScopedPointer<ctkVTKDataSetArrayComboBoxPrivate> d_ptr;

private:
  Q_DECLARE_PRIVATE(ctkVTKDataSetArrayComboBox);
  Q_DISABLE_COPY(ctkVTKDataSetArrayComboBox);
};

#endif