epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2011 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
bungeman | f3c15b7 | 2015-08-19 11:56:48 -0700 | [diff] [blame] | 7 | |
| 8 | #include "SkBitmapProcShader.h" |
| 9 | #include "SkBitmapProcState.h" |
reed | 013e9e3 | 2015-09-15 14:46:27 -0700 | [diff] [blame] | 10 | #include "SkBitmapProvider.h" |
reed@android.com | 8a1c16f | 2008-12-17 15:59:43 +0000 | [diff] [blame] | 11 | |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 12 | static bool only_scale_and_translate(const SkMatrix& matrix) { |
| 13 | unsigned mask = SkMatrix::kTranslate_Mask | SkMatrix::kScale_Mask; |
| 14 | return (matrix.getType() & ~mask) == 0; |
| 15 | } |
| 16 | |
| 17 | class BitmapProcInfoContext : public SkShader::Context { |
| 18 | public: |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 19 | // The info has been allocated elsewhere, but we are responsible for calling its destructor. |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 20 | BitmapProcInfoContext(const SkShader& shader, const SkShader::ContextRec& rec, |
| 21 | SkBitmapProcInfo* info) |
| 22 | : INHERITED(shader, rec) |
| 23 | , fInfo(info) |
| 24 | { |
| 25 | fFlags = 0; |
| 26 | if (fInfo->fPixmap.isOpaque() && (255 == this->getPaintAlpha())) { |
| 27 | fFlags |= SkShader::kOpaqueAlpha_Flag; |
| 28 | } |
| 29 | |
| 30 | if (1 == fInfo->fPixmap.height() && only_scale_and_translate(this->getTotalInverse())) { |
| 31 | fFlags |= SkShader::kConstInY32_Flag; |
| 32 | } |
| 33 | } |
| 34 | |
| 35 | ~BitmapProcInfoContext() override { |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 36 | fInfo->~SkBitmapProcInfo(); |
| 37 | } |
herb | 6eff52a | 2016-03-23 09:00:33 -0700 | [diff] [blame] | 38 | |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 39 | uint32_t getFlags() const override { return fFlags; } |
| 40 | |
| 41 | private: |
| 42 | SkBitmapProcInfo* fInfo; |
| 43 | uint32_t fFlags; |
| 44 | |
| 45 | typedef SkShader::Context INHERITED; |
| 46 | }; |
| 47 | |
| 48 | /////////////////////////////////////////////////////////////////////////////////////////////////// |
| 49 | |
| 50 | class BitmapProcShaderContext : public BitmapProcInfoContext { |
| 51 | public: |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 52 | BitmapProcShaderContext(const SkShader& shader, const SkShader::ContextRec& rec, |
| 53 | SkBitmapProcState* state) |
| 54 | : INHERITED(shader, rec, state) |
| 55 | , fState(state) |
| 56 | {} |
| 57 | |
| 58 | void shadeSpan(int x, int y, SkPMColor dstC[], int count) override { |
| 59 | const SkBitmapProcState& state = *fState; |
| 60 | if (state.getShaderProc32()) { |
| 61 | state.getShaderProc32()(&state, x, y, dstC, count); |
| 62 | return; |
| 63 | } |
| 64 | |
| 65 | const int BUF_MAX = 128; |
| 66 | uint32_t buffer[BUF_MAX]; |
| 67 | SkBitmapProcState::MatrixProc mproc = state.getMatrixProc(); |
| 68 | SkBitmapProcState::SampleProc32 sproc = state.getSampleProc32(); |
| 69 | const int max = state.maxCountForBufferSize(sizeof(buffer[0]) * BUF_MAX); |
| 70 | |
| 71 | SkASSERT(state.fPixmap.addr()); |
| 72 | |
| 73 | for (;;) { |
| 74 | int n = SkTMin(count, max); |
| 75 | SkASSERT(n > 0 && n < BUF_MAX*2); |
| 76 | mproc(state, buffer, n, x, y); |
| 77 | sproc(state, buffer, n, dstC); |
| 78 | |
| 79 | if ((count -= n) == 0) { |
| 80 | break; |
| 81 | } |
| 82 | SkASSERT(count > 0); |
| 83 | x += n; |
| 84 | dstC += n; |
| 85 | } |
| 86 | } |
| 87 | |
| 88 | ShadeProc asAShadeProc(void** ctx) override { |
| 89 | if (fState->getShaderProc32()) { |
| 90 | *ctx = fState; |
| 91 | return (ShadeProc)fState->getShaderProc32(); |
| 92 | } |
| 93 | return nullptr; |
| 94 | } |
| 95 | |
| 96 | private: |
| 97 | SkBitmapProcState* fState; |
| 98 | |
| 99 | typedef BitmapProcInfoContext INHERITED; |
| 100 | }; |
| 101 | |
| 102 | /////////////////////////////////////////////////////////////////////////////////////////////////// |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 103 | #include "SkLinearBitmapPipeline.h" |
| 104 | #include "SkPM4f.h" |
| 105 | #include "SkXfermode.h" |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 106 | |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 107 | class LinearPipelineContext : public BitmapProcInfoContext { |
| 108 | public: |
| 109 | LinearPipelineContext(const SkShader& shader, const SkShader::ContextRec& rec, |
| 110 | SkBitmapProcInfo* info) |
| 111 | : INHERITED(shader, rec, info) |
| 112 | { |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 113 | // Save things off in case we need to build a blitter pipeline. |
| 114 | fSrcPixmap = info->fPixmap; |
| 115 | fAlpha = SkColorGetA(info->fPaintColor) / 255.0f; |
| 116 | fXMode = info->fTileModeX; |
| 117 | fYMode = info->fTileModeY; |
| 118 | fFilterQuality = info->fFilterQuality; |
| 119 | fMatrixTypeMask = info->fRealInvMatrix.getType(); |
| 120 | |
herb | 57a69dc | 2016-05-19 14:19:23 -0700 | [diff] [blame] | 121 | fShaderPipeline.init( |
| 122 | info->fRealInvMatrix, info->fFilterQuality, |
| 123 | info->fTileModeX, info->fTileModeY, |
| 124 | info->fPaintColor, |
| 125 | info->fPixmap); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 126 | |
| 127 | // To implement the old shadeSpan entry-point, we need to efficiently convert our native |
| 128 | // floats into SkPMColor. The SkXfermode::D32Procs do exactly that. |
| 129 | // |
reed | cfb6bdf | 2016-03-29 11:32:50 -0700 | [diff] [blame] | 130 | sk_sp<SkXfermode> xfer(SkXfermode::Make(SkXfermode::kSrc_Mode)); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 131 | fXferProc = SkXfermode::GetD32Proc(xfer.get(), 0); |
| 132 | } |
| 133 | |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 134 | void shadeSpan4f(int x, int y, SkPM4f dstC[], int count) override { |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 135 | fShaderPipeline->shadeSpan4f(x, y, dstC, count); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | void shadeSpan(int x, int y, SkPMColor dstC[], int count) override { |
| 139 | const int N = 128; |
| 140 | SkPM4f tmp[N]; |
| 141 | |
| 142 | while (count > 0) { |
| 143 | const int n = SkTMin(count, N); |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 144 | fShaderPipeline->shadeSpan4f(x, y, tmp, n); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 145 | fXferProc(nullptr, dstC, tmp, n, nullptr); |
| 146 | dstC += n; |
| 147 | x += n; |
| 148 | count -= n; |
| 149 | } |
| 150 | } |
| 151 | |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 152 | bool onChooseBlitProcs(const SkImageInfo& dstInfo, BlitState* state) override { |
| 153 | SkXfermode::Mode mode; |
| 154 | if (!SkXfermode::AsMode(state->fXfer, &mode)) { return false; } |
| 155 | |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 156 | if (SkLinearBitmapPipeline::ClonePipelineForBlitting( |
herb | 57a69dc | 2016-05-19 14:19:23 -0700 | [diff] [blame] | 157 | &fBlitterPipeline, *fShaderPipeline, |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 158 | fMatrixTypeMask, |
| 159 | fXMode, fYMode, |
| 160 | fFilterQuality, fSrcPixmap, |
herb | 9f666a1 | 2016-04-25 09:05:14 -0700 | [diff] [blame] | 161 | fAlpha, mode, dstInfo)) |
| 162 | { |
herb | 57a69dc | 2016-05-19 14:19:23 -0700 | [diff] [blame] | 163 | state->fStorage[0] = fBlitterPipeline.get(); |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 164 | state->fBlitBW = &LinearPipelineContext::ForwardToPipeline; |
| 165 | |
| 166 | return true; |
| 167 | } |
| 168 | |
herb | 69076fe | 2016-04-12 14:07:59 -0700 | [diff] [blame] | 169 | return false; |
| 170 | } |
| 171 | |
| 172 | static void ForwardToPipeline(BlitState* state, int x, int y, const SkPixmap& dst, int count) { |
| 173 | SkLinearBitmapPipeline* pipeline = static_cast<SkLinearBitmapPipeline*>(state->fStorage[0]); |
| 174 | void* addr = dst.writable_addr32(x, y); |
| 175 | pipeline->blitSpan(x, y, addr, count); |
| 176 | } |
| 177 | |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 178 | private: |
herb | 57a69dc | 2016-05-19 14:19:23 -0700 | [diff] [blame] | 179 | SkEmbeddableLinearPipeline fShaderPipeline; |
| 180 | SkEmbeddableLinearPipeline fBlitterPipeline; |
| 181 | SkXfermode::D32Proc fXferProc; |
| 182 | SkPixmap fSrcPixmap; |
| 183 | float fAlpha; |
| 184 | SkShader::TileMode fXMode; |
| 185 | SkShader::TileMode fYMode; |
| 186 | SkMatrix::TypeMask fMatrixTypeMask; |
| 187 | SkFilterQuality fFilterQuality; |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 188 | |
| 189 | typedef BitmapProcInfoContext INHERITED; |
| 190 | }; |
| 191 | |
| 192 | /////////////////////////////////////////////////////////////////////////////////////////////////// |
| 193 | |
| 194 | static bool choose_linear_pipeline(const SkShader::ContextRec& rec, const SkImageInfo& srcInfo) { |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 195 | // If we get here, we can reasonably use either context, respect the caller's preference |
| 196 | // |
herb | a4e4b79 | 2016-06-04 13:27:10 -0700 | [diff] [blame] | 197 | bool needsPremul = srcInfo.alphaType() == kUnpremul_SkAlphaType; |
| 198 | bool needsSwizzle = srcInfo.bytesPerPixel() == 4 && srcInfo.colorType() != kN32_SkColorType; |
| 199 | return SkShader::ContextRec::kPM4f_DstType == rec.fPreferredDstType |
| 200 | || needsPremul || needsSwizzle; |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 201 | } |
| 202 | |
reed | 320a40d | 2016-08-02 06:12:06 -0700 | [diff] [blame] | 203 | size_t SkBitmapProcLegacyShader::ContextSize(const ContextRec& rec, const SkImageInfo& srcInfo) { |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 204 | size_t size0 = sizeof(BitmapProcShaderContext) + sizeof(SkBitmapProcState); |
| 205 | size_t size1 = sizeof(LinearPipelineContext) + sizeof(SkBitmapProcInfo); |
reed | 320a40d | 2016-08-02 06:12:06 -0700 | [diff] [blame] | 206 | size_t s = SkTMax(size0, size1); |
| 207 | return s; |
reed | 7a4d847 | 2015-09-15 13:33:58 -0700 | [diff] [blame] | 208 | } |
| 209 | |
reed | 320a40d | 2016-08-02 06:12:06 -0700 | [diff] [blame] | 210 | SkShader::Context* SkBitmapProcLegacyShader::MakeContext(const SkShader& shader, |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 211 | TileMode tmx, TileMode tmy, |
| 212 | const SkBitmapProvider& provider, |
| 213 | const ContextRec& rec, void* storage) { |
| 214 | SkMatrix totalInverse; |
| 215 | // Do this first, so we know the matrix can be inverted. |
| 216 | if (!shader.computeTotalInverse(rec, &totalInverse)) { |
| 217 | return nullptr; |
| 218 | } |
| 219 | |
herb | 6eff52a | 2016-03-23 09:00:33 -0700 | [diff] [blame] | 220 | // Decide if we can/want to use the new linear pipeline |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 221 | bool useLinearPipeline = choose_linear_pipeline(rec, provider.info()); |
Brian Osman | 7b8400d | 2016-11-08 17:08:54 -0500 | [diff] [blame^] | 222 | SkDestinationSurfaceColorMode colorMode = SkMipMap::DeduceColorMode(rec); |
reed | cd660e1 | 2016-03-03 09:36:50 -0800 | [diff] [blame] | 223 | |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 224 | if (useLinearPipeline) { |
| 225 | void* infoStorage = (char*)storage + sizeof(LinearPipelineContext); |
Brian Osman | 7b8400d | 2016-11-08 17:08:54 -0500 | [diff] [blame^] | 226 | SkBitmapProcInfo* info = new (infoStorage) SkBitmapProcInfo(provider, tmx, tmy, colorMode); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 227 | if (!info->init(totalInverse, *rec.fPaint)) { |
| 228 | info->~SkBitmapProcInfo(); |
| 229 | return nullptr; |
| 230 | } |
herb | 670f01f | 2016-05-13 10:04:46 -0700 | [diff] [blame] | 231 | |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 232 | return new (storage) LinearPipelineContext(shader, rec, info); |
| 233 | } else { |
| 234 | void* stateStorage = (char*)storage + sizeof(BitmapProcShaderContext); |
reed | 6644d93 | 2016-06-10 11:41:47 -0700 | [diff] [blame] | 235 | SkBitmapProcState* state = new (stateStorage) SkBitmapProcState(provider, tmx, tmy, |
Brian Osman | 7b8400d | 2016-11-08 17:08:54 -0500 | [diff] [blame^] | 236 | colorMode); |
reed | d882901 | 2016-03-04 11:07:43 -0800 | [diff] [blame] | 237 | if (!state->setup(totalInverse, *rec.fPaint)) { |
| 238 | state->~SkBitmapProcState(); |
| 239 | return nullptr; |
| 240 | } |
| 241 | return new (storage) BitmapProcShaderContext(shader, rec, state); |
| 242 | } |
reed | 05a5647 | 2016-03-02 09:49:02 -0800 | [diff] [blame] | 243 | } |