PDF 阅读器简单实现

前言

最近需要实现一个 pdf 的阅读器，抱着很简单的心态就开始了调研，结果发现 Android 对 pdf 的展示只有 PdfRenderer 相关的类可以用，原理还是把 pdf 每一页转成图片来展示！并不能实现文字的复制粘贴等功能。要想有复杂功能的话需要接入第三方的库，怎么也要 10MB了 ，基于此，就用第一种方案进行了开发。

把主要代码贴出来，供大家参考。

效果图

device-2020-11-20-190333.png

实现方案

整体是用一个RecyclerView 来实现的，每一页是一个 Bitmap 的图片，支持展示文件中的 pdf 。

PDF 转换器：

怕你直接贴代码跑不起来，提前说好：

1. PDFManager 这个类，建议每个 ViewHolder 持有一个实例对象，避免创建太多实例对象。
2. 其中的方法都要在子线程中调用，我是用 rxjava 切换的线程，你可以选择你项目里通用的方式

class PDFManager {
    var renderer: PdfRenderer? = null
    var page: PdfRenderer.Page? = null

    /**
         * 获取总页数，需要在子线程中请求，会耗时大概 200 ms
         */
    @kotlin.jvm.Throws(Exception::class, IllegalStateException::class)
    @Synchronized // 一定要加 Synchronized，PdfRenderer 底层不支持多线程调用！
    fun getPageCount(pdfFile: File?): Int {
        check(pdfFile != null && pdfFile.exists()) { "PDF File Does Not Exist" }
        var pageCount: Int = 0
        try {
            var renderer = PdfRenderer(ParcelFileDescriptor.open(pdfFile, ParcelFileDescriptor.MODE_READ_ONLY))
            pageCount = renderer.pageCount
        } catch (ex: Exception) {
            throw ex
        } finally {
            renderer?.close()
        }
        return pageCount
    }

    /**
     * pdf 的第 {pageIndex} 页，转成一个 bitmap
     *
     * @param pdfFile pdf file
     * @param pageIndex the index of the page
     * @return bitmap
     * @throws Exception
     */
    @kotlin.jvm.Throws(Exception::class, IllegalStateException::class)
    @Synchronized
    fun pdfToBitmap(pdfFile: File?, pageIndex: Int, targetWidth: Int): Bitmap? {

        check(pdfFile != null && pdfFile.exists()) { "PDF File Does Not Exist" }
        var pBitmap: Bitmap? = null
        try {
            renderer = PdfRenderer(ParcelFileDescriptor.open(pdfFile, ParcelFileDescriptor.MODE_READ_ONLY))
            page = renderer?.openPage(pageIndex)

            val width = targetWidth
            val height = width * (page?.height ?: 1) / (page?.width ?: 1)

            pBitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888)

            page?.render(pBitmap!!, null, null, PdfRenderer.Page.RENDER_MODE_FOR_DISPLAY)

        } catch (ex: Exception) {
            throw ex
        } finally {
            page?.close()
            renderer?.close()
        }
        return pBitmap
    }

}

然后就是 ViewHolder 里进行调用这些代码了，代码如下：

class PdfItemViewHolder(itemView: View) : RecyclerView.ViewHolder(itemView), View.OnAttachStateChangeListener {
        val imageView = itemView.pdf_item_view
        lateinit var pdfPath: String // pdf 文件位置
    var disposable: Disposable? = null
    var bitmap: Bitmap? = null
    val pdfManager: PDFManager = PDFManager()

    init {
        itemView.addOnAttachStateChangeListener(this)
    }
        
    fun setData(pdfPath: String) {
        this.pdfPath = pdfPath
    }

    //1.view 展示时开始加载 bitmap
      override fun onViewAttachedToWindow(v: View?) {
        loadBitmap()
    }

    //2.view 隐藏时停止加载 bitmap
    override fun onViewDetachedFromWindow(v: View?) {
        RxSafeDispose.safeDispose(disposable)
        imageView.setImageBitmap(null) //3.必须写的！不然 imageView 会报错 "bitmap is recycled"
        bitmap?.recycle() // 回收 bitmap， 释放内存
        bitmap = null
    }

      private fun loadBitmap() {
        val uri = Uri.parse(pdfPath)
        val file = File(uri.path)
        Observable.create {
            try {
                bitmap = pdfManager.pdfToBitmap(file, adapterPosition, pageWidth)
                if (bitmap != null) {
                    it.onNext(bitmap!!)
                } else {
                    it.onError(Throwable("pdf 页面转换失败: file: ${pdfPath}, pageIndex: $adapterPosition"))
                }
            } catch (e: Exception) {
                it.onError(Throwable("pdf 页面转换失败: ${e.message}"))
            }
        }
                .subscribeOn(Schedulers.io()) //4.io 线程里转换 bitmap 图片
                .observeOn(AndroidSchedulers.mainThread())
                .subscribe({               
                    imageView.setImageBitmap(it)                   
                }, {
                    Log.e("pdf", "pdf 页面转换失败: ${it.message}")
                })
                .apply {
                    disposable = this
                }

    }

}

核心代码就在这了， 至于要实现 pdf 的放大缩小也是可以做的，自己在 RecyclerView 外面套一层，对 RecyclerView 整体进行缩放就可以了。

RecyclerView 会在快速滑动时创建大量的 bitmap，对内存来说明显是压力很大的，所以可以在 view 层进行处理一下，滑动时不去调用 loadBitmap，比较简单就不上代码了。

一些思考

方案如上，下面是我开发过程中的一些思考。

上面的方案不是一下子想出来的，刚开始会考虑做 Bitmap 的 LruCache ，但是缓存的目的是什么呢，一是加快第二次加载的速度，二是网络请求中的话可以节省流量。但是我打印出每张图的转换时间最多 200ms，这也可以接受，而且 bitmap 的创建也不用流量，只是消耗一些 cpu 资源。缓存的话内存里也会放一下不太使用的bitmap，相对来说内存压力最大。所以直接放弃了 LruCache。

然后是 bitmap 对象要不要做个复用的池子呢？因为网上确实有很多 bitmap 复用的方案，比如

针对 bitmap 还会有采样率和压缩等优化，我也是跃跃欲试想用，但是，但是，PdfRenderer 并不支持啊，PdfRenderer.Page 的方法只有 page.render(pBitmap, null, null, PdfRenderer.Page.RENDER_MODE_FOR_DISPLAY) 这一种，只能传一个 bitmap 对象进入，但是压缩等技术都是 BitmapFactory 里才支持的，所以，也放弃。

还有是要仿照网络请求，一下转换 5 个 bitmap 出来，然后再一起更新数据呢，因为那样整体时间能降一些，但是那样就比较类似网络请求了，而本文的实现比较类似 ImageLoader，根据实际应用，还是每张图单独加载会好一些。

虽然没有很多代码，却要有很多思考，还要有足够的知识储备。