import pygame as pg
from pygame import *

pg.init()
scr = pg.display.set_mode((1200,600))
kol = 3
rasst = round(400/kol*3)
green = (200,255,200)
brown = (180,180,180)
back = (155,220,255)
def a(k,n):
    num = round(n*k/10)
    return num

def DrawTree(k,x0,y0):
    pl_1 = ((x0-a(k,30),y0-a(k,10)),(x0,y0-a(k,40)),(x0+a(k,30),y0-a(k,10)))
    pl_2 = ((x0-a(k,20),y0-a(k,40)),(x0,y0-a(k,60)),(x0+a(k,20),y0-a(k,40)))
    pl_3 = ((x0-a(k,10),y0-a(k,60)),(x0,y0-a(k,70)),(x0+a(k,10),y0-a(k,60)))
    pl_4 = ((x0-a(k,5),y0-a(k,10)),(x0-a(k,5),y0),(x0+a(k,5),y0),(x0+a(k,5),y0-a(k,10)))
    trunk = pg.draw.polygon(scr,brown,pl_4)
    part1 = pg.draw.polygon(scr,green,pl_1)
    part2 = pg.draw.polygon(scr,green,pl_2)
    part3 = pg.draw.polygon(scr,green,pl_3)

k = 0
while 1:
    k1 = k%60
    k2 = k//60
    scr.fill(back)

    for j in range(k2):
        DrawTree(60,150+rasst*j,600)
    if k2<=kol:
        DrawTree(k1,150+rasst*k2,600)
    pg.time.wait(70)
    pg.display.update()
    k += 1