如何在android中通过display android获取截图

public void takeScreenshot() {
    try {
        // 获取Display对象
        Display display = ((WindowManager) getSystemService(Context.WINDOW_SERVICE)).getDefaultDisplay();
        
        // 创建Bitmap对象
        Point size = new Point();
        display.getSize(size);
        Bitmap bitmap = Bitmap.createBitmap(size.x, size.y, Bitmap.Config.ARGB_8888);
        
        // 使用Canvas绘制屏幕内容到Bitmap
        Canvas canvas = new Canvas(bitmap);
        View decorView = getWindow().getDecorView();
        decorView.draw(canvas);
        
        // 保存Bitmap为图片文件
        FileOutputStream fos = new FileOutputStream(new File(getExternalFilesDir(null), "screenshot.png"));
        bitmap.compress(Bitmap.CompressFormat.PNG, 100, fos);
        fos.close();
        
        Toast.makeText(this, "截图已保存", Toast.LENGTH_SHORT).show();
    } catch (IOException e) {
        e.printStackTrace();
        Toast.makeText(this, "保存截图失败", Toast.LENGTH_SHORT).show();
    }
}