Make DCShader serializable.

gm/dcshader.cpp

 
 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "gm.h"
 #if SK_SUPPORT_GPU
 #include "GrFragmentProcessor.h"
 #include "GrCoordTransform.h"
 #include "gl/GrGLProcessor.h"
 #include "gl/builders/GrGLProgramBuilder.h"
 #include "Resources.h"
+#include "SkReadBuffer.h"
 #include "SkShader.h"
 #include "SkStream.h"
 #include "SkTypeface.h"
+#include "SkWriteBuffer.h"
 
 namespace skiagm {
 
 ///////////////////////////////////////////////////////////////////////////////
 
 class DCShader : public SkShader {
 public:
     DCShader(const SkMatrix& matrix) : fDeviceMatrix(matrix) {}
 
-    // This is a custom shader, so we don't need to make it
-    // flattenable.  Since this class is not part of the skia library,
-    // it wouldn't deserialize without linking this library anyway.
-    SK_DECLARE_NOT_FLATTENABLE_PROCS(DCShader)
+    SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(DCShader);
+
+    void flatten(SkWriteBuffer& buf) const SK_OVERRIDE {
+        buf.writeMatrix(fDeviceMatrix);
+    }
 
     bool asFragmentProcessor(GrContext*, const SkPaint& paint, const SkMatrix& viewM,
                              const SkMatrix* localMatrix, GrColor* color,
                              GrFragmentProcessor** fp) const SK_OVERRIDE;
 private:
     const SkMatrix fDeviceMatrix;
 };
 
+SkFlattenable* DCShader::CreateProc(SkReadBuffer& buf) {
+    SkMatrix matrix;
+    buf.readMatrix(&matrix);
+    return SkNEW_ARGS(DCShader, (matrix));
+}
+
 class DCFP : public GrFragmentProcessor {
 public:
     DCFP(const SkMatrix& m) : fDeviceTransform(kDevice_GrCoordSet, m) {
         this->addCoordTransform(&fDeviceTransform);
         this->initClassID<DCFP>();
     }
 
     void getGLProcessorKey(const GrGLCaps& caps,
                             GrProcessorKeyBuilder* b) const SK_OVERRIDE {}
 
@@ skipping 242 matching lines @@
     };
 
     SkTArray<Prim*> fPrims;
 
     typedef GM INHERITED;
 };
 
 DEF_GM( return SkNEW(DCShaderGM); )
 }
 #endif