# -*- coding: utf-8 -*-
#!/usr/bin/python
from PIL import Image
import sys

relief = Image.open("data/default.png") # Карта высот
mask = Image.new("RGBA", (512,512), (0,0,0,0)) # Создаем новый файл изображения, с 4-мя каналами.
rgba = mask.split() # Разбиваем на каналы

red = rgba[0] # Переменная для красного канала
green = rgba[1] # Переменная для зеленого канала   
blue = rgba[2] # Переменная для синего канала
alfa = rgba[3] # Переменная для альфа канала

for y in range(512):
    for x in range(512):
          # Определяем уровень песка от 0 до 75
          if 0 <= relief.getpixel((x, y)) < 75:
              red.putpixel((x, y), 255 )
          # Определяем уровень травы от 75 до 120
          if 74 < relief.getpixel((x, y)) < 80:
              green.putpixel((x, y), 255 )
          # Определяем уровень скал от 75 до 170
          if 79 < relief.getpixel((x, y)) < 140:
              blue.putpixel((x, y), 255 )
          # Определяем уровень снега от 170 до 255
          if 139 < relief.getpixel((x, y)) < 255:
              alfa.putpixel((x, y), 255 )

im = Image.merge("RGBA", (red,green,blue,alfa)) # Собираем слои
im.save("data/default_d.png") # И сохраняем

    def scan1(self, task):
         for x in range(512):
             for y in range(512):
                 n = self.terrain.getNormal(x,y)
                 print Vec3(0,1,0).dot(n)
                 if 0.001 < Vec3(0,1,0).dot(n) < 0.01:  
                     self.red.putpixel((x, y), 0 )
                     self.green.putpixel((x, y), 0 )
                     self.blue.putpixel((x, y), 0 )
                     self.alfa.putpixel((x, y), 255 )

n = self.terrain.getNormal(x,y)
n.normalize()
dot = abs(Vec3(0,1,0).dot(n))

        self.terrain = GeoMipTerrain("Terrain")
         self.terrain.setHeightfield(Filename('data/land01-map.png'))
         self.terrain.setBlockSize(32)
         self.terrain.setFactor(100)
         #self.terrain.setFocalPoint(base.camera)
         self.terrain.getRoot().reparentTo(render)
         self.terrain.getRoot().setPos(0, 0, -500)
         self.terrain.getRoot().setSz(200)
         self.terrain.generate()

self.mask = Image.new("RGBA", (512,512), (0,0,255,0)) # Создаем новый файл изображения, с 4-мя каналами.

    def scan1(self, task):
         for x in range(512):
             for y in range(512):
                 n = self.terrain.getNormal(x,y)
                 n = Vec3(n.x*150, n.y*150, n.z) # 150 - масштаб террайна по высоте
                 n.normalize()
                 dot = abs(Vec3(0,0,1).dot(n))
                 r,g,b,a = self.mask.getpixel((x,y))
                 a = 255 - min(255, int(255 * dot ** 4 * 1.5)) # последние два коэффициента влияют на вертикальность и яркость прорисовки камней
                 r = max(0, r-a)
                 g = max(0, g-a)
                 b = max(0, b-a)
                 self.mask.putpixel((x, y), (r,g,b,a))
         self.mask.save("data/default_d.png")
         sys.exit()