Android截屏工具类的使用

此截屏工具采用MediaProjectionManager类实现帧截图，截屏的结果会包含状态栏和底部虚拟按键，涉及到以下三个工具

CaptureUtil：主要负责截屏

ImageUtil：主要负责对截屏后的Bitmap的处理

ScreenUtil：主要负责截图前屏幕参数的初始化

import android.annotation.SuppressLint;
import android.app.Dialog;
import android.content.Context;
import android.content.Intent;
import android.graphics.Bitmap;
import android.graphics.PixelFormat;
import android.hardware.display.DisplayManager;
import android.media.Image;
import android.media.ImageReader;
import android.media.projection.MediaProjection;
import android.media.projection.MediaProjectionManager;
import android.widget.ImageView;
import com.alrey.robot.R;
import static com.alrey.robot.utils.ImageUtil.covetBitmap;public class CaptureUtil {private static CaptureUtil instance;private static MediaProjection sMediaProjection;private static Bitmap bitmapVertical;private static ImageReader imageReaderVertical;private static Bitmap bitmapHorizontal;private static ImageReader imageReaderHorizontal;public static CaptureUtil getInstance() {if (instance == null){synchronized (CaptureUtil.class){if (instance == null){instance = new CaptureUtil();}}}return instance;}@SuppressLint("WrongConstant")public void init(MediaProjectionManager manager,int resultCode, Intent data){sMediaProjection = manager.getMediaProjection(resultCode, data);if (ScreenUtil.getInstance().isVertical()){//竖屏模式下的截屏初始化imageReaderVertical = ImageReader.newInstance(ScreenUtil.getInstance().getScreenWidth(), ScreenUtil.getInstance().getScreenHeight(), PixelFormat.RGBA_8888, 2);sMediaProjection.createVirtualDisplay("VerticalScreenShot",ScreenUtil.getInstance().getScreenWidth(),ScreenUtil.getInstance().getScreenHeight(),ScreenUtil.getInstance().getDpi(),DisplayManager.VIRTUAL_DISPLAY_FLAG_OWN_CONTENT_ONLY | DisplayManager.VIRTUAL_DISPLAY_FLAG_PUBLIC,imageReaderVertical.getSurface(), null, null);}else {//横屏模式下的截屏初始化imageReaderHorizontal = ImageReader.newInstance(ScreenUtil.getInstance().getScreenWidth(), ScreenUtil.getInstance().getScreenHeight(), PixelFormat.RGBA_8888, 2);sMediaProjection.createVirtualDisplay("VerticalScreenShot",ScreenUtil.getInstance().getScreenWidth(),ScreenUtil.getInstance().getScreenHeight(),ScreenUtil.getInstance().getDpi(),DisplayManager.VIRTUAL_DISPLAY_FLAG_OWN_CONTENT_ONLY | DisplayManager.VIRTUAL_DISPLAY_FLAG_PUBLIC,imageReaderHorizontal.getSurface(), null, null);}}/*** 获取横屏模式下的截屏* */public Bitmap getScreenCapHorizontal() {Bitmap bitmap;Image image;do {image = imageReaderHorizontal.acquireLatestImage();if (image == null && bitmapHorizontal != null) {return bitmapHorizontal;}} while (image == null);bitmap = covetBitmap(image);if (bitmapHorizontal != null) {bitmapHorizontal.recycle();bitmapHorizontal = null;}bitmapHorizontal = bitmap;return bitmap;}/*** 获取竖屏模式下的截屏* */public Bitmap getScreenCapVertical(){Bitmap bitmap;android.media.Image image;do {image = imageReaderVertical.acquireLatestImage();if (image == null && bitmapVertical != null) {return bitmapVertical;}} while (image == null);bitmap = covetBitmap(image);if (bitmapVertical != null) {bitmapVertical.recycle();bitmapVertical = null;}bitmapVertical = bitmap;return bitmap;}/*** 返回对应模式下的屏幕截图* */public Bitmap getScreenCapture(){if (ScreenUtil.getInstance().isVertical()){return getScreenCapVertical();}return getScreenCapHorizontal();}/*** 预览截图* @param img* @param context*/public void show(Bitmap img, Context context) {img = ImageUtil.resize(img,img.getWidth() / 2,img.getHeight() / 2);Dialog dia = new Dialog(context, R.style.edit_AlertDialog_style2);dia.setContentView(R.layout.activity_start_dialog);ImageView imageView = (ImageView) dia.findViewById(R.id.start_img);imageView.setImageBitmap(img);dia.show();dia.setCanceledOnTouchOutside(true);}public void stop(){if (sMediaProjection != null){sMediaProjection.stop();sMediaProjection = null;}}
}

import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.graphics.Matrix;
import android.media.Image;
import android.util.Base64;
import com.orhanobut.logger.Logger;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;public class ImageUtil {/*** Image转Bitmap* */public static Bitmap covetBitmap(Image image) {int width = image.getWidth();int height = image.getHeight();final android.media.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);bitmap = Bitmap.createBitmap(bitmap, 0, 0, width, height);image.close();return bitmap;}/*** 根据给定的宽和高进行resize** @param origin    原图* @param newWidth  新图的宽* @param newHeight 新图的高* @return new Bitmap*/public static Bitmap resize(Bitmap origin, int newWidth, int newHeight) {if (origin == null) {return null;}int height = origin.getHeight();int width = origin.getWidth();float scaleWidth = ((float) newWidth) / width;float scaleHeight = ((float) newHeight) / height;Matrix matrix = new Matrix();matrix.postScale(scaleWidth, scaleHeight);// 使用后乘Bitmap newBM = Bitmap.createBitmap(origin, 0, 0, width, height, matrix, false);if (!origin.isRecycled()) {origin.recycle();}return newBM;}
}

import android.app.Activity;
import android.content.res.Configuration;
import android.util.DisplayMetrics;public class ScreenUtil {private static ScreenUtil instance;private int screenHeight;private int screenWidth;private int dpi;private boolean vertical;public static ScreenUtil getInstance() {if (instance == null){instance = new ScreenUtil();}return instance;}public void init(Activity activity){DisplayMetrics outMetrics = new DisplayMetrics();activity.getWindowManager().getDefaultDisplay().getRealMetrics(outMetrics);screenWidth = outMetrics.widthPixels;screenHeight = outMetrics.heightPixels;dpi = outMetrics.densityDpi;vertical = activity.getResources().getConfiguration().orientation == Configuration.ORIENTATION_PORTRAIT;}public int getScreenHeight() {return screenHeight;}public int getScreenWidth() {return screenWidth;}public int getDpi() {return dpi;}public boolean isVertical() {return vertical;}
}

使用方法

public class MainActivity extends Activity implements Handler.Callback {@BindView(R.id.imageView)ImageView imageView;private final int MEDIA_PROJECTION_REQUEST_CODE = 11;private MediaProjectionManager mediaProjectionManager;@Overrideprotected void onCreate(Bundle savedInstanceState) {super.onCreate(savedInstanceState);setContentView(R.layout.activity_main);ButterKnife.bind(this);ScreenUtil.getInstance().init(this);initCapture();}@OnClick(R.id.start)public void onViewClicked() {Bitmap bitmap = CaptureUtil.getInstance().getScreenCapture();Logger.d(bitmap);}private void initCapture(){mediaProjectionManager = (MediaProjectionManager) getSystemService(MEDIA_PROJECTION_SERVICE);startActivityForResult(mediaProjectionManager.createScreenCaptureIntent(),MEDIA_PROJECTION_REQUEST_CODE);}@Overrideprotected void onActivityResult(int requestCode, int resultCode, Intent data) {super.onActivityResult(requestCode, resultCode, data);//处理截屏权限申请后的结果if (RESULT_OK == resultCode && MEDIA_PROJECTION_REQUEST_CODE == requestCode){CaptureUtil.getInstance().init(mediaProjectionManager,resultCode,data);}}@Overrideprotected void onStop() {super.onStop();CaptureUtil.getInstance().stop();}
}

最后给出edit_AlertDialog_style2和activity_start_dialog

<style name="edit_AlertDialog_style2" parent="@android:style/Theme.Dialog"><item name="android:windowIsFloating">true</item><item name="android:windowIsTranslucent">true</item><item name="android:windowNoTitle">true</item><!-- 是否启用标题栏 --><item name="android:windowBackground">@android:color/transparent</item><item name="android:background">@android:color/transparent</item><item name="android:backgroundDimEnabled">true</item><!-- 是否使用背景半透明 --></style>

<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"android:layout_width="wrap_content"android:layout_height="wrap_content"android:layout_gravity="center_horizontal"><FrameLayoutandroid:layout_width="wrap_content"android:layout_height="wrap_content"android:layout_marginTop="30dp"><ImageViewandroid:id="@+id/start_img"android:layout_width="wrap_content"android:layout_height="wrap_content"android:layout_gravity="center"android:background="@android:color/transparent"android:paddingRight="7dp"android:paddingTop="7dp"android:scaleType="fitXY" /></FrameLayout></LinearLayout>