public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use Integer class for NumberLine

Browse source
This commit is contained in:
Grant Sanderson 2018-08-25 18:32:09 -07:00
parent 979cd3a38f
commit 94b1c9d3af
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
mobject/number_line.py
View file

@ -2,7 +2,7 @@
from constants import * from constants import *
from mobject.svg.tex_mobject import TexMobject from mobject.numbers import DecimalNumber
from mobject.types.vectorized_mobject import VGroup from mobject.types.vectorized_mobject import VGroup
from mobject.types.vectorized_mobject import VMobject from mobject.types.vectorized_mobject import VMobject
from mobject.geometry import Arrow from mobject.geometry import Arrow
@ -33,6 +33,9 @@ class NumberLine(VMobject):
"line_to_number_buff": MED_SMALL_BUFF, "line_to_number_buff": MED_SMALL_BUFF,
"include_tip": False, "include_tip": False,
"propagate_style_to_family": True, "propagate_style_to_family": True,
"decimal_number_config": {
"num_decimal_places": 0,
}
} }
def __init__(self, **kwargs): def __init__(self, **kwargs):
@ -120,11 +123,11 @@ class NumberLine(VMobject):
# TODO, handle decimals # TODO, handle decimals
if len(numbers) == 0: if len(numbers) == 0:
numbers = self.default_numbers_to_display() numbers = self.default_numbers_to_display()
if "force_integers" in kwargs and kwargs["force_integers"]:
numbers = list(map(int, numbers))
result = VGroup() result = VGroup()
for number in numbers: for number in numbers:
mob = TexMobject(str(number)) mob = DecimalNumber(
number, **self.decimal_number_config
)
mob.scale(self.number_scale_val) mob.scale(self.number_scale_val)
mob.next_to( mob.next_to(
self.number_to_point(number), self.number_to_point(number),