SpikeGL/GLSpatialVis.h at master · dsdanielko/SpikeGL · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
#ifndef GLSpatialVis_H
#define GLSpatialVis_H

#include <QGLWidget>
#include <QColor>
#include <vector>
#include <QVariant>
#include <QImage>

#include "Vec.h"

class GLSpatialVis : public QGLWidget
{
    Q_OBJECT
public:

	enum Sel { Box = 0, Outline = 1, N_Sel };

    GLSpatialVis(QWidget *parent = 0);
    virtual ~GLSpatialVis();

    /// Reset the graph to default params, as if it were freshly constructed
    void reset();

	void setPoints(const QVector<Vec2> & points);
	void setColors(const QVector<Vec4f> & colors);

	Vec2f glyphSize() const { return glyph_size; }
	void setGlyphSize(Vec2f gs);

	enum GlyphType { Square=0, Circle, N_GlyphType };

	GlyphType glyphType() const { return glyph; }
	void setGlyphType(GlyphType g);

    const QColor & bgColor() const { return bg_Color; }
    const QColor & gridColor() const { return grid_Color; }
	void setBGColor(QColor c);
	void setGridColor(QColor c);

    unsigned numVGridLines() const { return nVGridLines; }
    void setNumVGridLines(unsigned);

    unsigned numHGridLines() const { return nHGridLines; }
    void setNumHGridLines(unsigned);

    bool autoUpdate() const { return auto_update; }
    void setAutoUpdate(bool b) { auto_update = b; }

    bool needsUpdateGL() const { return need_update; }

	void setSelectionRange(double begin_x, double end_x, double begin_y, double end_y, Sel = Box);
	void setSelectionEnabled(bool onoff, Sel = Box);
	bool isSelectionEnabled(Sel s = Box) const { return hasSelection[s]; }
	bool isSelectionVisible(Sel = Box) const;

	QVector<unsigned> selectAllGlyphsIntersectingRect(Vec2 corner1, Vec2 corner2, Sel = Box, Vec2 margin = Vec2());

	void setOverlay(const void *data, int w, int h, int fmt=GL_RGBA); ///< if set to a null, disable overlay
	void setOverlayAlpha(float alpha); ///< if set to 0, then disable overlay
	float overlayAlpha() const { return overlay_alpha; }

	void setXForm(float translate_x, float translate_y, float scale_x, float scale_y);
	void unsetXForm() { setXForm(0.f,0.f,1.f,1.f); xform_is_set = false; }
	bool xformIsSet() const { return xform_is_set; }
signals:
	/// like cursorOver(), except emitted x,y units are in window coordinates, not graph coordinates
	void cursorOverWindowCoords(int x, int y);
	/// like clicked(), except emitted x,y units are in window coordinates
	void clickedWindowCoords(int x, int y);
	void clickReleasedWindowCoords(int x, int y);
    /// for all the below: x is a value in range [0,1], y is a graph Y-pos in range [0,1]
    void cursorOver(double x, double y);
    void clicked(double x, double y); ///< this only emitted on Left mouse button clicks
    void clickReleased(double x, double y); ///< this only emitted on Right mouse button clicks
    void doubleClicked(double x, double y); ///< this only emitted on Left dbl-click

protected:
    void initializeGL();
    void resizeGL(int w, int h);
    void paintGL();

    Vec2 pos2Vec(const QPoint & pos);
    void mouseMoveEvent(QMouseEvent *evt);
    void mousePressEvent(QMouseEvent *evt);
    void mouseReleaseEvent(QMouseEvent *evt);
    void mouseDoubleClickEvent(QMouseEvent *evt);

    unsigned short & gridLineStipple() { return gridLineStipplePattern; }

private:
    void drawGrid() const;
    void drawPoints() const;
	void drawCircles() const;
	void drawSquares() const;
	void drawSelection() const;
	void drawOverlay();
	void drawBoundingBox() const;

	void updateColorBuf();
	void updateVertexBuf();

	int w_pix, h_pix;
    QColor bg_Color, grid_Color;
	QColor sel_Color[N_Sel];
	double point_size;
	Vec2f glyph_size;
	GlyphType glyph;
    unsigned nHGridLines, nVGridLines;
    unsigned short gridLineStipplePattern;
    QVector<Vec2> pointsBuf;
	QVector<Vec4f> colorsBuf;
	QVector<Vec2f> vbuf;///< scratch buff for vertices in squares mode, mostly
	QVector<Vec4f> cbuf;
    std::vector<Vec2> gridVs, gridHs;
    bool auto_update, need_update;
	double selx1[N_Sel],selx2[N_Sel],sely1[N_Sel],sely2[N_Sel];
	bool hasSelection[N_Sel];

	double timeNow;

	GLuint tex;
	const void * overlay;
	int overlay_width, overlay_height, overlay_fmt;
	float overlay_alpha;
	bool overlay_changed, overlay_subimg;

	Vec4f xform;
	bool xform_is_set;
};

#endif