Я пытаюсь создать GIF для масштабирования в наборе Мандельброта в Python, но у меня возникли проблемы с поплавками - PullRequest
Новая
       39

Я пытаюсь создать GIF для масштабирования в наборе Мандельброта в Python, но у меня возникли проблемы с поплавками

1 голос
/ 21 апреля 2020 
from PIL import Image
import numpy as np
import math

class Fractal(object):

    def __init__(self, resolution=(720,720), x_range=(-2.0,2.0), y_range=(-2.0,2.0), maxIterations=255, zoom=0, zoomRate=0):
        self.resolution = resolution
        self.x_range = x_range
        self.y_range = y_range
        self.maxIterations = maxIterations
        self.zoom = zoom
        self.zoomRate = zoomRate

    def computePixel(self, x, y):
        z = complex(0.0,0.0)
        c = complex(x,y)
        iterations = 0
        while iterations < self.maxIterations or abs(z)<2:
            z = z*z + c
            iterations += 1
        # pixelHue = (360/self.maxIterations)*iterations
        # RGB = hsv_to_rgb(pixelHue, 0.0, 0.0)
        # return RGB
        return iterations

    def generateFrame(self):
        #x_max, x_min, y_max, y_min = self.zoom()
        x_max, x_min, y_max, y_min = (self.x_range[1],self.x_range[0],self.y_range[1],self.y_range[0])
        x_increment = (x_max - x_min)/self.resolution[0]
        print(x_increment)
        y_increment = (y_max - y_min)/self.resolution[1]
        print(y_increment)
        frame = []
        yPixels = []
        for x in np.arange(x_min, x_max, x_increment):
            for y in np.arange(y_min, y_max, y_increment):
                x,y = float(x), float(y)
                iterations = self.computePixel(x,y)
                print(x,y,iterations)
        #         pixel_RGB= self.computePixel(x,y)
        #         yPixels.append(pixel_RGB)
        #     frame.append(yPixels)
        # return frame


    def zoom(self):
        '''this is the change that will happen in one second'''
        x_max = self.x_range[1]-(self.zoomRate/2)
        x_min = self.x_range[0]+(self.zoomRate/2)
        y_max = self.y_range[1]-(self.zoomRate/2)
        y_min = self.y_range[0]+(self.zoomRate/2)
        return (x_max, x_min, y_max, y_min)

    def sweep(self):
        for x in np.arange(self.size[0]):
            for y in np.arange(self.size[1]):
                pass



def hsv_to_rgb(h, s, v):
    if s == 0.0: return (v, v, v)
    i = int(h*6.) # XXX assume int() truncates!
    f = (h*6.)-i; p,q,t = v*(1.-s), v*(1.-s*f), v*(1.-s*(1.-f)); i%=6
    if i == 0: return (v, t, p)
    if i == 1: return (q, v, p)
    if i == 2: return (p, v, t)
    if i == 3: return (p, q, v)
    if i == 4: return (t, p, v)
    if i == 5: return (v, p, q)

f = Fractal()
f.generateFrame()

В настоящее время я только пытаюсь отобразить первый кадр GIF, то есть набор Мандельброта без увеличения. Однако, когда я пытаюсь перебрать все значения пикселей, итерация останавливается на x=-1.8611111111111116 и y=-0.005555555555562641

Я предполагаю, что это из-за некоторой ошибки переполнения, возникающей, когда y должен быть равен нулю , но я не знаю, как решить эту проблему.

Вот последний снимок экрана командной строки, где я печатаю действительные и мнимые части всех комплексных чисел с количеством итераций, которое потребовалось для этого числа. сходиться (или нет, если на выходе 255).

enter image description here

1 Ответ

0 голосов
/ 21 апреля 2020

Вам необходимо изменить условие while l oop на: 

while iterations < self.maxIterations and abs(z)<2:

В противном случае, l oop работает в течение длительного времени, поскольку abs(z) меньше 2.

...