import rhinoscriptsyntax as rs
import scriptcontext as sc
import math
import random as ran

allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)

rs.EnableRedraw(enable=False)

# Anmerkug: 
# Sollte beim Laufen des Codes ein Fehler auftreten bzw. die Ringe des Kraters 
# komisch aussehen, bitte nochmals laufen lassen! Löst sich von selbst!

####
# all changable Parameters:
a = 250                 # Ground circle radius
tiefe = -15             # Ground circle thickness

# Second Plane can be disabled by "if 0:" in the code! (87)
b = 500                 # lower plane width (render/environment)
tiefe2 = -1             # lower plane thickness (render/environment)

rings = 35              # Amount of rings on the ground | #rings = ran.randint(20,35)

b_hgt = 120             # builing height in the center (Maß bevor der Planet Radius abgezogen wird zusätzlich!)
aa = 16                 # builing length
bb = 8                  # builing width

offset = 1.5            # Sockel Breite
seam = 8                # Sockelhöhe

planet_rad = 50          # Planet radius
planet_hgt = 150         # distance of Planet on the z-axis

sky_cir_dis = 200       # distance outer planet ring
ring_width = -50        # width of outer planet ring
####

# Crater Ring Creation:
seg = 40        # Inner Cicle Segments, don't change
rad = 35        # Radius of Inner Circle, don't change

cir1 = rs.AddCircle([0,0,0], rad)
pts1 = rs.DivideCurve(cir1, seg, True, True)
#print(pts1)

crv_pts=[]

for i in range(seg):
    xran = ran.randint(0,6)
    yran =ran.randint(0,6)
    crv_pts.append(rs.MoveObject(pts1[i],(xran,yran,0)))
#print(crv_pts)

curve=rs.AddCurve(crv_pts)
curve2=rs.CloseCurve(curve,tolerance=10) #Hauptkurve
rs.RebuildCurve(curve2,point_count=seg)
rs.DeleteObjects((cir1,curve))

curves=[]

curve3=curve2
for k in range(rings):
    off_dis = ran.randint(1, 5)
    curve3 = rs.OffsetCurve(curve2, [0,0, 5],-i*(off_dis/30))
    curves.append(curve3)
    curve2=curve3

zran = ran.randint(2,20)
start = [0,0,0]
end = [0,0,zran]

# Extrude Crater Rings:
for s in range(len(curves)):
    if s<len(curves)-1:
        zran=ran.randint(1,8)
        extr1=rs.ExtrudeCurveStraight((curves[s]), [0,0,0], [0,0,zran])
        extr2=rs.ExtrudeCurveStraight((curves[s+1]), [0,0,0], [0,0,zran])
        rs.CapPlanarHoles(extr1)
        rs.CapPlanarHoles(extr2)
        rs.BooleanDifference(extr2,extr1)

# make a cone, ground plane:
ground_cir = rs.AddCircle([0,0,0], a)
ground_srf = rs.AddPlanarSrf(ground_cir)
path_ground = rs.AddLine([0,0,0], [0,0, tiefe])     # path for extruding
rs.ExtrudeSurface(ground_srf, path_ground, True)

# make a second cone, ground plane (for render purposes):
if 1:
    ground_cir1 = rs.AddCircle([0,0,0], b)
    ground_srf1 = rs.AddPlanarSrf(ground_cir1)
    path_ground1 = rs.AddLine([0,0,0], [0,0, tiefe2])    # path for extruding
    ground_plane1 = rs.ExtrudeSurface(ground_srf1, path_ground1, True)

    rs.MoveObject(ground_plane1, [0,0, tiefe])
    rs.DeleteObjects([ground_cir1, ground_srf1])

# Buiding in the crater center:
corners = [(-aa,-bb,0), (aa,-bb,0), (aa,bb,0), (-aa,bb,0), 
            (-aa,-bb,b_hgt), (aa,-bb,b_hgt), (aa,bb,b_hgt), (-aa,bb,b_hgt)]
box1 = rs.AddBox(corners)

# Comet in building (only cutout, non visible):
comet = rs.AddSphere([0,0,0], planet_rad)
rs.MoveObject(comet, [0,0, b_hgt])

# Building:
broken_build = rs.BooleanDifference(box1, comet, True)
rs.DeleteObject(box1)

# comet 2 in the sky:
comet2 = rs.AddSphere([0,0,0], planet_rad)
comet_reb = rs.RebuildSurface(comet2, (3,3), (20,20))
rs.MoveObject(comet2, [0,0,planet_hgt])

# comet 2 rings:
sky_pos  = rs.AddCircle([0,0,0], sky_cir_dis)
sky_cir = rs.MoveObject( sky_pos, [0,0, planet_hgt])

sky_offs = rs.OffsetCurve(sky_cir, [1,1,0], ring_width)
ring_srf = rs.AddPlanarSrf([sky_offs, sky_cir])
ring_srf2 = rs.CopyObject(ring_srf, [0,0,0])
ring2 = rs.ScaleObject(ring_srf2, [0,0,0], [0.5, 0.5, 0.5], False)
rs.MoveObject(ring2, [0,0,planet_hgt-planet_rad])

# Sweep Sockel:
press = rs.AddPolyline([(-aa*offset,-bb*offset, 0), (aa*offset,-bb*offset,0), (aa*offset,bb*offset,0), (-aa*offset,bb*offset,0), (-aa*offset,-bb*offset,0)])
no_press = rs.AddPolyline([(-aa,-bb, seam), (aa,-bb, seam), (aa,bb, seam), (-aa,bb, seam), (-aa,-bb, seam)])

press_crv = rs.AddCurve([(-aa*offset,-bb*offset, 0), (-aa,-bb, seam)])

# loft Sockel:
rs.AddLoftSrf([press, no_press])

rs.EnableRedraw(True)