o o h@s@ddlmZddlmmmZddgiZGdddZdS)) import_moduleN)PlotGrid matplotlibc@s`eZdZdZdddddZddZed d Zed d Zd dZ ddZ ddZ ddZ dS)ra This class helps to plot subplots from already created SymPy plots in a single figure. Examples ======== .. plot:: :context: close-figs :format: doctest :include-source: True >>> from sympy import symbols >>> from sympy.plotting import plot, plot3d, PlotGrid >>> x, y = symbols('x, y') >>> p1 = plot(x, x**2, x**3, (x, -5, 5)) >>> p2 = plot((x**2, (x, -6, 6)), (x, (x, -5, 5))) >>> p3 = plot(x**3, (x, -5, 5)) >>> p4 = plot3d(x*y, (x, -5, 5), (y, -5, 5)) Plotting vertically in a single line: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(2, 1, p1, p2) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x for x over (-5.0, 5.0) [1]: cartesian line: x**2 for x over (-5.0, 5.0) [2]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plotting horizontally in a single line: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(1, 3, p2, p3, p4) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[2]:Plot object containing: [0]: cartesian surface: x*y for x over (-5.0, 5.0) and y over (-5.0, 5.0) Plotting in a grid form: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(2, 2, p1, p2, p3, p4) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x for x over (-5.0, 5.0) [1]: cartesian line: x**2 for x over (-5.0, 5.0) [2]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plot[2]:Plot object containing: [0]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[3]:Plot object containing: [0]: cartesian surface: x*y for x over (-5.0, 5.0) and y over (-5.0, 5.0) TN)showsizecOsvtddgdidtfd|_||_||_g|_d|_||_|D] }|j|jq||_ |r7|jr9| dSdSdS)a Parameters ========== nrows : The number of rows that should be in the grid of the required subplot. ncolumns : The number of columns that should be in the grid of the required subplot. nrows and ncolumns together define the required grid. Arguments ========= A list of predefined plot objects entered in a row-wise sequence i.e. plot objects which are to be in the top row of the required grid are written first, then the second row objects and so on Keyword arguments ================= show : Boolean The default value is set to ``True``. Set show to ``False`` and the function will not display the subplot. The returned instance of the ``PlotGrid`` class can then be used to save or display the plot by calling the ``save()`` and ``show()`` methods respectively. size : (float, float), optional A tuple in the form (width, height) in inches to specify the size of the overall figure. The default value is set to ``None``, meaning the size will be set by the default backend. rfromlist)pyplotcm collectionsz1.1.0) import_kwargsmin_module_versioncatchN) r RuntimeErrorrnrowsncolumns_series_figargsappendrr)selfrrrrrkwargsargrk/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/plotting/plotgrid.py__init__\s#   zPlotGrid.__init__c Cs|jj|j|j}i}d}t|jD] }t|jD]}|t|jkr.|j|||||f<|d7}qq|js9ind|ji}|jj j di||_ | D],\}}t|j dkrb|j djrbddini}|j j|fi|} |j |_| |_|qMdS)Nrfigsize projection3dr)rgridspecGridSpecrrrangelenrrrfigureritemsris_3D add_subplot _plotgrid_fig _plotgrid_axprocess_series) rgsmappingcijkwspecpcur_axrrr_create_figures,    zPlotGrid._create_figurecCs|js||jSN)rr3rrrrfigsz PlotGrid.figcCs|Sr4rr5rrr_backendszPlotGrid._backendcCs|jj|jdSr4)rrcloser6r5rrrr8szPlotGrid.closecCs,tjr|j|jjdS|dSr4) base_backend_showr6 tight_layoutrrrr8r5rrrrs  z PlotGrid.showcCs|j|dSr4)r6savefig)rpathrrrsavesz PlotGrid.savecCs"ddt|jD}dd|S)NcSs g|] \}}d|t|qS)z Plot[%d]:)str).0r-plotrrr sz$PlotGrid.__str__..zPlotGrid object containing:  ) enumeraterjoin)r plot_strsrrr__str__szPlotGrid.__str__) __name__ __module__ __qualname____doc__rr3propertyr6r7r8rr>rGrrrrrsK1   r)sympy.externalr$sympy.plotting.backends.base_backendplottingbackendsr9__doctest_requires__rrrrrs