def __init__(self):
self.logger = log.get_logger(__name__)
self.logger.debug('Initialising grabber')
self.api = LowLevelApi(TITLE)
self.assets = self.api.load_assets()
self.field_size = Point(int(self.api.size.x / CELL_SIZE), int((self.api.size.y - HEADER_HEIGHT - FOOTER_HEIGHT) / CELL_SIZE))
def __init__(self):
self.api = LowLevelApi()
# Event callbacks
self.stateChange = self.nope
self.lowHp = self.nope
self.criticalHp = self.nope
self.hasLoot = self.nope
self.stateChange = self.nope
# Loading assets
self.icons = {}
for filename in os.listdir('assets'):
if filename.endswith('.png'):
self.icons[os.path.splitext(filename)[0]] = self.api.loadIcon(os.path.join('assets', filename))
print '[d] Assets:', len(self.icons)
class Grabber:
def __init__(self):
self.logger = log.get_logger(__name__)
self.logger.debug('Initialising grabber')
self.api = LowLevelApi(TITLE)
self.assets = self.api.load_assets()
self.field_size = Point(int(self.api.size.x / CELL_SIZE), int((self.api.size.y - HEADER_HEIGHT - FOOTER_HEIGHT) / CELL_SIZE))
def _get_local_offset(self):
return Point(0, HEADER_HEIGHT)
def get_offset(self):
return self._get_local_offset() + self.api.offset
def grab_field(self):
offset = self._get_local_offset()
snapshot = self.api.get_snapshot()
cells = [[None for _ in range(self.field_size.x)] for _ in range(self.field_size.y)]
for x in range(self.field_size.x):
for y in range(self.field_size.y):
coords = Point(CELL_SIZE * x, CELL_SIZE * y)
cell = self.subimage(snapshot, coords)
if self.compare(cell, 'empty'):
cells[y][x] = '?'
elif self.compare(cell, '0'):
cells[y][x] = '0'
elif self.compare(cell, '1'):
cells[y][x] = '1'
elif self.compare(cell, '2'):
cells[y][x] = '2'
elif self.compare(cell, '3'):
cells[y][x] = '3'
elif self.compare(cell, '4'):
cells[y][x] = '4'
elif self.compare(cell, '5'):
cells[y][x] = '5'
elif self.compare(cell, '6'):
cells[y][x] = '6'
elif self.compare(cell, '7'):
cells[y][x] = '7'
elif self.compare(cell, '8'):
cells[y][x] = '8'
elif self.compare(cell, 'mine') or self.compare(cell, 'bang'):
cells[y][x] = 'M'
elif self.compare(cell, 'flag'):
cells[y][x] = '+'
elif self.compare(cell, 'win'):
cells[y][x] = '!'
else:
cells[y][x] = '#'
cell.save('test/test{}{}.png'.format(x, y))
return cells
def subimage(self, image, point, size=Point(CELL_SIZE, CELL_SIZE)):
point = point + self._get_local_offset()
rect = (point.x, point.y, point.x + size.x, point.y + size.y)
result = image.crop(rect)
result.load()
return result
def compare(self, image, sample):
is0 = (sample + '@0') in self.assets and self.api.is_images_equal(image, self.assets[sample + '@0'])
is1 = (sample + '@1') in self.assets and self.api.is_images_equal(image, self.assets[sample + '@1'])
return is0 or is1
class GameApi:
state = UNDEFINED
icons = None
lowHpThreshold = 50
criticalHpThreshold = 30
def __init__(self):
self.api = LowLevelApi()
# Event callbacks
self.stateChange = self.nope
self.lowHp = self.nope
self.criticalHp = self.nope
self.hasLoot = self.nope
self.stateChange = self.nope
# Loading assets
self.icons = {}
for filename in os.listdir('assets'):
if filename.endswith('.png'):
self.icons[os.path.splitext(filename)[0]] = self.api.loadIcon(os.path.join('assets', filename))
print '[d] Assets:', len(self.icons)
def nope(self):
print '[w] Not implemented'
def checkState(self):
if self.api.testPixel(NEXUS, COLORS['NEXUS_MARKER']):
if self.state != CAN_TELEPORT:
self.state = CAN_TELEPORT
self.stateChange(CAN_TELEPORT)
elif self.state != UNDEFINED:
self.state = UNDEFINED
self.stateChange(UNDEFINED)
def checkHP(self):
criticalHealth = Point(HEALTH_BASE.x + int(1.0 * self.criticalHpThreshold / 100 * HEALTH_SIZE), HEALTH_BASE.y)
lowHealth = Point(HEALTH_BASE.x + int(1.0 * self.lowHpThreshold / 100 * HEALTH_SIZE), HEALTH_BASE.y)
if self.api.testPixel(criticalHealth, COLORS['NOHP']):
self.criticalHp()
elif self.api.testPixel(lowHealth, COLORS['NOHP']):
self.lowHp()
def checkLoot(self):
loot8 = self.api.getLootImage(8)
return loot8.isEqual(self.icons['empty'])
def checkLootItemInSlot(self, slot, acceptedItems):
lootSlot = self.api.getLootImage(slot, checkHashes = True)
for item in acceptedItems:
if lootSlot.isEqual(self.icons[item]):
return item
return None
def run(self):
while True:
self.api.enterFrame()
self.checkState()
if self.state == CAN_TELEPORT:
self.checkHP()
if self.checkLoot():
self.hasLoot()
self.api.exitFrame()
time.sleep(CHECK_INTERVAL)
