YCbCr viewer

rgb.py

#!/usr/bin/env python
import pygame
import Image
import sys

W = 352
H = 288
WH = (W, H)

pygame.init()
screen = pygame.display.set_mode(WH)

fd = open('foreman.yuv', 'rb')
y = fd.read(W * H)
u = fd.read(W * H / 4)
v = fd.read(W * H / 4)

# Use PIL to get a black-n-white version of the YCbCr
# and convert it to RGB
im = Image.fromstring("L",WH, y)
rgb = im.convert('RGB')
bin_str = im.tostring()

# Also tried to skip PIL and use
#s = pygame.image.fromstring(y, WH, 'P').convert()

# no errors or anything, but all black is displayed

s = pygame.image.fromstring(bin_str, WH, 'RGB').convert()
print s

screen.blit(s, (0,0))
pygame.display.flip()