Fix modular channels section assignment.

Channels after the first "large" one are not in LF global; the existing code assumed decisions were made per channel.
libjxl · Dec 24, 2024 · d160829 · d160829
1 parent c138847
commit d160829
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jxl/src/frame/modular.rs b/jxl/src/frame/modular.rs
@@ -138,7 +138,7 @@ impl FullModularImage {
                 .iter()
                 .enumerate()
                 .filter(|x| x.1.is_coded)
-                .filter(|x| x.1.info.is_meta_or_small(frame_header.group_dim()))
+                .take_while(|x| x.1.info.is_meta_or_small(frame_header.group_dim()))
                 .map(|x| x.0)
                 .collect(),
         );
@@ -148,7 +148,7 @@ impl FullModularImage {
                 .iter()
                 .enumerate()
                 .filter(|x| x.1.is_coded)
-                .filter(|x| !x.1.info.is_meta_or_small(frame_header.group_dim()))
+                .skip_while(|x| x.1.info.is_meta_or_small(frame_header.group_dim()))
                 .filter(|x| x.1.info.is_shift_in_range(3, usize::MAX))
                 .map(|x| x.0)
                 .collect(),