/**
 * 截取當(dāng)前窗體的截圖，根據(jù)[isShowStatusBar]判斷是否包含當(dāng)前窗體的狀態(tài)欄
 * 原理是獲取當(dāng)前窗體decorView的緩存生成圖片
 */
fun captureWindow(activity: Activity, isShowStatusBar: Boolean): Bitmap? {
    // 獲取當(dāng)前窗體的View對(duì)象
    val view = activity.window.decorView
    view.isDrawingCacheEnabled = true
    // 生成緩存
    view.buildDrawingCache()

    val bitmap = if (isShowStatusBar) {
        // 繪制整個(gè)窗體，包括狀態(tài)欄
        Bitmap.createBitmap(view.drawingCache, 0, 0, view.measuredWidth, view.measuredHeight)
    } else {
        // 獲取狀態(tài)欄高度
        val rect = Rect()
        view.getWindowVisibleDisplayFrame(rect)
        val display = activity.windowManager.defaultDisplay

        // 減去狀態(tài)欄高度
        Bitmap.createBitmap(view.drawingCache, 0,
                rect.top, display.width, display.height - rect.top)
    }

    view.isDrawingCacheEnabled = false
    view.destroyDrawingCache()

    return bitmap
}

/**
  * View已經(jīng)在界面上展示了，可以直接獲取View的緩存
  * 對(duì)View進(jìn)行量測(cè)，布局后生成View的緩存
  * View為固定大小的View，包括TextView,ImageView,LinearLayout,FrameLayout,RelativeLayout等
  * @param view 截取的View,View必須有固定的大小，不然drawingCache返回null
  * @return 生成的Bitmap
  */
fun captureView(view: View): Bitmap? {
        view.isDrawingCacheEnabled = true
        view.buildDrawingCache()
        // 重新測(cè)量一遍View的寬高
        view.measure(View.MeasureSpec.makeMeasureSpec(view.width, View.MeasureSpec.EXACTLY),
                View.MeasureSpec.makeMeasureSpec(view.height, View.MeasureSpec.EXACTLY))
        // 確定View的位置
        view.layout(view.x.toInt(), view.y.toInt(), view.x.toInt() + view.measuredWidth,
                view.y.toInt() + view.measuredHeight)
        // 生成View寬高一樣的Bitmap
        val bitmap = Bitmap.createBitmap(view.drawingCache, 0, 0, view.measuredWidth,
                view.measuredHeight)
        view.isDrawingCacheEnabled = false
        view.destroyDrawingCache()
        return bitmap
}

/**
 * 截取ScrollerView
 * 原理是獲取scrollView的子View的高度，然后創(chuàng)建一個(gè)子View寬高的畫(huà)布，將ScrollView繪制在畫(huà)布上
 * @param scrollView 控件
 * @return 返回截圖后的Bitmap
 */
fun captureScrollView(scrollView: ScrollView): Bitmap? {
     var h = 0
     for (i in 0 until scrollView.childCount) {
         val childView = scrollView.getChildAt(i)
         // 獲取子View的高度
         h += childView.height
         // 設(shè)置背景顏色，避免布局里未設(shè)置背景顏色，截的圖背景黑色
         childView.setBackgroundColor(Color.parseColor("#FFFFFF"))
     }

     val bitmap = createBitmap(scrollView.width, h)
     val canvas = Canvas(bitmap)
     // 將ScrollView繪制在畫(huà)布上
     scrollView.draw(canvas)
     return bitmap
}

/**
 * 截取ListView
 * 原理：獲取到每一個(gè)子View，將子View生成的bitmap存入集合，并且累積ListView高度
 * 遍歷完成后，創(chuàng)建一個(gè)ListView大小的畫(huà)布，將集合的Bitmap繪制到畫(huà)布上
 * @param listView 截圖控件對(duì)象
 * @return 生成的截圖對(duì)象
 */
fun captureListView(listView: ListView): Bitmap? {
        val adapter = listView.adapter
        val itemCount = adapter.count
        var allitemsheight = 0
        val bitmaps = ArrayList<Bitmap>()

        for (i in 0 until itemCount) {
            // 獲取每一個(gè)子View
            val childView = adapter.getView(i, null, listView)
            // 測(cè)量寬高
            childView.measure(
                    View.MeasureSpec.makeMeasureSpec(listView.width, View.MeasureSpec.EXACTLY),
                    View.MeasureSpec.makeMeasureSpec(0, View.MeasureSpec.UNSPECIFIED))

            // 布局位置
            childView.layout(0, 0, childView.measuredWidth, childView.measuredHeight)
            // 設(shè)置背景顏色，避免是黑色的
            childView.setBackgroundColor(Color.parseColor("#FFFFFF"))
            childView.isDrawingCacheEnabled = true
            // 生成緩存
            childView.buildDrawingCache()
            // 將每一個(gè)View的截圖加入集合
            bitmaps.add(childView.drawingCache)
            // 疊加截圖高度
            allitemsheight += childView.measuredHeight
        }

        // 創(chuàng)建和ListView寬高一樣的畫(huà)布
        val bitmap = createBitmap(listView.measuredWidth, allitemsheight)
        val canvas = Canvas(bitmap)

        val paint = Paint()
        var iHeight = 0f

        for (i in bitmaps.indices) {
            val bmp: Bitmap = bitmaps[i]
            // 將每一個(gè)生成的bitmap繪制在畫(huà)布上
            canvas.drawBitmap(bmp, 0f, iHeight, paint)
            iHeight += bmp.height

            bmp.recycle()
        }
        return bitmap
}

/**
 * 截取RecyclerView
 * 原理和ListView集合是一樣的，獲取到每一個(gè)Holder的截圖放入集合，最后統(tǒng)一繪制到Bitmap上
 * @param recyclerView 要截圖的控件
 * @return 生成的截圖
 */
fun captureRecyclerView(recyclerView: RecyclerView): Bitmap? {
        val adapter = recyclerView.adapter
        var bigBitmap: Bitmap? = null
        if (adapter != null) {
            val size = adapter.itemCount
            var height = 0
            val paint = Paint()
            var iHeight = 0
            val maxMemory = (Runtime.getRuntime().maxMemory() / 1024).toInt()

            // Use 1/8th of the available memory for this memory cache.
            val cacheSize = maxMemory / 8
            val bitmapCache = LruCache<String, Bitmap>(cacheSize)
            for (i in 0 until size) {
                val holder = adapter.createViewHolder(recyclerView, adapter.getItemViewType(i))
                adapter.onBindViewHolder(holder, i)
                holder.itemView.measure(
                        View.MeasureSpec.makeMeasureSpec(recyclerView.width, View.MeasureSpec.EXACTLY),
                        View.MeasureSpec.makeMeasureSpec(0, View.MeasureSpec.UNSPECIFIED))
                holder.itemView.layout(0, 0, holder.itemView.measuredWidth,
                        holder.itemView.measuredHeight)
                holder.itemView.setBackgroundColor(Color.parseColor("#FFFFFF"))
                holder.itemView.isDrawingCacheEnabled = true
                holder.itemView.buildDrawingCache()
                val drawingCache = holder.itemView.drawingCache
                if (drawingCache != null) {
                    bitmapCache.put(i.toString(), drawingCache)
                }
                height += holder.itemView.measuredHeight
            }

            bigBitmap = createBitmap(recyclerView.measuredWidth, height)
            val bigCanvas = Canvas(bigBitmap!!)
            val lBackground = recyclerView.background
            if (lBackground is ColorDrawable) {
                val lColor = lBackground.color
                bigCanvas.drawColor(lColor)
            }

            for (i in 0 until size) {
                val bitmap = bitmapCache.get(i.toString())
                bigCanvas.drawBitmap(bitmap, 0f, iHeight.toFloat(), paint)
                iHeight += bitmap.height
                bitmap.recycle()
            }
        }
        return bigBitmap
}

/**
 * 截取WebView，包含WebView的整個(gè)長(zhǎng)度
 * 在WebView渲染之前要加上以下代碼，開(kāi)啟Html緩存，不然會(huì)截屏空白
 *  if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
 *      WebView.enableSlowWholeDocumentDraw()
 *  }
 *  WebView的截圖很容易遇到內(nèi)存溢出的問(wèn)題，因?yàn)閃ebView可以加載很多內(nèi)容，導(dǎo)致生成的圖片特別長(zhǎng)，創(chuàng)建Bitmap時(shí)容易OOM
 */
fun captureWebView(webView: WebView): Bitmap? {
        // 重新調(diào)用WebView的measure方法測(cè)量實(shí)際View的大?。▽y(cè)量模式設(shè)置為UNSPECIFIED模式也就是需要多大就可以獲得多大的空間）
        webView.measure(View.MeasureSpec.makeMeasureSpec(View.MeasureSpec.UNSPECIFIED, View.MeasureSpec.UNSPECIFIED),
                View.MeasureSpec.makeMeasureSpec(0, View.MeasureSpec.UNSPECIFIED))
        // 調(diào)用layout方法設(shè)置布局（使用新測(cè)量的大小）
        webView.layout(0, 0, webView.measuredWidth, webView.measuredHeight)
        // 開(kāi)啟WebView的緩存(當(dāng)開(kāi)啟這個(gè)開(kāi)關(guān)后下次調(diào)用getDrawingCache()方法的時(shí)候會(huì)把view繪制到一個(gè)bitmap上)
        webView.isDrawingCacheEnabled = true
        // 強(qiáng)制繪制緩存（必須在setDrawingCacheEnabled(true)之后才能調(diào)用，否者需要手動(dòng)調(diào)用destroyDrawingCache()清楚緩存）
        webView.buildDrawingCache()

        val bitmap = createBitmap(webView.measuredWidth, webView.measuredHeight)

        // 已picture為背景創(chuàng)建一個(gè)畫(huà)布
        val canvas = Canvas(bitmap)  // 畫(huà)布的寬高和 WebView 的網(wǎng)頁(yè)保持一致
        val paint = Paint()
        // 設(shè)置畫(huà)筆的定點(diǎn)位置，也就是左上角
        canvas.drawBitmap(bitmap, 0f, webView.measuredHeight * 1f, paint)
        // 將WebView繪制在剛才創(chuàng)建的畫(huà)板上
        webView.draw(canvas)
        webView.isDrawingCacheEnabled = false
        webView.destroyDrawingCache()
        return bitmap
}