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

reorganize media file tree

Browse source
This commit is contained in:
Devin Neal 2019-06-03 23:41:05 -07:00
parent c203f8e8c0
commit caa4577cd1
8 changed files with 36 additions and 52 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
manim.py Normal file → Executable file
View file

21
manimlib/constants.py
View file

@ -20,28 +20,19 @@ if not os.path.isdir(MEDIA_DIR):
"this behavior by writing a different directory to media_dir.txt." "this behavior by writing a different directory to media_dir.txt."
) )
VIDEO_DIR = os.path.join(MEDIA_DIR, "videos") VIDEO_DIR = os.path.join(MEDIA_DIR, "video")
RASTER_IMAGE_DIR = os.path.join(MEDIA_DIR, "designs", "raster_images") ASSETS_DIR = os.path.join(MEDIA_DIR, "assets")
SVG_IMAGE_DIR = os.path.join(MEDIA_DIR, "designs", "svg_images") TEX_DIR = os.path.join(MEDIA_DIR, "Tex")
SOUND_DIR = os.path.join(MEDIA_DIR, "designs", "sounds")
###
THIS_DIR = os.path.dirname(os.path.realpath(__file__))
FILE_DIR = os.path.join(os.getenv("FILE_DIR", default="."), "files")
TEX_DIR = os.path.join(FILE_DIR, "Tex")
# These two may be depricated now.
MOBJECT_DIR = os.path.join(FILE_DIR, "mobjects")
IMAGE_MOBJECT_DIR = os.path.join(MOBJECT_DIR, "image")
for folder in [FILE_DIR, RASTER_IMAGE_DIR, SVG_IMAGE_DIR, VIDEO_DIR, for folder in [VIDEO_DIR, ASSETS_DIR, TEX_DIR]:
TEX_DIR, MOBJECT_DIR, IMAGE_MOBJECT_DIR]:
if not os.path.exists(folder): if not os.path.exists(folder):
os.makedirs(folder) os.makedirs(folder)
TEX_USE_CTEX = False TEX_USE_CTEX = False
TEX_TEXT_TO_REPLACE = "YourTextHere" TEX_TEXT_TO_REPLACE = "YourTextHere"
TEMPLATE_TEX_FILE = os.path.join( TEMPLATE_TEX_FILE = os.path.join(
THIS_DIR, "tex_template.tex" if not TEX_USE_CTEX os.path.dirname(os.path.realpath(__file__)),
else "ctex_template.tex" "tex_template.tex" if not TEX_USE_CTEX else "ctex_template.tex"
) )
with open(TEMPLATE_TEX_FILE, "r") as infile: with open(TEMPLATE_TEX_FILE, "r") as infile:
TEMPLATE_TEXT_FILE_BODY = infile.read() TEMPLATE_TEXT_FILE_BODY = infile.read()

4
manimlib/for_3b1b_videos/pi_creature.py
View file

@ -15,11 +15,11 @@ from manimlib.utils.config_ops import digest_config
from manimlib.utils.space_ops import get_norm from manimlib.utils.space_ops import get_norm
from manimlib.utils.space_ops import normalize from manimlib.utils.space_ops import normalize
pi_creature_dir_maybe = os.path.join(MEDIA_DIR, "designs", "PiCreature") pi_creature_dir_maybe = os.path.join(MEDIA_DIR, "assets", "PiCreature")
if os.path.exists(pi_creature_dir_maybe): if os.path.exists(pi_creature_dir_maybe):
PI_CREATURE_DIR = pi_creature_dir_maybe PI_CREATURE_DIR = pi_creature_dir_maybe
else: else:
PI_CREATURE_DIR = os.path.join(FILE_DIR) PI_CREATURE_DIR = os.path.join("assets")
PI_CREATURE_SCALE_FACTOR = 0.5 PI_CREATURE_SCALE_FACTOR = 0.5

6
manimlib/mobject/svg/svg_mobject.py
View file

@ -50,9 +50,9 @@ class SVGMobject(VMobject):
if self.file_name is None: if self.file_name is None:
raise Exception("Must specify file for SVGMobject") raise Exception("Must specify file for SVGMobject")
possible_paths = [ possible_paths = [
os.path.join(SVG_IMAGE_DIR, self.file_name), os.path.join(os.path.join("assets", "svg_images"), self.file_name),
os.path.join(SVG_IMAGE_DIR, self.file_name + ".svg"), os.path.join(os.path.join("assets", "svg_images"), self.file_name + ".svg"),
os.path.join(SVG_IMAGE_DIR, self.file_name + ".xdv"), os.path.join(os.path.join("assets", "svg_images"), self.file_name + ".xdv"),
self.file_name, self.file_name,
] ]
for path in possible_paths: for path in possible_paths:

49
manimlib/scene/scene_file_writer.py
View file

@ -13,7 +13,7 @@ from manimlib.constants import STREAMING_PORT
from manimlib.constants import STREAMING_PROTOCOL from manimlib.constants import STREAMING_PROTOCOL
from manimlib.constants import VIDEO_DIR from manimlib.constants import VIDEO_DIR
from manimlib.utils.config_ops import digest_config from manimlib.utils.config_ops import digest_config
from manimlib.utils.file_ops import guarantee_existance from manimlib.utils.file_ops import guarantee_existence
from manimlib.utils.file_ops import add_extension_if_not_present from manimlib.utils.file_ops import add_extension_if_not_present
from manimlib.utils.file_ops import get_sorted_integer_files from manimlib.utils.file_ops import get_sorted_integer_files
from manimlib.utils.sounds import get_full_sound_file_path from manimlib.utils.sounds import get_full_sound_file_path
@ -35,7 +35,6 @@ class SceneFileWriter(object):
# TODO, address this in extract_scene et. al. # TODO, address this in extract_scene et. al.
"file_name": None, "file_name": None,
"output_directory": None, "output_directory": None,
"file_name": None,
} }
def __init__(self, scene, **kwargs): def __init__(self, scene, **kwargs):
@ -47,66 +46,62 @@ class SceneFileWriter(object):
# Output directories and files # Output directories and files
def init_output_directories(self): def init_output_directories(self):
output_directory = self.output_directory or self.get_default_output_directory() module_directory = self.output_directory or self.get_default_module_directory()
file_name = self.file_name or self.get_default_file_name() scene_name = self.file_name or self.get_default_scene_name()
if self.save_last_frame: if self.save_last_frame:
image_dir = guarantee_existance(os.path.join( image_dir = guarantee_existence(os.path.join(
VIDEO_DIR, VIDEO_DIR,
output_directory, module_directory,
self.get_image_directory(), scene_name,
"images",
)) ))
self.image_file_path = os.path.join( self.image_file_path = os.path.join(
image_dir, image_dir,
add_extension_if_not_present(file_name, ".png") add_extension_if_not_present(scene_name, ".png")
) )
if self.write_to_movie: if self.write_to_movie:
movie_dir = guarantee_existance(os.path.join( movie_dir = guarantee_existence(os.path.join(
VIDEO_DIR, VIDEO_DIR,
output_directory, module_directory,
self.get_movie_directory(), scene_name,
self.get_resolution_directory(),
)) ))
self.movie_file_path = os.path.join( self.movie_file_path = os.path.join(
movie_dir, movie_dir,
add_extension_if_not_present( add_extension_if_not_present(
file_name, self.movie_file_extension scene_name, self.movie_file_extension
) )
) )
self.gif_file_path = os.path.join( self.gif_file_path = os.path.join(
movie_dir, movie_dir,
add_extension_if_not_present( add_extension_if_not_present(
file_name, self.gif_file_extension scene_name, self.gif_file_extension
) )
) )
self.partial_movie_directory = guarantee_existance(os.path.join( self.partial_movie_directory = guarantee_existence(os.path.join(
movie_dir, movie_dir,
self.get_partial_movie_directory(), "partial_movie_files",
file_name, scene_name,
)) ))
def get_default_output_directory(self): def get_default_module_directory(self):
filename = os.path.basename(self.input_file_path) filename = os.path.basename(self.input_file_path)
root, ext = os.path.splitext(filename) root, _ = os.path.splitext(filename)
return root if root else ext[1:] return root
def get_default_file_name(self): def get_default_scene_name(self):
if self.file_name is None: if self.file_name is None:
return self.scene.__class__.__name__ return self.scene.__class__.__name__
else: else:
return self.file_name return self.file_name
def get_movie_directory(self): def get_resolution_directory(self):
pixel_height = self.scene.camera.pixel_height pixel_height = self.scene.camera.pixel_height
frame_rate = self.scene.camera.frame_rate frame_rate = self.scene.camera.frame_rate
return "{}p{}".format( return "{}p{}".format(
pixel_height, frame_rate pixel_height, frame_rate
) )
def get_image_directory(self):
return "images"
def get_partial_movie_directory(self):
return "partial_movie_files"
# Directory getters # Directory getters
def get_image_file_path(self): def get_image_file_path(self):
return self.image_file_path return self.image_file_path

2
manimlib/utils/file_ops.py
View file

@ -10,7 +10,7 @@ def add_extension_if_not_present(file_name, extension):
return file_name return file_name
def guarantee_existance(path): def guarantee_existence(path):
if not os.path.exists(path): if not os.path.exists(path):
os.makedirs(path) os.makedirs(path)
return os.path.abspath(path) return os.path.abspath(path)

3
manimlib/utils/images.py
View file

@ -3,14 +3,13 @@ import os
from PIL import Image from PIL import Image
from manimlib.constants import RASTER_IMAGE_DIR
from manimlib.utils.file_ops import seek_full_path_from_defaults from manimlib.utils.file_ops import seek_full_path_from_defaults
def get_full_raster_image_path(image_file_name): def get_full_raster_image_path(image_file_name):
return seek_full_path_from_defaults( return seek_full_path_from_defaults(
image_file_name, image_file_name,
default_dir=RASTER_IMAGE_DIR, default_dir=os.path.join("assets", "raster_images"),
extensions=[".jpg", ".png", ".gif"] extensions=[".jpg", ".png", ".gif"]
) )

3
manimlib/utils/sounds.py
View file

@ -1,5 +1,4 @@
import os import os
from manimlib.constants import SOUND_DIR
from manimlib.utils.file_ops import seek_full_path_from_defaults from manimlib.utils.file_ops import seek_full_path_from_defaults
@ -35,6 +34,6 @@ def play_finish_sound():
def get_full_sound_file_path(sound_file_name): def get_full_sound_file_path(sound_file_name):
return seek_full_path_from_defaults( return seek_full_path_from_defaults(
sound_file_name, sound_file_name,
default_dir=SOUND_DIR, default_dir=os.path.join("assets", "sounds"),
extensions=[".wav", ".mp3"] extensions=[".wav", ".mp3"]
) )