linetypes.py

Mon, 22 Jan 2018 21:04:53 +0200

author
Santeri Piippo
date
Mon, 22 Jan 2018 21:04:53 +0200
changeset 25
8990ac138cc2
parent 7
0ab0d61ccee8
child 35
865cd526e8b6
permissions
-rw-r--r--

added check for invalid colours

3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
1 def ldraw_str(value):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
2 ' Like str() except removes unneeded ".0"-suffixes '
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
3 rep = str(value)
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
4 if isinstance(value, float):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
5 if rep.endswith('.0'):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
6 rep = rep[:-2]
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
7 if rep == '-0':
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
8 rep = '0'
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
9 return rep
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
10
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
11 class EmptyLine:
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
12 def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
13 return 'linetypes.EmptyLine()'
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
14 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
15 return ''
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
16 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
17 return 'empty line'
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
18
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
19 class Comment:
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
20 def __init__(self, text, style = 'old'):
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
21 if style == 'old' and text.startswith('//'):
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
22 self.text = text[2:].strip()
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
23 self.style = 'new'
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
24 else:
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
25 self.text = text
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
26 self.style = style
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
27 def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
28 return str.format('linetypes.Comment({text!r}, {style!r})',
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
29 text = self.text,
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
30 style = self.style,
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
31 )
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
32 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
33 if self.style == 'old':
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
34 return '0 ' + self.text
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
35 else:
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
36 return '0 // ' + self.text
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
37 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
38 return 'comment'
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
39
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
40 class SubfileReference:
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
41 def __init__(self, *, colour, subfile_path, anchor, matrix):
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
42 self.colour, self.subfile_path, = colour, subfile_path
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
43 self.anchor, self.matrix = anchor, matrix
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
44 def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
45 return str.format('linetypes.SubfileReference(' \
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
46 'colour = {colour!r}, ' \
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
47 'subfile_path = {subfile_path!r}, ' \
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
48 'anchor = {anchor!r}, ' \
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
49 'matrix = {matrix!r})', **self.__dict__)
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
50 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
51 args = [
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
52 1,
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
53 self.colour,
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
54 *self.anchor.coordinates,
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
55 *self.matrix.values,
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
56 self.subfile_path,
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
57 ]
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
58 return ' '.join(ldraw_str(arg) for arg in args)
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
59 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
60 return 'subfile reference'
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
61
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
62 class BasePolygon:
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
63 def __init__(self, *, colour, geometry):
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
64 self.colour, self.geometry = colour, geometry
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
65 def __repr__(self):
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
66 return str.format('linetypes.{typename}(' \
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
67 'colour = {colour!r}, ' \
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
68 'geometry = {geometry!r})',
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
69 typename = type(self).__name__,
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
70 colour = self.colour,
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
71 geometry = self.geometry,
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
72 )
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
73 def base_textual_representation(self):
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
74 args = [self.colour]
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
75 for vertex in self.geometry.vertices:
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
76 args += vertex.coordinates
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
77 return ' '.join(ldraw_str(arg) for arg in args)
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
78
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
79 class LineSegment(BasePolygon):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
80 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
81 return '2 ' + self.base_textual_representation()
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
82 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
83 return 'line segment'
0
55b4c97d44c5 Initial commit with half-done parsing function
Santeri Piippo
parents:
diff changeset
84
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
85 class Triangle(BasePolygon):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
86 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
87 return '3 ' + self.base_textual_representation()
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
88 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
89 return 'triangle'
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
90
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
91 class Quadrilateral(BasePolygon):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
92 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
93 return '4 ' + self.base_textual_representation()
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
94 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
95 return 'quadrilateral'
1
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
96
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
97 class Contour(LineSegment):
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
98 def __init__(self, *, colour, geometry, control_points):
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
99 super().__init__(colour = colour, geometry = geometry)
1
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
100 self.control_points = control_points
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
101 assert(len(self.control_points) == 2)
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
102 def __repr__(self):
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
103 return str.format('linetypes.Contour(' \
4
8eb83f200486 color -> colour
Santeri Piippo
parents: 3
diff changeset
104 'colour = {colour!r}, ' \
1
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
105 'geometry = {geometry!r}, ' \
5411a25cfca7 Parsing function complete
Santeri Piippo
parents: 0
diff changeset
106 'control_points = {control_points!r})', **self.__dict__)
3
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
107 def textual_representation(self):
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
108 result = '5 ' + self.base_textual_representation()
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
109 for control_point in self.control_points:
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
110 strings = (ldraw_str(value) for value in control_point.coordinates)
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
111 result += ' '
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
112 result += ' '.join(strings)
1dc58f44d556 Can now write dat files, added direct color handling
Santeri Piippo
parents: 1
diff changeset
113 return result
7
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
114 def typename(self):
0ab0d61ccee8 Smallest angles
Santeri Piippo
parents: 4
diff changeset
115 return 'contour line segment'

mercurial