def __init__(self, *args):
QtGui.QWidget.__init__(self, None)
self.ui = Ui_Form()
self.ui.setupUi(self)
self.camActive = False
Widget.desc = QVariant(unicode(i18n("Create Your User Picture")))
self.pictureTaken = 0
self.ui.takeButton.hide()
for dev in os.listdir("/dev"):
if dev.startswith("video"):
cam = v4l2capture.Video_device(os.path.join("/dev", dev))
cam_driver, cam_card, cam_bus, cam_capabilities = cam.get_info(
)
cam_str = "%s - %s" % (cam_card, os.path.join("/dev", dev))
if "video_capture" in cam_capabilities:
if "radio" in cam_capabilities or "tuner" in cam_capabilities:
continue
self.ui.comboBox.addItem(
cam_str, QVariant(os.path.join("/dev", dev)))
self.timer = QtCore.QTimer(self)
self.connect(self.timer, QtCore.SIGNAL("timeout()"), self.refreshCam)
self.connect(self.ui.comboBox, QtCore.SIGNAL('activated(QString)'),
self.processSelection)
self.connect(self.ui.takeButton, QtCore.SIGNAL('clicked()'),
self.showPicture)
self.connect(self.ui.takeAgainButton, QtCore.SIGNAL('clicked()'),
self.activateCam)
self.ui.takeAgainButton.hide()
class Widget(QtGui.QWidget, Screen):
title = i18n("User Picture")
screenSettings = {}
screenSettings["hasChanged"] = False
def __init__(self, *args):
QtGui.QWidget.__init__(self, None)
self.ui = Ui_Form()
self.ui.setupUi(self)
self.camActive = False
Widget.desc = QVariant(unicode(i18n("Create Your User Picture")))
self.pictureTaken = 0
self.ui.takeButton.hide()
for dev in os.listdir("/dev"):
if dev.startswith("video"):
cam = v4l2capture.Video_device(os.path.join("/dev", dev))
cam_driver, cam_card, cam_bus, cam_capabilities = cam.get_info(
)
cam_str = "%s - %s" % (cam_card, os.path.join("/dev", dev))
if "video_capture" in cam_capabilities:
if "radio" in cam_capabilities or "tuner" in cam_capabilities:
continue
self.ui.comboBox.addItem(
cam_str, QVariant(os.path.join("/dev", dev)))
self.timer = QtCore.QTimer(self)
self.connect(self.timer, QtCore.SIGNAL("timeout()"), self.refreshCam)
self.connect(self.ui.comboBox, QtCore.SIGNAL('activated(QString)'),
self.processSelection)
self.connect(self.ui.takeButton, QtCore.SIGNAL('clicked()'),
self.showPicture)
self.connect(self.ui.takeAgainButton, QtCore.SIGNAL('clicked()'),
self.activateCam)
self.ui.takeAgainButton.hide()
def processSelection(self):
if self.ui.comboBox.currentIndex() == 1:
self.chooseFile()
if self.ui.comboBox.currentIndex() > 1:
self.activateCam()
def activateCam(self):
self.drawCrop = DrawCropMask(self.ui.camGoruntu)
self.camActive = True
self.ui.takeButton.show()
self.ui.takeAgainButton.hide()
self.timer.stop()
cam = str(
self.ui.comboBox.itemData(
self.ui.comboBox.currentIndex()).toString())
self.video = v4l2capture.Video_device(cam)
self.size_x, self.size_y = self.video.set_format(320, 240)
self.video.create_buffers(10)
self.video.queue_all_buffers()
self.video.start()
select.select((self.video, ), (), ())
self.timer.setInterval(30)
self.timer.start()
def refreshCam(self):
try:
self.image_data = self.video.read_and_queue()
self.image_raw = Image.fromstring("RGB",
(self.size_x, self.size_y),
self.image_data)
self.image = ImageQt.ImageQt(self.image_raw).mirrored(True, False)
self.ui.camGoruntu.setPixmap(QtGui.QPixmap.fromImage(self.image))
self.updateCropMask()
except:
pass
def showPicture(self):
self.__class__.screenSettings["hasChanged"] = True
self.camActive = False
self.deactivateCam()
self.ui.takeAgainButton.show()
self.ui.takeButton.hide()
self.savePicture()
def deactivateCam(self):
self.timer.stop()
self.video.close()
def savePicture(self, picturePath=None):
home = os.path.expanduser("~")
if picturePath:
self.picturePath = str(picturePath)
else:
self.pictureTaken = 1
self.picturePath = home + "/temp.jpg"
self.image.save(self.picturePath)
self.foto = Image.open(self.picturePath)
self.space = (85, 70, 235, 195)
self.paste = self.foto.crop(self.space)
self.paste.save(home + "/temp.jpg")
os.rename(home + "/temp.jpg", home + "/.face.icon")
def chooseFile(self):
self.ui.takeButton.hide()
self.__class__.screenSettings["hasChanged"] = True
if self.camActive == True:
self.deactivateCam()
self.selectedFile = QFileDialog.getOpenFileName(
None, "Open Image", os.environ["HOME"],
'Image Files (*.png *.jpg *bmp)')
if self.selectedFile.isNull():
return
else:
home = os.path.expanduser("~")
self.ui.camGoruntu.setPixmap(QtGui.QPixmap(str(self.selectedFile)))
self.ui.camGoruntu.update()
self.savePicture(self.selectedFile)
def shown(self):
pass
def execute(self):
if not self.pictureTaken == 1:
self.timer.stop()
try:
self.video.close()
except:
pass
return True
