Advanced Code Extensions

This Manim extension allows you to quickly create arrays, assign elements, point to elements, push, pop and etc. It sits on top of the Manim Community Edition.

Overview:

Array

In the works:

Code Highlighting by text search

TODO:

Make the array flexible i.e. each element with different Rectangles
2D Array, maybe even 3D arrays
Stack
Tests

Array

The Array object has many functionalities. It creates a Rectangle and Tex object for each element, also it creates a Tex object for its name.

Example code:

array = Array(
    "Array",
    *range(6), # this is the array
    name_config={"fill_color": WHITE}
)
array.create_array(sq_size=0.5, name_size=1.2)
for anims in array.draw_array():
    self.play(anims, run_time=0.5)

Creating a pointer to an element in the array using the TextPointer class from our library:

# create a pointer with a text label
pointer = TextPointer("Here", 0.8, UP)
pointer.next_to(array.get_square(2), DOWN)
self.play(*pointer.draw(), run_time=0.5)

# connect the pointer, so when you transform the array
# it will transform the pointer as well
array.connect_pointer(pointer)

self.play(
    ApplyMethod(pointer.next_to, array.get_square(4), DOWN, aligned_edge=UP),
    run_time=0.5
)
self.play(*array.indicate_at(4), run_time=0.5)
self.play(Indicate(pointer), run_time=0.5)

The array is a VGroup, so you can transform it as a whole.

self.play(ApplyMethod(array.scale, 1.1), run_time=0.5)

Popping an element:

pop_anims = list(array.pop(index=3))
self.play(*pop_anims[:-1]) # pop the element
self.wait()
self.play(*pop_anims[-1:]) # shift the others
self.wait()

In the examples folder you can find the full working example.

Functions

create_array (sq_size: int, name_size: int = 1, **kwargs): Creates all the necessary objects with the given text and square size.
draw_array (): Returns the animations that draw the array on the screen.
pop (index: int): Returns the animation of poping an element from the array. It deletes the objects associated with it, but not the pointers. It shifts all the other elements appropriately.
get_square (index: int): Gets the Square object of an element with the given index.

TextPointer and Pointer

These two classes allow you to create a simple arrow pointers with/without a label.

# create a pointer with a text label
pointer = TextPointer("Here", 0.8, UP)

# this is an example of positioning
pointer.next_to(array.get_square(2), DOWN)

# drawing the pointer and its label
self.play(*pointer.draw(), run_time=0.5)

Contributing

This is a fairly small project, so I think there are many of you guys who can easily contribute to it! Feel free to contribute and suggest new features! The plans for the future are to make the array much more flexible and create different representations, such as 2D and 3D arrays, stacks, etc.

Notes