Реализация параметрической L-системы с использованием Python - PullRequest
Новая
       39

Реализация параметрической L-системы с использованием Python

0 голосов
/ 23 января 2019

Я работал над реализацией параметрической L-системы с python. Мой случай - 2D дерево, поэтому я использую библиотеку Python Turtle. Я написал этот простой код для непараметрической L-системы, которая работает довольно хорошо. Вот мой код: 

import turtle as T


class LSystem:
    def __init__(self, axiom, width, length, angle):
        self.axiom = axiom
        self.state = axiom
        self.width = width
        self.length = length
        self.angle = angle
        self.rules = {}
        T.pensize(self.width)

    def add_rules(self, *rules):
        for arg in rules:
            self.rules[arg[0]] = arg[-1]

    def single_generation(self):
        next_state = ""
        for char in self.state:
            if char in self.rules:
                next_state += self.rules[char]
            else:
                next_state += char
        self.state = next_state

    def set_turtle(self, my_tuple):
        T.up()
        T.goto(my_tuple[0], my_tuple[1])
        T.seth(my_tuple[2])
        T.down()

    def length_function(self, r):
        self.length *= r

    def width_function(self, q, e):
        self.width *= q**e

    def draw_turtle(self):
        turtle_state = []
        # ***************
        T.tracer(0, 0)
        T.down()
        T.ht()
        T.speed(0)
        T.seth(90)
        # ***************
        for move in self.state:
            if move == "[":
                turtle_state.append((T.xcor(), T.ycor(), T.heading(), T.pensize()))
            elif move == "]":
                xcor, ycor, head, wid = turtle_state.pop()
                self.set_turtle((xcor, ycor, head))
                self.width = wid
            elif move == 'F':
                T.fd(self.length)
                T.pensize(self.width)
            elif move == '+':
                T.left(self.angle)
            elif move == '-':
                T.right(self.angle)
        T.getscreen()
        T.done()


if __name__ == '__main__':
    my_axiom = "A"
    my_width = 10
    my_length = 60
    my_angle = 33.5
    LSys = LSystem(my_axiom, my_width, my_length, my_angle)
    my_rule = ("A", "F[+A][-A]")
    LSys.add_rules(my_rule)
    LSys.single_generation()
    LSys.single_generation()
    LSys.single_generation()
    LSys.single_generation()
    LSys.single_generation()
    LSys.draw_turtle()

Проблема в том, что этот код работает для простой непараметрической L-системы, но я хочу уменьшать диаметр и длину моего дерева на каждом шаге. Я написал две функции для этого, length_function и width_function, но я не знаю, где и как его использовать. Вот мои правила L-системы и аксиома: 

A(s,w) ------> F(s,w)[+A(s*r1, w* q^e)][-A(s*r2, w* (1-q)^e]
axiom = "A(s0, w0)"

Значения для r1, r2, q, e известны. также известны s0 & w0. Я не хочу хранить значения этих параметров в строковом формате, я хочу их в списке или массиве, но я не знаю как. Вот изображение дерева, которое я пытаюсь нарисовать против того, что рисует мой код: Parametric Tree enter image description here

...