db_tree_wid.hpp

/* hds_db_tree_wid.hpp */
/* $Id$ */
#ifndef __db_tree_wid_hpp
#define __db_tree_wid_hpp

#include <QTreeWidgetItem>

class QTreeWidget;
class QAction;
class QMenu;

namespace hds
{
  class DbTreeWid : public QWidget
  {
    Q_OBJECT
    
  public:  
    
    enum ItemType
      {
    IT_NONE = QTreeWidgetItem::UserType, 
    IT_PROJECT,                          
    IT_FOLDER,                           
    IT_SEISMIC,                          
    IT_TABLE                             
      };
    
    enum ItemOptions
      {
    IOM_OPENABLE  = 1, 
    IOM_REMOVABLE = 2, 
    IOM_SELECTABLE = 4 
      };
    
    DbTreeWid(QWidget *parent = 0, Qt::WindowFlags f = 0);
    
    virtual ~DbTreeWid();
    
    
    void selectProject(const QString &project_str);
    
    
    void setRemoveObjectsMode(const bool mode);
    
    void setOpenObjectsMode(const bool mode);
    
    
  signals:
    
    void itemChanged(const int type, 
             const QString &project, 
             const QString &name);
    
    void errorOccurred(const int &ec); 
                     
  public slots:
    
    virtual void refreshTree();
    
    virtual void expandTree();
    
    virtual void collapseTree();
                         
  protected slots:
    
    virtual void showContextMenu(QPoint pos);
    
    virtual void openObject(QTreeWidgetItem*) 
    {/* do nothing by default. */}

    virtual void removeObject(QTreeWidgetItem*) 
    {/* do nothing by default. */}
    
    virtual void changeCurrentItem(QTreeWidgetItem* /*cur_item*/, 
                   QTreeWidgetItem* /*prev_item*/)
    {/* do nothing by default. */}
    
  protected:
    
    QTreeWidget* ptw() {return mp_tw;}
    
    const QTreeWidget* ptw() const {return mp_tw;}

    const QPoint& contextMenuPos() const {return m_cm_pos;} 


    virtual QTreeWidgetItem* newTreeWidgetItem(const ItemType type,
                           const QString &name,
                           QTreeWidgetItem *parent = NULL) const;

    QTreeWidgetItem* projectItem(QTreeWidgetItem *item) const;
    
    QTreeWidgetItem* projectItem(const QString &project_str) const;
    
    QString projectNameStr(QTreeWidgetItem *item) const;
    
    int itemOptions(const QTreeWidgetItem *item) const;
    
    void setItemOptions(QTreeWidgetItem *item, const int &mask) const;
    
    virtual bool event(QEvent *ev);
            
  private:
    
    // context menu actions
    QAction *mp_open_object_action;
    QAction *mp_refresh_tree_action;
    QAction *mp_expand_tree_action;
    QAction *mp_collapse_tree_action;
    QAction *mp_remove_object_action;
    
    QMenu *mp_context_menu;

    QPoint m_cm_pos;
    
    QTreeWidget *mp_tw;
    
    QString m_selected_project_str;
    
    bool m_open_objects_flag;
    
    bool m_remove_objects_flag;
    
    void prepare_context_menu();

  private slots:
    
    void openObjectFromContextMenu();
    
    void openObjectFromDoubleClick(QTreeWidgetItem *item, int column);

    void removeObjectFromContextMenu();
  };
  
}; // hds

#endif /* db_tree_wid.hpp */