Add "GLTextItem" (#1776)
* Add GLTextItem
* Fixed DocString.
* Delete unnecessary function.
* Add `from .items.GLTextItem import *`
* Add an example.
* [Combines two `isinstance()` into one.](https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633046120)
* [Combines two `isinstance()` into one.](https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633046120)
* [The long code has been broken up into separate lines.](https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633046234)
* [Moved `pos`, `color`, `text`, and `font` to the `__init__` method.](https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633047216)
* Add `import initExample` and fix `mkQApp().exec_()` to `pg.exec()` (https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633046878, https://github.com/pyqtgraph/pyqtgraph/pull/1776#discussion_r633046781)
* Fix `pg.exec()` to `pg.mkQApp().exec()`
* Revert "Fix `pg.exec()` to `pg.mkQApp().exec()`"
This reverts commit 67d397d803
.
* Remove type-hints.
* Fix `glColor4d(float(self.color[0]), float(self.color[1]), float(self.color[2]), float(self.color[3]))` to `glColor4d(*self.color)`
* Add `value = fn.glColor(value)`
* Remove debug print.
* Add GLGridItem and GLAxisItem
* Remove if-check for "color" argument
* Draw text without using GLUT.
* Divide the text position by the device pixel ratio
* Fixed bare exceptions to ValueError and TypeError.
* Add 'GLTextItem.py' to utils.py.
* Fixed a bare exception to ArgumentErrror.
* Add `__all__ = ['GLTextItem']`
This commit is contained in:
parent
e206ea5ae9
commit
f13002b251
|
@ -0,0 +1,35 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
Simple examples demonstrating the use of GLTextItem.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
import initExample
|
||||||
|
|
||||||
|
import pyqtgraph as pg
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui, mkQApp
|
||||||
|
import pyqtgraph.opengl as gl
|
||||||
|
|
||||||
|
app = mkQApp("GLTextItem Example")
|
||||||
|
|
||||||
|
gvw = gl.GLViewWidget()
|
||||||
|
gvw.show()
|
||||||
|
gvw.setWindowTitle('pyqtgraph example: GLTextItem')
|
||||||
|
|
||||||
|
griditem = gl.GLGridItem()
|
||||||
|
griditem.setSize(10, 10)
|
||||||
|
griditem.setSpacing(1, 1)
|
||||||
|
gvw.addItem(griditem)
|
||||||
|
|
||||||
|
axisitem = gl.GLAxisItem()
|
||||||
|
gvw.addItem(axisitem)
|
||||||
|
|
||||||
|
txtitem1 = gl.GLTextItem(pos=(0.0, 0.0, 0.0), text='text1')
|
||||||
|
gvw.addItem(txtitem1)
|
||||||
|
|
||||||
|
txtitem2 = gl.GLTextItem()
|
||||||
|
txtitem2.setData(pos=(1.0, -1.0, 2.0), color=(127, 255, 127, 255), text='text2')
|
||||||
|
gvw.addItem(txtitem2)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
pg.exec()
|
|
@ -69,6 +69,7 @@ examples = OrderedDict([
|
||||||
('Line Plot', 'GLLinePlotItem.py'),
|
('Line Plot', 'GLLinePlotItem.py'),
|
||||||
('Mesh', 'GLMeshItem.py'),
|
('Mesh', 'GLMeshItem.py'),
|
||||||
('Image', 'GLImageItem.py'),
|
('Image', 'GLImageItem.py'),
|
||||||
|
('Text', 'GLTextItem.py'),
|
||||||
])),
|
])),
|
||||||
('Widgets', OrderedDict([
|
('Widgets', OrderedDict([
|
||||||
('PlotWidget', 'PlotWidget.py'),
|
('PlotWidget', 'PlotWidget.py'),
|
||||||
|
|
|
@ -14,6 +14,7 @@ from .items.GLImageItem import *
|
||||||
from .items.GLSurfacePlotItem import *
|
from .items.GLSurfacePlotItem import *
|
||||||
from .items.GLBoxItem import *
|
from .items.GLBoxItem import *
|
||||||
from .items.GLVolumeItem import *
|
from .items.GLVolumeItem import *
|
||||||
|
from .items.GLTextItem import *
|
||||||
|
|
||||||
from .MeshData import MeshData
|
from .MeshData import MeshData
|
||||||
## for backward compatibility:
|
## for backward compatibility:
|
||||||
|
|
|
@ -0,0 +1,97 @@
|
||||||
|
from OpenGL.GL import *
|
||||||
|
import numpy as np
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui
|
||||||
|
from pyqtgraph.opengl.GLGraphicsItem import GLGraphicsItem
|
||||||
|
import pyqtgraph.functions as fn
|
||||||
|
|
||||||
|
__all__ = ['GLTextItem']
|
||||||
|
|
||||||
|
class GLTextItem(GLGraphicsItem):
|
||||||
|
"""Draws text in 3D."""
|
||||||
|
|
||||||
|
def __init__(self, **kwds):
|
||||||
|
"""All keyword arguments are passed to setData()"""
|
||||||
|
GLGraphicsItem.__init__(self)
|
||||||
|
glopts = kwds.pop('glOptions', 'additive')
|
||||||
|
self.setGLOptions(glopts)
|
||||||
|
|
||||||
|
self.pos = np.array([0.0, 0.0, 0.0])
|
||||||
|
self.color = QtCore.Qt.white
|
||||||
|
self.text = ''
|
||||||
|
self.font = QtGui.QFont('Helvetica', 16)
|
||||||
|
|
||||||
|
self.setData(**kwds)
|
||||||
|
|
||||||
|
def setData(self, **kwds):
|
||||||
|
"""
|
||||||
|
Update the data displayed by this item. All arguments are optional;
|
||||||
|
for example it is allowed to update text while leaving colors unchanged, etc.
|
||||||
|
|
||||||
|
==================== ==================================================
|
||||||
|
**Arguments:**
|
||||||
|
------------------------------------------------------------------------
|
||||||
|
pos (3,) array of floats specifying text location.
|
||||||
|
color QColor or array of ints [R,G,B] or [R,G,B,A]. (Default: Qt.white)
|
||||||
|
text String to display.
|
||||||
|
font QFont (Default: QFont('Helvetica', 16))
|
||||||
|
==================== ==================================================
|
||||||
|
"""
|
||||||
|
args = ['pos', 'color', 'text', 'font']
|
||||||
|
for k in kwds.keys():
|
||||||
|
if k not in args:
|
||||||
|
raise ArgumentError('Invalid keyword argument: %s (allowed arguments are %s)' % (k, str(args)))
|
||||||
|
for arg in args:
|
||||||
|
if arg in kwds:
|
||||||
|
value = kwds[arg]
|
||||||
|
if arg == 'pos':
|
||||||
|
if isinstance(value, np.ndarray):
|
||||||
|
if value.shape != (3,):
|
||||||
|
raise ValueError('"pos.shape" must be (3,).')
|
||||||
|
elif isinstance(value, (tuple, list)):
|
||||||
|
if len(value) != 3:
|
||||||
|
raise ValueError('"len(pos)" must be 3.')
|
||||||
|
elif arg == 'color':
|
||||||
|
value = fn.mkColor(value)
|
||||||
|
elif arg == 'font':
|
||||||
|
if isinstance(value, QtGui.QFont) is False:
|
||||||
|
raise TypeError('"font" must be QFont.')
|
||||||
|
setattr(self, arg, value)
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def paint(self):
|
||||||
|
if len(self.text) < 1:
|
||||||
|
return
|
||||||
|
self.setupGLState()
|
||||||
|
|
||||||
|
modelview = glGetDoublev(GL_MODELVIEW_MATRIX)
|
||||||
|
projection = glGetDoublev(GL_PROJECTION_MATRIX)
|
||||||
|
viewport = glGetIntegerv(GL_VIEWPORT)
|
||||||
|
|
||||||
|
text_pos = self.__project(self.pos, modelview, projection, viewport)
|
||||||
|
text_pos[1] = viewport[3] - text_pos[1]
|
||||||
|
|
||||||
|
text_pos /= self.view().devicePixelRatio()
|
||||||
|
|
||||||
|
painter = QtGui.QPainter(self.view())
|
||||||
|
painter.setPen(self.color)
|
||||||
|
painter.setFont(self.font)
|
||||||
|
painter.setRenderHints(QtGui.QPainter.Antialiasing | QtGui.QPainter.TextAntialiasing)
|
||||||
|
painter.drawText(text_pos[0], text_pos[1], self.text)
|
||||||
|
painter.end()
|
||||||
|
|
||||||
|
def __project(self, obj_pos, modelview, projection, viewport):
|
||||||
|
obj_vec = np.append(np.array(obj_pos), [1.0])
|
||||||
|
|
||||||
|
view_vec = np.matmul(modelview.T, obj_vec)
|
||||||
|
proj_vec = np.matmul(projection.T, view_vec)
|
||||||
|
|
||||||
|
if proj_vec[3] == 0.0:
|
||||||
|
return
|
||||||
|
|
||||||
|
proj_vec[0:3] /= proj_vec[3]
|
||||||
|
|
||||||
|
return np.array([
|
||||||
|
viewport[0] + (1.0 + proj_vec[0]) * viewport[2] / 2.0,
|
||||||
|
viewport[1] + (1.0 + proj_vec[1]) * viewport[3] / 2.0,
|
||||||
|
(1.0 + proj_vec[2]) / 2.0
|
||||||
|
])
|
Loading…
Reference in New Issue