Home | History | Annotate | Download | only in core
      1 
      2 /*
      3  * Copyright 2006 The Android Open Source Project
      4  *
      5  * Use of this source code is governed by a BSD-style license that can be
      6  * found in the LICENSE file.
      7  */
      8 
      9 
     10 #ifndef SkBitmapProcShader_DEFINED
     11 #define SkBitmapProcShader_DEFINED
     12 
     13 #include "SkShader.h"
     14 #include "SkSmallAllocator.h"
     15 
     16 struct SkBitmapProcState;
     17 class SkBitmapProvider;
     18 
     19 class SkBitmapProcShader : public SkShader {
     20 public:
     21     SkBitmapProcShader(const SkBitmap& src, TileMode tx, TileMode ty,
     22                        const SkMatrix* localMatrix = nullptr);
     23 
     24     bool isOpaque() const override;
     25 
     26     size_t contextSize(const ContextRec&) const override { return ContextSize(); }
     27 
     28     SK_TO_STRING_OVERRIDE()
     29     SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkBitmapProcShader)
     30 
     31 #if SK_SUPPORT_GPU
     32     const GrFragmentProcessor* asFragmentProcessor(GrContext*, const SkMatrix& viewM,
     33                                                    const SkMatrix*, SkFilterQuality) const override;
     34 #endif
     35 
     36 protected:
     37     class BitmapProcShaderContext : public SkShader::Context {
     38     public:
     39         // The context takes ownership of the state. It will call its destructor
     40         // but will NOT free the memory.
     41         BitmapProcShaderContext(const SkShader&, const ContextRec&, SkBitmapProcState*);
     42         ~BitmapProcShaderContext() override;
     43 
     44         void shadeSpan(int x, int y, SkPMColor dstC[], int count) override;
     45         ShadeProc asAShadeProc(void** ctx) override;
     46 
     47         uint32_t getFlags() const override { return fFlags; }
     48 
     49     private:
     50         SkBitmapProcState*  fState;
     51         uint32_t            fFlags;
     52 
     53         typedef SkShader::Context INHERITED;
     54     };
     55 
     56     void flatten(SkWriteBuffer&) const override;
     57     Context* onCreateContext(const ContextRec&, void* storage) const override;
     58     bool onIsABitmap(SkBitmap*, SkMatrix*, TileMode*) const override;
     59 
     60     SkBitmap    fRawBitmap;   // experimental for RLE encoding
     61     uint8_t     fTileModeX, fTileModeY;
     62 
     63 private:
     64     friend class SkImageShader;
     65 
     66     static size_t ContextSize();
     67     static Context* MakeContext(const SkShader&, TileMode tmx, TileMode tmy,
     68                                 const SkBitmapProvider&, const ContextRec&, void* storage);
     69 
     70     typedef SkShader INHERITED;
     71 };
     72 
     73 // Commonly used allocator. It currently is only used to allocate up to 3 objects. The total
     74 // bytes requested is calculated using one of our large shaders, its context size plus the size of
     75 // an Sk3DBlitter in SkDraw.cpp
     76 // Note that some contexts may contain other contexts (e.g. for compose shaders), but we've not
     77 // yet found a situation where the size below isn't big enough.
     78 typedef SkSmallAllocator<3, 1500> SkTBlitterAllocator;
     79 
     80 // If alloc is non-nullptr, it will be used to allocate the returned SkShader, and MUST outlive
     81 // the SkShader.
     82 SkShader* SkCreateBitmapShader(const SkBitmap& src, SkShader::TileMode, SkShader::TileMode,
     83                                const SkMatrix* localMatrix, SkTBlitterAllocator* alloc);
     84 
     85 #endif
     86