1 # SPDX-License-Identifier: GPL-2.0-or-later
4 from mathutils
import Vector
5 from math
import cos
, sin
, pi
8 def create_widget(name
):
9 """Create an empty widget object and return the object"""
10 prefs
= bpy
.context
.preferences
.addons
["add_camera_rigs"].preferences
11 widget_prefix
= prefs
.widget_prefix
12 obj_name
= widget_prefix
+ name
13 scene
= bpy
.context
.scene
15 obj
= bpy
.data
.objects
.get(obj_name
)
18 mesh
= bpy
.data
.meshes
.new(obj_name
)
19 obj
= bpy
.data
.objects
.new(obj_name
, mesh
)
21 # Create a new collection for the widgets
22 collection_name
= prefs
.camera_widget_collection_name
23 coll
= bpy
.data
.collections
.get(collection_name
)
25 coll
= bpy
.data
.collections
.new(collection_name
)
26 scene
.collection
.children
.link(coll
)
27 coll
.hide_viewport
= True
28 coll
.hide_render
= True
31 coll
.objects
.link(obj
)
36 def create_corner_widget(name
, reverse
=False):
37 """Create a wedge-shaped widget"""
38 obj
= create_widget(name
)
39 if not obj
.data
.vertices
:
40 reverse
= -1 if reverse
else 1
41 verts
= (Vector((reverse
* 0.0, 0.0, 0.0)),
42 Vector((reverse
* 0.0, 1.0, 0.0)),
43 Vector((reverse
* -0.1, 1.0, 0.0)),
44 Vector((reverse
* -0.1, 0.1, 0.0)),
45 Vector((reverse
* -1.0, 0.1, 0.0)),
46 Vector((reverse
* -1.0, 0.0, 0.0)),
48 edges
= [(n
, (n
+1) % len(verts
)) for n
in range(len(verts
))]
51 mesh
.from_pydata(verts
, edges
, ())
56 def create_circle_widget(name
, radius
=1.0):
57 """Create a circle-shaped widget"""
58 obj
= create_widget(name
)
59 if not obj
.data
.vertices
:
62 for n
in range(vert_n
):
63 angle
= n
/ vert_n
* 2*pi
64 verts
.append(Vector((cos(angle
) * radius
,
65 sin(angle
) * radius
, 0.0)))
66 edges
= [(n
, (n
+1) % len(verts
)) for n
in range(len(verts
))]
69 mesh
.from_pydata(verts
, edges
, ())
74 def create_root_widget(name
):
75 """Create a compass-shaped widget"""
76 obj
= create_widget(name
)
77 if not obj
.data
.vertices
:
78 verts
= [(0.6307649612426758, 0.6271349787712097, 0.0),
79 (0.3413670063018799, 0.8205029964447021, 0.0),
80 (0.0, 0.8884050250053406, 0.0),
81 (-0.3413670063018799, 0.8205029964447021, 0.0),
82 (-0.6307649612426758, 0.6271349787712097, 0.0),
83 (-0.8241360187530518, 0.3377370238304138, 0.0),
84 (-0.8920379877090454, -0.003631560131907463, 0.0),
85 (0.8920379877090454, -0.003631379920989275, 0.0),
86 (0.824133038520813, 0.3377370238304138, 0.0),
87 (0.21458700299263, 1.5175920724868774, 0.0),
88 (-0.21458669006824493, 1.5175920724868774, 0.0),
89 (-0.21458700299263, 1.1372729539871216, 0.0),
90 (0.21458700299263, 1.1372729539871216, 0.0),
91 (-0.3977609872817993, 1.5175920724868774, 0.0),
92 (0.3977609872817993, 1.5175920724868774, 0.0),
93 (0.0, 2.002802848815918, 0.0),
94 (-0.8241360187530518, -0.3449999988079071, 0.0),
95 (0.8241360187530518, -0.3449999988079071, 0.0),
96 (0.6307649612426758, -0.6343979835510254, 0.0),
97 (0.3413670063018799, -0.8277660012245178, 0.0),
98 (0.0, -0.8956680297851562, 0.0),
99 (-0.3413670063018799, -0.8277660012245178, 0.0),
100 (-0.6307649612426758, -0.6343979835510254, 0.0),
101 (-2.0064330101013184, -0.003630870021879673, 0.0),
102 (-1.5212249755859375, 0.39413100481033325, 0.0),
103 (-1.5212249755859375, -0.4013940095901489, 0.0),
104 (-1.1409029960632324, 0.21095609664916992, 0.0),
105 (-1.1409029960632324, -0.2182179093360901, 0.0),
106 (-1.5212249755859375, -0.2182179093360901, 0.0),
107 (-1.5212249755859375, 0.21095609664916992, 0.0),
108 (1.5212249755859375, -0.2182179093360901, 0.0),
109 (1.5212249755859375, 0.21095609664916992, 0.0),
110 (1.1409029960632324, 0.21095609664916992, 0.0),
111 (1.1409029960632324, -0.2182179093360901, 0.0),
112 (1.5212249755859375, 0.39413100481033325, 0.0),
113 (1.5212249755859375, -0.4013940095901489, 0.0),
114 (2.0064330101013184, -0.0036309000570327044, 0.0),
115 (0.0, -2.0100629329681396, 0.0),
116 (-0.3977609872817993, -1.5248548984527588, 0.0),
117 (0.3977609872817993, -1.5248548984527588, 0.0),
118 (-0.21458669006824493, -1.144536018371582, 0.0),
119 (0.21458730101585388, -1.144536018371582, 0.0),
120 (0.21458730101585388, -1.5248548984527588, 0.0),
121 (-0.21458669006824493, -1.5248548984527588, 0.0)]
123 edges
= [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (7, 8), (0, 8),
124 (10, 11), (9, 12), (11, 12), (10, 13), (9, 14), (13, 15), (14, 15),
125 (16, 22), (17, 18), (18, 19), (19, 20), (20, 21), (21, 22), (7, 17),
126 (6, 16), (23, 24), (23, 25), (24, 29), (25, 28), (26, 27), (26, 29),
127 (27, 28), (31, 32), (30, 33), (32, 33), (31, 34), (30, 35), (34, 36),
128 (35, 36), (37, 38), (37, 39), (38, 43), (39, 42), (40, 41), (40, 43),
132 mesh
.from_pydata(verts
, edges
, [])
137 def create_camera_widget(name
):
138 """Create a camera control widget"""
139 obj
= create_widget(name
)
140 if not obj
.data
.vertices
:
141 verts
= [(0.275136, 0, -0.275136), (0.359483, 0, -0.148903),
142 (0.389102, 0, 0), (0.359483, 0, 0.148903),
143 (0.275136, 0, 0.275136), (0.148903, 0, 0.359483),
144 (-1.94818e-07, 0, 0.389102), (-1.17505e-07, 0, -0.389102),
145 (0.148903, 0, -0.359483), (0.663549, 0, -0.0936016),
146 (0.663549, 0, 0.0936016), (0.497656, 0, 0.0936016),
147 (0.497656, 0, -0.0936017), (0.663549, 0, 0.173501),
148 (0.663549, 0, -0.173501), (0.875195, 0, 0),
149 (-0.148903, 0, 0.359483), (-0.148903, 0, -0.359483),
150 (-0.275136, 0, -0.275136), (-0.359483, 0, -0.148903),
151 (-0.389102, 0, 0), (-0.359483, 0, 0.148903),
152 (-0.275136, 0, 0.275136), (1.03426e-07, 0, 0.875195),
153 (0.173502, 0, 0.663549), (-0.173501, 0, 0.663549),
154 (0.0936017, 0, 0.497656), (-0.0936016, 0, 0.497656),
155 (-0.0936016, 0, 0.663549), (0.0936017, 0, 0.663549),
156 (-0.0936015, 0, -0.663549), (0.0936017, 0, -0.663549),
157 (0.0936017, 0, -0.497656), (-0.0936016, 0, -0.497656),
158 (0.173502, 0, -0.663549), (-0.173501, 0, -0.663549),
159 (9.42269e-08, 0, -0.875195), (-0.875195, 0, 0),
160 (-0.663549, 0, 0.173501), (-0.663549, 0, -0.173502),
161 (-0.497656, 0, 0.0936015), (-0.497656, 0, -0.0936018),
162 (-0.663549, 0, -0.0936018), (-0.663549, 0, 0.0936015),
164 edges
= [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (7, 8), (0, 8),
165 (10, 11), (9, 12), (11, 12), (10, 13), (9, 14), (13, 15), (14, 15),
166 (16, 22), (17, 18), (18, 19), (19, 20), (20, 21), (21, 22), (7, 17),
167 (6, 16), (23, 24), (23, 25), (24, 29), (25, 28), (26, 29), (27, 28),
168 (31, 32), (30, 33), (32, 33), (31, 34), (30, 35), (34, 36), (35, 36),
169 (37, 38), (37, 39), (38, 43), (39, 42), (40, 41), (40, 43), (41, 42),
173 mesh
.from_pydata(verts
, edges
, [])
178 def create_camera_offset_widget(name
):
179 """Create a camera offset control widget"""
180 obj
= create_widget(name
)
181 if not obj
.data
.vertices
:
182 verts
= [(0.23286, 0, 0), (0.215135, 0, 0.0891117),
183 (0.164657, 0, 0.164657), (0.0891117, 0, 0.215135),
184 (0, 0, 0.23286), (-0.0891117, 0, 0.215135),
185 (-0.164657, 0, 0.164657), (-0.215135, 0, 0.0891117),
186 (-0.23286, 0, 0), (-0.215135, 0, -0.0891117),
187 (-0.164657, 0, -0.164657), (-0.0891117, 0, -0.215135),
188 (0, 0, -0.23286), (0.0891117, 0, -0.215135),
189 (0.164657, 0, -0.164657), (0.215135, 0, -0.0891117),
191 edges
= [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (6, 7), (7, 8),
192 (8, 9), (9, 10), (10, 11), (11, 12), (12, 13), (13, 14), (14, 15),
196 mesh
.from_pydata(verts
, edges
, [])
201 def create_aim_widget(name
):
202 """Create a camera aim widget"""
203 obj
= create_widget(name
)
204 if not obj
.data
.vertices
:
205 verts
= [(0.31008288264274597, 0.0, 0.31008288264274597),
206 (0.40514281392097473, 0.0, 0.1678156554698944),
207 (0.43852344155311584, 0.0, 0.0),
208 (0.40514281392097473, 0.0, -0.1678156852722168),
209 (0.31008288264274597, 0.0, -0.31008288264274597),
210 (0.1678156703710556, 0.0, -0.40514281392097473),
211 (-2.1956294915526087e-07, 0.0, -0.43852344155311584),
212 (-1.3243040086763358e-07, 0.0, 0.43852344155311584),
213 (0.16781596839427948, 0.0, 0.4051426947116852),
214 (0.7993937134742737, 0.0, 0.10549049079418182),
215 (0.7993937134742737, 0.0, -0.10549048334360123),
216 (0.9863580465316772, 0.0, -0.10549048334360123),
217 (0.9863580465316772, 0.0, 0.105490542948246),
218 (0.7993937134742737, 0.0, -0.19553886353969574),
219 (0.7993937134742737, 0.0, 0.19553886353969574),
220 (0.5608659386634827, 0.0, 0.0),
221 (-0.1678159236907959, 0.0, -0.4051426947116852),
222 (-0.16781575977802277, 0.0, 0.40514281392097473),
223 (-0.31008294224739075, 0.0, 0.3100828230381012),
224 (-0.40514281392097473, 0.0, 0.16781564056873322),
225 (-0.43852344155311584, 0.0, 0.0),
226 (-0.40514281392097473, 0.0, -0.16781572997570038),
227 (-0.3100828230381012, 0.0, -0.31008294224739075),
228 (5.8281088399780856e-08, 0.0, 0.5608659982681274),
229 (-0.19553889334201813, 0.0, 0.7993938326835632),
230 (0.1955389529466629, 0.0, 0.7993938326835632),
231 (-0.10549052804708481, 0.0, 0.9863581657409668),
232 (0.10549058020114899, 0.0, 0.9863581657409668),
233 (0.10549052804708481, 0.0, 0.7993938326835632),
234 (-0.10549046844244003, 0.0, 0.7993938326835632),
235 (0.10549034923315048, 0.0, -0.7993939518928528),
236 (-0.10549066960811615, 0.0, -0.7993939518928528),
237 (-0.10549074411392212, 0.0, -0.9863584041595459),
238 (0.10549039393663406, 0.0, -0.9863584041595459),
239 (-0.19553910195827484, 0.0, -0.7993938326835632),
240 (0.19553880393505096, 0.0, -0.7993940711021423),
241 (-1.4296951178494055e-07, 0.0, -0.5608659982681274),
242 (-0.5608660578727722, 0.0, 0.0),
243 (-0.7993939518928528, 0.0, -0.19553877413272858),
244 (-0.7993937134742737, 0.0, 0.19553901255130768),
245 (-0.9863580465316772, 0.0, -0.10549040883779526),
246 (-0.9863580465316772, 0.0, 0.10549063980579376),
247 (-0.7993938326835632, 0.0, 0.10549062490463257),
248 (-0.7993938326835632, 0.0, -0.10549038648605347),
249 (-0.12803608179092407, 0.0, -0.12803612649440765),
250 (-0.167287215590477, 0.0, -0.06929267197847366),
251 (-0.18107034265995026, 0.0, 0.0),
252 (-0.167287215590477, 0.0, 0.06929262727499008),
253 (-0.12803612649440765, 0.0, 0.12803608179092407),
254 (-0.06929267197847366, 0.0, 0.167287215590477),
255 (-0.06929274648427963, 0.0, -0.1672871708869934),
256 (0.06929276883602142, 0.0, 0.1672871708869934),
257 (-5.468173824851874e-08, 0.0, 0.18107034265995026),
258 (-9.065958295195742e-08, 0.0, -0.18107034265995026),
259 (0.06929264962673187, 0.0, -0.167287215590477),
260 (0.12803609669208527, 0.0, -0.12803609669208527),
261 (0.167287215590477, 0.0, -0.06929264962673187),
262 (0.18107034265995026, 0.0, 0.0),
263 (0.167287215590477, 0.0, 0.06929264217615128),
264 (0.12803609669208527, 0.0, 0.12803609669208527),
265 (-6.667435314966497e-08, 0.0, 0.060356780886650085),
266 (-7.866697160352487e-08, 0.0, -0.060356780886650085),
267 (-0.060356780886650085, 0.0, 0.0),
268 (0.060356780886650085, 0.0, 0.0)]
270 edges
= [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (7, 8), (0, 8),
271 (10, 11), (9, 12), (11, 12), (10, 13), (9, 14), (13, 15), (14, 15),
272 (16, 22), (17, 18), (18, 19), (19, 20), (20, 21), (21, 22), (7, 17),
273 (6, 16), (23, 24), (23, 25), (24, 29), (25, 28), (26, 29), (27, 28),
274 (31, 32), (30, 33), (32, 33), (31, 34), (30, 35), (34, 36), (35, 36),
275 (37, 38), (37, 39), (38, 43), (39, 42), (40, 41), (40, 43), (41, 42),
276 (50, 53), (49, 52), (44, 45), (45, 46), (46, 47), (47, 48), (48, 49),
277 (44, 50), (51, 59), (51, 52), (53, 54), (54, 55), (55, 56), (56, 57),
278 (57, 58), (58, 59), (27, 26), (60, 61), (62, 63)]
281 mesh
.from_pydata(verts
, edges
, [])