ParticleTreeWidget.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 PARTICLETREEWIDGET_H
#define PARTICLETREEWIDGET_H

#include <QtGui/QTreeWidget>
#include <Phast.h>
#include <osg/ref_ptr>
#include <osg/Node>
#include <osg/Vec3>

#include <map>

#include "PhastNode.h"
class PhastScene;

// #define PARTICLETREEWIDGET_COLORS_AS_TEXT    // <--- show only id's instead of color

class ParticleTreeWidget : public QTreeWidget
{
        Q_OBJECT
public:
        
        enum ItemDataRoles //user roles must be >= 32
        {
                ROLE_OSG_NODE           =       100,
                ROLE_LINE_WIDTH         =       101
        };
        
        enum ColumnType
        {
                COL_INDEX = 0,
                COL_VISIBLE,
                COL_COLOR,
                
                // track momentum (if PaTrack::hasMom())
                COL_CHARGE,
                COL_MOMENTUM,

                // Rich information (accessed via PaPid and track)
                COL_RICHINFO,  // "yes"/"no" if rich info is available
                COL_TYPE,
                COL_MASS,
                
                // particle information (if associated particle is given)
                COL_PARTICLE,      // associated particle (if any)
                COL_NVERTEX,       // int  PaParticle::NVertex()
                COL_NCALORIM,      // int  PaParticle::NCalorim()
                COL_PARTICLE_TYPE, // PaParticle::PID(), particle ID in GEANT3 notation
                COL_IS_BEAM,       // bool PaParticle::IsBeam()
                COL_IS_MUPRIM,     // bool PaParticle::IsMuPrim()
                
                // additional track info
                COL_ZFIRST,
                COL_ZLAST,
                COL_LENGTH,
                COL_NHITS,
                COL_CHI2TOT,
                COL_MEANTIME,
                
                NUM_COLUMNS
        };
        
        ParticleTreeWidget( QWidget* parent = 0 );

        void updateContent( const PaEvent& event, PhastScene* scene );
        
signals:
        void updateNeeded();
        void particleActivated( int index );
        
public slots:
        void highlight( int uid );
        void autoResizeColumns()
        {
                for( int i=0; i < columnCount(); ++i )
                        resizeColumnToContents( i );
        }
        
protected slots:
        void updateItem( QTreeWidgetItem* item, int col );      

        // REMARK: only one of the following two functions get connected!
        void onItemActivated( QTreeWidgetItem* item, int col );
        void onCurrentItemChanged( QTreeWidgetItem* item, QTreeWidgetItem* );

protected:
        QColor getPIDColor( int PID );

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

private:
        QMap<int, QTreeWidgetItem*> m_UID2ItemMap;

        bool m_Lock; 
};

#endif /*PARTICLETREEWIDGET_H_*/