72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import argparse
|
||
|
|
from turtle import Turtle
|
||
|
|
import random
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
description="Draw Barnsley fern using Python's Turtle graphics.")
|
||
|
|
parser.add_argument('-d', '--dots', metavar='number',
|
||
|
|
help='Number of dots to draw. Deefault is 20000',
|
||
|
|
type=int, default=20000)
|
||
|
|
parser.add_argument('-bg', '--background', metavar='color',
|
||
|
|
type=str, default='white',
|
||
|
|
help='Sets background color.')
|
||
|
|
parser.add_argument('-fg', '--foreground', metavar='color',
|
||
|
|
type=str, default='green',
|
||
|
|
help='Sets pen color color.')
|
||
|
|
parser.add_argument('-s', '--scale', metavar='pixels',
|
||
|
|
type=int, default=500,
|
||
|
|
help='Chooses scale. Default is 500 pixels.')
|
||
|
|
parser.add_argument('-ww', '--width', metavar='pixels',
|
||
|
|
type=int, default=600,
|
||
|
|
help='Window width.')
|
||
|
|
parser.add_argument('-wh', '--height', metavar='pixels',
|
||
|
|
type=int, default=800,
|
||
|
|
help='Window height.')
|
||
|
|
args = parser.parse_args()
|
||
|
|
return args
|
||
|
|
|
||
|
|
def draw_fern(pen):
|
||
|
|
x = 0.5
|
||
|
|
y = 0.0
|
||
|
|
for i in range(args.dots):
|
||
|
|
r = random.randint(0, 100)
|
||
|
|
if r <= 1:
|
||
|
|
tempx = 0.5
|
||
|
|
tempy = 0.16 * y
|
||
|
|
elif r <= 8:
|
||
|
|
tempx = 0.2 * x - 0.26 * y + 0.400
|
||
|
|
tempy = 0.23 * x + 0.22 * y - 0.045
|
||
|
|
elif r <= 15:
|
||
|
|
tempx = -0.15 * x + 0.28 * y + 0.575
|
||
|
|
tempy = 0.26 * x + 0.24 * y - 0.086
|
||
|
|
else:
|
||
|
|
tempx = 0.85 * x + 0.04 * y + 0.075
|
||
|
|
tempy = -0.04 * x + 0.850 * y + 0.180
|
||
|
|
if r == 100:
|
||
|
|
r = 0
|
||
|
|
r += 1
|
||
|
|
x, y = tempx, tempy
|
||
|
|
pen.setpos(x * args.scale - args.scale // 2, y * args.scale - args.scale // 2)
|
||
|
|
pen.dot(1)
|
||
|
|
|
||
|
|
def setup_pen():
|
||
|
|
pen = Turtle()
|
||
|
|
pen.speed(0)
|
||
|
|
pen.ht()
|
||
|
|
pen.pu()
|
||
|
|
pen.screen.setup (width=args.width, height=args.height)
|
||
|
|
pen.screen.title('Barnsley fern')
|
||
|
|
pen.screen.bgcolor(args.background)
|
||
|
|
pen.screen.tracer(n=15000) # skips frames to speed up drawing
|
||
|
|
pen.color(args.foreground)
|
||
|
|
return pen
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
args = parse_args()
|
||
|
|
pen = setup_pen()
|
||
|
|
draw_fern(pen)
|
||
|
|
input('All done!\nPress Enter to quit')
|
||
|
|
exit()
|