Source code for pynq.lib.video.common

#   Copyright (c) 2018, Xilinx, Inc.
#   SPDX-License-Identifier: BSD-3-Clause



[docs]class VideoMode: """Class for holding the information about a video mode Attributes ---------- height : int Height of the video frame in lines width : int Width of the video frame in pixels stride : int Width of a line in the video frame in bytes bits_per_pixel : int Bits per pixel bytes_per_Pixel : int Bytes required to represent each pixel shape : tuple of int Numpy-style tuple describing the video frame """ def __init__(self, width, height, bits_per_pixel, fps=60, stride=None): self.width = width self.height = height self.bits_per_pixel = bits_per_pixel self.bytes_per_pixel = ((bits_per_pixel - 1) // 8) + 1 self.fps = fps if stride: self.stride = stride else: self.stride = width * self.bytes_per_pixel if self.bytes_per_pixel == 1: self.shape = (self.height, self.width) else: self.shape = (self.height, self.width, self.bytes_per_pixel) def __repr__(self): return ("VideoMode: width={} height={} bpp={} fps={}" .format(self.width, self.height, self.bits_per_pixel, self.fps)) def __eq__(self, mode): return self.width == mode.width and self.height == mode.height and \ self.bits_per_pixel == mode.bits_per_pixel and \ self.fps == mode.fps and self.stride == mode.stride
[docs]class PixelFormat: """Wrapper for all of the information about a video format Attributes ---------- bits_per_pixel : int Number of bits for each pixel in_color : list of float Coefficients from BGR stream to pixel format out_color : list of float Coefficient from pixel format to BGR stream """ def __init__(self, bits_per_pixel, in_color, out_color, fourcc=None): self.bits_per_pixel = bits_per_pixel self.in_color = in_color self.out_color = out_color self.fourcc = fourcc
COLOR_IN_BGR = [1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0] COLOR_OUT_BGR = [1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0] COLOR_IN_RGB = [0, 0, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0] COLOR_OUT_RGB = [0, 0, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0] COLOR_IN_YCBCR = [0.114, 0.587, 0.299, 0.5, -0.331264, -0.168736, -0.081312, -0.41866, 0.5, 0, 0.5, 0.5] COLOR_OUT_YCBCR = [1, 1.772, 0, 1, -0.3344136, -0.714136, 1, 0, 1.402, -0.886, 0.529136, -0.701] COLOR_OUT_GRAY = [1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0] PIXEL_RGB = PixelFormat(24, COLOR_IN_RGB, COLOR_OUT_RGB, "RG24") PIXEL_RGBA = PixelFormat(32, COLOR_IN_RGB, COLOR_OUT_RGB, "RA24") PIXEL_BGR = PixelFormat(24, COLOR_IN_BGR, COLOR_OUT_BGR, "BG24") PIXEL_YCBCR = PixelFormat(24, COLOR_IN_YCBCR, COLOR_OUT_YCBCR, "YU24") PIXEL_GRAY = PixelFormat(8, COLOR_IN_YCBCR, COLOR_OUT_GRAY)