ImageTreeWidget.h

#ifndef ImageTreeWidget_H
#define ImageTreeWidget_H

#include <QPointer>
#include <QTreeWidget>

class QProgressBar;

namespace Isis {
  class Directory;
  class DisplayProperties;
  class Image;
  class ImageList;
  class ImageTreeWidgetItem;
  class ProgressBar;

  class ImageTreeWidget : public QTreeWidget {
      Q_OBJECT

    public:

      enum TreeItemContainerType {
        ImageGroupType = 1,
        ImageListNameType
      };

      ImageTreeWidget(Directory *directory = 0, QWidget *parent = 0);
      virtual ~ImageTreeWidget();

      QList<QAction *> actions();

      QTreeWidgetItem *addGroup(QString imageListName, QString groupName = "", int index = -1);

      QTreeWidgetItem *createGroup(QTreeWidgetItem *imageListItem, QString groupName = "",
                                   int index = -1);
      QTreeWidgetItem *createImageListNameItem(QString name);

      void refit();

      void disableSort();
      void enqueueReadDisplayProperties(ImageTreeWidgetItem *);

      QList<QAction *> getViewActions();
      bool groupInList(QList<QTreeWidgetItem *>);
      void updateViewActs();

      ImageTreeWidgetItem *prepCube(ImageList *imageList, Image *image);

      ImageList imagesInView();

    public:
      class ImagePosition {
        public:
          ImagePosition();
          ImagePosition(int outerIndex, int innerIndex);
          ImagePosition(const ImagePosition &other);

          virtual ~ImagePosition();

          void setPosition(int outerIndex, int innerIndex);

          int group() const;
          int index() const;
          bool isValid() const;

          void swap(ImagePosition &other);

          bool operator<(const ImagePosition &rhs);
          ImagePosition &operator=(const ImagePosition &rhs);

        private:
          void init();

        private:
          int m_group;
          int m_index;
      };

    signals:
      void queueSelectionChanged();
      void queueReadDisplayProperties();

    protected:
      void dropEvent(QDropEvent *event);
      void mousePressEvent(QMouseEvent *event);
      void contextMenuEvent(QContextMenuEvent *event);

    private slots:
      QTreeWidgetItem *addGroup();
      void deleteSelectedGroups();
      void imageDeleted(QObject *image);
      void onItemChanged(QTreeWidgetItem *, int);
      void onSelectionChanged();
      void onQueuedReadDisplayProperties();
      void onQueuedSelectionChanged();
      void propertiesChanged(DisplayProperties *);
      void renameSelectedGroup();
      void requestCloseSelected();
      void toggleColumnVisible();
      void updateDragAndDropability();
      void setDefaultFileListCols();

    private:
      bool mosaicItemInList(QList<QTreeWidgetItem *>);
      ImageList selectedDisplays();

      QTreeWidgetItem *imageListTreeItem(QString imageListName);
      ImageTreeWidgetItem *treeItem(DisplayProperties *);
      ImageTreeWidgetItem *treeItem(Image *);

      Directory *m_directory;
      QList<QAction *> m_viewActs;
      QPointer<QAction> m_setFileListColsAct;
      QMap<DisplayProperties *, ImageTreeWidgetItem *> m_displayPropsToTreeItemLookup;
      bool m_queuedSelectionChanged;

      QList<ImageTreeWidgetItem *> m_queuedReadDisplayPropertiesItems;
  };
};

#endif