pigeon-tem-comms 2.0.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

TEM_comms/camera.py

@@ -1,27 +1,14 @@
 from pigeon import BaseMessage
-from .tile.metadata import TileMetadata
+from .tile.metadata import TileMetadata, ProcessingOptions
 from typing import Optional
 from pydantic import Field
 
 
-class Command(TileMetadata):
+class Command(TileMetadata, ProcessingOptions):
     """
     This message is used to instruct the camera to capture an image.
     """
 
-    brightfield: Optional[bool] = Field(
-        default=False,
-        description="If true, this tile should be used in creating the brightfield image.",
-    )
-    darkfield: Optional[bool] = Field(
-        default=False,
-        description="If true, this tile should be used in creating the darkfield image.",
-    )
-    lens_correction: Optional[bool] = Field(
-        default=True,
-        description="If true, this tile is part of a lens correction montage.",
-    )
-
 
 class Image(BaseMessage):
     """

TEM_comms/tile/__init__.py

@@ -1,7 +1,7 @@
-from .metadata import TileMetadata
+from .metadata import TileMetadata, ProcessingOptions
 from . import statistics
 from pydantic import BaseModel, Field
-from typing import Literal, List, Tuple
+from typing import Literal, List, Tuple, Optional
 
 
 class Preview(TileMetadata):
@@ -20,7 +20,7 @@ class Mini(TileMetadata):
     image: str = Field(description="The downsampled tile as a base 64 encoded string.")
 
 
-class Raw(TileMetadata):
+class Raw(TileMetadata, ProcessingOptions):
     """
     This message is sent whenever a new tile is stored on the filesystem and is ready for processing.
     """
@@ -74,37 +74,42 @@ class Matches(TileMetadata):
 class TemplateMatch(BaseModel):
     model_config = {"extra": "forbid"}
 
-    row: int = Field(description="The row of the neighboring tile.")
-    column: int = Field(description="The column of the neighboring tile.")
-    position: Literal["top", "bottom", "left", "right"] = Field(
-        description="The position of matched tile relative to the captured tile."
+    offset: Tuple[float, float] = Field(
+        description="The offset between the expected and actual template positions."
     )
-
-    offsets: List[Tuple[float, float]] = Field(
-        description="The offsets between expected and actual template positions."
+    distance: float = Field(description="The distance of the offset.")
+    rotation: float = Field(description="The angle of the offset.")
+    maxVal: float = Field(
+        description="The maximum value of the template match."
     )
-    distance: List[float] = Field(description="The distance of each offset.")
-    rotation: List[float] = Field(description="The angle of each offset.")
-    maxVal: List[float] = Field(
-        description="The maximum value from each of the template matches."
+    minVal: float = Field(
+        description="The minimum value of the template match."
     )
-    minVal: List[float] = Field(
-        description="The minimum value from each of the template matches."
+    expected_offset_in_crop: Tuple[int, int] = Field(description="")
+    maxLoc: Tuple[int, int] = Field(
+        description="The maximum location of the template match."
     )
-    expected_offset_in_crop: List[Tuple[int, int]] = Field(description="")
-    maxLoc: List[Tuple[int, int]] = Field(
-        description="The maximum location of each template match."
+    matched_pos_img2: Tuple[int, int] = Field(
+        description="The template top left corner absolute location."
     )
-    matched_pos_img2: List[Tuple[int, int]] = Field(
-        description="Each of the template top left corner absolute locations."
+    matched_center_img2: Tuple[int, int] = Field(
+        description="The template center absolute location."
     )
-    matched_center_img2: List[Tuple[int, int]] = Field(
-        description="Each of the template center absolute locations."
+    good: bool = Field(description="True if the match is good.")
+    reject_reason: Optional[str] = Field(
+        default="", description="The reason why the match is not good."
     )
-    good: List[bool] = Field(description="True if an individual match is good.")
-    reject_reason: List[str] = Field(
-        description="The reason why an individual match is not good."
+
+
+class TemplateMatchContainer(BaseModel):
+    model_config = {"extra": "forbid"}
+
+    row: int = Field(description="The row of the neighboring tile.")
+    column: int = Field(description="The column of the neighboring tile.")
+    position: Literal["top", "bottom", "left", "right"] = Field(
+        description="The position of matched tile relative to the captured tile."
     )
+    matches: List[TemplateMatch] = Field(description="A list of data structures containing the information about each individual match.")
 
 
 class TemplateMatches(TileMetadata):
@@ -112,7 +117,7 @@ class TemplateMatches(TileMetadata):
     This message contains data reltaing to template matches used for calculating the lens correction transform.
     """
 
-    matches: List[TemplateMatch] = Field(
+    matches: List[TemplateMatchContainer] = Field(
         description="Information about how the captured tile matches to each of its available neighbors."
     )
 

TEM_comms/tile/metadata.py

@@ -1,5 +1,6 @@
 from pigeon import BaseMessage
 from pydantic import Field
+from typing import Optional
 
 
 class TileMetadata(BaseMessage):
@@ -20,3 +21,18 @@ class TileMetadata(BaseMessage):
     overlap: int = Field(
         description="The number of pixels of overlap between tiles.", examples=[512]
     )
+
+
+class ProcessingOptions(BaseMessage):
+    brightfield: Optional[bool] = Field(
+        default=False,
+        description="If true, this tile should be used in creating the brightfield image.",
+    )
+    darkfield: Optional[bool] = Field(
+        default=False,
+        description="If true, this tile should be used in creating the darkfield image.",
+    )
+    lens_correction: Optional[bool] = Field(
+        default=False,
+        description="If true, this tile is part of a lens correction montage.",
+    )

{pigeon_tem_comms-2.0.0.dist-info → pigeon_tem_comms-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pigeon-tem-comms
-Version: 2.0.0
+Version: 2.1.1
 Summary: Pigeon messages for the next generation TEM imaging system.
 Author-email: Cameron Devine <cameron.devine@alleninstitute.org>
 License: BSD 3 Clause

{pigeon_tem_comms-2.0.0.dist-info → pigeon_tem_comms-2.1.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 TEM_comms/__init__.py,sha256=c2UGXOvlwg_Xy5WTQBaIS2ebmgEtJIbIwyoh7AzDjDk,1850
 TEM_comms/buffer.py,sha256=bQiYDQDOMDBEfx6QKWFKUJE14z-xn8SqwnqE1LqL5Io,546
 TEM_comms/calibration.py,sha256=Nd5u1HA5qwA_04pOeyklVLZkdV06e9y0kS2Gv9qvJNg,1383
-TEM_comms/camera.py,sha256=nUIL7nW2u7BNtA-96cHL2NSwH8kJ_5AgkWmspGjItEs,2600
+TEM_comms/camera.py,sha256=XrRVU-4hAGHJEbfvoj3b7359NEuTuRyqtMLL9e0V3xI,2164
 TEM_comms/lens_correction.py,sha256=KCsI1jMuPwHTxfHzx4_MP7VXdgcq6DpESQOYXV4NVqQ,290
 TEM_comms/montage.py,sha256=GevXbNdf8NGmvxR-2RPEK4JeSGtNdTXpNE_I2PSjTZ8,2157
 TEM_comms/qc.py,sha256=sFUTBjroGZrKjuL9u0LO6DT9mn10IEQsfuMv1QjgiLU,627
@@ -14,12 +14,12 @@ TEM_comms/stage/__init__.py,sha256=UVtYKzMBI01FUADS24xSItNIjhqdZtlDeJdqsDJNVIc,6
 TEM_comms/stage/aperture.py,sha256=RoZjwmUMsgZcFTW6OYasK7g9pbz5hE0fUAOvWqVPZeA,1029
 TEM_comms/stage/motion.py,sha256=Ox_WIdhKxhkr7Q9QxbdKRKx-84az5tT_L1yUlxxNrXo,1730
 TEM_comms/stage/rotation.py,sha256=_rMk9ES7_ivMZ1AypqFVdvcuTbTQ32g9La2L1ZLdMTE,1476
-TEM_comms/tile/__init__.py,sha256=KHzx0GHWiIggphOkUq4bHRBrKCLYrW_VKf49lqbpTfY,4725
-TEM_comms/tile/metadata.py,sha256=k5YYn5UDTNcIDw0Cjm1nrajd6BGAvctOUwEWn2ZJ9O4,790
+TEM_comms/tile/__init__.py,sha256=_SN7yS_rbosNsZ0n9M62WOlD_WSHnjaLoeg-vvdDeEY,4895
+TEM_comms/tile/metadata.py,sha256=iLKEmi-ocC0I_LU4YVk0AymL5zxSmmquk-fKDiKyUg0,1332
 TEM_comms/tile/statistics.py,sha256=hinAROuYdL4l0SNAdvndqetbWhDiF1ANjMoLJoPubpM,976
-pigeon_tem_comms-2.0.0.dist-info/licenses/LICENSE,sha256=tZfQIUX7uNGacTb751ZWInVenJl6OwYQihcGbYrKSts,1463
-pigeon_tem_comms-2.0.0.dist-info/METADATA,sha256=kdfkvH5q1KSRMhMm8IooD1tlAeqaeWhKFTgzZf_3ao4,1575
-pigeon_tem_comms-2.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pigeon_tem_comms-2.0.0.dist-info/entry_points.txt,sha256=X2WfBCDgfH9XVVqiI4rLu7BLyMNuLW6WWckqXxfbrgk,38
-pigeon_tem_comms-2.0.0.dist-info/top_level.txt,sha256=3_1RcQ7xvS2dJEYnPIigo9snoKcSXSNJzPqunY0qydI,10
-pigeon_tem_comms-2.0.0.dist-info/RECORD,,
+pigeon_tem_comms-2.1.1.dist-info/licenses/LICENSE,sha256=tZfQIUX7uNGacTb751ZWInVenJl6OwYQihcGbYrKSts,1463
+pigeon_tem_comms-2.1.1.dist-info/METADATA,sha256=3WsqVZnGiI0xuEf3dUnyXqg0aeS65lRNpWkADV6xzkU,1575
+pigeon_tem_comms-2.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pigeon_tem_comms-2.1.1.dist-info/entry_points.txt,sha256=X2WfBCDgfH9XVVqiI4rLu7BLyMNuLW6WWckqXxfbrgk,38
+pigeon_tem_comms-2.1.1.dist-info/top_level.txt,sha256=3_1RcQ7xvS2dJEYnPIigo9snoKcSXSNJzPqunY0qydI,10
+pigeon_tem_comms-2.1.1.dist-info/RECORD,,