用python实现参数化Lsystem

2024-06-16 13:19:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我一直在用python开发参数化L系统的实现。我的例子是一个2D树,因此我使用pythonturtle库。我为一个非参数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和{},但我不知道在哪里或如何使用它。以下是我的L系统规则和公理:

^{2}$

r1, r2, q, e的值是已知的。也知道s0 & w0。我不想以字符串格式存储这些参数的值,我希望它们在列表或数组中,但我不知道如何存储。下面是我要绘制的树与我的代码绘制的树的对比图: Parametric Treeenter image description here


Tags: self参数movemydefwidthlengthrules