From 528149fc1c63b1e146b0cfc1eacd59886e3b15bb Mon Sep 17 00:00:00 2001
From: winscripter <142818255+winscripter@users.noreply.github.com>
Date: Thu, 16 Jul 2026 14:05:35 +0400
Subject: [PATCH] Implement CreateFlatHistogram

See ans_common.h
---
 src/ImageSharp/Formats/Jxl/IO/JxlAnsHelper.cs | 24 +++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/src/ImageSharp/Formats/Jxl/IO/JxlAnsHelper.cs b/src/ImageSharp/Formats/Jxl/IO/JxlAnsHelper.cs
index babbfe8fd5..33b15e6dc5 100644
--- a/src/ImageSharp/Formats/Jxl/IO/JxlAnsHelper.cs
+++ b/src/ImageSharp/Formats/Jxl/IO/JxlAnsHelper.cs
@@ -22,4 +22,28 @@ internal static class JxlAnsHelper
 
         return r;
     }
+
+    // NOTE: The result may potentially be large, so prefer using a memory allocator
+    public static IMemoryOwner<int> CreateFlatHistogram(Configuration configuration, int length, int totalCount)
+    {
+        Debug.Assert(length <= 0, "Length should be >= 0");
+        Debug.Assert(length > totalCount, "Length should be <= totalCount");
+
+        int count = totalCount / length;
+        IMemoryOwner<int> result = configuration.MemoryAllocator.Allocate<int>(length);
+        Span<int> resultSpan = result.Memory.Span;
+
+        for (int i = 0; i < length; i++)
+        {
+            resultSpan[i] = count;
+        }
+
+        int remCounts = totalCount % length;
+        for (int i = 0; i < remCounts; i++)
+        {
+            resultSpan[i]++;
+        }
+
+        return result;
+    }
 }