| /* |
| * Copyright (C) 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 com.android.test.voiceenrollment; |
| |
| import java.util.Random; |
| import java.util.UUID; |
| |
| import android.app.Activity; |
| import android.hardware.soundtrigger.SoundTrigger; |
| import android.hardware.soundtrigger.SoundTrigger.Keyphrase; |
| import android.hardware.soundtrigger.SoundTrigger.KeyphraseSoundModel; |
| import android.os.Bundle; |
| import android.os.UserManager; |
| import android.util.Log; |
| import android.view.View; |
| import android.widget.Toast; |
| |
| public class TestEnrollmentActivity extends Activity { |
| private static final String TAG = "TestEnrollmentActivity"; |
| private static final boolean DBG = true; |
| |
| /** Keyphrase related constants, must match those defined in enrollment_application.xml */ |
| private static final int KEYPHRASE_ID = 101; |
| private static final int RECOGNITION_MODES = SoundTrigger.RECOGNITION_MODE_VOICE_TRIGGER; |
| private static final String BCP47_LOCALE = "fr-FR"; |
| private static final String TEXT = "Hello There"; |
| |
| private EnrollmentUtil mEnrollmentUtil; |
| private Random mRandom; |
| |
| @Override |
| protected void onCreate(Bundle savedInstanceState) { |
| if (DBG) Log.d(TAG, "onCreate"); |
| super.onCreate(savedInstanceState); |
| setContentView(R.layout.main); |
| mEnrollmentUtil = new EnrollmentUtil(); |
| mRandom = new Random(); |
| } |
| |
| /** |
| * Called when the user clicks the enroll button. |
| * Performs a fresh enrollment. |
| */ |
| public void onEnrollButtonClicked(View v) { |
| Keyphrase kp = new Keyphrase(KEYPHRASE_ID, RECOGNITION_MODES, BCP47_LOCALE, TEXT, |
| new int[] { UserManager.get(this).getUserHandle() /* current user */}); |
| UUID modelUuid = UUID.randomUUID(); |
| // Generate a fake model to push. |
| byte[] data = new byte[1024]; |
| mRandom.nextBytes(data); |
| KeyphraseSoundModel soundModel = new KeyphraseSoundModel(modelUuid, null, data, |
| new Keyphrase[] { kp }); |
| boolean status = mEnrollmentUtil.addOrUpdateSoundModel(soundModel); |
| if (status) { |
| Toast.makeText( |
| this, "Successfully enrolled, model UUID=" + modelUuid, Toast.LENGTH_SHORT) |
| .show(); |
| } else { |
| Toast.makeText(this, "Failed to enroll!!!" + modelUuid, Toast.LENGTH_SHORT).show(); |
| } |
| } |
| |
| /** |
| * Called when the user clicks the un-enroll button. |
| * Clears the enrollment information for the user. |
| */ |
| public void onUnEnrollButtonClicked(View v) { |
| KeyphraseSoundModel soundModel = mEnrollmentUtil.getSoundModel(KEYPHRASE_ID, BCP47_LOCALE); |
| if (soundModel == null) { |
| Toast.makeText(this, "Sound model not found!!!", Toast.LENGTH_SHORT).show(); |
| return; |
| } |
| boolean status = mEnrollmentUtil.deleteSoundModel(KEYPHRASE_ID, BCP47_LOCALE); |
| if (status) { |
| Toast.makeText(this, "Successfully un-enrolled, model UUID=" + soundModel.uuid, |
| Toast.LENGTH_SHORT) |
| .show(); |
| } else { |
| Toast.makeText(this, "Failed to un-enroll!!!", Toast.LENGTH_SHORT).show(); |
| } |
| } |
| |
| /** |
| * Called when the user clicks the re-enroll button. |
| * Uses the previously enrolled sound model and makes changes to it before pushing it back. |
| */ |
| public void onReEnrollButtonClicked(View v) { |
| KeyphraseSoundModel soundModel = mEnrollmentUtil.getSoundModel(KEYPHRASE_ID, BCP47_LOCALE); |
| if (soundModel == null) { |
| Toast.makeText(this, "Sound model not found!!!", Toast.LENGTH_SHORT).show(); |
| return; |
| } |
| // Generate a fake model to push. |
| byte[] data = new byte[2048]; |
| mRandom.nextBytes(data); |
| KeyphraseSoundModel updated = new KeyphraseSoundModel(soundModel.uuid, |
| soundModel.vendorUuid, data, soundModel.keyphrases); |
| boolean status = mEnrollmentUtil.addOrUpdateSoundModel(updated); |
| if (status) { |
| Toast.makeText(this, "Successfully re-enrolled, model UUID=" + updated.uuid, |
| Toast.LENGTH_SHORT) |
| .show(); |
| } else { |
| Toast.makeText(this, "Failed to re-enroll!!!", Toast.LENGTH_SHORT).show(); |
| } |
| } |
| } |