##################################
### DM2_w23 # AGruber@tugraz.at ###
### ue_05 UN_headquaters  NYC   ###
###################################

# script: Regana Gjeli


##############################
import rhinoscriptsyntax as rs
import random, time, sys   ###
sys.path.append("P:/")     ###
sys.path.append("P:/WWW/flobber27/dm2")     # path for library !!
import DM_lib as dm        ### reload(dm)
##############################      

rs.UnitSystem(4)                                        # km = 5, meters = 4, cm = 3 etc
rs.ShowGrid(None, 0)                                    # grid > 0 = off
rs.ShowGridAxes(None, 1)                                # y/y/z axen display > 0/1 = off/on
rs.ViewDisplayMode(rs.CurrentView(), "wireframe")
rs.Command("cplane w t enter", 0)						# cPlane World Top
dm.PointRadius(displayModeX=0, rad=3, styl=3)
dm.printDisplay(0)                                      # nomen est omen
rs.EnableRedraw(0) ### MAC >> uncomment !



###_________________________________________#
### basic settings for grid to fit UN_slab  #
###                                         # ! no need 2 change !
floors = H  = dm.H = 40                     # default=40 / incl roof
slabs  = L  = dm.L = 11                     # default=11
depth  = D  = dm.D =  4                     # default= 4 / division in building_depth
floorHeight = fH = dm.fH = 4.0              # default= 4.0 / 4.0*(H-1) = 156 meters
                                            #
#############                               # get gridCoords L*D*H = 1760
UnoGridCoords = dm.getUnoGridCoords()       # get from DM_lib as dm
#################___________________________#


dm.newEmptyLayer("UNO::setUp::points", [150,150,200])
dm.newEmptyLayer("UNO::setUp::geo", [150, 20,150])
dm.newEmptyLayer("UNO::setUp::crv", [200, 100, 100])
dm.newEmptyLayer("UNO::setUp", [120,120,140])


##################
depthVec  = dVec = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(0, 2, 0) ))
lengthVec = lVec = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(1, 0, 0) ))
##################


if 1: ### SETUP >> dont' exec @ homework !
    if 1:
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(1, 0, 0) )
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(0, 1, 0) )
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(0, 0, 1) )
        rs.ObjectColor(rs.AllObjects()[2], [0, 222, 0] )
        rs.ObjectColor(rs.AllObjects()[1], [222, 0, 0] )
        rs.ObjectColor(rs.AllObjects()[0], [0, 0, 222] )
        rs.ObjectPrintWidth( rs.AllObjects()[0:3], 1.0 )
        
        p0 = dm.getUnoCoord(  0, 0, 0)
        p1 = dm.getUnoCoord( 10, 0, 0)
        p2 = dm.getUnoCoord( 10, 3, 0)
        p3 = dm.getUnoCoord(  0, 3, 0)
        unoBaseCoords = [p0, p1, p2, p3, p0]
        unoBaseCrv = rs.AddCurve( unoBaseCoords, 1)
        unoTopCoords = [dm.getUnoCoord(  0, 0, 39),
                        dm.getUnoCoord( 10, 0, 39),
                        dm.getUnoCoord( 10, 3, 39),
                        dm.getUnoCoord(  0, 3, 39),
                        dm.getUnoCoord(  0, 0, 39),
                       ]
        unoTopCrv = rs.AddCurve( unoTopCoords, 1)
        rs.ZoomExtents()
    if 0:
        ##############################
        rs.CurrentLayer("UNO::setUp::points")
        rs.AddPoints( UnoGridCoords )
        for i in range(L): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(i, 0, 0) ), [0, 222, 0] )
        for i in range(D): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(0, i, 0) ), [222, 0, 0] )
        for i in range(H): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(0, 0, i) ), [0, 0, 222] )
        
        ##############################
    if 0:
        rs.CurrentLayer("UNO::setUp::crv")
        for i in range(8):
            p0 = dm.getUnoCoord( random.randint(0,L-1), random.randint(0,D-1), random.randint(0,H-1) )
            p1 = dm.getUnoCoord( random.randint(0,L-1), random.randint(0,D-1), random.randint(0,H-1) )
            rs.AddCurve( [p0, p1] )
            dm.textDots(  [p0, p1] )
        rs.ZoomExtents()
    if 0:
        coords = UnoGridCoords[:] ### make COPY - not equal !
        #random.shuffle( coords )
        rs.AddCurve( coords[0:], 2 )

#_____________________________here you go:
dm.newEmptyLayer("myPROJ", [100,100,100])
if 1:
    pX= dm.getUnoCoord(0,0,0)
    rs.AddPoint(pX)
#    rs.AddPoint(dm.getUnoCoord(10,0,0))
#    rs.AddPoint(dm.getUnoCoord(10,0,10))
#    rs.AddPoint(dm.getUnoCoord(0,20,10))
#    rs.AddPoint(dm.getUnoCoord(5,10,5))



def makeLeitCrv():
    #global dom
    #global crv
    p0=rs.VectorAdd(dm.getUnoCoord(5,0,20),rs.VectorScale(dVec,50.0))
    p1=dm.getUnoCoord(10,0,15)
    p2=dm.getUnoCoord(5,3,20)
    p3=rs.VectorAdd(dm.getUnoCoord(10,3,20),rs.VectorScale(lVec,-50.0))
    p4=rs.VectorAdd(dm.getUnoCoord(10,0,10),rs.VectorScale(dVec,50.0))
    
    crv=rs.AddCurve([p0,p1,p2,p3,p4,p0],2)
    dom= rs.CurveDomain(crv)[1]

    print "dom",dom
    pX= rs.EvaluateCurve(crv,dom*0.3)
    return [dom, crv]


def makeRoller( circDivision=32, anzahl =128 , makeStutz = 1):
    allCoords=[]
    
    
    rad=6.00
    
    for i in range(anzahl+1):
        para = dom/(anzahl)*i
        pX=rs.EvaluateCurve(crv,para)
        rs.AddPoint(pX)
        planeX=rs.CurvePerpFrame(crv,para)
        circ=rs.AddCircle(planeX,rad)
        coords = rs.DivideCurve(circ,circDivision,0)
        coord = rs.EvaluateCurve( crv, dom*(i/anzahl)  )
        para = rs.CurveClosestPoint(unoBaseCrv, coord)

        coordOnCurve = rs.EvaluateCurve( unoBaseCrv, para)
        coordOnCrv = rs.EvaluateCurve( crv, para )
        
        if makeStutz:
            rs.ObjectColor(rs.AddCurve( [coord, coordOnCrv ] ),[20,17,50])
            rs.ObjectColor(rs.AddCurve( [coord, coordOnCurve ] ),[100,60,70])

        
        
        rs.DeleteObject(circ)
        allCoords.append(coords)
        rs.AddCurve(coords[int(circDivision*0.5):],1)
        const_1=rs.AddCurve(coords[int(circDivision*0.5):],1)
        midPt =rs.CurveMidPoint(const_1,2)
        if i%2==0:
            const_2= rs.AddLine(midPt,pX)
        
        
        

    for i in range( int(circDivision*0.5)+1 ):
        corrdsX = []
        for circList in allCoords:
            corrdsX.append( circList[i] )
        rs.ObjectColor(rs.AddCurve(corrdsX, 2), [0,99,90] )
        rs.Redraw()


leitValues = makeLeitCrv()
dom = leitValues[0]
crv = leitValues[1]


makeRoller(circDivision=50, anzahl=100, makeStutz = 0)



#____________________________:here you end

###################################
rs.ZoomExtents()
rs.EnableRedraw(0) ### 4 the MACs
dm.printDisplay(1)
dm.newEmptyLayer("Default")