81 |
82 |
83 |
190 |
191 |
214 |
215 |
239 |
240 |
262 |
263 |
284 |
285 |
84 |
99 |
100 | ai03 Plate Generator
85 |An advanced plate generator which doesn't result in 13.99999mm cutouts.
86 |Also fillets the cutouts so you don't have to waste hours later. Enjoy.
87 | 88 | {% with messages = get_flashed_messages(with_categories=true) %} 89 | {% if messages %} 90 | {% for message in messages %} 91 |
92 |
94 | {% endfor %}
95 | {% endif %}
96 | {% endwith %}
97 |
98 | Error: {{ message[1] }}
93 |
101 |
102 |
187 |
188 |
189 |
286 |
299 |
300 |
301 |
302 |
319 |
320 |
321 |
--------------------------------------------------------------------------------
/plategen2.py:
--------------------------------------------------------------------------------
1 | #=================================#
2 | # Plate Generator 2 #
3 | #=================================#
4 |
5 | # By ai03
6 | # Credits to
7 | # Amtra5, Mxblue, Bakingpy,
8 | # Senter, Pwner, Kevinplus,
9 | # Peioris, Deskthority Wiki,
10 | # and any others I may have missed
11 |
12 | # Automated production of high-end mechanical keyboard plate data
13 | # No float rounding issues, pre-filleted corners, ready for production.
14 |
15 | # EXTRA SUPPORTED JSON FIELDS:
16 | # In addition to the KLE fields such as w: for width,
17 | # _rs: Rotate the stabilizers
18 | # _rc: Rotate switch cutout
19 |
20 | #=================================#
21 | # #
22 | #=================================#
23 |
24 | # Import necessities
25 | import ezdxf
26 | import sys
27 | import json5
28 | import argparse
29 |
30 | from mpmath import *
31 | from decimal import *
32 |
33 | class PlateGenerator(object):
34 |
35 | #init
36 | def __init__(self, arg_ct, arg_cr, arg_st, arg_sr, arg_at, arg_ar, arg_uw, arg_uh, arg_db):
37 |
38 | # Set up decimal and mpmath
39 | getcontext().prec = 50
40 | mp.dps = 50
41 | mp.pretty = True
42 |
43 | # Create blank dxf workspace
44 | self.plate = ezdxf.new(dxfversion='AC1024')
45 | self.modelspace = self.plate.modelspace()
46 |
47 | # Cutout type: mx, alps
48 | self.cutout_type = arg_ct
49 |
50 | # Cutout radius: The fillet radius ( 0 <= x <= 1/2 cutout width or height )
51 | try:
52 | self.cutout_radius = Decimal(arg_cr)
53 | except:
54 | raise ValueError
55 |
56 | # Stab type: mx-simple, large-cuts, alps-aek, alps-at101
57 | self.stab_type = arg_st
58 |
59 | # Stab radius: The fillet radius for stab cutouts ( 0 <= x <= 1 )
60 | try:
61 | self.stab_radius = Decimal(arg_sr)
62 | except:
63 | raise ValueError
64 |
65 | # Acoustic cuts: The cutouts typically found on high end plates beside the switches.
66 | # This script only handles the thin short cuts vertically beside each switch cut, not the large ones, i.e. between fn row and alphas.
67 | # none = disabled, typical = 1.5-1.75U only, extreme = On 1.5-2.75U
68 | self.acoustics_type = arg_at
69 |
70 | # Acoustic radius: Fillet radius for cuts mentioned above.
71 | try:
72 | self.acoustics_radius = Decimal(arg_ar)
73 | except:
74 | raise ValueError
75 |
76 | # Unit size (i.e. 1U = 19.05mm). ( 0 <= x <= inf, cap at 1000 for now )
77 | try:
78 | self.unit_width = Decimal(arg_uw)
79 | except:
80 | raise ValueError
81 | try:
82 | self.unit_height = Decimal(arg_uh)
83 | except:
84 | raise ValueError
85 |
86 |
87 | #== Debug parameters ==#
88 |
89 | # The debug_log flag makes the script spit out debug info and NOT write out the plate file.
90 | self.debug_log = arg_db
91 |
92 | # Other variables
93 | self.cutout_width = Decimal("0");
94 | self.cutout_height = Decimal("0");
95 |
96 | #=================================#
97 | # Classes #
98 | #=================================#
99 |
100 | class Switch:
101 |
102 | def __init__(self):
103 | # These fields correspond to the respective kle data
104 |
105 | # Universal fields
106 | self.width = 1 # Width
107 | self.height = 1 # Height
108 | self.width2 = 1 # Width2: For oddly sized keys such as ISO, stepped
109 | self.height2 = 1 # Height2: For oddly sized keys such as ISO, stepped
110 | self.rotated_zone = False # Rotation zone for rotation zone (coords handled differently)
111 |
112 | # Coord fields (in mm) generated based on variables
113 | self.coord_x = 0;
114 | self.coord_y = 0;
115 |
116 | # Notice about offset fields (x: and y: fields):
117 | # They are calculated into the x/y_var or x_y_offset respectively during switch creation
118 |
119 | # Non-rotated fields
120 | self.pos_x = x_var # Implied coord X (Does not have a field in KLE)
121 | self.pos_y = y_var # Implied coord Y (Does not have a field in KLE)
122 |
123 | # Rotated fields
124 | self.rot_anchor_x = 0 # RotationX: Rotation anchor
125 | self.rot_anchor_y = 0 # RotationY: Rotation anchor
126 | self.rot_angle = 0 # Rotation: Angle
127 | self.rot_x_offset = 0 # x: X offset from anchor
128 | self.rot_y_offset = 0 # y: Y offset from anchor
129 |
130 | # Custom fields unique to plategen
131 | self.cutout_angle = 0 # Custom field for switch-independent cutout rotation
132 | self.stab_angle = 0 # Custom field for switch-independent stabilizer rotation
133 |
134 | #=================================#
135 | # DXF Blocks #
136 | #=================================#
137 |
138 | # These basically act as stamps.
139 | # Saves from having to write each and every entity individually, as was done before.
140 |
141 | def build_blocks(self):
142 |
143 | # Switch cutout
144 | block_switch_cutout = self.plate.blocks.new(name='SWITCH_CUTOUT')
145 | switch_half_width = self.cutout_width / 2
146 | switch_half_height = self.cutout_height / 2
147 | block_switch_cutout.add_line((-switch_half_width + self.cutout_radius, -switch_half_height), (switch_half_width - self.cutout_radius, -switch_half_height)) # Top
148 | block_switch_cutout.add_line((-switch_half_width + self.cutout_radius, switch_half_height), (switch_half_width - self.cutout_radius, switch_half_height)) # Bottom
149 | block_switch_cutout.add_line((-switch_half_width, -switch_half_height + self.cutout_radius), (-switch_half_width, switch_half_height - self.cutout_radius)) # Left
150 | block_switch_cutout.add_line((switch_half_width, -switch_half_height + self.cutout_radius), (switch_half_width, switch_half_height - self.cutout_radius)) # Right
151 | block_switch_cutout.add_arc((switch_half_width - self.cutout_radius, -switch_half_height + self.cutout_radius), self.cutout_radius, 0, 90) # Top right
152 | block_switch_cutout.add_arc((-switch_half_width + self.cutout_radius, -switch_half_height + self.cutout_radius), self.cutout_radius, 90, 180) # Top left
153 | block_switch_cutout.add_arc((-switch_half_width + self.cutout_radius, switch_half_height - self.cutout_radius), self.cutout_radius, 180, 270) # Bottom left
154 | block_switch_cutout.add_arc((switch_half_width - self.cutout_radius, switch_half_height - self.cutout_radius), self.cutout_radius, 270, 360) # Bottom right
155 |
156 | # Stab cutout for mx-simple
157 | # Fairly tight rectangles with width based on official spec
158 | stab_mx_simple_cutout = self.plate.blocks.new(name='STAB_MX_SIMPLE_CUTOUT')
159 | stab_mx_simple_cutout.add_line((Decimal('-3.375') + self.stab_radius, Decimal('6')), (Decimal('3.375') - self.stab_radius, Decimal('6')))
160 | stab_mx_simple_cutout.add_line((Decimal('-3.375') + self.stab_radius, Decimal('-8')), (Decimal('3.375') - self.stab_radius, Decimal('-8')))
161 | stab_mx_simple_cutout.add_line((Decimal('-3.375'), Decimal('6') - self.stab_radius), (Decimal('-3.375'), Decimal('-8') + self.stab_radius))
162 | stab_mx_simple_cutout.add_line((Decimal('3.375'), Decimal('6') - self.stab_radius), (Decimal('3.375'), Decimal('-8') + self.stab_radius))
163 | stab_mx_simple_cutout.add_arc((Decimal('-3.375') + self.stab_radius, Decimal('6') - self.stab_radius), self.stab_radius, 90, 180)
164 | stab_mx_simple_cutout.add_arc((Decimal('3.375') - self.stab_radius, Decimal('6') - self.stab_radius), self.stab_radius, 0, 90)
165 | stab_mx_simple_cutout.add_arc((Decimal('-3.375') + self.stab_radius, Decimal('-8') + self.stab_radius), self.stab_radius, 180, 270)
166 | stab_mx_simple_cutout.add_arc((Decimal('3.375') - self.stab_radius, Decimal('-8') + self.stab_radius), self.stab_radius, 270, 360)
167 |
168 | # Stab cutout for large-cuts
169 | # Large, spacious 15x7 cutouts, the recommended standard
170 | stab_large_cuts_cutout = self.plate.blocks.new(name='STAB_LARGE_CUTS_CUTOUT')
171 | stab_large_cuts_cutout.add_line((Decimal('-3.5') + self.stab_radius, Decimal('6')), (Decimal('3.5') - self.stab_radius, Decimal('6')))
172 | stab_large_cuts_cutout.add_line((Decimal('-3.5') + self.stab_radius, Decimal('-9')), (Decimal('3.5') - self.stab_radius, Decimal('-9')))
173 | stab_large_cuts_cutout.add_line((Decimal('-3.5'), Decimal('6') - self.stab_radius), (Decimal('-3.5'), Decimal('-9') + self.stab_radius))
174 | stab_large_cuts_cutout.add_line((Decimal('3.5'), Decimal('6') - self.stab_radius), (Decimal('3.5'), Decimal('-9') + self.stab_radius))
175 | stab_large_cuts_cutout.add_arc((Decimal('-3.5') + self.stab_radius, Decimal('6') - self.stab_radius), self.stab_radius, 90, 180)
176 | stab_large_cuts_cutout.add_arc((Decimal('3.5') - self.stab_radius, Decimal('6') - self.stab_radius), self.stab_radius, 0, 90)
177 | stab_large_cuts_cutout.add_arc((Decimal('-3.5') + self.stab_radius, Decimal('-9') + self.stab_radius), self.stab_radius, 180, 270)
178 | stab_large_cuts_cutout.add_arc((Decimal('3.5') - self.stab_radius, Decimal('-9') + self.stab_radius), self.stab_radius, 270, 360)
179 |
180 | # Alps cutouts
181 | # Rectangles 2.67 wide, 5.21 high.
182 | stab_alps_cutout = self.plate.blocks.new(name='STAB_ALPS_CUTOUT')
183 | stab_alps_cutout.add_line((Decimal('-1.335') + self.stab_radius, Decimal('-3.875')), (Decimal('1.335') - self.stab_radius, Decimal('-3.875')))
184 | stab_alps_cutout.add_line((Decimal('-1.335') + self.stab_radius, Decimal('-9.085')), (Decimal('1.335') - self.stab_radius, Decimal('-9.085')))
185 | stab_alps_cutout.add_line((Decimal('-1.335'), Decimal('-3.875') - self.stab_radius), (Decimal('-1.335'), Decimal('-9.085') + self.stab_radius))
186 | stab_alps_cutout.add_line((Decimal('1.335'), Decimal('-3.875') - self.stab_radius), (Decimal('1.335'), Decimal('-9.085') + self.stab_radius))
187 | stab_alps_cutout.add_arc((Decimal('-1.335') + self.stab_radius, Decimal('-3.875') - self.stab_radius), self.stab_radius, 90, 180)
188 | stab_alps_cutout.add_arc((Decimal('1.335') - self.stab_radius, Decimal('-3.875') - self.stab_radius), self.stab_radius, 0, 90)
189 | stab_alps_cutout.add_arc((Decimal('-1.335') + self.stab_radius, Decimal('-9.085') + self.stab_radius), self.stab_radius, 180, 270)
190 | stab_alps_cutout.add_arc((Decimal('1.335') - self.stab_radius, Decimal('-9.085') + self.stab_radius), self.stab_radius, 270, 360)
191 |
192 | # Acoustic cutouts
193 | acoustic_cutout = self.plate.blocks.new(name='ACOUSTIC_CUTOUT')
194 | acoustic_cutout.add_line((Decimal('-1') + self.acoustics_radius, (self.cutout_height / Decimal('2'))), (Decimal('1') - self.acoustics_radius, (self.cutout_height / Decimal('2'))))
195 | acoustic_cutout.add_line((Decimal('-1') + self.acoustics_radius, (self.cutout_height / -Decimal('2'))), (Decimal('1') - self.acoustics_radius, (self.cutout_height / -Decimal('2'))))
196 | acoustic_cutout.add_line((Decimal('-1'), (self.cutout_height / Decimal('2')) - self.acoustics_radius), (Decimal('-1'), (self.cutout_height / -Decimal('2')) + self.acoustics_radius))
197 | acoustic_cutout.add_line((Decimal('1'), (self.cutout_height / Decimal('2')) - self.acoustics_radius), (Decimal('1'), (self.cutout_height / -Decimal('2')) + self.acoustics_radius))
198 | acoustic_cutout.add_arc((Decimal('-1') + self.acoustics_radius, (self.cutout_height / Decimal('2')) - self.acoustics_radius), self.stab_radius, 90, 180)
199 | acoustic_cutout.add_arc((Decimal('1') - self.acoustics_radius, (self.cutout_height / Decimal('2')) - self.acoustics_radius), self.stab_radius, 0, 90)
200 | acoustic_cutout.add_arc((Decimal('-1') + self.acoustics_radius, (self.cutout_height / -Decimal('2')) + self.acoustics_radius), self.stab_radius, 180, 270)
201 | acoustic_cutout.add_arc((Decimal('1') - self.acoustics_radius, (self.cutout_height / -Decimal('2')) + self.acoustics_radius), self.stab_radius, 270, 360)
202 |
203 |
204 | #=================================#
205 | # Functions #
206 | #=================================#
207 |
208 | # Check if string is valid number
209 | # Credits to https://stackoverflow.com/questions/4138202/using-isdigit-for-floats
210 | def is_a_number(self, s):
211 | return_value = True
212 | try:
213 | test_float = float(s)
214 | except ValueError:
215 | return_value = False
216 | return return_value
217 |
218 | # Modifies a point with rotation
219 | def rotate_point_around_anchor(self, x, y, anchor_x, anchor_y, angle):
220 | radius_squared = ((x - anchor_x) ** Decimal('2')) + ((y-anchor_y) ** Decimal('2'))
221 | radius = Decimal.sqrt(radius_squared)
222 | anglefrac = angle.as_integer_ratio()
223 | radian_qty = radians(anglefrac[0]/anglefrac[1])
224 | cos_result = Decimal(str(cos(radian_qty)))
225 | sin_result = Decimal(str(sin(radian_qty)))
226 |
227 | old_x = x - anchor_x
228 | old_y = y - anchor_y
229 |
230 | coord = matrix([float(old_x), float(old_y)])
231 | transform = matrix([[cos(radian_qty), -sin(radian_qty)], [sin(radian_qty), cos(radian_qty)]])
232 | result = transform * coord
233 |
234 | new_x = Decimal(str(result[0]))
235 | new_y = Decimal(str(result[1]))
236 |
237 | new_x += anchor_x
238 | new_y += anchor_y
239 |
240 | return (new_x, new_y)
241 |
242 | # Stab cutout maker
243 | # The x and y are center, like this:
244 | #
245 | # -------
246 | # | |
247 | # | X | - - - Center Y of switch
248 | # | |
249 | # |_ _|
250 | # |_|
251 |
252 |
253 | # Acoustics cuts maker
254 |
255 | def make_acoustic_cutout(self, x, y, anchor_x, anchor_y, angle):
256 |
257 |
258 |
259 | # Calls make stab cutout based on unit width and style
260 | def generate_stabs(self, center_x, center_y, angle, unitwidth):
261 |
262 | if (self.stab_type == "mx-simple" or self.stab_type == "large-cuts"):
263 | # Switch based on unit width
264 | # These spacings are based on official mx datasheets and deskthority measurements
265 | if (unitwidth >= 8):
266 | # self.make_stab_cutout(x, y, anchor_x, anchor_y, angle)
267 | self.make_stab_cutout(center_x + Decimal('66.675'), center_y, center_x, center_y, angle)
268 | self.make_stab_cutout(center_x - Decimal('66.675'), center_y, center_x, center_y, angle)
269 | elif (unitwidth >= 7):
270 | self.make_stab_cutout(center_x + Decimal('57.15'), center_y, center_x, center_y, angle)
271 | self.make_stab_cutout(center_x - Decimal('57.15'), center_y, center_x, center_y, angle)
272 | elif (unitwidth == 6.25):
273 | self.make_stab_cutout(center_x + Decimal('50'), center_y, center_x, center_y, angle)
274 | self.make_stab_cutout(center_x - Decimal('50'), center_y, center_x, center_y, angle)
275 | elif (unitwidth == 6):
276 | #self.make_stab_cutout(center_x + Decimal('38.1'), center_y, center_x, center_y, angle)
277 | #self.make_stab_cutout(center_x - Decimal('57.15'), center_y, center_x, center_y, angle)
278 | self.make_stab_cutout(center_x - Decimal('47.625'), center_y, center_x, center_y, angle)
279 | self.make_stab_cutout(center_x - Decimal('47.625'), center_y, center_x, center_y, angle)
280 |
281 | elif (unitwidth >= 3):
282 | self.make_stab_cutout(center_x + Decimal('19.05'), center_y, center_x, center_y, angle)
283 | self.make_stab_cutout(center_x - Decimal('19.05'), center_y, center_x, center_y, angle)
284 | elif (unitwidth >= 2):
285 | self.make_stab_cutout(center_x + Decimal('11.938'), center_y, center_x, center_y, angle)
286 | self.make_stab_cutout(center_x - Decimal('11.938'), center_y, center_x, center_y, angle)
287 | if (self.acoustics_type == "extreme"):
288 | self.make_acoustic_cutout(center_x + Decimal('18.25'), center_y, center_x, center_y, angle)
289 | self.make_acoustic_cutout(center_x - Decimal('18.25'), center_y, center_x, center_y, angle)
290 | elif (unitwidth >= 1.5):
291 | if (self.acoustics_type == "typical" or (self.acoustics_type == "extreme")):
292 | self.make_acoustic_cutout(center_x + Decimal('11.6'), center_y, center_x, center_y, angle)
293 | self.make_acoustic_cutout(center_x - Decimal('11.6'), center_y, center_x, center_y, angle)
294 | elif (self.stab_type == "alps-aek"):
295 | # These are mostly based on measurements.
296 | # If someone has datasheets, please let me know
297 | if (unitwidth >= 6.5):
298 | self.make_stab_cutout(center_x + Decimal('45.3'), center_y, center_x, center_y, angle)
299 | self.make_stab_cutout(center_x - Decimal('45.3'), center_y, center_x, center_y, angle)
300 | elif (unitwidth >= 6.25):
301 | self.make_stab_cutout(center_x + Decimal('41.86'), center_y, center_x, center_y, angle)
302 | self.make_stab_cutout(center_x - Decimal('41.86'), center_y, center_x, center_y, angle)
303 | elif (unitwidth >= 2):
304 | self.make_stab_cutout(center_x + Decimal('14'), center_y, center_x, center_y, angle)
305 | self.make_stab_cutout(center_x - Decimal('14'), center_y, center_x, center_y, angle)
306 | elif (unitwidth >= 1.75):
307 | self.make_stab_cutout(center_x + Decimal('12'), center_y, center_x, center_y, angle)
308 | self.make_stab_cutout(center_x - Decimal('12'), center_y, center_x, center_y, angle)
309 | elif (self.stab_type == "alps-at101"):
310 | # These are mostly based on measurements.
311 | # If someone has datasheets, please let me know
312 | if (unitwidth >= 6.5):
313 | self.make_stab_cutout(center_x + Decimal('45.3'), center_y, center_x, center_y, angle)
314 | self.make_stab_cutout(center_x - Decimal('45.3'), center_y, center_x, center_y, angle)
315 | elif (unitwidth >= 6.25):
316 | self.make_stab_cutout(center_x + Decimal('41.86'), center_y, center_x, center_y, angle)
317 | self.make_stab_cutout(center_x - Decimal('41.86'), center_y, center_x, center_y, angle)
318 | elif (unitwidth >= 2.75):
319 | self.make_stab_cutout(center_x + Decimal('20.5'), center_y, center_x, center_y, angle)
320 | self.make_stab_cutout(center_x - Decimal('20.5'), center_y, center_x, center_y, angle)
321 | elif (unitwidth >= 2):
322 | self.make_stab_cutout(center_x + Decimal('14'), center_y, center_x, center_y, angle)
323 | self.make_stab_cutout(center_x - Decimal('14'), center_y, center_x, center_y, angle)
324 | elif (unitwidth >= 1.75):
325 | self.make_stab_cutout(center_x + Decimal('12'), center_y, center_x, center_y, angle)
326 | self.make_stab_cutout(center_x - Decimal('12'), center_y, center_x, center_y, angle)
327 |
328 | def place_switch_cutouts(self, switch):
329 |
330 | # Handle 6U
331 | if (switch.width == 6):
332 | coord = self.rotate_point_around_anchor(switch.coord_x + Decimal("9.525"), switch.coord_y, switch.coord_x, switch.coord_y, switch.rot_angle)
333 |
334 | # Place switch object
335 | self.modelspace.add_blockref('SWITCH_CUTOUT', (switch.coord_x, switch.coord_y), dxfattribs={
336 | 'xscale': 1,
337 | 'yscale': 1,
338 | 'rotation': switch.rot_angle
339 | })
340 |
341 |
342 | # Adjust width for vertically tall keys, and generate stabs
343 | apparent_width = switch.width;
344 | if (switch.width < switch.height):
345 | apparent_width = switch.height;
346 | switch.stab_angle += Decimal("90");
347 |
348 | self.generate_stabs(mm_center_x, mm_center_y, switch.angle + switch.stab_angle, apparent_width)
349 |
350 | def render_switch(self, switch):
351 |
352 | # First, generate the switch's coord variables based on other fields
353 |
354 | # If non-rotated, use regular x/y_var
355 | if (switch.rotated_zone == False):
356 |
357 | switch.coord_x = self.unit_width * switch.x_var;
358 | switch.coord_y = self.unit_height * switch.y_var;
359 |
360 | # Otherwise, do some magic with rotation fields
361 | else:
362 |
363 | coords = rotate_point_around_anchor(switch.rot_x_offset, switch.rot_y_offset, switch.rot_anchor_x, switch.rot_anchor_y, switch.rot_angle)
364 | switch.coord_x = coords[0] * self.unit_width;
365 | switch.corod_y = coords[1] * self.unit_height;
366 |
367 | # TODO: self.max_width and self.max_height
368 |
369 | # Draw cutouts
370 | self.place_switch_cutouts(switch)
371 |
372 |
373 | # Generate switch cutout sizes
374 | def initialize_variables(self):
375 | if (self.cutout_type == "mx"):
376 | self.cutout_width = Decimal('14');
377 | self.cutout_height = Decimal('14');
378 | elif (self.cutout_type == "alps"):
379 | self.cutout_width = Decimal('15.50');
380 | self.cutout_height = Decimal('12.80');
381 | elif (self.cutout_type == "omron"):
382 | self.cutout_width = Decimal('13.50');
383 | self.cutout_height = Decimal('13.50');
384 | else:
385 | print("Unsupported cutout type.", file=sys.stderr)
386 | print("Supported: mx, alps, omron", file=sys.stderr)
387 | #exit(1)
388 | return 3
389 |
390 | # Check if values legal
391 |
392 | # Cutout radius: The fillet radius ( 0 <= x <= 1/2 width or height)
393 | if ((self.cutout_radius < 0) or (self.cutout_radius > (self.cutout_width/2)) or (self.cutout_radius > (self.cutout_height/2))) :
394 | print("Radius must be between 0 and half the cutout width/height.", file=sys.stderr)
395 | #exit(1)
396 | return 4
397 |
398 | # Unit size ( 0 <= x <= inf, cap at 1000 for now )
399 | if (self.unit_width < 0 or self.unit_width > 1000):
400 | print("Unit size must be between 0 and 1000", file=sys.stderr)
401 | #exit(1)
402 | return 5
403 |
404 | if (self.unit_height < 0 or self.unit_height > 1000):
405 | print("Unit size must be between 0 and 1000", file=sys.stderr)
406 | #exit(1)
407 | return 5
408 |
409 | if (self.stab_radius < 0 or self.stab_radius > 5):
410 | return 6
411 | if (self.acoustics_radius < 0 or self.acoustics_radius > 5):
412 | return 7
413 | if (self.stab_type != "mx-simple" and self.stab_type != "large-cuts" and self.stab_type != "alps-aek" and self.stab_type != "alps-at101"):
414 | print("Unsupported stab type.", file=sys.stderr)
415 | print("Stab types: mx-simple, large-cuts, alps-aek, alps-at101", file=sys.stderr)
416 | #exit(1)
417 | return(8)
418 |
419 |
420 | return 0
421 |
422 | def generate_plate(self, file, input_data=None):
423 |
424 | # Init vars
425 | init_code = self.initialize_variables()
426 | if (init_code != 0):
427 | return init_code
428 |
429 | # Sanitize by removing \" (KLE's literal " for a label)
430 | #input_data = input_data.replace('\n', '')
431 | #input_data = input_data.replace(r'\"', '')
432 |
433 | # TODO: Filter out improper quotes from " being in a label!
434 |
435 | if (self.debug_log):
436 | print("Filtered input data:")
437 | print(input_data)
438 | print("")
439 |
440 | # Parse KLE data
441 | all_switches = []
442 |
443 | # Switch to rotation parse mode once KLE enters the rotation syntax zone
444 | rotate_zone = False
445 |
446 | try:
447 | json_data = json5.loads('[' + input_data + ']')
448 | except(ValueError):
449 | #print("Invalid KLE data", file=sys.stderr)
450 | return(1)
451 |
452 | for row in json_data:
453 | if (self.debug_log):
454 | print (">>> ROW BEGIN")
455 | print (str(row))
456 |
457 | # KLE standard supports first row being metadata.
458 | # If it is, ignore.
459 | if isinstance(row, dict):
460 | if (self.debug_log):
461 | print ("!!! Row is metadata. Skip.")
462 | continue
463 | for key in row:
464 | # The "key" can either be a legend (actual key) or dictionary of data (for succeeding key).
465 |
466 | # If it's just a string, it's just a key. Create one and add to list
467 | if isinstance(key, str):
468 |
469 | if (self.current_deco):
470 | self.reset_key_parameters()
471 | continue
472 |
473 | # First, we simply make the switch
474 | current_switch = self.Switch(self.current_x, self.current_y)
475 |
476 | # TODO: Write the key creation code
477 |
478 | all_switches.append(current_switch)
479 |
480 | # Reset the fields to their defaults
481 | self.reset_key_parameters()
482 |
483 | # Otherwise, it's a data dictionary. We must parse it properly
484 | else:
485 | for i in key:
486 | # i = The dictionary key. Not the keyboard kind of key
487 | # j = The corresponding value.
488 | j = key[i]
489 |
490 | # Large if-else chain to set params
491 | if (str(i) == "w"):
492 | # w = Width
493 | self.current_width = Decimal(str(j))
494 |
495 | elif (str(i) == "h"):
496 | # h = Height
497 | self.current_height = Decimal(str(j))
498 |
499 | elif (str(i) == "w2"):
500 | # w2 = Secondary width
501 | self.current_width_secondary = Decimal(str(j))
502 |
503 | elif (str(i) == "h2"):
504 | # h2 = Secondary height
505 | self.current_height_secondary = Decimal(str(j))
506 |
507 | elif (str(i) == "rx"):
508 | # rx = Rotation anchor x
509 | self.current_rotx = Decimal(str(j))
510 |
511 | elif (str(i) == "ry"):
512 | # ry = Rotation anchor y
513 | self.current_roty = Decimal(str(j))
514 |
515 | elif (str(i) == "r"):
516 | # r = Rotation angle OPPOSITE OF typical counterclockwise-from-xpositive
517 | self.current_angle = -Decimal(str(j))
518 |
519 | elif (str(i) == "_rs"):
520 | # _rs = Rotation angle offset for stabilizer OPPOSITE OF typical counterclockwise-from-xpositive
521 | self.current_stab_angle = -Decimal(str(j))
522 |
523 | elif (str(i) == "_rc"):
524 | # _rs = Switch cutout angle offset for stabilizer OPPOSITE OF typical counterclockwise-from-xpositive
525 | self.current_cutout_angle = -Decimal(str(j))
526 |
527 | elif (str(i) == "x"):
528 | # x = X offset for next keys OR offset from rotation anchor (seriously kle?)
529 | self.current_offset_x = Decimal(str(j))
530 |
531 | elif (str(i) == "y"):
532 | # y = Y offset for next keys OR offset from rotation anchor (seriously kle?)
533 | self.current_offset_y = Decimal(str(j))
534 |
535 | elif (str(i) == "d"):
536 | # Key is decoration.
537 | self.current_deco = True
538 |
539 | # Finished row
540 | self.current_y -= Decimal('1')
541 | self.current_x = Decimal('0')
542 |
543 |
544 | # At this point, the keys are built.
545 |
546 | # Adjust max width/height from units to mm
547 |
548 | self.max_width = self.max_width * self.unit_width
549 | self.max_height = self.max_height * self.unit_height
550 |
551 | # Render each one by one.
552 | for switch in all_switches:
553 | self.render_switch(switch)
554 |
555 | # Draw outer bounds - top, bottom, left, right
556 | self.modelspace.add_line((0, 0), (self.max_width, 0))
557 | self.modelspace.add_line((0, self.max_height), (self.max_width, self.max_height))
558 | self.modelspace.add_line((0, 0), (0, self.max_height))
559 | self.modelspace.add_line((self.max_width, 0), (self.max_width, self.max_height))
560 |
561 | if (self.debug_log):
562 | print("Complete! Saving plate to specified output")
563 |
564 | if (file == "stdout"):
565 | self.plate.write(sys.stdout)
566 | else:
567 | self.plate.write(file)
568 | return 0
569 |
570 |
571 |
572 | if __name__ == "__main__":
573 |
574 | parser = argparse.ArgumentParser(description='Create a plate DXF based on KLE raw data.')
575 |
576 | # Note: The args will be fed into Decimal(), which takes strings
577 |
578 | parser.add_argument("-ct", "--cutout-type", help="Switch cutout type. Supported: mx, alps, omron; Default: mx", type=str, default='mx')
579 | parser.add_argument("-cr", "--cutout-radius", help="Switch cutout fillet radius. Default: 0.5", type=str, default='0.5')
580 | parser.add_argument("-st", "--stab-type", help="Stabilizer type. Supported: mx-simple, large-cuts, alps-aek, alps-at101; Default: mx-simple", type=str, default='mx-simple')
581 | parser.add_argument("-sr", "--stab-radius", help="Stabilizer cutout fillet radius. Default: 0.5", type=str, default='0.5')
582 | parser.add_argument("-at", "--acoustics-type", help="Acoustic cutouts type. Supported: none, typical, extreme; Default: none", type=str, default='none')
583 | parser.add_argument("-ar", "--acoustics-radius", help="Acoustic cutouts fillet radius. Default: 0.5", type=str, default='0.5')
584 | parser.add_argument("-uw", "--unit-width", help="Key unit width. Default: 19.05", type=str, default='19.05')
585 | parser.add_argument("-uh", "--unit-height", help="Key unit height. Default: 19.05", type=str, default='19.05')
586 | #parser.add_argument("-om", "--output-method", help="The save method for data. Supported: stdout, file; Default: stdout", type=str, default='stdout')
587 | #parser.add_argument("-of", "--output-file", help="Output file name if using file output-method. Default: plate.dxf", type=str, default='plate.dxf')
588 | parser.add_argument("--debug-log", help="Spam output with useless info.", action="store_true", default = False)
589 |
590 | args = parser.parse_args()
591 |
592 | gen = PlateGenerator(args.cutout_type, args.cutout_radius, args.stab_type, args.stab_radius, args.acoustics_type, args.acoustics_radius,
593 | args.unit_width, args.unit_height, args.debug_log)
594 |
595 | input_data = sys.stdin.read()
596 | gen.generate_plate("stdout", input_data)
597 |
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | GNU AFFERO GENERAL PUBLIC LICENSE
2 | Version 3, 19 November 2007
3 |
4 | Copyright (C) 2007 Free Software Foundation, Inc.
287 |
288 |
298 |
289 |
292 |
293 |
294 |
295 |
296 |
297 | Written by ai03
290 |Credits to Amtra5, Mxblue, Bakingpy, Senter, Pwner, and Kevinplus for help and info
291 |292 |
293 |