convert opencv image to pyglet image

cv2glet.py

"""
convert opencv image to pyglet image

"""  
import pyglet
import cv2
from PIL import Image
import numpy as np

def cv2glet(img,format):
    '''Assumes image is in BGR color space. Returns a pyimg object'''
    if format == 'GRAY':
      rows, cols = img.shape
      channels = 1
    else:
      rows, cols, channels = img.shape

    raw_img = Image.fromarray(img).tobytes()

    top_to_bottom_flag = -1
    bytes_per_row = channels*cols
    pyimg = pyglet.image.ImageData(width=cols, 
                                   height=rows, 
                                   format=format, 
                                   data=raw_img, 
                                   pitch=top_to_bottom_flag*bytes_per_row)
    return pyimg