sleep time.

Codecs/Image/Jpeg/Classes/HuffmanTable.cs

@@ -49,12 +49,9 @@ internal void BuildCodeTable()
         throw new ArgumentException($"Value {value} not found in Huffman table.");
     }
 
-    public bool TryGetCode(int code, int length, out (ushort code, byte length) value)
+    public bool TryGetCode(int code, out (ushort code, byte length) value)
     {
-        // Convert the code and length to a byte key
-        byte key = (byte)((code << 4) | length);
-
-        if (_codeTable.TryGetValue(key, out value))
+        if (_codeTable.TryGetValue((byte)code, out value))
         {
             return true;
         }

Codecs/Image/Jpeg/JpegCodec.cs

@@ -255,9 +255,9 @@ private static int DecodeHuffman(BitReader stream, HuffmanTable table)
                 length++;
 
                 // Try to get the Huffman code from the table
-                if (table.TryGetCode(code, length, out var value))
+                if (table.TryGetCode(code, out var value))
                 {
-                    return value.code;
+                    return value.length;
                 }
 
                 // Check for an invalid length (e.g., exceeding the maximum code length)
@@ -270,8 +270,8 @@ private static int DecodeHuffman(BitReader stream, HuffmanTable table)
 
         private static short[] ReadBlock(BitReader stream, HuffmanTable dcTable, HuffmanTable acTable, ref int previousDC)
         {
+            //Todo, should not build every time called.
             dcTable.BuildCodeTable();
-
             acTable.BuildCodeTable();
 
             var block = new short[BlockSize * BlockSize];  // Assuming 8x8 block
@@ -337,6 +337,10 @@ internal static void Decompress(JpegImage jpegImage)
 
                 var quantTable = jpegImage.JpegState.GetQuantizationTable(component.Tqi);
 
+                //Should have logging support
+                if (dcTable == null || acTable == null || quantTable == null)
+                    continue;
+
                 int previousDC = 0;
                 for (int i = 0; i < BlockSize; i++)
                 {