tests/tests/media/src/android/media/cts/VisualizerTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2010 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.media.cts;

 import android.media.audiofx.AudioEffect;
 import android.media.AudioFormat;
 import android.media.AudioManager;
 import android.media.audiofx.Visualizer;
 import android.os.Looper;
 import android.test.AndroidTestCase;
 import android.util.Log;

 public class VisualizerTest extends AndroidTestCase {

     private String TAG = "VisualizerTest";
     private final static int MIN_CAPTURE_RATE_MAX = 10000; // 10Hz
     private final static int MIN_CAPTURE_SIZE_MAX = 1024;
     private final static int MAX_CAPTURE_SIZE_MIN = 512;
     private final static int MAX_LOOPER_WAIT_COUNT = 10;

     private Visualizer mVisualizer = null;
     private int mSession = -1;
     private boolean mInitialized = false;
     private Looper mLooper = null;
     private final Object mLock = new Object();
     private byte[] mWaveform = null;
     private byte[] mFft = null;
     private boolean mCaptureWaveform = false;
     private boolean mCaptureFft = false;
     private Thread mListenerThread;

     //-----------------------------------------------------------------
     // VISUALIZER TESTS:
     //----------------------------------

     //-----------------------------------------------------------------
     // 0 - constructor
     //----------------------------------

     //Test case 0.0: test constructor and release
     public void test0_0ConstructorAndRelease() throws Exception {
         Visualizer visualizer = null;
          try {
             visualizer = new Visualizer(0);
             assertNotNull("could not create Visualizer", visualizer);
         } catch (IllegalArgumentException e) {
             fail("Visualizer not found");
         } catch (UnsupportedOperationException e) {
             fail("Effect library not loaded");
         } finally {
             if (visualizer != null) {
                 visualizer.release();
             }
         }
     }


     //-----------------------------------------------------------------
     // 1 - get/set parameters
     //----------------------------------

     //Test case 1.0: capture rates
     public void test1_0CaptureRates() throws Exception {
         getVisualizer(0);
         try {
             int captureRate = mVisualizer.getMaxCaptureRate();
             assertTrue("insufficient max capture rate",
                     captureRate >= MIN_CAPTURE_RATE_MAX);
             int samplingRate = mVisualizer.getSamplingRate();
         } catch (IllegalArgumentException e) {
             fail("Bad parameter value");
         } catch (UnsupportedOperationException e) {
             fail("get parameter() rejected");
         } catch (IllegalStateException e) {
             fail("get parameter() called in wrong state");
         } finally {
             releaseVisualizer();
         }
     }

     //Test case 1.1: test capture size
     public void test1_1CaptureSize() throws Exception {
         getVisualizer(0);
         try {
             int[] range = mVisualizer.getCaptureSizeRange();
             assertTrue("insufficient min capture size",
                     range[0] <= MAX_CAPTURE_SIZE_MIN);
             assertTrue("insufficient min capture size",
                     range[1] >= MIN_CAPTURE_SIZE_MAX);
             mVisualizer.setCaptureSize(range[0]);
             assertEquals("insufficient min capture size",
                     range[0], mVisualizer.getCaptureSize());
             mVisualizer.setCaptureSize(range[1]);
             assertEquals("insufficient min capture size",
                     range[1], mVisualizer.getCaptureSize());
         } catch (IllegalArgumentException e) {
             fail("Bad parameter value");
         } catch (UnsupportedOperationException e) {
             fail("get parameter() rejected");
         } catch (IllegalStateException e) {
             fail("get parameter() called in wrong state");
         } finally {
             releaseVisualizer();
         }
     }

     //-----------------------------------------------------------------
     // 2 - check capture
     //----------------------------------

     //Test case 2.0: test cature in polling mode
     public void test2_0PollingCapture() throws Exception {
         try {
             getVisualizer(0);
             mVisualizer.setEnabled(true);
             assertTrue("visualizer not enabled", mVisualizer.getEnabled());
             Thread.sleep(100);
             // check capture on silence
             byte[] data = new byte[mVisualizer.getCaptureSize()];
             mVisualizer.getWaveForm(data);
             int energy = computeEnergy(data, true);
             assertEquals("getWaveForm reports energy for silence",
                     0, energy);
             mVisualizer.getFft(data);
             energy = computeEnergy(data, false);
             assertEquals("getFft reports energy for silence",
                     0, energy);

         } catch (IllegalStateException e) {
             fail("method called in wrong state");
         } catch (InterruptedException e) {
             fail("sleep() interrupted");
         } finally {
             releaseVisualizer();
         }
     }

     //Test case 2.1: test capture with listener
     public void test2_1ListenerCapture() throws Exception {
         try {
             getVisualizer(0);
             synchronized(mLock) {
                 mInitialized = false;
                 createListenerLooper();
                 waitForLooperInitialization_l();
             }
             mVisualizer.setEnabled(true);
             assertTrue("visualizer not enabled", mVisualizer.getEnabled());

             Thread.sleep(100);

             // check capture on silence
             synchronized(mLock) {
                 mCaptureWaveform = true;
                 int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
                 while ((mWaveform == null) && (looperWaitCount-- > 0)) {
                     try {
                         mLock.wait();
                     } catch(Exception e) {
                     }
                 }
                 mCaptureWaveform = false;
             }
             assertNotNull("waveform capture failed", mWaveform);
             int energy = computeEnergy(mWaveform, true);
             assertEquals("getWaveForm reports energy for silence",
                     0, energy);

             synchronized(mLock) {
                 mCaptureFft = true;
                 int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
                 while ((mFft == null) && (looperWaitCount-- > 0)) {
                     try {
                         mLock.wait();
                     } catch(Exception e) {
                     }
                 }
                 mCaptureFft = false;
             }
             assertNotNull("FFT capture failed", mFft);
             energy = computeEnergy(mFft, false);
             assertEquals("getFft reports energy for silence",
                     0, energy);

         } catch (IllegalStateException e) {
             fail("method called in wrong state");
         } catch (InterruptedException e) {
             fail("sleep() interrupted");
         } finally {
             terminateListenerLooper();
             releaseVisualizer();
         }
     }

     //-----------------------------------------------------------------
     // private methods
     //----------------------------------

     private int computeEnergy(byte[] data, boolean pcm) {
         int energy = 0;
         if (data.length != 0) {
             if (pcm) {
                 for (int i = 0; i < data.length; i++) {
                     int tmp = ((int)data[i] & 0xFF) - 128;
                     energy += tmp*tmp;
                 }
             } else {
                 energy = (int)data[0] * (int)data[0];
                 for (int i = 2; i < data.length; i += 2) {
                     int real = (int)data[i];
                     int img = (int)data[i + 1];
                     energy += real * real + img * img;
                 }
             }
         }
         return energy;
     }

     private void getVisualizer(int session) {
          if (mVisualizer == null || session != mSession) {
              if (session != mSession && mVisualizer != null) {
                  mVisualizer.release();
                  mVisualizer = null;
              }
              try {
                 mVisualizer = new Visualizer(session);
                 mSession = session;
             } catch (IllegalArgumentException e) {
                 Log.e(TAG, "getVisualizer() Visualizer not found exception: "+e);
             } catch (UnsupportedOperationException e) {
                 Log.e(TAG, "getVisualizer() Effect library not loaded exception: "+e);
             }
          }
          assertNotNull("could not create mVisualizer", mVisualizer);
     }

     private void releaseVisualizer() {
         if (mVisualizer != null) {
             mVisualizer.release();
             mVisualizer = null;
         }
     }

     private void waitForLooperInitialization_l() {
         int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
         while (!mInitialized && (looperWaitCount-- > 0)) {
             try {
                 mLock.wait();
             } catch(Exception e) {
             }
         }
         assertTrue(mInitialized);
     }

     private void createListenerLooper() {
         mListenerThread = new Thread() {
             @Override
             public void run() {
                 // Set up a looper to be used by mEffect.
                 Looper.prepare();

                 // Save the looper so that we can terminate this thread
                 // after we are done with it.
                 mLooper = Looper.myLooper();

                 synchronized(mLock) {
                     if (mVisualizer != null) {
                         mVisualizer.setDataCaptureListener(new Visualizer.OnDataCaptureListener() {
                             public void onWaveFormDataCapture(
                                     Visualizer visualizer, byte[] waveform, int samplingRate) {
                                 synchronized(mLock) {
                                     if (visualizer == mVisualizer) {
                                         if (mCaptureWaveform) {
                                             mWaveform = waveform;
                                             mLock.notify();
                                         }
                                     }
                                 }
                             }

                             public void onFftDataCapture(
                                     Visualizer visualizer, byte[] fft, int samplingRate) {
                                 synchronized(mLock) {
                                     Log.e(TAG, "onFftDataCapture 2 mCaptureFft: "+mCaptureFft);
                                     if (visualizer == mVisualizer) {
                                         if (mCaptureFft) {
                                             mFft = fft;
                                             mLock.notify();
                                         }
                                     }
                                 }
                             }
                         },
                         10000,
                         true,
                         true);
                     }
                     mInitialized = true;
                     mLock.notify();
                 }
                 Looper.loop();  // Blocks forever until Looper.quit() is called.
             }
         };
         mListenerThread.start();
     }
     /*
      * Terminates the listener looper thread.
      */
     private void terminateListenerLooper() {
         if (mListenerThread != null) {
             if (mLooper != null) {
                 mLooper.quit();
                 mLooper = null;
             }
             try {
                 mListenerThread.join();
             } catch(InterruptedException e) {
             }
             mListenerThread = null;
         }
     }
 }
	/*
	* Copyright (C) 2010 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.media.cts;

	import android.media.audiofx.AudioEffect;
	import android.media.AudioFormat;
	import android.media.AudioManager;
	import android.media.audiofx.Visualizer;
	import android.os.Looper;
	import android.test.AndroidTestCase;
	import android.util.Log;

	public class VisualizerTest extends AndroidTestCase {

	private String TAG = "VisualizerTest";
	private final static int MIN_CAPTURE_RATE_MAX = 10000; // 10Hz
	private final static int MIN_CAPTURE_SIZE_MAX = 1024;
	private final static int MAX_CAPTURE_SIZE_MIN = 512;
	private final static int MAX_LOOPER_WAIT_COUNT = 10;

	private Visualizer mVisualizer = null;
	private int mSession = -1;
	private boolean mInitialized = false;
	private Looper mLooper = null;
	private final Object mLock = new Object();
	private byte[] mWaveform = null;
	private byte[] mFft = null;
	private boolean mCaptureWaveform = false;
	private boolean mCaptureFft = false;
	private Thread mListenerThread;

	//-----------------------------------------------------------------
	// VISUALIZER TESTS:
	//----------------------------------

	//-----------------------------------------------------------------
	// 0 - constructor
	//----------------------------------

	//Test case 0.0: test constructor and release
	public void test0_0ConstructorAndRelease() throws Exception {
	Visualizer visualizer = null;
	try {
	visualizer = new Visualizer(0);
	assertNotNull("could not create Visualizer", visualizer);
	} catch (IllegalArgumentException e) {
	fail("Visualizer not found");
	} catch (UnsupportedOperationException e) {
	fail("Effect library not loaded");
	} finally {
	if (visualizer != null) {
	visualizer.release();
	}
	}
	}


	//-----------------------------------------------------------------
	// 1 - get/set parameters
	//----------------------------------

	//Test case 1.0: capture rates
	public void test1_0CaptureRates() throws Exception {
	getVisualizer(0);
	try {
	int captureRate = mVisualizer.getMaxCaptureRate();
	assertTrue("insufficient max capture rate",
	captureRate >= MIN_CAPTURE_RATE_MAX);
	int samplingRate = mVisualizer.getSamplingRate();
	} catch (IllegalArgumentException e) {
	fail("Bad parameter value");
	} catch (UnsupportedOperationException e) {
	fail("get parameter() rejected");
	} catch (IllegalStateException e) {
	fail("get parameter() called in wrong state");
	} finally {
	releaseVisualizer();
	}
	}

	//Test case 1.1: test capture size
	public void test1_1CaptureSize() throws Exception {
	getVisualizer(0);
	try {
	int[] range = mVisualizer.getCaptureSizeRange();
	assertTrue("insufficient min capture size",
	range[0] <= MAX_CAPTURE_SIZE_MIN);
	assertTrue("insufficient min capture size",
	range[1] >= MIN_CAPTURE_SIZE_MAX);
	mVisualizer.setCaptureSize(range[0]);
	assertEquals("insufficient min capture size",
	range[0], mVisualizer.getCaptureSize());
	mVisualizer.setCaptureSize(range[1]);
	assertEquals("insufficient min capture size",
	range[1], mVisualizer.getCaptureSize());
	} catch (IllegalArgumentException e) {
	fail("Bad parameter value");
	} catch (UnsupportedOperationException e) {
	fail("get parameter() rejected");
	} catch (IllegalStateException e) {
	fail("get parameter() called in wrong state");
	} finally {
	releaseVisualizer();
	}
	}

	//-----------------------------------------------------------------
	// 2 - check capture
	//----------------------------------

	//Test case 2.0: test cature in polling mode
	public void test2_0PollingCapture() throws Exception {
	try {
	getVisualizer(0);
	mVisualizer.setEnabled(true);
	assertTrue("visualizer not enabled", mVisualizer.getEnabled());
	Thread.sleep(100);
	// check capture on silence
	byte[] data = new byte[mVisualizer.getCaptureSize()];
	mVisualizer.getWaveForm(data);
	int energy = computeEnergy(data, true);
	assertEquals("getWaveForm reports energy for silence",
	0, energy);
	mVisualizer.getFft(data);
	energy = computeEnergy(data, false);
	assertEquals("getFft reports energy for silence",
	0, energy);

	} catch (IllegalStateException e) {
	fail("method called in wrong state");
	} catch (InterruptedException e) {
	fail("sleep() interrupted");
	} finally {
	releaseVisualizer();
	}
	}

	//Test case 2.1: test capture with listener
	public void test2_1ListenerCapture() throws Exception {
	try {
	getVisualizer(0);
	synchronized(mLock) {
	mInitialized = false;
	createListenerLooper();
	waitForLooperInitialization_l();
	}
	mVisualizer.setEnabled(true);
	assertTrue("visualizer not enabled", mVisualizer.getEnabled());

	Thread.sleep(100);

	// check capture on silence
	synchronized(mLock) {
	mCaptureWaveform = true;
	int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
	while ((mWaveform == null) && (looperWaitCount-- > 0)) {
	try {
	mLock.wait();
	} catch(Exception e) {
	}
	}
	mCaptureWaveform = false;
	}
	assertNotNull("waveform capture failed", mWaveform);
	int energy = computeEnergy(mWaveform, true);
	assertEquals("getWaveForm reports energy for silence",
	0, energy);

	synchronized(mLock) {
	mCaptureFft = true;
	int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
	while ((mFft == null) && (looperWaitCount-- > 0)) {
	try {
	mLock.wait();
	} catch(Exception e) {
	}
	}
	mCaptureFft = false;
	}
	assertNotNull("FFT capture failed", mFft);
	energy = computeEnergy(mFft, false);
	assertEquals("getFft reports energy for silence",
	0, energy);

	} catch (IllegalStateException e) {
	fail("method called in wrong state");
	} catch (InterruptedException e) {
	fail("sleep() interrupted");
	} finally {
	terminateListenerLooper();
	releaseVisualizer();
	}
	}

	//-----------------------------------------------------------------
	// private methods
	//----------------------------------

	private int computeEnergy(byte[] data, boolean pcm) {
	int energy = 0;
	if (data.length != 0) {
	if (pcm) {
	for (int i = 0; i < data.length; i++) {
	int tmp = ((int)data[i] & 0xFF) - 128;
	energy += tmp*tmp;
	}
	} else {
	energy = (int)data[0] * (int)data[0];
	for (int i = 2; i < data.length; i += 2) {
	int real = (int)data[i];
	int img = (int)data[i + 1];
	energy += real * real + img * img;
	}
	}
	}
	return energy;
	}

	private void getVisualizer(int session) {
	if (mVisualizer == null \|\| session != mSession) {
	if (session != mSession && mVisualizer != null) {
	mVisualizer.release();
	mVisualizer = null;
	}
	try {
	mVisualizer = new Visualizer(session);
	mSession = session;
	} catch (IllegalArgumentException e) {
	Log.e(TAG, "getVisualizer() Visualizer not found exception: "+e);
	} catch (UnsupportedOperationException e) {
	Log.e(TAG, "getVisualizer() Effect library not loaded exception: "+e);
	}
	}
	assertNotNull("could not create mVisualizer", mVisualizer);
	}

	private void releaseVisualizer() {
	if (mVisualizer != null) {
	mVisualizer.release();
	mVisualizer = null;
	}
	}

	private void waitForLooperInitialization_l() {
	int looperWaitCount = MAX_LOOPER_WAIT_COUNT;
	while (!mInitialized && (looperWaitCount-- > 0)) {
	try {
	mLock.wait();
	} catch(Exception e) {
	}
	}
	assertTrue(mInitialized);
	}

	private void createListenerLooper() {
	mListenerThread = new Thread() {
	@Override
	public void run() {
	// Set up a looper to be used by mEffect.
	Looper.prepare();

	// Save the looper so that we can terminate this thread
	// after we are done with it.
	mLooper = Looper.myLooper();

	synchronized(mLock) {
	if (mVisualizer != null) {
	mVisualizer.setDataCaptureListener(new Visualizer.OnDataCaptureListener() {
	public void onWaveFormDataCapture(
	Visualizer visualizer, byte[] waveform, int samplingRate) {
	synchronized(mLock) {
	if (visualizer == mVisualizer) {
	if (mCaptureWaveform) {
	mWaveform = waveform;
	mLock.notify();
	}
	}
	}
	}

	public void onFftDataCapture(
	Visualizer visualizer, byte[] fft, int samplingRate) {
	synchronized(mLock) {
	Log.e(TAG, "onFftDataCapture 2 mCaptureFft: "+mCaptureFft);
	if (visualizer == mVisualizer) {
	if (mCaptureFft) {
	mFft = fft;
	mLock.notify();
	}
	}
	}
	}
	},
	10000,
	true,
	true);
	}
	mInitialized = true;
	mLock.notify();
	}
	Looper.loop(); // Blocks forever until Looper.quit() is called.
	}
	};
	mListenerThread.start();
	}
	/*
	* Terminates the listener looper thread.
	*/
	private void terminateListenerLooper() {
	if (mListenerThread != null) {
	if (mLooper != null) {
	mLooper.quit();
	mLooper = null;
	}
	try {
	mListenerThread.join();
	} catch(InterruptedException e) {
	}
	mListenerThread = null;
	}
	}
	}