#Test pick centers
#Should give S points evenly spaced across the space
gridInterval = S.calcGridInterval(imgSpace.shape, K)
centers = S.pickCenters(K, gridInterval, imgSpace, imgSize)
print centers.shape


# print centers
X0 = centers[:,3]
Y0 = centers[:,4]
# print X,Y
plt.plot(X0,Y0,"rx")


#Test rectangle extraction
indexes = S.calcRectIndexes(imgSpace, centers[12], gridInterval)
rectangle = imgSpace[indexes]

X1 = rectangle[:,3]
Y1 = rectangle[:,4]
plt.plot(X1,Y1,"bx")
plt.plot(X0,Y0,"rx")

#Test distance calculation
distance = S.calcDistance(gridInterval, rectangle, centers[12])
print rectangle.shape
print distance.shape

#Test clustering
distances = S.calcAllDistances(imgSpace, centers, K, gridInterval)
print distances.shape