| 1 | /* | 
|---|
| 2 | * GLMoleculeObject.hpp | 
|---|
| 3 | * | 
|---|
| 4 | *  This is based on the Qt3D example "teaservice", specifically meshobject.h. | 
|---|
| 5 | * | 
|---|
| 6 | *  Created on: Aug 17, 2011 | 
|---|
| 7 | *      Author: heber | 
|---|
| 8 | */ | 
|---|
| 9 |  | 
|---|
| 10 | #ifndef GLMOLECULEOBJECT_HPP_ | 
|---|
| 11 | #define GLMOLECULEOBJECT_HPP_ | 
|---|
| 12 |  | 
|---|
| 13 | // include config.h | 
|---|
| 14 | #ifdef HAVE_CONFIG_H | 
|---|
| 15 | #include <config.h> | 
|---|
| 16 | #endif | 
|---|
| 17 |  | 
|---|
| 18 |  | 
|---|
| 19 | #include <QtCore/qobject.h> | 
|---|
| 20 | #include <QtGui/qevent.h> | 
|---|
| 21 |  | 
|---|
| 22 | #include <Qt3D/qglpainter.h> | 
|---|
| 23 | #include <Qt3D/qglabstractscene.h> | 
|---|
| 24 |  | 
|---|
| 25 | class QGLView; | 
|---|
| 26 | class QGLSceneNode; | 
|---|
| 27 | class GLMoleculeScene; | 
|---|
| 28 |  | 
|---|
| 29 | /** This class represents a single object within a molecule, e.g. atom or bond. | 
|---|
| 30 | * | 
|---|
| 31 | */ | 
|---|
| 32 | class GLMoleculeObject : public QObject | 
|---|
| 33 | { | 
|---|
| 34 | Q_OBJECT | 
|---|
| 35 |  | 
|---|
| 36 | //!> Allow it to call cleanMaterialMap() | 
|---|
| 37 | friend class GLWorldScene; | 
|---|
| 38 | public: | 
|---|
| 39 | explicit GLMoleculeObject(QGLSceneNode *mesh[], QObject *parent=0); | 
|---|
| 40 | explicit GLMoleculeObject(QGLSceneNode *mesh, QObject *parent=0); | 
|---|
| 41 | explicit GLMoleculeObject(QGLAbstractScene *scene, QObject *parent=0); | 
|---|
| 42 | virtual ~GLMoleculeObject(); | 
|---|
| 43 |  | 
|---|
| 44 | QVector3D position() const { return m_position; } | 
|---|
| 45 | void setPosition(const QVector3D& value) { m_position = value; } | 
|---|
| 46 |  | 
|---|
| 47 | void setScale(qreal value) { m_scaleX = m_scaleY = m_scaleZ = value; } | 
|---|
| 48 |  | 
|---|
| 49 | qreal scaleX() const { return m_scaleX; } | 
|---|
| 50 | void setScaleX(qreal value) { m_scaleX = value; } | 
|---|
| 51 |  | 
|---|
| 52 | qreal scaleY() const { return m_scaleY; } | 
|---|
| 53 | void setScaleY(qreal value) { m_scaleY = value; } | 
|---|
| 54 |  | 
|---|
| 55 | qreal scaleZ() const { return m_scaleZ; } | 
|---|
| 56 | void setScaleZ(qreal value) { m_scaleZ = value; } | 
|---|
| 57 |  | 
|---|
| 58 | qreal rotationAngle() const { return m_rotationAngle; } | 
|---|
| 59 | void setRotationAngle(qreal value) { m_rotationAngle = value; } | 
|---|
| 60 |  | 
|---|
| 61 | QVector3D rotationVector() const { return m_rotationVector; } | 
|---|
| 62 | void setRotationVector(const QVector3D& value) { m_rotationVector = value; } | 
|---|
| 63 |  | 
|---|
| 64 | QGLMaterial *material() const { return m_material; } | 
|---|
| 65 | void setMaterial(QGLMaterial *value) { m_material = value; } | 
|---|
| 66 |  | 
|---|
| 67 | QGLAbstractEffect *effect() const { return m_effect; } | 
|---|
| 68 | void setEffect(QGLAbstractEffect *value) { m_effect = value; } | 
|---|
| 69 |  | 
|---|
| 70 | int objectId() const { return m_objectId; } | 
|---|
| 71 | void setObjectId(int id) { m_objectId = id; } | 
|---|
| 72 |  | 
|---|
| 73 | bool selected() const { return m_selected; } | 
|---|
| 74 | void setSelected(bool value); | 
|---|
| 75 |  | 
|---|
| 76 | bool visible() const { return m_visible; } | 
|---|
| 77 | virtual void setVisible(bool value); | 
|---|
| 78 |  | 
|---|
| 79 | void initStaticMaterials(); | 
|---|
| 80 | virtual void initialize(QGLView *view, QGLPainter *painter); | 
|---|
| 81 | virtual void draw(QGLPainter *painter, const QVector4D &cameraPlane); | 
|---|
| 82 | void drawSelectionBox(QGLPainter *painter); | 
|---|
| 83 |  | 
|---|
| 84 | signals: | 
|---|
| 85 | void pressed(); | 
|---|
| 86 | void released(); | 
|---|
| 87 | void clicked(); | 
|---|
| 88 | void doubleClicked(); | 
|---|
| 89 | void hoverChanged(GLMoleculeObject *ob); | 
|---|
| 90 | void selectionChanged(); | 
|---|
| 91 | void changed(); | 
|---|
| 92 |  | 
|---|
| 93 | protected: | 
|---|
| 94 | bool event(QEvent *e); | 
|---|
| 95 |  | 
|---|
| 96 | static QGLMaterial* getMaterial(size_t); | 
|---|
| 97 | static void cleanMaterialMap(); | 
|---|
| 98 |  | 
|---|
| 99 | void updateMesh(QGLSceneNode *scene); | 
|---|
| 100 |  | 
|---|
| 101 | typedef std::map< size_t, QGLMaterial *> ElementMaterialMap; | 
|---|
| 102 | static ElementMaterialMap ElementNoMaterialMap; | 
|---|
| 103 |  | 
|---|
| 104 | public: | 
|---|
| 105 | enum{DETAIL_HIGHEST, DETAIL_HIGH, DETAIL_MEDIUM, DETAIL_LOW, DETAILTYPES_MAX} DetailType; | 
|---|
| 106 |  | 
|---|
| 107 | static QGLSceneNode *meshEmpty[DETAILTYPES_MAX]; | 
|---|
| 108 | static QGLSceneNode *meshSphere[DETAILTYPES_MAX]; | 
|---|
| 109 | static QGLSceneNode *meshCylinder[DETAILTYPES_MAX]; | 
|---|
| 110 |  | 
|---|
| 111 | protected: | 
|---|
| 112 |  | 
|---|
| 113 | static double detailMinDistance[DETAILTYPES_MAX]; | 
|---|
| 114 |  | 
|---|
| 115 | QGLSceneNode *m_mesh[DETAILTYPES_MAX]; | 
|---|
| 116 | enum enum_MeshFreeType { | 
|---|
| 117 | free_all, | 
|---|
| 118 | free_single, | 
|---|
| 119 | free_none | 
|---|
| 120 | } MeshFreeType; | 
|---|
| 121 | QGLAbstractScene *m_scene; | 
|---|
| 122 | QVector3D m_position; | 
|---|
| 123 | qreal m_scaleX; | 
|---|
| 124 | qreal m_scaleY; | 
|---|
| 125 | qreal m_scaleZ; | 
|---|
| 126 | qreal m_rotationAngle; | 
|---|
| 127 | QVector3D m_rotationVector; | 
|---|
| 128 | QGLMaterial *m_material; | 
|---|
| 129 | static QGLMaterial *m_hoverMaterial; | 
|---|
| 130 | static QGLMaterial *m_selectionMaterial; | 
|---|
| 131 | static QGLMaterial *m_selectionBoxMaterial; | 
|---|
| 132 | QGLAbstractEffect *m_effect; | 
|---|
| 133 | int m_objectId; | 
|---|
| 134 | bool m_hovering; | 
|---|
| 135 | bool m_selected; | 
|---|
| 136 | bool m_visible; | 
|---|
| 137 | }; | 
|---|
| 138 |  | 
|---|
| 139 |  | 
|---|
| 140 | #endif /* GLMOLECULEOBJECT_HPP_ */ | 
|---|