1. 程式人生 > >Android三種方式擷取任意介面螢幕

Android三種方式擷取任意介面螢幕

一、使用MediaProjectionManager

Android5.0之後,開放擷取螢幕的API,也就是利用MediaProjectionManager建立VirtualDisplay,傳入與ImageReader關聯的Surface,這樣就可以從ImageReader中獲取到Image,然後把Image的畫素陣列拷貝到Bitmap,如果要儲存為圖片,就用得到的Bitmap壓縮為JPEG格式的圖片。

首先是使用startActivityForResult發起錄屏的請求:

    private void startScreenShot(){
        WindowManager windowManager = (WindowManager) getSystemService(Context.WINDOW_SERVICE);
        if (windowManager != null) {
            DisplayMetrics displayMetrics = new DisplayMetrics();
            windowManager.getDefaultDisplay().getMetrics(displayMetrics);
            width = displayMetrics.widthPixels;
            height = displayMetrics.heightPixels;
            dpi = displayMetrics.densityDpi;
        }

        mediaProjectionManager = (MediaProjectionManager) getSystemService(MEDIA_PROJECTION_SERVICE);
        if (mediaProjectionManager != null) {
            startActivityForResult(mediaProjectionManager.createScreenCaptureIntent(), 123);
        }
    }

在得到使用者授權錄屏後,在onActivityResult中獲取MediaProjection-->VirtualDisplay-->ImagReader-->Image-->Bitmap:

    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        super.onActivityResult(requestCode, resultCode, data);
        MediaProjection mediaProjection = mediaProjectionManager.getMediaProjection(resultCode, data);
        if (mediaProjection != null){
            getBitmap(mediaProjection);
        }
    }
private void getBitmap(MediaProjection mediaProjection){
        ImageReader imageReader = ImageReader.newInstance(width, height, PixelFormat.RGBA_8888, 3);
        mediaProjection.createVirtualDisplay("screen_shot",
                width, height, dpi, DisplayManager.VIRTUAL_DISPLAY_FLAG_AUTO_MIRROR,
                imageReader.getSurface(), null, null);
        imageReader.setOnImageAvailableListener(new ImageReader.OnImageAvailableListener() {
            @Override
            public void onImageAvailable(ImageReader reader) {
                Image image = reader.acquireNextImage();
                int width = image.getWidth();
                int height = image.getHeight();
                final Image.Plane[] planes = image.getPlanes();
                final ByteBuffer buffer = planes[0].getBuffer();
                int pixelStride = planes[0].getPixelStride();
                int rowStride = planes[0].getRowStride();
                int rowPadding = rowStride - pixelStride * width;
                Bitmap bitmap = Bitmap.createBitmap(width+rowPadding/pixelStride, height, Bitmap.Config.ARGB_8888);
                bitmap.copyPixelsFromBuffer(buffer);
                String filePath = Environment.getExternalStorageDirectory().getPath() + "/hello.jpg";
                //bitmap儲存為圖片
                saveBitmap(bitmap, filePath);
                image.close();
            }
        }, null);

    }
private void saveBitmap(Bitmap bitmap, String filePath){
        try {
            FileOutputStream outputStream = new FileOutputStream(filePath);
            bitmap.compress(Bitmap.CompressFormat.JPEG, 100, outputStream);
            outputStream.flush();
            outputStream.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

優點:不用系統簽名,不依賴系統底層API;

缺點:彈出確認框,需要使用者授權錄屏;

二、使用SurfaceControl

從上面分析可知,MediaProjectionManager錄屏依賴Surface,分析Surface原始碼後,發現Surface其實是呼叫SurfaceControl,也就是說可以用SurfaceControl走捷徑去截圖。其實系統框架層的截圖也是呼叫SurfaceControl。但是,由於SurfaceControl屬於系統API,對使用者不開放,我們無法直接呼叫。說到這裡,大家應該都想到,用反射機制來呼叫。

正確的呼叫方式是這樣的:

Bitmap bitmap = SurfaceControl.screenshot(width, height);

通過系統原始碼,可以知道它在Android.view.SurfaceControl路徑下。既然路徑、類、方法、引數都知道了,反射就沒問題:

	//使用反射呼叫截圖
	private void screenShotByReflect(){
		DisplayMetrics mDisplayMetrics = new DisplayMetrics();
		float[] dims = { mDisplayMetrics.widthPixels,
				mDisplayMetrics.heightPixels };
		try {
			Class<?> demo = Class.forName("android.view.SurfaceControl");
			Method method = demo.getDeclaredMethod("screenshot", int.class,int.class);
			mScreenBitmap = (Bitmap) method.invoke(null,(int) dims[0],(int) dims[1]);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

在呼叫截圖之前,判斷螢幕是否發生旋轉:

	private Bitmap takeScreenshot() {
		 mDisplay.getRealMetrics(mDisplayMetrics);
		 float[] dims = {mDisplayMetrics.widthPixels, mDisplayMetrics.heightPixels};
		float degrees = getDegreesForRotation(mDisplay.getRotation());
		boolean requiresRotation = (degrees > 0);
		//如果螢幕發生旋轉,通過matrix旋轉回來
		if (requiresRotation) {
			mDisplayMatrix.reset();
			mDisplayMatrix.preRotate(-degrees);
			mDisplayMatrix.mapPoints(dims);
			dims[0] = Math.abs(dims[0]);
			dims[1] = Math.abs(dims[1]);
		}
		//呼叫截圖
		screenShotByReflect();
		return mScreenBitmap;
	}

優點:不用彈框授權,不用系統簽名;

缺點:使用反射機制,如果系統API或者方法發生更改,導致無法呼叫;

三、使用screencap的adb命令

命令列是這樣的:adb shell screencap -p file_path

在程式碼中執行,就不用adb shell,直接screencap -p file_path,呼叫Runtime的程序來執行:

	public static void screenShotByShell(String filePath){
		String shotCmd = "screencap -p " + filePath + " \n";
		try {
			Runtime.getRuntime().exec(shotCmd);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

優點:程式碼簡單,直接獲取到圖片;

缺點:需要系統簽名;

以上三種截圖方式,大家可以根據應用場景來使用。如果是使用者app,需要連續錄屏,建議採用MediaProjectionManager;如果是希望得到單個Bitmap,那麼可以用SurfaceControl;如果是系統app,並且希望得到圖片,首選screencap的adb命令列。