mirror of
https://github.com/panda3d/panda3d.git
synced 2025-10-04 10:54:24 -04:00
137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
from panda3d.core import Texture, PNMImage, LColor
|
|
from array import array
|
|
import math
|
|
|
|
|
|
def image_from_stored_pixel(component_type, format, data):
|
|
""" Creates a 1-pixel texture with the given settings and pixel data,
|
|
then returns a PNMImage as result of calling texture.store(). """
|
|
|
|
tex = Texture("")
|
|
tex.setup_1d_texture(1, component_type, format)
|
|
tex.set_ram_image(data)
|
|
|
|
img = PNMImage()
|
|
assert tex.store(img)
|
|
return img
|
|
|
|
|
|
def peek_tex_with_clear_color(component_type, format, clear_color):
|
|
""" Creates a 1-pixel texture with the given settings and clear color,
|
|
then peeks the value at this pixel and returns it. """
|
|
|
|
tex = Texture("")
|
|
tex.setup_1d_texture(1, component_type, format)
|
|
tex.set_clear_color(clear_color)
|
|
tex.make_ram_image()
|
|
|
|
col = LColor()
|
|
tex.peek().fetch_pixel(col, 0, 0)
|
|
return col
|
|
|
|
|
|
def test_texture_store_unsigned_byte():
|
|
data = array('B', (2, 1, 0, 0xff))
|
|
img = image_from_stored_pixel(Texture.T_unsigned_byte, Texture.F_rgba, data)
|
|
|
|
assert img.maxval == 0xff
|
|
pix = img.get_pixel(0, 0)
|
|
assert tuple(pix) == (0, 1, 2, 0xff)
|
|
|
|
|
|
def test_texture_store_unsigned_short():
|
|
data = array('H', (2, 1, 0, 0xffff))
|
|
img = image_from_stored_pixel(Texture.T_unsigned_short, Texture.F_rgba, data)
|
|
|
|
assert img.maxval == 0xffff
|
|
pix = img.get_pixel(0, 0)
|
|
assert tuple(pix) == (0, 1, 2, 0xffff)
|
|
|
|
|
|
def test_texture_store_unsigned_int():
|
|
data = array('I', (0x30000, 2, 0, 0xffffffff))
|
|
img = image_from_stored_pixel(Texture.T_unsigned_int, Texture.F_rgba, data)
|
|
|
|
assert img.maxval == 0xffff
|
|
pix = img.get_pixel(0, 0)
|
|
assert tuple(pix) == (0, 0, 3, 0xffff)
|
|
|
|
|
|
def test_texture_store_float():
|
|
data = array('f', (0.5, 0.0, -2.0, 10000.0))
|
|
img = image_from_stored_pixel(Texture.T_float, Texture.F_rgba, data)
|
|
|
|
assert img.maxval == 0xffff
|
|
col = img.get_xel_a(0, 0)
|
|
assert col.almost_equal((0.0, 0.0, 0.5, 1.0), 1 / 255.0)
|
|
|
|
|
|
def test_texture_store_half():
|
|
# Python's array class doesn't support half floats, so we hardcode the
|
|
# binary representation of these numbers:
|
|
data = array('H', (
|
|
# -[exp][mantissa]
|
|
0b0011110000000000, # 1.0
|
|
0b1100000000000000, # -2.0
|
|
0b0111101111111111, # 65504.0
|
|
0b0011010101010101, # 0.333251953125
|
|
))
|
|
img = image_from_stored_pixel(Texture.T_half_float, Texture.F_rgba, data)
|
|
|
|
assert img.maxval == 0xffff
|
|
col = img.get_xel_a(0, 0)
|
|
assert col.almost_equal((1.0, 0.0, 1.0, 0.333251953125), 1 / 255.0)
|
|
|
|
|
|
def test_texture_store_srgb():
|
|
# 188 = roughly middle gray
|
|
data = array('B', [188, 188, 188])
|
|
img = image_from_stored_pixel(Texture.T_unsigned_byte, Texture.F_srgb, data)
|
|
|
|
# We allow some imprecision.
|
|
assert img.maxval == 0xff
|
|
col = img.get_xel(0, 0)
|
|
assert col.almost_equal((0.5, 0.5, 0.5), 1 / 255.0)
|
|
|
|
|
|
def test_texture_store_srgb_alpha():
|
|
# 188 = middle gray
|
|
data = array('B', [188, 188, 188, 188])
|
|
img = image_from_stored_pixel(Texture.T_unsigned_byte, Texture.F_srgb_alpha, data)
|
|
|
|
# We allow some imprecision.
|
|
assert img.maxval == 0xff
|
|
col = img.get_xel_a(0, 0)
|
|
assert col.almost_equal((0.5, 0.5, 0.5, 188 / 255.0), 1 / 255.0)
|
|
|
|
|
|
def test_texture_clear_unsigned_byte():
|
|
col = peek_tex_with_clear_color(Texture.T_unsigned_byte, Texture.F_rgba, (0, 1 / 255.0, 254 / 255.0, 1.0))
|
|
assert col == LColor(0, 1 / 255.0, 254 / 255.0, 1.0)
|
|
|
|
|
|
def test_texture_clear_float():
|
|
col = peek_tex_with_clear_color(Texture.T_float, Texture.F_rgba, (0, 0.25, -0.5, 2))
|
|
assert col == LColor(0, 0.25, -0.5, 2)
|
|
|
|
|
|
def test_texture_clear_half():
|
|
col = peek_tex_with_clear_color(Texture.T_half_float, Texture.F_rgba, (0, 0.25, -0.5, 2))
|
|
assert col == LColor(0, 0.25, -0.5, 2)
|
|
|
|
# Test edge cases
|
|
inf = float('inf')
|
|
nan = float('nan')
|
|
col = peek_tex_with_clear_color(Texture.T_half_float, Texture.F_rgba, (65504, 65536, inf, nan))
|
|
assert col.x == 65504
|
|
assert col.y == inf
|
|
assert col.z == inf
|
|
assert math.isnan(col.w)
|
|
|
|
# Negative edge case
|
|
col = peek_tex_with_clear_color(Texture.T_half_float, Texture.F_rgba, (-65504, -65536, -inf, -nan))
|
|
assert col.x == -65504
|
|
assert col.y == -inf
|
|
assert col.z == -inf
|
|
assert math.isnan(col.w)
|