bohrhilfe.py

#!/usr/bin/python

# Es wird das Paket python-opencv benoetigt!

import cv

class DrillHelp:

    def __init__(self, camera_id = 0):

        self.mirror_image = True

        self.cross = (50, 50)

        cv.NamedWindow("camera", 1)

        cv.SetMouseCallback('camera', self._set_cross, None)

        self.capture = cv.CreateCameraCapture(camera_id)

        height = int(cv.GetCaptureProperty(self.capture, cv.CV_CAP_PROP_FRAME_HEIGHT))

        width = int(cv.GetCaptureProperty(self.capture, cv.CV_CAP_PROP_FRAME_WIDTH))

        self.img = cv.CreateImage( (width, height), 8, 3)

    def _set_cross(self, event, x, y, flags, foo):

        if event == cv.CV_EVENT_LBUTTONDOWN:

            self.cross = (x, y)

    def _draw_cross(self):

        cv.Circle(self.img, self.cross, 33, cv.RGB(255, 255, 255), 2)

        cv.Circle(self.img, self.cross, 13, cv.RGB(255, 255, 255), 1)

        cv.Line(self.img, (self.cross[0] - 20, self.cross[1]  -20), (self.cross[0] + 20, self.cross[1]  + 20), cv.RGB(0,200,0), 1)

        cv.Line(self.img, (self.cross[0] + 20, self.cross[1]  -20), (self.cross[0] - 20, self.cross[1]  + 20), cv.RGB(0,200,0), 1)

    def run(self):

        while True:

            frame = cv.QueryFrame(self.capture)

            if self.mirror_image:

                cv.Flip(frame, self.img, 0)

            else:    

                cv.Copy(frame, self.img)

            self._draw_cross()

            cv.ShowImage("camera", self.img)

            k = cv.WaitKey(10)

            if k != -1:

                break

        del(self.capture)

if __name__ == '__main__':

    d = DrillHelp()

    d.run()