def test_gray2qimage():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
qImg = qimage2ndarray.gray2qimage(a)
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(42,42,42)))
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(0,0,0)))
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(0,0,0)))
def test_gray2qimage():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
qImg = qimage2ndarray.gray2qimage(a)
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(42,42,42)))
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(0,0,0)))
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(0,0,0)))
def test_gray2qimage_normalize_domain():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
qImg = qimage2ndarray.gray2qimage(a, normalize = (-100, 100))
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
x = int(255 * 142.42 / 200.0)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(x,x,x)))
x = int(255 * 90.0 / 200.0)
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(x,x,x)))
x = int(255 * 100.0 / 200.0)
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(x,x,x)))
def test_gray2qimage_normalize_domain():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
qImg = qimage2ndarray.gray2qimage(a, normalize = (-100, 100))
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
x = int(255 * 142.42 / 200.0)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(x,x,x)))
x = int(255 * 90.0 / 200.0)
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(x,x,x)))
x = int(255 * 100.0 / 200.0)
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(x,x,x)))
def test_gray2qimage_masked():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
a[:,160:] = 100
a = numpy.ma.masked_greater(a, 99)
qImg = qimage2ndarray.gray2qimage(a, normalize = True)
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(255,255,255)))
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(0,0,0)))
x = int(255 * 10.0 / 52.42)
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(x,x,x)))
assert_equal(QtGui.qAlpha(qImg.pixel(0,10)), 255)
assert_equal(QtGui.qAlpha(qImg.pixel(200,10)), 0)
def test_gray2qimage_masked():
a = numpy.zeros((240, 320), dtype = float)
a[12,10] = 42.42
a[13,10] = -10
a[:,160:] = 100
a = numpy.ma.masked_greater(a, 99)
qImg = qimage2ndarray.gray2qimage(a, normalize = True)
assert not qImg.isNull()
assert_equal(qImg.width(), 320)
assert_equal(qImg.height(), 240)
assert_equal(qImg.format(), QtGui.QImage.Format_Indexed8)
assert_equal(a.nbytes, numBytes(qImg) * a.itemsize)
assert_equal(numColors(qImg), 256)
assert_equal(hex(qImg.pixel(10,12)), hex(QtGui.qRgb(255,255,255)))
assert_equal(hex(qImg.pixel(10,13)), hex(QtGui.qRgb(0,0,0)))
x = int(255 * 10.0 / 52.42)
assert_equal(hex(qImg.pixel(10,14)), hex(QtGui.qRgb(x,x,x)))
assert_equal(QtGui.qAlpha(qImg.pixel(0,10)), 255)
assert_equal(QtGui.qAlpha(qImg.pixel(200,10)), 0)
