前言
??????在前一篇介紹了Extractor的整個框架以及關(guān)鍵的AnohterPacketSource之后, 我們來介紹一下Extractor中用來讀取文件二進制bit數(shù)據(jù)的類. 她的芳名為: ABitReader(哈哈, 還真是顧名思義呀~).
1. 從"頭"開始
??????按照慣例, 我們首先來看一下這個類的頭文件聲明. 文件路徑為: /frameworks/av/include/media/stagefright/fundation/ABitReader.h(我會在關(guān)鍵方法后加上本菜鳥自己的理解)
28class ABitReader {
29public:
30 ABitReader(const uint8_t *data, size_t size);
31 virtual ~ABitReader();
32
33 // Tries to get |n| bits. If not successful, returns |fallback|. Otherwise, returns result.
34 // Reading 0 bits will always succeed and return 0.
35 uint32_t getBitsWithFallback(size_t n, uint32_t fallback);
36
37 // Tries to get |n| bits. If not successful, returns false. Otherwise, stores result in |out|
38 // and returns true. Use !overRead() to determine if this call was successful. Reading 0 bits
39 // will always succeed and write 0 in |out|.
40 bool getBitsGraceful(size_t n, uint32_t *out);
41
42 // Gets |n| bits and returns result. ABORTS if unsuccessful. Reading 0 bits will always
43 // succeed.
44 uint32_t getBits(size_t n);
45
46 // Tries to skip |n| bits. Returns true iff successful. Skipping 0 bits will always succeed.
47 bool skipBits(size_t n);
48
49 // "Puts" |n| bits with the value |x| back virtually into the bit stream. The put-back bits
50 // are not actually written into the data, but are tracked in a separate buffer that can
51 // store at most 32 bits. This is a no-op if the stream has already been over-read.
52 void putBits(uint32_t x, size_t n);
53 //返回當(dāng)前一共讀了多少個bit
54 size_t numBitsLeft() const;
55 //返回本次讀取bit時, 起始Byte的指針
56 const uint8_t *data() const;
57
58 // Returns true iff the stream was over-read (e.g. any getBits operation has been unsuccessful
59 // due to overread (and not trying to read >32 bits).)
60 bool overRead() const { return mOverRead; }
61
62protected:
63 const uint8_t *mData; //要開始讀取的位置的指針
64 size_t mSize; //要讀取的Byte數(shù)
65
66 uint32_t mReservoir; // left-aligned bits
67 size_t mNumBitsLeft; //本次讀取工作剩余bit數(shù)(多少bit是剩下的)
68 bool mOverRead; //標(biāo)識是否讀取bit越界
69 //這是個基本的函數(shù),名字叫:"填充蓄水池". 作用是檢查mOverRead, 如果沒有OverReader, mNumBitsLedt記錄讀取了多少bit(8的倍數(shù))
70 virtual bool fillReservoir();
71
72 DISALLOW_EVIL_CONSTRUCTORS(ABitReader);
73};
74
75class NALBitReader : public ABitReader {
76public:
77 NALBitReader(const uint8_t *data, size_t size);
78
79 bool atLeastNumBitsLeft(size_t n) const;
80
81private:
82 int32_t mNumZeros;
83
84 virtual bool fillReservoir();
85
86 DISALLOW_EVIL_CONSTRUCTORS(NALBitReader);
87};
2. 追尋本"源"
??????這個頭文件還是非常友好的說, 關(guān)鍵的方法都加了注釋, 小弟只是在沒有注釋的地方, 加了點自己的理解.
??????其實, 常用方法只有getBits或者是skipBits. 讓我們一起來看下.
2.1. getBits
54 uint32_t ABitReader::getBits(size_t n) {
55 uint32_t ret;
56 CHECK(getBitsGraceful(n, &ret));
57 return ret;
58 }
??????我們可以看到, 直接簡單的調(diào)用了getBitGraceful, 然后將拿到的result放入ret中. 哈哈, 接下來看看是如何get bit Graceful(優(yōu)雅的).
2.2 getBitsGraceful
66bool ABitReader::getBitsGraceful(size_t n, uint32_t *out) {
67 if (n > 32) {
68 return false;
69 }
70
71 uint32_t result = 0;
72 while (n > 0) {
73 if (mNumBitsLeft == 0) {
74 if (!fillReservoir()) {
75 return false;
76 }
77 }
78
79 size_t m = n;
80 if (m > mNumBitsLeft) {
81 m = mNumBitsLeft;
82 }
83
84 result = (result << m) | (mReservoir >> (32 - m));
85 mReservoir <<= m;
86 mNumBitsLeft -= m;
87
88 n -= m;
89 }
90
91 *out = result;
92 return true;
93}
67 - 69: 如果有人要搞事情, 一次性讀取bit數(shù)超過了32, 那還是無情的返回false吧;
71 - 89: 開始循環(huán)讀取bit位. 大家一定發(fā)現(xiàn)了一個有趣的方法fillReservoir(), 顧名思義, 他是一個容量最大為32bit的容器. 我們來看看他的實現(xiàn).
34 bool ABitReader::fillReservoir() { 35 if (mSize == 0) { 36 mOverRead = true; 37 return false; 38 } 39 40 mReservoir = 0; 41 size_t i; 42 for (i = 0; mSize > 0 && i < 4; ++i) { 43 mReservoir = (mReservoir << 8) | *mData; 44 45 ++mData; 46 --mSize; 47 } 48 49 mNumBitsLeft = 8 * i; 50 mReservoir <<= 32 - mNumBitsLeft; 51 return true; 52}35- 37: 如果讀到
mSize都沒有了, 那就直接返回false吧;
42- 47: 逐個Byte(8bit去讀mData指向的內(nèi)容), 放入mReservoir;
49 - 50: 更新mNumBitsLeft,mReservoir;
之后就是更新下各私有變量的值, 以便下次讀取之用.
91 - 92 把結(jié)果放入out, 返回讀取成功true.
2.2. skipBits
95 bool ABitReader::skipBits(size_t n) {
96 uint32_t dummy;
97 while (n > 32) {
98 if (!getBitsGraceful(32, &dummy)) {
99 return false;
100 }
101 n -= 32;
102 }
103
104 if (n > 0) {
105 return getBitsGraceful(n, &dummy);
106 }
107 return true;
108}
96: 造一個假的接收數(shù)據(jù)的變量dummy;
97 - 108: 優(yōu)雅的去使用getBitsGraceful去讀取bit位, 反正讀取出來的放在dummy中, 我們不用管. 這樣就達到了我們skip n bit的目的.(有點意思)
小結(jié)
??????這是Extractor中讀取bit位做判斷的基礎(chǔ). 核心方法是getBitsGraceful和fillReservoir. 我們搞清楚他的實現(xiàn), 對于以后自己寫Extractror還是很有幫助的, (更何況, 他的命名規(guī)則和邏輯處理方法也比較有意思_).
