from picamera import PiCamera
from gpiozero import Button
from mcpi.minecraft import Minecraft
from time import sleep
from skimage import io, color

mc = Minecraft.create()

button = Button(4, pull_up=False)

## Taking a picture with Micro:bit

mc.postToChat("Press the Button!")
sleep(2)
button.wait_for_press()


with PiCamera() as camera:
    camera.resolution = (100,80)
    camera.start_preview()
    sleep(3)
    camera.capture('/home/pi/Desktop/Microbit/selfie.jpg')
    camera.stop_preview()
    mc.postToChat('fly up in air and look for shadow on the ground')
    sleep(5)


## Rendering the picture

### load selfie and map
selfie_rgb = io.imread("/home/pi/Desktop/Microbit/selfie.jpg")
map_rgb = io.imread("/home/pi/Desktop/Microbit/colour_map.png")


### Convert to Lab

selfie_lab = color.rgb2lab(selfie_rgb)
map_lab = color.rgb2lab(map_rgb)

### Mapping colours on colour map to Minecraft blocks
### First tuple is coordinates of colour map
### Second tuple is Minecraft block

colours={(0,0):(2,0),(0,1):(3,0),(0,2):(4,0),(0,3):(5,0),(0,4):(7,0),(0,5):(14,0),(0,6):(15,0),(1,0):(16,0),(1,1):(17,0),(1,2):(21,0),(1,3):(22,0),(1,4):(24,0),(1,5):(35,0),(1,6):(35,1),(2,0):(35,2),(2,1):(35,3),(2,2):(35,4),(2,3):(35,5),(2,4):(35,6),(2,5):(35,7),(2,6):(35,8),(3,0):(35,9),(3,1):(35,10),(3,2):(35,11),(3,3):(35,12),(3,4):(35,13),(3,5):(35,14),(3,6):(35,15),(4,0):(41,0),(4,1):(42,0),(4,2):(43,0),(4,3):(45,0),(4,4):(46,0),(4,5):(47,0),(4,6):(48,0),(5,0):(49,0),(5,1):(54,0),(5,2):(56,0),(5,3):(57,0),(5,4):(58,0),(5,5):(60,0),(5,6):(61,0),(6,0):(73,0),(6,1):(79,0),(6,2):(80,0),(6,3):(82,0),(6,4):(89,0),(6,5):(103,0),(6,6):(246,0)}

## Iterate over image and then over map. Find closest colour from map, and then look up that block and place

mc = Minecraft.create()
x, y, z = mc.player.getPos()

for i, selfie_column in enumerate(selfie_lab):
        for j, selfie_pixel in enumerate(selfie_column):
                distance = 300
                for k, map_column in enumerate(map_lab):
                        for l, map_pixel in enumerate(map_column):
                                delta = color.deltaE_ciede2000(selfie_pixel,map_pixel)
                                if delta < distance:
                                        distance = delta
                                        block = colours[(k,l)]
                mc.setBlock(x-j, y-i+60, z+5, block[0], block[1])