apps/CtsVerifier/src/com/android/cts/verifier/presence/BleRssiPrecisionActivity.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2022 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.cts.verifier.presence;

 import android.bluetooth.BluetoothAdapter;
 import android.bluetooth.BluetoothManager;
 import android.content.pm.PackageManager;
 import android.os.Build;
 import android.os.Bundle;
 import android.util.Log;
 import android.view.View;
 import android.widget.Button;
 import android.widget.CheckBox;
 import android.widget.EditText;
 import android.widget.LinearLayout;
 import android.widget.TextView;
 import android.widget.Toast;

 import com.android.compatibility.common.util.ResultType;
 import com.android.compatibility.common.util.ResultUnit;
 import com.android.cts.verifier.PassFailButtons;
 import com.android.cts.verifier.presence.ble.BleAdvertisingPacket;
 import com.android.cts.verifier.presence.ble.BleScanner;
 import com.android.cts.verifier.presence.ble.BleAdvertiser;
 import com.android.cts.verifier.R;

 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Random;

 public class BleRssiPrecisionActivity extends PassFailButtons.Activity {
     private static final String TAG = BleRssiPrecisionActivity.class.getName();
     private static final String DEVICE_NAME = Build.MODEL;

     // Report log schema
     private static final String KEY_REFERENCE_DEVICE = "reference_device";
     private static final String KEY_RSSI_RANGE_95_PERCENTILE = "rssi_range_95_percentile";

     // Thresholds
     private static final int MAX_RSSI_RANGE_DBM = 18;

     private boolean isReferenceDevice;
     private BleScanner mBleScanner;
     private BleAdvertiser mBleAdvertiser;
     private BluetoothAdapter mBluetoothAdapter;
     private HashMap<String, ArrayList<Integer>> mRssiResultMap;
     private Button mStartTestButton;
     private Button mStopTestButton;
     private Button mStartAdvertisingButton;
     private Button mStopAdvertisingButton;
     private LinearLayout mDutModeLayout;
     private LinearLayout mRefModeLayout;
     private TextView mDeviceIdInfoTextView;
     private TextView mDeviceFoundTextView;
     private EditText mReferenceDeviceIdInput;
     private String mReferenceDeviceName;
     private CheckBox mIsReferenceDeviceCheckbox;
     private CheckBox mIsManualPassCheckbox;
     private boolean mIsManualPass;
     private boolean mTestCompleted;
     private int mRssiRange;

     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         setContentView(R.layout.ble_rssi_precision);
         setPassFailButtonClickListeners();
         getPassButton().setEnabled(false);
         mStartTestButton = findViewById(R.id.start_test);
         mStopTestButton = findViewById(R.id.stop_test);
         mStartAdvertisingButton = findViewById(R.id.start_advertising);
         mStopAdvertisingButton = findViewById(R.id.stop_advertising);
         mDeviceIdInfoTextView = findViewById(R.id.device_id_info);
         mDeviceFoundTextView = findViewById(R.id.device_found_info);
         mReferenceDeviceIdInput = findViewById(R.id.ref_device_id_input);
         mIsReferenceDeviceCheckbox = findViewById(R.id.is_reference_device);
         mIsManualPassCheckbox = findViewById(R.id.is_manual_pass);
         mDutModeLayout = findViewById(R.id.dut_mode_layout);
         mRefModeLayout = findViewById(R.id.ref_mode_layout);
         DeviceFeatureChecker.checkFeatureSupported(this, getPassButton(),
                 PackageManager.FEATURE_BLUETOOTH_LE);
         BluetoothManager bluetoothManager = getSystemService(BluetoothManager.class);
         mBluetoothAdapter = bluetoothManager.getAdapter();
         mBleAdvertiser = new BleAdvertiser();
         mBleScanner = new BleScanner();
         mRssiResultMap = new HashMap<>();
         mStopTestButton.setEnabled(false);
         mStopAdvertisingButton.setEnabled(false);
         mDeviceIdInfoTextView.setVisibility(View.GONE);
         mDeviceFoundTextView.setVisibility(View.GONE);
         isReferenceDevice = mIsReferenceDeviceCheckbox.isChecked();
         mIsManualPass = mIsManualPassCheckbox.isChecked();
         checkUiMode();
         mIsReferenceDeviceCheckbox.setOnCheckedChangeListener((buttonView, isChecked) -> {
             isReferenceDevice = isChecked;
             checkUiMode();
         });
         mIsManualPassCheckbox.setOnCheckedChangeListener((buttonView, isChecked) -> {
             mIsManualPass = isChecked;
         });
         mStartTestButton.setOnClickListener(v -> startTest());
         mStopTestButton.setOnClickListener(v -> stopTest());
         mStartAdvertisingButton.setOnClickListener(v -> startAdvertising());
         mStopAdvertisingButton.setOnClickListener(v -> stopAdvertising());
     }

     @Override
     public boolean requiresReportLog() {
         return true;
     }

     private void startTest() {
         mTestCompleted = false;
         if (!checkBluetoothEnabled()) {
             return;
         }
         if (mReferenceDeviceIdInput.getText().toString().isEmpty()) {
             makeToast("Input the device ID shown on the advertising device before commencing test");
             return;
         }
         mStartTestButton.setEnabled(false);
         mStopTestButton.setEnabled(true);
         mIsReferenceDeviceCheckbox.setEnabled(false);
         mBleScanner.startScanning((uuids,
                 macAddress,
                 deviceName,
                 referenceDeviceName,
                 deviceId,
                 rssiMedian,
                 rawRssi) -> {

             if (deviceId != Byte.parseByte(mReferenceDeviceIdInput.getText().toString())) {
                 //reference device does not match discovered device and scan should be discarded
                 Log.i(TAG, "Reference device does not match discovered device. Skipping");
                 return;
             }
             mRssiResultMap.computeIfAbsent(deviceName, k -> new ArrayList<>());
             ArrayList<Integer> resultList = mRssiResultMap.get(deviceName);
             resultList.add(rawRssi);
             mDeviceFoundTextView.setVisibility(View.VISIBLE);
             mReferenceDeviceName = referenceDeviceName;
             String deviceFoundText = getString(R.string.device_found_presence,
                     resultList.size(), 1000);
             mDeviceFoundTextView.setText(deviceFoundText);
             if (resultList.size() >= 1000) {
                 Log.i(TAG, "Data collection complete");
                 mBleScanner.stopScanning();
                 mStartTestButton.setEnabled(true);
                 mStopTestButton.setEnabled(false);
                 mIsReferenceDeviceCheckbox.setEnabled(true);
                 computeTestResults(resultList);
             }
         });
     }

     private void computeTestResults(ArrayList<Integer> data) {
         Collections.sort(data);
         // Calculate range at 95th percentile
         int rssiRange = data.get(975) - data.get(25);
         mRssiRange = rssiRange;
         mTestCompleted = true;
         if (rssiRange <= MAX_RSSI_RANGE_DBM) {
             makeToast("Test passed! Rssi range is: " + rssiRange);
             if (mIsManualPass) {
                 getPassButton().setEnabled(true);
             } else {
                 getPassButton().performClick();
             }
         } else {
             makeToast("Test failed! Rssi range is: " + rssiRange);
         }
         data.clear();
     }

     private void stopTest() {
         mBleScanner.stopScanning();
         for (String device : mRssiResultMap.keySet()) {
             mRssiResultMap.get(device).clear();
         }
         mStopTestButton.setEnabled(false);
         mStartTestButton.setEnabled(true);
         mIsReferenceDeviceCheckbox.setEnabled(true);
     }

     private void startAdvertising() {
         if (!checkBluetoothEnabled()) {
             return;
         }
         byte randomAdvertiserDeviceId = getRandomDeviceId();
         String deviceIdInfoText = getString(R.string.device_id_info_presence,
                 randomAdvertiserDeviceId);
         mDeviceIdInfoTextView.setText(deviceIdInfoText);
         mDeviceIdInfoTextView.setVisibility(View.VISIBLE);
         String packetDeviceName = "";
         if (DEVICE_NAME.length() > BleAdvertisingPacket.MAX_REFERENCE_DEVICE_NAME_LENGTH) {
             packetDeviceName = DEVICE_NAME.substring(0,
                     BleAdvertisingPacket.MAX_REFERENCE_DEVICE_NAME_LENGTH - 1);
         } else {
             packetDeviceName = DEVICE_NAME;
         }
         mBleAdvertiser.startAdvertising(
                 new BleAdvertisingPacket(packetDeviceName, randomAdvertiserDeviceId,
                         (byte) 0).toBytes());
         mStartAdvertisingButton.setEnabled(false);
         mStopAdvertisingButton.setEnabled(true);
     }

     private void stopAdvertising() {
         mBleAdvertiser.stopAdvertising();
         mStopAdvertisingButton.setEnabled(false);
         mStartAdvertisingButton.setEnabled(true);
         mDeviceIdInfoTextView.setVisibility(View.GONE);
     }

     private boolean checkBluetoothEnabled() {
         if (!mBluetoothAdapter.isEnabled()) {
             makeToast("Bluetooth is not enabled, turn on bluetooth before starting test");
             Log.w(TAG, "Could not start test because Bluetooth is not enabled");
             return false;
         }
         return true;
     }

     private void checkUiMode() {
         if (isReferenceDevice) {
             mDutModeLayout.setVisibility(View.GONE);
             mRefModeLayout.setVisibility(View.VISIBLE);
         } else {
             mRefModeLayout.setVisibility(View.GONE);
             mDutModeLayout.setVisibility(View.VISIBLE);
         }
     }

     private static byte getRandomDeviceId() {
         Random random = new Random();
         byte[] randomDeviceIdArray = new byte[1];
         random.nextBytes(randomDeviceIdArray);
         return randomDeviceIdArray[0];
     }

     private void makeToast(String message) {
         Toast.makeText(this, message, Toast.LENGTH_SHORT).show();
     }

     @Override
     public void recordTestResults() {
         if (mTestCompleted) {
             getReportLog().addValue(KEY_REFERENCE_DEVICE, mReferenceDeviceName,
                     ResultType.NEUTRAL, ResultUnit.NONE);
             getReportLog().addValue(KEY_RSSI_RANGE_95_PERCENTILE, mRssiRange, ResultType.NEUTRAL,
                     ResultUnit.NONE);
             getReportLog().submit();
         }
     }
 }
	/*
	* Copyright (C) 2022 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.cts.verifier.presence;

	import android.bluetooth.BluetoothAdapter;
	import android.bluetooth.BluetoothManager;
	import android.content.pm.PackageManager;
	import android.os.Build;
	import android.os.Bundle;
	import android.util.Log;
	import android.view.View;
	import android.widget.Button;
	import android.widget.CheckBox;
	import android.widget.EditText;
	import android.widget.LinearLayout;
	import android.widget.TextView;
	import android.widget.Toast;

	import com.android.compatibility.common.util.ResultType;
	import com.android.compatibility.common.util.ResultUnit;
	import com.android.cts.verifier.PassFailButtons;
	import com.android.cts.verifier.presence.ble.BleAdvertisingPacket;
	import com.android.cts.verifier.presence.ble.BleScanner;
	import com.android.cts.verifier.presence.ble.BleAdvertiser;
	import com.android.cts.verifier.R;

	import java.util.ArrayList;
	import java.util.Collections;
	import java.util.HashMap;
	import java.util.Random;

	public class BleRssiPrecisionActivity extends PassFailButtons.Activity {
	private static final String TAG = BleRssiPrecisionActivity.class.getName();
	private static final String DEVICE_NAME = Build.MODEL;

	// Report log schema
	private static final String KEY_REFERENCE_DEVICE = "reference_device";
	private static final String KEY_RSSI_RANGE_95_PERCENTILE = "rssi_range_95_percentile";

	// Thresholds
	private static final int MAX_RSSI_RANGE_DBM = 18;

	private boolean isReferenceDevice;
	private BleScanner mBleScanner;
	private BleAdvertiser mBleAdvertiser;
	private BluetoothAdapter mBluetoothAdapter;
	private HashMap<String, ArrayList<Integer>> mRssiResultMap;
	private Button mStartTestButton;
	private Button mStopTestButton;
	private Button mStartAdvertisingButton;
	private Button mStopAdvertisingButton;
	private LinearLayout mDutModeLayout;
	private LinearLayout mRefModeLayout;
	private TextView mDeviceIdInfoTextView;
	private TextView mDeviceFoundTextView;
	private EditText mReferenceDeviceIdInput;
	private String mReferenceDeviceName;
	private CheckBox mIsReferenceDeviceCheckbox;
	private CheckBox mIsManualPassCheckbox;
	private boolean mIsManualPass;
	private boolean mTestCompleted;
	private int mRssiRange;

	@Override
	protected void onCreate(Bundle savedInstanceState) {
	super.onCreate(savedInstanceState);
	setContentView(R.layout.ble_rssi_precision);
	setPassFailButtonClickListeners();
	getPassButton().setEnabled(false);
	mStartTestButton = findViewById(R.id.start_test);
	mStopTestButton = findViewById(R.id.stop_test);
	mStartAdvertisingButton = findViewById(R.id.start_advertising);
	mStopAdvertisingButton = findViewById(R.id.stop_advertising);
	mDeviceIdInfoTextView = findViewById(R.id.device_id_info);
	mDeviceFoundTextView = findViewById(R.id.device_found_info);
	mReferenceDeviceIdInput = findViewById(R.id.ref_device_id_input);
	mIsReferenceDeviceCheckbox = findViewById(R.id.is_reference_device);
	mIsManualPassCheckbox = findViewById(R.id.is_manual_pass);
	mDutModeLayout = findViewById(R.id.dut_mode_layout);
	mRefModeLayout = findViewById(R.id.ref_mode_layout);
	DeviceFeatureChecker.checkFeatureSupported(this, getPassButton(),
	PackageManager.FEATURE_BLUETOOTH_LE);
	BluetoothManager bluetoothManager = getSystemService(BluetoothManager.class);
	mBluetoothAdapter = bluetoothManager.getAdapter();
	mBleAdvertiser = new BleAdvertiser();
	mBleScanner = new BleScanner();
	mRssiResultMap = new HashMap<>();
	mStopTestButton.setEnabled(false);
	mStopAdvertisingButton.setEnabled(false);
	mDeviceIdInfoTextView.setVisibility(View.GONE);
	mDeviceFoundTextView.setVisibility(View.GONE);
	isReferenceDevice = mIsReferenceDeviceCheckbox.isChecked();
	mIsManualPass = mIsManualPassCheckbox.isChecked();
	checkUiMode();
	mIsReferenceDeviceCheckbox.setOnCheckedChangeListener((buttonView, isChecked) -> {
	isReferenceDevice = isChecked;
	checkUiMode();
	});
	mIsManualPassCheckbox.setOnCheckedChangeListener((buttonView, isChecked) -> {
	mIsManualPass = isChecked;
	});
	mStartTestButton.setOnClickListener(v -> startTest());
	mStopTestButton.setOnClickListener(v -> stopTest());
	mStartAdvertisingButton.setOnClickListener(v -> startAdvertising());
	mStopAdvertisingButton.setOnClickListener(v -> stopAdvertising());
	}

	@Override
	public boolean requiresReportLog() {
	return true;
	}

	private void startTest() {
	mTestCompleted = false;
	if (!checkBluetoothEnabled()) {
	return;
	}
	if (mReferenceDeviceIdInput.getText().toString().isEmpty()) {
	makeToast("Input the device ID shown on the advertising device before commencing test");
	return;
	}
	mStartTestButton.setEnabled(false);
	mStopTestButton.setEnabled(true);
	mIsReferenceDeviceCheckbox.setEnabled(false);
	mBleScanner.startScanning((uuids,
	macAddress,
	deviceName,
	referenceDeviceName,
	deviceId,
	rssiMedian,
	rawRssi) -> {

	if (deviceId != Byte.parseByte(mReferenceDeviceIdInput.getText().toString())) {
	//reference device does not match discovered device and scan should be discarded
	Log.i(TAG, "Reference device does not match discovered device. Skipping");
	return;
	}
	mRssiResultMap.computeIfAbsent(deviceName, k -> new ArrayList<>());
	ArrayList<Integer> resultList = mRssiResultMap.get(deviceName);
	resultList.add(rawRssi);
	mDeviceFoundTextView.setVisibility(View.VISIBLE);
	mReferenceDeviceName = referenceDeviceName;
	String deviceFoundText = getString(R.string.device_found_presence,
	resultList.size(), 1000);
	mDeviceFoundTextView.setText(deviceFoundText);
	if (resultList.size() >= 1000) {
	Log.i(TAG, "Data collection complete");
	mBleScanner.stopScanning();
	mStartTestButton.setEnabled(true);
	mStopTestButton.setEnabled(false);
	mIsReferenceDeviceCheckbox.setEnabled(true);
	computeTestResults(resultList);
	}
	});
	}

	private void computeTestResults(ArrayList<Integer> data) {
	Collections.sort(data);
	// Calculate range at 95th percentile
	int rssiRange = data.get(975) - data.get(25);
	mRssiRange = rssiRange;
	mTestCompleted = true;
	if (rssiRange <= MAX_RSSI_RANGE_DBM) {
	makeToast("Test passed! Rssi range is: " + rssiRange);
	if (mIsManualPass) {
	getPassButton().setEnabled(true);
	} else {
	getPassButton().performClick();
	}
	} else {
	makeToast("Test failed! Rssi range is: " + rssiRange);
	}
	data.clear();
	}

	private void stopTest() {
	mBleScanner.stopScanning();
	for (String device : mRssiResultMap.keySet()) {
	mRssiResultMap.get(device).clear();
	}
	mStopTestButton.setEnabled(false);
	mStartTestButton.setEnabled(true);
	mIsReferenceDeviceCheckbox.setEnabled(true);
	}

	private void startAdvertising() {
	if (!checkBluetoothEnabled()) {
	return;
	}
	byte randomAdvertiserDeviceId = getRandomDeviceId();
	String deviceIdInfoText = getString(R.string.device_id_info_presence,
	randomAdvertiserDeviceId);
	mDeviceIdInfoTextView.setText(deviceIdInfoText);
	mDeviceIdInfoTextView.setVisibility(View.VISIBLE);
	String packetDeviceName = "";
	if (DEVICE_NAME.length() > BleAdvertisingPacket.MAX_REFERENCE_DEVICE_NAME_LENGTH) {
	packetDeviceName = DEVICE_NAME.substring(0,
	BleAdvertisingPacket.MAX_REFERENCE_DEVICE_NAME_LENGTH - 1);
	} else {
	packetDeviceName = DEVICE_NAME;
	}
	mBleAdvertiser.startAdvertising(
	new BleAdvertisingPacket(packetDeviceName, randomAdvertiserDeviceId,
	(byte) 0).toBytes());
	mStartAdvertisingButton.setEnabled(false);
	mStopAdvertisingButton.setEnabled(true);
	}

	private void stopAdvertising() {
	mBleAdvertiser.stopAdvertising();
	mStopAdvertisingButton.setEnabled(false);
	mStartAdvertisingButton.setEnabled(true);
	mDeviceIdInfoTextView.setVisibility(View.GONE);
	}

	private boolean checkBluetoothEnabled() {
	if (!mBluetoothAdapter.isEnabled()) {
	makeToast("Bluetooth is not enabled, turn on bluetooth before starting test");
	Log.w(TAG, "Could not start test because Bluetooth is not enabled");
	return false;
	}
	return true;
	}

	private void checkUiMode() {
	if (isReferenceDevice) {
	mDutModeLayout.setVisibility(View.GONE);
	mRefModeLayout.setVisibility(View.VISIBLE);
	} else {
	mRefModeLayout.setVisibility(View.GONE);
	mDutModeLayout.setVisibility(View.VISIBLE);
	}
	}

	private static byte getRandomDeviceId() {
	Random random = new Random();
	byte[] randomDeviceIdArray = new byte[1];
	random.nextBytes(randomDeviceIdArray);
	return randomDeviceIdArray[0];
	}

	private void makeToast(String message) {
	Toast.makeText(this, message, Toast.LENGTH_SHORT).show();
	}

	@Override
	public void recordTestResults() {
	if (mTestCompleted) {
	getReportLog().addValue(KEY_REFERENCE_DEVICE, mReferenceDeviceName,
	ResultType.NEUTRAL, ResultUnit.NONE);
	getReportLog().addValue(KEY_RSSI_RANGE_95_PERCENTILE, mRssiRange, ResultType.NEUTRAL,
	ResultUnit.NONE);
	getReportLog().submit();
	}
	}
	}