python and firware code added
This commit is contained in:
27
src/python/toQImage.py
Normal file
27
src/python/toQImage.py
Normal file
@@ -0,0 +1,27 @@
|
||||
from PyQt5.QtGui import QImage, qRgb
|
||||
import numpy as np
|
||||
|
||||
class NotImplementedException:
|
||||
pass
|
||||
|
||||
gray_color_table = [qRgb(i, i, i) for i in range(256)]
|
||||
|
||||
def toQImage(im, copy=False):
|
||||
if im is None:
|
||||
return QImage()
|
||||
|
||||
if im.dtype == np.uint8:
|
||||
if len(im.shape) == 2:
|
||||
qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_Indexed8)
|
||||
qim.setColorTable(gray_color_table)
|
||||
return qim.copy() if copy else qim
|
||||
|
||||
elif len(im.shape) == 3:
|
||||
if im.shape[2] == 3:
|
||||
qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_RGB888);
|
||||
return qim.copy() if copy else qim
|
||||
elif im.shape[2] == 4:
|
||||
qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_ARGB32);
|
||||
return qim.copy() if copy else qim
|
||||
|
||||
#raise NotImplementedException
|
||||
Reference in New Issue
Block a user