import os, glob, subprocess, platform\r
import wx\r
\r
+from util import profile\r
+from gui import toolbarUtil\r
+\r
try:\r
#Try to find the OpenCV library for video capture.\r
from opencv import cv\r
class webcam(object):\r
def __init__(self):\r
self._cam = None\r
+ self._overlayImage = toolbarUtil.getBitmapImage("cura-overlay.png")\r
+ self._overlayUltimaker = toolbarUtil.getBitmapImage("ultimaker-overlay.png")\r
if cv != None:\r
self._cam = highgui.cvCreateCameraCapture(-1)\r
elif win32vidcap != None:\r
if cv != None:\r
frame = cv.QueryFrame(self._cam)\r
cv.CvtColor(frame, frame, cv.CV_BGR2RGB)\r
- self._bitmap = wx.BitmapFromBuffer(frame.width, frame.height, frame.imageData)\r
+ bitmap = wx.BitmapFromBuffer(frame.width, frame.height, frame.imageData)\r
elif win32vidcap != None:\r
buffer, width, height = self._cam.getbuffer()\r
try:\r
wxImage.SetData(buffer[::-1])\r
if self._bitmap != None:\r
del self._bitmap\r
- self._bitmap = wxImage.ConvertToBitmap()\r
+ bitmap = wxImage.ConvertToBitmap()\r
del wxImage\r
del buffer\r
except:\r
pass\r
\r
+ dc = wx.MemoryDC()\r
+ dc.SelectObject(bitmap)\r
+ dc.DrawBitmap(self._overlayImage, bitmap.GetWidth() - self._overlayImage.GetWidth() - 5, 5, True)\r
+ if profile.getPreference('machine_type') == 'ultimaker':\r
+ dc.DrawBitmap(self._overlayUltimaker, (bitmap.GetWidth() - self._overlayUltimaker.GetWidth()) / 2, bitmap.GetHeight() - self._overlayUltimaker.GetHeight() - 5, True)\r
+ dc.SelectObject(wx.NullBitmap)\r
+\r
+ self._bitmap = bitmap\r
+\r
if self._doTimelaps:\r
filename = os.path.normpath(os.path.join(os.path.split(__file__)[0], "../__tmp_snap", "__tmp_snap_%04d.jpg" % (self._snapshotCount)))\r
self._snapshotCount += 1\r
- self._bitmap.SaveFile(filename, wx.BITMAP_TYPE_JPEG)\r
+ bitmap.SaveFile(filename, wx.BITMAP_TYPE_JPEG)\r
\r
return self._bitmap\r
\r