From 8228ffb56ddd02eeaf0ff2fe4c98ef16cdd9a420 Mon Sep 17 00:00:00 2001
From: Timofey Solomko <tsolomko@gmail.com>
Date: Sun, 13 Nov 2016 20:30:13 +0300
Subject: [PATCH] Added reversed argument back to findNextSymbol function.

---
 Sources/HuffmanTable.swift | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Sources/HuffmanTable.swift b/Sources/HuffmanTable.swift
index 92b7a19c..ccb7f438 100644
--- a/Sources/HuffmanTable.swift
+++ b/Sources/HuffmanTable.swift
@@ -90,7 +90,7 @@ class HuffmanTable: CustomStringConvertible {
         self.init(bootstrap: (zip(range, addedLengths)).map { [$0, $1] })
     }
 
-    func findNextSymbol(in bitArray: [UInt8]) -> HuffmanLength? {
+    func findNextSymbol(in bitArray: [UInt8], reversed: Bool = true) -> HuffmanLength? {
         var cachedLength = -1
         var cached: Int = -1
 
@@ -101,7 +101,8 @@ class HuffmanTable: CustomStringConvertible {
                 cached = convertToInt(uint8Array: Array(bitArray[0..<lbits]))
                 cachedLength = lbits
             }
-            if length.reversedSymbol == cached {
+            if (reversed && length.reversedSymbol == cached) ||
+                (!reversed && length.symbol == cached){
                 return length
             }
         }