import itertools as it import re import string import warnings import os import hashlib from xml.dom import minidom from manimlib.constants import DEFAULT_STROKE_WIDTH from manimlib.constants import ORIGIN, UP, DOWN, LEFT, RIGHT from manimlib.constants import BLACK from manimlib.constants import WHITE from manimlib.constants import DEGREES, PI from manimlib.mobject.geometry import Circle from manimlib.mobject.geometry import Rectangle from manimlib.mobject.geometry import RoundedRectangle from manimlib.mobject.types.vectorized_mobject import VGroup from manimlib.mobject.types.vectorized_mobject import VMobject from manimlib.utils.color import * from manimlib.utils.config_ops import digest_config from manimlib.utils.directories import get_mobject_data_dir from manimlib.utils.images import get_full_vector_image_path from manimlib.utils.simple_functions import clip def string_to_numbers(num_string): num_string = num_string.replace("-", ",-") num_string = num_string.replace("e,-", "e-") return [ float(s) for s in re.split("[ ,]", num_string) if s != "" ] class SVGMobject(VMobject): CONFIG = { "should_center": True, "height": 2, "width": None, # Must be filled in in a subclass, or when called "file_name": None, "unpack_groups": True, # if False, creates a hierarchy of VGroups # TODO, style components should be read in, not defaulted "stroke_width": DEFAULT_STROKE_WIDTH, "fill_opacity": 1.0, "path_string_config": {} } def __init__(self, file_name=None, **kwargs): digest_config(self, kwargs) self.file_name = file_name or self.file_name if file_name is None: raise Exception("Must specify file for SVGMobject") self.file_path = get_full_vector_image_path(file_name) super().__init__(**kwargs) self.move_into_position() def move_into_position(self): if self.should_center: self.center() if self.height is not None: self.set_height(self.height) if self.width is not None: self.set_width(self.width) def init_points(self): doc = minidom.parse(self.file_path) self.ref_to_element = {} for child in doc.childNodes: if child.tagName != 'svg': continue mobjects = self.get_mobjects_from(child) if self.unpack_groups: self.add(*mobjects) else: self.add(*mobjects[0].submobjects) doc.unlink() def get_mobjects_from(self, element): result = [] if not isinstance(element, minidom.Element): return result if element.tagName == 'defs': self.update_ref_to_element(element) elif element.tagName == 'style': pass # TODO, handle style elif element.tagName in ['g', 'svg', 'symbol']: result += it.chain(*( self.get_mobjects_from(child) for child in element.childNodes )) elif element.tagName == 'path': result.append(self.path_string_to_mobject( element.getAttribute('d') )) elif element.tagName == 'use': result += self.use_to_mobjects(element) elif element.tagName == 'rect': result.append(self.rect_to_mobject(element)) elif element.tagName == 'circle': result.append(self.circle_to_mobject(element)) elif element.tagName == 'ellipse': result.append(self.ellipse_to_mobject(element)) elif element.tagName in ['polygon', 'polyline']: result.append(self.polygon_to_mobject(element)) else: pass # TODO # warnings.warn("Unknown element type: " + element.tagName) result = [m for m in result if m is not None] self.handle_transforms(element, VGroup(*result)) if len(result) > 1 and not self.unpack_groups: result = [VGroup(*result)] return result def g_to_mobjects(self, g_element): mob = VGroup(*self.get_mobjects_from(g_element)) self.handle_transforms(g_element, mob) return mob.submobjects def path_string_to_mobject(self, path_string): return VMobjectFromSVGPathstring( path_string, **self.path_string_config, ) def use_to_mobjects(self, use_element): # Remove initial "#" character ref = use_element.getAttribute("xlink:href")[1:] if ref not in self.ref_to_element: warnings.warn(f"{ref} not recognized") return VGroup() return self.get_mobjects_from( self.ref_to_element[ref] ) def attribute_to_float(self, attr): stripped_attr = "".join([ char for char in attr if char in string.digits + "." + "-" ]) return float(stripped_attr) def polygon_to_mobject(self, polygon_element): path_string = polygon_element.getAttribute("points") for digit in string.digits: path_string = path_string.replace(f" {digit}", f"L {digit}") path_string = path_string.replace("L", "M", 1) return self.path_string_to_mobject(path_string) def circle_to_mobject(self, circle_element): x, y, r = [ self.attribute_to_float( circle_element.getAttribute(key) ) if circle_element.hasAttribute(key) else 0.0 for key in ("cx", "cy", "r") ] return Circle(radius=r).shift(x * RIGHT + y * DOWN) def ellipse_to_mobject(self, circle_element): x, y, rx, ry = [ self.attribute_to_float( circle_element.getAttribute(key) ) if circle_element.hasAttribute(key) else 0.0 for key in ("cx", "cy", "rx", "ry") ] result = Circle() result.stretch(rx, 0) result.stretch(ry, 1) result.shift(x * RIGHT + y * DOWN) return result def rect_to_mobject(self, rect_element): fill_color = rect_element.getAttribute("fill") stroke_color = rect_element.getAttribute("stroke") stroke_width = rect_element.getAttribute("stroke-width") corner_radius = rect_element.getAttribute("rx") # input preprocessing fill_opacity = 1 if fill_color in ["", "none", "#FFF", "#FFFFFF"] or Color(fill_color) == Color(WHITE): fill_opacity = 0 fill_color = BLACK # shdn't be necessary but avoids error msgs if fill_color in ["#000", "#000000"]: fill_color = WHITE if stroke_color in ["", "none", "#FFF", "#FFFFFF"] or Color(stroke_color) == Color(WHITE): stroke_width = 0 stroke_color = BLACK if stroke_color in ["#000", "#000000"]: stroke_color = WHITE if stroke_width in ["", "none", "0"]: stroke_width = 0 if corner_radius in ["", "0", "none"]: corner_radius = 0 corner_radius = float(corner_radius) if corner_radius == 0: mob = Rectangle( width=self.attribute_to_float( rect_element.getAttribute("width") ), height=self.attribute_to_float( rect_element.getAttribute("height") ), stroke_width=stroke_width, stroke_color=stroke_color, fill_color=fill_color, fill_opacity=fill_opacity ) else: mob = RoundedRectangle( width=self.attribute_to_float( rect_element.getAttribute("width") ), height=self.attribute_to_float( rect_element.getAttribute("height") ), stroke_width=stroke_width, stroke_color=stroke_color, fill_color=fill_color, fill_opacity=opacity, corner_radius=corner_radius ) mob.shift(mob.get_center() - mob.get_corner(UP + LEFT)) return mob def handle_transforms(self, element, mobject): # TODO, this could use some cleaning... x, y = 0, 0 try: x = self.attribute_to_float(element.getAttribute('x')) # Flip y y = -self.attribute_to_float(element.getAttribute('y')) mobject.shift([x, y, 0]) except Exception: pass transform = element.getAttribute('transform') try: # transform matrix prefix = "matrix(" suffix = ")" if not transform.startswith(prefix) or not transform.endswith(suffix): raise Exception() transform = transform[len(prefix):-len(suffix)] transform = string_to_numbers(transform) transform = np.array(transform).reshape([3, 2]) x = transform[2][0] y = -transform[2][1] matrix = np.identity(self.dim) matrix[:2, :2] = transform[:2, :] matrix[1] *= -1 matrix[:, 1] *= -1 for mob in mobject.family_members_with_points(): mob.apply_matrix(matrix.T) mobject.shift(x * RIGHT + y * UP) except: pass try: # transform scale prefix = "scale(" suffix = ")" if not transform.startswith(prefix) or not transform.endswith(suffix): raise Exception() transform = transform[len(prefix):-len(suffix)] scale_values = string_to_numbers(transform) if len(scale_values) == 2: scale_x, scale_y = scale_values mobject.scale(np.array([scale_x, scale_y, 1]), about_point=ORIGIN) elif len(scale_values) == 1: scale = scale_values[0] mobject.scale(np.array([scale, scale, 1]), about_point=ORIGIN) except: pass try: # transform translate prefix = "translate(" suffix = ")" if not transform.startswith(prefix) or not transform.endswith(suffix): raise Exception() transform = transform[len(prefix):-len(suffix)] x, y = string_to_numbers(transform) mobject.shift(x * RIGHT + y * DOWN) except: pass # TODO, ... def flatten(self, input_list): output_list = [] for i in input_list: if isinstance(i, list): output_list.extend(self.flatten(i)) else: output_list.append(i) return output_list def get_all_childNodes_have_id(self, element): all_childNodes_have_id = [] if not isinstance(element, minidom.Element): return if element.hasAttribute('id'): return [element] for e in element.childNodes: all_childNodes_have_id.append(self.get_all_childNodes_have_id(e)) return self.flatten([e for e in all_childNodes_have_id if e]) def update_ref_to_element(self, defs): new_refs = dict([(e.getAttribute('id'), e) for e in self.get_all_childNodes_have_id(defs)]) self.ref_to_element.update(new_refs) class VMobjectFromSVGPathstring(VMobject): CONFIG = { "long_lines": False, "should_subdivide_sharp_curves": False, "should_remove_null_curves": False, } def __init__(self, path_string, **kwargs): self.path_string = path_string super().__init__(**kwargs) def init_points(self): # After a given svg_path has been converted into points, the result # will be saved to a file so that future calls for the same path # don't need to retrace the same computation. hasher = hashlib.sha256(self.path_string.encode()) path_hash = hasher.hexdigest()[:16] points_filepath = os.path.join(get_mobject_data_dir(), f"{path_hash}_points.npy") tris_filepath = os.path.join(get_mobject_data_dir(), f"{path_hash}_tris.npy") if os.path.exists(points_filepath) and os.path.exists(tris_filepath): self.set_points(np.load(points_filepath)) self.triangulation = np.load(tris_filepath) self.needs_new_triangulation = False else: self.handle_commands() if self.should_subdivide_sharp_curves: # For a healthy triangulation later self.subdivide_sharp_curves() if self.should_remove_null_curves: # Get rid of any null curves self.set_points(self.get_points_without_null_curves()) # SVG treats y-coordinate differently self.stretch(-1, 1, about_point=ORIGIN) # Save to a file for future use np.save(points_filepath, self.get_points()) np.save(tris_filepath, self.get_triangulation()) def get_commands_and_coord_strings(self): all_commands = list(self.get_command_to_function_map().keys()) all_commands += [c.lower() for c in all_commands] pattern = "[{}]".format("".join(all_commands)) return zip( re.findall(pattern, self.path_string), re.split(pattern, self.path_string)[1:] ) def handle_commands(self): relative_point = ORIGIN for command, coord_string in self.get_commands_and_coord_strings(): func, number_types_str = self.command_to_function(command) upper_command = command.upper() if upper_command == "Z": func() # `close_path` takes no arguments continue number_types = np.array(list(number_types_str)) n_numbers = len(number_types_str) number_groups = np.array(string_to_numbers(coord_string)).reshape((-1, n_numbers)) for numbers in number_groups: if command.islower(): # Treat it as a relative command numbers[number_types == "x"] += relative_point[0] numbers[number_types == "y"] += relative_point[1] if upper_command == "A": args = [*numbers[:5], np.array([*numbers[5:7], 0.0])] elif upper_command == "H": args = [np.array([numbers[0], relative_point[1], 0.0])] elif upper_command == "V": args = [np.array([relative_point[0], numbers[0], 0.0])] else: args = list(np.hstack(( numbers.reshape((-1, 2)), np.zeros((n_numbers // 2, 1)) ))) func(*args) relative_point = self.get_last_point() def add_elliptical_arc_to(self, rx, ry, x_axis_rotation, large_arc_flag, sweep_flag, point): def close_to_zero(a, threshold=1e-5): return abs(a) < threshold def solve_2d_linear_equation(a, b, c): """ Using Crammer's rule to solve the linear equation `[a b]x = c` where `a`, `b` and `c` are all 2d vectors. """ def det(a, b): return a[0] * b[1] - a[1] * b[0] d = det(a, b) if close_to_zero(d): raise Exception("Cannot handle 0 determinant.") return [det(c, b) / d, det(a, c) / d] def get_arc_center_and_angles(x0, y0, rx, ry, phi, large_arc_flag, sweep_flag, x1, y1): """ The parameter functions of an ellipse rotated `phi` radians counterclockwise is (on `alpha`): x = cx + rx * cos(alpha) * cos(phi) + ry * sin(alpha) * sin(phi), y = cy + rx * cos(alpha) * sin(phi) - ry * sin(alpha) * cos(phi). Now we have two points sitting on the ellipse: `(x0, y0)`, `(x1, y1)`, corresponding to 4 equations, and we want to hunt for 4 variables: `cx`, `cy`, `alpha0` and `alpha_1`. Let `d_alpha = alpha1 - alpha0`, then: if `sweep_flag = 0` and `large_arc_flag = 1`, then `PI <= d_alpha < 2 * PI`; if `sweep_flag = 0` and `large_arc_flag = 0`, then `0 < d_alpha <= PI`; if `sweep_flag = 1` and `large_arc_flag = 0`, then `-PI <= d_alpha < 0`; if `sweep_flag = 1` and `large_arc_flag = 1`, then `-2 * PI < d_alpha <= -PI`. """ xd = x1 - x0 yd = y1 - y0 if close_to_zero(xd) and close_to_zero(yd): raise Exception("Cannot find arc center since the start point and the end point meet.") # Find `p = cos(alpha1) - cos(alpha0)`, `q = sin(alpha1) - sin(alpha0)` eq0 = [rx * np.cos(phi), ry * np.sin(phi), xd] eq1 = [rx * np.sin(phi), -ry * np.cos(phi), yd] p, q = solve_2d_linear_equation(*zip(eq0, eq1)) # Find `s = (alpha1 - alpha0) / 2`, `t = (alpha1 + alpha0) / 2` # If `sin(s) = 0`, this requires `p = q = 0`, # implying `xd = yd = 0`, which is impossible. sin_s = (p ** 2 + q ** 2) ** 0.5 / 2 if sweep_flag: sin_s = -sin_s sin_s = clip(sin_s, -1, 1) s = np.arcsin(sin_s) if large_arc_flag: if not sweep_flag: s = PI - s else: s = -PI - s sin_t = -p / (2 * sin_s) cos_t = q / (2 * sin_s) cos_t = clip(cos_t, -1, 1) t = np.arccos(cos_t) if sin_t <= 0: t = -t # We can make sure `0 < abs(s) < PI`, `-PI <= t < PI`. alpha0 = t - s alpha_1 = t + s cx = x0 - rx * np.cos(alpha0) * np.cos(phi) - ry * np.sin(alpha0) * np.sin(phi) cy = y0 - rx * np.cos(alpha0) * np.sin(phi) + ry * np.sin(alpha0) * np.cos(phi) return cx, cy, alpha0, alpha_1 def get_point_on_ellipse(cx, cy, rx, ry, phi, angle): return np.array([ cx + rx * np.cos(angle) * np.cos(phi) + ry * np.sin(angle) * np.sin(phi), cy + rx * np.cos(angle) * np.sin(phi) - ry * np.sin(angle) * np.cos(phi), 0 ]) def convert_elliptical_arc_to_quadratic_bezier_curve( cx, cy, rx, ry, phi, start_angle, end_angle, n_components=8 ): theta = (end_angle - start_angle) / n_components / 2 handles = np.array([ get_point_on_ellipse(cx, cy, rx / np.cos(theta), ry / np.cos(theta), phi, a) for a in np.linspace( start_angle + theta, end_angle - theta, n_components, ) ]) anchors = np.array([ get_point_on_ellipse(cx, cy, rx, ry, phi, a) for a in np.linspace( start_angle + theta * 2, end_angle, n_components, ) ]) return handles, anchors phi = x_axis_rotation * DEGREES x0, y0 = self.get_last_point()[:2] cx, cy, start_angle, end_angle = get_arc_center_and_angles( x0, y0, rx, ry, phi, large_arc_flag, sweep_flag, point[0], point[1] ) handles, anchors = convert_elliptical_arc_to_quadratic_bezier_curve( cx, cy, rx, ry, phi, start_angle, end_angle ) for handle, anchor in zip(handles, anchors): self.add_quadratic_bezier_curve_to(handle, anchor) def command_to_function(self, command): return self.get_command_to_function_map()[command.upper()] def get_command_to_function_map(self): """ Associates svg command to VMobject function, and the types of arguments it takes in """ return { "M": (self.start_new_path, "xy"), "L": (self.add_line_to, "xy"), "H": (self.add_line_to, "x"), "V": (self.add_line_to, "y"), "C": (self.add_cubic_bezier_curve_to, "xyxyxy"), "S": (self.add_smooth_cubic_curve_to, "xyxy"), "Q": (self.add_quadratic_bezier_curve_to, "xyxy"), "T": (self.add_smooth_curve_to, "xy"), "A": (self.add_elliptical_arc_to, "-----xy"), "Z": (self.close_path, ""), } def get_original_path_string(self): return self.path_string