From 3a66d270b10133232d6e0d54ec28cdb27a006763 Mon Sep 17 00:00:00 2001
From: greatroar <61184462+greatroar@users.noreply.github.com>
Date: Sat, 26 Nov 2022 19:06:37 +0100
Subject: [PATCH] zstd: Revert #699

This reverts commit 2878205bda99754286840d77f1d27331374f5e3a.

Turns out that the speedup was due to a bug. In:

        offset := uint32(m.rep)
        if offset < 0 {
                offset = uint32(m.s-m.offset) + 3
        }

offset is never < 0 because it's cast to uint32 too early.
---
 zstd/enc_best.go | 32 +++++++++++++++-----------------
 1 file changed, 15 insertions(+), 17 deletions(-)

diff --git a/zstd/enc_best.go b/zstd/enc_best.go
index ab0d1b7cd9..b1c3db8f7b 100644
--- a/zstd/enc_best.go
+++ b/zstd/enc_best.go
@@ -38,29 +38,27 @@ type match struct {
 const highScore = 25000
 
 // estBits will estimate output bits from predefined tables.
-func (m *match) estBits(bitsPerByte int32, ofc, mlc uint8) {
+func (m *match) estBits(bitsPerByte int32) {
+	mlc := mlCode(uint32(m.length - zstdMinMatch))
+	var ofc uint8
+	if m.rep < 0 {
+		ofc = ofCode(uint32(m.s-m.offset) + 3)
+	} else {
+		ofc = ofCode(uint32(m.rep))
+	}
 	// Cost, excluding
 	ofTT, mlTT := fsePredefEnc[tableOffsets].ct.symbolTT[ofc], fsePredefEnc[tableMatchLengths].ct.symbolTT[mlc]
 
 	// Add cost of match encoding...
-	est := int32(ofTT.outBits + mlTT.outBits)
-	est += int32(ofTT.deltaNbBits>>16 + mlTT.deltaNbBits>>16)
+	m.est = int32(ofTT.outBits + mlTT.outBits)
+	m.est += int32(ofTT.deltaNbBits>>16 + mlTT.deltaNbBits>>16)
 	// Subtract savings compared to literal encoding...
-	length := m.length
-	est -= (length * bitsPerByte) >> 10
-	if est > 0 {
+	m.est -= (m.length * bitsPerByte) >> 10
+	if m.est > 0 {
 		// Unlikely gain..
-		est, length = highScore, 0
-	}
-	m.est, m.length = est, length
-}
-
-func (m *match) ofCode() uint8 {
-	offset := uint32(m.rep)
-	if offset < 0 {
-		offset = uint32(m.s-m.offset) + 3
+		m.length = 0
+		m.est = highScore
 	}
-	return ofCode(offset)
 }
 
 // bestFastEncoder uses 2 tables, one for short matches (5 bytes) and one for long matches.
@@ -218,7 +216,7 @@ encodeLoop:
 				}
 			}
 			m := match{offset: offset, s: s, length: 4 + e.matchlen(s+4, offset+4, src), rep: rep}
-			m.estBits(bitsPerByte, m.ofCode(), mlCode(uint32(m.length-zstdMinMatch)))
+			m.estBits(bitsPerByte)
 			return m
 		}