skia2/tools/ToolUtils.cpp
Mike Klein f7eb0544a8 basic, untested BGR 1010102 and 101010x
Updated every switch that yelled at me, and added support to dm and fm,
and then founds some more switches that shouldn't have defaults...

The tricky spots outside those were mips and dither,
since they aren't simply exhaustive switches.

_Now_ no diffs between RGB/BGR 1010102 and 101010x.

No GPU support.

Bug: skia:9893
Change-Id: I73ab3fd22bdef0519296dfe4cb84031e23ca0be3
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/270114
Commit-Queue: Mike Klein <mtklein@google.com>
Reviewed-by: Brian Salomon <bsalomon@google.com>
2020-02-11 21:44:57 +00:00

563 lines
20 KiB
C++

/*
* 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 "include/core/SkBitmap.h"
#include "include/core/SkBlendMode.h"
#include "include/core/SkCanvas.h"
#include "include/core/SkColorPriv.h"
#include "include/core/SkImage.h"
#include "include/core/SkMatrix.h"
#include "include/core/SkPaint.h"
#include "include/core/SkPath.h"
#include "include/core/SkPixelRef.h"
#include "include/core/SkPixmap.h"
#include "include/core/SkPoint3.h"
#include "include/core/SkRRect.h"
#include "include/core/SkShader.h"
#include "include/core/SkSurface.h"
#include "include/core/SkTextBlob.h"
#include "include/ports/SkTypeface_win.h"
#include "include/private/SkColorData.h"
#include "include/private/SkFloatingPoint.h"
#include "src/core/SkFontMgrPriv.h"
#include "src/core/SkFontPriv.h"
#include "tools/ToolUtils.h"
#include "tools/flags/CommandLineFlags.h"
#include "tools/fonts/TestFontMgr.h"
#include <cmath>
#include <cstring>
#include <memory>
namespace ToolUtils {
const char* alphatype_name(SkAlphaType at) {
switch (at) {
case kUnknown_SkAlphaType: return "Unknown";
case kOpaque_SkAlphaType: return "Opaque";
case kPremul_SkAlphaType: return "Premul";
case kUnpremul_SkAlphaType: return "Unpremul";
}
SkASSERT(false);
return "unexpected alphatype";
}
const char* colortype_name(SkColorType ct) {
switch (ct) {
case kUnknown_SkColorType: return "Unknown";
case kAlpha_8_SkColorType: return "Alpha_8";
case kA16_unorm_SkColorType: return "Alpha_16";
case kA16_float_SkColorType: return "A16_float";
case kRGB_565_SkColorType: return "RGB_565";
case kARGB_4444_SkColorType: return "ARGB_4444";
case kRGBA_8888_SkColorType: return "RGBA_8888";
case kRGB_888x_SkColorType: return "RGB_888x";
case kBGRA_8888_SkColorType: return "BGRA_8888";
case kRGBA_1010102_SkColorType: return "RGBA_1010102";
case kRGB_101010x_SkColorType: return "RGB_101010x";
case kBGRA_1010102_SkColorType: return "BGRA_1010102";
case kBGR_101010x_SkColorType: return "BGR_101010x";
case kGray_8_SkColorType: return "Gray_8";
case kRGBA_F16Norm_SkColorType: return "RGBA_F16Norm";
case kRGBA_F16_SkColorType: return "RGBA_F16";
case kRGBA_F32_SkColorType: return "RGBA_F32";
case kR8G8_unorm_SkColorType: return "R8G8_unorm";
case kR16G16_unorm_SkColorType: return "R16G16_unorm";
case kR16G16_float_SkColorType: return "R16G16_float";
case kR16G16B16A16_unorm_SkColorType: return "R16G16B16A16_unorm";
}
SkASSERT(false);
return "unexpected colortype";
}
const char* colortype_depth(SkColorType ct) {
switch (ct) {
case kUnknown_SkColorType: return "Unknown";
case kAlpha_8_SkColorType: return "A8";
case kA16_unorm_SkColorType: return "A16";
case kA16_float_SkColorType: return "AF16";
case kRGB_565_SkColorType: return "565";
case kARGB_4444_SkColorType: return "4444";
case kRGBA_8888_SkColorType: return "8888";
case kRGB_888x_SkColorType: return "888";
case kBGRA_8888_SkColorType: return "8888";
case kRGBA_1010102_SkColorType: return "1010102";
case kRGB_101010x_SkColorType: return "101010";
case kBGRA_1010102_SkColorType: return "1010102";
case kBGR_101010x_SkColorType: return "101010";
case kGray_8_SkColorType: return "G8";
case kRGBA_F16Norm_SkColorType: return "F16Norm"; // TODO: "F16"?
case kRGBA_F16_SkColorType: return "F16";
case kRGBA_F32_SkColorType: return "F32";
case kR8G8_unorm_SkColorType: return "88";
case kR16G16_unorm_SkColorType: return "1616";
case kR16G16_float_SkColorType: return "F16F16";
case kR16G16B16A16_unorm_SkColorType: return "16161616";
}
SkASSERT(false);
return "unexpected colortype";
}
const char* tilemode_name(SkTileMode mode) {
switch (mode) {
case SkTileMode::kClamp: return "clamp";
case SkTileMode::kRepeat: return "repeat";
case SkTileMode::kMirror: return "mirror";
case SkTileMode::kDecal: return "decal";
}
SkASSERT(false);
return "unexpected tilemode";
}
SkColor color_to_565(SkColor color) {
// Not a good idea to use this function for greyscale colors...
// it will add an obvious purple or green tint.
SkASSERT(SkColorGetR(color) != SkColorGetG(color) || SkColorGetR(color) != SkColorGetB(color) ||
SkColorGetG(color) != SkColorGetB(color));
SkPMColor pmColor = SkPreMultiplyColor(color);
U16CPU color16 = SkPixel32ToPixel16(pmColor);
return SkPixel16ToColor(color16);
}
sk_sp<SkShader> create_checkerboard_shader(SkColor c1, SkColor c2, int size) {
SkBitmap bm;
bm.allocPixels(SkImageInfo::MakeS32(2 * size, 2 * size, kPremul_SkAlphaType));
bm.eraseColor(c1);
bm.eraseArea(SkIRect::MakeLTRB(0, 0, size, size), c2);
bm.eraseArea(SkIRect::MakeLTRB(size, size, 2 * size, 2 * size), c2);
return bm.makeShader(SkTileMode::kRepeat, SkTileMode::kRepeat);
}
SkBitmap create_checkerboard_bitmap(int w, int h, SkColor c1, SkColor c2, int checkSize) {
SkBitmap bitmap;
bitmap.allocPixels(SkImageInfo::MakeS32(w, h, kPremul_SkAlphaType));
SkCanvas canvas(bitmap);
ToolUtils::draw_checkerboard(&canvas, c1, c2, checkSize);
return bitmap;
}
void draw_checkerboard(SkCanvas* canvas, SkColor c1, SkColor c2, int size) {
SkPaint paint;
paint.setShader(create_checkerboard_shader(c1, c2, size));
paint.setBlendMode(SkBlendMode::kSrc);
canvas->drawPaint(paint);
}
SkBitmap
create_string_bitmap(int w, int h, SkColor c, int x, int y, int textSize, const char* str) {
SkBitmap bitmap;
bitmap.allocN32Pixels(w, h);
SkCanvas canvas(bitmap);
SkPaint paint;
paint.setColor(c);
SkFont font(ToolUtils::create_portable_typeface(), textSize);
canvas.clear(0x00000000);
canvas.drawSimpleText(str,
strlen(str),
SkTextEncoding::kUTF8,
SkIntToScalar(x),
SkIntToScalar(y),
font,
paint);
// Tag data as sRGB (without doing any color space conversion). Color-space aware configs
// will process this correctly but legacy configs will render as if this returned N32.
SkBitmap result;
result.setInfo(SkImageInfo::MakeS32(w, h, kPremul_SkAlphaType));
result.setPixelRef(sk_ref_sp(bitmap.pixelRef()), 0, 0);
return result;
}
void add_to_text_blob_w_len(SkTextBlobBuilder* builder,
const char* text,
size_t len,
SkTextEncoding encoding,
const SkFont& font,
SkScalar x,
SkScalar y) {
int count = font.countText(text, len, encoding);
if (count < 1) {
return;
}
auto run = builder->allocRun(font, count, x, y);
font.textToGlyphs(text, len, encoding, run.glyphs, count);
}
void add_to_text_blob(SkTextBlobBuilder* builder,
const char* text,
const SkFont& font,
SkScalar x,
SkScalar y) {
add_to_text_blob_w_len(builder, text, strlen(text), SkTextEncoding::kUTF8, font, x, y);
}
void get_text_path(const SkFont& font,
const void* text,
size_t length,
SkTextEncoding encoding,
SkPath* dst,
const SkPoint pos[]) {
SkAutoToGlyphs atg(font, text, length, encoding);
const int count = atg.count();
SkAutoTArray<SkPoint> computedPos;
if (pos == nullptr) {
computedPos.reset(count);
font.getPos(atg.glyphs(), count, &computedPos[0]);
pos = computedPos.get();
}
struct Rec {
SkPath* fDst;
const SkPoint* fPos;
} rec = {dst, pos};
font.getPaths(atg.glyphs(),
atg.count(),
[](const SkPath* src, const SkMatrix& mx, void* ctx) {
Rec* rec = (Rec*)ctx;
if (src) {
SkMatrix tmp(mx);
tmp.postTranslate(rec->fPos->fX, rec->fPos->fY);
rec->fDst->addPath(*src, tmp);
}
rec->fPos += 1;
},
&rec);
}
SkPath make_star(const SkRect& bounds, int numPts, int step) {
SkASSERT(numPts != step);
SkPath path;
path.setFillType(SkPathFillType::kEvenOdd);
path.moveTo(0, -1);
for (int i = 1; i < numPts; ++i) {
int idx = i * step % numPts;
SkScalar theta = idx * 2 * SK_ScalarPI / numPts + SK_ScalarPI / 2;
SkScalar x = SkScalarCos(theta);
SkScalar y = -SkScalarSin(theta);
path.lineTo(x, y);
}
path.transform(SkMatrix::MakeRectToRect(path.getBounds(), bounds, SkMatrix::kFill_ScaleToFit));
return path;
}
static inline void norm_to_rgb(SkBitmap* bm, int x, int y, const SkVector3& norm) {
SkASSERT(SkScalarNearlyEqual(norm.length(), 1.0f));
unsigned char r = static_cast<unsigned char>((0.5f * norm.fX + 0.5f) * 255);
unsigned char g = static_cast<unsigned char>((-0.5f * norm.fY + 0.5f) * 255);
unsigned char b = static_cast<unsigned char>((0.5f * norm.fZ + 0.5f) * 255);
*bm->getAddr32(x, y) = SkPackARGB32(0xFF, r, g, b);
}
void create_hemi_normal_map(SkBitmap* bm, const SkIRect& dst) {
const SkPoint center =
SkPoint::Make(dst.fLeft + (dst.width() / 2.0f), dst.fTop + (dst.height() / 2.0f));
const SkPoint halfSize = SkPoint::Make(dst.width() / 2.0f, dst.height() / 2.0f);
SkVector3 norm;
for (int y = dst.fTop; y < dst.fBottom; ++y) {
for (int x = dst.fLeft; x < dst.fRight; ++x) {
norm.fX = (x + 0.5f - center.fX) / halfSize.fX;
norm.fY = (y + 0.5f - center.fY) / halfSize.fY;
SkScalar tmp = norm.fX * norm.fX + norm.fY * norm.fY;
if (tmp >= 1.0f) {
norm.set(0.0f, 0.0f, 1.0f);
} else {
norm.fZ = sqrtf(1.0f - tmp);
}
norm_to_rgb(bm, x, y, norm);
}
}
}
void create_frustum_normal_map(SkBitmap* bm, const SkIRect& dst) {
const SkPoint center =
SkPoint::Make(dst.fLeft + (dst.width() / 2.0f), dst.fTop + (dst.height() / 2.0f));
SkIRect inner = dst;
inner.inset(dst.width() / 4, dst.height() / 4);
SkPoint3 norm;
const SkPoint3 left = SkPoint3::Make(-SK_ScalarRoot2Over2, 0.0f, SK_ScalarRoot2Over2);
const SkPoint3 up = SkPoint3::Make(0.0f, -SK_ScalarRoot2Over2, SK_ScalarRoot2Over2);
const SkPoint3 right = SkPoint3::Make(SK_ScalarRoot2Over2, 0.0f, SK_ScalarRoot2Over2);
const SkPoint3 down = SkPoint3::Make(0.0f, SK_ScalarRoot2Over2, SK_ScalarRoot2Over2);
for (int y = dst.fTop; y < dst.fBottom; ++y) {
for (int x = dst.fLeft; x < dst.fRight; ++x) {
if (inner.contains(x, y)) {
norm.set(0.0f, 0.0f, 1.0f);
} else {
SkScalar locX = x + 0.5f - center.fX;
SkScalar locY = y + 0.5f - center.fY;
if (locX >= 0.0f) {
if (locY > 0.0f) {
norm = locX >= locY ? right : down; // LR corner
} else {
norm = locX > -locY ? right : up; // UR corner
}
} else {
if (locY > 0.0f) {
norm = -locX > locY ? left : down; // LL corner
} else {
norm = locX > locY ? up : left; // UL corner
}
}
}
norm_to_rgb(bm, x, y, norm);
}
}
}
void create_tetra_normal_map(SkBitmap* bm, const SkIRect& dst) {
const SkPoint center =
SkPoint::Make(dst.fLeft + (dst.width() / 2.0f), dst.fTop + (dst.height() / 2.0f));
static const SkScalar k1OverRoot3 = 0.5773502692f;
SkPoint3 norm;
const SkPoint3 leftUp = SkPoint3::Make(-k1OverRoot3, -k1OverRoot3, k1OverRoot3);
const SkPoint3 rightUp = SkPoint3::Make(k1OverRoot3, -k1OverRoot3, k1OverRoot3);
const SkPoint3 down = SkPoint3::Make(0.0f, SK_ScalarRoot2Over2, SK_ScalarRoot2Over2);
for (int y = dst.fTop; y < dst.fBottom; ++y) {
for (int x = dst.fLeft; x < dst.fRight; ++x) {
SkScalar locX = x + 0.5f - center.fX;
SkScalar locY = y + 0.5f - center.fY;
if (locX >= 0.0f) {
if (locY > 0.0f) {
norm = locX >= locY ? rightUp : down; // LR corner
} else {
norm = rightUp;
}
} else {
if (locY > 0.0f) {
norm = -locX > locY ? leftUp : down; // LL corner
} else {
norm = leftUp;
}
}
norm_to_rgb(bm, x, y, norm);
}
}
}
#if !defined(__clang__) && defined(_MSC_VER)
// MSVC takes ~2 minutes to compile this function with optimization.
// We don't really care to wait that long for this function.
#pragma optimize("", off)
#endif
void make_big_path(SkPath& path) {
#include "BigPathBench.inc" // IWYU pragma: keep
}
void set_path_pt(int index, const SkPoint& pt, SkPath* path) {
SkPath result;
SkPoint pts[4];
SkPath::Verb verb;
SkPath::RawIter iter(*path);
int startIndex = 0;
int endIndex = 0;
while ((verb = iter.next(pts)) != SkPath::kDone_Verb) {
switch (verb) {
case SkPath::kMove_Verb:
endIndex += 1;
break;
case SkPath::kLine_Verb:
endIndex += 1;
break;
case SkPath::kQuad_Verb:
case SkPath::kConic_Verb:
endIndex += 2;
break;
case SkPath::kCubic_Verb:
endIndex += 3;
break;
case SkPath::kClose_Verb:
break;
case SkPath::kDone_Verb:
break;
default:
SkASSERT(0);
}
if (startIndex <= index && index < endIndex) {
pts[index - startIndex] = pt;
}
switch (verb) {
case SkPath::kMove_Verb:
result.moveTo(pts[0]);
break;
case SkPath::kLine_Verb:
result.lineTo(pts[1]);
startIndex += 1;
break;
case SkPath::kQuad_Verb:
result.quadTo(pts[1], pts[2]);
startIndex += 2;
break;
case SkPath::kConic_Verb:
result.conicTo(pts[1], pts[2], iter.conicWeight());
startIndex += 2;
break;
case SkPath::kCubic_Verb:
result.cubicTo(pts[1], pts[2], pts[3]);
startIndex += 3;
break;
case SkPath::kClose_Verb:
result.close();
startIndex += 1;
break;
case SkPath::kDone_Verb:
break;
default:
SkASSERT(0);
}
}
#if 0
SkDebugf("\n\noriginal\n");
path->dump();
SkDebugf("\nedited\n");
result.dump();
#endif
*path = result;
}
bool copy_to(SkBitmap* dst, SkColorType dstColorType, const SkBitmap& src) {
SkPixmap srcPM;
if (!src.peekPixels(&srcPM)) {
return false;
}
SkBitmap tmpDst;
SkImageInfo dstInfo = srcPM.info().makeColorType(dstColorType);
if (!tmpDst.setInfo(dstInfo)) {
return false;
}
if (!tmpDst.tryAllocPixels()) {
return false;
}
SkPixmap dstPM;
if (!tmpDst.peekPixels(&dstPM)) {
return false;
}
if (!srcPM.readPixels(dstPM)) {
return false;
}
dst->swap(tmpDst);
return true;
}
void copy_to_g8(SkBitmap* dst, const SkBitmap& src) {
SkASSERT(kBGRA_8888_SkColorType == src.colorType() ||
kRGBA_8888_SkColorType == src.colorType());
SkImageInfo grayInfo = src.info().makeColorType(kGray_8_SkColorType);
dst->allocPixels(grayInfo);
uint8_t* dst8 = (uint8_t*)dst->getPixels();
const uint32_t* src32 = (const uint32_t*)src.getPixels();
const int w = src.width();
const int h = src.height();
const bool isBGRA = (kBGRA_8888_SkColorType == src.colorType());
for (int y = 0; y < h; ++y) {
if (isBGRA) {
// BGRA
for (int x = 0; x < w; ++x) {
uint32_t s = src32[x];
dst8[x] = SkComputeLuminance((s >> 16) & 0xFF, (s >> 8) & 0xFF, s & 0xFF);
}
} else {
// RGBA
for (int x = 0; x < w; ++x) {
uint32_t s = src32[x];
dst8[x] = SkComputeLuminance(s & 0xFF, (s >> 8) & 0xFF, (s >> 16) & 0xFF);
}
}
src32 = (const uint32_t*)((const char*)src32 + src.rowBytes());
dst8 += dst->rowBytes();
}
}
//////////////////////////////////////////////////////////////////////////////////////////////
bool equal_pixels(const SkPixmap& a, const SkPixmap& b) {
if (a.width() != b.width() || a.height() != b.height() || a.colorType() != b.colorType()) {
return false;
}
for (int y = 0; y < a.height(); ++y) {
const char* aptr = (const char*)a.addr(0, y);
const char* bptr = (const char*)b.addr(0, y);
if (memcmp(aptr, bptr, a.width() * a.info().bytesPerPixel())) {
return false;
}
aptr += a.rowBytes();
bptr += b.rowBytes();
}
return true;
}
bool equal_pixels(const SkBitmap& bm0, const SkBitmap& bm1) {
SkPixmap pm0, pm1;
return bm0.peekPixels(&pm0) && bm1.peekPixels(&pm1) && equal_pixels(pm0, pm1);
}
bool equal_pixels(const SkImage* a, const SkImage* b) {
// ensure that peekPixels will succeed
auto imga = a->makeRasterImage();
auto imgb = b->makeRasterImage();
SkPixmap pm0, pm1;
return imga->peekPixels(&pm0) && imgb->peekPixels(&pm1) && equal_pixels(pm0, pm1);
}
sk_sp<SkSurface> makeSurface(SkCanvas* canvas,
const SkImageInfo& info,
const SkSurfaceProps* props) {
auto surf = canvas->makeSurface(info, props);
if (!surf) {
surf = SkSurface::MakeRaster(info, props);
}
return surf;
}
static DEFINE_bool(nativeFonts, true,
"If true, use native font manager and rendering. "
"If false, fonts will draw as portably as possible.");
#if defined(SK_BUILD_FOR_WIN)
static DEFINE_bool(gdi, false,
"Use GDI instead of DirectWrite for font rendering.");
#endif
void SetDefaultFontMgr() {
if (!FLAGS_nativeFonts) {
gSkFontMgr_DefaultFactory = &ToolUtils::MakePortableFontMgr;
}
#if defined(SK_BUILD_FOR_WIN)
if (FLAGS_gdi) {
gSkFontMgr_DefaultFactory = &SkFontMgr_New_GDI;
}
#endif
}
} // namespace ToolUtils