rsplotlib 0.3.5__cp39-cp39-win_amd64.whl

rsplotlib/dates.py

@@ -0,0 +1,85 @@
+"""rsplotlib.dates - Matplotlib dates 兼容接口 (子集)。
+
+提供日期刻度格式化器 ConciseDateFormatter，兼容 matplotlib.dates 的常用 API。
+日期数值沿用 matplotlib 约定：浮点数表示自 1970-01-01 (UTC) 起的天数。
+"""
+
+import datetime as _datetime
+
+from .ticks.ticker import Formatter
+
+# matplotlib 自 3.3 起默认纪元为 1970-01-01
+_EPOCH = _datetime.datetime(1970, 1, 1)
+
+
+def num2date(x):
+    """将 matplotlib 日期数值 (自纪元起的天数) 转为 datetime。"""
+    return _EPOCH + _datetime.timedelta(days=float(x))
+
+
+class ConciseDateFormatter(Formatter):
+    """紧凑日期格式化器，兼容 matplotlib.dates.ConciseDateFormatter。
+
+    根据刻度跨度自动选择合适的日期/时间粒度，尽量少地重复上层信息
+    (年/月/日)，产出简洁的刻度标签。
+
+    Args:
+        locator: 关联的刻度定位器 (与 matplotlib 一致，可选用于确定跨度)。
+        tz, formats, offset_formats, zero_formats, show_offset, usetex:
+            为兼容 matplotlib 签名而接受，当前实现下部分参数不生效。
+    """
+
+    def __init__(self, locator=None, tz=None, formats=None,
+                 offset_formats=None, zero_formats=None, show_offset=True,
+                 *, usetex=None):
+        self._locator = locator
+        self._tz = tz
+        self.formats = formats
+        self.zero_formats = zero_formats
+        self.offset_formats = offset_formats
+        self.show_offset = show_offset
+        self._usetex = usetex
+        self.offset_string = ''
+
+    def _to_datetime(self, value):
+        if isinstance(value, _datetime.datetime):
+            return value
+        if isinstance(value, _datetime.date):
+            return _datetime.datetime(value.year, value.month, value.day)
+        if isinstance(value, str):
+            return None
+        return num2date(value)
+
+    def _choose_scale(self, values):
+        """根据刻度跨度选择格式粒度。返回 strftime 模式。"""
+        dts = [self._to_datetime(v) for v in values]
+        dts = [d for d in dts if d is not None]
+        if len(dts) < 2:
+            return '%b %d'
+        span = max(dts) - min(dts)
+        secs = abs(span.total_seconds())
+        if secs > 365 * 24 * 3600:
+            return '%Y'
+        if secs > 30 * 24 * 3600:
+            return '%Y-%m'
+        if secs > 24 * 3600:
+            return '%b %d'
+        if secs > 3600:
+            return '%H:%M'
+        return '%H:%M:%S'
+
+    def format_ticks(self, values):
+        fmt = self._choose_scale(values)
+        return [self._format_one(v, fmt) for v in values]
+
+    def _format_one(self, value, fmt):
+        dt = self._to_datetime(value)
+        if dt is None:
+            return str(value)
+        return dt.strftime(fmt)
+
+    def __call__(self, value, pos=None):
+        return self._format_one(value, '%b %d')
+
+    def __repr__(self):
+        return 'ConciseDateFormatter()'

rsplotlib/figure/__init__.py

@@ -0,0 +1,2 @@
+"""rsplotlib.figure - 图形相关模块"""
+from ._defaults import DEFAULT_DPI, DEFAULT_FIGSIZE  # noqa: F401

rsplotlib/figure/_defaults.py

@@ -0,0 +1,13 @@
+"""rsplotlib._figure_defaults - 图形创建相关的默认值
+
+集中管理 figure 尺寸、DPI 等默认值，避免散落在 pyplot 模块中。
+底层实现: Rust figure 模块
+"""
+
+from .. import rsplotlib as _rs
+
+# 默认图形尺寸（英寸），与 matplotlib 默认一致
+DEFAULT_FIGSIZE = _rs.get_default_figsize()
+
+# 默认 DPI
+DEFAULT_DPI = _rs.get_default_dpi()

rsplotlib/gridspec.py

@@ -0,0 +1,6 @@
+"""rsplotlib.gridspec - 向后兼容的导入模块
+
+此模块提供从旧路径 `rsplotlib.gridspec` 导入的支持。
+实际实现在 `rsplotlib.layout.gridspec` 中。
+"""
+from .layout.gridspec import *  # noqa: F403, F401

rsplotlib/layout/__init__.py

@@ -0,0 +1,2 @@
+"""rsplotlib.layout - 布局管理"""
+from . import gridspec  # noqa: F401

rsplotlib/layout/gridspec.py

@@ -0,0 +1,161 @@
+"""rsplotlib.gridspec - Matplotlib GridSpec 兼容接口
+
+提供子图网格布局管理。
+底层实现已迁移至 Rust 层，此模块为保持完整 Python 接口的薄包装层。
+"""
+
+from .. import rsplotlib as _rs
+
+
+class GridSpec:
+    """GridSpec 布局管理器
+
+    用于在 Figure 中创建子图网格布局。
+    底层实现: Rust GridSpec
+
+    Args:
+        nrows: 行数
+        ncols: 列数
+        left: 左边界 (0-1)
+        bottom: 下边界 (0-1)
+        right: 右边界 (0-1)
+        top: 上边界 (0-1)
+        wspace: 列间距
+        hspace: 行间距
+        width_ratios: 列宽比例
+        height_ratios: 行高比例
+    """
+
+    def __init__(self, nrows=1, ncols=1, left=None, bottom=None, right=None,
+                 top=None, wspace=None, hspace=None, width_ratios=None,
+                 height_ratios=None):
+        self.nrows = nrows
+        self.ncols = ncols
+        self.left = left
+        self.bottom = bottom
+        self.right = right
+        self.top = top
+        self.wspace = wspace
+        self.hspace = hspace
+        self.width_ratios = width_ratios or [1] * ncols
+        self.height_ratios = height_ratios or [1] * nrows
+        self._impl = _rs.GridSpec(nrows, ncols, left, bottom, right, top, wspace, hspace, width_ratios, height_ratios)
+
+    def __getitem__(self, key):
+        """支持 gs[row, col] 和 gs[row_start:row_end, col_start:col_end] 语法
+
+        底层实现: Rust GridSpec.__getitem__
+
+        Returns:
+            SubplotSpec: 子图定位器
+        """
+        result = self._impl.__getitem__(key)
+        # 将 Rust SubplotSpec 包装为 Python SubplotSpec
+        return SubplotSpec._from_rust(result)
+
+    def get_subplot_params(self, figure=None):
+        """获取子图布局参数"""
+        return {
+            'left': self.left,
+            'bottom': self.bottom,
+            'right': self.right,
+            'top': self.top,
+            'wspace': self.wspace,
+            'hspace': self.hspace,
+        }
+
+    def tight_layout(self, figure=None, renderer=None, pad=1.08, h_pad=None, w_pad=None, rect=None):
+        """自动调整布局"""
+        self._impl.tight_layout(figure, renderer)
+
+
+class SubplotSpec:
+    """子图定位器
+
+    底层实现: Rust SubplotSpec
+    """
+
+    def __init__(self, gridspec, row_start=0, row_end=1, col_start=0, col_end=1):
+        self.gridspec = gridspec
+        self.row_start = row_start
+        self.row_end = row_end
+        self.col_start = col_start
+        self.col_end = col_end
+        if gridspec is not None:
+            self.numRows = gridspec.nrows
+            self.numCols = gridspec.ncols
+        else:
+            self.numRows = max(row_end, 1)
+            self.numCols = max(col_end, 1)
+        self.rowStart = row_start
+        self.rowStop = row_end
+        self.colStart = col_start
+        self.colStop = col_end
+        # 创建 Rust 实现
+        if gridspec is not None:
+            self._impl = _rs.SubplotSpec(gridspec._impl, row_start, row_end, col_start, col_end)
+        else:
+            self._impl = _rs.SubplotSpec(None, row_start, row_end, col_start, col_end)
+
+    @classmethod
+    def _from_rust(cls, rust_spec):
+        """从 Rust SubplotSpec 创建 Python 包装"""
+        instance = cls.__new__(cls)
+        instance._impl = rust_spec
+        instance.numRows = rust_spec.numRows
+        instance.numCols = rust_spec.numCols
+        instance.rowStart = rust_spec.rowStart
+        instance.rowStop = rust_spec.rowStop
+        instance.colStart = rust_spec.colStart
+        instance.colStop = rust_spec.colStop
+        instance.row_start = rust_spec.rowStart
+        instance.row_end = rust_spec.rowStop
+        instance.col_start = rust_spec.colStart
+        instance.col_end = rust_spec.colStop
+        instance.gridspec = None  # 简化处理
+        return instance
+
+    def get_position(self, figure):
+        """返回子图位置 (left, bottom, width, height)
+
+        底层实现: Rust SubplotSpec.get_position
+        """
+        return self._impl.get_position(figure)
+
+    def get_grid_span(self):
+        """返回网格跨度"""
+        return (self.rowStart, self.rowStop, self.colStart, self.colStop)
+
+
+# 便利函数
+def GridSpecFromSubplotSpec(nrows, ncols, subplot_spec, **kwargs):
+    """从 SubplotSpec 创建 GridSpec"""
+    from ..rsplotlib import gridspec_from_subplotspec
+    # 参数提取
+    left = kwargs.get('left')
+    bottom = kwargs.get('bottom')
+    right = kwargs.get('right')
+    top = kwargs.get('top')
+    wspace = kwargs.get('wspace')
+    hspace = kwargs.get('hspace')
+    width_ratios = kwargs.get('width_ratios')
+    height_ratios = kwargs.get('height_ratios')
+    rust_gs = gridspec_from_subplotspec(
+        nrows, ncols, left=left, bottom=bottom, right=right, top=top,
+        wspace=wspace, hspace=hspace, width_ratios=width_ratios,
+        height_ratios=height_ratios,
+    )
+    # 包装为 Python GridSpec
+    gs = GridSpec.__new__(GridSpec)
+    gs.nrows = nrows
+    gs.ncols = ncols
+    gs.left = left
+    gs.bottom = bottom
+    gs.right = right
+    gs.top = top
+    gs.wspace = wspace
+    gs.hspace = hspace
+    gs.width_ratios = width_ratios or [1] * ncols
+    gs.height_ratios = height_ratios or [1] * nrows
+    gs._impl = rust_gs
+    return gs

rsplotlib/patches.py

@@ -0,0 +1,54 @@
+class FancyArrowPatch:
+    def __init__(self, posA, posB, arrowstyle=None, connectionstyle=None,
+                 color=None, linestyle=None, linewidth=None, mutation_scale=None,
+                 shrinkA=None, shrinkB=None, zorder=None):
+        self.posA = posA
+        self.posB = posB
+        self.arrowstyle = arrowstyle
+        self.connectionstyle = connectionstyle
+        self.color = color
+        self.linestyle = linestyle
+        self.linewidth = linewidth
+        self.mutation_scale = mutation_scale
+        self.shrinkA = shrinkA if shrinkA is not None else 0.0
+        self.shrinkB = shrinkB if shrinkB is not None else 0.0
+        self.zorder = zorder
+        self._dpi_cor = 1.0
+        self._posA_posB = (posA, posB)
+        self.patchA = None
+        self.patchB = None
+
+    def _convert_xy_units(self, xy):
+        return xy
+
+    def get_connectionstyle(self):
+        return self.connectionstyle
+
+
+class _Path:
+    def __init__(self, vertices, codes=None):
+        self.vertices = vertices
+        self.codes = codes
+
+
+class _ConnectionStyleType:
+    pass
+
+
+class ConnectionStyle(_ConnectionStyleType):
+    def __init__(self, style):
+        self.style = style
+
+    def __call__(self, posA, posB, *args, **kwargs):
+        x1, y1 = posA
+        x2, y2 = posB
+        cx = (x1 + x2) / 2
+        cy = (y1 + y2) / 2
+        return _Path([posA, (cx, cy), posB], [1, 2, 2])
+
+
+ConnectionStyle.Angle3 = _ConnectionStyleType
+ConnectionStyle.Arc3 = _ConnectionStyleType
+ConnectionStyle.Angle = _ConnectionStyleType
+ConnectionStyle.Arc = _ConnectionStyleType
+ConnectionStyle.Bar = _ConnectionStyleType

rsplotlib/path.py

@@ -0,0 +1,4 @@
+class Path:
+    def __init__(self, vertices, codes=None):
+        self.vertices = vertices
+        self.codes = codes

rsplotlib/pylab.py

@@ -0,0 +1,22 @@
+"""rsplotlib.pylab - Matplotlib pylab 兼容接口"""
+
+from .utils._rcparams import rcParams, rcParamsOrig
+
+# mpl 模块可以从 pylab 导入
+
+
+class MplModule:
+    """兼容 matplotlib 的 mpl 模块"""
+
+    def __init__(self):
+        self.rcParams = rcParams
+        self.rcParamsOrig = rcParamsOrig
+
+    def get_backend(self):
+        return 'Agg'
+
+    def set_backend(self, backend):
+        pass
+
+
+mpl = MplModule()