Преглед на файлове

LibGfx/JPEGXL: Put frame's image inside the `Frame` struct

The image was previously managed as an output parameter of the
`read_frame` function. This reorganisation also brings us closer to the
spec. As it is specified that each frame have its own image that will
later on compose a greater bitmap.
Lucas CHOLLET преди 1 година
родител
ревизия
49c55447d9
променени са 1 файла, в които са добавени 18 реда и са изтрити 19 реда
  1. 18 19
      Userland/Libraries/LibGfx/ImageFormats/JPEGXLLoader.cpp

+ 18 - 19
Userland/Libraries/LibGfx/ImageFormats/JPEGXLLoader.cpp

@@ -2310,10 +2310,11 @@ struct Frame {
 
     u64 num_groups {};
     u64 num_lf_groups {};
+
+    Image image {};
 };
 
 static ErrorOr<Frame> read_frame(LittleEndianInputBitStream& stream,
-    Image& image,
     SizeHeader const& size_header,
     ImageMetadata const& metadata,
     Optional<EntropyDecoder>& entropy_decoder)
@@ -2352,12 +2353,12 @@ static ErrorOr<Frame> read_frame(LittleEndianInputBitStream& stream,
 
     frame.toc = TRY(read_toc(stream, frame.frame_header, frame.num_groups, frame.num_lf_groups));
 
-    image = TRY(Image::create({ frame.width, frame.height }, metadata));
+    frame.image = TRY(Image::create({ frame.width, frame.height }, metadata));
 
-    frame.lf_global = TRY(read_lf_global(stream, image, frame.frame_header, metadata, entropy_decoder));
+    frame.lf_global = TRY(read_lf_global(stream, frame.image, frame.frame_header, metadata, entropy_decoder));
 
     for (u32 i {}; i < frame.num_lf_groups; ++i)
-        TRY(read_lf_group(stream, image, frame.frame_header));
+        TRY(read_lf_group(stream, frame.image, frame.frame_header));
 
     if (frame.frame_header.encoding == FrameHeader::Encoding::kVarDCT) {
         TODO();
@@ -2366,13 +2367,13 @@ static ErrorOr<Frame> read_frame(LittleEndianInputBitStream& stream,
     auto const num_pass_group = frame.num_groups * frame.frame_header.passes.num_passes;
     auto const& transform_infos = frame.lf_global.gmodular.modular_header.transform;
     for (u64 i {}; i < num_pass_group; ++i)
-        TRY(read_pass_group(stream, image, frame.frame_header, group_dim));
+        TRY(read_pass_group(stream, frame.image, frame.frame_header, group_dim));
 
     // G.4.2 - Modular group data
     // When all modular groups are decoded, the inverse transforms are applied to
     // the at that point fully decoded GlobalModular image, as specified in H.6.
     for (auto const& transformation : transform_infos.in_reverse())
-        apply_transformation(image, transformation);
+        apply_transformation(frame.image, transformation);
 
     return frame;
 }
@@ -2391,7 +2392,7 @@ static u32 mirror_1d(i32 coord, u32 size)
 ///
 
 /// K - Image features
-static ErrorOr<void> apply_upsampling(Image& image, ImageMetadata const& metadata, Frame const& frame)
+static ErrorOr<void> apply_upsampling(Frame& frame, ImageMetadata const& metadata)
 {
     Optional<u32> ec_max;
     for (auto upsampling : frame.frame_header.ec_upsampling) {
@@ -2414,7 +2415,7 @@ static ErrorOr<void> apply_upsampling(Image& image, ImageMetadata const& metadat
         };
 
         // FIXME: Use ec_upsampling for extra-channels
-        for (auto& channel : image.channels()) {
+        for (auto& channel : frame.image.channels()) {
             auto upsampled = TRY(Channel::create(k * channel.width(), k * channel.height()));
 
             // Loop over the original image
@@ -2464,9 +2465,9 @@ static ErrorOr<void> apply_upsampling(Image& image, ImageMetadata const& metadat
     return {};
 }
 
-static ErrorOr<void> apply_image_features(Image& image, ImageMetadata const& metadata, Frame const& frame)
+static ErrorOr<void> apply_image_features(Frame& frame, ImageMetadata const& metadata)
 {
-    TRY(apply_upsampling(image, metadata, frame));
+    TRY(apply_upsampling(frame, metadata));
 
     if (frame.frame_header.flags != FrameHeader::Flags::None)
         TODO();
@@ -2497,10 +2498,10 @@ static void ycbcr_to_rgb(Image& image, u8 bits_per_sample)
     }
 }
 
-static void apply_colour_transformation(Image& image, ImageMetadata const& metadata, Frame const& frame)
+static void apply_colour_transformation(Frame& frame, ImageMetadata const& metadata)
 {
     if (frame.frame_header.do_YCbCr)
-        ycbcr_to_rgb(image, metadata.bit_depth.bits_per_sample);
+        ycbcr_to_rgb(frame.image, metadata.bit_depth.bits_per_sample);
 
     if (metadata.xyb_encoded) {
         TODO();
@@ -2548,20 +2549,18 @@ public:
 
     ErrorOr<void> decode_frame()
     {
-        Image image {};
-
-        auto const frame = TRY(read_frame(m_stream, image, m_header, m_metadata, m_entropy_decoder));
+        auto frame = TRY(read_frame(m_stream, m_header, m_metadata, m_entropy_decoder));
 
         if (frame.frame_header.restoration_filter.gab || frame.frame_header.restoration_filter.epf_iters != 0)
             TODO();
 
-        TRY(apply_image_features(image, m_metadata, frame));
+        TRY(apply_image_features(frame, m_metadata));
 
-        apply_colour_transformation(image, m_metadata, frame);
+        apply_colour_transformation(frame, m_metadata);
 
-        TRY(render_extra_channels(image, m_metadata));
+        TRY(render_extra_channels(frame.image, m_metadata));
 
-        m_bitmap = TRY(image.to_bitmap(m_metadata));
+        m_bitmap = TRY(frame.image.to_bitmap(m_metadata));
 
         return {};
     }