Refugee from Dead Machine 11: Add SkCanvas::drawLitAtlas call

src/core/SkDevice.cpp

 /*
  * Copyright 2011 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkColorFilter.h"
 #include "SkDevice.h"
 #include "SkDraw.h"
 #include "SkDrawFilter.h"
 #include "SkImage_Base.h"
+#include "SkLightingShader.h"
 #include "SkMetaData.h"
 #include "SkNinePatchIter.h"
 #include "SkPatchUtils.h"
 #include "SkPathMeasure.h"
 #include "SkRasterClip.h"
 #include "SkRSXform.h"
 #include "SkShader.h"
 #include "SkTextBlob.h"
 #include "SkTextToPathIter.h"
 
@@ skipping 182 matching lines @@
             pnt.setColorFilter(cf);
         }
         
         path.rewind();
         path.addPoly(quad, 4, true);
         path.setConvexity(SkPath::kConvex_Convexity);
         this->drawPath(draw, path, pnt, NULL, true);
     }
 }
 
+#include "SkGr.h"
+
+static bool wrap_as_bm(const SkImage* image, SkBitmap* bm) {
+    GrTexture* tex = as_IB(image)->getTexture();
+    if (tex) {
+        GrWrapTextureInBitmap(tex, image->width(), image->height(), image->isOpaque(), bm);
+        return true;
+    } else {
+        return as_IB(image)->getROPixels(bm);
+    }
+}
+
+void SkBaseDevice::drawLitAtlas(const SkDraw& draw, const SkImage* atlas, const SkRSXform xform[],
+                                const SkRect diffTex[], const SkRect normTex[],
+                                const SkColor colors[], int count, SkXfermode::Mode mode,
+                                const SkPaint& paint,
+                                const SkLight lights[], int numLights) {
+    SkPath path;
+    path.setIsVolatile(true);
+
+    SkBitmap bm;
+    if (!wrap_as_bm(atlas, &bm)) {
+        return;
+    }
+
+    SkLightingShader::Lights::Builder builder(lights, numLights);
+    SkAutoTUnref<const SkLightingShader::Lights> lights2(builder.finish());
+
+    for (int i = 0; i < count; ++i) {
+        SkASSERT(diffTex[i].width() == normTex[i].width() && 
+                 diffTex[i].height() == normTex[i].height());
+
+        SkPoint quad[4];
+        xform[i].toQuad(diffTex[i].width(), diffTex[i].height(), quad);
+        
+        SkMatrix diffLocalM;
+        diffLocalM.setRSXform(xform[i]);
+        diffLocalM.preTranslate(-diffTex[i].left(), -diffTex[i].top());
+
+        SkMatrix normLocalM;
+        normLocalM.setRSXform(xform[i]);
+        normLocalM.preTranslate(diffTex[i].left()-normTex[i].left(),
+                                diffTex[i].top()-normTex[i].top());
+        normLocalM.preTranslate(-diffTex[i].left(), -diffTex[i].top());
+
+        SkPaint pnt(paint);
+
+        pnt.setShader(SkLightingShader::Create(bm, bm,
+                                               lights2,
+                                               xform[i], &diffLocalM, &normLocalM));
+
+        if (colors && colors[i] != SK_ColorWHITE) {
+            SkAutoTUnref<SkColorFilter> cf(SkColorFilter::CreateModeFilter(colors[i], mode));
+            pnt.setColorFilter(cf);
+        }
+        
+        path.rewind();
+        path.addPoly(quad, 4, true);
+        path.setConvexity(SkPath::kConvex_Convexity);
+        this->drawPath(draw, path, pnt, NULL, true);
+    }
+}
+
 ///////////////////////////////////////////////////////////////////////////////////////////////////
 
 bool SkBaseDevice::readPixels(const SkImageInfo& info, void* dstP, size_t rowBytes, int x, int y) {
 #ifdef SK_DEBUG
     SkASSERT(info.width() > 0 && info.height() > 0);
     SkASSERT(dstP);
     SkASSERT(rowBytes >= info.minRowBytes());
     SkASSERT(x >= 0 && y >= 0);
 
     const SkImageInfo& srcInfo = this->imageInfo();
@@ skipping 183 matching lines @@
     if (kUnknown_SkPixelGeometry == fSurfaceProps.pixelGeometry()
         || this->onShouldDisableLCD(paint)) {
 
         flags &= ~SkPaint::kLCDRenderText_Flag;
         flags |= SkPaint::kGenA8FromLCD_Flag;
     }
 
     return flags;
 }