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