from PIL import Image
import os, shutil, sys

maps_folder = "just_maps/"

print "MAP\tGROUND\tTREES\tWATER"

color_ground = (252, 216, 168)
color_tree = (128, 208, 16)
color_water = (32, 56, 236)

map_number = 0
i = 0
for f in os.listdir(maps_folder):
	if f.lower().endswith(".bmp"):
		im = Image.open(os.path.join(maps_folder, f))
		# https://stackoverflow.com/a/29660960/1172196
		ground = 0
		water = 0
		tree = 0
		for pixel in im.getdata():
			if pixel == color_ground:
				ground += 1
			elif pixel == color_tree:
				tree += 1
			elif pixel == color_water:
				water += 1
			else:
				print "error, unknown color"
				sys.exit(1)
		print os.path.splitext(f)[0] + "\t" + str(ground) + "\t" + str(tree) + "\t" + str(water)