from SimpleCV import Camera, VideoStream, Color, Display, Image, VirtualCamera

import cv2

import numpy as np

import time

import re

# deprecated, checks if point in the sphere is in our output

def isInROI(x,y,R1,R2,Cx,Cy):

return isInOuter and not isInInner

# build the mapping

def buildMap(Ws,Hs,Wd,Hd,R1,R2,Cx,Cy):

#return map1, map2

# do the unwarping

def unwarp(img,xmap,ymap):

return result

#disp = Display((800,600))

disp = Display((1296,972))

vals = []

last = (0,0)

# Load the video from the rpi

#vc = VirtualCamera("/var/www/html/vid_files/video.h264","video")

# Sometimes there is crud at the begining, buffer it out

#for i in range(0,10):

# img = vc.getImage()

img = Image("pi3.jpg")

img.save(disp)

# Show the user a frame let them left click the center

# of the "donut" and the right inner and outer edge

# in that order. Press esc to exit the display

while not disp.isDone():

test = disp.leftButtonDownPosition()

if( test != last and test is not None):

last = test

vals.append(test)

# 0 = xc yc

# 1 = r1

# 2 = r2

# center of the "donut"

Cx = vals[0][0]

Cy = vals[0][1]

#print str(Cx) + " " + str(Cy)

# Inner donut radius

R1x = vals[1][0]

R1y = vals[1][1]

R1 = R1x-Cx

#print str(R1)

# outer donut radius

R2x = vals[2][0]

R2y = vals[2][1]

R2 = R2x-Cx

#print str(R2)

# our input and output image siZes

Wd = round(float(max(R1, R2)) * 2.0 * np.pi)

#Wd = 2.0*((R2+R1)/2)*np.pi

#Hd = (2.0*((R2+R1)/2)*np.pi) * (90/360)

Hd = (R2-R1)

Ws = img.width

Hs = img.height

# build the pixel map, this could be sped up

print "BUILDING MAP!"

#print "HD value" + str(Hd)

#print "WD value" + str(Wd)

xmap,ymap = buildMap(Ws,Hs,Wd,Hd,R1,R2,Cx,Cy)

#np.savetxt('xmap.txt', xmap.flatten(), header=str(xmap.shape))

#np.savetxt('ymap.txt', ymap.flatten(), header=str(ymap.shape))

print "MAP DONE!"

# do an unwarping and show it to us

result = unwarp(img,xmap,ymap)

#result = result.adaptiveScale(resolution=(640,480))

#result = result.adaptiveScale(resolution=(1024,768))

result.save("pi_unwarped3.png")

# SimpleCV/OpenCV video out was giving problems

# decided to output frames and convert using

# avconv / ffmpeg. Comment out the block below

# to save to video

#ofname = 'OUT.AVI'

#vs = VideoStream(fps=20,filename=ofname,framefill=False)

#vs.initializeWriter((640,480))

# I used these params for converting the raw frames to video

# avconv -f image2 -r 30 -v:b 1024K -i samples/lapinsnipermin/%03d.jpeg output.mpeg

# i = 0

# while img is not None:

# print img.width,img.height

# result = unwarp(img,xmap,ymap)

# #derp = result.adaptiveScale(resolution=(640,480))

# #result = result.resize(w=img.width)

# # Once we get an image overlay it on the source

# #derp = img.blit(result,(0,img.height-result.height))

# #derp = derp.applyLayers()

# #derp = derp.resize(640,480)

# #result = result.adaptiveScale(resolution=(960,480))

# result.save(disp)

# # Save to file

# fname = "/var/www/html/vid_files/FRAME{num:05d}.png".format(num=i)

# result.save(fname)

# #vs.writeFrame(derp)

# # get the next frame

# img = vc.getImage()

# i = i + 1