def tex_to_dvi(tex_file): tex_config = get_tex_config() program = tex_config["executable"] file_type = tex_config["intermediate_filetype"] result = tex_file.replace(".tex", "." + file_type) if not os.path.exists(result): commands = [ program, "-interaction=batchmode", "-halt-on-error", f"-output-directory=\"{os.path.dirname(tex_file)}\"", f"\"{tex_file}\"", ">", os.devnull ] exit_code = os.system(" ".join(commands)) if exit_code != 0: log_file = tex_file.replace(".tex", ".log") log.error("LaTeX Error! Not a worry, it happens to the best of us.") with open(log_file, "r", encoding="utf-8") as file: for line in file.readlines(): if line.startswith("!"): log.debug(f"The error could be: `{line[2:-1]}`") raise LatexError() return result
def prompt_user_for_choice(scene_classes): name_to_class = {} max_digits = len(str(len(scene_classes))) for idx, scene_class in enumerate(scene_classes, start=1): name = scene_class.__name__ print(f"{str(idx).zfill(max_digits)}: {name}") name_to_class[name] = scene_class try: user_input = input( "\nThat module has multiple scenes, " "which ones would you like to render?" "\nScene Name or Number: " ) return [ name_to_class[split_str] if not split_str.isnumeric() else scene_classes[int(split_str) - 1] for split_str in user_input.replace(" ", "").split(",") ] except IndexError: log.error("Invalid scene number") sys.exit(2) except KeyError: log.error("Invalid scene name") sys.exit(2) except EOFError: sys.exit(1)
def get_scenes_to_render(scene_classes, scene_config, config): if config["write_all"]: return [sc(**scene_config) for sc in scene_classes] result = [] for scene_name in config["scene_names"]: found = False for scene_class in scene_classes: if scene_class.__name__ == scene_name: fw_config = scene_config["file_writer_config"] if fw_config["write_to_movie"]: fw_config["total_frames"] = compute_total_frames(scene_class, scene_config) scene = scene_class(**scene_config) result.append(scene) found = True break if not found and (scene_name != ""): log.error(f"No scene named {scene_name} found") if result: return result if len(scene_classes) == 1: result = [scene_classes[0]] else: result = prompt_user_for_choice(scene_classes) return [scene_class(**scene_config) for scene_class in result]
def insert_embed_line(file_name: str, scene_name: str, line_marker: str): """ This is hacky, but convenient. When user includes the argument "-e", it will try to recreate a file that inserts the line `self.embed()` into the end of the scene's construct method. If there is an argument passed in, it will insert the line after the last line in the sourcefile which includes that string. """ with open(file_name, 'r') as fp: lines = fp.readlines() try: scene_line_number = next(i for i, line in enumerate(lines) if line.startswith(f"class {scene_name}")) except StopIteration: log.error(f"No scene {scene_name}") prev_line_num = None n_spaces = None if len(line_marker) == 0: # Find the end of the construct method in_construct = False for index in range(scene_line_number, len(lines) - 1): line = lines[index] if line.lstrip().startswith("def construct"): in_construct = True n_spaces = get_indent(line) + 4 elif in_construct: if len(line.strip()) > 0 and get_indent(line) < n_spaces: prev_line_num = index - 2 break if prev_line_num is None: prev_line_num = len(lines) - 2 elif line_marker.isdigit(): # Treat the argument as a line number prev_line_num = int(line_marker) - 1 elif len(line_marker) > 0: # Treat the argument as a string try: prev_line_num = next(i for i in range(scene_line_number, len(lines) - 1) if line_marker in lines[i]) except StopIteration: log.error(f"No lines matching {line_marker}") sys.exit(2) # Insert and write new file if n_spaces is None: n_spaces = get_indent(lines[prev_line_num]) new_lines = list(lines) new_lines.insert(prev_line_num + 1, " " * n_spaces + "self.embed()\n") alt_file = file_name.replace(".py", "_insert_embed.py") with open(alt_file, 'w') as fp: fp.writelines(new_lines) try: yield alt_file finally: os.remove(alt_file)
def revert_to_checkpoint(self, key: str): if key not in self.checkpoint_states: log.error(f"No checkpoint at {key}") return all_keys = list(self.checkpoint_states.keys()) index = all_keys.index(key) for later_key in all_keys[index + 1:]: self.checkpoint_states.pop(later_key) self.restore_state(self.checkpoint_states[key])
def check_if_overlap(self): span_tuples = sorted(self.tex_spans_dict.keys(), key=lambda t: (t[0], -t[1])) overlapping_span_pairs = [] for i, span_0 in enumerate(span_tuples): for span_1 in span_tuples[i + 1:]: if span_0[1] <= span_1[0]: continue if span_0[1] < span_1[1]: overlapping_span_pairs.append((span_0, span_1)) if overlapping_span_pairs: tex_string = self.tex_string log.error("Overlapping substring pairs occur in MTex:") for span_tuple_pair in overlapping_span_pairs: log.error(", ".join(f"\"{tex_string[slice(*span_tuple)]}\"" for span_tuple in span_tuple_pair)) sys.exit(2)
def get_camera_configuration(args, custom_config): camera_config = {} camera_resolutions = get_custom_config()["camera_resolutions"] if args.low_quality: resolution = camera_resolutions["low"] elif args.medium_quality: resolution = camera_resolutions["med"] elif args.hd: resolution = camera_resolutions["high"] elif args.uhd: resolution = camera_resolutions["4k"] else: resolution = camera_resolutions[ camera_resolutions["default_resolution"]] if args.fps: fps = int(args.fps) else: fps = get_custom_config()["fps"] width_str, height_str = resolution.split("x") width = int(width_str) height = int(height_str) camera_config.update({ "pixel_width": width, "pixel_height": height, "fps": fps, }) try: bg_color = args.color or custom_config["style"]["background_color"] camera_config["background_color"] = colour.Color(bg_color) except ValueError as err: log.error("Please use a valid color") log.error(err) sys.exit(2) # If rendering a transparent image/move, make sure the # scene has a background opacity of 0 if args.transparent: camera_config["background_opacity"] = 0 return camera_config
def get_containing_labels_dict(self) -> dict[tuple[int, int], list[int]]: tex_span_list = self.tex_span_list result = { tex_span: [] for tex_span in tex_span_list } overlapping_tex_span_pairs = [] for index_0, span_0 in enumerate(tex_span_list): for index_1, span_1 in enumerate(tex_span_list[index_0:]): if span_0[1] <= span_1[0]: continue if span_0[1] < span_1[1]: overlapping_tex_span_pairs.append((span_0, span_1)) result[span_0].append(index_0 + index_1) if overlapping_tex_span_pairs: tex_string = self.tex_string log.error("Partially overlapping substrings detected:") for tex_span_pair in overlapping_tex_span_pairs: log.error(", ".join( f"\"{tex_string[slice(*tex_span)]}\"" for tex_span in tex_span_pair )) raise ValueError return result
def get_configuration(args): global __config_file__ # ensure __config_file__ always exists if args.config_file is not None: if not os.path.exists(args.config_file): log.error(f"Can't find {args.config_file}.") if sys.platform == 'win32': log.info( f"Copying default configuration file to {args.config_file}..." ) os.system(f"copy default_config.yml {args.config_file}") elif sys.platform in ["linux2", "darwin"]: log.info( f"Copying default configuration file to {args.config_file}..." ) os.system(f"cp default_config.yml {args.config_file}") else: log.info("Please create the configuration file manually.") log.info("Read configuration from default_config.yml.") else: __config_file__ = args.config_file global_defaults_file = os.path.join(get_manim_dir(), "manimlib", "default_config.yml") if not (os.path.exists(global_defaults_file) or os.path.exists(__config_file__)): log.info( "There is no configuration file detected. Switch to the config file initializer:" ) init_customization() elif not os.path.exists(__config_file__): log.info( f"Using the default configuration file, which you can modify in `{global_defaults_file}`" ) log.info( "If you want to create a local configuration file, you can create a file named" f" `{__config_file__}`, or run `manimgl --config`") custom_config = get_custom_config() check_temporary_storage(custom_config) write_file = any([args.write_file, args.open, args.finder]) if args.transparent: file_ext = ".mov" elif args.gif: file_ext = ".gif" else: file_ext = ".mp4" dir_config = custom_config["directories"] output_directory = args.video_dir or dir_config["output"] if dir_config["mirror_module_path"] and args.file: to_cut = dir_config["removed_mirror_prefix"] ext = os.path.abspath(args.file) ext = ext.replace(to_cut, "").replace(".py", "") if ext.startswith("_"): ext = ext[1:] output_directory = os.path.join(output_directory, ext) file_writer_config = { "write_to_movie": not args.skip_animations and write_file, "break_into_partial_movies": custom_config["break_into_partial_movies"], "save_last_frame": args.skip_animations and write_file, "save_pngs": args.save_pngs, # If -t is passed in (for transparent), this will be RGBA "png_mode": "RGBA" if args.transparent else "RGB", "movie_file_extension": file_ext, "output_directory": output_directory, "file_name": args.file_name, "input_file_path": args.file or "", "open_file_upon_completion": args.open, "show_file_location_upon_completion": args.finder, "quiet": args.quiet, } module = get_module(args.file) if args.embed is not None: with insert_embed_line(args.file, args.scene_names[0], args.embed) as alt_file: module = get_module(alt_file) config = { "module": module, "scene_names": args.scene_names, "file_writer_config": file_writer_config, "quiet": args.quiet or args.write_all, "write_all": args.write_all, "skip_animations": args.skip_animations, "start_at_animation_number": args.start_at_animation_number, "end_at_animation_number": None, "preview": not write_file, "presenter_mode": args.presenter_mode, "leave_progress_bars": args.leave_progress_bars, "show_animation_progress": args.show_animation_progress, } # Camera configuration config["camera_config"] = get_camera_configuration(args, custom_config) # Default to making window half the screen size # but make it full screen if -f is passed in monitors = get_monitors() mon_index = custom_config["window_monitor"] monitor = monitors[min(mon_index, len(monitors) - 1)] window_width = monitor.width if not (args.full_screen or custom_config["full_screen"]): window_width //= 2 window_height = window_width * 9 // 16 config["window_config"] = { "size": (window_width, window_height), } # Arguments related to skipping stan = config["start_at_animation_number"] if stan is not None: if "," in stan: start, end = stan.split(",") config["start_at_animation_number"] = int(start) config["end_at_animation_number"] = int(end) else: config["start_at_animation_number"] = int(stan) return config
def parse_cli(): try: parser = argparse.ArgumentParser() module_location = parser.add_mutually_exclusive_group() module_location.add_argument( "file", nargs="?", help="Path to file holding the python code for the scene", ) parser.add_argument( "scene_names", nargs="*", help="Name of the Scene class you want to see", ) parser.add_argument( "-w", "--write_file", action="store_true", help="Render the scene as a movie file", ) parser.add_argument( "-s", "--skip_animations", action="store_true", help="Save the last frame", ) parser.add_argument( "-l", "--low_quality", action="store_true", help="Render at a low quality (for faster rendering)", ) parser.add_argument( "-m", "--medium_quality", action="store_true", help="Render at a medium quality", ) parser.add_argument( "--hd", action="store_true", help="Render at a 1080p", ) parser.add_argument( "--uhd", action="store_true", help="Render at a 4k", ) parser.add_argument( "-f", "--full_screen", action="store_true", help="Show window in full screen", ) parser.add_argument( "-p", "--presenter_mode", action="store_true", help="Scene will stay paused during wait calls until " "space bar or right arrow is hit, like a slide show") parser.add_argument( "-g", "--save_pngs", action="store_true", help="Save each frame as a png", ) parser.add_argument( "-i", "--gif", action="store_true", help="Save the video as gif", ) parser.add_argument( "-t", "--transparent", action="store_true", help="Render to a movie file with an alpha channel", ) parser.add_argument( "-q", "--quiet", action="store_true", help="", ) parser.add_argument( "-a", "--write_all", action="store_true", help="Write all the scenes from a file", ) parser.add_argument( "-o", "--open", action="store_true", help="Automatically open the saved file once its done", ) parser.add_argument( "--finder", action="store_true", help="Show the output file in finder", ) parser.add_argument( "--config", action="store_true", help="Guide for automatic configuration", ) parser.add_argument( "--file_name", help="Name for the movie or image file", ) parser.add_argument( "-n", "--start_at_animation_number", help="Start rendering not from the first animation, but " "from another, specified by its index. If you pass " "in two comma separated values, e.g. \"3,6\", it will end " "the rendering at the second value", ) parser.add_argument( "-e", "--embed", nargs="?", const="", help="Creates a new file where the line `self.embed` is inserted " "into the Scenes construct method. " "If a string is passed in, the line will be inserted below the " "last line of code including that string.") parser.add_argument( "-r", "--resolution", help="Resolution, passed as \"WxH\", e.g. \"1920x1080\"", ) parser.add_argument( "--fps", help="Frame rate, as an integer", ) parser.add_argument( "-c", "--color", help="Background color", ) parser.add_argument( "--leave_progress_bars", action="store_true", help="Leave progress bars displayed in terminal", ) parser.add_argument( "--show_animation_progress", action="store_true", help="Show progress bar for each animation", ) parser.add_argument( "--video_dir", help="Directory to write video", ) parser.add_argument( "--config_file", help="Path to the custom configuration file", ) parser.add_argument("-v", "--version", action="store_true", help="Display the version of manimgl") parser.add_argument( "--log-level", help= "Level of messages to Display, can be DEBUG / INFO / WARNING / ERROR / CRITICAL" ) args = parser.parse_args() return args except argparse.ArgumentError as err: log.error(str(err)) sys.exit(2)