libs/deviceutil/src/android/cts/util/MediaUtils.java - platform/cts - Git at Google

 /*
  * Copyright 2014 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package android.cts.util;

 import android.content.Context;
 import android.content.res.AssetFileDescriptor;
 import android.media.MediaCodecInfo;
 import android.media.MediaCodecList;
 import android.media.MediaExtractor;
 import android.media.MediaFormat;
 import android.net.Uri;
 import java.lang.reflect.Method;
 import static java.lang.reflect.Modifier.isPublic;
 import static java.lang.reflect.Modifier.isStatic;
 import java.util.Map;
 import android.util.Log;

 import java.io.IOException;

 public class MediaUtils {
     private static final String TAG = "MediaUtils";

     private static final int ALL_AV_TRACKS = -1;

     private static final MediaCodecList sMCL = new MediaCodecList(MediaCodecList.REGULAR_CODECS);


     /**
      * Finds test name (heuristically) and prints out standard skip message.
      *
      * Since it uses heuristics, this method has only been verified for media
      * tests. This centralizes the way to signal a skipped test.
      */
     public static void skipTest(String tag, String reason) {
         int bestScore = -1;
         String testName = "test???";
         Map<Thread, StackTraceElement[]> traces = Thread.getAllStackTraces();
         for (Map.Entry<Thread, StackTraceElement[]> entry : traces.entrySet()) {
             StackTraceElement[] stack = entry.getValue();
             for (int index = 0; index < stack.length; ++index) {
                 // method name must start with "test"
                 String methodName = stack[index].getMethodName();
                 if (!methodName.startsWith("test")) {
                     continue;
                 }

                 int score = 0;
                 // see if there is a public non-static void method that takes no argument
                 Class<?> clazz;
                 try {
                     clazz = Class.forName(stack[index].getClassName());
                     ++score;
                     for (final Method method : clazz.getDeclaredMethods()) {
                         if (method.getName().equals(methodName)
                                 && isPublic(method.getModifiers())
                                 && !isStatic(method.getModifiers())
                                 && method.getParameterTypes().length == 0
                                 && method.getReturnType().equals(Void.TYPE)) {
                             ++score;
                             break;
                         }
                     }
                     if (score == 1) {
                         // if we could read the class, but method is not public void, it is
                         // not a candidate
                         continue;
                     }
                 } catch (ClassNotFoundException e) {
                 }

                 // even if we cannot verify the method signature, there are signals in the stack

                 // usually test method is invoked by reflection
                 int depth = 1;
                 while (index + depth < stack.length
                         && stack[index + depth].getMethodName().equals("invoke")
                         && stack[index + depth].getClassName().equals(
                                 "java.lang.reflect.Method")) {
                     ++depth;
                 }
                 if (depth > 1) {
                     ++score;
                     // and usually test method is run by runMethod method in android.test package
                     if (index + depth < stack.length) {
                         if (stack[index + depth].getClassName().startsWith("android.test.")) {
                             ++score;
                         }
                         if (stack[index + depth].getMethodName().equals("runMethod")) {
                             ++score;
                         }
                     }
                 }

                 if (score > bestScore) {
                     bestScore = score;
                     testName = methodName;
                 }
             }
         }

         Log.i(tag, "SKIPPING " + testName + "(): " + reason);
     }

     /**
      * Finds test name (heuristically) and prints out standard skip message.
      *
      * Since it uses heuristics, this method has only been verified for media
      * tests.  This centralizes the way to signal a skipped test.
      */
     public static void skipTest(String reason) {
         skipTest(TAG, reason);
     }

     public static boolean check(boolean result, String message) {
         if (!result) {
             skipTest(message);
         }
         return result;
     }

     public static boolean canDecode(MediaFormat format) {
         if (sMCL.findDecoderForFormat(format) == null) {
             Log.i(TAG, "no decoder for " + format);
             return false;
         }
         return true;
     }

     public static boolean hasCodecForTrack(MediaExtractor ex, int track) {
         int count = ex.getTrackCount();
         if (track < 0 || track >= count) {
             throw new IndexOutOfBoundsException(track + " not in [0.." + (count - 1) + "]");
         }
         return canDecode(ex.getTrackFormat(track));
     }

     /**
      * return true iff all audio and video tracks are supported
      */
     public static boolean hasCodecsForMedia(MediaExtractor ex) {
         for (int i = 0; i < ex.getTrackCount(); ++i) {
             MediaFormat format = ex.getTrackFormat(i);
             // only check for audio and video codecs
             String mime = format.getString(MediaFormat.KEY_MIME).toLowerCase();
             if (!mime.startsWith("audio/") && !mime.startsWith("video/")) {
                 continue;
             }
             if (!canDecode(format)) {
                 return false;
             }
         }
         return true;
     }

     /**
      * return true iff any track starting with mimePrefix is supported
      */
     public static boolean hasCodecForMediaAndDomain(MediaExtractor ex, String mimePrefix) {
         mimePrefix = mimePrefix.toLowerCase();
         for (int i = 0; i < ex.getTrackCount(); ++i) {
             MediaFormat format = ex.getTrackFormat(i);
             String mime = format.getString(MediaFormat.KEY_MIME);
             if (mime.toLowerCase().startsWith(mimePrefix)) {
                 if (canDecode(format)) {
                     return true;
                 }
                 Log.i(TAG, "no decoder for " + format);
             }
         }
         return false;
     }

     private static boolean hasCodecsForResourceCombo(
             Context context, int resourceId, int track, String mimePrefix) {
         try {
             AssetFileDescriptor afd = null;
             MediaExtractor ex = null;
             try {
                 afd = context.getResources().openRawResourceFd(resourceId);
                 ex = new MediaExtractor();
                 ex.setDataSource(afd.getFileDescriptor(), afd.getStartOffset(), afd.getLength());
                 if (mimePrefix != null) {
                     return hasCodecForMediaAndDomain(ex, mimePrefix);
                 } else if (track == ALL_AV_TRACKS) {
                     return hasCodecsForMedia(ex);
                 } else {
                     return hasCodecForTrack(ex, track);
                 }
             } finally {
                 if (ex != null) {
                     ex.release();
                 }
                 if (afd != null) {
                     afd.close();
                 }
             }
         } catch (IOException e) {
             Log.i(TAG, "could not open resource");
         }
         return false;
     }

     /**
      * return true iff all audio and video tracks are supported
      */
     public static boolean hasCodecsForResource(Context context, int resourceId) {
         return hasCodecsForResourceCombo(context, resourceId, ALL_AV_TRACKS, null /* mimePrefix */);
     }

     public static boolean checkCodecsForResource(Context context, int resourceId) {
         return check(hasCodecsForResource(context, resourceId), "no decoder found");
     }

     /**
      * return true iff track is supported.
      */
     public static boolean hasCodecForResource(Context context, int resourceId, int track) {
         return hasCodecsForResourceCombo(context, resourceId, track, null /* mimePrefix */);
     }

     public static boolean checkCodecForResource(Context context, int resourceId, int track) {
         return check(hasCodecForResource(context, resourceId, track), "no decoder found");
     }

     /**
      * return true iff any track starting with mimePrefix is supported
      */
     public static boolean hasCodecForResourceAndDomain(
             Context context, int resourceId, String mimePrefix) {
         return hasCodecsForResourceCombo(context, resourceId, ALL_AV_TRACKS, mimePrefix);
     }

     /**
      * return true iff all audio and video tracks are supported
      */
     public static boolean hasCodecsForPath(Context context, String path) {
         MediaExtractor ex = null;
         try {
             ex = new MediaExtractor();
             Uri uri = Uri.parse(path);
             String scheme = uri.getScheme();
             if (scheme == null) { // file
                 ex.setDataSource(path);
             } else if (scheme.equalsIgnoreCase("file")) {
                 ex.setDataSource(uri.getPath());
             } else {
                 ex.setDataSource(context, uri, null);
             }
             return hasCodecsForMedia(ex);
         } catch (IOException e) {
             Log.i(TAG, "could not open path " + path);
         } finally {
             if (ex != null) {
                 ex.release();
             }
         }
         return false;
     }

     public static boolean checkCodecsForPath(Context context, String path) {
         return check(hasCodecsForPath(context, path), "no decoder found");
     }

     private static boolean hasCodecForMime(boolean encoder, String mime) {
         for (MediaCodecInfo info : sMCL.getCodecInfos()) {
             if (encoder != info.isEncoder()) {
                 continue;
             }

             for (String type : info.getSupportedTypes()) {
                 if (type.equalsIgnoreCase(mime)) {
                     Log.i(TAG, "found codec " + info.getName() + " for mime " + mime);
                     return true;
                 }
             }
         }
         return false;
     }

     private static boolean hasCodecForMimes(boolean encoder, String[] mimes) {
         for (String mime : mimes) {
             if (!hasCodecForMime(encoder, mime)) {
                 Log.i(TAG, "no " + (encoder ? "encoder" : "decoder") + " for mime " + mime);
                 return false;
             }
         }
         return true;
     }


     public static boolean hasEncoder(String... mimes) {
         return hasCodecForMimes(true /* encoder */, mimes);
     }

     public static boolean hasDecoder(String... mimes) {
         return hasCodecForMimes(false /* encoder */, mimes);
     }

     public static boolean checkDecoder(String... mimes) {
         return check(hasCodecForMimes(false /* encoder */, mimes), "no decoder found");
     }

     public static boolean checkEncoder(String... mimes) {
         return check(hasCodecForMimes(true /* encoder */, mimes), "no encoder found");
     }

     public static boolean canDecodeVideo(String mime, int width, int height, float rate) {
         MediaFormat format = MediaFormat.createVideoFormat(mime, width, height);
         format.setFloat(MediaFormat.KEY_FRAME_RATE, rate);
         return canDecode(format);
     }

     public static boolean checkDecoderForFormat(MediaFormat format) {
         return check(canDecode(format), "no decoder for " + format);
     }
 }
	/*
	* Copyright 2014 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package android.cts.util;

	import android.content.Context;
	import android.content.res.AssetFileDescriptor;
	import android.media.MediaCodecInfo;
	import android.media.MediaCodecList;
	import android.media.MediaExtractor;
	import android.media.MediaFormat;
	import android.net.Uri;
	import java.lang.reflect.Method;
	import static java.lang.reflect.Modifier.isPublic;
	import static java.lang.reflect.Modifier.isStatic;
	import java.util.Map;
	import android.util.Log;

	import java.io.IOException;

	public class MediaUtils {
	private static final String TAG = "MediaUtils";

	private static final int ALL_AV_TRACKS = -1;

	private static final MediaCodecList sMCL = new MediaCodecList(MediaCodecList.REGULAR_CODECS);


	/**
	* Finds test name (heuristically) and prints out standard skip message.
	*
	* Since it uses heuristics, this method has only been verified for media
	* tests. This centralizes the way to signal a skipped test.
	*/
	public static void skipTest(String tag, String reason) {
	int bestScore = -1;
	String testName = "test???";
	Map<Thread, StackTraceElement[]> traces = Thread.getAllStackTraces();
	for (Map.Entry<Thread, StackTraceElement[]> entry : traces.entrySet()) {
	StackTraceElement[] stack = entry.getValue();
	for (int index = 0; index < stack.length; ++index) {
	// method name must start with "test"
	String methodName = stack[index].getMethodName();
	if (!methodName.startsWith("test")) {
	continue;
	}

	int score = 0;
	// see if there is a public non-static void method that takes no argument
	Class<?> clazz;
	try {
	clazz = Class.forName(stack[index].getClassName());
	++score;
	for (final Method method : clazz.getDeclaredMethods()) {
	if (method.getName().equals(methodName)
	&& isPublic(method.getModifiers())
	&& !isStatic(method.getModifiers())
	&& method.getParameterTypes().length == 0
	&& method.getReturnType().equals(Void.TYPE)) {
	++score;
	break;
	}
	}
	if (score == 1) {
	// if we could read the class, but method is not public void, it is
	// not a candidate
	continue;
	}
	} catch (ClassNotFoundException e) {
	}

	// even if we cannot verify the method signature, there are signals in the stack

	// usually test method is invoked by reflection
	int depth = 1;
	while (index + depth < stack.length
	&& stack[index + depth].getMethodName().equals("invoke")
	&& stack[index + depth].getClassName().equals(
	"java.lang.reflect.Method")) {
	++depth;
	}
	if (depth > 1) {
	++score;
	// and usually test method is run by runMethod method in android.test package
	if (index + depth < stack.length) {
	if (stack[index + depth].getClassName().startsWith("android.test.")) {
	++score;
	}
	if (stack[index + depth].getMethodName().equals("runMethod")) {
	++score;
	}
	}
	}

	if (score > bestScore) {
	bestScore = score;
	testName = methodName;
	}
	}
	}

	Log.i(tag, "SKIPPING " + testName + "(): " + reason);
	}

	/**
	* Finds test name (heuristically) and prints out standard skip message.
	*
	* Since it uses heuristics, this method has only been verified for media
	* tests. This centralizes the way to signal a skipped test.
	*/
	public static void skipTest(String reason) {
	skipTest(TAG, reason);
	}

	public static boolean check(boolean result, String message) {
	if (!result) {
	skipTest(message);
	}
	return result;
	}

	public static boolean canDecode(MediaFormat format) {
	if (sMCL.findDecoderForFormat(format) == null) {
	Log.i(TAG, "no decoder for " + format);
	return false;
	}
	return true;
	}

	public static boolean hasCodecForTrack(MediaExtractor ex, int track) {
	int count = ex.getTrackCount();
	if (track < 0 \|\| track >= count) {
	throw new IndexOutOfBoundsException(track + " not in [0.." + (count - 1) + "]");
	}
	return canDecode(ex.getTrackFormat(track));
	}

	/**
	* return true iff all audio and video tracks are supported
	*/
	public static boolean hasCodecsForMedia(MediaExtractor ex) {
	for (int i = 0; i < ex.getTrackCount(); ++i) {
	MediaFormat format = ex.getTrackFormat(i);
	// only check for audio and video codecs
	String mime = format.getString(MediaFormat.KEY_MIME).toLowerCase();
	if (!mime.startsWith("audio/") && !mime.startsWith("video/")) {
	continue;
	}
	if (!canDecode(format)) {
	return false;
	}
	}
	return true;
	}

	/**
	* return true iff any track starting with mimePrefix is supported
	*/
	public static boolean hasCodecForMediaAndDomain(MediaExtractor ex, String mimePrefix) {
	mimePrefix = mimePrefix.toLowerCase();
	for (int i = 0; i < ex.getTrackCount(); ++i) {
	MediaFormat format = ex.getTrackFormat(i);
	String mime = format.getString(MediaFormat.KEY_MIME);
	if (mime.toLowerCase().startsWith(mimePrefix)) {
	if (canDecode(format)) {
	return true;
	}
	Log.i(TAG, "no decoder for " + format);
	}
	}
	return false;
	}

	private static boolean hasCodecsForResourceCombo(
	Context context, int resourceId, int track, String mimePrefix) {
	try {
	AssetFileDescriptor afd = null;
	MediaExtractor ex = null;
	try {
	afd = context.getResources().openRawResourceFd(resourceId);
	ex = new MediaExtractor();
	ex.setDataSource(afd.getFileDescriptor(), afd.getStartOffset(), afd.getLength());
	if (mimePrefix != null) {
	return hasCodecForMediaAndDomain(ex, mimePrefix);
	} else if (track == ALL_AV_TRACKS) {
	return hasCodecsForMedia(ex);
	} else {
	return hasCodecForTrack(ex, track);
	}
	} finally {
	if (ex != null) {
	ex.release();
	}
	if (afd != null) {
	afd.close();
	}
	}
	} catch (IOException e) {
	Log.i(TAG, "could not open resource");
	}
	return false;
	}

	/**
	* return true iff all audio and video tracks are supported
	*/
	public static boolean hasCodecsForResource(Context context, int resourceId) {
	return hasCodecsForResourceCombo(context, resourceId, ALL_AV_TRACKS, null /* mimePrefix */);
	}

	public static boolean checkCodecsForResource(Context context, int resourceId) {
	return check(hasCodecsForResource(context, resourceId), "no decoder found");
	}

	/**
	* return true iff track is supported.
	*/
	public static boolean hasCodecForResource(Context context, int resourceId, int track) {
	return hasCodecsForResourceCombo(context, resourceId, track, null /* mimePrefix */);
	}

	public static boolean checkCodecForResource(Context context, int resourceId, int track) {
	return check(hasCodecForResource(context, resourceId, track), "no decoder found");
	}

	/**
	* return true iff any track starting with mimePrefix is supported
	*/
	public static boolean hasCodecForResourceAndDomain(
	Context context, int resourceId, String mimePrefix) {
	return hasCodecsForResourceCombo(context, resourceId, ALL_AV_TRACKS, mimePrefix);
	}

	/**
	* return true iff all audio and video tracks are supported
	*/
	public static boolean hasCodecsForPath(Context context, String path) {
	MediaExtractor ex = null;
	try {
	ex = new MediaExtractor();
	Uri uri = Uri.parse(path);
	String scheme = uri.getScheme();
	if (scheme == null) { // file
	ex.setDataSource(path);
	} else if (scheme.equalsIgnoreCase("file")) {
	ex.setDataSource(uri.getPath());
	} else {
	ex.setDataSource(context, uri, null);
	}
	return hasCodecsForMedia(ex);
	} catch (IOException e) {
	Log.i(TAG, "could not open path " + path);
	} finally {
	if (ex != null) {
	ex.release();
	}
	}
	return false;
	}

	public static boolean checkCodecsForPath(Context context, String path) {
	return check(hasCodecsForPath(context, path), "no decoder found");
	}

	private static boolean hasCodecForMime(boolean encoder, String mime) {
	for (MediaCodecInfo info : sMCL.getCodecInfos()) {
	if (encoder != info.isEncoder()) {
	continue;
	}

	for (String type : info.getSupportedTypes()) {
	if (type.equalsIgnoreCase(mime)) {
	Log.i(TAG, "found codec " + info.getName() + " for mime " + mime);
	return true;
	}
	}
	}
	return false;
	}

	private static boolean hasCodecForMimes(boolean encoder, String[] mimes) {
	for (String mime : mimes) {
	if (!hasCodecForMime(encoder, mime)) {
	Log.i(TAG, "no " + (encoder ? "encoder" : "decoder") + " for mime " + mime);
	return false;
	}
	}
	return true;
	}


	public static boolean hasEncoder(String... mimes) {
	return hasCodecForMimes(true /* encoder */, mimes);
	}

	public static boolean hasDecoder(String... mimes) {
	return hasCodecForMimes(false /* encoder */, mimes);
	}

	public static boolean checkDecoder(String... mimes) {
	return check(hasCodecForMimes(false /* encoder */, mimes), "no decoder found");
	}

	public static boolean checkEncoder(String... mimes) {
	return check(hasCodecForMimes(true /* encoder */, mimes), "no encoder found");
	}

	public static boolean canDecodeVideo(String mime, int width, int height, float rate) {
	MediaFormat format = MediaFormat.createVideoFormat(mime, width, height);
	format.setFloat(MediaFormat.KEY_FRAME_RATE, rate);
	return canDecode(format);
	}

	public static boolean checkDecoderForFormat(MediaFormat format) {
	return check(canDecode(format), "no decoder for " + format);
	}
	}