adopt-plot 0.0.1__py3-none-any.whl

adopt_plot/__init__.py

@@ -0,0 +1,8 @@
+"""Adopt Plot package"""
+# SPDX-License-Identifier: BSD-3-Clause
+# Copyright (c) 2026 VartRusData. All rights reserved.
+class TooManyVariablesError(ValueError):
+    """Исключение, возникающее при попытке построить график для более чем двух переменных."""
+    pass
+from .plot import AdoptPlot
+__all__ = ["AdoptPlot", "TooManyVariablesError"]

adopt_plot/plot.py

@@ -0,0 +1,401 @@
+
+# SPDX-License-Identifier: BSD-3-Clause
+# Copyright (c) 2026 VartRusData. All rights reserved.
+from sympy import *
+import numpy as np
+import matplotlib.pyplot as plt
+import logging
+import re
+import sympy
+
+
+from . import TooManyVariablesError
+
+def get_all_sympy_function_names():
+    """Собирает имена всех встроенных функций и констант SymPy."""
+    names = set(sympy.__all__)
+    #print(names)
+    return names
+
+
+def insert_multiplication_signs(expr: str, extra_functions=None) -> str:
+    """
+    Вставляет знаки умножения с учётом неявного умножения.
+    Поддерживает латиницу и кириллицу в именах переменных.
+    Защищает все известные SymPy-функции и константы от разбиения.
+    """
+    func_set = get_all_sympy_function_names()  # теперь там и функции, и константы
+    if extra_functions:
+        func_set.update(extra_functions)
+
+    sorted_funcs = sorted(func_set, key=len, reverse=True)
+    func_pattern = '|'.join(re.escape(f) for f in sorted_funcs)
+
+    LETTER = r'[A-Za-zА-Яа-яёЁ_]'
+    LETTER_DIGIT = r'[\dA-Za-zА-Яа-яёЁ_]'
+
+    # === ШАГ 1: Защита известных имён (функций и констант) ===
+    protected = {}
+
+    def protect_known(match):
+        name = match.group(0)
+        if name not in func_set:
+            return name
+        placeholder = f'\ue000{len(protected)}\ue001'
+        protected[placeholder] = name
+        return placeholder
+
+    #print(expr)
+    expr = re.sub(r'[A-Za-z_]\w*', protect_known, expr)
+    #print(expr)
+    # === ШАГ 2: Правила вставки умножения ===
+    expr = re.sub(rf'(\d+)(\ue000\d+\ue001)(?=\()', r'\1*\2', expr)
+    # Явно вставляем * между цифрой/буквой и известным именем перед '('
+    expr = re.sub(rf'(\d)({func_pattern})(?=\()', r'\1*\2', expr)
+    expr = re.sub(rf'({LETTER})({func_pattern})(?=\()', r'\1*\2', expr)
+
+    # Основные правила с кириллицей
+    expr = re.sub(rf'(\d)({LETTER})', r'\1*\2', expr)  # 2x, 2я
+    # РАЗРЫВАЕМ ВСЕ ЦЕПОЧКИ БУКВ (кроме защищённых имён)
+    expr = re.sub(rf'({LETTER})(?={LETTER})', r'\1*', expr)  # a*b*c, а*я
+    expr = re.sub(rf'({LETTER_DIGIT})(\()', r'\1*\2', expr)  # a(, 3(, я(
+    expr = re.sub(rf'(\))({LETTER_DIGIT}\()', r'\1*\2', expr)  # )a, )(, )я
+
+    # === ШАГ 3: Возвращаем защищённые имена на место ===
+    for placeholder, name in protected.items():
+        expr = expr.replace(placeholder, name)
+
+    return expr
+
+
+
+
+
+
+
+from typing import Literal, Optional, Tuple
+
+class AdoptPlot():
+    def __init__(self,
+               expr: str,
+               lib: Literal['contour', 'implicit', None] = None,
+               xlims: Tuple[float, float] = (-20, 20),
+               ylims: Tuple[float, float] = (-20, 20),
+               n: int = 10000,
+
+               depth: int = 9,
+               linewidth: float = 2.0,
+               limit: float | Tuple[float, float, float, float] = 100,
+               color: str = 'blue',
+               legend: bool = True,
+               points: bool = True,
+               grid: bool = True,
+               interact: bool = True,
+               show: bool = True,
+               text_legend: Optional[str] = None
+               ):
+        """
+        Plots an implicit function with adaptive engine selection.
+
+        Parameters:
+            expr: String containing the equation (e.g., "y = 1/x").
+            lib: Forced plotting engine ('contour', 'implicit', or None for auto-selection).
+            xlims :Initial visible range for the plot.
+            ylims: Initial visible range for the plot.
+            n: Grid density for the contour engine.
+            depth: Adaptive refinement depth for plot_implicit.
+            linewidth: Thickness of the plot line.
+            limit: Defines the computational domain for the plot. Accepts a single number (for a square area) or a tuple of 4 floats (x_min, x_max, y_min, y_max) for a rectangular area.
+            color: Line color.
+            legend: Whether to display the legend.
+            points: Whether to display axis intersection points.
+            grid: Whether to display the grid.
+            interact: Allow zooming/panning (does not fix limits rigidly).
+            show: If False, returns a PlotHandle object instead of displaying the plot.
+            text_legend: Additional text for the legend (supports "\\n" and a list of strings).
+    """
+        try:
+            self.lib = lib
+            self.xlims = xlims
+            self.ylims = ylims
+            self.n = n
+            self.depth = depth
+            self.linewidth = linewidth
+            self.color = color
+            self.legend = legend
+            self.points = points
+            self.grid = grid
+            self.interact = interact
+            self.show = show
+            self.limit = limit
+            self.text_legend = text_legend
+            # Получаем уравнения из поля ввода
+            equations_str = expr
+            logging.info(f"Полученная строка уравнений: {equations_str}")
+            if equations_str == "":
+                raise ValueError("Empty string is not allowed.")
+            # Проверяем наличие запятых в строке
+            equation = equations_str
+            # Разбиение строки на отдельные уравнения
+
+
+
+            # Преобразование уравнений в объекты Sympy
+            expressions = []
+            used_variables = set()  # Множество переменных, используемых в уравнениях
+
+            logging.info(f"Преобразование уравнения: {equation}")
+
+            equation = insert_multiplication_signs(equation)
+
+            lhs, rhs = equation.split('=')
+            logging.info(str(lhs))
+            logging.info(str(rhs))
+            expressions.append(Eq(sympify(lhs), sympify(rhs)))
+            logging.info(f"Добавлено уравнение: {expressions[-1]}")
+
+            # Определяем переменные, участвующие в текущем уравнении
+            used_variables.update(list(expressions[-1].free_symbols))
+
+            logging.info(f"Переменные, задействованные в уравнениях: {used_variables}")
+
+            # Проверка на недоопределённость системы
+            # Строим график для первого уравнения
+            self.eq = expressions[0]
+            self.vars_list = list(self.eq.free_symbols)
+
+            if len(self.vars_list) == 2:
+                self.var1, self.var2 = self.vars_list[0], self.vars_list[1]
+                self.plot_equation()
+                return
+            else:
+                raise TooManyVariablesError(
+                f"adopt_plot поддерживает только 2D-графики. Получено {len(self.vars_list)} переменных: {self.vars_list}. "
+                f"Если вам нужен 3D-график, подождите следующей версии!"
+            )
+
+
+
+
+
+        # Обновляем историю
+
+        except Exception as e:
+            raise Exception(e)
+
+    def plot_equation(self):
+        try:
+            def has_odz(expr, var):
+                """
+                Проверяет, есть ли у выражения ограничения на область допустимых значений (ОДЗ).
+                Если есть - лучше рисовать через plot_implicit.
+                """
+                # 1. Проверяем наличие нелинейных функций с ограничениями (log, sqrt, asin, acos)
+                # В SymPy log и ln — одно и то же.
+                if expr.has(log) or expr.has(sqrt) or expr.has(asin) or expr.has(acos):
+                    return True
+
+                # 2. Проверяем, есть ли переменная в знаменателе (деление на ноль)
+                numer, denom = expr.as_numer_denom()
+                if var in denom.free_symbols:
+                    return True
+
+                return False
+
+
+            expr = self.eq.lhs - self.eq.rhs
+            vars_list = sorted(self.eq.free_symbols, key=lambda s: str(s))
+            var1, var2 = vars_list[0], vars_list[1]
+            # === 1. Безопасный поиск пересечений (ловит ошибки abs) ===
+            x_intercepts = []
+            y_intercepts = []
+            try:
+                x_sol = solve(expr.subs(var2, 0), var1)
+                x_intercepts = [float(s) for s in x_sol if s.is_real]
+            except Exception:
+                pass  # Если sympy не может решить (abs), пропускаем
+
+            try:
+                y_sol = solve(expr.subs(var1, 0), var2)
+                y_intercepts = [float(s) for s in y_sol if s.is_real]
+            except Exception:
+                pass
+
+
+            def zoom_key_handler(event):
+                # Проверяем, что нажатие произошло на оси и это сочетание с Ctrl
+                if event.inaxes != self.ax:
+                    return
+
+                # Коэффициент масштабирования (1.2 = 120%, 0.8 = 80%)
+                scale_factor = 1.2
+
+                # Получаем текущие границы
+                cur_xlim = self.ax.get_xlim()
+                cur_ylim = self.ax.get_ylim()
+
+                # Вычисляем центр текущего обзора
+                x_center = (cur_xlim[0] + cur_xlim[1]) / 2
+                y_center = (cur_ylim[0] + cur_ylim[1]) / 2
+
+                # Длина текущего диапазона
+                cur_xrange = cur_xlim[1] - cur_xlim[0]
+                cur_yrange = cur_ylim[1] - cur_ylim[0]
+                # print(event.key)
+                # Если нажато Ctrl + (или Ctrl + =) — приближаем
+                if event.key == 'ctrl+plus' or event.key == 'ctrl+equal' or event.key == 'ctrl+=' or event.key == 'ctrl++':
+                    new_xrange = cur_xrange / scale_factor
+                    new_yrange = cur_yrange / scale_factor
+                # Если нажато Ctrl - — отдаляем
+                elif event.key == 'ctrl+minus' or event.key == 'ctrl+-':
+                    new_xrange = cur_xrange * scale_factor
+                    new_yrange = cur_yrange * scale_factor
+                else:
+                    return  # Если нажата другая клавиша, ничего не делаем
+
+                # Устанавливаем новые границы, центрируя обзор
+                self.ax.set_xlim([x_center - new_xrange / 2, x_center + new_xrange / 2])
+                self.ax.set_ylim([y_center - new_yrange / 2, y_center + new_yrange / 2])
+
+                # Принудительно перерисовываем график
+                event.canvas.draw_idle()
+            # === 2. Увеличиваем разрешение для гладкости (было 500, стало 1000) ===
+            if self.lib == 'implicit' or (has_odz and self.lib is None):
+                from sympy.plotting import plot_implicit
+                if isinstance(self.limit, (int, float)):
+                    self.p = plot_implicit(self.eq, (var1, -self.limit, self.limit), (var2, -self.limit, self.limit), show=False, n=self.n, depth=self.depth)
+                elif isinstance(self.limit, tuple):
+                    if len(self.limit) == 4:
+                        self.p = plot_implicit(self.eq, (var1, *self.limit[:2]), (var2, *self.limit[2:]), show=False, n=self.n, depth=self.depth)
+                    else:
+                        raise ValueError(f"Length mismatch (expected 4), got {len(self.limit)}.")
+                else:
+                    raise TypeError(f"Type mismatch.\nExpected float | tuple[float, float, float, float], got {type(self.limit)}.")
+                self.p.process_series()  # Отрисовываем линию
+
+                self.ax = self.p.ax
+                logging.info(f"{self.ax}")
+                self.fig = self.p.fig
+                if self.interact:
+                    self.fig.canvas.mpl_connect('key_press_event', zoom_key_handler)
+                x_coords_str = ", ".join([f"({x:.2f}, 0)" for x in x_intercepts]) if x_intercepts else "Нет"
+                y_coords_str = ", ".join([f"(0, {y:.2f})" for y in y_intercepts]) if y_intercepts else "Нет"
+
+                # Три пустые линии для легенды (они не видны на графике, но создают текст в легенде)
+                self.ax.plot([], [], ' ', label=f"${latex(self.eq)}$")  # Красивое уравнение через LaTeX
+                self.ax.plot([], [], ' ', label=f"Пересечения с X: {x_coords_str}")  # Текст про X
+                self.ax.plot([], [], ' ', label=f"Пересечения с Y: {y_coords_str}")  # Текст про Y
+
+
+                patches = self.ax.patches
+                logging.info(f"{patches}")
+                # === УНИВЕРСАЛЬНОЕ ИЗМЕНЕНИЕ ТОЛЬКО ГРАФИКА (БЕЗ РАМОК) ===
+
+                # 1. Работаем с контурами и коллекциями (contour/plot_implicit)
+                for collection in self.ax.collections:
+                    collection.set_linewidth(self.linewidth)
+                    collection.set_edgecolor(self.color)
+
+                # 2. Работаем с патчами (заливки в implicit), но пропускаем оси (Spines)
+                from matplotlib.spines import Spine
+                for patch in self.ax.patches:
+                    # Если это рамка — пропускаем
+                    if isinstance(patch, Spine):
+                        continue
+                    patch.set_linewidth(self.linewidth)
+                    patch.set_edgecolor(self.color)
+
+                # 3. Работаем с обычными линиями (например, ax.plot)
+                for line in self.ax.lines:
+                    line.set_linewidth(self.linewidth)
+                    line.set_color(self.color)
+
+
+
+                # Дальше твой код с точками и оформлением...
+                try:
+
+                    try:
+
+                        for x in x_intercepts: self.ax.scatter(x, 0, s=60, color='red', marker='o', zorder=5)
+                    except:
+                        pass
+                    try:
+
+                        for y in y_intercepts: self.ax.scatter(0, y, s=60, color='blue', marker='o', zorder=5)
+                    except:
+                        pass
+
+                    self.ax.axhline(0, color='black', linewidth=1)
+                    self.ax.axvline(0, color='black', linewidth=1)
+                    self.ax.set_xlim(self.xlims)
+                    self.ax.set_ylim(self.ylims)
+                    self.ax.set_xlabel(str(var1))
+                    self.ax.set_ylabel(str(var2))
+                    if self.text_legend:
+                        self.ax.plot([], [], ' ', label=f"{self.text_legend}")
+                    if self.legend:
+                        self.ax.legend()
+                    self.ax.grid(True, linestyle='--', alpha=0.7)
+                except Exception as e:
+                    logging.warning(f"Ошибка при добавлении оформления на p.ax: {e}")
+
+                self.p.fig.tight_layout()
+                if self.show:
+                    self.p.plt.show()
+                return
+            else:
+                if isinstance(self.limit, (float, int)):
+
+                    x_vals = np.linspace(-self.limit, self.limit, self.n)
+                    y_vals = np.linspace(-self.limit, self.limit, self.n)
+                elif isinstance(self.limit, tuple):
+                    if len(self.limit) == 4:
+                        x_vals = np.linspace(self.limit[0], self.limit[1], self.n)
+                        y_vals = np.linspace(self.limit[2], self.limit[3], self.n)
+                    else:
+                        raise ValueError(f"Length mismatch (expected 4), got {len(self.limit)}.")
+                else:
+                    raise TypeError(f"Type mismatch.\nExpected float | tuple[float, float, float, float], got {type(self.limit)}.")
+                y_vals[~np.isfinite(y_vals)] = np.nan
+                X, Y = np.meshgrid(x_vals, y_vals)
+                f = lambdify((var1, var2), expr, modules='numpy')
+                # print(x_vals, y_vals, X, Y, f, sep='\n<------------------------------->\n')
+                Z = f(X, Y)
+                # print(Z)
+                self.fig, self.ax = plt.subplots()
+                x_coords_str = ", ".join([f"({x:.2f}, 0)" for x in x_intercepts]) if x_intercepts else "Нет"
+                y_coords_str = ", ".join([f"(0, {y:.2f})" for y in y_intercepts]) if y_intercepts else "Нет"
+                self.ax.contour(X, Y, Z, levels=[0], colors='blue', linewidths=self.linewidth)
+
+                self.ax.plot([], [], ' ', label=f"Уравнение: ${latex(self.eq)}$")
+
+                # Оси X и Y
+                self.ax.axhline(0, color='black', linewidth=1)
+                self.ax.axvline(0, color='black', linewidth=1)
+
+                # Точки пересечения (теперь они не вызовут ошибку, если не найдутся)
+                self.ax.scatter([x for x in x_intercepts], [0] * len(x_intercepts), s=60, color='red', marker='o', zorder=5,
+                           label=f"Пересечения с X: {x_coords_str}")
+
+                self.ax.scatter([0] * len(y_intercepts), [y for y in y_intercepts], s=60, color='blue', marker='o', zorder=5,
+                           label=f"Пересечения с Y: {y_coords_str}")
+
+                # Подписи осей строго по переменным
+                self.ax.set_xlabel(str(var1))
+                self.ax.set_ylabel(str(var2))
+                self.ax.grid(True, linestyle='--', alpha=0.7)
+                self.ax.set_xlim(self.xlims)
+                self.ax.set_ylim(self.ylims)
+                if self.text_legend:
+                    self.ax.plot([], [], ' ', label=f"{self.text_legend}")
+                if self.legend:
+                    self.ax.legend()
+                plt.show()
+
+        except Exception as e1:
+            logging.warning(f"Matplotlib contour не сработал: {e1}")
+            # Если контур упал (сингулярности), используем sympy.plot_implicit
+            from sympy.plotting import plot_implicit
+            p = plot_implicit(self.eq, (var1, -10, 10), (var2, -10, 10), show=False)
+            p.show()

adopt_plot-0.0.1.dist-info/METADATA

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: adopt-plot
+Version: 0.0.1
+Summary: A universal plotter for implicit functions using adaptive engine selection.
+Author: VartRusData
+License: BSD 3-Clause
+Classifier: License :: OSI Approved :: BSD License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: sympy
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Dynamic: license-file
+
+# adopt-plot
+
+A simple Python library for plotting implicit functions with adaptive engine selection. 
+Just pass an equation and get a beautiful, mathematically correct plot.

adopt_plot-0.0.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+adopt_plot/__init__.py,sha256=lxUd6MgfND2muGCURby4s4R3SmAaW8qWGCNTC_0nT_E,418
+adopt_plot/plot.py,sha256=sUmjMOMVPYbv6gYGnQbkMPvnjrsKmqO5CzxJBbiLE8Q,19714
+adopt_plot-0.0.1.dist-info/licenses/LICENSE,sha256=WZte-rQIp6QmVj6-5DLhGEK2By1GnzykObQqbcazK-Y,1478
+adopt_plot-0.0.1.dist-info/METADATA,sha256=7_uWqtIYzU-f8TmtuCv45R30yC6Pc2bb37V2fpdRNws,602
+adopt_plot-0.0.1.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+adopt_plot-0.0.1.dist-info/top_level.txt,sha256=pFg71VDYed8NjYV3j_hv7GPLBiqJjDd-MkN6yJPOmHM,11
+adopt_plot-0.0.1.dist-info/RECORD,,

adopt_plot-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

adopt_plot-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2026 VartRusData.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice,
+this list of conditions and the following disclaimer.
+2. Redistributions in binary form must reproduce the above copyright notice,
+this list of conditions and the following disclaimer in the documentation and/or other materials provided
+with the distribution.
+3. Neither the name of the copyright holder nor the names of its contributors
+may be used to endorse or promote products derived from this software
+without specific prior written permission.
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
+INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
+OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

adopt_plot-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+adopt_plot