diff --git a/S3MP/utils/image_utils.py b/S3MP/utils/image_utils.py index f79ae08..15396d8 100644 --- a/S3MP/utils/image_utils.py +++ b/S3MP/utils/image_utils.py @@ -24,6 +24,7 @@ def __init__( focal_length: float | None = None, altitude: float | None = None, distortion_params: DistortionParams | None = None, + dewarp_flag: bool = False, ): """Initialize ImageMetadata object. @@ -37,6 +38,7 @@ def __init__( focal_length: Focal length in pixels if available altitude: Altitude in meters if available distortion_params: Distortion parameters if available + dewarp_flag: Flag indicating if image should be dewarped based on metadata """ self.mirror_path = mirror_path self.name = self.mirror_path.local_path.stem @@ -63,6 +65,7 @@ def __init__( self.focal_length = focal_length self.altitude = altitude self.distortion_params = distortion_params + self.dewarp_flag = dewarp_flag @classmethod def parse_metadata(cls, mirror_path: MirrorPath) -> ImageMetadata: @@ -84,6 +87,11 @@ def parse_metadata(cls, mirror_path: MirrorPath) -> ImageMetadata: except Exception: distortion_params = None + try: + dewarp_flag = parser.dewarp_flag() + except KeyError: + dewarp_flag = False + return cls( mirror_path, parser.dimensions(), @@ -94,6 +102,7 @@ def parse_metadata(cls, mirror_path: MirrorPath) -> ImageMetadata: parser.focal_length_pixels(), parser.relative_altitude(), distortion_params, + dewarp_flag, ) @property diff --git a/pyproject.toml b/pyproject.toml index c45d305..4b34468 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "S3MP" -version = "0.8.0" +version = "0.8.1" description = "" authors = [ {name = "Joshua Dean", email = "joshua.dean@sentera.com"}, diff --git a/uv.lock b/uv.lock index d8ee49a..b6eb775 100644 --- a/uv.lock +++ b/uv.lock @@ -1515,7 +1515,7 @@ wheels = [ [[package]] name = "s3mp" -version = "0.8.0" +version = "0.8.1" source = { editable = "." } dependencies = [ { name = "aioboto3" },