private unsafe byte[] EncodePrediction(Bitmap Frame) { //We'll just look if there are blocks that are 100% the same first BitmapData d = Frame.LockBits(new Rectangle(0, 0, Frame.Width, Frame.Height), ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb); BitmapData prev = PastFrames[1].LockBits(new Rectangle(0, 0, Frame.Width, Frame.Height), ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb); for (int y = 0; y < Height; y += 16) { for (int x = 0; x < Width; x += 16) { MacroBlocks[y / 16][x / 16] = new MacroBlock(d, x, y); bool same = true; for (int y2 = 0; y2 < 16; y2++) { ulong* pA = (ulong*)(((byte*)d.Scan0) + (y + y2) * d.Stride + x * 4); ulong* pB = (ulong*)(((byte*)prev.Scan0) + (y + y2) * prev.Stride + x * 4); for (int x2 = 0; x2 < 16; x2+=2) { ulong color = *pA++; ulong color2 = *pB++; if (color != color2) { same = false; break; } } if (!same) break; } if (same) MacroBlocks[y / 16][x / 16].Predict = true; else { Analyzer.ConfigureBlockY(this, MacroBlocks[y / 16][x / 16]); int blocktype2 = Analyzer.AnalyzeBlockUV(this, MacroBlocks[y / 16][x / 16]); MacroBlocks[y / 16][x / 16].UVPredictionMode = blocktype2; MacroBlocks[y / 16][x / 16].UVUseComplex8x8[0] = true; MacroBlocks[y / 16][x / 16].UVUseComplex8x8[1] = true; MacroBlocks[y / 16][x / 16].SetupDCTs(this); } } } Frame.UnlockBits(d); PastFrames[1].UnlockBits(prev); BitWriter b = new BitWriter(); b.WriteBits(0, 1);//Interframe b.WriteVarIntSigned(0); for (int y = 0; y < Height; y += 16) { for (int x = 0; x < Width; x += 16) { MacroBlock curblock = MacroBlocks[y / 16][x / 16]; if (curblock.Predict) { b.WriteBits(1, 1); b.WriteVarIntUnsigned(0); } else { b.WriteBits(0xE >> 1, 5); EncodeBlockIntra(curblock, b); } } } b.WriteBits(0, 16); b.Flush(); byte[] result = b.ToArray(); return result; }
private void EncodeBlockIntra(MacroBlock Block, BitWriter b) { uint dctmask = (Block.YUseComplex8x8[0] ? 1u : 0) | ((Block.YUseComplex8x8[1] ? 1u : 0) << 1) | ((Block.YUseComplex8x8[2] ? 1u : 0) << 2) | ((Block.YUseComplex8x8[3] ? 1u : 0) << 3) | ((Block.UVUseComplex8x8[0] ? 1u : 0) << 4) | ((Block.UVUseComplex8x8[1] ? 1u : 0) << 5); b.WriteVarIntUnsigned(REV_byte_115FC4[dctmask]); b.WriteBits((uint)Block.YPredictionMode, 3);//Block type for (int y2 = 0; y2 < 2; y2++) { for (int x2 = 0; x2 < 2; x2++) { if (Block.YUseComplex8x8[x2 + y2 * 2] && !Block.YUse4x4[x2 + y2 * 2]) { b.WriteBits(1, 1);//Don't use 4x4 blocks EncodeDCT(Block.YDCT8x8[x2 + y2 * 2], 0, b); } else if (Block.YUseComplex8x8[x2 + y2 * 2] && Block.YUse4x4[x2 + y2 * 2]) { uint dctmask2 = (Block.YUseDCT4x4[x2 + y2 * 2][0] ? 1u : 0) | ((Block.YUseDCT4x4[x2 + y2 * 2][1] ? 1u : 0) << 1) | ((Block.YUseDCT4x4[x2 + y2 * 2][2] ? 1u : 0) << 2) | ((Block.YUseDCT4x4[x2 + y2 * 2][3] ? 1u : 0) << 3); b.WriteVarIntUnsigned(REV_byte_1164F4[dctmask2]); for (int y3 = 0; y3 < 2; y3++) { for (int x3 = 0; x3 < 2; x3++) { if (Block.YUseDCT4x4[x2 + y2 * 2][x3 + y3 * 2]) EncodeDCT(Block.YDCT4x4[x2 + y2 * 2][x3 + y3 * 2], 0, b); } } } } } b.WriteBits((uint)Block.UVPredictionMode, 3);//Block type for (int q = 0; q < 2; q++) { if (Block.UVUseComplex8x8[q] && !Block.UVUse4x4[q]) { b.WriteBits(1, 1);//Don't use 4x4 blocks EncodeDCT(Block.UVDCT8x8[q], 0, b); } } }