# File: staircase.py
# Author: Petr Sorfa
# Desc: Creates a stair case

import am
import math

am.begin()

in_r = 0.4
out_r = 0.8
levels = 4
steps = 10

am.surface_begin()

for level in range(levels):
    for step in range(steps):
        anglex = math.cos (math.pi*2*step/9.0)
        anglez = math.sin (math.pi*2*step/9.0)
        x0 = in_r*anglex
        z0 = in_r*anglez
        x1 = out_r*anglex
        z1 = out_r*anglez

        y = level + step/9.0

        am.surface_add_row([x0, y, z0, x1, y, z1])
        am.surface_add_row([x0, y+1.0/steps, z0, x1, y+1.0/steps, z1])

am.surface_end()

am.end()