mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 1 | #include "SkColorPriv.h" |
| 2 | #include "SkTableColorFilter.h" |
| 3 | #include "SkUnPreMultiply.h" |
| 4 | |
| 5 | class SkTable_ColorFilter : public SkColorFilter { |
| 6 | public: |
| 7 | SkTable_ColorFilter(const uint8_t tableA[], const uint8_t tableR[], |
| 8 | const uint8_t tableG[], const uint8_t tableB[]) { |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 9 | fBitmap = NULL; |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 10 | fFlags = 0; |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 11 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 12 | uint8_t* dst = fStorage; |
| 13 | if (tableA) { |
| 14 | memcpy(dst, tableA, 256); |
| 15 | dst += 256; |
| 16 | fFlags |= kA_Flag; |
| 17 | } |
| 18 | if (tableR) { |
| 19 | memcpy(dst, tableR, 256); |
| 20 | dst += 256; |
| 21 | fFlags |= kR_Flag; |
| 22 | } |
| 23 | if (tableG) { |
| 24 | memcpy(dst, tableG, 256); |
| 25 | dst += 256; |
| 26 | fFlags |= kG_Flag; |
| 27 | } |
| 28 | if (tableB) { |
| 29 | memcpy(dst, tableB, 256); |
| 30 | fFlags |= kB_Flag; |
| 31 | } |
| 32 | } |
| 33 | |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 34 | virtual bool asComponentTable(SkBitmap* table) SK_OVERRIDE; |
| 35 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 36 | virtual void filterSpan(const SkPMColor src[], int count, |
| 37 | SkPMColor dst[]) SK_OVERRIDE; |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 38 | |
djsollen@google.com | ba28d03 | 2012-03-26 17:57:35 +0000 | [diff] [blame] | 39 | SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkTable_ColorFilter) |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 40 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 41 | protected: |
| 42 | SkTable_ColorFilter(SkFlattenableReadBuffer& buffer); |
djsollen@google.com | 5492424 | 2012-03-29 15:18:04 +0000 | [diff] [blame] | 43 | virtual void flatten(SkFlattenableWriteBuffer&) const SK_OVERRIDE; |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 44 | |
| 45 | private: |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 46 | SkBitmap* fBitmap; |
| 47 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 48 | enum { |
| 49 | kA_Flag = 1 << 0, |
| 50 | kR_Flag = 1 << 1, |
| 51 | kG_Flag = 1 << 2, |
| 52 | kB_Flag = 1 << 3, |
| 53 | }; |
| 54 | uint8_t fStorage[256 * 4]; |
| 55 | unsigned fFlags; |
| 56 | |
| 57 | typedef SkColorFilter INHERITED; |
| 58 | }; |
| 59 | |
| 60 | static const uint8_t gIdentityTable[] = { |
| 61 | 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, |
| 62 | 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, |
| 63 | 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, |
| 64 | 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F, |
| 65 | 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, |
| 66 | 0x28, 0x29, 0x2A, 0x2B, 0x2C, 0x2D, 0x2E, 0x2F, |
| 67 | 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, |
| 68 | 0x38, 0x39, 0x3A, 0x3B, 0x3C, 0x3D, 0x3E, 0x3F, |
| 69 | 0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, |
| 70 | 0x48, 0x49, 0x4A, 0x4B, 0x4C, 0x4D, 0x4E, 0x4F, |
| 71 | 0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, |
| 72 | 0x58, 0x59, 0x5A, 0x5B, 0x5C, 0x5D, 0x5E, 0x5F, |
| 73 | 0x60, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, |
| 74 | 0x68, 0x69, 0x6A, 0x6B, 0x6C, 0x6D, 0x6E, 0x6F, |
| 75 | 0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, |
| 76 | 0x78, 0x79, 0x7A, 0x7B, 0x7C, 0x7D, 0x7E, 0x7F, |
| 77 | 0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, |
| 78 | 0x88, 0x89, 0x8A, 0x8B, 0x8C, 0x8D, 0x8E, 0x8F, |
| 79 | 0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, |
| 80 | 0x98, 0x99, 0x9A, 0x9B, 0x9C, 0x9D, 0x9E, 0x9F, |
| 81 | 0xA0, 0xA1, 0xA2, 0xA3, 0xA4, 0xA5, 0xA6, 0xA7, |
| 82 | 0xA8, 0xA9, 0xAA, 0xAB, 0xAC, 0xAD, 0xAE, 0xAF, |
| 83 | 0xB0, 0xB1, 0xB2, 0xB3, 0xB4, 0xB5, 0xB6, 0xB7, |
| 84 | 0xB8, 0xB9, 0xBA, 0xBB, 0xBC, 0xBD, 0xBE, 0xBF, |
| 85 | 0xC0, 0xC1, 0xC2, 0xC3, 0xC4, 0xC5, 0xC6, 0xC7, |
| 86 | 0xC8, 0xC9, 0xCA, 0xCB, 0xCC, 0xCD, 0xCE, 0xCF, |
| 87 | 0xD0, 0xD1, 0xD2, 0xD3, 0xD4, 0xD5, 0xD6, 0xD7, |
| 88 | 0xD8, 0xD9, 0xDA, 0xDB, 0xDC, 0xDD, 0xDE, 0xDF, |
| 89 | 0xE0, 0xE1, 0xE2, 0xE3, 0xE4, 0xE5, 0xE6, 0xE7, |
| 90 | 0xE8, 0xE9, 0xEA, 0xEB, 0xEC, 0xED, 0xEE, 0xEF, |
| 91 | 0xF0, 0xF1, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7, |
| 92 | 0xF8, 0xF9, 0xFA, 0xFB, 0xFC, 0xFD, 0xFE, 0xFF |
| 93 | }; |
| 94 | |
| 95 | void SkTable_ColorFilter::filterSpan(const SkPMColor src[], int count, |
| 96 | SkPMColor dst[]) { |
| 97 | const uint8_t* table = fStorage; |
| 98 | const uint8_t* tableA = gIdentityTable; |
| 99 | const uint8_t* tableR = gIdentityTable; |
| 100 | const uint8_t* tableG = gIdentityTable; |
| 101 | const uint8_t* tableB = gIdentityTable; |
| 102 | if (fFlags & kA_Flag) { |
| 103 | tableA = table; table += 256; |
| 104 | } |
| 105 | if (fFlags & kR_Flag) { |
| 106 | tableR = table; table += 256; |
| 107 | } |
| 108 | if (fFlags & kG_Flag) { |
| 109 | tableG = table; table += 256; |
| 110 | } |
| 111 | if (fFlags & kB_Flag) { |
| 112 | tableB = table; |
| 113 | } |
| 114 | |
| 115 | const SkUnPreMultiply::Scale* scaleTable = SkUnPreMultiply::GetScaleTable(); |
| 116 | for (int i = 0; i < count; ++i) { |
| 117 | SkPMColor c = src[i]; |
| 118 | unsigned a, r, g, b; |
| 119 | if (0 == c) { |
| 120 | a = r = g = b = 0; |
| 121 | } else { |
| 122 | a = SkGetPackedA32(c); |
| 123 | r = SkGetPackedR32(c); |
| 124 | g = SkGetPackedG32(c); |
| 125 | b = SkGetPackedB32(c); |
| 126 | |
| 127 | if (a < 255) { |
| 128 | SkUnPreMultiply::Scale scale = scaleTable[a]; |
| 129 | r = SkUnPreMultiply::ApplyScale(scale, r); |
| 130 | g = SkUnPreMultiply::ApplyScale(scale, g); |
| 131 | b = SkUnPreMultiply::ApplyScale(scale, b); |
| 132 | } |
| 133 | } |
| 134 | dst[i] = SkPremultiplyARGBInline(tableA[a], tableR[r], |
| 135 | tableG[g], tableB[b]); |
| 136 | } |
| 137 | } |
| 138 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 139 | static const uint8_t gCountNibBits[] = { |
| 140 | 0, 1, 1, 2, |
| 141 | 1, 2, 2, 3, |
| 142 | 1, 2, 2, 3, |
| 143 | 2, 3, 3, 4 |
| 144 | }; |
| 145 | |
| 146 | #include "SkPackBits.h" |
| 147 | |
djsollen@google.com | 5492424 | 2012-03-29 15:18:04 +0000 | [diff] [blame] | 148 | void SkTable_ColorFilter::flatten(SkFlattenableWriteBuffer& buffer) const { |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 149 | this->INHERITED::flatten(buffer); |
| 150 | |
| 151 | uint8_t storage[5*256]; |
| 152 | int count = gCountNibBits[fFlags & 0xF]; |
| 153 | size_t size = SkPackBits::Pack8(fStorage, count * 256, storage); |
| 154 | SkASSERT(size <= sizeof(storage)); |
| 155 | |
| 156 | // SkDebugf("raw %d packed %d\n", count * 256, size); |
| 157 | |
| 158 | buffer.writeInt(fFlags); |
| 159 | buffer.writeInt(size); |
| 160 | buffer.write(storage, size); |
| 161 | } |
| 162 | |
| 163 | SkTable_ColorFilter::SkTable_ColorFilter(SkFlattenableReadBuffer& buffer) : INHERITED(buffer) { |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 164 | fBitmap = NULL; |
| 165 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 166 | uint8_t storage[5*256]; |
| 167 | |
| 168 | fFlags = buffer.readInt(); |
| 169 | size_t size = buffer.readInt(); |
| 170 | buffer.read(storage, size); |
| 171 | |
| 172 | size_t raw = SkPackBits::Unpack8(storage, size, fStorage); |
| 173 | |
| 174 | SkASSERT(raw <= sizeof(fStorage)); |
bsalomon@google.com | cadbcb8 | 2012-01-06 19:22:11 +0000 | [diff] [blame] | 175 | size_t count = gCountNibBits[fFlags & 0xF]; |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 176 | SkASSERT(raw == count * 256); |
| 177 | } |
| 178 | |
mike@reedtribe.org | f23f5f7 | 2012-01-07 03:48:45 +0000 | [diff] [blame] | 179 | bool SkTable_ColorFilter::asComponentTable(SkBitmap* table) { |
| 180 | if (table) { |
| 181 | if (NULL == fBitmap) { |
| 182 | fBitmap = new SkBitmap; |
| 183 | fBitmap->setConfig(SkBitmap::kA8_Config, 256, 4, 256); |
| 184 | fBitmap->allocPixels(); |
| 185 | memcpy(fBitmap->getAddr8(0, 0), fStorage, 256 * 4); |
| 186 | } |
| 187 | *table = *fBitmap; |
| 188 | } |
| 189 | return true; |
| 190 | } |
| 191 | |
mike@reedtribe.org | a69b48c | 2011-12-28 20:31:00 +0000 | [diff] [blame] | 192 | /////////////////////////////////////////////////////////////////////////////// |
| 193 | |
| 194 | #ifdef SK_CPU_BENDIAN |
| 195 | #else |
| 196 | #define SK_A32_INDEX (3 - (SK_A32_SHIFT >> 3)) |
| 197 | #define SK_R32_INDEX (3 - (SK_R32_SHIFT >> 3)) |
| 198 | #define SK_G32_INDEX (3 - (SK_G32_SHIFT >> 3)) |
| 199 | #define SK_B32_INDEX (3 - (SK_B32_SHIFT >> 3)) |
| 200 | #endif |
| 201 | |
| 202 | /////////////////////////////////////////////////////////////////////////////// |
| 203 | |
| 204 | SkColorFilter* SkTableColorFilter::Create(const uint8_t table[256]) { |
| 205 | return SkNEW_ARGS(SkTable_ColorFilter, (table, table, table, table)); |
| 206 | } |
| 207 | |
| 208 | SkColorFilter* SkTableColorFilter::CreateARGB(const uint8_t tableA[256], |
| 209 | const uint8_t tableR[256], |
| 210 | const uint8_t tableG[256], |
| 211 | const uint8_t tableB[256]) { |
| 212 | return SkNEW_ARGS(SkTable_ColorFilter, (tableA, tableR, tableG, tableB)); |
| 213 | } |
djsollen@google.com | a2ca41e | 2012-03-23 19:00:34 +0000 | [diff] [blame] | 214 | |
| 215 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_START(SkTableColorFilter) |
| 216 | SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkTable_ColorFilter) |
| 217 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_END |