detector_tree.h

Go to the documentation of this file.

#ifndef detector_tree_H
#define detector_tree_H 1

// Qt headers
#include <QTreeWidgetItem>
#include <QAction>
#include <QContextMenuEvent>
#include <QtWidgets>

// G4 headers
#include "G4RunManager.hh"
#include "G4VisManager.hh"
#include "G4UImanager.hh"
#include "G4GDMLParser.hh"

// gemc headers
#include "detector.h"
#include "options.h"
#include "detector_editor.h"

// C++ headers
#include <string>
#include <map>
using namespace std;


// Class definition
class tree_item
{
    public:
        tree_item(){;}
        ~tree_item(){;}
        
    public:
        string volume;
        string mother;
        QTreeWidgetItem *treeItem;
        int scanned;
        int sensitive;
        int exist;
        int visible;
        int wiresolid;
};



class detector_tree : public QWidget
{
    // metaobject required for non-qt slots
    Q_OBJECT
    
    public:
        detector_tree(){;}
        detector_tree(QWidget *parent, goptions, G4RunManager*, map<string, detector>*, map<string, G4Material*>);
        ~detector_tree();
        
        goptions gemcOpt;
        map<string, detector> *Hall_Map;
        
        map<string, tree_item> tree_map;
        map<string, tree_item> read_geometry(QTreeWidget *motherWidget);
        map<string, G4Material*> MMats;
        
        QLinearGradient ActiveGrad;
        QLinearGradient SensitiveGrad;
        QLinearGradient NonActiveGrad;
        QLinearGradient NonVisibleGrad;
        
        QBrush ActiveBrush;
        QBrush SensitiveBrush;
        QBrush NonActiveBrush;
        QBrush NonVisibleBrush;
        
        descriptionTab  *dTab;
    
        QPushButton *writeToGDML, *writeToWRL, *showDetInNewWindow;
    
        
    private:
        QTreeWidget *treeWidget;
        QAction *Switch_visibility;
        QAction *Switch_visibility_daughters;
        QAction *Switch_wiresolid;
        QAction *Switch_color;
        QAction *Write_GDML_File;
        QAction *Write_WRL_File;
    
        // passing G4 managers to QT so we can delete them when QT quits
        // and can access directly the UImanager
        G4RunManager *runManager;
        G4UImanager  *UImanager;
        
    private slots:
        void switch_visibility();
        void switch_visibility_daughters();
        void switch_wiresolid();
        void switch_color();
        void show_detector();
    
        // writing to gdml
        void write_gdml_file(string);
        void set_gdml_name();
        void set_gdml_nameAll();
    
        // writing to wrl
        void write_wrl_file(string);
        void set_wrl_name();
        void set_wrl_nameAll();
    
    
        void inspectDetector();
        void change_placement();                

    private:
        void createActions();
    
};

#endif