def __init__(self, directory, CELLS_PER_ROW, CELLS_PER_COLUMN):
self.directory = directory
self.images = []
self.cpr = CELLS_PER_ROW
self.cpc = CELLS_PER_COLUMN
self.histograms = [[numpy.zeros(256) for i in xrange(self.cpc)] for i in xrange(self.cpr)]
jpgs = glob.glob("%s/*.jpg" % self.directory)
print "Found %s images" % len(jpgs)
default = pImage(jpgs[0])
default.makeDefault()
default.cellify(self.cpr, self.cpc)
self.generateLuminance(default)
default.classifyCells(True)
default.setBoardArea()
default.free()
self.images = [0 for x in xrange(len(jpgs))]
for index in xrange(len(jpgs)):
pImg = pImage(jpgs[index])
pImg.cellify(self.cpr, self.cpc)
self.images[index] = pImg
def getStartingKeyImage(self):
keyImage = pImage(self.images[0].filename)
keyImage.cellify(self.cpr, self.cpc)
keyImage.classifyCells(True)
return keyImage
