add flatten() to CodeMobject

eulertour · Dec 3, 2018 · a8fba5d · a8fba5d
1 parent 183bab9
commit a8fba5d
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/mobject/svg/tex_mobject.py b/mobject/svg/tex_mobject.py
@@ -160,6 +160,19 @@ def has_header(self):
                 break
         return ret
 
+    def flatten(self):
+        if type(self.submobjects[0]).__name__ == "TexSymbol":
+            return [self]
+        elif type(self.submobjects[0]).__name__ == "SingleStringTexMobject":
+            result = []
+            for mob in self.submobjects:
+                result.extend(mob.flatten())
+            return result
+        else:
+            raise Exception("malformed SingleStringTexMobject")
+            breakpoint(context=9)
+
+
 
 class TexMobject(SingleStringTexMobject):
     CONFIG = {
@@ -319,7 +332,7 @@ def break_up_tex_strings(self, tex_string):
     def modify_special_strings(self, tex):
         return tex
 
-    def break_up_by_substrings(self):
+    def break_up_by_substrings(self, **kwargs):
         index, mob = self.organize_by_blocks(self.tex_string)
         self.submobjects = mob.submobjects