Compatibility increased

This commit is contained in:
Enrique Moran 2016-12-08 16:12:27 +01:00 committed by GitHub
parent cb84e739c4
commit fd4781d4d5

View file

@ -6,11 +6,11 @@ class lumberjackBot():
__author__ = "EnriqueMoran" __author__ = "EnriqueMoran"
def __init__(self, playX, playY, playerX, playerY): def __init__(self, playX, playY, treeX, treeY):
self.playX = playX self.playX = playX
self.playY = playY self.playY = playY
self.playerX = playerX self.treeX = treeX
self.playerY = playerY self.treeY = treeY
def move(self, direction): def move(self, direction):
if direction == "left": if direction == "left":
@ -22,14 +22,17 @@ class lumberjackBot():
def play(self): def play(self):
while(1): while(1):
branch = pyautogui.screenshot(region=(playerX, playerY - 85, 1, 1)) # Pixel of tree right side branch = pyautogui.screenshot(region=(treeX - 30, treeY - 130, 1, 1)) # Pixel of tree left side
obstacle = branch.getpixel((0,0)) obstacle = branch.getpixel((0,0))
end = pyautogui.screenshot(region=(treeX - 30, treeY - 100, 1, 1))
stop = end.getpixel((0,0))
if obstacle == (161, 116, 56): if obstacle == (161, 116, 56):
self.move("left")
else:
self.move("right") self.move("right")
time.sleep(0.09) # Speed of lumberman else:
self.move("left")
time.sleep(0.1) # Speed of lumberjack
if stop == (255,255,255):
return 0
if __name__ == "__main__": if __name__ == "__main__":
@ -38,7 +41,22 @@ if __name__ == "__main__":
pyautogui.moveTo(playX, playY) # Start the game by pressing play button pyautogui.moveTo(playX, playY) # Start the game by pressing play button
pyautogui.click() pyautogui.click()
time.sleep(1) time.sleep(1)
playerX, playerY = pyautogui.locateCenterOnScreen('player.png') # Recognize position of player treeX, treeY = pyautogui.locateCenterOnScreen('tree.png') # Recognize tree position
time.sleep(1) time.sleep(1)
print("Im playing... To stop me click on IDLE window and press ctrl + C") print("Im playing...")
lumberjackBot(playX, playY, playerX, playerY).play() lumberjackBot(playX, playY, treeX, treeY).play()
''' SCORE:
0.15 -> 180, 188, 180
0.12 -> 246, 204, 72X, 222
0.11 -> 218X, 128X, 62X, 232, 6X
0.1 -> 12X, 84X, 16X, 6X, 28X
0.099 -> 164X, 108X, 114X
0.08 -> 24X, 34X, 6X, 4X, 24X
0.07 -> 4X, 8X, 2X
'''