l jzdZddlmZgdZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZdd lmZd d lmZd dlmZd dlmZmZd dlmZd dl m!Z!m"Z"d dl#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)Gdde!Z*Gdde*Z+Gdde*Z,Gdde*Z-Gdde*Z.dS)aMobjects representing tables. Examples -------- .. manim:: TableExamples :save_last_frame: class TableExamples(Scene): def construct(self): t0 = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")], top_left_entry=Text("TOP")) t0.add_highlighted_cell((2,2), color=GREEN) x_vals = np.linspace(-2,2,5) y_vals = np.exp(x_vals) t1 = DecimalTable( [x_vals, y_vals], row_labels=[MathTex("x"), MathTex("f(x)")], include_outer_lines=True) t1.add(t1.get_cell((2,2), color=RED)) t2 = MathTable( [["+", 0, 5, 10], [0, 0, 5, 10], [2, 2, 7, 12], [4, 4, 9, 14]], include_outer_lines=True) t2.get_horizontal_lines()[:3].set_color(BLUE) t2.get_vertical_lines()[:3].set_color(BLUE) t2.get_horizontal_lines()[:3].set_z_index(1) cross = VGroup( Line(UP + LEFT, DOWN + RIGHT), Line(UP + RIGHT, DOWN + LEFT)) a = Circle().set_color(RED).scale(0.5) b = cross.set_color(BLUE).scale(0.5) t3 = MobjectTable( [[a.copy(),b.copy(),a.copy()], [b.copy(),a.copy(),a.copy()], [a.copy(),b.copy(),b.copy()]]) t3.add(Line( t3.get_corner(DL), t3.get_corner(UR) ).set_color(RED)) vals = np.arange(1,21).reshape(5,4) t4 = IntegerTable( vals, include_outer_lines=True ) g1 = Group(t0, t1).scale(0.5).arrange(buff=1).to_edge(UP, buff=1) g2 = Group(t2, t3, t4).scale(0.5).arrange(buff=1).to_edge(DOWN, buff=1) self.add(g1, g2) ) annotations)Table MathTable MobjectTable IntegerTable DecimalTableN)CallableIterableSequence)Line)Polygon)BackgroundRectangle) DecimalNumberInteger)MathTex) Paragraph) Animation)AnimationGroup)CreateWrite)FadeIn)VGroupVMobject)BLACK PURE_YELLOW ManimColorParsableManimColor)get_vectorized_mobject_classceZdZdZdddddddedeeiiifdLfd ZdMd ZdNd#ZdOd%Z dPd&Z dPd'Z dQd(Z dQd)Z dQd*ZdQd+ZdRd.ZdRd/Z dSdTd3Z dSdTd4ZdQd5ZdQd6ZdQd7ZefdUd9ZdVdWd=Zd:efdXd?Zd:efdYd@ZdAeeeefdZdIZd[fdK Z xZ!S)\raA mobject that displays a table on the screen. Parameters ---------- table A 2D array or list of lists. Content of the table has to be a valid input for the callable set in ``element_to_mobject``. row_labels List of :class:`~.VMobject` representing the labels of each row. col_labels List of :class:`~.VMobject` representing the labels of each column. top_left_entry The top-left entry of the table, can only be specified if row and column labels are given. v_buff Vertical buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 0.8. h_buff Horizontal buffer passed to :meth:`~.Mobject.arrange_in_grid`, by default 1.3. include_outer_lines ``True`` if the table should include outer lines, by default False. add_background_rectangles_to_entries ``True`` if background rectangles should be added to entries, by default ``False``. entries_background_color Background color of entries if ``add_background_rectangles_to_entries`` is ``True``. include_background_rectangle ``True`` if the table should have a background rectangle, by default ``False``. background_rectangle_color Background color of table if ``include_background_rectangle`` is ``True``. element_to_mobject The :class:`~.Mobject` class applied to the table entries. by default :class:`~.Paragraph`. For common choices, see :mod:`~.text_mobject`/:mod:`~.tex_mobject`. element_to_mobject_config Custom configuration passed to :attr:`element_to_mobject`, by default {}. arrange_in_grid_config Dict passed to :meth:`~.Mobject.arrange_in_grid`, customizes the arrangement of the table. line_config Dict passed to :class:`~.Line`, customizes the lines of the table. kwargs Additional arguments to be passed to :class:`~.VGroup`. Examples -------- .. manim:: TableExamples :save_last_frame: class TableExamples(Scene): def construct(self): t0 = Table( [["This", "is a"], ["simple", "Table in \\n Manim."]]) t1 = Table( [["This", "is a"], ["simple", "Table."]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")]) t1.add_highlighted_cell((2,2), color=YELLOW) t2 = Table( [["This", "is a"], ["simple", "Table."]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")], top_left_entry=Star().scale(0.3), include_outer_lines=True, arrange_in_grid_config={"cell_alignment": RIGHT}) t2.add(t2.get_cell((2,2), color=RED)) t3 = Table( [["This", "is a"], ["simple", "Table."]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")], top_left_entry=Star().scale(0.3), include_outer_lines=True, line_config={"stroke_width": 1, "color": YELLOW}) t3.remove(*t3.get_vertical_lines()) g = Group( t0,t1,t2,t3 ).scale(0.7).arrange_in_grid(buff=1) self.add(g) .. manim:: BackgroundRectanglesExample :save_last_frame: class BackgroundRectanglesExample(Scene): def construct(self): background = Rectangle(height=6.5, width=13) background.set_fill(opacity=.5) background.set_color([TEAL, RED, YELLOW]) self.add(background) t0 = Table( [["This", "is a"], ["simple", "Table."]], add_background_rectangles_to_entries=True) t1 = Table( [["This", "is a"], ["simple", "Table."]], include_background_rectangle=True) g = Group(t0, t1).scale(0.7).arrange(buff=0.5) self.add(g) Ng?g?Ftable*Iterable[Iterable[float | str | VMobject]] row_labelsIterable[VMobject] | None col_labelstop_left_entryVMobject | Nonev_bufffloath_buffinclude_outer_linesbool$add_background_rectangles_to_entriesentries_background_colorrinclude_background_rectanglebackground_rectangle_colorelement_to_mobject,Callable[[float | str | VMobject], VMobject]element_to_mobject_configdictarrange_in_grid_config line_configc ||_||_||_t||_t|d|_||_||_||_||_ t| |_ | |_ t| |_ | |_| |_||_||_|D]7}t|t|dkr)t%dt'jdi|||}t-t/j||_||}||t-t/j||_t|jddkr%|j|jd||j| ||_!|"|#|j r|$|j |j r|%|j dSdS)Nrz+Not all rows in table have the same length.color)&r$r&r'lenrow_dimcol_dimr)r+r,r.rr/r0r1r2r4r6r7 ValueErrorsuper__init___table_to_mob_tableritchainelements_without_labels _add_labels_organize_mob_tableelementsget_all_pointsremoveaddcenter mob_table_add_horizontal_lines_add_vertical_linesadd_background_to_entriesadd_background_rectangle)selfr"r$r&r'r)r+r,r.r/r0r1r2r4r6r7kwargsrowrM __class__s N/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/table.pyrAzTable.__init__sJ,%$,5zz 58}}   #6 4X1(23K(L(L%,H)*45O*P*P'"4)B&&<#& P PC3xx3uQx==(( !NOOO""6""",,U33 '-rx/C'D$$$Y//    +++) 45 t}Q..00 1 1Q 6 6 M q!1 2 2 2  " ""$$$   """  4 P  * *1N * O O O  , Q  ) )0O ) P P P P P Q Qreturnlistc fd|DS)aInitializes the entries of ``table`` as :class:`~.VMobject`. Parameters ---------- table A 2D array or list of lists. Content of the table has to be a valid input for the callable set in ``element_to_mobject``. Returns -------- List List of :class:`~.VMobject` from the entries of ``table``. c,g|]}fd|DS)c6g|]}j|fijSr;r2r4).0itemrRs rV z8Table._table_to_mob_table...sA   ('OO0NOO   rWr;)r^rTrRs rVr`z-Table._table_to_mob_table.. sL              rWr;)rRr"s` rVrBzTable._table_to_mob_tables0"         rWIterable[Iterable[VMobject]]rc<t}t|D];\}}t|D]&\}}||||'<|jdt |t |d|j|jfd|j|S)atArranges the :class:`~.VMobject` of ``table`` in a grid. Parameters ---------- table A 2D iterable object with :class:`~.VMobject` entries. Returns -------- :class:`~.VGroup` The :class:`~.VMobject` of the ``table`` in a :class:`~.VGroup` already arranged in a table-like grid. r)rowscolsbuffr;)r enumeraterKarrange_in_gridr<r+r)r6)rRr" help_tableirTj_s rVrGzTable._organize_mob_tablesXX && , ,FAs!# , ,1uQx{++++ ," " UU1X+t{+  )    rWrMc|j>tt|jD]}|j|g||z||<|j|jf|j'|jg|jz}|d|nSt }|g|jz}|d|n|d|j|S)a#Adds labels to an in a grid arranged :class:`~.VGroup`. Parameters ---------- mob_table An in a grid organized class:`~.VGroup`. Returns -------- :class:`~.VGroup` Returns the ``mob_table`` with added labels. Nr)r$ranger<r&r'insertr )rRrMkr& dummy_mobjects rVrFzTable._add_labels/s ? &3t//00 C C $ 23ilB ! ? &*&2"&"5!6!HJ$$Q 3333%C$@$B$B$D$DM"/4?!BJ$$Q 3333  DO444rWc|dd|jzz }|dd|jzz}t}|jr|ddd|jzz}t||dg||dgfi|j }| || ||d dd|jzz }t||dg||dgfi|j }| || |tt|jdz D]}||dzdd|| d||dzdz zz}t||dg||dgfi|j }| || |||_|S)z'Adds the horizontal lines to the table.r?r)get_leftr+ get_rightrr,get_rowsget_topr)r r7rK get_bottomrmr<rMhorizontal_lines)rR anchor_left anchor_right line_groupanchorlineros rVrNzTable._add_horizontal_linesNsdmmooa(3+<< ~~''*S4;->> XX  # ]]__Q'//11!4sT[7HHFfa(<*CGKGWD NN4 HHTNNN]]__R(3355a83;LLFfa(<*CGKGWD NN4 HHTNNNs4>**Q.//  A]]__QU+3355a83 "--//2T]]__QU5K5S5S5U5UVW5XX<Ffa(<*CGKGWD NN4 HHTNNNN * rWc>|dd|jzz}|dd|jzz }t }|jr|ddd|jzz }t||dg||dgfi|j }| || ||d dd|jzz}t||dg||dgfi|j }| || |tt|jddz D]}||dzdd|| d||dzdz zz}t||dg||dgfi|j }| || |||_|S)z$Adds the vertical lines to the tablerrrrrs)rvrwr)rxrr, get_columnsrtr+r r7rKrurmr<rMvertical_lines)rR anchor_top anchor_bottomr|r}r~ros rVrOzTable._add_vertical_linesls]]__,,..q1C$+4EE  2244Q7# :KK XX  # %%''*3355a83;LLFQ'&-)CGKGWD NN4 HHTNNN%%''+5577:S4;=NNFQ'&-)CGKGWD NN4 HHTNNNs4>!,--122  A%%''A.7799!z.Table.get_columns...s'::CQ::::::rWNrrM)r^rirRs @rVrz$Table.get_columns..sT::::4>:::;rWr)rrmr<rMrs`rVrzTable.get_columnssO0s4>!#45566  rWc2td|jDS)aReturn the rows of the table as a :class:`~.VGroup` of :class:`~.VGroup`. Returns -------- :class:`~.VGroup` :class:`~.VGroup` containing each row in a :class:`~.VGroup`. Examples -------- .. manim:: GetRowsExample :save_last_frame: class GetRowsExample(Scene): def construct(self): table = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")]) table.add(SurroundingRectangle(table.get_rows()[1])) self.add(table) c3(K|] }t|VdSr)r)r^rTs rVrz!Table.get_rows..s&?? ??????rWrrs rVrvzTable.get_rowss 0?????@@rWcolorsIterable[ParsableManimColor]c|}t||dD]\}}|||S)aSet individual colors for each column of the table. Parameters ---------- colors An iterable of colors; each color corresponds to a column. Examples -------- .. manim:: SetColumnColorsExample :save_last_frame: class SetColumnColorsExample(Scene): def construct(self): table = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")] ).set_column_colors([RED,BLUE], GREEN) self.add(table) Fstrict)rzip set_color)rRrcolumnsr:columns rVset_column_colorszTable.set_column_colorssS0""$$ ??? $ $ME6   U # # # # rWc|}t||dD]\}}|||S)aSet individual colors for each row of the table. Parameters ---------- colors An iterable of colors; each color corresponds to a row. Examples -------- .. manim:: SetRowColorsExample :save_last_frame: class SetRowColorsExample(Scene): def construct(self): table = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")] ).set_row_colors([RED,BLUE], GREEN) self.add(table) Fr)rvrr)rRrrcr:rTs rVset_row_colorszTable.set_row_colorssM0}}fd5999 ! !JE3 MM%  rWposSequence[int] | NoneVMobject | VGroupc:||jM|jF|j?t|jd|ddz z|dzdz }|j|St|jd|ddz z|dzdz }|j|S|jS)aReturn the individual entries of the table (including labels) or one specific entry if the parameter, ``pos``, is set. Parameters ---------- pos The position of a specific entry on the table. ``(1,1)`` being the top left entry of the table. Returns ------- Union[:class:`~.VMobject`, :class:`~.VGroup`] :class:`~.VGroup` containing all entries of the table (including labels) or the :class:`~.VMobject` at the given position if ``pos`` is set. Examples -------- .. manim:: GetEntriesExample :save_last_frame: class GetEntriesExample(Scene): def construct(self): table = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")]) ent = table.get_entries() for item in ent: item.set_color(random_bright_color()) table.get_entries((2,2)).rotate(PI) self.add(table) Nrrr)r$r&r'r<rMrHrRrindexs rV get_entrieszTable.get_entries2sL ?+O/'/DN1-..#a&1*=AFJ}U++DN1-..#a&1*=AFJ}U++= rWcl|,|j|ddz z|dzdz }|j|S|jS)aKReturn the individual entries of the table (without labels) or one specific entry if the parameter, ``pos``, is set. Parameters ---------- pos The position of a specific entry on the table. ``(1,1)`` being the top left entry of the table (without labels). Returns ------- Union[:class:`~.VMobject`, :class:`~.VGroup`] :class:`~.VGroup` containing all entries of the table (without labels) or the :class:`~.VMobject` at the given position if ``pos`` is set. Examples -------- .. manim:: GetEntriesWithoutLabelsExample :save_last_frame: class GetEntriesWithoutLabelsExample(Scene): def construct(self): table = Table( [["First", "Second"], ["Third","Fourth"]], row_labels=[Text("R1"), Text("R2")], col_labels=[Text("C1"), Text("C2")]) ent = table.get_entries_without_labels() colors = [BLUE, GREEN, YELLOW, RED] for k in range(len(colors)): ent[k].set_color(colors[k]) table.get_entries_without_labels((2,2)).rotate(PI) self.add(table) Nrr)r>rErs rVget_entries_without_labelsz Table.get_entries_without_labelsfsCN ?LCFQJ/#a&81 *,')'?Q?Q?Q?Q?Q?Q?QB    24><>%%%%4####4    >AAAA4:>%)2!2!2!2!2!l%)+0+0+0+0+0Z((((8((((8!!!!FEJ 77777v$$/%%%%%R$$/&&&&&TCIDIFLDJ H@H@H@H@H@TrWrc(eZdZdZefdfd ZxZS)raA specialized :class:`~.Table` mobject for use with LaTeX. Examples -------- .. manim:: MathTableExample :save_last_frame: class MathTableExample(Scene): def construct(self): t0 = MathTable( [["+", 0, 5, 10], [0, 0, 5, 10], [2, 2, 7, 12], [4, 4, 9, 14]], include_outer_lines=True) self.add(t0) r"Iterable[Iterable[float | str]]r2!Callable[[float | str], VMobject]c @tj|fd|i|dS)a* Special case of :class:`~.Table` with `element_to_mobject` set to :class:`~.MathTex`. Every entry in `table` is set in a Latex `align` environment. Parameters ---------- table A 2d array or list of lists. Content of the table have to be valid input for :class:`~.MathTex`. element_to_mobject The :class:`~.Mobject` class applied to the table entries. Set as :class:`~.MathTex`. kwargs Additional arguments to be passed to :class:`~.Table`. r2Nr@rArRr"r2rSrUs rVrAzMathTable.__init__sD(    1      rWr"rr2r)rrrrrrArrs@rVrrsS,AH           rWrc*eZdZdZdfdfd ZxZS) ra[A specialized :class:`~.Table` mobject for use with :class:`~.Mobject`. Examples -------- .. manim:: MobjectTableExample :save_last_frame: class MobjectTableExample(Scene): def construct(self): cross = VGroup( Line(UP + LEFT, DOWN + RIGHT), Line(UP + RIGHT, DOWN + LEFT), ) a = Circle().set_color(RED).scale(0.5) b = cross.set_color(BLUE).scale(0.5) t0 = MobjectTable( [[a.copy(),b.copy(),a.copy()], [b.copy(),a.copy(),a.copy()], [a.copy(),b.copy(),b.copy()]] ) line = Line( t0.get_corner(DL), t0.get_corner(UR) ).set_color(RED) self.add(t0, line) c|Srr;)ms rVzMobjectTable.sqrWr"rar2Callable[[VMobject], VMobject]c @tj|fd|i|dS)a2 Special case of :class:`~.Table` with ``element_to_mobject`` set to an identity function. Here, every item in ``table`` must already be of type :class:`~.Mobject`. Parameters ---------- table A 2D array or list of lists. Content of the table must be of type :class:`~.Mobject`. element_to_mobject The :class:`~.Mobject` class applied to the table entries. Set as ``lambda m : m`` to return itself. kwargs Additional arguments to be passed to :class:`~.Table`. r2Nrrs rVrAzMobjectTable.__init__s0& PP3EPPPPPPrW)r"rar2r)rrrrrArrs@rVrrs_<>I[QQQQQQQQQQQrWrc(eZdZdZefdfd ZxZS)raiA specialized :class:`~.Table` mobject for use with :class:`~.Integer`. Examples -------- .. manim:: IntegerTableExample :save_last_frame: class IntegerTableExample(Scene): def construct(self): t0 = IntegerTable( [[0,30,45,60,90], [90,60,45,30,0]], col_labels=[ MathTex(r"\frac{ \sqrt{0} }{2}"), MathTex(r"\frac{ \sqrt{1} }{2}"), MathTex(r"\frac{ \sqrt{2} }{2}"), MathTex(r"\frac{ \sqrt{3} }{2}"), MathTex(r"\frac{ \sqrt{4} }{2}")], row_labels=[MathTex(r"\sin"), MathTex(r"\cos")], h_buff=1, element_to_mobject_config={"unit": r"^{\circ}"}) self.add(t0) r"rr2rc @tj|fd|i|dS)a! Special case of :class:`~.Table` with `element_to_mobject` set to :class:`~.Integer`. Will round if there are decimal entries in the table. Parameters ---------- table A 2d array or list of lists. Content of the table has to be valid input for :class:`~.Integer`. element_to_mobject The :class:`~.Mobject` class applied to the table entries. Set as :class:`~.Integer`. kwargs Additional arguments to be passed to :class:`~.Table`. r2Nrrs rVrAzIntegerTable.__init__Js0( PP3EPPPPPPrWr)rrrrrrArrs@rVrr0s^8AHQQQQQQQQQQQrWrc.eZdZdZeddifd fd ZxZS) ratA specialized :class:`~.Table` mobject for use with :class:`~.DecimalNumber` to display decimal entries. Examples -------- .. manim:: DecimalTableExample :save_last_frame: class DecimalTableExample(Scene): def construct(self): x_vals = [-2,-1,0,1,2] y_vals = np.exp(x_vals) t0 = DecimalTable( [x_vals, y_vals], row_labels=[MathTex("x"), MathTex("f(x)=e^{x}")], h_buff=1, element_to_mobject_config={"num_decimal_places": 2}) self.add(t0) num_decimal_placesrr"rr2rr4r5c Btj|f||d|dS)a Special case of :class:`~.Table` with ``element_to_mobject`` set to :class:`~.DecimalNumber`. By default, ``num_decimal_places`` is set to 1. Will round/truncate the decimal places based on the provided ``element_to_mobject_config``. Parameters ---------- table A 2D array, or a list of lists. Content of the table must be valid input for :class:`~.DecimalNumber`. element_to_mobject The :class:`~.Mobject` class applied to the table entries. Set as :class:`~.DecimalNumber`. element_to_mobject_config Element to mobject config, here set as {"num_decimal_places": 1}. kwargs Additional arguments to be passed to :class:`~.Table`. r]Nr)rRr"r2r4rSrUs rVrAzDecimalTable.__init__vsG0   1&?        rW)r"rr2rr4r5)rrrrrrArrs@rVrras[.AN+?*C            rWr)/r __future__r__all__ itertoolsrCcollections.abcr r r manim.mobject.geometry.liner manim.mobject.geometry.polygramr %manim.mobject.geometry.shape_matchersrmanim.mobject.text.numbersrrmanim.mobject.text.tex_mobjectrmanim.mobject.text.text_mobjectranimation.animationranimation.compositionranimation.creationrranimation.fadingr mobject.types.vectorized_mobjectrr utils.colorrrrrutilsr rrrrrr;rWrVrsg55n#"""""   8888888888,,,,,,333333EEEEEE========222222555555++++++222222........%%%%%%????????LLLLLLLLLLLL//////v v v v v Fv v v r, , , , , , , , ^/Q/Q/Q/Q/Q5/Q/Q/Qd.Q.Q.Q.Q.Q5.Q.Q.Qb2 2 2 2 2 52 2 2 2 2 rW