manim_matrix.py

from manimlib.scene.scene import Scene
from manimlib.mobject.matrix import Matrix
from manimlib import *

class manim_mobject_matrix_001a(Scene):
    def construct(self):
        sans = 'monospace'
        size = 0.5
        play_kw = {"run_time": 0.5}

        lines = VGroup(

            Tex("Kernel Matrix", size=size, color=BLUE, t2c={"Command:": BLUE}),
            Matrix([("0", "-1", "0"),
                    ("-1", "5", "-1"),
                    ("0", "-1", "0")], size=size),

            Matrix([("10", "20", "45"),
                    ("45", "97  ", "123"),
                    ("133", "56", "75")], size=size),

            Tex("*", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("=", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex(".", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("(", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex(")", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("+", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD),

            Matrix([("0", "-20", "0"),
                    ("-45", "485", "-123"),
                    ("0", "-56", "0")], size=size),

            Tex("241", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD)

        )


        op = VGroup(
            Tex("0", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("-20", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("0", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("-45", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("485", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("-123", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("0", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("-56", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD),
            Tex("0", size=size, t2c={"Command:": BLUE}, font=sans, weight=BOLD)
        )

        t1_1 = lines[0]
        kernel_matrix = lines[1]
        image_matrix = lines[2]
        multi = lines[3]
        equal = lines[4]
        m_mul = lines[5]
        b_open = lines[6]
        b_close = lines[7]
        plus = lines[8]
        mm = lines[9]
        sum_oppp = lines[-1]
        t1_1.shift(UL * 3)

        kernel_matrix.shift(UP)
        self.play(Write(kernel_matrix))
        self.play(ApplyMethod(kernel_matrix.shift, LEFT*5))
        multi.next_to(kernel_matrix)
        image_matrix.next_to(multi)
        equal.next_to(image_matrix)
        self.play(Write(multi))
        self.play(Write(image_matrix))
        self.play(Write(equal))
        mm.next_to(equal)


        final_mat = VGroup(*mm)
        final_mat1 = VGroup(*final_mat[0])

        for i in final_mat[1:]:
            self.play(Write(i))

        kk = VGroup(*kernel_matrix)
        k1 = VGroup(*kk[0])
        ii = VGroup(*image_matrix)
        i1 = VGroup(*ii[0])

        runtime = 0.1
        flag = True

        for i, j, final_mm, o1 in zip(k1.copy(), i1.copy(), final_mat1.copy(), op):
            i2 = i.copy()
            j2 = j.copy()
            f2 = final_mm.copy()
            m_mul1 = m_mul.copy()
            b_o1 = b_open.copy()
            b_c1 = b_close.copy()
            eq1 = equal.copy()

            b_o1.shift(DOWN*2)
            b_o1.shift(LEFT * 3)
            i2.set_color(RED)
            self.play(Write(b_o1), **play_kw)
            self.play(ApplyMethod(i2.next_to, b_o1), **play_kw)

            m_mul1.next_to(i2)
            self.play(Write(m_mul1), **play_kw)
            j2.set_color(RED)
            self.play(ApplyMethod(j2.next_to, m_mul1), **play_kw)
            b_c1.next_to(j2)
            self.play(Write(b_c1), **play_kw)
            eq1.next_to(b_c1)
            self.play(Write(eq1), **play_kw)
            o1.next_to(eq1)
            self.play(Write(o1),**play_kw)

            self.play(FadeOut(b_o1), FadeOut(i2), FadeOut(m_mul1), FadeOut(j2),
                      FadeOut(b_c1), FadeOut(eq1), Transform(o1, f2))

        self.play(FadeOut(equal), FadeOut(image_matrix), FadeOut(multi), FadeOut(kernel_matrix))
        summ = Tex(r"\sum", size=size, color=BLUE, t2c={"Command:": BLUE}, font=sans, weight=BOLD)
        summ.shift(UP)
        summ.shift(LEFT*3)

        self.play(Write(summ))
        self.play(FadeOut(op), ApplyMethod(mm.next_to, summ))

        ii = VGroup(*mm)
        m1 = VGroup(*ii[0])
        len_mm = len(m1)
        count = 0
        flag = True
        for i in m1.copy():
            count = count + 1
            i1 = i.copy()
            ss = sum_oppp.copy()
            if flag:
                p1 = plus.copy()
                i1.set_color(RED)
                self.play(ApplyMethod(i1.shift, DOWN*4), **play_kw)
                self.play(ApplyMethod(i1.shift, LEFT * 5), **play_kw)
                p1.next_to(i1)
                self.play(Write(p1), **play_kw)
                flag = False
            else:
                p1 = p1.copy()
                i1.set_color(RED)
                self.play(ApplyMethod(i1.next_to, p1), **play_kw)
                if count >= len_mm:
                    equal.next_to(i1)
                    self.play(Write(equal), **play_kw)
                    ss.next_to(equal)
                    self.play(Write(ss), **play_kw)
                    equal.set_color(YELLOW)
                    self.play(ApplyMethod(equal.next_to, mm), **play_kw)
                    ss.set_color(YELLOW)
                    self.play(ApplyMethod(ss.next_to, equal), **play_kw)

                else:
                    p1.next_to(i1)
                    self.play(Write(p1), **play_kw)
        self.wait(2)
        self.clear()
        self.wait(2)