// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright 2008 Sandia Corporation
// SPDX-License-Identifier: LicenseRef-BSD-3-Clause-Sandia-USGov
/**
 * @class   vtkQtTableView
 * @brief   A VTK view based on a Qt Table view.
 *
 *
 * vtkQtTableView is a VTK view using an underlying QTableView.
 *
 * @par Thanks:
 * Thanks to Brian Wylie from Sandia National Laboratories for implementing
 * this class
 */

#ifndef vtkQtTableView_h
#define vtkQtTableView_h

#include "vtkQtView.h"
#include "vtkViewsQtModule.h" // For export macro

#include "vtkSmartPointer.h" // Needed for member variables
#include <QPointer>          // Needed to hold the view

class QItemSelection;
class QSortFilterProxyModel;
class QTableView;

VTK_ABI_NAMESPACE_BEGIN
class vtkAddMembershipArray;
class vtkApplyColors;
class vtkAttributeDataToTableFilter;
class vtkIdTypeArray;
class vtkQtTableModelAdapter;

class VTKVIEWSQT_EXPORT vtkQtTableView : public vtkQtView
{
  Q_OBJECT

public:
  static vtkQtTableView* New();
  vtkTypeMacro(vtkQtTableView, vtkQtView);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  /**
   * Get the main container of this view (a  QWidget).
   * The application typically places the view with a call
   * to GetWidget(): something like this
   * this->ui->box->layout()->addWidget(this->View->GetWidget());
   */
  QWidget* GetWidget() override;

  /**
   * Have the view show/hide its column headers
   */
  void SetShowVerticalHeaders(bool);

  /**
   * Have the view show/hide its row headers
   */
  void SetShowHorizontalHeaders(bool);

  enum
  {
    FIELD_DATA = 0,
    POINT_DATA = 1,
    CELL_DATA = 2,
    VERTEX_DATA = 3,
    EDGE_DATA = 4,
    ROW_DATA = 5,
  };

  ///@{
  /**
   * The field type to copy into the output table.
   * Should be one of FIELD_DATA, POINT_DATA, CELL_DATA, VERTEX_DATA, EDGE_DATA.
   */
  vtkGetMacro(FieldType, int);
  void SetFieldType(int);
  ///@}

  /**
   * Enable drag and drop on this widget
   */
  void SetEnableDragDrop(bool);

  /**
   * Whether the table allows individual columns to be sorted upon
   * Sorting is enabled by default (turn off for large tables);
   */
  void SetSortingEnabled(bool);

  ///@{
  /**
   * Whether or not to display all columns from the input table or to use the
   * ColumnName provided.
   * FIXME: This should be replaced with an Add/Remove column API.
   */
  void SetShowAll(bool);
  vtkGetMacro(ShowAll, bool);
  ///@}

  ///@{
  /**
   * The name of a single column to display.
   * FIXME: This should be replaced with an Add/Remove column API.
   */
  vtkSetStringMacro(ColumnName);
  vtkGetStringMacro(ColumnName);
  ///@}

  void SetColumnVisibility(const QString& name, bool status);

  /**
   * Set whether or not the table view should split multi-component columns
   * into multiple single-component columns
   */
  void SetSplitMultiComponentColumns(bool value);

  /**
   * Get whether or not the table view splits multi-component columns into
   * multiple single-component columns
   */
  bool GetSplitMultiComponentColumns();

  ///@{
  /**
   * Whether or not to sort selections that the view receives to the top
   */
  void SetSortSelectionToTop(bool value);
  vtkGetMacro(SortSelectionToTop, bool);
  ///@}

  ///@{
  /**
   * Whether or not to add an icon to the row header denoting the color
   * of an annotated row.
   */
  void SetApplyRowColors(bool value);
  vtkGetMacro(ApplyRowColors, bool);
  ///@}

  /**
   * Updates the view.
   */
  void Update() override;

  ///@{
  /**
   * The array to use for coloring items in view.  Default is "color".
   */
  void SetColorArrayName(const char* name);
  const char* GetColorArrayName();
  ///@}

  ///@{
  /**
   * Whether to color vertices.  Default is off.
   */
  void SetColorByArray(bool vis);
  bool GetColorByArray();
  vtkBooleanMacro(ColorByArray, bool);
  ///@}

  /**
   * Apply a view theme to this view.
   */
  void ApplyViewTheme(vtkViewTheme* theme) override;

  enum
  {
    SELECT_ITEMS = 0,
    SELECT_ROWS,
    SELECT_COLUMNS
  };

  ///@{
  /**
   * The selection mode for this view.
   * SELECT_ITEMS (0) selects single items.
   * SELECT_ROWS (1) selects rows.
   * SELECT_COLUMNS (2) selects columns.
   * Linked selection only works when in the default mode
   * SELECT_ROWS. Selections from other modes may be retrieved
   * using GetSelectedItems().
   */
  virtual void SetSelectionBehavior(int type);
  virtual int GetSelectionBehavior();
  ///@}

  /**
   * Fills the array with the selected items of the view.
   * If the selection behavior is SELECT_ITEMS,
   * arr will be a 2-component array containing (row,column)
   * for each selected item.
   * If the selection behavior is SELECT_ROWS or SELECT_COLUMNS,
   * arr will contain a list of row or column indices.
   */
  virtual void GetSelectedItems(vtkIdTypeArray* arr);

protected:
  vtkQtTableView();
  ~vtkQtTableView() override;

  void AddRepresentationInternal(vtkDataRepresentation* rep) override;
  void RemoveRepresentationInternal(vtkDataRepresentation* rep) override;

private Q_SLOTS:
  void slotQtSelectionChanged(const QItemSelection&, const QItemSelection&);

private: // NOLINT(readability-redundant-access-specifiers)
  void SetVTKSelection();
  vtkMTimeType LastSelectionMTime;
  vtkMTimeType LastInputMTime;
  vtkMTimeType LastMTime;

  vtkSetStringMacro(ColorArrayNameInternal);
  vtkGetStringMacro(ColorArrayNameInternal);

  QPointer<QTableView> TableView;
  vtkQtTableModelAdapter* TableAdapter;
  QSortFilterProxyModel* TableSorter;
  int FieldType;
  bool ShowAll;
  char* ColumnName;
  bool InSelectionChanged;
  bool SortSelectionToTop;
  bool ApplyRowColors;
  char* ColorArrayNameInternal;

  vtkSmartPointer<vtkAddMembershipArray> AddSelectedColumn;
  vtkSmartPointer<vtkAttributeDataToTableFilter> DataObjectToTable;
  vtkSmartPointer<vtkApplyColors> ApplyColors;

  vtkQtTableView(const vtkQtTableView&) = delete;
  void operator=(const vtkQtTableView&) = delete;
};

VTK_ABI_NAMESPACE_END
#endif
