def test_scaled_extra_byte_array_type(simple_las_path):
"""
To make sure we handle scaled extra bytes
"""
las = pylas.read(simple_las_path)
las.add_extra_dim(
pylas.ExtraBytesParams(
name="test_dim",
type="3int32",
scales=np.array([1.0, 2.0, 3.0], np.float64),
offsets=np.array([10.0, 20.0, 30.0], np.float64),
)
)
assert np.allclose(las.test_dim[..., 0], 10.0)
assert np.allclose(las.test_dim[..., 1], 20.0)
assert np.allclose(las.test_dim[..., 2], 30.0)
las.test_dim[..., 0][:] = 42.0
las.test_dim[..., 1][:] = 82.0
las.test_dim[..., 2][:] = 123.0
assert np.allclose(las.test_dim[..., 0], 42.0)
assert np.allclose(las.test_dim[..., 1], 82.0)
assert np.allclose(las.test_dim[..., 2], 123.0)
las = write_then_read_again(las)
assert np.allclose(las.test_dim[..., 0], 42.0)
assert np.allclose(las.test_dim[..., 1], 82.0)
assert np.allclose(las.test_dim[..., 2], 123.0)
def test_read_write_example_extra_bytes_file(las_file_path_with_extra_bytes):
"""
Test that we can write extra bytes without problem
"""
original = pylas.read(las_file_path_with_extra_bytes)
las = write_then_read_again(original)
for name in original.point_format.dimension_names:
assert np.allclose(las[name], original[name])
def test_rw_all_set_one(las):
for dim_name in las.point_format.dimension_names:
las[dim_name][:] = 1
for dim_name in las.point_format.dimension_names:
assert np.alltrue(las[dim_name] == 1), "{} not equal".format(dim_name)
las2 = write_then_read_again(las)
for dim_name in las.point_format.dimension_names:
assert np.alltrue(las[dim_name] == las2[dim_name]), "{} not equal".format(
dim_name
)
def test_change_format(las):
in_version = las.header.version
las = pylas.convert(las, point_format_id=2)
las = write_then_read_again(las)
assert las.points.point_format.id == 2
assert las.header.point_format.id == 2
assert las.header.version == in_version
assert dim_does_not_exists(las, "gps_time")
las = pylas.convert(las, point_format_id=1)
las = write_then_read_again(las)
assert las.points.point_format.id == 1
assert las.header.point_format.id == 1
assert las.header.version == in_version
assert dim_does_not_exists(las, "red")
assert dim_does_not_exists(las, "green")
assert dim_does_not_exists(las, "blue")
las = pylas.convert(las, point_format_id=0)
las = write_then_read_again(las)
assert las.points.point_format.id == 0
assert las.header.point_format.id == 0
assert las.header.version == in_version
assert dim_does_not_exists(las, "red")
assert dim_does_not_exists(las, "green")
assert dim_does_not_exists(las, "blue")
assert dim_does_not_exists(las, "gps_time")
las = pylas.convert(las, point_format_id=8)
las = write_then_read_again(las)
assert str(las.header.version) == "1.4"
assert las.points.point_format.id == 8
assert las.header.point_format.id == 8
assert dim_does_exists(las, "red")
assert dim_does_exists(las, "green")
assert dim_does_exists(las, "blue")
assert dim_does_exists(las, "nir")
las = pylas.convert(las, point_format_id=7)
las = write_then_read_again(las)
assert str(las.header.version) == "1.4"
assert las.points.point_format.id == 7
assert las.header.point_format.id == 7
assert dim_does_exists(las, "red")
assert dim_does_exists(las, "green")
assert dim_does_exists(las, "blue")
assert dim_does_not_exists(las, "nir")
las = pylas.convert(las, point_format_id=6)
las = write_then_read_again(las)
assert str(las.header.version) == "1.4"
assert las.points.point_format.id == 6
assert las.header.point_format.id == 6
assert dim_does_not_exists(las, "red")
assert dim_does_not_exists(las, "green")
assert dim_does_not_exists(las, "blue")
assert dim_does_not_exists(las, "nir")
def test_extra_bytes_with_spaces_in_name(simple_las_path):
"""
Test that we can create extra bytes with spaces in their name
and that they can be accessed using __getitem__ ( [] )
as de normal '.name' won't work
"""
las = pylas.read(simple_las_path)
las.add_extra_dim(pylas.ExtraBytesParams(name="Name With Spaces", type="int32"))
assert np.alltrue(las["Name With Spaces"] == 0)
las["Name With Spaces"][:] = 789_464
las = write_then_read_again(las)
np.alltrue(las["Name With Spaces"] == 789_464)
def test_creating_extra_bytes(extra_bytes_params, simple_las_path):
"""
Test that we can create extra byte dimensions for each
data type. And they can be written then read.
"""
las = pylas.read(simple_las_path)
las.add_extra_dim(extra_bytes_params)
assert np.allclose(las[extra_bytes_params.name], 0)
las[extra_bytes_params.name][:] = 42
assert np.allclose(las[extra_bytes_params.name], 42)
las = write_then_read_again(las)
assert np.allclose(las[extra_bytes_params.name], 42)
def test_point_format_conversion_copies_field_values(file_path,
target_point_format_id):
original = pylas.read(file_path)
converted = pylas.convert(original, point_format_id=target_point_format_id)
converted = write_then_read_again(converted)
converted_dimension_names = set(converted.point_format.dimension_names)
dimension_expected_to_be_kept = [
dim_name for dim_name in original.point_format.dimension_names
if dim_name in converted_dimension_names
]
for dim_name in dimension_expected_to_be_kept:
assert np.allclose(converted[dim_name],
original[dim_name]), "{} not equal".format(dim_name)
def test_extra_bytes_description_is_ok(extra_bytes_params, simple_las_path):
"""
Test that the description in ok
"""
las = pylas.read(simple_las_path)
las.add_extra_dim(extra_bytes_params)
extra_dim_info = list(las.point_format.extra_dimensions)
assert len(extra_dim_info) == 1
assert extra_dim_info[0].description == extra_bytes_params.description
las = write_then_read_again(las)
extra_dim_info = list(las.point_format.extra_dimensions)
assert len(extra_dim_info) == 1
assert extra_dim_info[0].description == extra_bytes_params.description
def test_creating_scaled_extra_bytes(extra_bytes_params, simple_las_path):
las = pylas.read(simple_las_path)
try:
num_elements = int(extra_bytes_params.type[0])
except ValueError:
num_elements = 1
params = pylas.ExtraBytesParams(
extra_bytes_params.name,
extra_bytes_params.type,
offsets=np.array([2.0] * num_elements),
scales=np.array([1.0] * num_elements),
)
las.add_extra_dim(params)
assert np.allclose(las[extra_bytes_params.name], 2.0)
las[params.name][:] = 42.0
assert np.allclose(las[extra_bytes_params.name], 42.0)
las = write_then_read_again(las)
assert np.allclose(las[extra_bytes_params.name], 42.0)
def test_can_write_then_re_read_files(las, do_compress):
_las = write_then_read_again(las, do_compress=do_compress)