manimの作法 その49

概要

manimの作法、調べてみた。
ComplexPlane使ってみた。

サンプルコード

from manimlib.imports import *

class test(Scene):
	CONFIG = {
		"plane_config": {
			"x_radius": 9,
			"y_radius": 9,
			"stroke_width": 3,
		},
		"background_plane_config": {
			"color": LIGHT_GREY,
			"secondary_color": DARK_GREY,
			"stroke_width": 0.5,
			"stroke_opacity": 0.5,
			"secondary_line_ratio": 0,
		}
	}
	def construct(self):
		self.add_planes()
		z_tuples = [(complex(2, 1), "2 + i", UP), (complex(5, 2), "5 + 2i", LEFT), (complex(-np.sqrt(2) / 2, np.sqrt(2) / 2), "-\\frac{\\sqrt{2}}{2} + \\frac{\\sqrt{2}}{2} i", LEFT, ), (complex(-4, 1.5), "-4 + 1.5i", RIGHT), (complex(3, 0), "3 + 0i", UP), (complex(4, -3), "4 + -3i", UP), ]
		for z, z_tex, label_vect in z_tuples:
			self.show_multiplication(z, z_tex, label_vect)
	def add_planes(self, include_title = True):
		plane = ComplexPlane(**self.plane_config)
		self.plane = plane
		background_plane = ComplexPlane(**self.background_plane_config)
		background_plane.add_coordinates()
		self.background_plane = background_plane
		self.add(background_plane)
		self.add(plane)
		if include_title:
			title = TextMobject("Complex plane")
			title.scale(1.5)
			title.to_corner(UL, buff = MED_LARGE_BUFF)
			title.shift(SMALL_BUFF * UR)
			self.title = title
			self.add_foreground_mobjects(title)
	def show_multiplication(self, z, z_tex, label_vect):
		z_color = WHITE
		plane = self.plane
		new_plane = plane.deepcopy()
		real_tex, imag_tex = z_tex.split("+")
		label = TexMobject("\\text{Multiply by}\\\\", real_tex, "+", imag_tex, alignment = "", )
		label[1].set_color(GREEN)
		label[3].set_color(RED)
		label.scale(1.2)
		h_line = Line(plane.number_to_point(0), plane.number_to_point(z.real), color = GREEN, stroke_width = 5, )
		v_line = Line(plane.number_to_point(z.real), plane.number_to_point(z), color = RED, stroke_width = 5, )
		lines = VGroup(h_line, v_line)
		z_point = plane.number_to_point(z)
		z_dot = Dot(z_point)
		z_dot.set_color(z_color)
		label[1:].next_to(z_dot, label_vect)
		label[0].next_to(label[1:], UP)
		for mob in label:
			label.add_to_back(BackgroundRectangle(mob))
		one_dot = Dot(plane.number_to_point(1))
		one_dot.set_color(YELLOW)
		for dot in z_dot, one_dot:
			dot.save_state()
			dot.scale(5)
			dot.set_fill(opacity = 0)
			dot.set_stroke(width = 1, opacity = 0.5)
		to_fade_out = VGroup(plane, label, lines, z_dot, one_dot)
		self.play(ShowCreation(lines), FadeInFromDown(label), Restore(z_dot), )
		self.play(Restore(one_dot))
		angle = np.log(z).imag
		self.play(one_dot.move_to, z_dot, plane.apply_complex_function, lambda w: z * w, path_arc = angle, run_time = 3)
		self.wait()
		self.play(FadeOut(to_fade_out), FadeIn(new_plane), )
		self.plane = new_plane

生成した動画

以上。