199 lines
6.8 KiB
Python
199 lines
6.8 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import math
|
|||
|
|
import sys
|
|||
|
|
import inkex
|
|||
|
|
from simpletransform import parseTransform, formatTransform
|
|||
|
|
from simpletransform import computeBBox, applyTransformToPoint
|
|||
|
|
|
|||
|
|
sys.path.append('/usr/share/inkscape/extensions')
|
|||
|
|
inkex.localize()
|
|||
|
|
|
|||
|
|
|
|||
|
|
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):
|
|||
|
|
"""
|
|||
|
|
Constructor.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
inkex.Effect.__init__(self)
|
|||
|
|
|
|||
|
|
self.OptionParser.add_option(
|
|||
|
|
'-c', '--conversion', action='store', type='string',
|
|||
|
|
dest='conversion', default='top',
|
|||
|
|
help='Conversion to perform: (top|left|right)')
|
|||
|
|
self.OptionParser.add_option(
|
|||
|
|
'-r', '--reverse', action='store', type='string',
|
|||
|
|
dest='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 getMidPoint(self, bbox, node):
|
|||
|
|
"""
|
|||
|
|
Get the coordinates of the center of the bounding box.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
x = bbox[1] - (bbox[1] - bbox[0]) / 2
|
|||
|
|
y = bbox[3] - (bbox[3] - bbox[2]) / 2
|
|||
|
|
|
|||
|
|
return [x, y]
|
|||
|
|
|
|||
|
|
def translateBetweenPoints(self, matrix, here, there):
|
|||
|
|
"""
|
|||
|
|
Add a translation to a matrix that moves between two points.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
x = there[0] - here[0]
|
|||
|
|
y = there[1] - here[1]
|
|||
|
|
matrix[0][2] += x
|
|||
|
|
matrix[1][2] += 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):
|
|||
|
|
"""
|
|||
|
|
Apply the transformation. If an element already has a transformation
|
|||
|
|
attribute, it will be combined with the transformation matrix for the
|
|||
|
|
requested conversion.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
if self.options.reverse == "true":
|
|||
|
|
conversion = "from_" + self.options.conversion
|
|||
|
|
else:
|
|||
|
|
conversion = "to_" + self.options.conversion
|
|||
|
|
|
|||
|
|
if len(self.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.selected.items():
|
|||
|
|
bbox = computeBBox([node])
|
|||
|
|
midpoint = self.getMidPoint(bbox, node)
|
|||
|
|
center_old = self.getTransformCenter(midpoint, node)
|
|||
|
|
transform = node.get("transform")
|
|||
|
|
# Combine our transformation matrix with any pre-existing
|
|||
|
|
# transform.
|
|||
|
|
matrix = parseTransform(transform, effect_matrix)
|
|||
|
|
|
|||
|
|
# Compute the location of the transformation center after applying
|
|||
|
|
# the transformation matrix.
|
|||
|
|
center_new = center_old[:]
|
|||
|
|
applyTransformToPoint(matrix, center_new)
|
|||
|
|
applyTransformToPoint(matrix, midpoint)
|
|||
|
|
|
|||
|
|
# Add a translation transformation that will move the object to
|
|||
|
|
# keep its transformation center in the same place.
|
|||
|
|
self.translateBetweenPoints(matrix, center_new, center_old)
|
|||
|
|
|
|||
|
|
node.set('transform', formatTransform(matrix))
|
|||
|
|
|
|||
|
|
# Adjust the transformation center.
|
|||
|
|
self.moveTransformationCenter(node, midpoint, center_new)
|
|||
|
|
|
|||
|
|
# Create effect instance and apply it.
|
|||
|
|
effect = IsometricProjectionTools()
|
|||
|
|
effect.affect()
|