blob: 2fa39f6c759f6ac08a505152caea78cc50f8f7b2 [file] [log] [blame]
/*
* Copyright 2017 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "SkColorSpacePriv.h"
#include "SkImage.h"
#include "SkPngEncoder.h"
#include "Resources.h"
/**
* Create a color space that swaps the red, green, and blue channels.
*/
static sk_sp<SkColorSpace> gbr_color_space() {
return as_CSB(SkColorSpace::MakeSRGB())->makeColorSpin();
}
/**
* Create a color space with a steep transfer function.
*/
static sk_sp<SkColorSpace> tf_color_space() {
SkColorSpaceTransferFn fn;
fn.fA = 1.f; fn.fB = 0.f; fn.fC = 0.f; fn.fD = 0.f; fn.fE = 0.f; fn.fF = 0.f; fn.fG = 50.f;
return SkColorSpace::MakeRGB(fn, SkColorSpace::kSRGB_Gamut);
}
/**
* Create a wide gamut color space.
*/
static sk_sp<SkColorSpace> wide_gamut_color_space() {
return SkColorSpace::MakeRGB(SkColorSpace::kSRGB_RenderTargetGamma,
SkColorSpace::kRec2020_Gamut);
}
int main(int argc, char** argv) {
sk_sp<SkImage> image = GetResourceAsImage("flutter_logo.jpg");
if (!image) {
SkDebugf("Cannot find flutter_logo.jpg in resources.\n");
return 1;
}
// Encode an image with a gbr color space.
SkImageInfo info = SkImageInfo::MakeN32(image->width(), image->height(), kOpaque_SkAlphaType,
gbr_color_space());
size_t rowBytes = info.minRowBytes();
SkAutoTMalloc<uint8_t> storage(rowBytes * image->height());
SkPixmap src(info, storage.get(), rowBytes);
image->readPixels(src, 0, 0, SkImage::kDisallow_CachingHint);
SkFILEWStream dst0("gbr.png");
SkPngEncoder::Options opts;
opts.fUnpremulBehavior = SkTransferFunctionBehavior::kIgnore; // Does not matter for opaque src
SkAssertResult(SkPngEncoder::Encode(&dst0, src, opts));
// Encode an image with steep transfer function.
src.setColorSpace(tf_color_space());
image->readPixels(src, 0, 0, SkImage::kDisallow_CachingHint);
SkFILEWStream dst1("tf.png");
SkAssertResult(SkPngEncoder::Encode(&dst1, src, opts));
// Encode a wide gamut image.
src.setColorSpace(wide_gamut_color_space());
image->readPixels(src, 0, 0, SkImage::kDisallow_CachingHint);
SkFILEWStream dst2("wide-gamut.png");
SkAssertResult(SkPngEncoder::Encode(&dst2, src, opts));
return 0;
}