updates to new classes

Author: jtsodapop

labelbox/data/annotation_types/data/tiled_image.py

@@ -1,10 +1,13 @@
 from enum import Enum
-from typing import Optional, List
+from typing import Optional, List, Any, Dict
 
 from pydantic import BaseModel, validator
+import numpy as np
+from pyproj import Transformer, transform
 
 from ..geometry import Point
 from .base_data import BaseData
+from .raster import RasterData
 
 
 class EPSG(Enum):
@@ -15,9 +18,9 @@ class EPSG(Enum):
     >>> epsg = EPSG()
     """
     SIMPLEPIXEL = 1
-    EPSG4326 = 2
-    EPSG3857 = 3
-    EPSG3395 = 4
+    EPSG4326 = 4326
+    EPSG3857 = 3857
+    EPSG3395 = 3395
 
 
 class TiledBounds(BaseModel):
@@ -71,38 +74,95 @@ class TiledImageData(BaseData):
     """ Represents tiled imagery 
 
     If specified version is 2, converts bounds from [lng,lat] to [lat,lng]
+
+    Requires the following args:
+        tile_layer: TileLayer
+        tile_bounds: TiledBounds
+        zoom_levels: List[int]
+    Optional args:
+        max_native_zoom: int = None
+        tile_size: Optional[int]
+        version: int = 2
+        alternative_layers: List[TileLayer]        
     """
     tile_layer: TileLayer
-    alternative_layers: List[TileLayer]
     tile_bounds: TiledBounds
+    alternative_layers: List[TileLayer] = None
     zoom_levels: List[int]
-    max_native_zoom: int = zoom_levels[1]
+    max_native_zoom: int = None
     tile_size: Optional[int]
     version: int = 2
 
-    # @property #TODO
-    # def value(self):
-    #     Return self._as_raster(self.min_zoom).value()
-
-    # def _as_raster(zoom): #TODO
-    #     stitched together tiles as a RasterData object
-    #     Return result
-
-    @property
-    def tile_layer_url(self):
-        return self.tile_layer.url
-
-    @property
-    def bounds(self):
-        return self.tile_bounds.bounds
-
+    def __post_init__(self):
+        if self.max_native_zoom is None:
+            self.max_native_zoom = zoom_levels[1]
+
+    #TODO: look further into Matt's code and how to reference the monorepo ?
+    def _as_raster(zoom):
+        # stitched together tiles as a RasterData object
+        # TileData.get_image(target_hw) ← we will be using this from Matt's precomputed embeddings
+        # more info found here: https://github.com/Labelbox/python-monorepo/blob/baac09cb89e083209644c9bdf1bc3d7cb218f147/services/precomputed_embeddings/precomputed_embeddings/tiled.py
+        image_as_np = None
+        return RasterData(arr=image_as_np)
+
+    @property  #TODO
+    def value(self) -> np.ndarray:
+        return self._as_raster(self.min_zoom).value()
+
+    #TODO: maybe not necessary, can remove
+    # @property
+    # def tile_layer_url(self) -> str:
+    #     return self.tile_layer.url
+
+    #TODO: maybe not necessary, can remove
+    # @property
+    # def bounds(self) -> List[Point]:
+    #     return self.tile_bounds.bounds
+
+    #TODO: wondering to keep this or not since epsg may be something of interest
+    # and they dont want to go through TiledImageData.tile_bounds.epsg?
+    # then can call TiledimageData.epsg
     @property
-    def epsg(self):
+    def epsg(self) -> EPSG:
         return self.tile_bounds.epsg
 
     @validator('zoom_levels')
     def validate_zooms(cls, zoom_levels):
         if len(zoom_levels) != 2:
             raise AssertionError(
-                f"zoom_levels should only contain min and max, found {len(zoom_levels)}"
+                f"zoom_levels should contain 2 values [min,max], found {len(zoom_levels)}"
+            )
+
+
+#TODO: we will need to update the [data] package to also require pyproj
+class EPSGTransformer(BaseModel):
+    """Transformer class between different EPSG's. Useful when wanting to project 
+    in different formats.
+
+    Requires as input a Point object.
+    """
+    class Config:
+        arbitrary_types_allowed = True
+
+    transform_function: Transformer = None
+
+    def is_simple(self, epsg: EPSG) -> bool:
+        return epsg == EPSG.SIMPLEPIXEL
+
+    def geo_and_geo(self, src_epsg: EPSG, tgt_epsg: EPSG) -> None:
+        if self.is_simple(src_epsg) or self.is_simple(tgt_epsg):
+            raise Exception(
+                f"Cannot be used for Simple transformations. Found {src_epsg} and {tgt_epsg}"
             )
+        self.transform_function = Transformer.from_crs(src_epsg.value,
+                                                       tgt_epsg.value)
+
+    def geo_and_pixel(self, src_epsg, geojson):
+        pass
+
+    def __call__(self, point: Point):
+        if self.transform_function is not None:
+            res = self.transform_function.transform(point.x, point.y)
+            return Point(x=res[0], y=res[1])
+        else:
+            raise Exception("No transformation has been set.")