Move DM png code to picture_utils, for use by other tools.

dm/DM.cpp

 /*
  * Copyright 2013 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "CrashHandler.h"
 #include "DMJsonWriter.h"
 #include "DMSrcSink.h"
 #include "DMSrcSinkAndroid.h"
 #include "ProcStats.h"
 #include "Resources.h"
 #include "SkBBHFactory.h"
 #include "SkChecksum.h"
 #include "SkCodec.h"
 #include "SkColorPriv.h"
 #include "SkCommonFlags.h"
 #include "SkCommonFlagsConfig.h"
+#include "SkData.h"
 #include "SkFontMgr.h"
 #include "SkGraphics.h"
 #include "SkHalf.h"
 #include "SkMD5.h"
 #include "SkMutex.h"
 #include "SkOSFile.h"
 #include "SkPM4fPriv.h"
 #include "SkSpinlock.h"
 #include "SkTHash.h"
 #include "SkTaskGroup.h"
 #include "SkThreadUtils.h"
 #include "Test.h"
 #include "Timer.h"
+#include "picture_utils.h"
 #include "sk_tool_utils.h"
 
 #ifdef SK_PDF_IMAGE_STATS
 extern void SkPDFImageDumpStats();
 #endif
 
 #include "png.h"
 
 #include <stdlib.h>
 
@@ skipping 874 matching lines @@
         }
         if (sink) {
             push_sink(config, sink);
         }
     }
 }
 
 static bool dump_png(SkBitmap bitmap, const char* path, const char* md5) {
     const int w = bitmap.width(),
               h = bitmap.height();
-    // PNG wants unpremultiplied 8-bit RGBA pixels (16-bit could work fine too).
-    // We leave the gamma of these bytes unspecified, to continue the status quo,
-    // which we think generally is to interpret them as sRGB.
 
-    SkAutoTMalloc<uint32_t> rgba(w*h);
-
-    if (bitmap.  colorType() ==  kN32_SkColorType &&
-        bitmap.profileType() == kSRGB_SkColorProfileType) {
-        // These are premul sRGB 8-bit pixels in SkPMColor order.
-        // We want unpremul sRGB 8-bit pixels in RGBA order.  We'll get there via floats.
-        bitmap.lockPixels();
-        auto px = (const uint32_t*)bitmap.getPixels();
-        if (!px) {
-            return false;
-        }
-        for (int i = 0; i < w*h; i++) {
-            Sk4f fs = Sk4f_fromS32(px[i]);         // Convert up to linear floats.
-        #if defined(SK_PMCOLOR_IS_BGRA)
-            fs = SkNx_shuffle<2,1,0,3>(fs);        // Shuffle to RGBA, if not there already.
-        #endif
-            float invA = 1.0f / fs[3];
-            fs = fs * Sk4f(invA, invA, invA, 1);   // Unpremultiply.
-            rgba[i] = Sk4f_toS32(fs);              // Pack down to sRGB bytes.
-        }
-
-    } else if (bitmap.colorType() == kRGBA_F16_SkColorType) {
-        // These are premul linear half-float pixels in RGBA order.
-        // We want unpremul sRGB 8-bit pixels in RGBA order.  We'll get there via floats.
-        bitmap.lockPixels();
-        auto px = (const uint64_t*)bitmap.getPixels();
-        if (!px) {
-            return false;
-        }
-        for (int i = 0; i < w*h; i++) {
-            // Convert up to linear floats.
-            Sk4f fs(SkHalfToFloat(static_cast<SkHalf>(px[i] >> (0 * 16))),
-                    SkHalfToFloat(static_cast<SkHalf>(px[i] >> (1 * 16))),
-                    SkHalfToFloat(static_cast<SkHalf>(px[i] >> (2 * 16))),
-                    SkHalfToFloat(static_cast<SkHalf>(px[i] >> (3 * 16))));
-            fs = Sk4f::Max(0.0f, Sk4f::Min(fs, 1.0f));  // Clamp
-            float invA = 1.0f / fs[3];
-            fs = fs * Sk4f(invA, invA, invA, 1);  // Unpremultiply.
-            rgba[i] = Sk4f_toS32(fs);             // Pack down to sRGB bytes.
-        }
-
-
-    } else {
-        // We "should" gamma correct in here but we don't.
-        // We want Gold to show exactly what our clients are seeing, broken gamma.
-
-        // Convert smaller formats up to premul linear 8-bit (in SkPMColor order).
-        if (bitmap.colorType() != kN32_SkColorType) {
-            SkBitmap n32;
-            if (!bitmap.copyTo(&n32, kN32_SkColorType)) {
-                return false;
-            }
-            bitmap = n32;
-        }
-
-        // Convert premul linear 8-bit to unpremul linear 8-bit RGBA.
-        if (!bitmap.readPixels(SkImageInfo::Make(w,h, kRGBA_8888_SkColorType,
-                                                      kUnpremul_SkAlphaType),
-                               rgba, 4*w, 0,0)) {
-            return false;
-        }
+    sk_sp<SkData> encodedBitmap = sk_tools::encode_bitmap_for_png(bitmap);
+    if (encodedBitmap.get() == nullptr) {
+        return false;
     }
+    uint32_t* rgba = static_cast<uint32_t*>(encodedBitmap.get()->writable_data());
 
     // We don't need bitmap anymore.  Might as well drop our ref.
     bitmap.reset();
 
     FILE* f = fopen(path, "wb");
     if (!f) { return false; }
 
     png_structp png = png_create_write_struct(PNG_LIBPNG_VER_STRING, nullptr, nullptr, nullptr);
     if (!png) {
         fclose(f);
@@ skipping 26 matching lines @@
     text[1].text = (png_charp)description.c_str();
     text[1].compression = PNG_TEXT_COMPRESSION_NONE;
     png_set_text(png, info, text, 2);
 
     png_init_io(png, f);
     png_set_IHDR(png, info, (png_uint_32)w, (png_uint_32)h, 8,
                  PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE,
                  PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
     png_write_info(png, info);
     for (int j = 0; j < h; j++) {
-        png_bytep row = (png_bytep)(rgba.get() + w*j);
+        png_bytep row = (png_bytep)(rgba + w*j);
         png_write_rows(png, &row, 1);
     }
     png_write_end(png, info);
 
     png_destroy_write_struct(&png, &info);
     fclose(f);
     return true;
 }
 
 static bool match(const char* needle, const char* haystack) {
@@ skipping 419 matching lines @@
 #endif
 }
 } // namespace skiatest
 
 #if !defined(SK_BUILD_FOR_IOS)
 int main(int argc, char** argv) {
     SkCommandLineFlags::Parse(argc, argv);
     return dm_main();
 }
 #endif