UI Automator Viewer二次开发（一）—— 提升快照速度

短命女 2023-02-23 10:54 13阅读 0赞

## 先上效果对比图 ##

![111.gif][] 改进前的快照速度![.gif][] 改进后的快照速度

> 速度直接翻倍，起飞！

## 修改方法 ##

### （一）在actions包中添加一个EasyScreenshotAction类 ###

顾名思义，就是一个方便的快照类

代码如下

package com.android.uiautomator.actions;
     
    import java.io.BufferedReader;
    import java.io.File;
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.InputStreamReader;
    import java.lang.reflect.InvocationTargetException;
    import java.text.SimpleDateFormat;
    import java.util.List;
    import java.util.concurrent.LinkedBlockingQueue;
    import java.util.concurrent.ThreadPoolExecutor;
    import java.util.concurrent.TimeUnit;
     
    import org.eclipse.core.runtime.IProgressMonitor;
    import org.eclipse.core.runtime.Status;
    import org.eclipse.jface.action.Action;
    import org.eclipse.jface.dialogs.Dialog;
    import org.eclipse.jface.dialogs.ErrorDialog;
    import org.eclipse.jface.dialogs.MessageDialog;
    import org.eclipse.jface.dialogs.ProgressMonitorDialog;
    import org.eclipse.jface.operation.IRunnableWithProgress;
    import org.eclipse.jface.resource.ImageDescriptor;
    import org.eclipse.swt.events.SelectionAdapter;
    import org.eclipse.swt.events.SelectionEvent;
    import org.eclipse.swt.graphics.Image;
    import org.eclipse.swt.graphics.ImageData;
    import org.eclipse.swt.graphics.ImageLoader;
    import org.eclipse.swt.layout.GridLayout;
    import org.eclipse.swt.widgets.Combo;
    import org.eclipse.swt.widgets.Composite;
    import org.eclipse.swt.widgets.Control;
    import org.eclipse.swt.widgets.Display;
    import org.eclipse.swt.widgets.Label;
    import org.eclipse.swt.widgets.Shell;
     
    import com.android.ddmlib.IDevice;
    import com.android.uiautomator.DebugBridge;
    import com.android.uiautomator.UiAutomatorModel;
    import com.android.uiautomator.UiAutomatorViewer;
     
    public class EasyScreenshotAction extends Action {
    	
    	UiAutomatorViewer mViewer;
    	
    	
    	public EasyScreenshotAction(UiAutomatorViewer viewer) {//构造方法
    		super("&Zoom");//图标名
    		this.mViewer = viewer;
    	}
    	
    	public ImageDescriptor getImageDescriptor() {//按钮使用图标
    		return ImageHelper.loadImageDescriptorFromResource("images/quickshot.png");
    	}
    	
    	public static String getCurrentTime(){//获取当前时间，并转换成固定格式
    		SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd_HH-mm-ss");   
    		return df.format(System.currentTimeMillis());   
    	}
    	
    	public void run() {
    		if(!DebugBridge.isInitialized()) {//无安卓设备时报错
    			MessageDialog.openError(this.mViewer.getShell(), "Error obtaining Device Screenshot", "Unable to connect to adb. Check if adb is installed correctly.");
    		} else {//建立文件夹用于存放两个文件
    			File saveDir=new File("C:\\TempForUiautomatorView") ;
    			if (!saveDir.exists()) {
    				saveDir.mkdirs();
    			}
    			final String saveDirString = saveDir.getAbsolutePath()+File.separator;//File.separator就是斜线
    			final IDevice device = this.pickDevice();//选择一个设备
    			if(device != null) {
    				ProgressMonitorDialog dialog = new ProgressMonitorDialog(this.mViewer.getShell());
    				try {
    					dialog.run(true, false, new IRunnableWithProgress() {
    						public void run(IProgressMonitor monitor) throws InvocationTargetException, InterruptedException {
    							String adbHead = "adb ";
    							try {
    								if (EasyScreenshotAction.DevicePickerDialog.serial ==null || EasyScreenshotAction.DevicePickerDialog.serial.length()<1) {
    									//serial是序列号
    								}else {
    									adbHead="adb -s "+EasyScreenshotAction.DevicePickerDialog.serial+" ";
    								}
    								String  mobileScreenshot=getCurrentTime()+"_screenshot.png";
    								String  mobileXml=getCurrentTime()+"_xml.uix";
    								//使用多线程，来达到更快的同步速度
    								MyThread screenThread=new MyThread(adbHead+"shell /system/bin/screencap -p /data/local/tmp/"+mobileScreenshot);
    								//System.out.println(adbHead+"shell /system/bin/screencap -p /data/local/tmp/"+mobileScreenshot);
    								MyThread xmlThread=new MyThread(adbHead+"shell uiautomator dump /data/local/tmp/"+mobileXml);
    								//System.out.println(adbHead+"shell uiautomator dump /data/local/tmp/"+mobileXml);
    								monitor.setTaskName("Obtaining device screenshot");
    								monitor.subTask("Taking UI XML snapshot...");
    								
    								ThreadPoolExecutor executor = new ThreadPoolExecutor(5,5, 60, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>());  							
    //								Runtime.getRuntime().exec("adb shell content insert --uri content://settings/system --bind name:s:user_rotation --bind value:i:0").waitFor();
    								
    								executor.execute(xmlThread);
    								executor.execute(screenThread);
    //								Runtime.getRuntime().exec("adb shell content insert --uri content://settings/system --bind name:s:accelerometer_rotation --bind value:i:0").waitFor();
    								
    //								executor.execute(new Runnable() {
    									String adbHead = this.adbHead;
    //									@Override
    //									public void run() {
    //										// TODO 自动生成的方法存根
    //										try {
    //											Runtime.getRuntime().exec("adb "+"shell /system/bin/screencap -p /data/local/tmp/"+mobileScreenshot).waitFor();
    //										} catch (Exception e) {
    //											// TODO: handle exception
    //										}
    //										
    //									}
    //								});System.out.println("截图执行完毕");
    //								executor.execute(new Runnable() {
    									String adbHead = this.adbHead;
    //									@Override
    //									public void run() {
    //										// TODO 自动生成的方法存根
    //										try {
    //											Runtime.getRuntime().exec("adb "+"shell uiautomator dump /data/local/tmp/"+mobileXml).waitFor();
    //										} catch (Exception e) {
    //											// TODO: handle exception
    //										}
    //										
    //									}
    //								});
    //								Runtime.getRuntime().exec(adbHead+"shell uiautomator dump /data/local/tmp/"+mobileXml).waitFor();
    //								executor.execute(new MyThread(adbHead+"shell uiautomator dump /data/local/tmp/"+mobileXml));
    								executor.shutdown();//启动关闭，执行原来提交的任务且不接受新的任务
    //								while(true){
    //									if(executor.isTerminated()){
    //										System.out.println("所有的子线程都结束了！");
    //										break;
    //									}
    //									Thread.sleep(1000);  
    //								}
    								
    //								try {
    //									  executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
    //									} catch (InterruptedException e) {
    //									 
    //									}
    //								try {  
    //									boolean loop = true; 
    //									do {    //等待所有任务完成  
    //										loop = !executor.awaitTermination(200, TimeUnit.MILLISECONDS);  //阻塞，直到线程池里所有任务结束
    //										System.out.println("循环中");
    //									} while(loop); 
    //									System.out.println("循环结束啦！");
    //								} catch (InterruptedException e) {  
    //									e.printStackTrace();  
    //								}  
    								try {
    									while(!executor.awaitTermination(200, TimeUnit.MILLISECONDS)) {
    										System.out.println("未结束！");
    									}
    									    System.out.println("jieshule");
    								} catch (InterruptedException e) {
    									// TODO Auto-generated catch block
    									e.printStackTrace();
    								}
    								System.out.println("dump执行完毕");
    								System.out.println("开始执行下面的啦！");
    								
    								//等待线程池中所有线程执行完毕后，再开始pull操作
    								Runtime.getRuntime().exec(adbHead+"pull /data/local/tmp/"+mobileScreenshot+" "+saveDirString+mobileScreenshot).waitFor();
    								monitor.subTask("Obtaining UI hierarchy");
    								Runtime.getRuntime().exec(adbHead+"pull /data/local/tmp/"+mobileXml+" "+saveDirString+mobileXml).waitFor();
    								UiAutomatorModel model = new UiAutomatorModel(new File(saveDirString+mobileXml));
    								Image img = null;
    								File screenshot = new File(saveDirString+mobileScreenshot);
    								//File xml = new File(saveDirString+mobileXml);
    								if(screenshot != null) {
    									ImageData[] e = (new ImageLoader()).load(screenshot.getAbsolutePath());
    									if(e.length < 1) {
    										throw new RuntimeException("Unable to load image: " + screenshot.getAbsolutePath());
    									}
    									img = new Image(Display.getDefault(), e[0]);
    								}
    	 
    	                            //调用setModel方法，在工具中生成快照
    								mViewer.setModel(model,new File(saveDirString+mobileXml), img);
    							} catch (Exception var4) {
    								monitor.done();
    								EasyScreenshotAction.this.showError(var4.getMessage(), var4);
    								return;
    							}
    							monitor.done();
    						}
    					});
    				} catch (Exception var4) {
    					this.showError("Unexpected error while obtaining UI hierarchy", var4);
    				}
    				
    			}
    		}
    	}
    
    	
    	private void showError(final String msg, final Throwable t) {
    		this.mViewer.getShell().getDisplay().syncExec(new Runnable() {
    			public void run() {
    				Status s = new Status(4, "Screenshot", msg, t);
    				ErrorDialog.openError(EasyScreenshotAction.this.mViewer.getShell(), "Error", "Error obtaining UI hierarchy", s);
    			}
    		});
    	}
    	
    	private IDevice pickDevice() {//选择一个设备
    		List<?> devices = DebugBridge.getDevices();
    		if(devices.size() == 0) {//生成一个没有安卓设备的报错对话框
    			MessageDialog.openError(this.mViewer.getShell(), "Error obtaining Device Screenshot", "No Android devices were detected by adb.");
    			return null;
    		} else if(devices.size() == 1) {
    			return (IDevice)devices.get(0);
    		} else {
    			EasyScreenshotAction.DevicePickerDialog dlg = new EasyScreenshotAction.DevicePickerDialog(this.mViewer.getShell(), devices);
    			return dlg.open() != 0?null:dlg.getSelectedDevice();
    		}
    	}
    	
    	private static class DevicePickerDialog extends Dialog {
    		
    		private final List<?> mDevices;
    		private final String[] mDeviceNames;
    		private static int sSelectedDeviceIndex;
    		private static String serial="";
    		
    		
    		public DevicePickerDialog(Shell parentShell, List<?> devices) {
    			super(parentShell);
    			this.mDevices = devices;
    			this.mDeviceNames = new String[this.mDevices.size()];
    			
    			for(int i = 0; i < devices.size(); ++i) {
    				this.mDeviceNames[i] = ((IDevice)devices.get(i)).getName();
    			}
    			
    		}
    		
    		protected Control createDialogArea(Composite parentShell) {
    			Composite parent = (Composite)super.createDialogArea(parentShell);
    			Composite c = new Composite(parent, 0);
    			c.setLayout(new GridLayout(2, false));
    			Label l = new Label(c, 0);
    			l.setText("Select device: ");
    			final Combo combo = new Combo(c, 2056);
    			combo.setItems(this.mDeviceNames);
    			int defaultSelection = sSelectedDeviceIndex < this.mDevices.size()?sSelectedDeviceIndex:0;
    			combo.select(defaultSelection);
    			sSelectedDeviceIndex = defaultSelection;
    			serial = combo.getItem(combo.getSelectionIndex());
    			combo.addSelectionListener(new SelectionAdapter() {
    				public void widgetSelected(SelectionEvent arg0) {
    					EasyScreenshotAction.DevicePickerDialog.sSelectedDeviceIndex = combo.getSelectionIndex();
    					serial = combo.getItem(combo.getSelectionIndex());
    //               System.out.println(serial);
    				}
    			});
    			
    //         System.out.println(serial);
    			return parent;
    		}
    		
    		public IDevice getSelectedDevice() {
    			return (IDevice)this.mDevices.get(sSelectedDeviceIndex);
    		}
    	}
    }
     
    class MyThread extends Thread {
    	public String cmd="";
    	public MyThread(String cmdString) {
    		this.cmd=cmdString;
    	}
    	
    	public void run() {
    		try {
    			Runtime.getRuntime().exec(cmd).waitFor();
    			
    		} catch (Exception e) {
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}  
    		
    		
    	}
    }

### （二）在UiAutomatorViewer类中，添加一个按钮 ###

代码如图所示

![1594083595-1024x161.png][]

### （三）在UiAutomatorViewer类的main函数中 ，添加如下代码 ###

//每次启动时，都要去检查是否有旧文件如果有的话，就删除旧文件
     	   File saveDir=new File("C:\\TempForUiautomatorView") ;
     	   if (saveDir.exists()) {
     		   File TrxFiles[] = saveDir.listFiles();
     		   for(File curFile:TrxFiles ){
     			   curFile.delete();  
     		   }
     	   }

避免文件积累太多，造成C盘的浪费

### （四）打一个jar包，替换原来的uiautomatorviewer.jar文件即可 ###

## 实现原理 ##

## 避免了原版uiautomatorviewer使用socket通信，降低的通信成本。 ##

直接利用adb命令进行截图和dump，将图片和xml dump文件自动传到电脑端（自动创建C:\\TempForUiautomatorView目录）

![15940839751.png][]

同时建立线程池，多线程执行数条adb指令，也能缩短快照时间

最终在电脑端对文件进行解析，获得控件树

[111.gif]: http://101.201.142.67:12/wp-content/uploads/2020/07/改进前速度111.gif
[.gif]: http://101.201.142.67:12/wp-content/uploads/2020/07/改进后速度.gif
[1594083595-1024x161.png]: http://101.201.142.67:12/wp-content/uploads/2020/07/1594083595-1024x161.png
[15940839751.png]: http://101.201.142.67:12/wp-content/uploads/2020/07/15940839751.png