37df17d164
partial impl for GPU on linux (need getProcAddress and SkEGLContext) git-svn-id: http://skia.googlecode.com/svn/trunk@661 2bbb7eff-a529-9590-31e7-b0007b416f81
305 lines
8.6 KiB
C++
305 lines
8.6 KiB
C++
#include "gm.h"
|
|
#include "SkColorPriv.h"
|
|
#include "SkGraphics.h"
|
|
#include "SkImageDecoder.h"
|
|
#include "SkImageEncoder.h"
|
|
#include "SkStream.h"
|
|
#include "SkRefCnt.h"
|
|
|
|
#include "GrContext.h"
|
|
#include "SkGpuCanvas.h"
|
|
#include "SkEGLContext.h"
|
|
#include "SkDevice.h"
|
|
|
|
#ifdef SK_SUPPORT_PDF
|
|
#include "SkPDFDevice.h"
|
|
#include "SkPDFDocument.h"
|
|
#endif
|
|
|
|
using namespace skiagm;
|
|
|
|
// need to explicitly declare this, or we get some weird infinite loop llist
|
|
template GMRegistry* GMRegistry::gHead;
|
|
|
|
class Iter {
|
|
public:
|
|
Iter() {
|
|
fReg = GMRegistry::Head();
|
|
}
|
|
|
|
GM* next() {
|
|
if (fReg) {
|
|
GMRegistry::Factory fact = fReg->factory();
|
|
fReg = fReg->next();
|
|
return fact(0);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
static int Count() {
|
|
const GMRegistry* reg = GMRegistry::Head();
|
|
int count = 0;
|
|
while (reg) {
|
|
count += 1;
|
|
reg = reg->next();
|
|
}
|
|
return count;
|
|
}
|
|
|
|
private:
|
|
const GMRegistry* fReg;
|
|
};
|
|
|
|
static SkString make_name(const char shortName[], const char configName[]) {
|
|
SkString name(shortName);
|
|
name.appendf("_%s", configName);
|
|
return name;
|
|
}
|
|
|
|
static SkString make_filename(const char path[], const SkString& name, const char suffix[]) {
|
|
SkString filename(path);
|
|
if (filename.size() && filename[filename.size() - 1] != '/') {
|
|
filename.append("/");
|
|
}
|
|
filename.appendf("%s.%s", name.c_str(), suffix);
|
|
return filename;
|
|
}
|
|
|
|
/* since PNG insists on unpremultiplying our alpha, we take no precision chances
|
|
and force all pixels to be 100% opaque, otherwise on compare we may not get
|
|
a perfect match.
|
|
*/
|
|
static void force_all_opaque(const SkBitmap& bitmap) {
|
|
SkAutoLockPixels lock(bitmap);
|
|
for (int y = 0; y < bitmap.height(); y++) {
|
|
for (int x = 0; x < bitmap.width(); x++) {
|
|
*bitmap.getAddr32(x, y) |= (SK_A32_MASK << SK_A32_SHIFT);
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool write_bitmap(const SkString& path, const SkBitmap& bitmap) {
|
|
SkBitmap copy;
|
|
bitmap.copyTo(©, SkBitmap::kARGB_8888_Config);
|
|
force_all_opaque(copy);
|
|
return SkImageEncoder::EncodeFile(path.c_str(), copy,
|
|
SkImageEncoder::kPNG_Type, 100);
|
|
}
|
|
|
|
static inline SkPMColor compute_diff_pmcolor(SkPMColor c0, SkPMColor c1) {
|
|
int dr = SkGetPackedR32(c0) - SkGetPackedR32(c1);
|
|
int dg = SkGetPackedG32(c0) - SkGetPackedG32(c1);
|
|
int db = SkGetPackedB32(c0) - SkGetPackedB32(c1);
|
|
return SkPackARGB32(0xFF, SkAbs32(dr), SkAbs32(dg), SkAbs32(db));
|
|
}
|
|
|
|
static void compute_diff(const SkBitmap& target, const SkBitmap& base,
|
|
SkBitmap* diff) {
|
|
SkAutoLockPixels alp(*diff);
|
|
|
|
const int w = target.width();
|
|
const int h = target.height();
|
|
for (int y = 0; y < h; y++) {
|
|
for (int x = 0; x < w; x++) {
|
|
SkPMColor c0 = *base.getAddr32(x, y);
|
|
SkPMColor c1 = *target.getAddr32(x, y);
|
|
SkPMColor d = 0;
|
|
if (c0 != c1) {
|
|
d = compute_diff_pmcolor(c0, c1);
|
|
}
|
|
*diff->getAddr32(x, y) = d;
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool compare(const SkBitmap& target, const SkBitmap& base,
|
|
const SkString& name, SkBitmap* diff) {
|
|
SkBitmap copy;
|
|
const SkBitmap* bm = ⌖
|
|
if (target.config() != SkBitmap::kARGB_8888_Config) {
|
|
target.copyTo(©, SkBitmap::kARGB_8888_Config);
|
|
bm = ©
|
|
}
|
|
|
|
force_all_opaque(*bm);
|
|
|
|
const int w = bm->width();
|
|
const int h = bm->height();
|
|
if (w != base.width() || h != base.height()) {
|
|
SkDebugf("---- dimensions mismatch for %s base [%d %d] current [%d %d]\n",
|
|
name.c_str(), base.width(), base.height(), w, h);
|
|
return false;
|
|
}
|
|
|
|
SkAutoLockPixels bmLock(*bm);
|
|
SkAutoLockPixels baseLock(base);
|
|
|
|
for (int y = 0; y < h; y++) {
|
|
for (int x = 0; x < w; x++) {
|
|
SkPMColor c0 = *base.getAddr32(x, y);
|
|
SkPMColor c1 = *bm->getAddr32(x, y);
|
|
if (c0 != c1) {
|
|
SkDebugf("----- pixel mismatch for %s at [%d %d] base 0x%08X current 0x%08X\n",
|
|
name.c_str(), x, y, c0, c1);
|
|
|
|
if (diff) {
|
|
diff->setConfig(SkBitmap::kARGB_8888_Config, w, h);
|
|
diff->allocPixels();
|
|
compute_diff(*bm, base, diff);
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
// they're equal
|
|
return true;
|
|
}
|
|
|
|
static void write_pdf(GM* gm, const char writePath[]) {
|
|
#ifdef SK_SUPPORT_PDF
|
|
SkISize size = gm->getISize();
|
|
SkPDFDevice* dev = new SkPDFDevice(size.width(), size.height());
|
|
SkAutoUnref aur(dev);
|
|
|
|
{
|
|
SkCanvas c(dev);
|
|
gm->draw(&c);
|
|
}
|
|
|
|
SkDynamicMemoryWStream output;
|
|
SkPDFDocument doc;
|
|
doc.appendPage(dev);
|
|
doc.emitPDF(&output);
|
|
|
|
SkString shortName(gm->shortName());
|
|
SkString path = make_filename(writePath, shortName, "pdf");
|
|
SkFILEWStream stream(path.c_str());
|
|
stream.write(output.getStream(), output.getOffset());
|
|
#endif
|
|
}
|
|
|
|
static const struct {
|
|
SkBitmap::Config fConfig;
|
|
bool fUseGPU;
|
|
const char* fName;
|
|
} gRec[] = {
|
|
{ SkBitmap::kARGB_8888_Config, false, "8888" },
|
|
{ SkBitmap::kARGB_4444_Config, false, "4444" },
|
|
{ SkBitmap::kRGB_565_Config, false, "565" },
|
|
{ SkBitmap::kARGB_8888_Config, true, "gpu" },
|
|
};
|
|
|
|
int main (int argc, char * const argv[]) {
|
|
SkAutoGraphics ag;
|
|
|
|
const char* writePath = NULL; // if non-null, where we write the originals
|
|
const char* readPath = NULL; // if non-null, were we read from to compare
|
|
const char* diffPath = NULL; // if non-null, where we write our diffs (from compare)
|
|
|
|
char* const* stop = argv + argc;
|
|
for (++argv; argv < stop; ++argv) {
|
|
if (strcmp(*argv, "-w") == 0) {
|
|
argv++;
|
|
if (argv < stop && **argv) {
|
|
writePath = *argv;
|
|
}
|
|
} else if (strcmp(*argv, "-r") == 0) {
|
|
argv++;
|
|
if (argv < stop && **argv) {
|
|
readPath = *argv;
|
|
}
|
|
} else if (strcmp(*argv, "-d") == 0) {
|
|
argv++;
|
|
if (argv < stop && **argv) {
|
|
diffPath = *argv;
|
|
}
|
|
}
|
|
}
|
|
|
|
// setup a GL context for drawing offscreen
|
|
GrContext* context = NULL;
|
|
SkEGLContext eglContext;
|
|
if (eglContext.init(1024, 1024)) {
|
|
context = GrContext::CreateGLShaderContext();
|
|
}
|
|
|
|
Iter iter;
|
|
GM* gm;
|
|
|
|
if (readPath) {
|
|
fprintf(stderr, "reading from %s\n", readPath);
|
|
} else if (writePath) {
|
|
fprintf(stderr, "writing to %s\n", writePath);
|
|
}
|
|
|
|
while ((gm = iter.next()) != NULL) {
|
|
SkISize size = gm->getISize();
|
|
SkDebugf("drawing... %s [%d %d]\n", gm->shortName(),
|
|
size.width(), size.height());
|
|
|
|
SkBitmap bitmap;
|
|
for (size_t i = 0; i < SK_ARRAY_COUNT(gRec); i++) {
|
|
bitmap.setConfig(gRec[i].fConfig, size.width(), size.height());
|
|
bitmap.allocPixels();
|
|
bitmap.eraseColor(0);
|
|
SkCanvas canvas(bitmap);
|
|
|
|
if (gRec[i].fUseGPU) {
|
|
if (NULL == context) {
|
|
continue;
|
|
}
|
|
SkGpuCanvas gc(context);
|
|
gc.setDevice(gc.createDevice(bitmap.config(), bitmap.width(), bitmap.height(),
|
|
bitmap.isOpaque(), false))->unref();
|
|
gm->draw(&gc);
|
|
gc.readPixels(&bitmap); // overwrite our previous allocation
|
|
} else {
|
|
gm->draw(&canvas);
|
|
}
|
|
SkString name = make_name(gm->shortName(), gRec[i].fName);
|
|
|
|
if (writePath) {
|
|
SkString path = make_filename(writePath, name, "png");
|
|
bool success = write_bitmap(path, bitmap);
|
|
if (!success) {
|
|
fprintf(stderr, "FAILED to write %s\n", path.c_str());
|
|
}
|
|
write_pdf(gm, writePath);
|
|
} else if (readPath) {
|
|
SkString path = make_filename(readPath, name, "png");
|
|
SkBitmap orig;
|
|
bool success = SkImageDecoder::DecodeFile(path.c_str(), &orig,
|
|
SkBitmap::kARGB_8888_Config,
|
|
SkImageDecoder::kDecodePixels_Mode, NULL);
|
|
if (success) {
|
|
SkBitmap diffBitmap;
|
|
success = compare(bitmap, orig, name, diffPath ? &diffBitmap : NULL);
|
|
if (!success && diffPath) {
|
|
SkString diffName = make_filename(diffPath, name, ".diff.png");
|
|
fprintf(stderr, "Writing %s\n", diffName.c_str());
|
|
write_bitmap(diffName, diffBitmap);
|
|
}
|
|
} else {
|
|
fprintf(stderr, "FAILED to read %s\n", path.c_str());
|
|
}
|
|
}
|
|
}
|
|
SkDELETE(gm);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
using namespace skiagm;
|
|
|
|
GM::GM() {}
|
|
GM::~GM() {}
|
|
|
|
void GM::draw(SkCanvas* canvas) {
|
|
this->onDraw(canvas);
|
|
}
|
|
|
|
|