162 lines
5.9 KiB
Python
162 lines
5.9 KiB
Python
#!/usr/bin/env python3
|
||
|
||
import math
|
||
import inkex
|
||
from inkex import Transform
|
||
|
||
class IsometricProjectionTools(inkex.Effect):
|
||
"""
|
||
Convert a flat 2D projection to one of the three visible sides in an
|
||
isometric projection, and vice versa.
|
||
"""
|
||
|
||
attrTransformCenterX = inkex.addNS('transform-center-x', 'inkscape')
|
||
attrTransformCenterY = inkex.addNS('transform-center-y', 'inkscape')
|
||
|
||
# Precomputed values for sine, cosine, and tangent of 30°.
|
||
rad_30 = math.radians(30)
|
||
cos_30 = math.cos(rad_30)
|
||
sin_30 = 0.5 # No point in using math.sin for 30°.
|
||
tan_30 = math.tan(rad_30)
|
||
|
||
# Combined affine transformation matrices. The bottom row of these 3×3
|
||
# matrices is omitted; it is always [0, 0, 1].
|
||
transformations = {
|
||
# From 2D to isometric top down view:
|
||
# * scale vertically by cos(30°)
|
||
# * shear horizontally by -30°
|
||
# * rotate clock-wise 30°
|
||
'to_top': [[cos_30, -cos_30, 0],
|
||
[sin_30, sin_30, 0]],
|
||
|
||
# From 2D to isometric left-hand side view:
|
||
# * scale horizontally by cos(30°)
|
||
# * shear vertically by -30°
|
||
'to_left': [[cos_30, 0, 0],
|
||
[sin_30, 1, 0]],
|
||
|
||
# From 2D to isometric right-hand side view:
|
||
# * scale horizontally by cos(30°)
|
||
# * shear vertically by 30°
|
||
'to_right': [[cos_30, 0, 0],
|
||
[-sin_30, 1, 0]],
|
||
|
||
# From isometric top down view to 2D:
|
||
# * rotate counter-clock-wise 30°
|
||
# * shear horizontally by 30°
|
||
# * scale vertically by 1 / cos(30°)
|
||
'from_top': [[tan_30, 1, 0],
|
||
[-tan_30, 1, 0]],
|
||
|
||
# From isometric left-hand side view to 2D:
|
||
# * shear vertically by 30°
|
||
# * scale horizontally by 1 / cos(30°)
|
||
'from_left': [[1 / cos_30, 0, 0],
|
||
[-tan_30, 1, 0]],
|
||
|
||
# From isometric right-hand side view to 2D:
|
||
# * shear vertically by -30°
|
||
# * scale horizontally by 1 / cos(30°)
|
||
'from_right': [[1 / cos_30, 0, 0],
|
||
[tan_30, 1, 0]]
|
||
}
|
||
|
||
def __init__(self):
|
||
inkex.Effect.__init__(self)
|
||
self.arg_parser.add_argument('--conversion', default='top', help='Conversion to perform: (top|left|right)')
|
||
self.arg_parser.add_argument('--reverse', default="false", help='Reverse the transformation from isometric projection to flat 2D')
|
||
|
||
def getTransformCenter(self, midpoint, node):
|
||
"""
|
||
Find the transformation center of an object. If the user set it
|
||
manually by dragging it in Inkscape, those coordinates are used.
|
||
Otherwise, an attempt is made to find the center of the object's
|
||
bounding box.
|
||
"""
|
||
|
||
c_x = node.get(self.attrTransformCenterX)
|
||
c_y = node.get(self.attrTransformCenterY)
|
||
|
||
# Default to dead-center.
|
||
if c_x is None:
|
||
c_x = 0.0
|
||
else:
|
||
c_x = float(c_x)
|
||
if c_y is None:
|
||
c_y = 0.0
|
||
else:
|
||
c_y = float(c_y)
|
||
|
||
x = midpoint[0] + c_x
|
||
y = midpoint[1] - c_y
|
||
|
||
return [x, y]
|
||
|
||
def translateBetweenPoints(self, tr, here, there):
|
||
"""
|
||
Add a translation to a matrix that moves between two points.
|
||
"""
|
||
|
||
x = there[0] - here[0]
|
||
y = there[1] - here[1]
|
||
tr.add_translate(x, y)
|
||
|
||
def moveTransformationCenter(self, node, midpoint, center_new):
|
||
"""
|
||
If a transformation center is manually set on the node, move it to
|
||
match the transformation performed on the node.
|
||
"""
|
||
|
||
c_x = node.get(self.attrTransformCenterX)
|
||
c_y = node.get(self.attrTransformCenterY)
|
||
|
||
if c_x is not None:
|
||
x = str(center_new[0] - midpoint[0])
|
||
node.set(self.attrTransformCenterX, x)
|
||
if c_y is not None:
|
||
y = str(midpoint[1] - center_new[1])
|
||
node.set(self.attrTransformCenterY, y)
|
||
|
||
def effect(self):
|
||
if self.options.reverse == "true":
|
||
conversion = "from_" + self.options.conversion
|
||
else:
|
||
conversion = "to_" + self.options.conversion
|
||
|
||
if len(self.svg.selected) == 0:
|
||
inkex.errormsg("Please select an object to perform the " +
|
||
"isometric projection transformation on.")
|
||
return
|
||
|
||
# Default to the flat 2D to isometric top down view conversion if an
|
||
# invalid identifier is passed.
|
||
effect_matrix = self.transformations.get(
|
||
conversion, self.transformations.get('to_top'))
|
||
|
||
for id, node in self.svg.selected.items():
|
||
bbox = node.bounding_box()
|
||
midpoint = [bbox.center_x, bbox.center_y]
|
||
center_old = self.getTransformCenter(midpoint, node)
|
||
transform = node.get("transform")
|
||
# Combine our transformation matrix with any pre-existing
|
||
# transform.
|
||
tr = Transform(transform) * Transform(effect_matrix)
|
||
|
||
# Compute the location of the transformation center after applying
|
||
# the transformation matrix.
|
||
center_new = center_old[:]
|
||
#Transform(matrix).apply_to_point(center_new)
|
||
tr.apply_to_point(center_new)
|
||
tr.apply_to_point(midpoint)
|
||
|
||
# Add a translation transformation that will move the object to
|
||
# keep its transformation center in the same place.
|
||
self.translateBetweenPoints(tr, center_new, center_old)
|
||
|
||
node.set('transform', str(tr))
|
||
|
||
# Adjust the transformation center.
|
||
self.moveTransformationCenter(node, midpoint, center_new)
|
||
|
||
# Create effect instance and apply it.
|
||
IsometricProjectionTools().run() |