cvs/fract/mandelzoom.cgi

#!/usr/bin/python

import cgi, Image, ImageDraw, sys, math
from string import atoi, atof

# This is to get backtrace output
sys.stderr = sys.stdout

# Uncomment this to get the backtrace more readable
#print "Content-Type: text/plain"
#print ""

form= cgi.FieldStorage()

# Image size
if form.has_key('width'):
    width= atoi(form['width'].value)
else:
    width= 480
if form.has_key('height'):
    height= atoi(form['height'].value)
else:
    height= 480

xmax, ymax = width-1, height-1          # Coordinate maximums

# Complex number in center of image is (cx+cy*j)
if form.has_key('cx'):
    cx= atof(form['cx'].value)
else:
    cx= 0

if form.has_key('cy'):
    cy= atof(form['cy'].value)
else:
    cy= 0

if form.has_key('diag'):
    diag= atof(form['diag'].value)
else:
    diag= math.sqrt(32) # sqrt(4**2 + 4**2)

debug= form.has_key('debug')

if form.has_key('iter'):
    maxiter= atoi(form['iter'].value)
else:
    maxiter= 270

# If type!=image, then output an HTML page, not an image
if not (form.has_key('type') and form['type'].value == "image"):
    print """Content-Type: text/html

<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN">
<HTML><HEAD>
<TITLE>Mandel Set Zoomer</TITLE>
</HEAD>
<BODY>
<H1>Mandel Set Zoomer</H1>

<FORM ACTION="mandelzoom.cgi" METHOD=GET>

<!-- implies TYPE=SUBMIT -->"""
    print '<INPUT TYPE=IMAGE NAME="image"',
    print 'WIDTH="%s"' % (str(width)),
    print 'HEIGHT="%s"' % (str(height)),
    if debug:
        ds="&debug=on"
    else:
        ds=""
    if form.has_key('zoom'):
        zoom=atof(form['zoom'].value)
        diag=diag/zoom
    else:
        # If no zoom provided, don't actually zoom
        zoom= 2
    print 'SRC="mandelzoom.cgi?type=image&width=%s&height=%s&iter=%s%s&diag=%s&cx=%s&cy=%s"' % (str(width), str(height), str(maxiter), ds, str(diag), str(cx), str(cy)),
    print 'ALIGN=BOTTOM HEIGHT="%s"' % (str(height)),
    print 'WIDTH="%s"><P>' % (str(width))
    print 'Zoom: <SELECT NAME="zoom">'
    for zv in [1.0/5, 1.0/3, 1.0/2, 1.0/1.5, 1.0, 1.5, 2.0, 3.0, 5.0]:
        print '<OPTION',
        if str(zv)==str(zoom):
            print 'SELECTED',
        print 'VALUE="%s"' % (str(zv))
        if zv == 1:
            print '>Pan'
        elif zv < 1:
            if int(1/zv) == 1/zv:
                print '>Out �%s' % (str(int(1/zv)))
            else:
                print '>Out �%s' % (str(1/zv))
        else:
            print '>In �%s' % (str(zv))
    print '</SELECT>'
    print """Width: <INPUT TYPE=TEXT NAME="width" MAXLENGTH="4" SIZE="3"
    VALUE="%s">
    Height: <INPUT TYPE=TEXT NAME="height" MAXLENGTH="4" SIZE="3"
    VALUE="%s">
    Iterations: <INPUT TYPE=TEXT NAME="iter" MAXLENGTH="4" SIZE="3"
    VALUE="%s">
    Debug mode: <INPUT TYPE=CHECKBOX NAME="debug" """ % (str(width), str(height), str(maxiter)),
    if debug:
        print 'CHECKED>'
    else:
        print '>'
    print '<INPUT TYPE=SUBMIT VALUE="Apply">'
    for var in (("diag", diag), ("cx", cx), ("cy", cy)):
        print '<INPUT TYPE=HIDDEN NAME="%s" VALUE="%s">' % var
    print '<INPUT TYPE=HIDDEN NAME=type VALUE="html">'

    print """</FORM>
</BODY></HTML>"""
    sys.exit(0)

# Figure out c1 and c2 from width, height, diag, cx, and cy.
# Diagonal in pixels
diagp= math.sqrt(width**2 + height**2)
# Scale between pixels and coordinates
scale= diagp/diag
x= (width/2.0)/scale
y= (height/2.0)/scale

c1= cx - x + cy - y * (0+1j)
c2= cx + x + cy + y * (0+1j)

#print width, height, diag, cx, cy, diagp, scale, x, y, c1, c2
#sys.exit(0)

#c1, c2 = (-2+2j), (2-2j)                # Corner coordinates

# Force c1 to be upper left and c2 to be lower right
c1, c2= complex(min(c1.real, c2.real), max(c1.imag, c2.imag)), \
        complex(max(c1.real, c2.real), min(c1.imag, c2.imag))

mandel= Image.new("L", (width, height))
draw= ImageDraw.Draw(mandel)

def drawrect((x1, y1), (x2, y2)):
    "Draw a rectangle defined by the corners (x1, y1) and (x2, y2)."
    color= plot(x1, y1) # Plot the upper left corner
    # Should we cut this rectangle and recurse into the pieces?
    cut= 0
    ## Plot the edges and at the same time check if they are all the
    ## same color.
    # Upper edge (left to right)
    x, y= x1+1, y1
    while x<=x2:
        if color<>plot(x, y):
            cut= 1
        x= x+1
    # Right edge (top to bottom)
    x, y= x2, y1+1
    while y<=y2:
        if color<>plot(x, y):
            cut= 1
        y= y+1
    # Bottom edge (right to left)
    x, y= x2-1, y2
    while x>=x1:
        if color<>plot(x, y):
            cut= 1
        x= x-1
    # Left edge (bottom to top)
    x, y= x1, y2-1
    while y>y1:
        if color<>plot(x, y):
            cut= 1
        y= y-1
    # If our rectangle is too small to have an inside, we have now
    # plotted every pixel in it.
    if x2-x1<=1 or y2-y1<=1:
        return
    if cut:                             # If we're cutting the rectangle
        if x2-x1 > y2-y1:               # If it's wider than high
            # Cut vertically
            xc= ((x2-x1)/2)+x1
            drawrect((x1, y1), (xc, y2))
            drawrect((xc, y1), (x2, y2))
        else:                           # if it's higher than wide
            # Cut horizontally
            yc= ((y2-y1)/2)+y1
            drawrect((x1, y1), (x2, yc))
            drawrect((x1, yc), (x2, y2))
    else:
        # If we're not cutting, it was the same color along the edge
        if not debug:
            # Fill in the rectangle
            draw.rectangle([x1, y1, x2, y2], fill= color)
            if yorig >= 0: # The x axis is visible on image
                # Fill in the mirror rectangle
                # Mirror y1 and y2 over yorig, but keep within image
                y1m= max(0, min(2*yorig-y1, ymax))
                y2m= max(0, min(2*yorig-y2, ymax))
                draw.rectangle([x1, y1m, x2, y2m], fill= color)

def constant(x, y):
    creal= (float(x)/xmax)*(c2.real - c1.real)+c1.real
    cimag= (float(y)/ymax)*(c2.imag - c1.imag)+c1.imag
    return complex(creal, cimag)

def plot(x, y):
    c= constant(x, y)
    z= i= 0
    color= mandel.getpixel((x, y))
    if color<>0:
        return color
    while i<maxiter and abs(z)<2:
        z= z**2+c
        i= i+1
    color= 16*i%256
    if i>=maxiter or color==0:
        if debug:
            color= 255
        else:
            color= 1
    try:
        mandel.putpixel((x, y), color)
        if yorig >= 0: # The x axis is visible on image
            ym= 2*yorig-y   # ym is y mirrored over yorig
            if ym >= 0 and ym <= ymax:
                # ym is on the image
                mandel.putpixel((x, ym), color)
    except IndexError, the_error:
        print "coord", x, y, xm, ym
        raise IndexError, the_error
    return color

# Initial default values
yfrom= 0
yto= ymax
yorig= -1 # The pixel coordinate of the x axis; negative means not on image

# Check if we should do mirroring
if c1.imag > 0 and c2.imag < 0: # the x axis is visible
    yorig= int((-c1.imag/(c2.imag-c1.imag))*height) # pixel pos. of x axis
    if c1.imag < abs(c2.imag): # the x axis is closer to top than bottom
        yfrom= yorig # Begin at x axis
    else:
        yto= yorig # End at x axis

drawrect((0, yfrom), (xmax, yto))

print "Content-type: image/png"
print
mandel.save(sys.stdout, "PNG")
1	#!/usr/bin/python
2
3	import cgi, Image, ImageDraw, sys, math
4	from string import atoi, atof
5
6	# This is to get backtrace output
7	sys.stderr = sys.stdout
8
9	# Uncomment this to get the backtrace more readable
10	#print "Content-Type: text/plain"
11	#print ""
12
13	form= cgi.FieldStorage()
14
15	# Image size
16	if form.has_key('width'):
17	width= atoi(form['width'].value)
18	else:
19	width= 480
20	if form.has_key('height'):
21	height= atoi(form['height'].value)
22	else:
23	height= 480
24
25	xmax, ymax = width-1, height-1 # Coordinate maximums
26
27	# Complex number in center of image is (cx+cy*j)
28	if form.has_key('cx'):
29	cx= atof(form['cx'].value)
30	else:
31	cx= 0
32
33	if form.has_key('cy'):
34	cy= atof(form['cy'].value)
35	else:
36	cy= 0
37
38	if form.has_key('diag'):
39	diag= atof(form['diag'].value)
40	else:
41	diag= math.sqrt(32) # sqrt(42 + 42)
42
43	debug= form.has_key('debug')
44
45	if form.has_key('iter'):
46	maxiter= atoi(form['iter'].value)
47	else:
48	maxiter= 270
49
50	# If type!=image, then output an HTML page, not an image
51	if not (form.has_key('type') and form['type'].value == "image"):
52	print """Content-Type: text/html
53
54	<!DOCTYPE HTML PUBLIC "-//IETF//DTD HTML 2.0//EN">
55	<HTML><HEAD>
56	<TITLE>Mandel Set Zoomer</TITLE>
57	</HEAD>
58	<BODY>
59	<H1>Mandel Set Zoomer</H1>
60
61	<FORM ACTION="mandelzoom.cgi" METHOD=GET>
62
63	<!-- implies TYPE=SUBMIT -->"""
64	print '<INPUT TYPE=IMAGE NAME="image"',
65	print 'WIDTH="%s"' % (str(width)),
66	print 'HEIGHT="%s"' % (str(height)),
67	if debug:
68	ds="&debug=on"
69	else:
70	ds=""
71	if form.has_key('zoom'):
72	zoom=atof(form['zoom'].value)
73	diag=diag/zoom
74	else:
75	# If no zoom provided, don't actually zoom
76	zoom= 2
77	print 'SRC="mandelzoom.cgi?type=image&width=%s&height=%s&iter=%s%s&diag=%s&cx=%s&cy=%s"' % (str(width), str(height), str(maxiter), ds, str(diag), str(cx), str(cy)),
78	print 'ALIGN=BOTTOM HEIGHT="%s"' % (str(height)),
79	print 'WIDTH="%s"><P>' % (str(width))
80	print 'Zoom: <SELECT NAME="zoom">'
81	for zv in [1.0/5, 1.0/3, 1.0/2, 1.0/1.5, 1.0, 1.5, 2.0, 3.0, 5.0]:
82	print '<OPTION',
83	if str(zv)==str(zoom):
84	print 'SELECTED',
85	print 'VALUE="%s"' % (str(zv))
86	if zv == 1:
87	print '>Pan'
88	elif zv < 1:
89	if int(1/zv) == 1/zv:
90	print '>Out �%s' % (str(int(1/zv)))
91	else:
92	print '>Out �%s' % (str(1/zv))
93	else:
94	print '>In �%s' % (str(zv))
95	print '</SELECT>'
96	print """Width: <INPUT TYPE=TEXT NAME="width" MAXLENGTH="4" SIZE="3"
97	VALUE="%s">
98	Height: <INPUT TYPE=TEXT NAME="height" MAXLENGTH="4" SIZE="3"
99	VALUE="%s">
100	Iterations: <INPUT TYPE=TEXT NAME="iter" MAXLENGTH="4" SIZE="3"
101	VALUE="%s">
102	Debug mode: <INPUT TYPE=CHECKBOX NAME="debug" """ % (str(width), str(height), str(maxiter)),
103	if debug:
104	print 'CHECKED>'
105	else:
106	print '>'
107	print '<INPUT TYPE=SUBMIT VALUE="Apply">'
108	for var in (("diag", diag), ("cx", cx), ("cy", cy)):
109	print '<INPUT TYPE=HIDDEN NAME="%s" VALUE="%s">' % var
110	print '<INPUT TYPE=HIDDEN NAME=type VALUE="html">'
111
112	print """</FORM>
113	</BODY></HTML>"""
114	sys.exit(0)
115
116	# Figure out c1 and c2 from width, height, diag, cx, and cy.
117	# Diagonal in pixels
118	diagp= math.sqrt(width2 + height2)
119	# Scale between pixels and coordinates
120	scale= diagp/diag
121	x= (width/2.0)/scale
122	y= (height/2.0)/scale
123
124	c1= cx - x + cy - y * (0+1j)
125	c2= cx + x + cy + y * (0+1j)
126
127	#print width, height, diag, cx, cy, diagp, scale, x, y, c1, c2
128	#sys.exit(0)
129
130	#c1, c2 = (-2+2j), (2-2j) # Corner coordinates
131
132	# Force c1 to be upper left and c2 to be lower right
133	c1, c2= complex(min(c1.real, c2.real), max(c1.imag, c2.imag)), \
134	complex(max(c1.real, c2.real), min(c1.imag, c2.imag))
135
136	mandel= Image.new("L", (width, height))
137	draw= ImageDraw.Draw(mandel)
138
139	def drawrect((x1, y1), (x2, y2)):
140	"Draw a rectangle defined by the corners (x1, y1) and (x2, y2)."
141	color= plot(x1, y1) # Plot the upper left corner
142	# Should we cut this rectangle and recurse into the pieces?
143	cut= 0
144	## Plot the edges and at the same time check if they are all the
145	## same color.
146	# Upper edge (left to right)
147	x, y= x1+1, y1
148	while x<=x2:
149	if color<>plot(x, y):
150	cut= 1
151	x= x+1
152	# Right edge (top to bottom)
153	x, y= x2, y1+1
154	while y<=y2:
155	if color<>plot(x, y):
156	cut= 1
157	y= y+1
158	# Bottom edge (right to left)
159	x, y= x2-1, y2
160	while x>=x1:
161	if color<>plot(x, y):
162	cut= 1
163	x= x-1
164	# Left edge (bottom to top)
165	x, y= x1, y2-1
166	while y>y1:
167	if color<>plot(x, y):
168	cut= 1
169	y= y-1
170	# If our rectangle is too small to have an inside, we have now
171	# plotted every pixel in it.
172	if x2-x1<=1 or y2-y1<=1:
173	return
174	if cut: # If we're cutting the rectangle
175	if x2-x1 > y2-y1: # If it's wider than high
176	# Cut vertically
177	xc= ((x2-x1)/2)+x1
178	drawrect((x1, y1), (xc, y2))
179	drawrect((xc, y1), (x2, y2))
180	else: # if it's higher than wide
181	# Cut horizontally
182	yc= ((y2-y1)/2)+y1
183	drawrect((x1, y1), (x2, yc))
184	drawrect((x1, yc), (x2, y2))
185	else:
186	# If we're not cutting, it was the same color along the edge
187	if not debug:
188	# Fill in the rectangle
189	draw.rectangle([x1, y1, x2, y2], fill= color)
190	if yorig >= 0: # The x axis is visible on image
191	# Fill in the mirror rectangle
192	# Mirror y1 and y2 over yorig, but keep within image
193	y1m= max(0, min(2*yorig-y1, ymax))
194	y2m= max(0, min(2*yorig-y2, ymax))
195	draw.rectangle([x1, y1m, x2, y2m], fill= color)
196
197	def constant(x, y):
198	creal= (float(x)/xmax)*(c2.real - c1.real)+c1.real
199	cimag= (float(y)/ymax)*(c2.imag - c1.imag)+c1.imag
200	return complex(creal, cimag)
201
202	def plot(x, y):
203	c= constant(x, y)
204	z= i= 0
205	color= mandel.getpixel((x, y))
206	if color<>0:
207	return color
208	while i<maxiter and abs(z)<2:
209	z= z**2+c
210	i= i+1
211	color= 16*i%256
212	if i>=maxiter or color==0:
213	if debug:
214	color= 255
215	else:
216	color= 1
217	try:
218	mandel.putpixel((x, y), color)
219	if yorig >= 0: # The x axis is visible on image
220	ym= 2*yorig-y # ym is y mirrored over yorig
221	if ym >= 0 and ym <= ymax:
222	# ym is on the image
223	mandel.putpixel((x, ym), color)
224	except IndexError, the_error:
225	print "coord", x, y, xm, ym
226	raise IndexError, the_error
227	return color
228
229	# Initial default values
230	yfrom= 0
231	yto= ymax
232	yorig= -1 # The pixel coordinate of the x axis; negative means not on image
233
234	# Check if we should do mirroring
235	if c1.imag > 0 and c2.imag < 0: # the x axis is visible
236	yorig= int((-c1.imag/(c2.imag-c1.imag))*height) # pixel pos. of x axis
237	if c1.imag < abs(c2.imag): # the x axis is closer to top than bottom
238	yfrom= yorig # Begin at x axis
239	else:
240	yto= yorig # End at x axis
241
242	drawrect((0, yfrom), (xmax, yto))
243
244	print "Content-type: image/png"
245	print
246	mandel.save(sys.stdout, "PNG")