Merge "Merge "Extend MediaCodecInfo to support AV1 decoder" into qt-dev am:... (8ef9c246) · Commits · e / os / android_frameworks_base

media/java/android/media/MediaCodecInfo.java

+87 −0

Original line number	Diff line number	Diff line
		@@ -3145,6 +3145,93 @@ public final class MediaCodecInfo {
		maxBlocks, maxBlocksPerSecond,
		8 /* blockWidth /, 8 / blockHeight */,
		1 /* widthAlignment /, 1 / heightAlignment */);
		} else if (mime.equalsIgnoreCase(MediaFormat.MIMETYPE_VIDEO_AV1)) {
		maxBlocksPerSecond = 829440;
		maxBlocks = 36864;
		maxBps = 200000;
		int maxDim = 512;

		// Sample rate, Picture Size, Bit rate and luma dimension for AV1 Codec,
		// corresponding to the definitions in
		// "AV1 Bitstream & Decoding Process Specification", Annex A
		// found at https://aomedia.org/av1-bitstream-and-decoding-process-specification/
		for (CodecProfileLevel profileLevel: profileLevels) {
		long SR = 0; // luma sample rate
		int FS = 0; // luma picture size
		int BR = 0; // bit rate kbps
		int D = 0; // luma D
		switch (profileLevel.level) {
		case CodecProfileLevel.AV1Level2:
		SR = 5529600; FS = 147456; BR = 1500; D = 2048; break;
		case CodecProfileLevel.AV1Level21:
		case CodecProfileLevel.AV1Level22:
		case CodecProfileLevel.AV1Level23:
		SR = 10454400; FS = 278784; BR = 3000; D = 2816; break;

		case CodecProfileLevel.AV1Level3:
		SR = 24969600; FS = 665856; BR = 6000; D = 4352; break;
		case CodecProfileLevel.AV1Level31:
		case CodecProfileLevel.AV1Level32:
		case CodecProfileLevel.AV1Level33:
		SR = 39938400; FS = 1065024; BR = 10000; D = 5504; break;

		case CodecProfileLevel.AV1Level4:
		SR = 77856768; FS = 2359296; BR = 12000; D = 6144; break;
		case CodecProfileLevel.AV1Level41:
		case CodecProfileLevel.AV1Level42:
		case CodecProfileLevel.AV1Level43:
		SR = 155713536; FS = 2359296; BR = 20000; D = 6144; break;

		case CodecProfileLevel.AV1Level5:
		SR = 273715200; FS = 8912896; BR = 30000; D = 8192; break;
		case CodecProfileLevel.AV1Level51:
		SR = 547430400; FS = 8912896; BR = 40000; D = 8192; break;
		case CodecProfileLevel.AV1Level52:
		SR = 1094860800; FS = 8912896; BR = 60000; D = 8192; break;
		case CodecProfileLevel.AV1Level53:
		SR = 1176502272; FS = 8912896; BR = 60000; D = 8192; break;

		case CodecProfileLevel.AV1Level6:
		SR = 1176502272; FS = 35651584; BR = 60000; D = 16384; break;
		case CodecProfileLevel.AV1Level61:
		SR = 2189721600L; FS = 35651584; BR = 100000; D = 16384; break;
		case CodecProfileLevel.AV1Level62:
		SR = 4379443200L; FS = 35651584; BR = 160000; D = 16384; break;
		case CodecProfileLevel.AV1Level63:
		SR = 4706009088L; FS = 35651584; BR = 160000; D = 16384; break;

		default:
		Log.w(TAG, "Unrecognized level "
		+ profileLevel.level + " for " + mime);
		errors \|= ERROR_UNRECOGNIZED;
		}
		switch (profileLevel.profile) {
		case CodecProfileLevel.AV1ProfileMain8:
		case CodecProfileLevel.AV1ProfileMain10:
		case CodecProfileLevel.AV1ProfileMain10HDR10:
		case CodecProfileLevel.AV1ProfileMain10HDR10Plus:
		break;
		default:
		Log.w(TAG, "Unrecognized profile "
		+ profileLevel.profile + " for " + mime);
		errors \|= ERROR_UNRECOGNIZED;
		}
		errors &= ~ERROR_NONE_SUPPORTED;
		maxBlocksPerSecond = Math.max(SR, maxBlocksPerSecond);
		maxBlocks = Math.max(FS, maxBlocks);
		maxBps = Math.max(BR * 1000, maxBps);
		maxDim = Math.max(D, maxDim);
		}

		final int blockSize = 8;
		int maxLengthInBlocks = Utils.divUp(maxDim, blockSize);
		maxBlocks = Utils.divUp(maxBlocks, blockSize * blockSize);
		maxBlocksPerSecond = Utils.divUp(maxBlocksPerSecond, blockSize * blockSize);
		applyMacroBlockLimits(
		maxLengthInBlocks, maxLengthInBlocks,
		maxBlocks, maxBlocksPerSecond,
		blockSize, blockSize,
		1 /* widthAlignment /, 1 / heightAlignment */);
		} else {
		Log.w(TAG, "Unsupported mime " + mime);
		// using minimal bitrate here. should be overriden by