loliball/media3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add stress test for image transform

Browse Source 
PiperOrigin-RevId: 642213253
This commit is contained in:
claincly 2024-06-11 03:57:25 -07:00 committed by Copybara-Service
parent acb6a89ff6
commit fd376bea07
2 changed files with 60 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libraries/transformer/src/androidTest/java/androidx/media3/transformer/AndroidTestUtil.java
View File

@ -815,17 +815,18 @@ public final class AndroidTestUtil {
*
* @param context The {@link Context context}.
* @param testId The test ID.
* @param inputFormat The {@link Format format} to decode.
* @param inputFormat The {@link Format format} to decode, or the input is not produced by
* MediaCodec, like an image.
* @param outputFormat The {@link Format format} to encode/mux or {@code null} if the output won't
* be encoded or muxed.
* @throws AssumptionViolatedException If the device does not support the formats. In this case,
* the reason for skipping the test is logged.
*/
public static void assumeFormatsSupported(
Context context, String testId, Format inputFormat, @Nullable Format outputFormat)
Context context, String testId, @Nullable Format inputFormat, @Nullable Format outputFormat)
throws IOException, JSONException, MediaCodecUtil.DecoderQueryException {
// TODO(b/278657595): Make this capability check match the default codec factory selection code.
boolean canDecode = canDecode(inputFormat);
boolean canDecode = inputFormat == null || canDecode(inputFormat);
boolean canEncode = outputFormat == null || canEncode(outputFormat);
boolean canMux = outputFormat == null || canMux(outputFormat);

59
libraries/transformer/src/androidTest/java/androidx/media3/transformer/TransformerMixedInputEndToEndTest.java
View File

@ -41,6 +41,8 @@ import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import com.google.common.collect.ImmutableList;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
@ -77,6 +79,8 @@ public class TransformerMixedInputEndToEndTest {
testId = testName.getMethodName();
}
// TODO: b/343362776 - Add tests to assert enough silence is generated.
@Test
public void videoEditing_withImageThenVideoInputs_completesWithCorrectFrameCount()
throws Exception {
@ -133,6 +137,47 @@ public class TransformerMixedInputEndToEndTest {
assertThat(new File(result.filePath).length()).isGreaterThan(0);
}
@Test
public void videoEditing_withShortAlternatingImages_completesWithCorrectFrameCount()
throws Exception {
Transformer transformer =
new Transformer.Builder(context)
.setEncoderFactory(
new DefaultEncoderFactory.Builder(context).setEnableFallback(false).build())
.build();
EditedMediaItem image1 =
new EditedMediaItem.Builder(MediaItem.fromUri(PNG_ASSET_URI_STRING))
.setDurationUs(100_000)
.setFrameRate(30)
.build();
int image1FrameCount = 3;
EditedMediaItem image2 =
new EditedMediaItem.Builder(MediaItem.fromUri(JPG_ASSET_URI_STRING))
.setDurationUs(200_000)
.setFrameRate(30)
.build();
int image2FrameCount = 6;
ArrayList<EditedMediaItem> editedMediaItems = new ArrayList<>(100);
for (int i = 0; i < 50; i++) {
editedMediaItems.add(image1);
editedMediaItems.add(image2);
}
ExportTestResult result =
new TransformerAndroidTestRunner.Builder(context, transformer)
.build()
.run(testId, buildComposition(editedMediaItems));
// TODO: b/346289922 - Check frame count with extractors.
assertThat(result.exportResult.videoFrameCount)
.isEqualTo(50 * image1FrameCount + 50 * image2FrameCount);
// 50 100ms-images and 50 200ms-images
assertThat(result.exportResult.durationMs).isEqualTo(14_966);
assertThat(new File(result.filePath).length()).isGreaterThan(0);
}
@Test
public void
videoEditing_withComplexMixedColorSpaceSdrVideoAndImageInputsEndWithVideo_completesWithCorrectFrameCount()
@ -343,9 +388,8 @@ public class TransformerMixedInputEndToEndTest {
.build();
}
private static Composition buildComposition(
EditedMediaItem editedMediaItem, EditedMediaItem... editedMediaItems) {
return new Composition.Builder(new EditedMediaItemSequence(editedMediaItem, editedMediaItems))
private static Composition buildComposition(List<EditedMediaItem> editedMediaItems) {
return new Composition.Builder(new EditedMediaItemSequence(editedMediaItems))
.setEffects(
new Effects(
/* audioProcessors= */ ImmutableList.of(),
@ -355,4 +399,13 @@ public class TransformerMixedInputEndToEndTest {
/* width= */ 480, /* height= */ 360, Presentation.LAYOUT_SCALE_TO_FIT))))
.build();
}
private static Composition buildComposition(
EditedMediaItem editedMediaItem, EditedMediaItem... editedMediaItems) {
return buildComposition(
new ImmutableList.Builder<EditedMediaItem>()
.add(editedMediaItem)
.add(editedMediaItems)
.build());
}
}