File: block.py

package info (click to toggle)
linuxcnc 1%3A2.9.4-2
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 282,780 kB
  • sloc: python: 201,110; ansic: 106,370; cpp: 99,219; tcl: 16,054; xml: 10,617; sh: 10,258; makefile: 1,251; javascript: 138; sql: 72; asm: 15
file content (347 lines) | stat: -rw-r--r-- 14,858 bytes parent folder | download | duplicates (3)
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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
'''
block.py

Copyright (C) 2020, 2021, 2022  Phillip A Carter
Copyright (C) 2020, 2021, 2022  Gregory D Carl

This program is free software; you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by the
Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
'''

'''
NOTE:
mirror is done externally by inverting convMirror and calling the preview function
flip is done externally by inverting convFlip and calling the preview function
'''

import os
import sys
from shutil import copy as COPY
import gettext

for f in sys.path:
    if '/lib/python' in f:
        if '/usr' in f:
            localeDir = 'usr/share/locale'
        else:
            localeDir = os.path.join('{}'.format(f.split('/lib')[0]),'share','locale')
        break
gettext.install("linuxcnc", localedir=localeDir)

# Conv is the upstream calling module
def preview(Conv, fNgc, fTmp, columns, rows, cOffset, \
            rOffset, xOffset, yOffset, angle, \
            scale, rotation, convBlock, convMirror, convFlip, \
            convMirrorToggle, convFlipToggle, g5xIndex, convUnits):
    error = ''
    msg1 = _('entry is invalid')
    valid, columns = Conv.conv_is_int(columns)
    if not valid:
        msg0 = _('COLUMNS NUMBER')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, cOffset = Conv.conv_is_float(cOffset)
    if not valid and cOffset:
        msg0 = _('COLUMNS OFFSET')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, rows = Conv.conv_is_int(rows)
    if not valid:
        msg0 = _('ROWS NUMBER')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, rOffset = Conv.conv_is_float(rOffset)
    if not valid and rOffset:
        msg0 = _('ROWS OFFSET')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, xOffset = Conv.conv_is_float(xOffset)
    if not valid and xOffset:
        msg0 = _('X OFFSET')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, yOffset = Conv.conv_is_float(yOffset)
    if not valid and yOffset:
        msg0 = _('Y OFFSET')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, angle = Conv.conv_is_float(angle)
    if not valid and angle:
        msg0 = _('PATTERN ANGLE')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, scale = Conv.conv_is_float(scale)
    if not valid:
        msg0 = _('SHAPE SCALE')
        error += '{} {}\n\n'.format(msg0, msg1)
    valid, rotation = Conv.conv_is_float(rotation)
    if not valid and rotation:
        msg0 = _('SHAPE ROTATION')
        error += '{} {}\n\n'.format(msg0, msg1)
    if error:
        return error
    if columns <= 0:
        msg = _('COLUMNS NUMBER cannot be zero')
        error += '{}\n\n'.format(msg)
    if rows <= 0:
        msg = _('ROWS NUMBER cannot be zero')
        error += '{}\n\n'.format(msg)
    # if columns == 1 and rows == 1:
    #     msg = _('Either COLUMNS NUMBER or ROWS NUMBER must be greater then one')
    #     error += '{}\n\n'.format(msg)
    if columns > 1 and not cOffset:
        msg = _('COLUMNS OFFSET is required')
        error += '{}\n\n'.format(msg)
    if rows > 1 and not rOffset:
        msg = _('ROWS OFFSET is required')
        error += '{}\n\n'.format(msg)
    if scale <=0:
        msg = _('SCALE cannot be zero')
        error += '{}\n\n'.format(msg)
    if error:
        return error
    COPY(fNgc, fTmp)
    inCode = open(fTmp, 'r')
    outNgc = open(fNgc, 'w')
    # edit existing parameters
    if convBlock[0]:
        indent = False
        for line in inCode:
            if line.startswith('#<array_x_offset>'):
                line = '#<array_x_offset> = {}\n'.format(cOffset)
            elif line.startswith('#<array_y_offset>'):
                line = '#<array_y_offset> = {}\n'.format(rOffset)
            elif line.startswith('#<array_columns>'):
                line = '#<array_columns> = {}\n'.format(columns)
            elif line.startswith('#<array_rows>'):
                line = '#<array_rows> = {}\n'.format(rows)
            elif line.startswith('#<origin_x_offset>'):
                line = '#<origin_x_offset> = {}\n'.format(xOffset)
            elif line.startswith('#<origin_y_offset>'):
                line = '#<origin_y_offset> = {}\n'.format(yOffset)
            elif line.startswith('#<array_angle>'):
                line ='#<array_angle> = {}\n'.format(angle)
            elif line.startswith('#<blk_scale>'):
                line ='#<blk_scale> = {}\n'.format(scale)
            elif line.startswith('#<shape_angle>'):
                line ='#<shape_angle> = {}\n'.format(rotation)
            elif line.startswith('#<shape_mirror>'):
                line ='#<shape_mirror> = {}\n'.format(convMirror)
            elif line.startswith('#<shape_flip>'):
                line ='#<shape_flip> = {}\n'.format(convFlip)
            elif '#<shape_mirror>' in line and (convMirrorToggle or convFlipToggle):
                if 'g2' in line:
                    line = line.replace('g2', 'g3')
                elif 'g3' in line:
                    line = line.replace('g3', 'g2')
            outNgc.write('{}'.format(line))
    # create new array
    else:
        xIndex = [5221,5241,5261,5281,5301,5321,5341,5361,5381][(g5xIndex - 1)]
        outNgc.write(';conversational block\n\n')
        # inputs
        outNgc.write(';inputs\n')
        outNgc.write('#<ucs_x_offset> = #{}\n'.format(xIndex))
        outNgc.write('#<ucs_y_offset> = #{}\n'.format(xIndex + 1))
        outNgc.write('#<ucs_r_offset> = #{}\n'.format(xIndex + 9))
        outNgc.write('#<array_x_offset> = {}\n'.format(cOffset))
        outNgc.write('#<array_y_offset> = {}\n'.format(rOffset))
        outNgc.write('#<array_columns> = {}\n'.format(columns))
        outNgc.write('#<array_rows> = {}\n'.format(rows))
        outNgc.write('#<origin_x_offset> = {}\n'.format(xOffset))
        outNgc.write('#<origin_y_offset> = {}\n'.format(yOffset))
        outNgc.write('#<array_angle> = {}\n'.format(angle))
        outNgc.write('#<blk_scale> = {}\n'.format(scale))
        outNgc.write('#<shape_angle> = {}\n'.format(rotation))
        outNgc.write('#<shape_mirror> = {}\n'.format(convMirror))
        outNgc.write('#<shape_flip> = {}\n\n'.format(convFlip))
        # calculations
        outNgc.write(';calculations\n')
        outNgc.write('#<this_col> = 0\n')
        outNgc.write('#<this_row> = 0\n')
        outNgc.write('#<array_rot> = [#<array_angle> + #<ucs_r_offset>]\n')
        outNgc.write('#<blk_x_offset> = [#<origin_x_offset> + [#<ucs_x_offset> * {}]]\n'.format(convUnits[0]))
        outNgc.write('#<blk_y_offset> = [#<origin_y_offset> + [#<ucs_y_offset> * {}]]\n'.format(convUnits[0]))
        outNgc.write('#<x_sin> = [[#<array_x_offset> * #<blk_scale>] * SIN[#<array_rot>]]\n')
        outNgc.write('#<x_cos> = [[#<array_x_offset> * #<blk_scale>] * COS[#<array_rot>]]\n')
        outNgc.write('#<y_sin> = [[#<array_y_offset> * #<blk_scale>] * SIN[#<array_rot>]]\n')
        outNgc.write('#<y_cos> = [[#<array_y_offset> * #<blk_scale>] * COS[#<array_rot>]]\n\n')
        # main loop
        outNgc.write(';main loop\n')
        outNgc.write('o<loop> while [#<this_row> LT #<array_rows>]\n')
        outNgc.write('    #<shape_x_start> = [[#<this_col> * #<x_cos>] - [#<this_row> * #<y_sin>] + #<blk_x_offset>]\n')
        outNgc.write('    #<shape_y_start> = [[#<this_row> * #<y_cos>] + [#<this_col> * #<x_sin>] + #<blk_y_offset>]\n')
        outNgc.write('    #<blk_angle> = [#<shape_angle> + #<array_rot>]\n')
        if convUnits[1]:
            outNgc.write('    {}\n'.format(convUnits[1]))
        outNgc.write('    G10 L2 P0 X#<shape_x_start> Y#<shape_y_start> R#<blk_angle>\n\n')
        # the shape
        started, ended = False, False
        for line in inCode:
            line = line.strip().lower()
            # remove line numbers
            if line.startswith('n'):
                line = line[1:]
                while line[0].isdigit() or line[0] == '.':
                    line = line[1:].lstrip()
                    if not line:
                        break
            # remove leading 0's from G & M codes
            elif (line.lower().startswith('g') or \
               line.lower().startswith('m')) and \
               len(line) > 2:
                while line[1] == '0' and len(line) > 2:
                    if line[2].isdigit():
                        line = line[:1] + line[2:]
                    else:
                        break
            # scale the shape
            if len(line) and line[0] in 'gxyz':
                started = True
                rLine = scale_shape(line, convMirrorToggle, convFlipToggle)
                if rLine is not None:
                    outNgc.write('    {}\n'.format(rLine))
                else:
                    return
            # loop counter
            elif not ended and ('m2' in line or 'm30' in line or (line.startswith('%') and started)):
                ended = True
                outNgc.write('\n    #<this_col> = [#<this_col> + 1]\n')
                outNgc.write('    o<count> if [#<this_col> EQ #<array_columns>]\n')
                outNgc.write('        #<this_col> = 0\n')
                outNgc.write('        #<this_row> = [#<this_row> + 1]\n')
                outNgc.write('    o<count> endif\n')
                outNgc.write('o<loop> endwhile\n')
            elif not line:
                outNgc.write('\n')
            elif ended and ('m2' in line or 'm30' in line or line.startswith('%')):
                pass
            else:
                outNgc.write('    {}\n'.format(line))
        # reset offsets to original
        outNgc.write('\nG10 L2 P0 X[#<ucs_x_offset> * {0}] Y[#<ucs_y_offset> * {0}] R#<ucs_r_offset>\n'.format(convUnits[0]))
        outNgc.write('\nM2\n')
    inCode.close()
    outNgc.close()
    return False

def scale_shape(line, convMirrorToggle, convFlipToggle):
    if line[0] == 'g' and (line[1] not in '0123' or (line[1] in '0123' and len(line) > 2 and line[2] in '0123456789')):
        return '{}'.format(line)
    newLine = ''
    multiAxis = False
    numParam = False
    namParam = False
    fWord = False
    lastAxis = ''
    offset = ''
    while 1:
        # remove spaces
        if line[0] == ' ':
            line = line[1:]
        # if beginning of comment
        if line[0] == '(' or line[0] == ';':
            if multiAxis and not fWord:
                if lastAxis == 'x':
                    newLine += '*#<blk_scale>*#<shape_mirror>]'
                elif lastAxis == 'i':
                    newLine += '*#<blk_scale>*#<shape_mirror>]'
                elif lastAxis == 'y':
                    newLine += '*#<blk_scale>*#<shape_flip>]'
                elif lastAxis == 'j':
                    newLine += '*#<blk_scale>*#<shape_flip>]'
                else:
                    newLine += '*#<blk_scale>]'
            newLine += line
            break
        # if beginning of parameter
        elif line[0] == 'p':
            if not numParam and not namParam:
                if multiAxis and not fWord:
                    if lastAxis == 'x':
                        newLine += '*#<blk_scale>*#<shape_mirror>]'
                    elif lastAxis == 'i':
                        newLine += '*#<blk_scale>*#<shape_mirror>]'
                    elif lastAxis == 'y':
                        newLine += '*#<blk_scale>*#<shape_flip>]'
                    elif lastAxis == 'j':
                        newLine += '*#<blk_scale>*#<shape_flip>]'
                    else:
                        newLine += '*#<blk_scale>]'
                lastAxis = line[0]
            newLine += line[0]
            line = line[1:]
        # if alpha character
        elif line[0].isalpha():
            if not numParam and not namParam:
                if multiAxis and not fWord:
                    if lastAxis == 'x':
                        newLine += '*#<blk_scale>*#<shape_mirror>]'
                    elif lastAxis == 'i':
                        newLine += '*#<blk_scale>*#<shape_mirror>]'
                    elif lastAxis == 'y':
                        newLine += '*#<blk_scale>*#<shape_flip>]'
                    elif lastAxis == 'j':
                        newLine += '*#<blk_scale>*#<shape_flip>]'
                    else:
#                    elif lastAxis not in 'p':
                        newLine += '*#<blk_scale>]'
                lastAxis = line[0]
                if line[0] == 'f':
                    fWord = True
            newLine += line[0]
            line = line[1:]
        # if beginning of parameter
        elif line[0] == '#':
            numParam = True
            newLine += line[0]
            line = line[1:]
        # if parameter should be a named parameter
        elif line[0] == '<' and numParam:
            numParam = False
            namParam = True
            newLine += line[0]
            line = line[1:]
        #if end of numbered parameter
        elif not line[0].isdigit() and numParam:
            numParam = False
            newLine += line[0]
            line = line[1:]
        # if end of named parameter
        elif line[0] == '>' and namParam:
            namParam = False
            newLine += line[0]
            line = line[1:]
        #if last axis was x, y, z, i, j, or r
        elif newLine[-1] in 'xyzijr' and not numParam and not namParam:
            multiAxis = True
            newLine += '[{}'.format(line[0])
            line = line[1:]
        # everything else
        else:
            newLine += line[0]
            line = line[1:]
        # empty line, must be finished
        if not line:
            if not fWord:
                if lastAxis == 'x':
                    newLine += '*#<blk_scale>*#<shape_mirror>]'
                elif lastAxis == 'i':
                    newLine += '*#<blk_scale>*#<shape_mirror>]'
                elif lastAxis == 'y':
                    newLine += '*#<blk_scale>*#<shape_flip>]'
                elif lastAxis == 'j':
                    newLine += '*#<blk_scale>*#<shape_flip>]'
                elif lastAxis not in 'p':
                    newLine += '*#<blk_scale>]'
            break
    if '#<shape_mirror>' in newLine and (convMirrorToggle or convFlipToggle):
        if 'g2' in newLine:
            newLine = newLine.replace('g2', 'g3')
        elif 'g3' in newLine:
            newLine = newLine.replace('g3', 'g2')
    return ('{}'.format(newLine))