skia2/tools/imgcvt.cpp
Mike Klein 06a215b768 add profile->profile mode to imgcvt
E.g.

out/imgcvt /System/Library/ColorSync/Profiles/ITU-2020.icc ~/skcms/profiles/mobile/sRGB_parametric.icc
(0, 0, 0) --> (+0.0000, +0.0000, +0.0000)
(1, 1, 1) --> (+0.9998, +0.9998, +0.9998)
(1, 0, 0) --> (+1.2481, -0.3879, -0.1435)
(0, 1, 0) --> (-0.7903, +1.0561, -0.3501)
(0, 0, 1) --> (-0.2992, -0.0886, +1.0503)
(0, 1, 1) --> (-0.8325, +1.0527, +1.0077)
(1, 0, 1) --> (+1.2239, -0.4000, +1.0428)
(1, 1, 0) --> (+1.0312, +1.0035, -0.3791)

Change-Id: Ieb38a3b519f00ae56491c3265ed9725888b692e9
Reviewed-on: https://skia-review.googlesource.com/152360
Reviewed-by: Brian Osman <brianosman@google.com>
Commit-Queue: Mike Klein <mtklein@google.com>
2018-09-06 18:54:19 +00:00

121 lines
3.7 KiB
C++

/*
* Copyright 2018 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "../third_party/skcms/skcms.h"
#include "SkColorSpacePriv.h"
#include "SkData.h"
#include "SkImage.h"
#include "SkStream.h"
int main(int argc, char** argv) {
const char* source_path = argc > 1 ? argv[1] : nullptr;
if (!source_path) {
SkDebugf("Please pass an image or profile to convert"
" as the first argument to this program.\n");
return 1;
}
const char* dst_profile_path = argc > 2 ? argv[2] : nullptr;
skcms_ICCProfile dst_profile = *skcms_sRGB_profile();
if (dst_profile_path) {
sk_sp<SkData> blob = SkData::MakeFromFileName(dst_profile_path);
if (!skcms_Parse(blob->data(), blob->size(), &dst_profile)) {
SkDebugf("Can't parse %s as an ICC profile.\n", dst_profile_path);
return 1;
}
}
auto blob = SkData::MakeFromFileName(source_path);
skcms_ICCProfile src_profile;
if (skcms_Parse(blob->data(), blob->size(), &src_profile)) {
// Transform white, black, primaries, and primary complements.
float src[] = {
0,0,0,
1,1,1,
1,0,0,
0,1,0,
0,0,1,
0,1,1,
1,0,1,
1,1,0,
};
float dst[24] = {0};
if (!skcms_Transform(
src, skcms_PixelFormat_RGB_fff, skcms_AlphaFormat_Unpremul, &src_profile,
dst, skcms_PixelFormat_RGB_fff, skcms_AlphaFormat_Unpremul, &dst_profile,
8)) {
SkDebugf("Cannot transform.\n");
return 1;
}
for (int i = 0; i < 8; i++) {
SkDebugf("(%g, %g, %g) --> (%+.4f, %+.4f, %+.4f)\n",
src[3*i+0], src[3*i+1], src[3*i+2],
dst[3*i+0], dst[3*i+1], dst[3*i+2]);
}
return 0;
}
sk_sp<SkImage> image = SkImage::MakeFromEncoded(blob);
if (!image) {
SkDebugf("Couldn't decode %s as an SkImage or an ICC profile.\n", source_path);
return 1;
}
image = image->makeRasterImage();
if (!image) {
SkDebugf("Converting to raster image failed.\n");
return 1;
}
SkPixmap pixmap;
if (!image->peekPixels(&pixmap)) {
SkDebugf("We really should be able to peek raster pixels.\n");
return 1;
}
SkColorSpace* src_cs = image->colorSpace() ? image->colorSpace()
: sk_srgb_singleton();
src_cs->toProfile(&src_profile);
skcms_PixelFormat fmt;
switch (pixmap.colorType()) {
case kRGBA_8888_SkColorType: fmt = skcms_PixelFormat_RGBA_8888; break;
case kBGRA_8888_SkColorType: fmt = skcms_PixelFormat_BGRA_8888; break;
default:
SkDebugf("color type %d not yet supported, imgcvt.cpp needs an update.\n",
pixmap.colorType());
return 1;
}
if (pixmap.alphaType() != kPremul_SkAlphaType) {
SkDebugf("not premul, that's weird.\n");
return 1;
}
auto alpha = skcms_AlphaFormat_PremulAsEncoded;
if (pixmap.rowBytes() != (size_t)pixmap.width() * pixmap.info().bytesPerPixel()) {
SkDebugf("not a tight pixmap, need a loop here\n");
return 1;
}
skcms_Transform(pixmap.addr(), fmt,alpha, &src_profile,
pixmap.writable_addr(), fmt,alpha, &dst_profile,
pixmap.width() * pixmap.height());
pixmap.setColorSpace(SkColorSpace::Make(dst_profile));
sk_sp<SkImage> transformed = SkImage::MakeRasterCopy(pixmap);
sk_sp<SkData> png = transformed->encodeToData();
SkFILEWStream("transformed.png").write(png->data(), png->size());
return 0;
}