loliball/media
1
0
Fork 0
mirror of https://github.com/androidx/media.git synced 2025-04-30 06:46:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Record transformation duration in TransformerAndroidTestRunner.

Browse Source 
The change will be useful in testing transcoding performance

PiperOrigin-RevId: 432956283
This commit is contained in:
claincly 2022-03-07 16:52:37 +00:00 committed by Ian Baker
parent 743c6cf5f8
commit 5ab32b6e0d
2 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libraries/transformer/src/androidTest/java/androidx/media3/transformer/TransformationTestResult.java
View File

@ -22,17 +22,24 @@ public class TransformationTestResult {
public final TransformationResult transformationResult;
public final String filePath;
/** The amount of time taken to perform the transformation in milliseconds. */
public final long transformationDurationMs;
/** The SSIM score of the transformation, {@link #SSIM_UNSET} if unavailable. */
public final double ssim;
public TransformationTestResult(TransformationResult transformationResult, String filePath) {
this(transformationResult, filePath, /* ssim= */ SSIM_UNSET);
public TransformationTestResult(
TransformationResult transformationResult, String filePath, long transformationDurationMs) {
this(transformationResult, filePath, transformationDurationMs, /* ssim= */ SSIM_UNSET);
}
public TransformationTestResult(
TransformationResult transformationResult, String filePath, double ssim) {
TransformationResult transformationResult,
String filePath,
long transformationDurationMs,
double ssim) {
this.transformationResult = transformationResult;
this.filePath = filePath;
this.transformationDurationMs = transformationDurationMs;
this.ssim = ssim;
}
}

10
libraries/transformer/src/androidTest/java/androidx/media3/transformer/TransformerAndroidTestRunner.java
View File

@ -25,6 +25,7 @@ import androidx.annotation.Nullable;
import androidx.media3.common.C;
import androidx.media3.common.MediaItem;
import androidx.media3.common.util.Log;
import androidx.media3.common.util.SystemClock;
import androidx.test.platform.app.InstrumentationRegistry;
import java.io.File;
import java.io.FileWriter;
@ -157,6 +158,7 @@ public class TransformerAndroidTestRunner {
AtomicReference<@NullableType TransformationResult> transformationResultReference =
new AtomicReference<>();
CountDownLatch countDownLatch = new CountDownLatch(1);
long startTimeMs = SystemClock.DEFAULT.elapsedRealtime();
Transformer testTransformer =
transformer
@ -199,6 +201,7 @@ public class TransformerAndroidTestRunner {
if (!countDownLatch.await(timeoutSeconds, SECONDS)) {
throw new TimeoutException("Transformer timed out after " + timeoutSeconds + " seconds.");
}
long transformationDurationMs = SystemClock.DEFAULT.elapsedRealtime() - startTimeMs;
@Nullable Exception unexpectedException = unexpectedExceptionReference.get();
if (unexpectedException != null) {
@ -220,13 +223,15 @@ public class TransformerAndroidTestRunner {
.build();
if (!calculateSsim) {
return new TransformationTestResult(transformationResult, outputVideoFile.getPath());
return new TransformationTestResult(
transformationResult, outputVideoFile.getPath(), transformationDurationMs);
}
double ssim =
SsimHelper.calculate(
context, /* expectedVideoPath= */ uriString, outputVideoFile.getPath());
return new TransformationTestResult(transformationResult, outputVideoFile.getPath(), ssim);
return new TransformationTestResult(
transformationResult, outputVideoFile.getPath(), transformationDurationMs, ssim);
}
private static void writeTestSummaryToFile(Context context, String testId, JSONObject resultJson)
@ -270,6 +275,7 @@ public class TransformerAndroidTestRunner {
if (testResult.ssim != TransformationTestResult.SSIM_UNSET) {
transformationResultJson.put("ssim", testResult.ssim);
}
transformationResultJson.put("transformationDurationMs", testResult.transformationDurationMs);
return transformationResultJson;
}