Hook in rough distance field support for fonts

src/gpu/GrTextStrike.cpp

 /*
  * Copyright 2010 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "GrAtlas.h"
 #include "GrGpu.h"
 #include "GrRectanizer.h"
 #include "GrTextStrike.h"
 #include "GrTextStrike_impl.h"
 #include "SkString.h"
 
+#include "edtaa3.h"
+
 SK_DEFINE_INST_COUNT(GrFontScaler)
 SK_DEFINE_INST_COUNT(GrKey)
 
 ///////////////////////////////////////////////////////////////////////////////
 
 #define FONT_CACHE_STATS 0
 #if FONT_CACHE_STATS
 static int g_PurgeCount = 0;
 #endif
 
@@ skipping 161 matching lines @@
     ++gDumpCount;
 }
 #endif
 
 ///////////////////////////////////////////////////////////////////////////////
 
 #ifdef SK_DEBUG
     static int gCounter;
 #endif
 
+#define DISTANCE_FIELD_PAD   4
+#define DISTANCE_FIELD_RANGE (4.0)
+
 /*
     The text strike is specific to a given font/style/matrix setup, which is
     represented by the GrHostFontScaler object we are given in getGlyph().
 
     We map a 32bit glyphID to a GrGlyph record, which in turn points to a
     atlas and a position within that texture.
  */
 
 GrTextStrike::GrTextStrike(GrFontCache* cache, const GrKey* key,
                            GrMaskFormat format,
@@ skipping 33 matching lines @@
 }
 
 GrGlyph* GrTextStrike::generateGlyph(GrGlyph::PackedID packed,
                                      GrFontScaler* scaler) {
     SkIRect bounds;
     if (!scaler->getPackedGlyphBounds(packed, &bounds)) {
         return NULL;
     }
 
     GrGlyph* glyph = fPool.alloc();
+    // expand bounds to hold full distance field data
+    if (fUseDistanceField) {
+        bounds.fLeft   -= DISTANCE_FIELD_PAD;
+        bounds.fRight  += DISTANCE_FIELD_PAD;
+        bounds.fTop    -= DISTANCE_FIELD_PAD;
+        bounds.fBottom += DISTANCE_FIELD_PAD;
+    }
     glyph->init(packed, bounds);
     fCache.insert(packed, glyph);
     return glyph;
 }
 
 bool GrTextStrike::removeUnusedPlots() {
     fCache.getArray().visitAll(invalidate_glyph);
     return fAtlasMgr->removeUnusedPlots(&fAtlas);
 }
 
+
 bool GrTextStrike::getGlyphAtlas(GrGlyph* glyph, GrFontScaler* scaler) {
 #if 0   // testing hack to force us to flush our cache often
     static int gCounter;
     if ((++gCounter % 10) == 0) return false;
 #endif
 
     SkASSERT(glyph);
     SkASSERT(scaler);
     SkASSERT(fCache.contains(glyph));
     SkASSERT(NULL == glyph->fPlot);
 
     SkAutoRef ar(scaler);
 
     int bytesPerPixel = GrMaskFormatBytesPerPixel(fMaskFormat);
-    size_t size = glyph->fBounds.area() * bytesPerPixel;
-    SkAutoSMalloc<1024> storage(size);
-    if (!scaler->getPackedGlyphImage(glyph->fPackedID, glyph->width(),
-                                     glyph->height(),
-                                     glyph->width() * bytesPerPixel,
-                                     storage.get())) {
-        return false;
+
+    GrPlot* plot;
+    if (fUseDistanceField) {
+        SkASSERT(1 == bytesPerPixel);
+
+        // we've already expanded the glyph dimensions to match the final size
+        // but must shrink back down to get the packed glyph data
+        int dfwidth = glyph->width();
+        int dfheight = glyph->height();
+        int width = dfwidth - 2*DISTANCE_FIELD_PAD;
+        int height = dfheight - 2*DISTANCE_FIELD_PAD;
+        size_t stride = width*bytesPerPixel;
+
+        size_t size = width * height * bytesPerPixel;
+        SkAutoSMalloc<1024> storage(size);
+        if (!scaler->getPackedGlyphImage(glyph->fPackedID, width, height, stride, storage.get())) {
+            return false;
+        }
+
+        // alloc storage for distance field glyph
+        size_t dfsize = dfwidth * dfheight * bytesPerPixel;
+        SkAutoSMalloc<1024> dfstorage(dfsize);
+
+        // copy glyph into distance field storage
+        memset(dfstorage.get(), 0, dfsize);
+
+        unsigned char* ptr = (unsigned char*) storage.get();
+        unsigned char* dfptr = (unsigned char*) dfstorage.get();
+        size_t dfstride = dfwidth*bytesPerPixel;
+        dfptr += DISTANCE_FIELD_PAD*dfstride;
+        dfptr += DISTANCE_FIELD_PAD*bytesPerPixel;
+
+        for (int i = 0; i < height; ++i) {
+            memcpy(dfptr, ptr, stride);
+
+            dfptr += dfstride;
+            ptr += stride;
+        }
+
+        // generate distance field data
+        short* distx = new short[dfwidth*dfheight];
+        short* disty = new short[dfwidth*dfheight];
+        double* outerdist = new double[dfwidth*dfheight];
+        double* gx = new double[dfwidth*dfheight];
+        double* gy = new double[dfwidth*dfheight];
+        dfptr = (unsigned char*) dfstorage.get();
+        computegradient(dfptr, dfwidth, dfheight, gx, gy);
+        edtaa3(dfptr, gx, gy, dfwidth, dfheight, distx, disty, outerdist);
+
+        for (int i = 0; i < dfwidth*dfheight; ++i) {
+            *dfptr++ = 255 - *dfptr;
+        }
+        double* innerdist = new double[dfwidth*dfheight];
+        dfptr = (unsigned char*) dfstorage.get();
+        memset(gx, 0, sizeof(double)*dfwidth*dfheight);
+        memset(gy, 0, sizeof(double)*dfwidth*dfheight);
+        computegradient(dfptr, dfwidth, dfheight, gx, gy);
+        edtaa3(dfptr, gx, gy, dfwidth, dfheight, distx, disty, innerdist);
+
+        for (int i = 0; i < dfwidth*dfheight; ++i) {
+            unsigned char val;
+            double outerval = outerdist[i];
+            if (outerval < 0.0) { outerval = 0.0; }
+            double innerval = innerdist[i];
+            if (innerval < 0.0) { innerval = 0.0; }
+            double dist = outerval - innerval;
+            if (dist <= -DISTANCE_FIELD_RANGE) {
+                val = 255;
+            } else if (dist > DISTANCE_FIELD_RANGE) {
+                val = 0;
+            } else {
+                val = (unsigned char)((DISTANCE_FIELD_RANGE-dist)*128.0/DISTANCE_FIELD_RANGE);
+            }
+            *dfptr++ = val;
+        }
+
+        delete [] distx;
+        delete [] disty;
+        delete [] innerdist;
+        delete [] outerdist;
+        
+        // copy to atlas
+        plot = fAtlasMgr->addToAtlas(&fAtlas, dfwidth, dfheight, dfstorage.get(), 
+                                             &glyph->fAtlasLocation);
+
+    } else {
+        size_t size = glyph->fBounds.area() * bytesPerPixel;
+        SkAutoSMalloc<1024> storage(size);
+        if (!scaler->getPackedGlyphImage(glyph->fPackedID, glyph->width(),
+                                         glyph->height(),
+                                         glyph->width() * bytesPerPixel,
+                                         storage.get())) {
+            return false;
+        }
+
+        plot = fAtlasMgr->addToAtlas(&fAtlas, glyph->width(), glyph->height(), storage.get(), 
+                                             &glyph->fAtlasLocation);
     }
 
-    GrPlot* plot = fAtlasMgr->addToAtlas(&fAtlas, glyph->width(),
-                                         glyph->height(), storage.get(),
-                                         &glyph->fAtlasLocation);
     if (NULL == plot) {
         return false;
     }
 
     glyph->fPlot = plot;
     return true;
 }
+