feat: added png support

Author: ForNeus57

src/python/app/io/bmp.py

@@ -38,7 +38,7 @@ class Signature(IntEnum):
 
 @dataclass(slots=True)
 class BitmapFileHeader:
-    """Implements the struct for the """
+    """Implements the struct for the Bitmap file header"""
 
     HEADER_LENGTH: ClassVar[int] = 14
 
@@ -53,7 +53,7 @@ def from_bytes(cls, data: bytes) -> 'BitmapFileHeader':
         if len(data) < cls.HEADER_LENGTH:
             raise InvalidFormatException(f"Header too short, received: {len(data)} instead of {cls.HEADER_LENGTH}")
 
-        return cls(*struct.unpack('HIHHI', data))
+        return cls(*struct.unpack('>HIHHI', data))
 
     def __post_init__(self) -> None:
         if self.signature not in set(e.value for e in Signature):
@@ -68,11 +68,11 @@ def __post_init__(self) -> None:
 
 @dataclass(slots=True)
 class BMP:
-    pass
+    header: BitmapFileHeader
 
 
 @final
-class BMPReader(IFormatReader):    # pylint: disable=too-few-public-methods
+class BMPReader(IFormatReader):
     """Class that deserializes BMP format to Image"""
 
     @override
@@ -108,12 +108,12 @@ def read_format(self, file: BinaryIO) -> Image:
 
 
 @final
-class BMPWriter(IFormatWriter):    # pylint: disable=too-few-public-methods
+class BMPWriter(IFormatWriter):
     """Class that serializes Image to BMP format"""
 
+    @override
     def write_format(self, file: BinaryIO, input_image: Image) -> None:
         input_arr = input_image.data
-        input_arr = np.flip(np.flip(input_arr[:, :, ::-1], axis=1), axis=0)
 
         image_height, image_width, num_colors = input_arr.shape
         bits_per_pixel = num_colors * 8

src/python/app/io/format_factory.py

@@ -6,6 +6,7 @@
 from app.io.bmp import BMPReader, BMPWriter
 from app.io.format_reader import IFormatReader
 from app.io.format_writer import IFormatWriter
+from app.io.png import PNGWriter, PNGReader
 
 
 class KnownFormat(enum.Enum):
@@ -25,6 +26,9 @@ def from_string(cls, data_format: str) -> 'KnownFormat':
             case 'bmp':
                 return cls.BMP
 
+            case 'png':
+                return cls.PNG
+
             case _:
                 raise UnknownFormatException(data_format)
 
@@ -38,7 +42,7 @@ def get_available_formats(cls) -> list[str]:
     def default(cls) -> 'KnownFormat':
         """Return default format for commandline to use"""
 
-        return KnownFormat.BMP
+        return KnownFormat.PNG
 
 
 def get_reader_from_format(data_format: KnownFormat) -> IFormatReader:
@@ -48,6 +52,9 @@ def get_reader_from_format(data_format: KnownFormat) -> IFormatReader:
         case KnownFormat.BMP:
             return BMPReader()
 
+        case KnownFormat.PNG:
+            return PNGReader()
+
         case _:
             assert False, "unreachable"
 
@@ -59,5 +66,8 @@ def get_writer_from_format(data_format: KnownFormat) -> IFormatWriter:
         case KnownFormat.BMP:
             return BMPWriter()
 
+        case KnownFormat.PNG:
+            return PNGWriter()
+
         case _:
             assert False, "unreachable"