l j[J*dZddlmZddgZddlmZmZmZddlm Z ddl Z ddl m Z mZdd lTdd lmZdd lmZdd lmZdd lmZddlmZmZddlmZmZddlmZddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*dZ+GddeZ,GddeZ-dS)zEMobjects representing objects from probability theory and statistics.) annotations SampleSpaceBarChart)IterableMutableSequenceSequence)AnyN)configlogger)*) Rectangle)Axes)OpenGLVMobject)Brace)MathTexTex)VGroupVMobject)Vector3D)BLUE_E DARK_GREYGREEN_E LIGHT_GREYMAROON_BYELLOWParsableManimColorcolor_gradient)tuplifyg-C6?ceZdZdZddeddedfd7fd Zdefd8dZd9dZ d:dZ d;d#Z e e gefdd0Zefd?d1Zd@d2Zd@d3ZdAd4ZdBd6ZxZS)CraA mobject representing a twodimensional rectangular sampling space. Examples -------- .. manim:: ExampleSampleSpace :save_last_frame: class ExampleSampleSpace(Scene): def construct(self): poly1 = SampleSpace(stroke_width=15, fill_opacity=1) poly2 = SampleSpace(width=5, height=3, stroke_width=5, fill_opacity=0.5) poly3 = SampleSpace(width=2, height=2, stroke_width=5, fill_opacity=0.1) poly3.divide_vertically(p_list=np.array([0.37, 0.13, 0.5]), colors=[BLACK, WHITE, GRAY], vect=RIGHT) poly_group = VGroup(poly1, poly2, poly3).arrange() self.add(poly_group) ?heightfloatwidth fill_colorr fill_opacity stroke_width stroke_colordefault_label_scale_valcdt||||||||_dS)N)r#r%r&r'r(r))super__init__r*) selfr#r%r&r'r(r)r* __class__s ]/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/graphing/probability.pyr-zSampleSpace.__init__8sH !%%%    (?$$$z Sample spacetitlestrbuffreturnNonect|}|j|jkr |j|_||t|||_||dS)Nr4)rr%next_toUPr2add)r.r2r4 title_mobs r0 add_titlezSampleSpace.add_titleLsaJJ ?TZ ' '"jIO$...  r1labelc||_dSN)r>)r.r>s r0 add_labelzSampleSpace.add_labelWs  r1p_listfloat | Iterable[float] list[float]ct|}t|}dt|z }t|tkr|||S)Ng?)rlistsumabsEPSILONappend)r.rBp_list_tuplified new_p_list remainders r0complete_p_listzSampleSpace.complete_p_listZsW)0*++ #j//) y>>G # #   i ( ( (r1dimintcolorsSequence[ParsableManimColor]vectrrc||}t|t|}|| }t }t ||dD]\} } t } | | d| |d| | || || | |}| | |S)NTstrictr!)stretch) rNrlenget_edge_centerrziprset_fillreplacerWmove_tor;) r.rBrOrQrSp_list_completecolors_in_gradient last_pointpartsfactorcolorparts r0get_division_along_dimensionz(SampleSpace.get_division_along_dimensionbs..v66+FC4H4HII))4%00  2DTRRR  MFE==D MM% # # # LLtL , , , LL % % % LLdU + + +--d33J IIdOOOO r1c2||d||S)Nr!rer.rBrQrSs r0get_horizontal_divisionz#SampleSpace.get_horizontal_divisionx 00FDIIIr1c2||d||S)Nrrgrhs r0get_vertical_divisionz!SampleSpace.get_vertical_divisionrjr1argsr kwargsc^|j|i||_||jdSr@)rihorizontal_partsr;r.rmrns r0divide_horizontallyzSampleSpace.divide_horizontallys7 < rdbrace label_mobs r0!get_subdivision_braces_and_labelsz-SampleSpace.get_subdivision_braces_and_labelss'XX vuU;;; & &KE4$ TRRRE%(N!;<< >! #ENN  <===   eY 5 5 5 JJu   i22 2 22 NN9 % % % % !  oo''"   elEL111r1c Tt|dsJ|j}|j|||fi|S)Nrp)hasattrrpr)r.rvrxrnras r0get_side_braces_and_labelsz&SampleSpace.get_side_braces_and_labelssN t/00000%5t5 69  (.   r1c ^t|dsJ|j}|j||tfi|SNrt)rrtrr:r.rvrnras r0get_top_braces_and_labelsz%SampleSpace.get_top_braces_and_labelssBt-.....#5t5eVRRR6RRRr1c ^t|dsJ|j}|j||tfi|Sr)rrtrDOWNrs r0get_bottom_braces_and_labelsz(SampleSpace.get_bottom_braces_and_labelssBt-.....#5t5eVTTTVTTTr1cdD][}t||st||}dD]5}t||r#|t||6\dS)N)rprt)r}rv)rgetattrr;)r.attrrasubattrs r0add_braces_and_labelsz!SampleSpace.add_braces_and_labelss8 6 6D4&& D$''E- 6 65'**6HHWUG44555 6  6 6r1indexct|dr|j|}|St|dr|j|}|S||S)Nrprt)rrprtsplit)r.rvals r0 __getitem__zSampleSpace.__getitem__s_ 4+ , , #4U;CJ T+ , , %e,CJzz||E""r1)r#r$r%r$r&rr'r$r(r$r)rr*r$)r2r3r4r$r5r6)r>r3r5r6)rBrCr5rD) rBrCrOrPrQrRrSrr5r)rBrCrQrRrSrr5r)rmr rnr r5r6) rarrvrwrxrr4r$ryrPr5r)rvrwrxrrnr r5r)rvrwrnr r5rr5r6)rrPr5r)__name__ __module__ __qualname____doc__rrr-MED_SMALL_BUFFr=rArNrerrrrirrRIGHTrlrrru SMALL_BUFFrLEFTrrrrr __classcell__r/s@r0rr%s()2!+5)*???????**     218/@ JJJJJ19&/A JJJJJ((((&&&&! 22222D#      SSSSUUUU6666########r1cxeZdZdZddddgddddfd1fd Zd2dZd2dZd3d!Zd2d"Zdd#e e fd4d+Z d5d6d0Z xZ S)7raaCreates a bar chart. Inherits from :class:`~.Axes`, so it shares its methods and attributes. Each axis inherits from :class:`~.NumberLine`, so pass in ``x_axis_config``/``y_axis_config`` to control their attributes. Parameters ---------- values A sequence of values that determines the height of each bar. Accepts negative values. bar_names A sequence of names for each bar. Does not have to match the length of ``values``. y_range The y_axis range of values. If ``None``, the range will be calculated based on the min/max of ``values`` and the step will be calculated based on ``y_length``. x_length The length of the x-axis. If ``None``, it is automatically calculated based on the number of values and the width of the screen. y_length The length of the y-axis. bar_colors The color for the bars. Accepts a sequence of colors (can contain just one item). If the length of``bar_colors`` does not match that of ``values``, intermediate colors will be automatically determined. bar_width The length of a bar. Must be between 0 and 1. bar_fill_opacity The fill opacity of the bars. bar_stroke_width The stroke width of the bars. Examples -------- .. manim:: BarChartExample :save_last_frame: class BarChartExample(Scene): def construct(self): chart = BarChart( values=[-5, 40, -10, 20, -3], bar_names=["one", "two", "three", "four", "five"], y_range=[-20, 50, 10], y_length=6, x_length=10, x_axis_config={"font_size": 36}, ) c_bar_lbls = chart.get_bar_labels(font_size=48) self.add(chart, c_bar_lbls) N)z#003f5cz#58508dz#bc5090z#ff6361z#ffa600g333333?gffffff?r valuesMutableSequence[float] bar_namesSequence[str] | Noney_rangeSequence[float] | Nonex_length float | Noney_length bar_colors Iterable[str] bar_widthr$bar_fill_opacitybar_stroke_widthrnr c vt|tr#tjdt |}||nt jdz }||_||_||_ ||_ ||_ | |_ dt|jdg} |itdt|jtdt|jt!t|j|z dg}n %  # # % % % !!!!!r1r5r6c.|jj|jdS)zInitialize the colors of the bars of the chart. Sets the color of ``self.bars`` via ``self.bar_colors``. Primarily used when the bars are initialized with ``self._add_bars`` or updated via ``self.change_bar_values``. N)rset_color_by_gradientr)r.s r0_update_colorszBarChart._update_colorsZs ( '9999r1ct|jtsJtjdt |jd}t }tt||jdD]\}\}}|j |dkrtnt}|j |}|j j|_||j |||j j||||j _|j |dS)zEssentially :meth`:~.NumberLine.add_labels`, but differs in that the direction of the label with respect to the x_axis changes to UP or DOWN depending on the value. UP for negative values and DOWN for positive values. r"r!TrUr)rxr4N)r{rrFnparangerXr enumeraterZrr:rx_axisrrr9number_to_pointline_to_number_buffr;rv)r. val_rangervivaluebar_namerxbar_name_labels r0rzBarChart._add_x_axis_labelsds3$.$/////I T^$$a  $-  4>$ 7 7 7% %  ' ' A x #k!nq00dI&*k&C&CH&M&MN'+{'>++A.!Q1BB..    aZZbbd DHHZ#-q113Q ??? r1ct|jD]6\}}|||}|j|7|||jdS)N)rr)rrrrr;r add_to_back)r.rrtmp_bars r0rzBarChart._add_barss|!$+.. # #HAu&&!5&AAG IMM' " " " "  #####r1rrcParsableManimColor | Nonerr4r type[MathTex]rct}t|j|jdD]\}}|t |}|(||n||||_|dkrtnt} | || || ||S)aAnnotates each bar with its corresponding value. Use ``self.bar_labels`` to access the labels after creation. Parameters ---------- color The color of each label. By default ``None`` and is based on the parent's bar color. font_size The font size of each label. buff The distance from each label to its bar. By default 0.4. label_constructor The Mobject class to construct the labels, by default :class:`~.Tex`. Examples -------- .. manim:: GetBarLabelsExample :save_last_frame: class GetBarLabelsExample(Scene): def construct(self): chart = BarChart(values=[10, 9, 8, 7, 6, 5, 4, 3, 2, 1], y_range=[0, 10, 1]) c_bar_lbls = chart.get_bar_labels( color=WHITE, label_constructor=MathTex, font_size=36 ) self.add(chart, c_bar_lbls) FrUNrr8) rrZrrr3 set_colorget_fill_colorrr:rr9r;) r.rcrr4rrrrbar_lblrs r0get_bar_labelszBarChart.get_bar_labelssHXX diUCCC $ $JC00U<BD2D2D2D2D2D2D2D2D2r1).r __future__r__all__collections.abcrrrtypingr numpyrmanimr r manim.constantsmanim.mobject.geometry.polygramr )manim.mobject.graphing.coordinate_systemsr.manim.mobject.opengl.opengl_vectorized_mobjectrmanim.mobject.svg.bracermanim.mobject.text.tex_mobjectrr&manim.mobject.types.vectorized_mobjectrr manim.typingrmanim.utils.colorrrrrrrrrmanim.utils.iterablesrrIrrrr1r0r sKK"""""" * %@????????? 555555::::::IIIIII))))))77777777CCCCCCCC!!!!!!                    *))))) t#t#t#t#t#)t#t#t#nL2L2L2L2L2tL2L2L2L2L2r1