Class PlotGraph

mbd/scripts/vis_manim.py:6–423

class PlotGraph(Scene):
    def construct(self):
        render = True
        dt = 1 / 15
        caption_size = 30

        # Create title "Model-Free Diffusion"
        title = Text("Standard Model-Free Diffusion", font_size=40).shift(UP * 3)

        # Create two box side by side
        box1 = Rectangle(width=4, height=4).shift(LEFT * 3)
        box2 = Rectangle(width=4, height=4).shift(RIGHT * 3)

        # Create circle in the center of each box filled with blue
        r = 0.5
        circle1 = Circle(radius=r, color=BLUE, fill_opacity=1).shift(box1.get_center())
        circle2 = Circle(radius=r, color=BLUE, fill_opacity=1).shift(box2.get_center())

        # Create curve from [0, -1.5] to [0, 1.5] passing [-0.5, 0]
        demo_curves = []
        for _ in range(4):
            x = np.random.uniform(0.5, 0.8)
            curve = ParametricFunction(
                lambda t: np.array([x * np.cos(t / 3.0 * np.pi), t, 0]),
                t_range=[-1.5, 1.5],
                color=RED,
            ).shift(box1.get_center())
            demo_curves.append(curve)
        for _ in range(4):
            x = np.random.uniform(0.5, 0.8)
            curve = ParametricFunction(
                lambda t: np.array([-x * np.cos(t / 3.0 * np.pi), t, 0]),
                t_range=[-1.5, 1.5],
                color=RED,
            ).shift(box1.get_center())
            demo_curves.append(curve)

        # Create vector field in the box2
        def demo_vector_field_fn(x):
            value = (x[0] / 0.6) ** 2 + (x[1] / 1.5) ** 2 - 1.0
            vec = 0.2 * np.array([x[0] / 0.6, x[1] / 1.5])
            if value < 0:
                return vec
            else:
                return -vec

        demo_vector_field = ArrowVectorField(
            demo_vector_field_fn,
            x_range=[-2, 2],
            y_range=[-2, 2],
            length_func=lambda norm: np.clip(norm, 0, 0.5),
            color=RED,
        ).shift(box2.get_center())

        demo_title = Text("Demonstrations", font_size=30).shift(
            box1.get_top() + UP * 0.5
        )
        diffusion_title = Text("Diffusion Process", font_size=30).shift(