content: /*
   * Copyright (C) 2011-2013 GUIGUI Simon, fyhertz@gmail.com
   * 
   * This file is part of Spydroid (http://code.google.com/p/spydroid-ipcamera/)
   * 
   * Spydroid is free software; you can redistribute it and/or modify
   * it under the terms of the GNU General Public License as published by
   * the Free Software Foundation; either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This source code is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU General Public License
   * along with this source code; if not, write to the Free Software
   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
   */
  
  package net.majorkernelpanic.streaming.video;
  
  import java.util.ArrayList;
  import java.util.HashMap;
  
  import android.annotation.SuppressLint;
  import android.media.MediaCodecInfo;
  import android.media.MediaCodecList;
  import android.os.Build;
  import android.util.Log;
  import android.util.SparseArray;
  
  @SuppressLint("InlinedApi")
  public class CodecManager {
  
  	public final static String TAG = "CodecManager";
  	
  	public static final int[] SUPPORTED_COLOR_FORMATS = {
  		MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Planar,
  		MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420SemiPlanar
  	};
  
  	/**
  	 * There currently is no way to know if an encoder is software or hardware from the MediaCodecInfo class,
  	 * so we need to maintain a list of known software encoders.
  	 */
  	public static final String[] SOFTWARE_ENCODERS = {
  		"OMX.google.h264.encoder"
  	};
  
  	/**
  	 * Contains a list of encoders and color formats that we may use with a {@link CodecManager.Translator}.  
  	 */
  	static class Codecs {
  		/** A hardware encoder supporting a color format we can use. */
  		public String hardwareCodec;
  		public int hardwareColorFormat;
  		/** A software encoder supporting a color format we can use. */
  		public String softwareCodec;
  		public int softwareColorFormat;
  	}
  
  	/**
  	 *  Contains helper functions to choose an encoder and a color format.
  	 */
  	static class Selector {
  
  		private static HashMap<String,SparseArray<ArrayList<String>>> sHardwareCodecs = new HashMap<String, SparseArray<ArrayList<String>>>();
  		private static HashMap<String,SparseArray<ArrayList<String>>> sSoftwareCodecs = new HashMap<String, SparseArray<ArrayList<String>>>();
  
  		/**
  		 * Determines the most appropriate encoder to compress the video from the Camera
  		 */
  		public static Codecs findCodecsFormMimeType(String mimeType, boolean tryColorFormatSurface) {
  			findSupportedColorFormats(mimeType);
  			SparseArray<ArrayList<String>> hardwareCodecs = sHardwareCodecs.get(mimeType);
  			SparseArray<ArrayList<String>> softwareCodecs = sSoftwareCodecs.get(mimeType);
  			Codecs list = new Codecs();
  
  			// On devices running 4.3, we need an encoder supporting the color format used to work with a Surface
  			if (Build.VERSION.SDK_INT>=18 && tryColorFormatSurface) {
  				int colorFormatSurface = MediaCodecInfo.CodecCapabilities.COLOR_FormatSurface;
  				try {
  					// We want a hardware encoder
  					list.hardwareCodec = hardwareCodecs.get(colorFormatSurface).get(0);
  					list.hardwareColorFormat = colorFormatSurface;
  				} catch (Exception e) {}
  				try {
  					// We want a software encoder
  					list.softwareCodec = softwareCodecs.get(colorFormatSurface).get(0);
  					list.softwareColorFormat = colorFormatSurface;
  				} catch (Exception e) {}
  
  				if (list.hardwareCodec != null) {
  					Log.v(TAG,"Choosen primary codec: "+list.hardwareCodec+" with color format: "+list.hardwareColorFormat);
  				} else {
  					Log.e(TAG,"No supported hardware codec found !");
  				}
  				if (list.softwareCodec != null) {
  					Log.v(TAG,"Choosen secondary codec: "+list.hardwareCodec+" with color format: "+list.hardwareColorFormat);
  				} else {
  					Log.e(TAG,"No supported software codec found !");
  				}
  				return list;
  			}
  
  			for (int i=0;i<SUPPORTED_COLOR_FORMATS.length;i++) {
  				try {
  					list.hardwareCodec = hardwareCodecs.get(SUPPORTED_COLOR_FORMATS[i]).get(0);
  					list.hardwareColorFormat = SUPPORTED_COLOR_FORMATS[i];
  					break;
  				} catch (Exception e) {}
  			}
  			for (int i=0;i<SUPPORTED_COLOR_FORMATS.length;i++) {
  				try {
  					list.softwareCodec = softwareCodecs.get(SUPPORTED_COLOR_FORMATS[i]).get(0);
  					list.softwareColorFormat = SUPPORTED_COLOR_FORMATS[i];
  					break;
  				} catch (Exception e) {}
  			}
  
  			if (list.hardwareCodec != null) {
  				Log.v(TAG,"Choosen primary codec: "+list.hardwareCodec+" with color format: "+list.hardwareColorFormat);
  			} else {
  				Log.e(TAG,"No supported hardware codec found !");
  			}
  			if (list.softwareCodec != null) {
  				Log.v(TAG,"Choosen secondary codec: "+list.hardwareCodec+" with color format: "+list.softwareColorFormat);
  			} else {
  				Log.e(TAG,"No supported software codec found !");
  			}
  
  			return list;
  		}			
  
  		/** 
  		 * Returns an associative array of the supported color formats and the names of the encoders for a given mime type
  		 * This can take up to sec on certain phones the first time you run it...
  		 **/
  		@SuppressLint("NewApi")
  		static private void findSupportedColorFormats(String mimeType) {
  			SparseArray<ArrayList<String>> softwareCodecs = new SparseArray<ArrayList<String>>();
  			SparseArray<ArrayList<String>> hardwareCodecs = new SparseArray<ArrayList<String>>();
  
  			if (sSoftwareCodecs.containsKey(mimeType)) {
  				return; 
  			}
  
  			Log.v(TAG,"Searching supported color formats for mime type \""+mimeType+"\"...");
  
  			// We loop through the encoders, apparently this can take up to a sec (testes on a GS3)
  			for(int j = MediaCodecList.getCodecCount() - 1; j >= 0; j--){
  				MediaCodecInfo codecInfo = MediaCodecList.getCodecInfoAt(j);
  				if (!codecInfo.isEncoder()) continue;
  
  				String[] types = codecInfo.getSupportedTypes();
  				for (int i = 0; i < types.length; i++) {
  					if (types[i].equalsIgnoreCase(mimeType)) {
  						MediaCodecInfo.CodecCapabilities capabilities = codecInfo.getCapabilitiesForType(mimeType);
  
  						boolean software = false;
  						for (int k=0;k<SOFTWARE_ENCODERS.length;k++) {
  							if (codecInfo.getName().equalsIgnoreCase(SOFTWARE_ENCODERS[i])) {
  								software = true;
  							}
  						}
  
  						// And through the color formats supported
  						for (int k = 0; k < capabilities.colorFormats.length; k++) {
  							int format = capabilities.colorFormats[k];
  							if (software) {
  								if (softwareCodecs.get(format) == null) softwareCodecs.put(format, new ArrayList<String>());
  								softwareCodecs.get(format).add(codecInfo.getName());
  							} else {
  								if (hardwareCodecs.get(format) == null) hardwareCodecs.put(format, new ArrayList<String>());
  								hardwareCodecs.get(format).add(codecInfo.getName());
  							}
  						}
  
  					}
  				}
  			}
  
  			// Logs the supported color formats on the phone
  			StringBuilder e = new StringBuilder();
  			e.append("Supported color formats on this phone: ");
  			for (int i=0;i<softwareCodecs.size();i++) e.append(softwareCodecs.keyAt(i)+", ");
  			for (int i=0;i<hardwareCodecs.size();i++) e.append(hardwareCodecs.keyAt(i)+(i==hardwareCodecs.size()-1?".":", "));
  			Log.v(TAG, e.toString());
  
  			sSoftwareCodecs.put(mimeType, softwareCodecs);
  			sHardwareCodecs.put(mimeType, hardwareCodecs);
  			return;
  		}
  
  
  	}
  
  	static class Translator {
  
  		private int mOutputColorFormat;
  		private int mWidth; 
  		private int mHeight;
  		private int mYStride;
  		private int mUVStride;
  		private int mYSize;
  		private int mUVSize;
  		private int bufferSize;
  		private int i;
  		private byte[] tmp;
  
  		public Translator(int outputColorFormat, int width, int height) {
  			mOutputColorFormat = outputColorFormat;
  			mWidth = width;
  			mHeight = height;
  			mYStride   = (int) Math.ceil(mWidth / 16.0) * 16;
  			mUVStride  = (int) Math.ceil( (mYStride / 2) / 16.0) * 16;
  			mYSize     = mYStride * mHeight;
  			mUVSize    = mUVStride * mHeight / 2;
  			bufferSize = mYSize + mUVSize * 2;
  			tmp = new byte[mUVSize*2];
  		}
  
  		public int getBufferSize() {
  			return bufferSize;
  		}
  
  		public int getUVStride() {
  			return mUVStride;
  		}
  
  		public int getYStride() {
  			return mYStride;
  		}
  
  		public byte[] translate(byte[] buffer) {
  
  			if (mOutputColorFormat == MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Planar) {
  				// FIXME: May be issues because of padding here :/
  				int wh4 = bufferSize/6; //wh4 = width*height/4
  				byte tmp;
  				for (i=wh4*4; i<wh4*5; i++) {
  					tmp = buffer[i];
  					buffer[i] = buffer[i+wh4];
  					buffer[i+wh4] = tmp;
  				}
  			}
  
  			else if (mOutputColorFormat == MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420SemiPlanar) {
  				// We need to interleave the U and V channel
  				System.arraycopy(buffer, mYSize, tmp, 0, mUVSize*2); // Y
  				for (i = 0; i < mUVSize; i++) {
  					buffer[mYSize + i*2] = tmp[i + mUVSize]; // Cb (U)
  					buffer[mYSize + i*2+1] = tmp[i]; // Cr (V)
  				}
  			}
  
  			return buffer;
  		}
  
  
  	}
  
  
  }
  