rsplotlib 0.3.5__cp39-cp39-win_amd64.whl

rsplotlib/__init__.py

@@ -0,0 +1,101 @@
+"""rsplotlib 包顶层导出。建议通过导入自 `rsplotlib.api` 使用公开 API。"""
+
+from .core.api import *  # noqa: F403, F401
+from .core.api import __all__ as _api_all
+from .rsplotlib import register_sans_serif_font
+from . import pyplot, pylab  # noqa: F401
+from .utils import _font_resolver, style  # noqa: F401
+from .layout import gridspec  # noqa: F401
+from .ticks import ticker  # noqa: F401
+import rsplotlib.text as text
+
+GridSpec = gridspec.GridSpec
+MaxNLocator = ticker.MaxNLocator
+MultipleLocator = ticker.MultipleLocator
+AutoMinorLocator = ticker.AutoMinorLocator
+
+__version__ = "0.3.5"
+
+
+def _round_float_for_display(value):
+    if isinstance(value, float):
+        rounded = round(value, 15)
+        rounded_int = round(rounded)
+        if abs(rounded - rounded_int) < 1e-10:
+            return rounded_int
+        return rounded
+    return value
+
+
+def _patch_rsnumpy_repr():
+    try:
+        import rsnumpy as np
+        ndarray_cls = np.ndarray
+        
+        original_repr = ndarray_cls.__repr__
+        original_str = ndarray_cls.__str__
+        
+        def patched_repr(self):
+            try:
+                data = self.tolist()
+                
+                def convert_to_python(obj):
+                    if hasattr(obj, 'tolist'):
+                        return convert_to_python(obj.tolist())
+                    elif isinstance(obj, list):
+                        return [convert_to_python(item) for item in obj]
+                    elif isinstance(obj, complex):
+                        real_part = _round_float_for_display(obj.real)
+                        imag_part = _round_float_for_display(obj.imag)
+                        return complex(real_part, imag_part)
+                    else:
+                        return _round_float_for_display(obj)
+                
+                converted_data = convert_to_python(data)
+                
+                if isinstance(converted_data, list) and len(converted_data) == 1:
+                    converted_data = converted_data[0]
+                
+                def format_list(lst):
+                    if not lst:
+                        return "[]"
+                    first = lst[0]
+                    if isinstance(first, list):
+                        inner = ", ".join(format_list(item) for item in lst)
+                        return f"[{inner}]"
+                    elif isinstance(first, complex):
+                        formatted = [f"({x.real}+{x.imag}j)" for x in lst]
+                        return f"[{', '.join(formatted)}]"
+                    else:
+                        return str(lst)
+                
+                if isinstance(converted_data, list):
+                    return format_list(converted_data)
+                else:
+                    return str(converted_data)
+            except Exception:
+                ndarray_cls.__str__ = original_str
+                result = original_repr(self)
+                ndarray_cls.__str__ = patched_str
+                return result
+        
+        def patched_str(self):
+            try:
+                return patched_repr(self)
+            except Exception:
+                return original_str(self)
+        
+        ndarray_cls.__repr__ = patched_repr
+        ndarray_cls.__str__ = patched_str
+    except ImportError:
+        pass
+
+
+_patch_rsnumpy_repr()
+
+
+__all__ = list(_api_all) + [
+    'pyplot', 'style', 'gridspec', 'ticker', 'text',
+    'GridSpec', 'MaxNLocator', 'MultipleLocator',
+    'AutoMinorLocator', 'register_sans_serif_font',
+]

rsplotlib/_font_resolver.py

@@ -0,0 +1,167 @@
+"""rsplotlib._font_resolver - 字体族名 → 字体文件路径解析
+
+将 matplotlib 风格的无衬线字体族名（如 "Arial Unicode MS"、"Helvetica Neue"）
+映射到本地字体文件路径。找不到时返回 None，由调用方决定回退到默认字体。
+"""
+import os
+from typing import Optional, List
+import sys
+
+
+# ====== 字体族名 → 候选文件路径映射（按平台分别维护）======
+
+_FONT_NAME_TO_PATHS = {
+    # macOS
+    "Arial Unicode MS": [
+        "/Library/Fonts/Arial Unicode.ttf",
+        "/System/Library/Fonts/Supplemental/Arial Unicode.ttf",
+    ],
+    "Arial": [
+        "/Library/Fonts/Arial.ttf",
+        "/System/Library/Fonts/Supplemental/Arial.ttf",
+    ],
+    "Helvetica": [
+        "/System/Library/Fonts/Helvetica.ttc",
+        "/System/Library/Fonts/HelveticaNeue.ttc",
+    ],
+    "Helvetica Neue": [
+        "/System/Library/Fonts/HelveticaNeue.ttc",
+    ],
+    "PingFang SC": [
+        "/System/Library/Fonts/PingFang.ttc",
+    ],
+    "Heiti SC": [
+        "/System/Library/Fonts/STHeiti Light.ttc",
+        "/System/Library/Fonts/STHeiti Medium.ttc",
+    ],
+    "Hiragino Sans GB": [
+        "/System/Library/Fonts/Hiragino Sans GB W3.otf",
+        "/System/Library/Fonts/Hiragino Sans GB W6.otf",
+    ],
+    # Linux
+    "DejaVu Sans": [
+        "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
+        "/usr/share/fonts/dejavu/DejaVuSans.ttf",
+    ],
+    "Liberation Sans": [
+        "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf",
+    ],
+    "Noto Sans CJK SC": [
+        "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
+        "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",
+        "/usr/share/fonts/noto-cjk/NotoSansCJK-Regular.ttc",
+    ],
+    "WenQuanYi Micro Hei": [
+        "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc",
+    ],
+    # Windows
+    "Microsoft YaHei": [
+        "C:/Windows/Fonts/msyh.ttc",
+        "C:/Windows/Fonts/msyh.ttf",
+        "C:/Windows/Fonts/msyhbd.ttc",
+    ],
+    "SimHei": [
+        "C:/Windows/Fonts/simhei.ttf",
+    ],
+    "SimSun": [
+        "C:/Windows/Fonts/simsun.ttc",
+    ],
+}
+
+
+# 跨平台按系统名归一化的额外兜底字体路径
+
+def _system_fallback_paths() -> List[str]:
+    """按当前操作系统返回一组"通用全功能字体"候选路径"""
+    system = sys.platform
+    if system == "darwin":
+        return [
+            "/Library/Fonts/Arial Unicode.ttf",
+            "/System/Library/Fonts/Supplemental/Arial Unicode.ttf",
+        ]
+    elif system == "linux":
+        return [
+            "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
+            "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",
+            "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
+            "/usr/share/fonts/dejavu/DejaVuSans.ttf",
+        ]
+    elif system == "win32":
+        return [
+            "C:/Windows/Fonts/msyh.ttc",
+            "C:/Windows/Fonts/msyh.ttf",
+            "C:/Windows/Fonts/msyhbd.ttc",
+        ]
+    elif system == "cygwin":
+        return [
+            "C:/Windows/Fonts/msyh.ttc",
+            "C:/Windows/Fonts/arial.ttf",
+        ]
+    return []
+
+
+def resolve_font_path(family: str) -> Optional[str]:
+    """根据字体族名查找本地的字体文件路径。
+
+    找不到时返回 None。
+    """
+    if not family:
+        return None
+    # 精确匹配
+    candidates = _FONT_NAME_TO_PATHS.get(family, [])
+    for path in candidates:
+        if os.path.isfile(path):
+            return path
+    # 大小写不敏感匹配
+    lower_map = {k.lower(): v for k, v in _FONT_NAME_TO_PATHS.items()}
+    candidates = lower_map.get(family.lower(), [])
+    for path in candidates:
+        if os.path.isfile(path):
+            return path
+    # 平台回退
+    for path in _system_fallback_paths():
+        if os.path.isfile(path):
+            return path
+    return None
+
+
+def apply_rcparams_font() -> Optional[str]:
+    """读取 rcParams["font.sans-serif"]，把第一个能解析到本地文件的字体注册到 plotters。
+
+    返回实际注册的字体文件路径，如果没找到任何字体则返回 None。
+    """
+    try:
+        # 延迟导入避免循环依赖
+        from . import rsplotlib as _rsplotlib
+        from .pylab import mpl
+    except Exception:
+        return None
+
+    sans_serif = mpl.rcParams.get("font.sans-serif")
+    if not sans_serif:
+        return None
+
+    if isinstance(sans_serif, str):
+        candidates = [sans_serif]
+    else:
+        try:
+            candidates = list(sans_serif)
+        except TypeError:
+            candidates = [str(sans_serif)]
+
+    # "sans-serif" 关键字跳过：让 plotters 使用内部默认
+    candidates = [c for c in candidates if c and c.lower() != "sans-serif"]
+
+    for family in candidates:
+        path = resolve_font_path(family)
+        if path is None:
+            # 可能是直接的字体文件路径
+            if os.path.isfile(family):
+                path = family
+        if path is not None:
+            try:
+                _rsplotlib.register_sans_serif_font(path)
+                return path
+            except Exception:
+                continue
+    return None

rsplotlib/cm.py

@@ -0,0 +1,15 @@
+class ScalarMappable:
+    def __init__(self, cmap=None, norm=None):
+        self.cmap = cmap
+        self.norm = norm
+        self.vmin = None
+        self.vmax = None
+
+    def set_clim(self, vmin=None, vmax=None):
+        self.vmin = vmin
+        self.vmax = vmax
+
+    def to_rgba(self, values):
+        if not isinstance(values, (list, tuple)):
+            values = [values]
+        return [[0.0, 0.0, 0.0, 1.0] for _ in values]

rsplotlib/collections.py

@@ -0,0 +1,35 @@
+class PathCollection:
+    def __init__(self, paths):
+        self.paths = paths
+        self.zorder = None
+
+    def set_zorder(self, zorder):
+        self.zorder = zorder
+
+
+class LineCollection:
+    def __init__(self, segments, colors=None, linewidths=None, linestyle=None,
+                 alpha=None, antialiaseds=None, zorder=None):
+        self.segments = segments
+        self.colors = colors
+        self.linewidths = linewidths
+        self.linestyle = linestyle
+        self.alpha = alpha
+        self.antialiaseds = antialiaseds
+        self.zorder = zorder
+        self.cmap = None
+        self.vmin = None
+        self.vmax = None
+
+    def set_cmap(self, cmap):
+        self.cmap = cmap
+
+    def set_clim(self, vmin, vmax):
+        self.vmin = vmin
+        self.vmax = vmax
+
+    def set_zorder(self, zorder):
+        self.zorder = zorder
+
+    def set_label(self, label):
+        self.label = label

rsplotlib/colors.py

@@ -0,0 +1,186 @@
+"""rsplotlib.colors - Matplotlib colors 兼容接口 (子集)。
+
+提供归一化类 Normalize / LogNorm，用于将数据映射到 [0, 1] 供 colormap 上色。
+当前 imshow/pcolormesh 等接受 norm 参数，Normalize/LogNorm 在此处提供兼容 API。
+"""
+
+import math
+
+
+class Normalize:
+    """将数据线性归一化到 [0, 1]，兼容 matplotlib.colors.Normalize。
+
+    Args:
+        vmin, vmax: 归一化值域端点，缺省时在首次调用时按数据自动推断。
+        clip: 是否将越界值裁剪到 [0, 1]。
+    """
+
+    # 归一化类型标记：供 pyplot 把归一化方式（线性/对数）下沉给 Rust 上色与颜色条。
+    _norm_kind = 'linear'
+
+    def __init__(self, vmin=None, vmax=None, clip=False):
+        self.vmin = vmin
+        self.vmax = vmax
+        self.clip = clip
+
+    def autoscale_None(self, values):
+        """当 vmin/vmax 未设置时，根据数据填充。"""
+        seq = _flatten(values)
+        if seq:
+            if self.vmin is None:
+                self.vmin = min(seq)
+            if self.vmax is None:
+                self.vmax = max(seq)
+
+    def _normalize_scalar(self, value):
+        vmin = 0.0 if self.vmin is None else float(self.vmin)
+        vmax = 1.0 if self.vmax is None else float(self.vmax)
+        if vmax == vmin:
+            return 0.0
+        t = (float(value) - vmin) / (vmax - vmin)
+        if self.clip:
+            t = max(0.0, min(1.0, t))
+        return t
+
+    def __call__(self, value, clip=None):
+        if clip is not None:
+            saved = self.clip
+            self.clip = clip
+            try:
+                return self._apply(value)
+            finally:
+                self.clip = saved
+        return self._apply(value)
+
+    def _apply(self, value):
+        if not _is_sequence(value):
+            return self._normalize_scalar(value)
+        # 循环不变量外提：vmin/vmax 的 float 转换与除数只算一次，再逐元素映射。
+        seq = _flatten(value)
+        vmin = 0.0 if self.vmin is None else float(self.vmin)
+        vmax = 1.0 if self.vmax is None else float(self.vmax)
+        if vmax == vmin:
+            return [0.0] * len(seq)
+        inv = 1.0 / (vmax - vmin)
+        if self.clip:
+            return [max(0.0, min(1.0, (v - vmin) * inv)) for v in seq]
+        return [(v - vmin) * inv for v in seq]
+
+    def __repr__(self):
+        return f'Normalize(vmin={self.vmin}, vmax={self.vmax})'
+
+
+class LogNorm(Normalize):
+    """对数归一化，兼容 matplotlib.colors.LogNorm。
+
+    将数据按对数刻度映射到 [0, 1]；非正值被视为越界。
+    """
+
+    _norm_kind = 'log'
+
+    def autoscale_None(self, values):
+        """当 vmin/vmax 未设置时，用数据的最小正值 / 最大值填充（对数刻度要求正值）。"""
+        seq = [v for v in _flatten(values) if v > 0]
+        if seq:
+            if self.vmin is None:
+                self.vmin = min(seq)
+            if self.vmax is None:
+                self.vmax = max(seq)
+
+    def _normalize_scalar(self, value):
+        vmin = self.vmin
+        vmax = self.vmax
+        if vmin is None or vmax is None or vmin <= 0 or vmax <= 0:
+            return 0.0
+        v = float(value)
+        if v <= 0:
+            return 0.0
+        t = (math.log(v) - math.log(vmin)) / (math.log(vmax) - math.log(vmin))
+        if self.clip:
+            t = max(0.0, min(1.0, t))
+        return t
+
+    def _apply(self, value):
+        if not _is_sequence(value):
+            return self._normalize_scalar(value)
+        # 循环不变量外提：log(vmin) 与对数跨度只算一次，避免逐元素重复求 log。
+        seq = _flatten(value)
+        vmin = self.vmin
+        vmax = self.vmax
+        if vmin is None or vmax is None or vmin <= 0 or vmax <= 0:
+            return [0.0] * len(seq)
+        log_vmin = math.log(vmin)
+        span = math.log(vmax) - log_vmin
+        if span == 0.0:
+            return [0.0] * len(seq)
+        inv = 1.0 / span
+        if self.clip:
+            return [max(0.0, min(1.0, (math.log(v) - log_vmin) * inv)) if v > 0 else 0.0
+                    for v in seq]
+        return [(math.log(v) - log_vmin) * inv if v > 0 else 0.0 for v in seq]
+
+    def __repr__(self):
+        return f'LogNorm(vmin={self.vmin}, vmax={self.vmax})'
+
+
+def _is_sequence(obj):
+    if obj is None or isinstance(obj, str):
+        return False
+    return hasattr(obj, 'tolist') or hasattr(obj, '__iter__')
+
+
+def _flatten(values):
+    """递归展开嵌套序列 / 数组对象为一维浮点列表。"""
+    if values is None:
+        return []
+    if hasattr(values, 'tolist'):
+        values = values.tolist()
+    out = []
+    if isinstance(values, (list, tuple)):
+        for item in values:
+            if isinstance(item, (list, tuple)):
+                out.extend(_flatten(item))
+            else:
+                out.append(float(item))
+    else:
+        out.append(float(values))
+    return out
+
+
+class _ColorConverter:
+    def to_rgba_array(self, colors, alpha=None):
+        if not isinstance(colors, (list, tuple)):
+            colors = [colors]
+        result = []
+        for color in colors:
+            if color is None or color == 'none':
+                rgba = [0.0, 0.0, 0.0, 0.0]
+            elif color == 'k' or color == 'black':
+                rgba = [0.0, 0.0, 0.0, 1.0]
+            elif color == 'w' or color == 'white':
+                rgba = [1.0, 1.0, 1.0, 1.0]
+            elif color == 'r' or color == 'red':
+                rgba = [1.0, 0.0, 0.0, 1.0]
+            elif color == 'g' or color == 'green':
+                rgba = [0.0, 1.0, 0.0, 1.0]
+            elif color == 'b' or color == 'blue':
+                rgba = [0.0, 0.0, 1.0, 1.0]
+            elif color == 'c' or color == 'cyan':
+                rgba = [0.0, 1.0, 1.0, 1.0]
+            elif color == 'm' or color == 'magenta':
+                rgba = [1.0, 0.0, 1.0, 1.0]
+            elif color == 'y' or color == 'yellow':
+                rgba = [1.0, 1.0, 0.0, 1.0]
+            elif isinstance(color, (list, tuple)) and len(color) in (3, 4):
+                rgba = list(color)
+                if len(rgba) == 3:
+                    rgba.append(1.0)
+            else:
+                rgba = [0.0, 0.0, 0.0, 1.0]
+            if alpha is not None:
+                rgba[3] = alpha
+            result.append(rgba)
+        return result
+
+
+colorConverter = _ColorConverter()

rsplotlib/core/__init__.py

@@ -0,0 +1,2 @@
+"""rsplotlib.core - 核心 API 函数"""
+from .api import *  # noqa: F403, F401