fractal 0.5.0__py3-none-any.whl

fractal/__init__.py

@@ -0,0 +1,8 @@
+"""
+分形相关操作封装
+"""
+from .lsystem import Pen
+from .ifs import IFS
+from .julia import Julia
+from .mandelbrot import Mandelbrot
+from .cifs import CIFS

fractal/base.py

@@ -0,0 +1,412 @@
+"""
+    基类
+    功能说明：
+    - 保存图片：点击"保存"按钮 或 Ctrl+P
+    - 放大：左键点击显示选择框，拖动调整位置，再次点击确认放大
+    - 缩小：右键点击缩小
+    - 取消选择：右键点击或按ESC键
+"""
+
+import tkinter as tk
+from tkinter import messagebox, filedialog
+import threading
+
+# 尝试导入Pillow和numpy用于高性能绘图
+try:
+    from PIL import Image, ImageTk, ImageDraw
+    HAS_PIL = True
+except ImportError:
+    HAS_PIL = False
+
+try:
+    import numpy as np
+    HAS_NUMPY = True
+except ImportError:
+    HAS_NUMPY = False
+
+
+class Base(object):
+
+    def __init__(self, size, callRun, title=""):
+        """
+            size: 画布尺寸
+            callRun: 子类的绘画函数回调
+            title: 画布标题
+        """
+        self._run = callRun
+        self.title = title if title else "Fractal"
+        self.width = size[0]
+        self.height = size[1]
+
+        # 绘图模式标记：'pixel'为像素绘图，'line'为线条绘图
+        self._draw_mode = 'pixel'
+
+        # 创建像素数组（用于存储计算结果）
+        if HAS_NUMPY:
+            self._pixel_array = np.zeros((self.height, self.width, 3), dtype=np.uint8)
+            self._pixel_array.fill(255)  # 白色背景
+        else:
+            self._pixel_array = None
+
+        # 创建主窗口
+        self.root = tk.Tk()
+        self.root.title(self.title)
+        self.root.resizable(False, False)
+
+        # 创建主框架
+        self.main_frame = tk.Frame(self.root)
+        self.main_frame.pack(padx=5, pady=5)
+
+        # 创建画布
+        self.canvas = tk.Canvas(self.main_frame, width=self.width, height=self.height, bg='white')
+        self.canvas.pack()
+
+        # 创建按钮框架
+        self.button_frame = tk.Frame(self.root)
+        self.button_frame.pack(fill=tk.X, padx=5, pady=5)
+
+        # 保存按钮
+        self.save_btn = tk.Button(self.button_frame, text="保存图片", command=self._on_save_click,
+                                   width=12, height=1, bg="#4CAF50", fg="white", font=("Arial", 10))
+        self.save_btn.pack(side=tk.LEFT, padx=5)
+
+        # 另存为按钮
+        self.save_as_btn = tk.Button(self.button_frame, text="另存为...", command=self._on_save_as_click,
+                                      width=12, height=1, bg="#2196F3", fg="white", font=("Arial", 10))
+        self.save_as_btn.pack(side=tk.LEFT, padx=5)
+
+        # 状态标签
+        self.status_label = tk.Label(self.button_frame, text="就绪", font=("Arial", 9), fg="gray")
+        self.status_label.pack(side=tk.RIGHT, padx=10)
+
+        # 创建图像对象
+        if HAS_PIL:
+            self._pil_image = Image.new('RGB', (self.width, self.height), 'white')
+            self._draw = ImageDraw.Draw(self._pil_image)  # PIL绘图对象
+            self.image = ImageTk.PhotoImage(self._pil_image, master=self.root)
+        else:
+            self.image = tk.PhotoImage(width=self.width, height=self.height, master=self.root)
+            self.image.put("white", to=(0, 0, self.width, self.height))
+
+        self.canvas_img = self.canvas.create_image(0, 0, anchor="nw", image=self.image)
+
+        self.psave = False
+        self.scalaRate = 4  # 放大（缩小）倍数
+        self.ctrl = False  # Ctrl键按下
+        self.cnt = 1  # 连续命名时标号
+        self._running = True
+        self._drawing = False
+        self._draw_thread = None
+
+        # 选择框相关
+        self._selection_rect = None  # 选择框ID
+        self._selecting = False  # 是否在选择模式
+        self._selection_center = None  # 选择框中心坐标
+        self._drag_start = None  # 拖动起始坐标
+
+        # 绑定事件
+        self.root.protocol("WM_DELETE_WINDOW", self._on_close)
+        self.root.bind("<Control-p>", self._on_save)
+        self.root.bind("<Control-s>", self._on_save_click)
+        self.root.bind("<Escape>", self._cancel_selection)  # ESC取消选择
+        self.canvas.bind("<Button-1>", self._on_left_click)  # 左键
+        self.canvas.bind("<Button-3>", self._on_right_click)  # 右键
+        self.canvas.bind("<B1-Motion>", self._on_drag)  # 拖动
+        self.canvas.bind("<ButtonRelease-1>", self._on_drag_end)  # 拖动结束
+        self.canvas.bind("<Double-Button-1>", self._on_double_click)  # 双击确认
+
+    def _on_close(self):
+        self._running = False
+        self.root.quit()
+        self.root.destroy()
+
+    def _update_status(self, text, color="gray"):
+        """更新状态标签"""
+        self.status_label.config(text=text, fg=color)
+
+    def _on_save(self, event=None):
+        """快捷键保存"""
+        self._on_save_click()
+
+    def _cancel_selection(self, event=None):
+        """取消选择框"""
+        if self._selection_rect is not None:
+            self.canvas.delete(self._selection_rect)
+            self._selection_rect = None
+            self._selecting = False
+            self._selection_center = None
+            self._update_status("已取消选择", "gray")
+
+    def _get_selection_rect_coords(self, center_x, center_y):
+        """获取选择框的坐标"""
+        w = self.width // (2 * self.scalaRate)
+        h = self.height // (2 * self.scalaRate)
+        return (center_x - w, center_y - h, center_x + w, center_y + h)
+
+    def _sync_pil_image(self):
+        """同步PIL图像从numpy数组"""
+        if HAS_PIL and HAS_NUMPY and self._draw_mode == 'pixel':
+            self._pil_image = Image.fromarray(self._pixel_array, 'RGB')
+            self._draw = ImageDraw.Draw(self._pil_image)
+
+    def _on_save_click(self):
+        """点击保存按钮"""
+        if self._drawing:
+            messagebox.showwarning("提示", "请等待绘图完成后再保存")
+            return
+        # 确保PIL图像与像素数组同步（仅对像素绘图模式）
+        self._sync_pil_image()
+        if self.title == "":
+            self.title = "fractal"
+        if HAS_PIL:
+            filename = self.title + str(self.cnt) + ".png"
+            self._pil_image.save(filename)
+            self._update_status(f"已保存: {filename}", "green")
+            messagebox.showinfo("保存成功", f"图片已保存为 {filename}")
+        else:
+            filename = self.title + str(self.cnt) + ".ps"
+            self.canvas.postscript(file=filename, colormode='color')
+            self._update_status(f"已保存: {filename}", "green")
+            messagebox.showinfo("保存成功", f"图片已保存为 {filename}\n(PostScript格式)")
+        self.cnt += 1
+
+    def _on_save_as_click(self):
+        """另存为对话框"""
+        if self._drawing:
+            messagebox.showwarning("提示", "请等待绘图完成后再保存")
+            return
+        # 确保PIL图像与像素数组同步
+        self._sync_pil_image()
+        if HAS_PIL:
+            filetypes = [
+                ("PNG 图片", "*.png"),
+                ("JPEG 图片", "*.jpg"),
+                ("BMP 图片", "*.bmp"),
+                ("所有文件", "*.*")
+            ]
+            filename = filedialog.asksaveasfilename(
+                title="保存图片",
+                initialfile=self.title + ".png",
+                filetypes=filetypes,
+                defaultextension=".png"
+            )
+            if filename:
+                self._pil_image.save(filename)
+                self._update_status(f"已保存: {filename}", "green")
+                messagebox.showinfo("保存成功", f"图片已保存为 {filename}")
+        else:
+            filetypes = [("PostScript", "*.ps"), ("所有文件", "*.*")]
+            filename = filedialog.asksaveasfilename(
+                title="保存图片",
+                initialfile=self.title + ".ps",
+                filetypes=filetypes,
+                defaultextension=".ps"
+            )
+            if filename:
+                self.canvas.postscript(file=filename, colormode='color')
+                self._update_status(f"已保存: {filename}", "green")
+                messagebox.showinfo("保存成功", f"图片已保存为 {filename}")
+
+    def _on_left_click(self, event):
+        """左键点击处理"""
+        if self._drawing or not hasattr(self, "scala"):
+            return
+
+        # 如果正在选择模式，确认放大
+        if self._selecting and self._selection_rect is not None:
+            self._confirm_zoom(event.x, event.y)
+            return
+
+        # 第一次点击，显示选择框
+        i, j = event.x, event.y
+        coords = self._get_selection_rect_coords(i, j)
+        self._selection_rect = self.canvas.create_rectangle(
+            coords[0], coords[1], coords[2], coords[3],
+            outline="red", width=2, dash=(4, 4)
+        )
+        self._selecting = True
+        self._selection_center = (i, j)
+        self._update_status("拖动调整位置，点击确认，ESC取消", "blue")
+
+    def _on_double_click(self, event):
+        """双击确认放大"""
+        if self._selecting and self._selection_rect is not None and not self._drawing:
+            self._confirm_zoom(self._selection_center[0], self._selection_center[1])
+
+    def _on_drag(self, event):
+        """拖动选择框"""
+        if not self._selecting or self._selection_rect is None:
+            return
+        # 更新选择框位置
+        coords = self._get_selection_rect_coords(event.x, event.y)
+        self.canvas.coords(self._selection_rect, coords[0], coords[1], coords[2], coords[3])
+        self._selection_center = (event.x, event.y)
+
+    def _on_drag_end(self, event):
+        """拖动结束"""
+        if self._selecting:
+            self._selection_center = (event.x, event.y)
+
+    def _confirm_zoom(self, x, y):
+        """确认放大"""
+        # 删除选择框
+        if self._selection_rect is not None:
+            self.canvas.delete(self._selection_rect)
+            self._selection_rect = None
+        self._selecting = False
+
+        # 执行放大
+        self._clear_pixels()
+        self.scala(x, y, self.scalaRate)
+        self._start_draw()
+
+    def _on_right_click(self, event):
+        """右键点击处理"""
+        if self._drawing:
+            return
+
+        # 如果在选择模式，取消选择
+        if self._selecting:
+            self._cancel_selection()
+            return
+
+        # 否则执行缩小
+        if hasattr(self, "scala"):
+            self._clear_pixels()
+            self.scala(event.x, event.y, 1 / self.scalaRate)
+            self._start_draw()
+
+    def _clear_pixels(self):
+        """清空像素数据"""
+        if HAS_NUMPY:
+            self._pixel_array.fill(255)
+        if HAS_PIL:
+            self._pil_image = Image.new('RGB', (self.width, self.height), 'white')
+            self._draw = ImageDraw.Draw(self._pil_image)
+
+    def _start_draw(self):
+        """开始绘图线程"""
+        self._drawing = True
+        self._update_status("绘图中...", "orange")
+        self._draw_thread = threading.Thread(target=self._run_and_update, daemon=True)
+        self._draw_thread.start()
+
+    def _run_and_update(self):
+        """绘图并在完成后更新显示"""
+        try:
+            self._run()
+            # 绘图完成后立即同步PIL图像（在线程中执行）
+            if HAS_PIL and HAS_NUMPY and self._draw_mode == 'pixel':
+                self._pil_image = Image.fromarray(self._pixel_array, 'RGB')
+                self._draw = ImageDraw.Draw(self._pil_image)
+        except Exception as e:
+            print(f"绘图错误: {e}")
+        finally:
+            self._drawing = False
+            # 在主线程中更新显示
+            self.root.after(0, self._finish_draw)
+
+    def _finish_draw(self):
+        """绘图完成后的处理"""
+        self._update_display()
+        self._update_status("绘图完成 - 左键选择放大区域", "green")
+
+    def _update_display(self):
+        """更新显示（在主线程中调用）"""
+        if HAS_PIL and HAS_NUMPY and self._draw_mode == 'pixel':
+            # 从numpy数组创建PIL图像
+            self._pil_image = Image.fromarray(self._pixel_array, 'RGB')
+            self._draw = ImageDraw.Draw(self._pil_image)
+        if HAS_PIL:
+            self.image = ImageTk.PhotoImage(self._pil_image, master=self.root)
+            self.canvas.itemconfig(self.canvas_img, image=self.image)
+
+    def save(self, title):
+        """保存图片"""
+        if HAS_PIL:
+            self._pil_image.save(title)
+        else:
+            self.canvas.postscript(file=title, colormode='color')
+
+    def set_pixel(self, x, y, color):
+        """设置单个像素"""
+        if isinstance(color, (list, tuple)):
+            r, g, b = int(color[0]), int(color[1]), int(color[2])
+        else:
+            r, g, b = 0, 0, 0
+
+        x, y = int(x), int(y)
+        if 0 <= x < self.width and 0 <= y < self.height:
+            if HAS_NUMPY:
+                self._pixel_array[y, x] = [r, g, b]
+            elif HAS_PIL:
+                self._pil_image.putpixel((x, y), (r, g, b))
+            else:
+                color_str = "#{:02x}{:02x}{:02x}".format(r, g, b)
+                self.image.put(color_str, (x, y))
+
+    def set_pixel_fast(self, x, y, r, g, b):
+        """快速设置单个像素"""
+        x, y = int(x), int(y)
+        if 0 <= x < self.width and 0 <= y < self.height:
+            if HAS_NUMPY:
+                self._pixel_array[y, x] = [r, g, b]
+            elif HAS_PIL:
+                self._pil_image.putpixel((x, y), (r, g, b))
+            else:
+                color_str = "#{:02x}{:02x}{:02x}".format(r, g, b)
+                self.image.put(color_str, (x, y))
+
+    def set_pixels(self, pixels):
+        """批量设置像素"""
+        for (x, y), color in pixels.items():
+            self.set_pixel(x, y, color)
+
+    def draw_line(self, start, end, color, width=1):
+        """
+        绘制线条
+        同时在Canvas和PIL图像上绘制，确保保存时图像正确
+        """
+        self._draw_mode = 'line'  # 标记为线条绘图模式
+
+        # 在Canvas上绘制
+        if isinstance(color, (list, tuple)):
+            r, g, b = int(color[0]), int(color[1]), int(color[2])
+            canvas_color = "#{:02x}{:02x}{:02x}".format(r, g, b)
+        else:
+            canvas_color = color
+            r, g, b = 0, 0, 0
+
+        self.canvas.create_line(start[0], start[1], end[0], end[1], fill=canvas_color, width=width)
+
+        # 在PIL图像上绘制（用于保存）
+        if HAS_PIL:
+            self._draw.line([start[0], start[1], end[0], end[1]], fill=(r, g, b), width=width)
+
+    def draw_rect(self, x, y, w, h, color, outline_only=True):
+        """绘制矩形"""
+        if isinstance(color, (list, tuple)):
+            r, g, b = color
+            color = "#{:02x}{:02x}{:02x}".format(int(r), int(g), int(b))
+        if outline_only:
+            self.canvas.create_rectangle(x, y, x + w, y + h, outline=color, width=1)
+        else:
+            self.canvas.create_rectangle(x, y, x + w, y + h, fill=color, outline=color)
+
+    def fill(self, color):
+        """填充背景色"""
+        if isinstance(color, (list, tuple)):
+            r, g, b = int(color[0]), int(color[1]), int(color[2])
+        else:
+            r, g, b = 255, 255, 255
+
+        if HAS_NUMPY:
+            self._pixel_array[:, :] = [r, g, b]
+        if HAS_PIL:
+            self._pil_image = Image.new('RGB', (self.width, self.height), (r, g, b))
+            self._draw = ImageDraw.Draw(self._pil_image)
+
+    def wait(self):
+        """等待用户关闭"""
+        self._start_draw()
+        self.root.mainloop()

fractal/cifs.py

@@ -0,0 +1,100 @@
+"""
+    Complex Iterated Function System.
+    复迭代函数系统
+"""
+
+from .base import Base, HAS_NUMPY
+from .colors import *
+from threading import Thread
+
+
+class CIFS(Base):
+
+    def __init__(self, size, title=""):
+        Base.__init__(self, size, self._run, title)
+        self.setRadius(10)
+        self.width = size[0]
+        self.height = size[1]
+        self.setRange(10, 10)
+        self.setCentre(0 + 0j)
+        self.ifunc = None
+
+    def setFunction(self, ifunc):
+        # 设置迭代函数
+        self.ifunc = ifunc
+
+    def setRadius(self, R):
+        # 设置逃逸半径
+        self.R = R
+
+    def color(self, n, r=2):
+        if n < len(reds):
+            return reds[n]
+        else:
+            if r < self.R:
+                return blues[int((len(blues) - 1) * r / self.R)]
+            else:
+                return purples[int((len(purples) - 1) * self.R / r)]
+
+    def setColor(self, call):
+        self.color = call
+
+    def setCentre(self, z0):
+        # 设置中心点
+        self.z0 = z0
+
+    def setRange(self, xmax, ymax):
+        # 设置坐标范围，范围越小图放大倍数越高
+        self.xmax = xmax
+        self.ymax = ymax
+
+    def __getXY(self, i, j):
+        # 通过像素坐标获取映射后的坐标
+        return complex((i / self.width - 0.5) * self.xmax + self.z0.real,
+                       (j / self.height - 0.5) * self.ymax + self.z0.imag)
+
+    def scala(self, i, j, rate):
+        # 将(i, j)像素点置于中心位置，放大rete倍
+        self.setCentre(self.__getXY(i, j))
+        self.xmax /= rate
+        self.ymax /= rate
+
+    def __calc(self, start, w, h):
+        # 绘制以start为起点，宽w,高h的子区域
+        for i in range(start[0], start[0] + w):
+            for j in range(start[1], start[1] + h):
+                ct = 0
+                z = self.__getXY(i, j)
+                for k in range(self.N):
+                    ct = k
+                    if abs(z) > self.R:
+                        break
+                    z = self.ifunc(z)
+                col = self.color(ct, abs(z))
+                if HAS_NUMPY and self._pixel_array is not None:
+                    self._pixel_array[j, i] = [int(col[0]), int(col[1]), int(col[2])]
+                else:
+                    self.set_pixel_fast(i, j, col[0], col[1], col[2])
+
+    def _run(self):
+        print("x range ：[-%.2e,%.2e]\ny range ：[-%.2e,%.2e]" % (
+            self.xmax, self.xmax, self.ymax, self.ymax))
+        if self.ifunc is None:
+            raise Exception("请设置迭代函数")
+        tn = 5
+        ci = self.width // tn
+        cj = self.height // tn
+        ts = []
+        for i in range(tn):
+            for j in range(tn):
+                t = Thread(target=self.__calc, args=([i * ci, j * cj], ci, cj))
+                t.start()
+                ts.append(t)
+        for t in ts:
+            t.join()
+        del ts
+
+    def doCifs(self, N):
+        # 进入迭代
+        # N: 单点最大迭代次数
+        self.N = N

fractal/colors.py

@@ -0,0 +1,21 @@
+"""
+
+颜色的RGB值，列表索引从小到大颜色依次加深
+
+"""
+
+blues = [(255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (254, 254, 254), (254, 254, 254), (254, 254, 254), (254, 254, 254), (253, 253, 253), (253, 253, 253), (254, 255, 255), (254, 255, 255), (253, 254, 255), (249, 253, 255), (248, 252, 255), (246, 250, 253), (245, 249, 252), (245, 249, 252), (245, 250, 253), (245, 250, 253), (245, 250, 253), (243, 251, 253), (242, 250, 252), (242, 250, 252), (242, 250, 252), (242, 250, 252), (241, 249, 252), (241, 249, 252), (241, 249, 252), (240, 248, 251), (240, 248, 251), (239, 248, 253), (238, 247, 252), (238, 247, 252), (238, 247, 254), (237, 248, 254), (237, 248, 254), (237, 248, 254), (237, 248, 254), (236, 246, 255), (233, 246, 254), (233, 246, 254), (231, 246, 253), (230, 244, 253), (230, 244, 253), (227, 244, 252), (226, 243, 251), (226, 243, 251), (225, 242, 250), (225, 242, 252), (223, 241, 251), (223, 241, 251), (222, 240, 252), (221, 241, 252), (220, 240, 251), (219, 239, 250), (219, 239, 250), (219, 239, 250), (212, 239, 250), (212, 239, 250), (211, 237, 250), (210, 236, 249), (208, 236, 248), (207, 235, 249), (205, 234, 248), (205, 234, 250), (202, 234, 249), (201, 232, 250), (200, 233, 250), (199, 232, 251), (196, 231, 250), (195, 230, 250), (194, 229, 249), (194, 229, 249), (187, 230, 249), (187, 230, 249), (186, 229, 248), (183, 228, 249), (180, 226, 249), (178, 226, 248), (175, 225, 248), (174, 226, 250), (170, 224, 250), (167, 223, 248), (164, 221, 248), (161, 221, 249), (160, 221, 249), (160, 221, 250), (156, 220, 248), (154, 218, 246), (151, 218, 247), (150, 217, 246), (147, 216, 247), (144, 215, 245), (139, 214, 245), (136, 212, 244), (132, 212, 245), (128, 212, 246), (122, 211, 245), (118, 211, 245), (114, 211, 246), (111, 209, 246), (106, 208, 246), (103, 207, 246), (102, 206, 245), (99, 205, 244), (86, 205, 248), (82, 203, 246), (77, 200, 244), (72, 199, 244), (67, 198, 244), (61, 198, 244), (54, 196, 244), (47, 195, 243), (39, 195, 244), (32, 193, 245), (24, 191, 243), (17, 191, 242), (13, 190, 244), (9, 190, 245), (5, 189, 243), (2, 187, 243), (2, 185, 243), (1, 184, 242), (0, 183, 241), (0, 182, 242), (0, 180, 241), (0, 179, 240), (0, 178, 240), (0, 177, 239), (0, 175, 240), (0, 174, 239), (0, 173, 239), (0, 173, 239), (1, 172, 241), (0, 171, 240), (0, 171, 240), (0, 171, 240), (0, 169, 236), (1, 170, 237), (1, 170, 237), (0, 169, 234), (1, 169, 234), (0, 168, 233), (1, 166, 232), (0, 165, 231), (1, 164, 229), (1, 164, 229), (1, 163, 228), (0, 162, 226), (2, 161, 226), (1, 160, 225), (0, 159, 224), (0, 159, 224), (0, 159, 221), (0, 159, 221),
+         (0, 159, 221), (0, 159, 221), (0, 158, 220), (0, 157, 220), (0, 157, 220), (1, 156, 220), (0, 155, 219), (0, 154, 218), (0, 154, 218), (0, 152, 217), (0, 152, 217), (0, 151, 216), (1, 150, 216), (0, 149, 215), (2, 150, 214), (2, 150, 212), (1, 149, 211), (1, 149, 211), (0, 148, 210), (0, 147, 209), (1, 146, 209), (1, 147, 207), (1, 147, 207), (1, 145, 206), (1, 145, 205), (0, 144, 204), (0, 144, 204), (0, 143, 203), (0, 142, 202), (0, 142, 202), (1, 141, 202), (1, 141, 202), (0, 140, 201), (0, 140, 201), (0, 139, 198), (0, 138, 197), (0, 138, 197), (0, 138, 197), (1, 137, 195), (0, 136, 194), (0, 136, 194), (0, 136, 194), (1, 136, 192), (0, 135, 191), (0, 135, 191), (0, 135, 191), (0, 132, 189), (0, 132, 189), (0, 132, 189), (0, 131, 188), (0, 131, 188), (0, 131, 188), (0, 130, 188), (0, 129, 187), (0, 128, 186), (0, 128, 186), (0, 128, 186), (0, 128, 186), (0, 127, 185), (0, 127, 185), (0, 127, 185), (0, 127, 185), (0, 125, 181), (0, 125, 181), (0, 124, 180), (0, 124, 180), (0, 124, 180), (0, 123, 180), (0, 122, 179), (0, 122, 179), (0, 121, 178), (1, 122, 179), (1, 122, 179), (1, 122, 179), (1, 122, 179), (1, 120, 178), (0, 119, 177), (0, 119, 177), (0, 117, 171), (0, 117, 171), (0, 115, 170), (0, 115, 170), (0, 115, 170), (0, 114, 169), (0, 114, 169), (0, 113, 169), (0, 113, 169), (0, 113, 169), (0, 113, 169), (1, 112, 168), (1, 112, 168), (1, 112, 168), (0, 111, 167), (0, 111, 167), (0, 109, 165), (0, 109, 165), (0, 109, 165), (0, 109, 165), (0, 109, 165), (2, 109, 165), (2, 109, 165), (2, 109, 165), (0, 107, 163), (1, 106, 163), (1, 106, 163), (0, 105, 162), (0, 104, 161), (0, 104, 161), (0, 102, 160), (0, 102, 160), (0, 105, 160), (0, 105, 160), (0, 104, 159), (0, 103, 158), (0, 103, 158), (0, 102, 155), (0, 101, 155), (0, 100, 154), (0, 98, 153), (0, 97, 150), (0, 97, 150), (1, 96, 150), (1, 96, 150), (0, 95, 149), (0, 96, 147), (1, 95, 147), (1, 95, 149), (1, 95, 149), (0, 94, 148), (0, 94, 148), (0, 94, 148), (0, 94, 148), (1, 92, 145), (1, 92, 145), (0, 90, 143), (0, 90, 143), (1, 91, 143), (1, 91, 143), (1, 91, 143), (2, 92, 144), (2, 92, 144), (4, 91, 144), (0, 89, 141), (0, 89, 141), (1, 88, 141), (1, 88, 141), (0, 87, 140), (0, 87, 140), (0, 87, 140), (1, 86, 140), (0, 85, 139), (0, 85, 139), (0, 85, 139), (2, 85, 139), (2, 85, 139), (2, 85, 139), (2, 85, 139), (2, 85, 139), (0, 83, 133), (0, 83, 133), (0, 83, 133), (0, 82, 132), (0, 82, 132), (0, 82, 132), (0, 82, 132), (0, 81, 131), (0, 80, 131), (0, 80, 131), (0, 80, 131), (0, 80, 131), (0, 79, 130), (1, 78, 130), (1, 78, 130), (1, 78, 130), (0, 78, 127), (0, 76, 126), (0, 76, 126), (0, 76, 126), (0, 76, 126), (0, 75, 125), (0, 75, 125), (1, 74, 125), (1, 74, 125), (1, 74, 125), (2, 73, 125), (1, 72, 124), (1, 72, 124), (1, 72, 124), (0, 71, 123), (0, 71, 123), (0, 71, 121), (0, 71, 121), (0, 71, 121), (0, 71, 121), (0, 69, 120), (0, 69, 120), (0, 69, 120), (0, 69, 120), (0, 67, 119), (0, 67, 119), (0, 66, 118), (0, 66, 118), (0, 66, 118), (0, 65, 117), (0, 65, 117), (0, 65, 117), (1, 65, 113), (1, 65, 113), (1, 65, 113), (0, 64, 112), (0, 64, 112), (0, 64, 112), (0, 62, 111), (0, 62, 111), (1, 63, 112), (1, 63, 112), (1, 61, 111), (1, 61, 111), (0, 60, 110), (0, 59, 109), (0, 59, 109)]
+
+reds = [(255, 254, 255), (255, 254, 255), (255, 255, 255), (255, 255, 255), (255, 255, 255), (254, 255, 255), (254, 255, 255), (255, 255, 255), (254, 254, 254), (253, 251, 252), (252, 248, 247), (250, 244, 244), (249, 240, 241), (248, 238, 239), (248, 236, 238), (248, 236, 238), (255, 228, 227), (255, 225, 225), (253, 221, 222), (252, 218, 219), (252, 216, 216), (252, 212, 213), (250, 208, 209), (251, 205, 207), (253, 203, 204), (254, 199, 202), (254, 195, 197), (254, 192, 195), (254, 190, 191), (253, 187, 189), (252, 183, 186), (250, 181, 184), (253, 177, 177), (252, 174, 174), (250, 170, 171), (249, 167, 169), (250, 164, 165), (249, 160, 162), (248, 156, 157), (251, 152, 155), (251, 148, 151), (251, 145, 149), (251, 141, 144), (252, 137, 142), (252, 136, 139), (253, 132, 137), (252, 129, 134), (249, 126, 131), (253, 120, 125), (252, 119, 122), (251, 116, 120), (249, 113, 117), (249, 108, 114), (248, 105, 109), (248, 101, 107), (249, 98, 103), (251, 96, 102), (251, 92, 97), (251, 88, 93), (250, 85, 91), (251, 82, 89), (250, 79, 85), (246, 75, 81), (246, 73, 79), (252, 67, 75), (250, 63, 72), (247, 60, 69), (248, 57, 65), (249, 55, 64), (249, 52, 62), (248, 48, 58), (249, 45, 56), (248, 41, 51), (249, 38, 47), (249, 33, 44), (248, 31, 40), (249, 27, 38), (248, 24, 35), (246, 20, 32),
+        (244, 18, 29), (247, 18, 25), (247, 18, 25), (247, 16, 24), (247, 16, 24), (246, 15, 23), (249, 15, 24), (249, 15, 24), (249, 15, 24), (246, 15, 23), (247, 16, 24), (246, 17, 24), (244, 17, 24), (243, 18, 24), (241, 18, 23), (238, 17, 22), (238, 17, 22), (236, 17, 23), (235, 16, 22), (235, 16, 22), (232, 15, 23), (231, 16, 23), (229, 16, 22), (226, 15, 21), (225, 16, 21), (221, 14, 20), (220, 15, 20), (219, 16, 20), (216, 15, 21), (214, 15, 20), (214, 15, 20), (211, 15, 19), (211, 15, 19), (208, 15, 18), (208, 15, 18), (207, 14, 19), (205, 14, 19), (203, 14, 18), (200, 14, 19), (199, 14, 19), (198, 15, 20), (193, 12, 17), (192, 13, 19), (189, 12, 18), (188, 13, 20), (186, 13, 19), (185, 12, 18), (183, 12, 20), (183, 12, 20), (174, 13, 19), (174, 13, 19), (172, 13, 18), (171, 12, 17), (169, 13, 17), (168, 12, 16), (167, 12, 16), (166, 11, 15), (166, 11, 15), (167, 11, 15), (169, 10, 15), (170, 11, 16), (171, 10, 16), (173, 9, 16), (174, 10, 17), (175, 10, 17), (172, 11, 17), (172, 11, 17), (171, 12, 17), (171, 12, 17), (172, 13, 18), (173, 14, 18), (175, 14, 19), (176, 15, 20), (179, 16, 19), (180, 15, 19), (182, 16, 20), (185, 14, 20), (188, 15, 21), (190, 15, 20), (191, 14, 20), (191, 14, 20), (199, 14, 20), (200, 13, 20), (201, 14, 21), (204, 15, 22), (205, 14, 22), (208, 14, 23)]
+
+
+purples = [(235, 233, 236), (235, 233, 236), (235, 233, 236), (236, 231, 235), (235, 230, 236), (235, 230, 236), (234, 229, 236), (234, 229, 236), (237, 229, 240), (236, 228, 239), (235, 227, 240), (236, 225, 239), (235, 224, 240), (234, 223, 239), (234, 223, 239), (233, 222, 238), (234, 221, 239), (233, 220, 238), (232, 219, 237), (232, 219, 237), (230, 217, 237), (230, 217, 237), (230, 215, 238), (229, 214, 237), (228, 212, 238), (229, 211, 237), (228, 209, 237), (228, 209, 237), (226, 207, 237), (226, 207, 237), (225, 206, 236), (225, 204, 235), (225, 200, 239), (225, 200, 239), (224, 199, 238), (223, 198, 237), (223, 196, 237), (222, 195, 236), (223, 194, 238), (222, 193, 237), (222, 190, 237), (222, 190, 237), (222, 188, 238), (221, 187, 237), (221, 184, 236), (220, 183, 235), (219, 182, 234), (218, 181, 233), (218, 179, 236), (218, 179, 236), (217, 177, 237), (216, 176, 236), (215, 173, 236), (214, 172, 235), (215, 170, 235), (215, 170, 237), (214, 167, 237), (213, 166, 236), (214, 164, 237), (213, 162, 237), (213, 160, 238), (211, 158, 236), (210, 157, 235), (210, 157, 235), (210, 155, 238), (209, 154, 237), (210, 152, 237), (209, 151, 236), (208, 150, 237), (207, 147, 235), (206, 146, 236), (208, 144, 238), (206, 142, 238), (206, 140, 238), (205, 139, 237), (205, 137, 238), (204, 136, 239), (205, 134, 238), (204, 133, 237), (203, 132, 238), (202, 128, 239), (202, 128, 239), (203, 126, 238), (201, 124, 238), (200, 123, 237),
+           (200, 121, 238), (199, 120, 239), (199, 118, 238), (197, 116, 237), (198, 114, 238), (197, 112, 239), (197, 111, 238), (195, 108, 238), (196, 106, 238), (195, 105, 237), (195, 105, 237), (192, 102, 236), (191, 101, 235), (191, 99, 236), (190, 98, 235), (189, 96, 236), (189, 95, 235), (188, 93, 235), (189, 91, 236), (187, 89, 236), (188, 88, 238), (187, 87, 237), (187, 84, 237), (186, 82, 237), (186, 81, 236), (185, 80, 235), (184, 78, 236), (182, 76, 236), (183, 75, 236), (182, 74, 237), (181, 73, 236), (179, 71, 235), (180, 69, 234), (179, 68, 235), (180, 67, 237), (181, 65, 238), (181, 64, 239), (181, 62, 238), (180, 61, 239), (178, 59, 239), (178, 57, 238), (177, 56, 237), (176, 55, 238), (176, 52, 238), (175, 51, 237), (175, 50, 238), (174, 49, 237), (174, 48, 236), (173, 46, 237), (174, 45, 236), (174, 44, 238), (173, 41, 236), (172, 40, 237), (174, 39, 238), (173, 38, 237), (173, 36, 236), (172, 35, 237), (171, 34, 236), (171, 34, 236), (169, 31, 238), (168, 30, 237), (169, 29, 239), (168, 28, 238), (167, 27, 237), (166, 26, 237), (165, 25, 236), (167, 24, 238), (166, 23, 237), (165, 22, 238), (164, 21, 237), (165, 19, 239), (163, 17, 237), (163, 17, 238), (162, 16, 237), (161, 15, 236), (163, 12, 239), (162, 11, 238), (162, 11, 238), (161, 10, 237), (161, 10, 237), (160, 9, 238), (160, 7, 237), (160, 7, 237), (158, 5, 235), (160, 4, 236), (160, 4, 236), (159, 3, 235), (158, 2, 234), (158, 2, 234), (158, 0, 235)]
+
+yellows = [(250, 248, 249), (249, 247, 248), (249, 247, 248), (249, 247, 248), (249, 248, 246), (248, 247, 245), (248, 247, 243), (248, 247, 243), (248, 247, 242), (248, 247, 242), (248, 248, 240), (248, 248, 240), (248, 248, 238), (248, 248, 238), (248, 248, 238), (248, 248, 238), (250, 248, 235), (250, 248, 235), (250, 248, 233), (250, 248, 233), (250, 249, 231), (250, 249, 231), (250, 249, 229), (250, 249, 228), (248, 248, 224), (248, 248, 222), (248, 248, 222), (248, 248, 220), (248, 249, 218), (248, 249, 218), (248, 249, 218), (248, 249, 217), (252, 249, 214), (252, 249, 214), (252, 249, 214), (251, 249, 211), (251, 249, 211), (251, 249, 210), (250, 248, 207), (250, 248, 207), (250, 249, 205), (250, 249, 203), (250, 249, 201), (250, 250, 200), (249, 249, 199), (249, 249, 197), (249, 249, 197), (249, 249, 197), (250, 249, 193), (250, 249, 192), (250, 249, 192), (250, 250, 190), (249, 249, 187), (249, 249, 187), (249, 249, 185), (249, 250, 182), (249, 250, 180), (249, 251, 178), (248, 250, 175), (248, 251, 174), (248, 251, 174), (247, 250, 171), (247, 250, 169), (247, 250, 169), (248, 250, 167), (250, 249, 166), (250, 249, 166), (249, 249, 163), (249, 249, 161), (249, 249, 161), (248, 248, 158), (249, 248, 155), (249, 249, 153), (249, 249, 151), (251, 249, 149), (250, 248, 147), (250, 248, 147), (250, 248, 145), (250, 249, 143), (249, 248, 142), (247, 249, 139), (247, 250, 137), (247, 250, 137), (249, 249, 135), (248, 249, 132), (248, 249, 132), (248, 249, 130),
+           (248, 250, 127), (250, 250, 126), (249, 250, 123), (249, 250, 122), (250, 250, 120), (250, 250, 120), (249, 249, 117), (249, 249, 115), (249, 249, 115), (249, 250, 112), (249, 250, 110), (248, 249, 109), (248, 250, 107), (248, 250, 105), (249, 248, 104), (249, 249, 103), (249, 249, 99), (248, 249, 96), (249, 248, 95), (249, 249, 93), (248, 248, 90), (248, 248, 90), (248, 248, 88), (250, 248, 86), (249, 247, 85), (249, 250, 84), (251, 250, 84), (251, 250, 84), (251, 250, 82), (250, 249, 79), (250, 250, 78), (250, 250, 76), (250, 249, 73), (249, 248, 70), (249, 248, 69), (250, 248, 67), (250, 248, 65), (249, 248, 62), (249, 248, 62), (249, 248, 60), (249, 248, 60), (251, 248, 57), (251, 248, 57), (250, 247, 54), (250, 247, 54), (250, 248, 52), (249, 247, 51), (249, 247, 50), (249, 247, 48), (250, 249, 47), (250, 249, 45), (249, 248, 44), (249, 248, 42), (249, 249, 41), (249, 249, 41), (248, 248, 40), (248, 248, 38), (250, 249, 35), (250, 249, 35), (250, 249, 35), (249, 248, 33), (249, 248, 33), (249, 249, 31), (248, 248, 28), (248, 248, 28), (248, 248, 26), (248, 249, 25), (248, 249, 23), (248, 249, 21), (248, 249, 21), (248, 250, 19), (248, 250, 19), (248, 250, 19), (249, 249, 15), (249, 249, 15), (249, 249, 15), (249, 249, 15), (249, 249, 13), (249, 249, 13), (249, 249, 11), (249, 249, 11), (249, 250, 9), (249, 250, 9), (249, 250, 8), (249, 250, 8), (248, 249, 5), (248, 249, 5), (248, 249, 5), (248, 249, 5), (248, 248, 0), (248, 248, 0), (248, 248, 0), (248, 248, 0)]
+
+rainbow = [(118, 16, 11), (120, 17, 12), (125, 18, 10), (131, 20, 11), (140, 23, 13), (149, 26, 11), (158, 28, 12), (166, 29, 10), (175, 34, 7), (184, 36, 6), (193, 41, 4), (200, 44, 3), (206, 48, 3), (212, 52, 4), (218, 57, 5), (222, 61, 7), (234, 65, 0), (237, 68, 0), (243, 72, 2), (246, 74, 2), (248, 76, 2), (250, 77, 1), (252, 79, 2), (255, 81, 0), (255, 81, 0), (255, 83, 0), (255, 84, 0), (255, 86, 1), (255, 86, 1), (255, 87, 2), (255, 89, 3), (255, 89, 3), (253, 90, 0), (254, 91, 0), (254, 91, 0), (255, 92, 1), (255, 92, 1), (254, 94, 0), (254, 94, 0), (254, 94, 0), (254, 94, 0), (253, 95, 0), (254, 96, 0), (254, 96, 0), (255, 97, 0), (255, 98, 1), (255, 100, 0), (255, 100, 0), (255, 101, 3), (254, 102, 1), (255, 104, 3), (255, 106, 3), (255, 109, 5), (255, 111, 4), (255, 114, 4), (255, 115, 2), (254, 118, 0), (255, 121, 0), (254, 124, 0), (254, 126, 0), (252, 127, 0), (253, 128, 0), (254, 131, 1), (255, 134, 1), (252, 136, 0), (254, 138, 1), (254, 141, 1), (255, 144, 2), (254, 145, 2), (254, 148, 0), (253, 152, 0), (254, 155, 0), (253, 159, 1), (254, 162, 1), (253, 166, 1), (253, 168, 0), (252, 170, 0), (253, 173, 0), (255, 175, 0), (255, 178, 2), (251, 186, 0), (252, 187, 1), (252, 189, 0), (255, 192, 3), (254, 195, 3), (255, 198, 3), (255, 201, 4), (253, 205, 1), (255, 209, 2), (253, 211, 2), (253, 213, 1), (253, 217, 1), (253, 220, 1), (255, 223, 1), (254, 224, 2), (255, 226, 2), (252, 230, 0), (252, 230, 0), (252, 231, 0), (251, 233, 1), (252, 236, 1), (251, 239, 3), (248, 241, 2), (245, 243, 2), (244, 245, 3), (241, 247, 1), (239, 248, 1), (238, 250, 2), (236, 251, 0), (236, 253, 1), (234, 253, 1), (233, 254, 1), (228, 255, 0), (225, 255, 0), (224, 254, 0), (223, 254, 0), (222, 255, 0), (219, 254, 1), (215, 255, 0), (213, 254, 0), (210, 255, 0), (208, 255, 1), (204, 255, 0), (202, 255, 1), (199, 255, 0), (198, 255, 0), (195, 255, 1), (194, 254, 0), (186, 255, 0), (183, 254, 0), (177, 254, 0), (169, 254, 2), (164, 255, 4), (156, 255, 6), (150, 254, 7), (146, 254, 8), (137, 251, 6), (129, 250, 7), (120, 249, 9), (109, 247, 11), (97, 247, 15), (87, 247, 17), (78, 245, 17), (72, 245, 16), (53, 245, 22), (49, 243, 24), (43, 240, 26), (37, 239, 31), (30, 239, 36), (23, 237, 41), (15, 236, 43), (9, 235, 47),
+           (5, 238, 51), (3, 238, 57), (2, 239, 63), (2, 239, 71), (2, 238, 78), (3, 237, 86), (5, 237, 93), (5, 236, 96), (2, 239, 109), (2, 238, 112), (3, 238, 120), (4, 237, 128), (6, 238, 139), (8, 238, 148), (6, 238, 154), (6, 237, 159), (5, 235, 162), (5, 237, 164), (5, 235, 170), (6, 235, 176), (6, 234, 181), (7, 234, 188), (7, 233, 193), (5, 234, 195), (4, 231, 204), (6, 230, 204), (6, 230, 206), (6, 229, 208), (6, 229, 211), (6, 228, 215), (6, 227, 218), (7, 226, 222), (7, 225, 226), (7, 224, 229), (8, 222, 232), (7, 220, 234), (5, 218, 236), (3, 215, 236), (3, 215, 237), (3, 215, 237), (6, 209, 239), (6, 207, 237), (4, 204, 237), (3, 201, 236), (4, 200, 238), (5, 196, 238), (5, 193, 238), (5, 189, 239), (6, 186, 239), (7, 183, 240), (8, 180, 242), (7, 176, 241), (6, 173, 241), (6, 171, 239), (5, 167, 239), (4, 166, 238), (5, 161, 238), (4, 160, 237), (5, 158, 236), (4, 155, 236), (3, 152, 236), (3, 149, 235), (5, 146, 236), (5, 144, 237), (6, 140, 237), (6, 138, 239), (8, 135, 240), (7, 132, 240), (7, 129, 240), (6, 126, 238), (5, 125, 238), (6, 123, 237), (5, 118, 238), (5, 115, 236), (3, 113, 236), (2, 110, 236), (4, 109, 237), (4, 105, 239), (4, 102, 239), (5, 98, 238), (3, 93, 239), (5, 91, 240), (7, 88, 240), (5, 84, 237), (4, 81, 237), (3, 77, 234), (3, 75, 235), (2, 74, 234), (8, 66, 238), (7, 64, 239), (6, 63, 238), (5, 61, 236), (3, 58, 236), (3, 55, 235), (4, 52, 236), (5, 50, 237), (6, 47, 237), (9, 43, 238), (10, 39, 237), (10, 36, 235), (12, 33, 236), (15, 31, 238), (16, 31, 238), (18, 30, 238), (23, 24, 238), (24, 23, 238), (26, 23, 238), (27, 20, 237), (32, 18, 237), (35, 15, 236), (38, 13, 236), (42, 11, 236), (44, 7, 234), (48, 7, 235), (51, 5, 235), (53, 6, 236), (54, 5, 236), (57, 5, 237), (57, 5, 237), (58, 4, 237), (63, 4, 236), (65, 6, 238), (68, 6, 239), (69, 5, 239), (71, 5, 237), (73, 3, 237), (77, 5, 237), (81, 5, 238), (82, 4, 237), (85, 3, 237), (87, 3, 236), (91, 3, 237), (94, 5, 237), (98, 6, 239), (99, 5, 239), (99, 5, 239), (106, 4, 238), (106, 4, 236), (108, 4, 237), (111, 5, 238), (113, 5, 239), (116, 4, 238), (118, 4, 239), (121, 3, 237), (123, 4, 236), (128, 4, 238), (131, 5, 237), (134, 4, 238), (136, 4, 238), (137, 3, 237), (141, 5, 239), (143, 7, 241), (143, 6, 236), (143, 6, 236)]

fractal/ifs.py

@@ -0,0 +1,98 @@
+"""
+    迭代函数系统（IFS）
+"""
+from random import random
+from math import sin, cos
+from .base import Base, HAS_NUMPY
+
+
+def getIfsCode(matrix):
+    # 极坐标形式变换到矩阵
+    res = []
+    for i in matrix:
+        r, s, the, fu, e, f, p = i
+        item = [r * cos(the), -s * sin(fu), r * sin(the), s * cos(fu), e, f, p]
+        res.append(item)
+    return res
+
+
+class IFS(Base):
+
+    def __init__(self, size, title="", color=None):
+        Base.__init__(self, size, self._run, title)
+        self.setPx()
+        self.setIfsCode()
+        self.__coo = True
+        if color:
+            self.color = color
+        else:
+            self.color = [255, 255, 255]
+        self.fill(self.color)
+
+    def setCoordinate(self):
+        # 纵轴的反向
+        self.__coo = not self.__coo
+
+    def ifsp(self, x, y):
+        # 变换规则(返回迭代后的坐标)
+        return (x, y)
+
+    def setIfsp(self, call):
+        self.ifsp = call
+
+    def setIfsCode(self, ifsCode=None):
+        # 设置ifs码
+        self.ifsCode = ifsCode
+        if ifsCode != None:
+            p = [i[-1] for i in self.ifsCode]
+            self.__fn = len(p)
+            sp = sum(p)
+            self.__pe = [sum(p[:i + 1])/sp for i in range(self.__fn)]
+
+    def __parseIfsCode(self, x, y):
+        rand = random()
+        i = 0
+        while i < self.__fn:
+            if rand <= self.__pe[i]:
+                a, b, c, d, e, f, p = self.ifsCode[i]
+                return (a * x + b * y + e, c * x + d * y + f)
+            i += 1
+        return (0, 0)
+
+    def setPx(self, enlarge=1, pl=0, pt=0):
+        # 平移、放大调整
+        self.enlarge = enlarge
+        self.pl = pl
+        self.pt = pt
+
+    def _run(self):
+        start = self.start
+        for i in range(self.n):
+            px = int(self.enlarge * start[0] + self.pl)
+            py = int(self.enlarge * start[1] + self.pt)
+            if not self.__coo:
+                py = self.height - py
+            if HAS_NUMPY and self._pixel_array is not None:
+                if 0 <= px < self.width and 0 <= py < self.height:
+                    self._pixel_array[py, px] = [int(self.pcolor[0]), int(self.pcolor[1]), int(self.pcolor[2])]
+            else:
+                self.set_pixel_fast(px, py, self.pcolor[0], self.pcolor[1], self.pcolor[2])
+            start = self.ifsp(*start)
+
+    def doIFS(self, n, start=None, color=None):
+        """
+            开始迭代
+            start: 迭代起点
+            color: 描点的颜色
+        """
+        self.n = n
+        if start == None:
+            self.start = (0, 0)
+        else:
+            self.start = start
+        if color == None:
+            self.pcolor = [0, 0, 0]
+        else:
+            self.pcolor = color
+        if self.ifsCode != None:
+            self.ifsp = self.__parseIfsCode

fractal/julia.py

@@ -0,0 +1,106 @@
+"""
+Julia集
+"""
+
+from .base import Base, HAS_NUMPY
+from .colors import *
+from threading import Thread
+
+
+class Julia(Base):
+
+    def __init__(self, size, title=""):
+        Base.__init__(self, size, self._run, title)
+        self.setExp(2)
+        self.setC(None)
+        self.setRadius(2)
+        self.width = size[0]
+        self.height = size[1]
+        self.setRange(3.5, 3.5)
+        self.setCentre(0 + 0j)
+
+    def setRadius(self, R):
+        # 设置逃逸半径
+        self.R = R
+
+    def setC(self, C):
+        # 设置参考值C
+        self.C = C
+
+    def setExp(self, expc):
+        # 设置指数，默认2
+        self.expc = expc
+
+    def color(self, n, r=2):
+        if n < len(reds):
+            return reds[n]
+        else:
+            if r < self.R:
+                return blues[int((len(blues) - 1) * r / self.R)]
+            else:
+                return purples[int((len(purples) - 1) * self.R / r)]
+
+    def setColor(self, call):
+        self.color = call
+
+    def setCentre(self, z0):
+        # 设置中心点
+        self.z0 = z0
+
+    def setRange(self, xmax, ymax):
+        # 设置坐标范围，范围越小图放大倍数越高
+        self.xmax = xmax
+        self.ymax = ymax
+
+    def __getXY(self, i, j):
+        # 通过像素坐标获取映射后的坐标
+        return complex((i / self.width - 0.5) * self.xmax + self.z0.real,
+                       (j / self.height - 0.5) * self.ymax + self.z0.imag)
+
+    def scala(self, i, j, rate):
+        # 将(i, j)像素点置于中心位置，放大rete倍
+        self.setCentre(self.__getXY(i, j))
+        self.xmax /= rate
+        self.ymax /= rate
+
+    def __calc(self, start, w, h):
+        """计算子区域像素"""
+        for i in range(start[0], start[0] + w):
+            for j in range(start[1], start[1] + h):
+                ct = 0
+                z = self.__getXY(i, j)
+                for k in range(self.N):
+                    ct = k
+                    if abs(z) > self.R:
+                        break
+                    z = z**self.expc + self.C
+                col = self.color(ct, abs(z))
+                # 直接写入numpy数组
+                if HAS_NUMPY and self._pixel_array is not None:
+                    self._pixel_array[j, i] = [int(col[0]), int(col[1]), int(col[2])]
+                else:
+                    self.set_pixel_fast(i, j, col[0], col[1], col[2])
+
+    def _run(self):
+        # 线程中
+        print("x range ：[-%.2e,%.2e]\ny range ：[-%.2e,%.2e]" % (
+            self.xmax, self.xmax, self.ymax, self.ymax))
+        if self.C == None:
+            raise Exception("请设置迭代常数")
+        tn = 5  # 25 个子线程绘图
+        ci = self.width // tn
+        cj = self.height // tn
+        ts = []
+        for i in range(tn):
+            for j in range(tn):
+                t = Thread(target=self.__calc, args=([i * ci, j * cj], ci, cj))
+                t.start()
+                ts.append(t)
+        for t in ts:
+            t.join()
+        del ts
+
+    def doJulia(self, N):
+        # 进入迭代
+        # N: 单点最大迭代次数
+        self.N = N

fractal/lsystem.py

@@ -0,0 +1,117 @@
+"""
+    L-System for fractal
+    已完成功能：
+        1.简单D0L-系统
+        2.合成D0L-系统
+        3.简单分叉结构
+    待完成功能：
+        1.带年龄树（分叉）的生成
+        2.随机L-系统
+        3.参数L-系统
+"""
+
+from math import sin, cos, pi
+from .base import Base
+
+
+def left(screen, st, angle, d):
+    # st: 起点坐标
+    # angle: 向左偏转的度数
+    # d: 距离
+    angle = pi * angle / 180
+    return [st[0] + d * cos(angle), st[1] - d * sin(angle)]
+
+
+class Pen(Base):
+
+    def __init__(self, size, title=""):
+        # size 画布的宽高 [width, hight]
+        # title 画布标题
+        Base.__init__(self, size, self._run, title)
+        self.setPoint([size[0] / 2, size[1] / 2])
+        self.setColor([0, 0, 0])
+        self.setWidth(1)
+        self.setAngle(0)
+
+    def setAngle(self, angle):
+        self.angle = angle
+
+    def setPoint(self, pos):
+        # 设置笔的位置
+        self.pos = pos
+
+    def setWidth(self, width):
+        # 设置线宽
+        self.width = width
+
+    def setColor(self, color):
+        # 设置颜色
+        self.color = color
+
+    def left(self, angle):
+        # 向左转angle度
+        self.angle = self.angle + angle
+
+    def right(self, angle):
+        # 向右转angle度
+        self.angle = self.angle - angle
+
+    def forward(self, d):
+        # 向前走d步长
+        np = left(self.canvas, self.pos, self.angle, d)
+        self.draw_line(self.pos, np, self.color, self.width)
+        self.pos = np
+
+    def _run(self):
+        # 线程
+        self.draw(self.omega, self.P, self.delta, self.length)
+
+    def draw(self, omega, P, delta, length):
+        i = 0
+        while i < len(omega):
+            if omega[i] == '+':
+                self.left(delta)
+            elif omega[i] == '-':
+                self.right(delta)
+            elif omega[i] == '[':
+                k = 0
+                st = i
+                while i < len(omega):
+                    if omega[i] == "[":
+                        k += 1
+                    elif omega[i] == "]":
+                        k -= 1
+                    if k == 0:
+                        break
+                    i += 1
+                sub = omega[st + 1:i]
+                curpoint = self.pos[:]
+                curangle = self.angle
+                self.draw(sub, P, delta, length)
+                self.pos = curpoint
+                self.angle = curangle
+            else:
+                self.forward(length)
+            i += 1
+
+    def doD0L(self, omega, P, delta, times, length, rate):
+        # omega: 公理（初始字符串）
+        # P: 产生式（映射规则）
+        # delta: 角度增量
+        # times: 迭代次数
+        # length: 初始线长
+        # rate: 每次迭代后缩小的倍数
+        length /= (rate**times)
+        for i in range(times):  # 完成字符串迭代
+            ct = 0
+            for key in P:
+                omega = omega.replace(key, str(ct))
+                ct += 1
+            ct = 0
+            for key in P:
+                omega = omega.replace(str(ct), P[key])
+                ct += 1
+        self.omega = omega
+        self.P = P
+        self.delta = delta
+        self.length = length

fractal/mandelbrot.py

@@ -0,0 +1,105 @@
+"""
+Mandelbrot集
+"""
+
+from .base import Base, HAS_NUMPY
+from .colors import *
+from threading import Thread
+
+
+class Mandelbrot(Base):
+
+    def __init__(self, size, title=""):
+        Base.__init__(self, size, self._run, title)
+        self.setExp(2)
+        self.setRadius(2)
+        self.setZ0(0 + 0j)
+        self.width = size[0]
+        self.height = size[1]
+        self.setRange(3.5, 3.5)
+        self.setCentre(0 + 0j)
+
+    def setRadius(self, R):
+        # 设置逃逸半径
+        self.R = R
+
+    def setZ0(self, Z0):
+        # 设置起始迭代复数（一般为0+0j）
+        self.Z0 = Z0
+
+    def setCentre(self, z0):
+        # 设置中心点
+        self.z0 = z0
+
+    def setRange(self, xmax, ymax):
+        # 设置坐标范围，范围越小图放大倍数越高
+        self.xmax = xmax
+        self.ymax = ymax
+
+    def __getXY(self, i, j):
+        # 通过像素坐标获取映射后的坐标
+        return complex((i / self.width - 0.5) * self.xmax + self.z0.real,
+                       (j / self.height - 0.5) * self.ymax + self.z0.imag)
+
+    def scala(self, i, j, rate):
+        # 将(i, j)像素点置于中心位置，放大rete倍
+        self.setCentre(self.__getXY(i, j))
+        self.xmax /= rate
+        self.ymax /= rate
+
+    def setExp(self, expc):
+        # 设置指数，默认2
+        self.expc = expc
+
+    def color(self, n, r=2):
+        if n < len(reds):
+            return reds[n]
+        else:
+            if r < self.R:
+                return blues[int((len(blues) - 1) * r / self.R)]
+            else:
+                return purples[int((len(purples) - 1) * self.R / r)]
+
+    def setColor(self, call):
+        self.color = call
+
+    def __calc(self, start, w, h):
+        """计算子区域像素"""
+        for i in range(start[0], start[0] + w):
+            for j in range(start[1], start[1] + h):
+                ct = 0
+                z = self.Z0
+                c = self.__getXY(i, j)
+                for k in range(self.N):
+                    ct = k
+                    if abs(z) > self.R:
+                        break
+                    z = z**self.expc + c
+                col = self.color(ct, abs(z))
+                # 直接写入numpy数组
+                if HAS_NUMPY and self._pixel_array is not None:
+                    self._pixel_array[j, i] = [int(col[0]), int(col[1]), int(col[2])]
+                else:
+                    self.set_pixel_fast(i, j, col[0], col[1], col[2])
+
+    def _run(self):
+        # 绘图
+        print("x range ：[-%.2e,%.2e]\ny range ：[-%.2e,%.2e]" % (
+            self.xmax, self.xmax, self.ymax, self.ymax))
+        tn = 5  # 25 个子线程绘图
+        ci = self.width // tn
+        cj = self.height // tn
+        ts = []
+        for i in range(tn):
+            for j in range(tn):
+                t = Thread(target=self.__calc, args=([i * ci, j * cj], ci, cj))
+                t.start()
+                ts.append(t)
+        for t in ts:
+            t.join()
+        del ts
+
+    def doMandelbrot(self, N):
+        # 开始迭代
+        # N: 最大迭代次数
+        self.N = N

fractal-0.5.0.dist-info/METADATA

@@ -0,0 +1,291 @@
+Metadata-Version: 2.4
+Name: fractal
+Version: 0.5.0
+Summary: 分形绘图库，支持Julia集、Mandelbrot集、IFS、L-System等
+Home-page: https://github.com/pysrc/fractal
+Author: L.Chen
+Author-email: 1570184051@qq.com
+License: MIT
+Keywords: fractal,分形,Julia,Mandelbrot,IFS,L-System
+Platform: any
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Topic :: Multimedia :: Graphics
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: Pillow>=8.0.0
+Requires-Dist: numpy>=1.20.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: platform
+Dynamic: requires-dist
+Dynamic: summary
+
+# Draw Fractal Image By Python
+
+A Python fractal drawing library supporting Julia Set, Mandelbrot Set, IFS, L-System and more.
+
+## Install
+
+```bash
+pip install fractal
+```
+
+Or download the package:
+
+```bash
+pip install -e .
+```
+
+Dependencies:
+- Pillow >= 8.0.0
+- numpy >= 1.20.0
+
+## Features
+
+- **Julia Set** - Interactive zoom with mouse
+- **Mandelbrot Set** - Classic Mandelbrot fractal
+- **IFS** - Iterated Function System (fern, leaf, tree, etc.)
+- **L-System** - Lindenmayer System (Koch curve, dragon, tree, etc.)
+- **CIFS** - Complex Iterated Function System
+
+## GUI Operations
+
+- **Save**: Click "Save" button or Ctrl+P
+- **Save As**: Click "Save As..." to choose format (PNG/JPG/BMP)
+- **Zoom In**: Left click to show selection box, drag to move, click again to confirm
+- **Zoom Out**: Right click
+- **Cancel Selection**: Press ESC or right click
+
+## Examples
+
+### L-System
+
+```python
+from fractal import Pen
+from math import sqrt
+p = Pen([350, 270])
+p.setPoint([140, 60])
+p.setWidth(1)
+p.doD0L(omega="L", P={"L": "L+R", "R": "L-R"},
+              delta=90, times=15, length=200, rate=sqrt(2))
+p.wait()
+```
+
+![](fenxing_images/dragon.jpg)
+
+```python
+from fractal import Pen
+
+p = Pen([500, 500], title="Window")
+p.setPoint([495, 495])
+p.setAngle(90)
+p.doD0L(omega="f+f+f+f", P={"f": "ff+f--f+f"},
+        delta=90, times=5, length=490, rate=3)
+p.wait()
+```
+
+![](fenxing_images/Window.jpg)
+
+```python
+from fractal import Pen
+p = Pen([420,420])
+p.setPoint([10,10])
+p.doD0L(omega = "L", P = {"L": "LFRFL-FF-RFLFR+FF+LFRFL", "R": "RFLFR+FF+LFRFL-FF-RFLFR"}, delta =  90, times = 4, length = 200 , rate = 3)
+p.wait()
+```
+
+![](fenxing_images/fass2.jpg)
+
+```python
+from fractal import Pen
+p = Pen([400, 470])
+p.setAngle(90)
+p.setPoint([200,470])
+p.doD0L(omega = "f", P = {"f": "h[-f][+f]hf", "h": "hh"}, delta = 25.7, times = 7, length = 400, rate = 2.17)
+p.wait()
+```
+
+![](fenxing_images/tree4.jpg)
+
+```python
+from fractal import Pen
+p = Pen([400, 470])
+p.setAngle(90)
+p.setPoint([170, 470])
+p.doD0L(omega="f", P={"f": "h+[[f]-f]-h[-hf]+f", "h": "hh"},
+        delta=22.5, times=6, length=400, rate=2.3)
+p.wait()
+```
+
+![](fenxing_images/tree5.jpg)
+
+### IFS
+
+```python
+from fractal import IFS
+from random import random
+
+def ifsp(x, y):
+    p = random()
+    if p < 0.01:
+        return (0, 0.16 * y)
+    elif p < 0.07:
+        if random() > 0.5:
+            return (0.21 * x - 0.25 * y, 0.25 * x + 0.21 * y + 0.44)
+        else:
+            return (-0.2 * x + 0.26 * y, 0.23 * x + 0.22 * y + 0.6)
+    else:
+        return (0.85 * x + 0.1 * y, -0.05 * x + 0.85 * y + 0.6)
+
+ob = IFS([400, 500], title = "Leaf")
+ob.setPx(100, 100, 100)
+ob.setIfsp(ifsp)
+ob.doIFS(200000)
+ob.wait()
+```
+
+![](fenxing_images/ifern.jpg)
+
+```python
+# Box IFS
+from fractal import IFS
+from random import randint
+
+
+def ifsp(x, y):
+    p = randint(1, 5)
+    if p == 1:
+        return (x / 3, y / 3)
+    elif p == 2:
+        return (x / 3 + 2 / 3, y / 3)
+    elif p == 3:
+        return (x / 3 + 1 / 3, y / 3 + 1 / 3)
+    elif p == 4:
+        return (x / 3, y / 3 + 2 / 3)
+    else:
+        return (x / 3 + 2 / 3, y / 3 + 2 / 3)
+
+ob = IFS([500, 500], title="Box")
+ob.setPx(490, 5, 5)
+ob.setIfsp(ifsp)
+ob.doIFS(200000)
+ob.wait()
+```
+
+![](fenxing_images/ibox.jpg)
+
+```python
+from fractal import IFS
+
+ifscode = [
+    [0.879, 0.054, -0.051, 0.878, 0.077, 0.123, 0.123],
+    [0.1, -0.193, 0.285, 0.224, 0.174, 0.169, 0.169],
+    [0.008, 0.135, 0, 0.204, 0.075, 0.074, 0.074],
+    [0.402, 0.045, 0.016, -0.197, 0.111, 0.193, 0.193]
+]
+
+ifs = IFS([500, 500])
+ifs.setPx(700, 0, 0)
+ifs.setIfsCode(ifscode)
+ifs.doIFS(200000)
+ifs.wait()
+```
+
+![](fenxing_images/leaf.jpg)
+
+```python
+from fractal import IFS
+
+code = [
+    [0.195, -0.488, 0.344, 0.443, 0.4431, 0.2452, 0.2],
+    [0.462, 0.414, -0.252, 0.361, 0.2511, 0.5692, 0.2],
+    [-0.637, 0, 0, 0.501, 0.8562, 0.2512, 0.2],
+    [-0.035, 0.07, -0.469, 0.022, 0.4884, 0.5069, 0.2],
+    [-0.058, -0.07, -0.453, -0.111, 0.5976, 0.0969, 0.2]
+]
+
+ifs = IFS([500,500])
+ifs.setCoordinate()
+ifs.setPx(500, 0, 0)
+ifs.setIfsCode(code)
+ifs.doIFS(200000)
+ifs.wait()
+```
+
+![](fenxing_images/itree.jpg)
+
+
+
+### Julia
+
+```python
+from fractal import Julia
+ju = Julia([500, 500])
+ju.setC(0 - 1j)
+ju.doJulia(500)
+ju.wait()
+```
+
+![](fenxing_images/ju2.jpg)
+
+**-1.25 + 0j**
+
+![](fenxing_images/ju6.jpg)
+
+![](fenxing_images/ju62.jpg)
+
+
+
+**-0.605-0.45j**
+
+![](fenxing_images/(-0.605-0.45j)1.jpg)
+
+![](fenxing_images/(-0.605-0.45j)2.jpg)
+
+![](fenxing_images/(-0.605-0.45j)3.jpg)
+
+### Mandelbrot
+
+
+
+```python
+from fractal import Mandelbrot
+man = Mandelbrot([500, 500])
+man.setRange(5, 5)
+man.doMandelbrot(200)
+man.wait()
+```
+
+![](fenxing_images/mandelbrot.jpg)
+
+![](fenxing_images/mandelbrot2.jpg)
+
+![](fenxing_images/mandelbrot3.jpg)
+
+![](fenxing_images/mandelbrot4.jpg)
+
+## License
+
+MIT License
+
+## Author
+
+L.Chen (pysrc)

fractal-0.5.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+fractal/__init__.py,sha256=aT4xfLoejCQqWE1nRrudKEXIP5a0osreLrPTaBxALk0,170
+fractal/base.py,sha256=kqtzt01ZgmqSLIeeXJkSBsL6Z3vzTHfhJKOaeW4sGHU,15526
+fractal/cifs.py,sha256=-TxYVvmLnF3OU5BtFFA3FM-zG8x-PfU6vsSzYLCmSak,2996
+fractal/colors.py,sha256=3zZKljXc5x3s_Xf_nReC9xwu7kei-armjhnxwlYH1r8,19581
+fractal/ifs.py,sha256=pwBViUOTHYjaynbeqbmr0_fJcla7bgfJ7hEs6M1kcB0,2815
+fractal/julia.py,sha256=JReJl_KTtR5AbW1Z4Nmyk0LzDzAPM9yYBA9Lpwb65uE,3100
+fractal/lsystem.py,sha256=cgTatLlprLnV1AEShPqHLU_hTd2UDqtuTVKQLoGAKlk,3221
+fractal/mandelbrot.py,sha256=aAYXzOXZD8X9OAMH03yT831LWVhu2yjNukBPaY_DP7c,3083
+fractal-0.5.0.dist-info/licenses/LICENSE,sha256=FVQI1mfynzIuUYjS7TllMka382YWm-oFVFwCnQfLTWU,1084
+fractal-0.5.0.dist-info/METADATA,sha256=ecPS7iKVhHI_Q6Yx06ufnBwfJleMBK_IpQOkryOM074,6404
+fractal-0.5.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+fractal-0.5.0.dist-info/top_level.txt,sha256=rEXdmNzoRgT8nHa7k96n4y-bmnhe0nrJ0qDl-5mnjog,8
+fractal-0.5.0.dist-info/RECORD,,

fractal-0.5.0.dist-info/WHEEL

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

fractal-0.5.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2018 L.Chen
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

fractal-0.5.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+fractal