SetupTreeWidget.h

Go to the documentation of this file.

/******************************************************************************
 *
 *  Copyright (C) 2008 Boris Iven, Max Hermann
 *
 *  This file is part of the "CEDRIC Event Display" application.
 *
 *****************************************************************************/

#ifndef SETUPTREEWIDGET_H
#define SETUPTREEWIDGET_H

#include <QtGui/QTreeWidget>
#include <QDomDocument>
#include <osg/Node>
#include <vector>

class QAction;
class PhastNode;

class SetupTreeWidget : public QTreeWidget
{
        Q_OBJECT

public:
        enum ItemTypes
        {
                //bit flags for QWidgetItem (must be >= 1000)
                ROOT_NODE               =               1024,
                INNER_NODE              =               2048,
                LEAF_NODE               =               4096,
                DETECTOR                =               8192,
                RICH                    =               16384,
                CALORIMETER             =               32768,
                MAGNET                  =               65536,
                TARGET                  =               131072
        };
        
        enum ItemDataRoles //user roles must be >= 32
        {
                ROLE_OSG_NODE   =       100,
                ROLE_DOM_NODE   =       101,
                ROLE_COLOR_TYPE =       102
        };
        
        enum ColumnType
        {
                COL_NAME = 0,
                COL_VISIBLE = 1,
                COL_DBG_OSG = 4,
                COL_DBG_DOM = 5,
                COL_COLOR = 2,
                COL_ALPHA = 3
        };
        
        enum ColorType
        {
                UNKNOWN = 0,
                MIXED = 1,
                RGB = 2
        };
        SetupTreeWidget( QWidget* parent=0 );

        //QDomDocument* getDomDocument() { loadConfiguration(); return domDocument; }
        QDomDocument* getDomDocument() { return domDocument; }
        
signals:

        void doReload();
        
        void updateNeeded();
        
public slots:

        bool loadConfiguration();
        void saveConfiguration();
        //~ void manageConfigurations(); // OBSOLETE
        void addGroup();
        void deleteGroup();
        void updateContextMenu(QTreeWidgetItem* current, int col);
        void updateItem(QTreeWidgetItem* item, int col);
        void changeSetupAlpha(int alpha);
        
        void highlight(int uid);
        void autoResizeColumns()
        {
                for( int i=0; i < columnCount(); ++i )
                        resizeColumnToContents( i );
        }

public:
        
        
        void setInitialBuild(bool state = false);
        void setOsgRootNode(osg::ref_ptr<osg::Node> rootNode){setupOsgRootNode = rootNode;}
        bool loadConfiguration(QString filename);
        void createUIDMapEntry(int uid, QTreeWidgetItem* item);
        
        //void select(QList<int> uids);
        
protected:
        void createActions();
        void createContextMenu();
        
        //Drag & Drop
        //void mouseMoveEvent(QMouseEvent* event);
        void mousePressEvent(QMouseEvent* event);
        void startDrag(Qt::DropActions supportedActions);
        //void dragEnterEvent(QDragEnterEvent* event);
        //void dragMoveEvent(QDragMoveEvent *event);
        void dropEvent(QDropEvent* event);

private slots:
        void highlight(QTreeWidgetItem* item, QTreeWidgetItem* unHighlightMe);

        
private:
        QAction* loadAction;
        QAction* saveAction;
        QAction* manageAction;
    QAction* newGroupAction;
    QAction* deleteGroupAction;
        QDomDocument* domDocument;
        
        QMap<int, QTreeWidgetItem*> UID2Item;
        
        osg::ref_ptr<osg::Node> setupOsgRootNode;
        
        //Drag & Drop
        QTreeWidgetItem* draggedItem;
        QTreeWidgetItem* oldParentItem;
        
        struct DragDropSet
        {
                QTreeWidgetItem* draggedItem;
                QTreeWidgetItem* oldParentItem;
                osg::Node::ParentList osgParentsBeforeDrag;
        };
        
        QList<DragDropSet> dragDropList;
        
        osg::Node::ParentList osgParentsBeforeDrag;
        QDomElement domParentBeforeDrag;
        
        
        void changeNodeColor(PhastNode* osgNode, QTreeWidgetItem* item); //remove ?
        void changeNodeAlpha(osg::ref_ptr<osg::Node> osgNode, int alpha); //remove ?
        bool initialBuild; //to prevent updateItem to be called during initial tree building
        bool traversingUpwards;
        
        bool updateMultiStateItems(QTreeWidgetItem* root, bool upwardsOnly = false); 
        
        
        //xml generation
        void createDomTree(QDomElement rootElement, QTreeWidgetItem* rootItem);
};

#endif