Android截屏工具类的使用

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>

猜你喜欢

转载自blog.csdn.net/qq_17282141/article/details/113520157