Source code for py3dtiles.tile

from __future__ import annotations

from pathlib import Path
from typing import List, Union

from .extendable import Extendable
from .tile_content import TileContent


[docs] class Tile(Extendable): def __init__(self, geometric_error=500, bounding_volume=None, refine_mode="ADD"): super().__init__() self.bounding_volume = bounding_volume self.geometric_error = geometric_error self._refine = "" self.set_refine_mode(refine_mode) self._content = None self._children = [] # Some possible valid properties left un-delt with viewerRequestVolume self._transform = None
[docs] def set_transform(self, transform: List[float]) -> None: """ :param transform: a flattened transformation matrix :return: """ self._transform = transform
[docs] def get_transform(self) -> list[float]: if self._transform is not None: return self._transform return [1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0]
[docs] def set_content(self, content: TileContent, force=True) -> None: if not force and self._content is not None: return self._content = content
[docs] def get_content(self) -> TileContent: return self._content
[docs] def set_content_uri(self, uri: str) -> None: if self._content is None: raise AttributeError('Tile with unset content.')
# self._content.set_uri(uri) # TODO add set_uri in TileContent
[docs] def get_content_uri(self) -> str: if self._content is None: raise AttributeError('Tile with unset content.') # return self._content.get_uri() # TODO add get_uri in TileContent return ""
[docs] def set_refine_mode(self, mode: str) -> None: if mode != 'ADD' and mode != 'REPLACE': raise ValueError(f"Unknown refinement mode {mode}. Should be either 'ADD' or 'REPLACE'.") self._refine = mode
[docs] def get_refine_mode(self) -> str: return self._refine
[docs] def add_child(self, tile: Tile) -> None: self._children.append(tile)
[docs] def has_children(self) -> bool: return len(self._children) != 0
[docs] def get_direct_children(self) -> List[Tile]: return self._children
[docs] def get_children(self) -> List[Tile]: """ :return: the recursive (across the children tree) list of the children tiles """ descendants = [] for child in self._children: # Add the child... descendants.append(child) # and if (and only if) they are grand-children then recurse if child.has_children(): descendants += child.get_children() return descendants
[docs] def sync_bounding_volume_with_children(self) -> None: if self.bounding_volume is None: raise AttributeError('This Tile has no bounding volume: exiting.') if not self.bounding_volume.is_box(): raise NotImplementedError("Don't know how to sync non box bounding volume.") # We consider that whatever information is present it is the # proper one (in other terms: when they are no sub-tiles this tile # is a leaf-tile and thus is has no synchronization to do) for child in self.get_direct_children(): child.sync_bounding_volume_with_children() # The information that depends on (is defined by) the children # nodes is limited to be bounding volume. self.bounding_volume.sync_with_children(self)
[docs] def write_content(self, directory: Path) -> None: """ Write (or overwrite) the tile _content_ to the directory specified as parameter and withing the relative filename designated by the tile's content uri. Note that it is the responsibility of the owning TileSet to - set those uris - to explicitly invoke write_content() (this is to be opposed with the Tile attributes which get serialized when recursing on the TileSet attributes) :param directory: the target directory """ file_name = self.get_content_uri() if not file_name: raise AttributeError("Tile with no content or no uri in content.") path_name = directory / file_name # Make sure the output directory exists (note that target_dir may # be a sub-directory of 'directory' because the uri might hold its # own path): path_name.parent.mkdir(parents=True, exist_ok=True) # Write the tile content of this tile: with path_name.open('wb') as f: f.write(self.get_content().to_array())
[docs] def to_dict(self) -> dict: dict_data = {} if self.bounding_volume is not None: bounding_volume = self.bounding_volume else: raise AttributeError('Bounding volume is not set') dict_data['boundingVolume'] = bounding_volume.to_dict() dict_data['geometricError'] = self.geometric_error if self._children: # The children list exists indeed (for technical reasons) yet it # happens to be still empty. This would pollute the json output # by adding a "children" entry followed by an empty list. In such # case just remove that attributes entry: dict_data['children'] = [child.to_dict() for child in self._children] if self._content is not None: # Refer to children related above comment (mutatis mutandis): dict_data['content'] = self._content return dict_data