From c144420be8e60861a7eac6a62f10896b041b6d98 Mon Sep 17 00:00:00 2001
From: Daniele Sparano <daniele.sparano@v-nova.com>
Date: Wed, 12 Feb 2025 14:21:16 +0000
Subject: [PATCH] Reset nal unit type after key frame determination

---
 .../src/main/java/androidx/media3/extractor/ts/H264Reader.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java b/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
index e755e1a57a..952940e1b7 100644
--- a/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
+++ b/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
@@ -518,9 +518,9 @@ public final class H264Reader implements ElementaryStreamReader {
         sampleIsKeyframe = false;
         readingSample = true;
       }
+      setSampleIsKeyframe();
       // Reset NAL unit type to avoid stale state
       nalUnitType = NalUnitUtil.H264_NAL_UNIT_TYPE_UNSPECIFIED;
-      setSampleIsKeyframe();
       return sampleIsKeyframe;
     }