123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- /*
- * 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 "include/core/SkBitmap.h"
- #include "include/core/SkCanvas.h"
- #include "include/core/SkColor.h"
- #include "include/core/SkImage.h"
- #include "include/core/SkImageGenerator.h"
- #include "include/core/SkImageInfo.h"
- #include "include/core/SkRefCnt.h"
- #include "include/core/SkTypes.h"
- #include "include/private/SkColorData.h"
- #include "src/core/SkMakeUnique.h"
- #include "src/core/SkUtils.h"
- #include "tests/Test.h"
- #include "tools/ToolUtils.h"
- #include <utility>
- class TestImageGenerator : public SkImageGenerator {
- public:
- enum TestType {
- kFailGetPixels_TestType,
- kSucceedGetPixels_TestType,
- kLast_TestType = kSucceedGetPixels_TestType
- };
- static int Width() { return 10; }
- static int Height() { return 10; }
- // value choosen so that there is no loss when converting to to RGB565 and back
- static SkColor Color() { return ToolUtils::color_to_565(0xffaabbcc); }
- static SkPMColor PMColor() { return SkPreMultiplyColor(Color()); }
- TestImageGenerator(TestType type, skiatest::Reporter* reporter,
- SkColorType colorType = kN32_SkColorType)
- : INHERITED(GetMyInfo(colorType)), fType(type), fReporter(reporter) {
- SkASSERT((fType <= kLast_TestType) && (fType >= 0));
- }
- ~TestImageGenerator() override {}
- protected:
- static SkImageInfo GetMyInfo(SkColorType colorType) {
- return SkImageInfo::Make(TestImageGenerator::Width(), TestImageGenerator::Height(),
- colorType, kOpaque_SkAlphaType);
- }
- bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
- const Options& options) override {
- REPORTER_ASSERT(fReporter, pixels != nullptr);
- REPORTER_ASSERT(fReporter, rowBytes >= info.minRowBytes());
- if (fType != kSucceedGetPixels_TestType) {
- return false;
- }
- if (info.colorType() != kN32_SkColorType && info.colorType() != getInfo().colorType()) {
- return false;
- }
- char* bytePtr = static_cast<char*>(pixels);
- switch (info.colorType()) {
- case kN32_SkColorType:
- for (int y = 0; y < info.height(); ++y) {
- sk_memset32((uint32_t*)bytePtr,
- TestImageGenerator::PMColor(), info.width());
- bytePtr += rowBytes;
- }
- break;
- case kRGB_565_SkColorType:
- for (int y = 0; y < info.height(); ++y) {
- sk_memset16((uint16_t*)bytePtr,
- SkPixel32ToPixel16(TestImageGenerator::PMColor()), info.width());
- bytePtr += rowBytes;
- }
- break;
- default:
- return false;
- }
- return true;
- }
- private:
- const TestType fType;
- skiatest::Reporter* const fReporter;
- typedef SkImageGenerator INHERITED;
- };
- ////////////////////////////////////////////////////////////////////////////////
- DEF_TEST(Image_NewFromGenerator, r) {
- const TestImageGenerator::TestType testTypes[] = {
- TestImageGenerator::kFailGetPixels_TestType,
- TestImageGenerator::kSucceedGetPixels_TestType,
- };
- const SkColorType testColorTypes[] = {
- kN32_SkColorType,
- kRGB_565_SkColorType
- };
- for (size_t i = 0; i < SK_ARRAY_COUNT(testTypes); ++i) {
- TestImageGenerator::TestType test = testTypes[i];
- for (const SkColorType testColorType : testColorTypes) {
- auto gen = skstd::make_unique<TestImageGenerator>(test, r, testColorType);
- sk_sp<SkImage> image(SkImage::MakeFromGenerator(std::move(gen)));
- if (nullptr == image) {
- ERRORF(r, "SkImage::NewFromGenerator unexpecedly failed ["
- SK_SIZE_T_SPECIFIER "]", i);
- continue;
- }
- REPORTER_ASSERT(r, TestImageGenerator::Width() == image->width());
- REPORTER_ASSERT(r, TestImageGenerator::Height() == image->height());
- REPORTER_ASSERT(r, image->isLazyGenerated());
- SkBitmap bitmap;
- bitmap.allocN32Pixels(TestImageGenerator::Width(), TestImageGenerator::Height());
- SkCanvas canvas(bitmap);
- const SkColor kDefaultColor = 0xffabcdef;
- canvas.clear(kDefaultColor);
- canvas.drawImage(image, 0, 0, nullptr);
- if (TestImageGenerator::kSucceedGetPixels_TestType == test) {
- REPORTER_ASSERT(
- r, TestImageGenerator::Color() == bitmap.getColor(0, 0));
- }
- else {
- REPORTER_ASSERT(r, kDefaultColor == bitmap.getColor(0, 0));
- }
- }
- }
- }
|