Esempio n. 1
0
def getZoneStats(gameFrame):

    centerPoint = [297,280]

    size = 180
    edgeBuffer = 10

    half = size//2
    threeHalf = int(1.5 *size)

    zoneFrame = gameFrame[(centerPoint[1] - threeHalf - edgeBuffer):(centerPoint[1] + threeHalf + edgeBuffer),
                            (centerPoint[0] - threeHalf - edgeBuffer):(centerPoint[0] + threeHalf + edgeBuffer)]

    obsFrame = gameFrame[(centerPoint[1] - 60):(centerPoint[1] + 60),
                            (centerPoint[0] - 60):(centerPoint[0] + 60)]

    bulletCenters = Bullets.getBulletCenters(zoneFrame)
    enemyCenters = GetData.getEnemiesScreen1(zoneFrame)
    obsMask = Trees.getObstacleMask(obsFrame,10)           # 10 - Color range +/- 10 BGR values

    obstacle_array = getObstacles(obsMask,50)               # 50 - Mean for mask value. 0 black, 255 white
    bulletZoneCount = getZoneCounts(zoneFrame.shape,bulletCenters,size)
    enemyZoneCount = getZoneCounts(zoneFrame.shape,enemyCenters,size)

    zoneStats = []

    for i in range(0,8):
        zoneStats.append((obstacle_array[i],bulletZoneCount[i],enemyZoneCount[i]))


    return zoneStats