Android音視頻【四】H264硬編碼

人間觀察
你什么時候覺得自己好孤獨

這篇我們介紹如何在Android中實現硬編碼為H264,用camera或者其它數據來源也行,我們先不用camera,camera還是蠻復雜的(比如前后置攝像頭預覽,yuv數據轉換,camera2等等)。本篇采用錄屏(錄制手機屏幕)然后編碼h264到文件。
如果你看了上篇文章了解了硬編碼和h264,這篇就很簡單了。

代碼用kotlin實現,學kotlin中,練練手。。。

錄屏

Android中的錄屏功能是在Android 5.0 API 21 上新增的,不需要root,但需要授權。一般直播的APP都會有這個功能,實現方案就是這個。

方案:
視頻來自屏幕,音頻來自麥克風或者播放器(如果只需要播放器的聲音的話),然后視頻編碼為h264,音頻編碼為aac,交給混合器MediaMuxer將音頻數據和視頻數據分別寫入同一個文件中的音頻軌道和視頻軌道合成為mp4,然后把mp4文件上傳到服務器。

在APP中錄屏是一種很危險的操作,必須很強制的提醒用戶,所以系統進行了彈窗授權。

錄屏請求授權

錄屏主要用到的API是MediaProjectionManagerMediaProjectionMediaProjectionManager和其它系統服務一樣,比如AlarmManager,ConnectivityManager等,通過getSystemService方法獲取。

private lateinit var mediaProjectionManager: MediaProjectionManager
   

mediaProjectionManager =
            getSystemService(Context.MEDIA_PROJECTION_SERVICE) as MediaProjectionManager
val intent: Intent = mediaProjectionManager.createScreenCaptureIntent()
startActivityForResult(intent, REQUEST_CODE)

請求后,系統會彈個對話框提醒是否允許用戶截取屏幕上顯示的所有內容。如果允許/拒絕,則回調onActivityResult方法,系統采用這種機制說明錄屏只能在Activity中使用。

override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?) {
        super.onActivityResult(requestCode, resultCode, data)
        if (resultCode == RESULT_OK && requestCode == REQUEST_CODE) {
            val mediaProjection = mediaProjectionManager.getMediaProjection(resultCode, data!!)
            encodeThread = H264EncodeThread(mediaProjection)
            encodeThread?.startEncode()
        }
    }

ok,簡單的介紹下使用方法即可,我們重點在h264的編碼,錄屏只是作為我們視頻數據的采集方式,所以錄屏這個名字應該加屏幕視頻數據的采集更貼切。

H264硬編碼

我們實現的demo是把錄屏采集的畫面經過MediaCodec編碼后寫入到文件中。還是用MediaCodec硬編碼,關于MediaCodec的使用可以參考上一篇。

初始化編碼器

 init {
        // H264/avc的編碼器
        mediaCodec = MediaCodec.createEncoderByType(MediaFormat.MIMETYPE_VIDEO_AVC)
        val mediaFormat =
            MediaFormat.createVideoFormat(MediaFormat.MIMETYPE_VIDEO_AVC, WIDTH, HEIGHT)

        mediaFormat.setInteger(
            MediaFormat.KEY_COLOR_FORMAT,
            MediaCodecInfo.CodecCapabilities.COLOR_FormatSurface
        )
        // 碼率
        mediaFormat.setInteger(MediaFormat.KEY_BIT_RATE, WIDTH * HEIGHT)
        // i幀間隔,MediaCodec一般會按照KEY_FRAME_RATE輸出
        mediaFormat.setInteger(MediaFormat.KEY_I_FRAME_INTERVAL, 2)
        // 幀率,每25幀后一次關鍵幀,即使畫面不動
        mediaFormat.setInteger(MediaFormat.KEY_FRAME_RATE, 25)

        // 第4個參數需要設置為CONFIGURE_FLAG_ENCODE=1
        mediaCodec.configure(mediaFormat, null, null, MediaCodec.CONFIGURE_FLAG_ENCODE)

        // surface,錄屏和編碼的進行關聯
        val surface = mediaCodec.createInputSurface()
        mediaProjection.createVirtualDisplay(
            "screen-h264", WIDTH, HEIGHT, 2,
            DisplayManager.VIRTUAL_DISPLAY_FLAG_PUBLIC, surface, null, null
        )
    }

其中主要的一點就是如何吧屏幕采集的數據和編碼器輸入進行關聯,我們這里是屏幕采集用的是surface。系統的編碼器提供了mediaCodec.createInputSurface方法來創建一個surface,然后mediaProjection

 mediaProjection.createVirtualDisplay(
            "screen-h264", WIDTH, HEIGHT, 2,
            DisplayManager.VIRTUAL_DISPLAY_FLAG_PUBLIC, surface, null, null
        )

進行關聯。這樣對于上層API調用的時候就簡單了,不需要手動的通過mediaCodecqueueInputBuffer來輸入數據給編碼器了。

  1. 我們傳入的是surface,這個就有了很好的可操控性,比如可以用opengl來截取屏幕指定區域的地址,或者加貼紙,水印,變色等等操作。后續文章可能會介紹
  2. 不需要手動給mediaCodec輸入數據了。

編碼處理

如上所訴這樣編碼的代碼就很簡單了如下:

            while (!isStop) {
                mediaCodec.queueInputBuffer()
                val outIndex = mediaCodec.dequeueOutputBuffer(info, 10_1000)
                if (outIndex >= 0) {
                    // 取出編碼后的H264數據
                    val byteBuffer = mediaCodec.getOutputBuffer(outIndex)
                    val data = ByteArray(info.size)
                    byteBuffer?.get(data)

                    check(data)
                    // to file
                    fos.write(data)

                    // 編碼。給false 不需要渲染
                    mediaCodec.releaseOutputBuffer(outIndex, false);
                }
            }

通過mediaCodec.getOutputBuffer(outIndex)采取出編碼后的h264數據寫入到文件中。同時我們有一個check方法,主要是看一下當前是SPS,PPS,IDR還是非IDR幀。

    private fun check(data: ByteArray) {
        var index = 4  // 00 00 00 01
        if (data[2].toInt() == 0X1) { // 00 00 01
            index = 3
        }
        // NALU的數據類型,header 1個字節的后五位
        val naluType = (data[index].and(0x1F)).toInt()
        if (naluType == 7) {
            Log.d(TAG, "SPS")
        } else if (naluType == 8) {
            Log.d(TAG, "PPS")
        } else if (naluType == 5) {
            Log.d(TAG, "IDR")
        } else {
            Log.d(TAG, "非IDR=" + naluType)
        }
    }

按照h264碼流的結構解析,檢查視頻壓縮數據的NAL單元(NAL Unit,NALU),取出NALU的類型進行判斷,然后進行了打印。

同時我們把生成的h264文件,16進制的方式打開,截取開頭的帶SPS,PPS,IDR幀的數據,和我們之前分析的H264碼流結構一樣。

0000 0001 6742 c01f da02 d028 4400 0000
0168 ca8f 2000 0000 0165 b841 dfff ff04
4245 0001 003f 1c00 0407 2381 be3e 8047
ee36 a6ff e00d 5d26 f9ec 6cc8 d973 dfff
ff82 1580 8666 44f9 95ba 5bff 8f0c 2bc5
952c 51bd 80de 68d2 7e7f fffe 081f 012c
4eec f9bf ffc2 0b89 8030 96cc f002 7538
8045 e9ff ff87 a05a f003 67b2 4383 dfcd
40cd 218c 9edf fff8 7a0e 930e b532 200f
// 省略其它數據...

h264碼流的結構分析參考:
Android音視頻【二】 H264碼流結構

編碼器的銷毀

最后就是釋放操作

isStop = true
mediaProjection.stop()
mediaCodec.stop()
mediaCodec.release()

整體代碼參考源碼

源碼

https://github.com/ta893115871/H264EncodeDemo

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,002評論 6 542
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,400評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,136評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,714評論 1 317
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,452評論 6 412
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,818評論 1 328
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,812評論 3 446
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,997評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,552評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,292評論 3 358
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,510評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,035評論 5 363
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,721評論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,121評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,429評論 1 294
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,235評論 3 398
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,480評論 2 379

推薦閱讀更多精彩內容