fix zora mask resurface bug when entering dungeons from ow after diving

Masks/zora_mask.asm

@@ -27,18 +27,18 @@ UpdateZoraPalette:
 
 ; =========================================================
 
-; TODO: Change from "bunny palette" to blue zora palette colors 
+; TODO: Finish the Zora palette
 zora_palette:
   dw #$7BDE, #$7FFF, #$2F7D, #$19B5, #$3A9C, #$14A5, #$4E48, #$3582
   dw #$55BB, #$6EF7, #$7BDE, #$55C7, #$6ECD, #$2E5A, #$1970, #$7616
-  ; dw #$6565, #$7271, #$2AB7, #$477E, #$1997, #$14B5, #$459B, #$69F2
-  ; dw #$7AB8, #$2609, #$19D8, #$3D95, #$567C, #$1890, #$52F6, #$2357, #$0000
+; dw #$6565, #$7271, #$2AB7, #$477E, #$1997, #$14B5, #$459B, #$69F2
+; dw #$7AB8, #$2609, #$19D8, #$3D95, #$567C, #$1890, #$52F6, #$2357
 
 ; zora_palette:
 ;   dw #$7BDE, #$7FFF, #$2F7D, #$19B5, #$3A9C, #$14A5, #$19FD, #$14B6
 ;   dw #$55BB, #$362A, #$3F4E, #$162B, #$22D0, #$2E5A, #$1970, #$7616
 ;   dw #$6565, #$7271, #$2AB7, #$477E, #$1997, #$14B5, #$459B, #$69F2
-;   dw #$7AB8, #$2609, #$19D8, #$3D95, #$567C, #$1890, #$52F6, #$2357, #$0000
+;   dw #$7AB8, #$2609, #$19D8, #$3D95, #$567C, #$1890, #$52F6, #$2357
 
 ; =========================================================
 
@@ -76,11 +76,10 @@ LinkState_UsingZoraMask:
 {
   ; Check if the mask is equipped 
   LDA $02B2 : CMP #$02 : BNE .normal : CLC
-
     ; Check if we are in water or not 
     LDA $5D : CMP #$04 : BEQ .swimming : CLC
   
-.normal
+  .normal
   ; Return to normal state 
   STZ $55
   STZ $5E     ; Reset speed to normal 
@@ -88,7 +87,7 @@ LinkState_UsingZoraMask:
   STZ $0351
   JMP .return
   
-.swimming
+  .swimming
   ; Check if we are indoors or outdoors 
   LDA $1B : BNE .dungeon ; z flag is 1 
 
@@ -137,13 +136,8 @@ LinkState_UsingZoraMask:
   {
     ; Check if we are in water or not 
     LDA $5D : CMP #$04 : BNE .return_dungeon : CLC
-
       ; Check if already underwater
       LDA !ZoraDiving : BNE .return_dungeon : CLC
-
-    ; Check if we are on a proper tile or not 
-    ; 
-
         ; Check the Y button and clear state if activated
         JSR Link_CheckNewY_ButtonPress : BCC .return_dungeon
           LDA $3A : AND.b #$BF : STA $3A
@@ -180,13 +174,10 @@ warnpc $078364
 
 pullpc
 
-.dungeon_resurface ; TODO: Fix resurfacing bug.
+.dungeon_resurface
 {
   LDA $1B : BEQ .return_default ; We are in overworld actually 
 
-  ; Check if we are swimming 
-  LDA $5D : CMP #$04 : BNE .return_default
-
   ; Check if the player is actually diving 
   LDA !ZoraDiving : BEQ .return_default
 
@@ -224,7 +215,8 @@ pullpc
     JMP .return_default
   }
 
-.return_default
+  .return_default
+  STZ !ZoraDiving
   STZ $0302
   RTS
 }
@@ -245,7 +237,7 @@ pullpc
     STZ $5E                                ; Reset speed to normal
     STZ !ZoraDiving                        ; Reset underwater flag 
     LDA #$62 : STA $9A                     ; Reset dungeon layer
-.return_hop
+  .return_hop
   LDA #$06 : STA $5D ; Set Link to Recoil State
   RTS
 }
@@ -262,11 +254,10 @@ LinkState_ResetMaskAnimated:
   CMP.b #$06 : BEQ .gbc_form
   CMP.b #$02 : BEQ .no_mask
   CMP   #$01 : BNE .transform
-
     ; Restore the sword, shield, and bow override
     LDA $0AAF : STA.l $7EF35A
 
-.transform
+  .transform
   LDY.b #$04 : LDA.b #$23
   JSL   AddTransformationCloud
   LDA.b #$14 : JSR Player_DoSfx2
@@ -274,8 +265,8 @@ LinkState_ResetMaskAnimated:
   STZ $02B2
   JSL Palette_ArmorAndGloves
   LDA #$10 : STA $BC
-.no_mask
-.gbc_form
+  .no_mask
+  .gbc_form
   RTL
 }