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

Add H265 as a parameter to SsimMapperTest.

Browse Source 
PiperOrigin-RevId: 463452893
This commit is contained in:
samrobinson 2022-07-26 23:24:03 +00:00 committed by tonihei
parent 302a742c94
commit 60d6654b8d
1 changed files with 48 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/analysis/SsimMapperTest.java
View File

@ -36,22 +36,27 @@ import static androidx.media3.transformer.AndroidTestUtil.MP4_REMOTE_7680W_4320H
import static androidx.media3.transformer.AndroidTestUtil.getFormatForTestFile;
import static androidx.media3.transformer.AndroidTestUtil.skipAndLogIfInsufficientCodecSupport;
import static androidx.media3.transformer.TransformationTestResult.SSIM_UNSET;
import static com.google.common.collect.Iterables.getLast;
import android.content.Context;
import android.net.Uri;
import androidx.annotation.Nullable;
import androidx.media3.common.Format;
import androidx.media3.common.MediaItem;
import androidx.media3.common.MimeTypes;
import androidx.media3.common.util.Util;
import androidx.media3.transformer.AndroidTestUtil;
import androidx.media3.transformer.DefaultEncoderFactory;
import androidx.media3.transformer.TransformationRequest;
import androidx.media3.transformer.Transformer;
import androidx.media3.transformer.TransformerAndroidTestRunner;
import androidx.media3.transformer.VideoEncoderSettings;
import androidx.test.core.app.ApplicationProvider;
import com.google.common.base.Splitter;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.junit.Test;
import org.junit.runner.RunWith;
@ -67,6 +72,8 @@ import org.junit.runners.Parameterized.Parameters;
@RunWith(Parameterized.class)
public class SsimMapperTest {
private static final Splitter FORWARD_SLASH_SPLITTER = Splitter.on('/');
// When running this test, input file list should be restricted more than this. Binary search can
// take up to 40 minutes to complete for a single clip on lower end devices.
private static final ImmutableList<String> INPUT_FILES =
@ -88,25 +95,46 @@ public class SsimMapperTest {
MP4_REMOTE_7680W_4320H_31_SECOND_ROOF_SAMSUNGS20ULTRA5G);
@Parameters
public static ImmutableList<String> parameters() {
return INPUT_FILES;
public static List<Object[]> parameters() {
List<Object[]> parameterList = new ArrayList<>();
for (String file : INPUT_FILES) {
parameterList.add(new Object[] {file, MimeTypes.VIDEO_H264});
// TODO(210593256): Test pre 24 once in-app muxing implemented.
if (Util.SDK_INT >= 24) {
parameterList.add(new Object[] {file, MimeTypes.VIDEO_H265});
}
}
return parameterList;
}
@Parameter @Nullable public String fileUri;
@Parameter(0)
@Nullable
public String fileUri;
@Parameter(1)
@Nullable
public String mimeType;
@Test
public void findSsimMapping() throws Exception {
String fileUri = checkNotNull(this.fileUri);
String mimeType = checkNotNull(this.mimeType);
String testIdPrefix =
String.format(
"ssim_search_VBR_%s", checkNotNull(getLast(FORWARD_SLASH_SPLITTER.split(mimeType))));
if (skipAndLogIfInsufficientCodecSupport(
ApplicationProvider.getApplicationContext(),
/* testId= */ "ssim_search_VBR_codecSupport",
testIdPrefix + "_codecSupport",
/* decodingFormat= */ getFormatForTestFile(fileUri),
/* encodingFormat= */ null)) {
return;
}
new SsimBinarySearcher(ApplicationProvider.getApplicationContext(), fileUri).search();
new SsimBinarySearcher(
ApplicationProvider.getApplicationContext(), testIdPrefix, fileUri, mimeType)
.search();
}
private static final class SsimBinarySearcher {
@ -116,8 +144,10 @@ public class SsimMapperTest {
private static final int MAX_TRANSFORMATIONS = 12;
private final Context context;
private final String testIdPrefix;
private final String videoUri;
private final Format format;
private final String outputMimeType;
private int transformationsLeft;
private double ssimLowerBound;
@ -129,11 +159,17 @@ public class SsimMapperTest {
* Creates a new instance.
*
* @param context The {@link Context}.
* @param testIdPrefix The test ID prefix.
* @param videoUri The URI of the video to transform.
* @param outputMimeType The video sample MIME type to output, see {@link
* TransformationRequest.Builder#setVideoMimeType}.
*/
public SsimBinarySearcher(Context context, String videoUri) {
public SsimBinarySearcher(
Context context, String testIdPrefix, String videoUri, String outputMimeType) {
this.context = context;
this.testIdPrefix = testIdPrefix;
this.videoUri = videoUri;
this.outputMimeType = outputMimeType;
transformationsLeft = MAX_TRANSFORMATIONS;
format = AndroidTestUtil.getFormatForTestFile(videoUri);
}
@ -222,8 +258,8 @@ public class SsimMapperTest {
private double transformAndGetSsim(int bitrate) throws Exception {
// TODO(b/238094555): Force specific encoders to be used.
String fileName = checkNotNull(Iterables.getLast(Splitter.on("/").split(videoUri)));
String testId = String.format("ssim_search_%s_VBR_%s", bitrate, fileName);
String fileName = checkNotNull(getLast(FORWARD_SLASH_SPLITTER.split(videoUri)));
String testId = String.format("%s_%s_%s", testIdPrefix, bitrate, fileName);
Map<String, Object> inputValues = new HashMap<>();
inputValues.put("targetBitrate", bitrate);
@ -236,6 +272,8 @@ public class SsimMapperTest {
Transformer transformer =
new Transformer.Builder(context)
.setRemoveAudio(true)
.setTransformationRequest(
new TransformationRequest.Builder().setVideoMimeType(outputMimeType).build())
.setEncoderFactory(
new DefaultEncoderFactory.Builder(context)
.setRequestedVideoEncoderSettings(
@ -243,6 +281,7 @@ public class SsimMapperTest {
.setBitrate(bitrate)
.setBitrateMode(BITRATE_MODE_VBR)
.build())
.setEnableFallback(false)
.build())
.build();