mandelbrot.py

import os
import cv2
import numpy as np
import time
import sys

LINUX = True if 'linux' in sys.platform else False
WINDOWS = True if 'win' in sys.platform else False
if (not (LINUX ^ WINDOWS)):
    raise "{} platform is not supported".format(sys.platform)
workDir = os.getcwd()

def getpng(n,c,name,variety=0,regenerate=False):
    if os.path.exists(os.path.join(workDir, name)) and not regenerate:
        # 已生成图像
        img = cv2.imread(name)
        cv2.imwrite(name,img)
        return img
    else:
        # 未生成图像
        if os.path.exists(os.path.join(workDir, 'Mandelbrot_python.out')) or os.path.exists(os.path.join(workDir, "Mandelbrot_python.exe")):
            # 已编译且存在可执行文件
            if WINDOWS:
                cmd = 'Mandelbrot_python {} {} {} {} {}'
            elif LINUX:
                cmd = './Mandelbrot_python.out {} {} {} "{}" {}'
            print(cmd.format(n, c.real, c.imag, name ,variety))
            print('Running...')
            os.system(cmd.format(n, c.real, c.imag, name ,variety))
            if not os.path.exists(os.path.join(workDir, name)):
                raise "Failed to generate the image\n生成图片失败"
            img = cv2.imread(name)
            cv2.imwrite(name,img)
            return img
        else:
            # 可执行文件未编译
            print('No executable file!\nCompiling...')
            if WINDOWS:
                cmd = 'gcc -lm Mandelbrot_python.c -o Mandelbrot_python.exe'
            elif LINUX:
                cmd = 'gcc Mandelbrot_python.c -o Mandelbrot_python.out -lm'
            print(cmd)
            os.system(cmd)
            if not (os.path.exists(os.path.join(workDir, 'Mandelbrot_python.out')) or os.path.exists(os.path.join(workDir, "Mandelbrot_python.exe"))):
                raise "Failed to compile\n编译失败"
            return getpng(n ,c, name, variety=variety, regenerate=regenerate)

if __name__=='__main__':
    from datetime import datetime

    try:
        os.mkdir('runs/')
    except:
        pass
    dirname='runs/'+datetime.now().strftime('%Y-%m-%d_%Hh%Mm%Ss')+'/'
    os.mkdir(dirname)

    n=0
    #c=-1.2228637871299002+0.31688226388713636j
    #c=-1.2603629819786142+0.3553264757232396j
    c=0+0j
    variety=0

    def enlarge(event,x,y,flags,param):
        global n,c
        print('\r'+str(n)+','+str(c+((x-127.5)+(127.5-y)*1j)/64/2**n),end='\t')
        if event==cv2.EVENT_LBUTTONDBLCLK:
            c+=((x-127.5)+(127.5-y)*1j)/64/2**n
            n+=1

    img=getpng(n,c,dirname+str(n)+'_'+str(c)+'.png',variety=variety)
    cv2.namedWindow('image')
    cv2.setMouseCallback('image',enlarge)

    while(True):
        img=getpng(n,c,dirname+str(n)+'_'+str(c)+'.png',variety=variety)
        cv2.imshow('image',img)
        if cv2.waitKey(100)&0xFF==27 or n>48:
            break
        #n+=1

    cv2.destroyAllWindows()