' stencil_shadows.bmx
' static and dynamic stencil shadows

Strict

Framework Openb3d.B3dglgraphics
Import Brl.Random

Graphics3D 800,600,0,2


Local camera:TCamera=CreateCamera()
PositionEntity camera,-20,10,0
CameraClsColor camera,0,0,255

' set light mode
Local lightmode%=2
Local light:TLight=CreateLight(lightmode)
Local light2:TLight=CreateLight(lightmode)

Local light_piv:TPivot=CreatePivot()
PositionEntity light_piv,50,0,50
Local light_piv2:TPivot=CreatePivot()
PositionEntity light_piv2,50,0,50

Local sphere:TMesh=CreateSphere(8,light_piv)
PositionEntity sphere,0,25,50
EntityColor sphere,255,0,0

Local sphere2:TMesh=CreateSphere(8,light_piv2)
PositionEntity sphere2,0,25,50
EntityColor sphere2,0,255,0

Local plane:TMesh=CreatePlane(16)
MoveEntity plane,0,-1,0
Local tex:TTexture=LoadTexture("media/Moss.bmp")
EntityTexture plane,tex

PointEntity camera,plane
PositionEntity camera,-40,25,55

' set shadow vars
Local static%=0 ' global static or dynamic shadows
Local numtypes%=25 ' number of cubes or cylinders
Local size%=100 ' size of area
Local numshadows%=0 ' shadow counter
Local animnumshadows%=0
Global scolor:Int[]=[100,100,100,100] ' global shadow colors 0..255
SetShadowColor(scolor[0],scolor[1],scolor[2],scolor[3])

Global lightcasters:TMesh[] ' all shadow casters
Global light1shadows:TShadowObject[] ' all lights if dynamic
Global light2shadows:TShadowObject[] ' second shadow array for static shadows
Global animlightcasters:TMesh[1]
Global animlight1shadows:TShadowObject[1]
Global animlight2shadows:TShadowObject[1]

' alphamapped quads
Local quad:TMesh=CreateQuad()
RotateMesh quad,90,0,0
ScaleMesh quad,15,15,15
PositionEntity quad,15,5,55

Local tex2:TTexture=LoadTexture("media/alpha_map.png")
EntityTexture(quad,tex2)
EntityFX(quad,32)
HideEntity quad

' load anim mesh
Local anim_time#=0
Local animmode%=1
Local ent:TMesh=LoadAnimMesh("media/zombie.b3d")
PositionEntity ent,55,0,50

animlightcasters[animnumshadows]=ent
animlight1shadows[animnumshadows]=CreateShadow(animlightcasters[animnumshadows],static)
If static=1
    animlight2shadows[animnumshadows]=CreateShadow(animlightcasters[animnumshadows],static)
EndIf
animnumshadows:+1 ' increment array index

' load cubes - static shadows need a separate shadow array for each light
For Local i%=0 To numtypes-1
    lightcasters=lightcasters[..numshadows+1]
    light1shadows=light1shadows[..numshadows+1]
    light2shadows=light2shadows[..numshadows+1]

    lightcasters[numshadows]=CreateCube()
    PositionEntity lightcasters[numshadows],Rnd(size),0,Rnd(size)
    EntityColor lightcasters[numshadows],Rnd(255),Rnd(255),Rnd(255)
    light1shadows[numshadows]=CreateShadow(lightcasters[numshadows],static)
    If static=1
        light2shadows[numshadows]=CreateShadow(lightcasters[numshadows],static)
    EndIf

    numshadows:+1
Next

' load cylinders
For Local i%=0 To numtypes-1
    lightcasters=lightcasters[..numshadows+1]
    light1shadows=light1shadows[..numshadows+1]
    light2shadows=light2shadows[..numshadows+1]

    lightcasters[numshadows]=CreateCylinder()
    PositionEntity lightcasters[numshadows],Rnd(size),1.5,Rnd(size)
    EntityColor lightcasters[numshadows],Rnd(255),Rnd(255),Rnd(255)
    FitMesh lightcasters[numshadows],-1,-1,-1,2,5,2 ' use ScaleMesh/FitMesh but not ScaleEntity
    'ScaleEntity cylinder[i],2,5,2
    light1shadows[numshadows]=CreateShadow(lightcasters[numshadows],static)
    If static=1
        light2shadows[numshadows]=CreateShadow(lightcasters[numshadows],static)
    EndIf

    numshadows:+1
Next

' initial static shadows render
TurnEntity light_piv2,0,-90,0
PositionEntity light,EntityX(sphere,1),EntityY(sphere,1),EntityZ(sphere,1)
PositionEntity light2,EntityX(sphere2,1),EntityY(sphere2,1),EntityZ(sphere2,1)
If static=1 Then CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)

Local wiretoggle%=-1
Local lightmove%=1
Local cylindermove%=1
Local hidelight1%=0
Local hidelight2%=0
Local max2dmode%=0
Local hidealpha%=1

' fps code
Local old_ms%=MilliSecs()
Local renders%, fps%, ticks%=0


Local mesh1:TMesh , mesh2:TMesh, mesh3:TMesh
Local surface1:TSurface , surface2:TSurface, surface3:TSurface
mesh1    = CreateMesh()
surface1 = CreateSurface( mesh1 )
mesh2    = CreateMesh()
surface2 = CreateSurface( mesh2 )
mesh3    = CreateMesh()
surface3 = CreateSurface( mesh3 )
EntityBlend( mesh1, 1 )
EntityBlend( mesh2 , 3 )
'EntityBlend( mesh3, 3 )
EntityFX   ( mesh1, 1+8+16+32 )
EntityFX   ( mesh2, 1+8+16+32 )


While Not KeyHit(KEY_ESCAPE) And Not AppTerminate()

    If KeyHit(KEY_SPACE) Then wiretoggle=-wiretoggle
    If wiretoggle=1 Then Wireframe True Else Wireframe False

    ' move camera
    If KeyDown(KEY_PAGEUP) Then TranslateEntity camera,1,0,0
    If KeyDown(KEY_PAGEDOWN) Then TranslateEntity camera,-1,0,0
    If KeyDown(KEY_UP) Then MoveEntity camera,0,0,1
    If KeyDown(KEY_DOWN) Then MoveEntity camera,0,0,-1
    If KeyDown(KEY_LEFT) Then MoveEntity camera,-1,0,0
    If KeyDown(KEY_RIGHT) Then MoveEntity camera,1,0,0

    ' start/stop light movement
    If KeyHit(KEY_L) Then lightmove=-lightmove
    If lightmove=1 Then TurnEntity light_piv,0,1,0 ; TurnEntity light_piv2,0,-1,0

    ' start/stop cube/cylinder movement
    If KeyHit(KEY_C) Then cylindermove=-cylindermove
    If static=0 '  static shadows only work for static casters - causes 'wrong' casting on casters
        For Local j%=1 To numshadows-1
            If cylindermove=1 Then TurnEntity lightcasters[j],0,0.25,-2.5
        Next
    EndIf

    ' reset static shadows
    If KeyHit(KEY_R)
        If static=1 Then CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)
    EndIf

    ' hide/show light1 - hide unwanted static shadows by freeing/re-creating
    If KeyHit(KEY_1)
        hidelight1=Not hidelight1
        If hidelight1
            HideEntity light
            HideEntity sphere
            If static=1
                FreeStaticShadows(numshadows,animnumshadows,camera,light,light2,1)
            EndIf
        Else
            ShowEntity light
            ShowEntity sphere
            If static=1
                CreateStaticShadows(numshadows,animnumshadows,camera,light,light2,1)
            EndIf
        EndIf
    EndIf

    ' hide/show light2
    If KeyHit(KEY_2)
        hidelight2=Not hidelight2
        If hidelight2
            HideEntity light2
            HideEntity sphere2
            If static=1
                FreeStaticShadows(numshadows,animnumshadows,camera,light,light2,2)
            EndIf
        Else
            ShowEntity light2
            ShowEntity sphere2
            If static=1
                CreateStaticShadows(numshadows,animnumshadows,camera,light,light2,2)
            EndIf
        EndIf
    EndIf

    ' max2d mode
    If KeyHit(KEY_M) Then max2dmode=Not max2dmode

    ' hide alpha quad
    If KeyHit(KEY_Q)
        hidealpha=Not hidealpha
        If hidealpha Then HideEntity quad Else ShowEntity quad
    EndIf

    ' anim mode
    If KeyHit(KEY_A) Then animmode:+1
    If animmode>2 Then animmode=0
    If animmode>0 Then anim_time#:+-0.5
    SetAnimTime(ent,anim_time#)

    If static=1 And animmode>0 ' shows dynamic with static shadows - anim is just set at array index 0
        CastStaticShadows(numshadows,animnumshadows,camera,light,light2,animmode)
    EndIf

    PositionEntity light,EntityX(sphere,1),EntityY(sphere,1),EntityZ(sphere,1)
    PointEntity light,plane
    PositionEntity light2,EntityX(sphere2,1),EntityY(sphere2,1),EntityZ(sphere2,1)
    PointEntity light2,plane

    If static=1 ' must have only one light on for main render or get doubled static shadows
        HideStaticLights(light,light2,hidelight1,hidelight2)
    EndIf

    If static=0 ' render first shadow color
        SetShadowColor(scolor[0],scolor[1],scolor[2],100) ' light1 color
    EndIf

    RenderWorld

    If static=0 ' render second shadow color
        SetShadowColor(scolor[0],scolor[1],scolor[2],250) ' light2 color
        If light2.Hidden() Then SetShadowColor(scolor[0],scolor[1],scolor[2],100)
        If Not light.Hidden() Or Not light2.Hidden() ' don't render if no lights
            TShadowObject.ShadowRenderWorldZFail() ' re-render shadows
        EndIf
    EndIf

    If static=1
        ShowStaticLights(light,light2,hidelight1,hidelight2)
    EndIf

    ' calculate fps
    renders=renders+1
    If MilliSecs()-old_ms>=1000
        old_ms=MilliSecs()
        fps=renders
        renders=0
    EndIf

    Text 0,0,"FPS: "+fps
    Text 0,20,"Arrows: move camera, L: light movement, C: cube/cylinder movement"
    Text 0,40,"R: reset static shadows, 1/2: hide lights, light mode = "+lightmode
    Text 0,60,"M: Max2d mode, Q: hide alpha quad, A: anim mode = "+animmode
    Text 0,80,"camera position = "+EntityX(camera)+" "+EntityY(camera)+" "+EntityZ(camera)

    If max2dmode
        BeginMax2D()
        DrawText "Testing Max2d",0,100
        EndMax2D()
    EndIf

    Flip

Wend


Function HideStaticLights( light:TLight,light2:TLight,light1hid%,light2hid% )

    If light1hid=1 Then HideEntity light ' light1 off
    If light2hid=1 Then HideEntity light2 ' light2 off
    If light1hid=0 And light2hid=0 Then HideEntity light2 ' light1 on, light2 on - so hide light2

End Function

Function ShowStaticLights( light:TLight,light2:TLight,light1hid%,light2hid% )

    If light1hid=1 Then ShowEntity light ' light1 off
    If light2hid=1 Then ShowEntity light2 ' light2 off
    If light1hid=0 And light2hid=0 Then ShowEntity light2 ' light1 on, light2 on

End Function

Function FreeStaticShadows( numshadows%,animnumshadows%,camera:TCamera,light:TLight,light2:TLight,lightid% )

    For Local i%=0 To numshadows-1
        If lightid=1 Then FreeShadow(light1shadows[i])
        If lightid=2 Then FreeShadow(light2shadows[i])
    Next
    CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)

    For Local i%=0 To animnumshadows-1
        If lightid=1 Then FreeShadow(animlight1shadows[i])
        If lightid=2 Then FreeShadow(animlight2shadows[i])
    Next
    CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)

End Function

Function CreateStaticShadows( numshadows%,animnumshadows%,camera:TCamera,light:TLight,light2:TLight,lightid% )

    For Local i%=0 To numshadows-1
        If lightid=1 Then light1shadows[i]=CreateShadow(lightcasters[i],1)
        If lightid=2 Then light2shadows[i]=CreateShadow(lightcasters[i],1)
    Next
    CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)

    For Local i%=0 To animnumshadows-1
        If lightid=1 Then animlight1shadows[i]=CreateShadow(animlightcasters[i],1)
        If lightid=2 Then animlight2shadows[i]=CreateShadow(animlightcasters[i],1)
    Next
    CastStaticShadows(numshadows,animnumshadows,camera,light,light2,2)

End Function

Function CastStaticShadows( numshadows%,animnumshadows%,camera:TCamera,light:TLight,light2:TLight,animmode% )

    Local currdist#, maxdist#=0
    Local light1hid%=light.hide[0]
    Local light2hid%=light2.hide[0]

    For Local i%=0 To numshadows-1 ' calculate furthest caster from camera to set camera range
        currdist=EntityDistance(camera,lightcasters[i])
        If currdist>maxdist Then maxdist=currdist
    Next
    For Local i%=0 To animnumshadows-1 ' calculate furthest caster from camera to set camera range
        currdist=EntityDistance(camera,animlightcasters[i])
        If currdist>maxdist Then maxdist=currdist
    Next

    CameraRange camera,1,maxdist+100 ' shorthen camera range to cap any 'detached' static shadows

    If light2hid=0 And light1hid=0 ' light2 on, light1 on - 2 lights, 3 possible states
        ShowEntity light2
        HideEntity light
        For Local i%=0 To animnumshadows-1
            animlight2shadows[i].ResetShadow()
        Next
        If animmode=2
            For Local i%=0 To numshadows-1
                light2shadows[i].ResetShadow()
            Next
        EndIf
        RenderWorld

        HideEntity light2
        ShowEntity light
        For Local i%=0 To animnumshadows-1
            animlight1shadows[i].ResetShadow()
        Next
        If animmode=2
            For Local i%=0 To numshadows-1
                light1shadows[i].ResetShadow()
            Next
        EndIf
        RenderWorld

        ShowEntity light
        ShowEntity light2
    EndIf

    If light2hid=1 And light1hid=0 ' light2 off, light1 on
        HideEntity light2
        ShowEntity light
        For Local i%=0 To animnumshadows-1
            animlight1shadows[i].ResetShadow()
        Next
        If animmode=2
            For Local i%=0 To numshadows-1
                light1shadows[i].ResetShadow()
            Next
        EndIf
        RenderWorld

        ShowEntity light
    EndIf

    If light2hid=0 And light1hid=1 ' light2 on, light1 off
        ShowEntity light2
        HideEntity light
        For Local i%=0 To animnumshadows-1
            animlight2shadows[i].ResetShadow()
        Next
        If animmode=2
            For Local i%=0 To numshadows-1
                light2shadows[i].ResetShadow()
            Next
        EndIf
        RenderWorld

        ShowEntity light2
    EndIf

    CameraRange camera,1,1000

End Function