From e13278c6a8bfa70b8c9b6576bac1852100a4ef0b Mon Sep 17 00:00:00 2001
From: Ishaan Bhardwaj <sashlilac109@gmail.com>
Date: Tue, 10 Nov 2020 22:34:48 -0500
Subject: [PATCH] Fixed safelock behavior for hard drop modes

---
 tetris/modes/marathon_a2.lua | 4 +++-
 tetris/modes/survival_a2.lua | 1 +
 tetris/modes/survival_a3.lua | 1 +
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/tetris/modes/marathon_a2.lua b/tetris/modes/marathon_a2.lua
index c6231df..5986584 100644
--- a/tetris/modes/marathon_a2.lua
+++ b/tetris/modes/marathon_a2.lua
@@ -34,6 +34,7 @@ function MarathonA2Game:new()
 	}
 
 	self.lock_drop = false
+	self.lock_hard_drop = false
 	self.enable_hold = false
 	self.next_queue_length = 1
 end
@@ -151,7 +152,8 @@ function MarathonA2Game:onLineClear(cleared_row_count)
 		if self:qualifiesForMRoll() then self.grade = 32 end
 		self.roll_frames = -150
 	end
-	if self.level >= 900 then self.lock_drop = true end
+	self.lock_drop = self.level >= 900
+	self.lock_hard_drop = self.level >= 900
 end
 
 function MarathonA2Game:updateSectionTimes(old_level, new_level)
diff --git a/tetris/modes/survival_a2.lua b/tetris/modes/survival_a2.lua
index cc659ca..aad38e2 100644
--- a/tetris/modes/survival_a2.lua
+++ b/tetris/modes/survival_a2.lua
@@ -27,6 +27,7 @@ function SurvivalA2Game:new()
 	}
 
 	self.lock_drop = true
+	self.lock_hard_drop = true
 end
 
 function SurvivalA2Game:getARE()
diff --git a/tetris/modes/survival_a3.lua b/tetris/modes/survival_a3.lua
index cf0c9f1..2038266 100644
--- a/tetris/modes/survival_a3.lua
+++ b/tetris/modes/survival_a3.lua
@@ -29,6 +29,7 @@ function SurvivalA3Game:new()
 	}
 	
 	self.lock_drop = true
+	self.lock_hard_drop = true
 	self.enable_hold = true
 	self.next_queue_length = 3