perf: EXRHeader에 변경사항을 적용하다

Author: 404Vector

openexr_python/__init__.py

@@ -9,8 +9,5 @@
 
 __version__ = "0.0.0"  # dynamic
 
-from .exr_enums import *
-from .exr_channel_info import *
-from .exr_channel_sampling_rate import *
 from .exr_header import *
 from .exr_window import *

openexr_python/exr_header.py

@@ -1,56 +1,63 @@
 from dataclasses import dataclass
-from typing import Any, Dict, Tuple
-from openexr_python.exr_enums import EXRCompressionType, EXRLineOrderType
-from openexr_python.exr_channel_info import EXRChannelInfo
-from openexr_python.exr_window import EXRWindow
+from typing import Any, Dict
+import OpenEXR
+import Imath
 
 
 @dataclass
 class EXRHeader:
-    _header_raw_dict: Dict[str, Any]
+    raw_header: Dict[str, Any]
 
     def __getitem__(self, key: str) -> Any:
-        return self._header_raw_dict[key]
+        return self.raw_header[key]
+
+    def add_channel(self, channel_name: str, channel_info: Imath.Channel):
+        self.channels[channel_name] = channel_info
+
+    def remove_channel(self, channel_name: str):
+        self.channels.pop(channel_name)
 
     @property
-    def channels(self) -> Dict[str, EXRChannelInfo]:
-        raw_channels: Dict[str, Any] = self["channels"]
-        return {
-            key: EXRChannelInfo.parse_raw2pyobj(raw_channel_info=value)
-            for key, value in raw_channels.items()
-        }
+    def channels(self) -> Dict[str, Imath.Channel]:
+        raw_channels: Dict[str, Imath.Channel] = self["channels"]
+        return raw_channels
 
     @property
-    def compression(self) -> EXRCompressionType:
-        raw_compression: Any = self["compression"]
-        return EXRCompressionType.parse_raw2pyobj(raw_compression=raw_compression)
+    def compression(self) -> Imath.Compression:
+        raw_compression: Imath.Compression = self["compression"]
+        return raw_compression
 
     @property
-    def data_window(self) -> EXRWindow:
-        raw_dataWindow: Any = self["dataWindow"]
-        return EXRWindow.parse_raw_to_window(raw_dataWindow)
+    def data_window(self) -> Imath.Box2i:
+        raw_dataWindow: Imath.Box2i = self["dataWindow"]
+        return raw_dataWindow
 
     @property
-    def display_window(self) -> EXRWindow:
-        raw_dataWindow: Any = self["displayWindow"]
-        return EXRWindow.parse_raw_to_window(raw_dataWindow)
+    def display_window(self) -> Imath.Box2i:
+        raw_dataWindow: Imath.Box2i = self["displayWindow"]
+        return raw_dataWindow
 
     @property
-    def line_order(self) -> EXRLineOrderType:
-        raw_line_order: Any = self["lineOrder"]
-        return EXRLineOrderType.parse_raw2pyobj(raw_line_order=raw_line_order)
+    def line_order(self) -> Imath.LineOrder:
+        raw_line_order: Imath.LineOrder = self["lineOrder"]
+        return raw_line_order
 
     @property
     def pixel_aspect_ratio(self) -> float:
         raw_pixel_aspect_ratio: float = self["pixelAspectRatio"]
         return raw_pixel_aspect_ratio
 
     @property
-    def screen_window_center(self) -> Tuple[float, float]:
-        raw_screen_window_center: Any = self["screenWindowCenter"]
-        return (raw_screen_window_center.x, raw_screen_window_center.y)
+    def screen_window_center(self) -> Imath.V2f:
+        raw_screen_window_center: Imath.V2f = self["screenWindowCenter"]
+        return raw_screen_window_center
 
     @property
     def screen_window_width(self) -> float:
         raw_screen_window_width: float = self["screenWindowWidth"]
         return raw_screen_window_width
+
+    @classmethod
+    def create_header(cls, width: int, height: int):
+        raw_header = OpenEXR.Header(width, height)
+        return EXRHeader(raw_header=raw_header)