Files
leporello-js/docs/examples/fractal_tree/fractal_tree.js
Dmitry Vasilev 0a26ac6fa5 canvas
2024-03-07 15:30:27 +08:00

63 lines
1.7 KiB
JavaScript

// Original source: http://bricault.mit.edu/recursive-drawing
// Author: Sarah Bricault
// Canvas setup
const canvas = document.createElement('canvas')
canvas.width = 700
canvas.height = 700
document.body.appendChild(canvas)
const ctx = canvas.getContext('2d')
ctx.translate(canvas.width / 2, canvas.height)
// Draw a tree
fractalTreeBasic({totalIterations: 10, basicLength: 10, rotate: 25})
function fractalTreeBasic({totalIterations, basicLength, rotate}) {
// Draw the tree trunk
const trunkLength = basicLength * 2 * Math.pow(1.2, totalIterations + 1)
const width = Math.pow(totalIterations, 0.6)
ctx.beginPath()
ctx.moveTo(0, 0)
ctx.lineTo(0, - trunkLength)
ctx.lineWidth = width
ctx.strokeStyle = 'black'
ctx.stroke()
drawBranch(90, [0, - trunkLength], totalIterations + 1)
function drawBranch(angle, startPoint, iterations) {
const len = basicLength * Math.pow(1.2, iterations)
const width = Math.pow(iterations, 0.6)
const red = Math.floor(255 - (iterations / totalIterations) * 255)
const green = 0
const blue = Math.floor( 255 - (iterations / totalIterations) * 255)
const color = `rgb(${red}, ${green}, ${blue})`
const x1 = startPoint[0]
const y1 = startPoint[1]
const y2 = y1 - len * Math.sin((angle * Math.PI) / 180)
const x2 = x1 + len * Math.cos((angle * Math.PI) / 180)
console.log('draw branch', x1, y1, x2, y2)
ctx.beginPath()
ctx.moveTo(x1, y1)
ctx.lineTo(x2, y2)
ctx.lineWidth = width
ctx.strokeStyle = color
ctx.stroke()
if (iterations - 1 > 0) {
// draw left branch
drawBranch(angle + rotate, [x2, y2], iterations - 1)
// draw right branch
drawBranch(angle - rotate, [x2, y2], iterations - 1)
}
}
}