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;
@Override
protected 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);
}
@Override
protected 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);
}
}
@Override
protected 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">
<FrameLayout
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_marginTop="30dp">
<ImageView
android: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>