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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
# Copyright 2004-2009 Nanorex, Inc. See LICENSE file for details.
"""
GLSphereBuffer.py -- Subclass of GLPrimitiveBuffer for sphere primitives.
@author: Russ
@version: $Id$
@copyright: 2004-2009 Nanorex, Inc. See LICENSE file for details.
History:
Originally written by Russ Fish; designed together with Bruce Smith.
================================================================
See design comments on:
* GL contexts, CSDLs and DrawingSet in DrawingSet.py
* TransformControl in TransformControl.py
* VBOs, IBOs, and GLPrimitiveBuffer in GLPrimitiveBuffer.py
* GLPrimitiveSet in GLPrimitiveSet in GLPrimitiveSet.py
"""
from graphics.drawing.GLPrimitiveBuffer import GLPrimitiveBuffer, HunkBuffer
from geometry.VQT import V, A
class GLSphereBuffer(GLPrimitiveBuffer):
"""
Encapsulate VBO/IBO handles for a batch of spheres.
See doc for common code in the base class, GLPrimitiveBuffer.
Draws a bounding-box of quads (or a single billboard quad) to a custom
sphere shader for each sphere primitive, along with control attribute data.
"""
def __init__(self, shaderGlobals):
"""
@param shaderGlobals: the instance of class ShaderGlobals
we will be associated with.
"""
super(GLSphereBuffer, self).__init__( shaderGlobals )
# Per-vertex attribute hunk VBOs that are specific to the sphere shader.
# Combine centers and radii into a 4-element vec4 attribute VBO. (Each
# attribute slot takes 4 floats, no matter how many of them are used.)
shader = self.shader
self.ctrRadHunks = HunkBuffer(shader, "center_rad", self.nVertices, 4)
self.hunkBuffers += [self.ctrRadHunks]
return
def addSpheres(self, centers, radii, colors, transform_ids, glnames):
"""
Sphere centers must be a list of VQT points.
Lists or single values may be given for the attributes of the spheres
(radii, colors, transform_ids, and selection glnames). A single value
is replicated for the whole batch. The lengths of attribute lists must
match the center points list.
radii and colors are required. Radii are numbers. Colors are tuples of
components: (R, G, B) or (R, G, B, A).
transform_ids may be None for centers in global modeling coordinates.
glnames may be None if mouseover drawing will not be done. glnames are
32-bit integers, allocated sequentially and associated with selected
objects in a global object dictionary
The return value is a list of allocated primitive IDs for the spheres.
"""
nSpheres = len(centers)
if type(radii) == type([]):
assert len(radii) == nSpheres
else:
radii = nSpheres * [float(radii)]
pass
if type(colors) == type([]):
assert len(colors) == nSpheres
colors = [self.color4(colors) for color in colors]
else:
colors = nSpheres * [self.color4(colors)]
pass
if type(transform_ids) == type([]):
assert len(transform_ids) == nSpheres
else:
if transform_ids is None:
# This bypasses transform logic in the shader for this sphere.
transform_ids = -1
pass
transform_ids = nSpheres * [transform_ids]
pass
if type(glnames) == type([]):
assert len(glnames) == nSpheres
else:
if glnames is None:
glnames = 0
pass
glnames = nSpheres * [glnames]
pass
newIDs = self.newPrimitives(nSpheres)
for (newID, ctr, radius, color, transform_id, glname) in \
zip(newIDs, centers, radii, colors, transform_ids, glnames):
# Combine the center and radius into one vertex attribute.
ctrRad = V(ctr[0], ctr[1], ctr[2], radius)
self.ctrRadHunks.setData(newID, ctrRad)
self.colorHunks.setData(newID, color)
if self.transform_id_Hunks:
self.transform_id_Hunks.setData(newID, transform_id)
# Break the glname into RGBA pixel color components, 0.0 to 1.0 .
# (Per-vertex attributes are all multiples (1-4) of Float32.)
##rgba = [(glname >> bits & 0xff) / 255.0 for bits in range(24,-1,-8)]
## Temp fix: Ignore the last byte, which always comes back 255 on Windows.
rgba = [(glname >> bits & 0xff) / 255.0 for bits in range(16,-1,-8)]+[0.0]
self.glname_color_Hunks.setData(newID, rgba)
continue
return newIDs
def grab_untransformed_data(self, primID): #bruce 090223
"""
"""
ctrRad = self.ctrRadHunks.getData(primID)
return A(ctrRad[:3]), ctrRad[3]
def store_transformed_primitive(self, primID, untransformed_data, transform):
#bruce 090223
"""
@param untransformed_data: something returned earlier from
self.grab_untransformed_data(primID)
"""
# todo: heavily optimize this, by combining multiple values of
# untransformed_data into a single array, like Chunk.baseposn
# (and/or by coding it in C)
point, radius = untransformed_data
point = transform.applyToPoint(point)
ctrRad = V(point[0], point[1], point[2], radius)
self.ctrRadHunks.setData(primID, ctrRad)
return
pass # End of class GLSphereBuffer.
|
