SSF2-AI-demos/Bert.lua at master · EHummerston/SSF2-AI-demos · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
-----------------------------------------------------------------------------
-- An implementation of the SSF2 AI which aims to be a simple algorithm which
-- mimics a human Ryu player.
--
-- 2017 Edward Hummerston
-----------------------------------------------------------------------------

require("SSF2-AI.Bot")

Bert = {}
setmetatable(Bert, {__index = Bot})
local Bert_mt = {__index = Bert}

-----------------------------------------------------------------------------
-- Override inherited constructor.
--
-- @param playerSlot     The controller port to represent that the algorithm
--                       will occupy.
-- @return               An instance of the created object.
-----------------------------------------------------------------------------
function Bert.new(playerSlot)
   local self = Bot.new(playerSlot, 10)
   setmetatable(self, Bert_mt)

   self.name = "Bert"
   self.i = 0
   self.walkTimer = 0
   self.newAttack = true
   return self
end

-----------------------------------------------------------------------------
-- Sets the controller state in order to repeatedly complete a Hadoken
-- command.
-- @param strength       A string of the button to use for the fireball.
-----------------------------------------------------------------------------
function Bert:fireball(strength)
   if self.i < 3 then   -- down
      self:setButton("Down",true)
   elseif self.i < 6 then  -- down forward
      self:setButton("Down",true)
      self:setButton("Toward",true)
   elseif self.i < 9 then  -- forward
      self:setButton("Toward",true)
   elseif self.i < 12 then -- forward + puinch
      self:setButton("Toward",true)
      self:setButton(strength,true)
   -- Long buffer to prevent accidental Shoryuken commands
   elseif self.i > 30 then
      self.i = -1
   end
end

-----------------------------------------------------------------------------
-- Sets the controller state in order to repeatedly complete a Shoryuken
-- command.
-----------------------------------------------------------------------------
function Bert:dragonPunch()
   if self.i < 3 then  -- forward
      self:setButton("Toward",true)
   elseif self.i < 6 then -- down
      self:setButton("Down",true)
   elseif self.i < 9 then -- down forward + punch
      self:setButton("Down",true)
      self:setButton("Toward",true)
      self:setButton("HP",true)
   elseif self.i > 12 then
      self.i = -1
   end
end

-----------------------------------------------------------------------------
-- Interprets the current game state and determines which action is
-- appropriate and thus what inputs the controller should be set to.
-----------------------------------------------------------------------------
function Bert:advance()
   self.i = self.i + 1

   if self:isAttacking() then
      return
   end

   if self:isOpponentAttacking() and not self:isOpponentAir() then
   -- if enemy is attacking and on the ground

      if self.newAttack and self:getDistance() < 0x24 then
      -- if enemy is close and they weren't attacking last frame
         self:dragonPunch()
         self.action = "ume shoryken"
      elseif self:hasOpponentFireball() then-- counter fireball with the same
         self:fireball("HP")
         self.action = "counter fireball"
      else  -- they are far away or this isn't the first frame of the attack
         self.newAttack = false
         self:setButton("Back",true)
         self.action = "block"
         self.i = -1
         if self:isOpponentCrouching() then  -- if enemy is crouching
            self:setButton("Down",true)
            self.action = "low block"
         end
      end
   else  -- the enemy is in the air or not attacking
      self.newAttack = true
      if self:getDistance() < 0x28 then   -- they are close
         if self:isOpponentAir() then  -- if enemy is in air
            self:dragonPunch()
            self.action = "anti-air"
         else  -- p2 is close on the ground
            self:setButton("Back",true)
            self.action = "walk back"
            self.i = -1
         end
      else  -- they are far away
         if not self:hasFireball() and not self:isOpponentAir() then
         -- there isn't a fireball and p2 is on the ground
            if self.walkTimer <= 0 then   -- we weren't walking too recently
               local fbStrength = self:getDistance() % 3
               if fbStrength == 0 then
                  self:fireball("LP")
               elseif fbStrength == 1 then
                  self:fireball("MP")
               else
                  self:fireball("HP")
               end
               self.action = "fireball"
            else  -- we were pressing forward recently
               self.walkTimer = self.walkTimer - 1
               self.action = "dp timeout"
            end
         else  -- we have a fireball out or the enemy is in the air
            self:setButton("Toward",true)
            self.action = "walk forward"
            self.walkTimer = 15
            self.i = -1
         end
      end
   end
end