Start

star.py

import copy

import numpy as np

from python.common.point import Point
from python.common.svg import Svg
from python.less9.transformations import scaling


def star(size, iterations=1000):
    transformations = [np.matrix([[0.255, 0.0, 0.3726],
                                  [0.0, 0.255, 0.6714],
                                  [0, 0, 1]]),
                       np.matrix([[0.255, 0.0, 0.1146],
                                  [0.0, 0.255, 0.2232],
                                  [0, 0, 1]]),
                       np.matrix([[0.255, 0.0, 0.6306],
                                  [0.0, 0.255, 0.2232],
                                  [0, 0, 1]]),
                       np.matrix([[0.370, -0.642, 0.6356],
                                  [0.642, 0.370, -0.0061],
                                  [0, 0, 1]])]

    probabilities = [0.07, 0.30, 0.33, 0.30]

    points = []
    current = Point(0, 0)
    points.append(current)
    for iteration in range(iterations):
        index = np.random.choice(4, 1, p=probabilities)[0]
        transformation = transformations[index]
        copy_p = copy.deepcopy(current)
        copy_p.apply_transformation(transformation)
        current = copy_p
        if iteration > 200:
            points.append(copy_p)

    svg = Svg()
    list(map(lambda y: y.draw_to_svg(svg), map(lambda x: x.apply_transformation(scaling(size, size)), points)))
    svg.save("output/" + "crystal")


star(1500, 100000)