def __init__(self): pygame.init() print "Pygame Version: " + pygame.version.ver self.mDisplaySurface = None self.mTicker = pygame.time.Clock() self.mImageManager = ImageManager(self) self.mSoundManager = SoundManager(self)
def restart(imagedir, seeddir):
#delete all of the images
print("Deleting images in the " + imagedir + " directory.")
print("Copying images in the " + seeddir + " directory to the " +
imagedir + " directory.")
cont = input("Continue (y/n)? ")
cont = cont.lower()
if (cont != "y" and cont != "yes"):
print("Aborting restart\n")
sys.exit(0)
imageman = ImageManager(imagedir, seeddir)
imageman.restart()
def open(self):
fileName = QtGui.QFileDialog.getOpenFileName(self)
if fileName:
with open(str(fileName)) as fp:
jsondict = json.load(fp)
self.im = ImageManager(jsondict)
for i in range(self.im.image_num):
self.createSubImageViewer(self.im.images[i].id,
self.im.images[i].pixmap,
self.im.images[i].labeled_pixels,
self.im.label_colors)
def communicateWithPeers(centralServerName, centralServerPort, peerServerPort, \
peerServerID, imagedir):
global centralServer
global peerServer
#connect to central server to add host and port to peer list
centralServer = ContactCentralServer(centralServerName, centralServerPort, \
peerServerPort, peerServerID)
response = centralServer.connect()
if (response.find("BAD") != -1):
print(response[4:])
print("Exiting")
sys.exit(0)
#Create an image manager object that will be passed to
#both the peer client and the peer server.
#The peer client will request images from a peer server.
#The peer server will return requested images to a peer client.
imageman = ImageManager(imagedir, "")
peerClient = PeerClient(centralServer, peerServerPort, imageman)
print("Peer server listening on port " + peerServerPort + ".\n")
peerClient.start() # start peer client thread
peerServer = PeerServer(peerServerPort, imageman)
peerServer.start() # start peer server thread
return peerClient
p2 = newpoints2[i]
#print str(p1) + " -- " + str(p2)
cv2.circle(cimg1, (p1[0] - point_radius, p1[1] - point_radius),
point_radius, rndColors[i], point_radius)
cv2.circle(cimg2, (p2[0] - point_radius, p2[1] - point_radius),
point_radius, rndColors[i], point_radius)
#cv2.imshow(filename1, cimg1)
cv2.imwrite(filename1 + "_l.jpg", cimg1)
#cv2.imshow(filename2, cimg2)
cv2.imwrite(filename2 + "_l.jpg", cimg2)
#cv2.waitKey(0)
print "matched len " + str(len(newpoints1))
numiters = 2000
im = ImageManager(None)
H, best_set = im.ransac2(img1, newpoints1, img2, newpoints2, numiters, 20)
#H = self.ransac(matchedpoints1, matchedpoints2, numiters)
print H
points_num = len(best_set)
rndColors = []
for i in range(points_num):
rndVals = np.random.randint(0, 255, 3)
rndColors.append((rndVals[0], rndVals[1], rndVals[2]))
point_radius = 20
cimg1 = cv2.cvtColor(img1, cv2.COLOR_GRAY2BGR)
cimg2 = cv2.cvtColor(img2, cv2.COLOR_GRAY2BGR)
for i in range(points_num):
#print best_set[i]
def _create(self):
open_canvas(self.WINDOW_WIDTH, self.WINDOW_HEIGHT, self.CAPTION)
self._m_ImageManager = ImageManager.ImageManager()
self._m_SoundManager = SoundManager.SoundManager()
self._build_scene()
self._m_bRun = True
selection = ''
# continue to prompt until user quits with '0' or 'q'
while (selection != '0') and (selection != 'q'):
print(
"\nPlease enter a number of the photo you would like to process (0 or q to quit):"
)
for key, value in photoFileDict.items():
photoOptions = photoFileDict[key]['fileName']
print(key, photoOptions)
selection = input("What is your choice? ")
if (selection != '0') and (selection != 'q'):
# ensure entered value is valid
if ((selection.isdigit() == False)
or (int(selection) > numberOfPhotos)):
print("INVALID ENTRY!")
else:
# valid selection, so parse the .jpg file selected
parsedImage = im.ImageManager(
photoFileDict[int(selection)]['filePath'])
parsedImage.parseImage()
print("Camera Make:", parsedImage.cameraMake)
print("Camera Model:", parsedImage.cameraModel)
print("Endian:", parsedImage.endian)
else:
print("No Jpg Photos Exist. Aborting")
except Exception as err:
print("Unknown error in ImageParser::Main: {0}".format(err))
im.ImageManager.printMoreErrInfo(err)
def onParseBtnPress(self, event):
if (self.bitmap.IsShown() == True):
parsedImage = im.ImageManager(self.fullPath)
parsedImage.parseImage()
self.makeTxt.SetLabel("Camera Make: " + parsedImage.cameraMake)
self.modelTxt.SetLabel("Camera Model: " + parsedImage.cameraModel)