def __init__(self, options):
FrameProcessor.__init__(self, options)
self.windowWidth = options.get('windowWidth', 640)
self.windowHeight = options.get('windowHeight', 480)
# * Initialize OpenGL texture and framebuffer used to render camera images
# NOTE: A valid OpenGL context must available at this point
self.texOutId = glGenTextures(1)
glBindTexture(GL_TEXTURE_2D, self.texOutId)
#glPixelStorei(GL_UNPACK_ALIGNMENT, 1) # image data is not padded (?)
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR) # GL_NEAREST
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR) # GL_NEAREST
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE) # GL_REPEAT
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE) # GL_REPEAT
self.framebufferId = glGenFramebuffers(1)
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, self.framebufferId)
glFramebufferTexture2D(GL_DRAW_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, self.texOutId, 0)
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0)
def __init__(self, options):
FrameProcessor.__init__(self, options)
self.windowWidth = options.get('windowWidth', 640)
self.windowHeight = options.get('windowHeight', 480)
# * Initialize OpenGL texture and framebuffer used to render camera images
# NOTE: A valid OpenGL context must available at this point
self.texOutId = glGenTextures(1)
glBindTexture(GL_TEXTURE_2D, self.texOutId)
#glPixelStorei(GL_UNPACK_ALIGNMENT, 1) # image data is not padded (?)
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER,
GL_LINEAR) # GL_NEAREST
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER,
GL_LINEAR) # GL_NEAREST
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S,
GL_CLAMP_TO_EDGE) # GL_REPEAT
glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T,
GL_CLAMP_TO_EDGE) # GL_REPEAT
self.framebufferId = glGenFramebuffers(1)
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, self.framebufferId)
glFramebufferTexture2D(GL_DRAW_FRAMEBUFFER, GL_COLOR_ATTACHMENT0,
GL_TEXTURE_2D, self.texOutId, 0)
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0)
def __init__(self, options):
FrameProcessor.__init__(self, options)
def __init__(self, options):
FrameProcessor.__init__(self, options)
if self.debug:
self.loop_delay = 0.025 # set some delay when debugging, in case we are running a video
self.state = ExposureNormalizer.State.NONE # set to NONE here, call start() to run through once
def __init__(self, options):
FrameProcessor.__init__(self, options)
self.filterBankFilename = options.get('filter_bank', self.defaultFilterBankFilename)
#self.debug = True # overriding debug flag
self.state = self.State.INIT
def __init__(self, options):
FrameProcessor.__init__(self, options)
self.filterBankFilename = options.get('filter_bank',
self.defaultFilterBankFilename)
#self.debug = True # overriding debug flag
self.state = self.State.INIT
def run(processor=FrameProcessor(options={
'gui': True,
'debug': True
}),
gui=True,
debug=True): # default options
"""Run a FrameProcessor object on a static image (repeatedly) or on frames from a camera/video."""
# TODO Use VideoInput instance instead of duplicating input logic
# * Initialize parameters and flags
delay = 10 # ms
delayS = delay / 1000.0 # sec; only used in non-GUI mode, so this can be set to 0
#gui = options.get('gui', True)
#debug = options.get('debug', True)
showInput = gui
showOutput = gui
showFPS = False
showKeys = False
isImage = False
isVideo = False
isOkay = False
isFrozen = False
# * Setup logging
logging.basicConfig(
format="%(levelname)s | %(module)s | %(funcName)s() | %(message)s",
level=logging.DEBUG if debug else logging.INFO)
# * Read input image or video, if specified
if len(sys.argv) > 1:
filename = sys.argv[1]
if isImageFile(filename):
print "run(): Reading image: \"" + filename + "\""
frame = cv2.imread(filename)
if frame is not None:
if showInput:
cv2.imshow("Input", frame)
isImage = True
isOkay = True
else:
print "run(): Error reading image; fallback to camera."
else:
print "run(): Reading video: \"" + filename + "\""
camera = cv2.VideoCapture(filename)
if camera.isOpened():
isVideo = True
isOkay = True
else:
print "run(): Error reading video; fallback to camera."
# * Open camera if image/video is not provided/available
if not isOkay:
print "run(): Opening camera..."
camera = cv2.VideoCapture(0)
# ** Final check before processing loop
if camera.isOpened():
result_width = camera.set(cv.CV_CAP_PROP_FRAME_WIDTH, cameraWidth)
result_height = camera.set(cv.CV_CAP_PROP_FRAME_HEIGHT,
cameraHeight)
print "run(): Camera frame size set to {width}x{height} (result: {result_width}, {result_height})".format(
width=cameraWidth,
height=cameraHeight,
result_width=result_width,
result_height=result_height)
isOkay = True
else:
print "run(): Error opening camera; giving up now."
return
# * Initialize supporting variables
fresh = True
# * Processing loop
timeStart = cv2.getTickCount() / cv2.getTickFrequency()
timeLast = timeNow = 0.0
while (1):
# ** [timing] Obtain relative timestamp for this loop iteration
timeNow = (cv2.getTickCount() / cv2.getTickFrequency()) - timeStart
if showFPS:
timeDiff = (timeNow - timeLast)
fps = (1.0 / timeDiff) if (timeDiff > 0.0) else 0.0
print "run(): {0:5.2f} fps".format(fps)
# ** If not static image, read frame from video/camera
if not isImage and not isFrozen:
isValid, frame = camera.read()
if not isValid:
break # camera disconnected or reached end of video
if showInput:
cv2.imshow("Input", frame)
# ** Initialize FrameProcessor, if required
if (fresh):
processor.initialize(
frame, timeNow) # timeNow should be zero on initialize
fresh = False
# ** Process frame
imageOut = processor.process(frame, timeNow)
# ** Show output image
if showOutput and imageOut is not None:
cv2.imshow("Output", imageOut)
# ** Check if GUI is available
if gui:
# *** If so, wait for inter-frame delay and process keyboard events using OpenCV
key = cv2.waitKey(delay)
if key != -1:
keyCode = key & 0x00007f # key code is in the last 8 bits, pick 7 bits for correct ASCII interpretation (8th bit indicates
keyChar = chr(keyCode) if not (
key & KeyCode.SPECIAL
) else None # if keyCode is normal, convert to char (str)
if showKeys:
print "run(): Key: " + KeyCode.describeKey(key)
#print "run(): key = {key:#06x}, keyCode = {keyCode}, keyChar = {keyChar}".format(key=key, keyCode=keyCode, keyChar=keyChar)
if keyCode == 0x1b or keyChar == 'q':
break
elif keyChar == ' ':
print "run(): [PAUSED] Press any key to continue..."
ticksPaused = cv2.getTickCount(
) # [timing] save time when paused
cv2.waitKey() # wait indefinitely for a key press
timeStart += (cv2.getTickCount() -
ticksPaused) / cv2.getTickFrequency(
) # [timing] compensate for duration paused
elif keyCode == 0x0d:
isFrozen = not isFrozen # freeze frame, but keep processors running
elif keyChar == 'f':
showFPS = not showFPS
elif keyChar == 'k':
showKeys = not showKeys
elif keyChar == 'i':
showInput = not showInput
if not showInput:
cv2.destroyWindow("Input")
elif keyChar == 'o':
showOutput = not showOutput
if not showOutput:
cv2.destroyWindow("Output")
elif not processor.onKeyPress(key, keyChar):
break
else:
# *** Else, wait for inter-frame delay using system method
sleep(delayS)
# ** [timing] Save timestamp for fps calculation
timeLast = timeNow
# * Clean-up
print "run(): Cleaning up..."
if gui:
cv2.destroyAllWindows()
if not isImage:
camera.release()
