avformat/mpegts: Simplify score compare

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>

diff --git a/libavformat/mpegts.c b/libavformat/mpegts.c
index 773b7a038486bcb5fd526ad4a79ca572cffa02b4..8f686393886e59a26b8eaa0d8913579f82863660 100644 (file)
--- a/libavformat/mpegts.c
+++ b/libavformat/mpegts.c
@@ -620,16 +620,16 @@ static int get_packet_size(AVFormatContext* s)
         av_log(s, AV_LOG_TRACE, "Probe: %d, score: %d, dvhs_score: %d, fec_score: %d \n",
             buf_size, score, dvhs_score, fec_score);
 
+        margin = mid_pred(score, fec_score, dvhs_score);
+
         if (buf_size < PROBE_PACKET_MAX_BUF)
-            margin = PROBE_PACKET_MARGIN; /*if buffer not filled */
-        else
-            margin = 0;
+            margin += PROBE_PACKET_MARGIN; /*if buffer not filled */
 
-        if (score > FFMAX(fec_score, dvhs_score) + margin)
+        if (score > margin)
             return TS_PACKET_SIZE;
-        else if (dvhs_score > FFMAX(score, fec_score) + margin)
+        else if (dvhs_score > margin)
             return TS_DVHS_PACKET_SIZE;
-        else if (fec_score > FFMAX(score, dvhs_score) + margin)
+        else if (fec_score > margin)
             return TS_FEC_PACKET_SIZE;
     }
     return AVERROR_INVALIDDATA;