From 6ac8a5d8b45c18603a6cb185ddfdc303cdc510ce Mon Sep 17 00:00:00 2001
From: Harald Kuhr <harald.kuhr@gmail.com>
Date: Tue, 20 Apr 2021 22:41:41 +0200
Subject: [PATCH] Minor optimization for standard case with only one image.

---
 .../twelvemonkeys/imageio/plugins/jpeg/JPEGImageReader.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imageio/imageio-jpeg/src/main/java/com/twelvemonkeys/imageio/plugins/jpeg/JPEGImageReader.java b/imageio/imageio-jpeg/src/main/java/com/twelvemonkeys/imageio/plugins/jpeg/JPEGImageReader.java
index a09fd4b3..43934b99 100644
--- a/imageio/imageio-jpeg/src/main/java/com/twelvemonkeys/imageio/plugins/jpeg/JPEGImageReader.java
+++ b/imageio/imageio-jpeg/src/main/java/com/twelvemonkeys/imageio/plugins/jpeg/JPEGImageReader.java
@@ -817,13 +817,14 @@ public final class JPEGImageReader extends ImageReaderBase {
 
                 // TODO: We should probably optimize this
                 try {
+                    segments = null;
                     getSOF(); // No SOF, no image
                     count++;
                 }
                 catch (IIOException ignore) {}
             }
 
-            currentStreamIndex = -1;
+            imageInput.seek(streamOffsets.get(currentStreamIndex));
 
             return count;
         }