add: Header에 필요한 data struct를 추가하다.

- EXRWindow : 해당 window 크기 및 위치

Author: 404Vector

openexr_python/__init__.py

@@ -9,4 +9,8 @@
 
 __version__ = "0.0.0"  # dynamic
 
+from .exr_enums import *
+from .exr_channel_info import *
+from .exr_channel_sampling_rate import *
 from openexr_python.header import Header
+from .exr_window import *

openexr_python/exr_channel_info.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+from typing import Any, Type
+from openexr_python.exr_enums.exr_image_data_type import EXRImageDataType
+from openexr_python.exr_channel_sampling_rate import EXRChannelSamplingRate
+
+
+@dataclass
+class EXRChannelInfo:
+    data_type: Type[EXRImageDataType]
+    sampling_rate: EXRChannelSamplingRate
+
+    @classmethod
+    def parse_raw2pyobj(cls, raw_channel_info: Any):
+        return EXRChannelInfo(
+            data_type=EXRImageDataType.parse_raw2pyobj(
+                raw_channel_info=raw_channel_info,
+            ),
+            sampling_rate=EXRChannelSamplingRate.parse_raw2pyobj(
+                raw_channel_info=raw_channel_info,
+            ),
+        )

openexr_python/exr_channel_sampling_rate.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass
+from typing import Any
+
+
+@dataclass
+class EXRChannelSamplingRate:
+    x_sampling: int
+    y_sampling: int
+
+    @classmethod
+    def parse_raw2pyobj(cls, raw_channel_info: Any):
+        return EXRChannelSamplingRate(
+            x_sampling=raw_channel_info.xSampling, y_sampling=raw_channel_info.ySampling
+        )

openexr_python/exr_window.py

@@ -0,0 +1,37 @@
+from dataclasses import dataclass
+from typing import Any, Tuple
+
+
+@dataclass
+class EXRWindow:
+    x_min: int
+    y_min: int
+    x_max: int
+    y_max: int
+
+    @property
+    def min(self) -> Tuple[int, int]:
+        return (self.x_min, self.y_min)
+
+    @property
+    def max(self) -> Tuple[int, int]:
+        return (self.x_max, self.y_max)
+
+    @property
+    def width(self) -> int:
+        assert self.x_max > self.x_min
+        return 1 + self.x_max - self.x_min
+
+    @property
+    def height(self) -> int:
+        assert self.y_max > self.y_min
+        return 1 + self.y_max - self.y_min
+
+    @classmethod
+    def parse_raw_to_window(cls, raw_window: Any):
+        return EXRWindow(
+            x_min=raw_window.min.x,
+            y_min=raw_window.min.y,
+            x_max=raw_window.max.x,
+            y_max=raw_window.max.y,
+        )