sjs/media
1
0
Fork 0
mirror of https://github.com/samsonjs/media.git synced 2026-04-16 13:05:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix H264 reader key frame detection

Browse source
This commit is contained in:
Daniele Sparano 2023-11-29 13:09:27 +00:00 committed by tonihei
parent 379cb3ba54
commit d4af13803a
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
View file

@ -183,7 +183,7 @@ public final class H264Reader implements ElementaryStreamReader {
pps.startNalUnit(nalUnitType);
}
sei.startNalUnit(nalUnitType);
sampleReader.startNalUnit(position, nalUnitType, pesTimeUs);
sampleReader.startNalUnit(position, nalUnitType, pesTimeUs, randomAccessIndicator);
}
@RequiresNonNull("sampleReader")
@ -254,7 +254,7 @@ public final class H264Reader implements ElementaryStreamReader {
seiReader.consume(pesTimeUs, seiWrapper);
}
boolean sampleIsKeyFrame =
sampleReader.endNalUnit(position, offset, hasOutputFormat, randomAccessIndicator);
sampleReader.endNalUnit(position, offset, hasOutputFormat);
if (sampleIsKeyFrame) {
// This is either an IDR frame or the first I-frame since the random access indicator, so mark
// it as a keyframe. Clear the flag so that subsequent non-IDR I-frames are not marked as
@ -297,6 +297,7 @@ public final class H264Reader implements ElementaryStreamReader {
private long samplePosition;
private long sampleTimeUs;
private boolean sampleIsKeyframe;
private boolean randomAccessIndicator;
public SampleReader(
TrackOutput output, boolean allowNonIdrKeyframes, boolean detectAccessUnits) {
@ -330,10 +331,11 @@ public final class H264Reader implements ElementaryStreamReader {
sliceHeader.clear();
}
public void startNalUnit(long position, int type, long pesTimeUs) {
public void startNalUnit(long position, int type, long pesTimeUs, boolean rai) {
nalUnitType = type;
nalUnitTimeUs = pesTimeUs;
nalUnitStartPosition = position;
randomAccessIndicator = rai;
if ((allowNonIdrKeyframes && nalUnitType == NalUnitUtil.NAL_UNIT_TYPE_NON_IDR)
|| (detectAccessUnits
&& (nalUnitType == NalUnitUtil.NAL_UNIT_TYPE_IDR
@ -482,7 +484,7 @@ public final class H264Reader implements ElementaryStreamReader {
}
public boolean endNalUnit(
long position, int offset, boolean hasOutputFormat, boolean randomAccessIndicator) {
long position, int offset, boolean hasOutputFormat) {
if (nalUnitType == NalUnitUtil.NAL_UNIT_TYPE_AUD
|| (detectAccessUnits && sliceHeader.isFirstVclNalUnitOfPicture(previousSliceHeader))) {
// If the NAL unit ending is the start of a new sample, output the previous one.