535 lines
17 KiB
C++
535 lines
17 KiB
C++
/*
|
|
* This file is part of RawTherapee.
|
|
*
|
|
* Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
|
|
*
|
|
* RawTherapee is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* RawTherapee is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with RawTherapee. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#ifdef WIN32
|
|
#include <winsock2.h>
|
|
#else
|
|
#include <netinet/in.h>
|
|
#endif
|
|
#include "iccstore.h"
|
|
#include "iccmatrices.h"
|
|
#include <glib/gstdio.h>
|
|
#include "safegtk.h"
|
|
#include "../rtgui/options.h"
|
|
|
|
#include <cstring>
|
|
|
|
namespace rtengine {
|
|
|
|
const double (*wprofiles[])[3] = {xyz_sRGB, xyz_adobe, xyz_prophoto, xyz_widegamut, xyz_bruce, xyz_beta, xyz_best};
|
|
const double (*iwprofiles[])[3] = {sRGB_xyz, adobe_xyz, prophoto_xyz, widegamut_xyz, bruce_xyz, beta_xyz, best_xyz};
|
|
const char* wpnames[] = {"sRGB", "Adobe RGB", "ProPhoto", "WideGamut", "BruceRGB", "Beta RGB", "BestRGB"};
|
|
const char* wpgamma[] = {"default","BT709_g2.2_s4.5", "sRGB_g2.4_s12.92", "linear_g1.0", "standard_g2.2", "standard_g1.8", "High_g1.3_s3.35","Low_g2.6_s6.9"}; //gamma free
|
|
//default = gamma inside profile
|
|
//BT709 g=2.22 s=4.5 sRGB g=2.4 s=12.92
|
|
//linear g=1.0
|
|
//std22 g=2.2 std18 g=1.8
|
|
// high g=1.3 s=3.35 for high dynamic images
|
|
//low g=2.6 s=6.9 for low contrast images
|
|
|
|
|
|
std::vector<Glib::ustring> getGamma () {//return gamma
|
|
|
|
std::vector<Glib::ustring> res;
|
|
for (unsigned int i=0; i<sizeof(wpgamma)/sizeof(wpgamma[0]); i++)
|
|
res.push_back (wpgamma[i]);
|
|
return res;
|
|
}
|
|
|
|
|
|
std::vector<Glib::ustring> getWorkingProfiles () {
|
|
|
|
std::vector<Glib::ustring> res;
|
|
for (unsigned int i=0; i<sizeof(wpnames)/sizeof(wpnames[0]); i++)
|
|
res.push_back (wpnames[i]);
|
|
return res;
|
|
}
|
|
|
|
std::vector<Glib::ustring> ICCStore::getOutputProfiles () {
|
|
|
|
MyMutex::MyLock lock(mutex_);
|
|
|
|
std::vector<Glib::ustring> res;
|
|
for (std::map<Glib::ustring, cmsHPROFILE>::iterator i=fileProfiles.begin(); i!=fileProfiles.end(); i++){
|
|
Glib::ustring name(i->first);
|
|
std::string::size_type i2 = name.find_last_of('/');
|
|
if( i2 == std::string::npos )
|
|
i2 = name.find_last_of('\\');
|
|
if( i2 == std::string::npos )
|
|
res.push_back ( name ); // list only profiles inside selected profiles directory
|
|
}
|
|
return res;
|
|
}
|
|
|
|
|
|
cmsHPROFILE
|
|
ICCStore::makeStdGammaProfile(cmsHPROFILE iprof)
|
|
{
|
|
// forgive me for the messy code, quick hack to change gamma of an ICC profile to the RT standard gamma
|
|
void *buf = NULL;
|
|
if (!iprof) {
|
|
return NULL;
|
|
}
|
|
cmsUInt32Number bytesNeeded = 0;
|
|
cmsSaveProfileToMem(iprof, 0, &bytesNeeded);
|
|
if (bytesNeeded == 0) {
|
|
return NULL;
|
|
}
|
|
uint8_t *data = new uint8_t[bytesNeeded+1];
|
|
cmsSaveProfileToMem(iprof, data, &bytesNeeded);
|
|
const size_t len = (int)bytesNeeded;
|
|
const uint8_t *p = &data[128]; // skip 128 byte header
|
|
uint32_t tag_count;
|
|
memcpy(&tag_count, p, 4);
|
|
p += 4;
|
|
tag_count = ntohl(tag_count);
|
|
|
|
struct icctag {
|
|
uint32_t sig;
|
|
uint32_t offset;
|
|
uint32_t size;
|
|
} tags[tag_count];
|
|
|
|
const uint32_t gamma = 0x239;
|
|
int gamma_size = (gamma == 0 || gamma == 256) ? 12 : 14;
|
|
int data_size = (gamma_size + 3) & ~3;
|
|
for (int i = 0; i < tag_count; i++) {
|
|
memcpy(&tags[i], p, 12);
|
|
tags[i].sig = ntohl(tags[i].sig);
|
|
tags[i].offset = ntohl(tags[i].offset);
|
|
tags[i].size = ntohl(tags[i].size);
|
|
p += 12;
|
|
if (tags[i].sig != 0x62545243 && // bTRC
|
|
tags[i].sig != 0x67545243 && // gTRC
|
|
tags[i].sig != 0x72545243 && // rTRC
|
|
tags[i].sig != 0x6B545243) // kTRC
|
|
{
|
|
data_size += (tags[i].size + 3) & ~3;
|
|
}
|
|
}
|
|
uint32_t sz = 128 + 4 + tag_count * 12 + data_size;
|
|
uint8_t *nd = new uint8_t[sz];
|
|
memset(nd, 0, sz);
|
|
memcpy(nd, data, 128 + 4);
|
|
sz = htonl(sz);
|
|
memcpy(nd, &sz, 4);
|
|
uint32_t offset = 128 + 4 + tag_count * 12;
|
|
uint32_t gamma_offset = 0;
|
|
for (int i = 0; i < tag_count; i++) {
|
|
struct icctag tag;
|
|
tag.sig = htonl(tags[i].sig);
|
|
if (tags[i].sig == 0x62545243 || // bTRC
|
|
tags[i].sig == 0x67545243 || // gTRC
|
|
tags[i].sig == 0x72545243 || // rTRC
|
|
tags[i].sig == 0x6B545243) // kTRC
|
|
{
|
|
if (gamma_offset == 0) {
|
|
gamma_offset = offset;
|
|
uint32_t pcurve[] = { htonl(0x63757276), htonl(0), htonl(gamma_size == 12 ? 0 : 1) };
|
|
memcpy(&nd[offset], pcurve, 12);
|
|
if (gamma_size == 14) {
|
|
uint16_t gm = htons(gamma);
|
|
memcpy(&nd[offset+12], &gm, 2);
|
|
}
|
|
offset += (gamma_size + 3) & ~3;
|
|
}
|
|
tag.offset = htonl(gamma_offset);
|
|
tag.size = htonl(gamma_size);
|
|
} else {
|
|
tag.offset = htonl(offset);
|
|
tag.size = htonl(tags[i].size);
|
|
memcpy(&nd[offset], &data[tags[i].offset], tags[i].size);
|
|
offset += (tags[i].size + 3) & ~3;
|
|
}
|
|
memcpy(&nd[128 + 4 + i * 12], &tag, 12);
|
|
}
|
|
|
|
cmsHPROFILE oprof = cmsOpenProfileFromMem (nd, ntohl(sz));
|
|
delete [] nd;
|
|
delete [] data;
|
|
return oprof;
|
|
}
|
|
|
|
ICCStore*
|
|
ICCStore::getInstance(void)
|
|
{
|
|
static ICCStore* instance_ = 0;
|
|
if ( instance_ == 0 )
|
|
{
|
|
static MyMutex smutex_;
|
|
MyMutex::MyLock lock(smutex_);
|
|
if ( instance_ == 0 )
|
|
{
|
|
instance_ = new ICCStore();
|
|
}
|
|
}
|
|
return instance_;
|
|
}
|
|
|
|
ICCStore::ICCStore ()
|
|
{
|
|
//cmsErrorAction (LCMS_ERROR_SHOW);
|
|
|
|
int N = sizeof(wpnames)/sizeof(wpnames[0]);
|
|
for (int i=0; i<N; i++) {
|
|
wProfiles[wpnames[i]] = createFromMatrix (wprofiles[i]);
|
|
wProfilesGamma[wpnames[i]] = createFromMatrix (wprofiles[i], true);
|
|
wMatrices[wpnames[i]] = wprofiles[i];
|
|
iwMatrices[wpnames[i]] = iwprofiles[i];
|
|
}
|
|
|
|
double mat[3][3]={ {1.0, 0, 0}, {0, 1.0, 0}, {0, 0, 1.0}};
|
|
xyz = createFromMatrix (mat, false, "XYZ");
|
|
srgb = cmsCreate_sRGBProfile ();
|
|
}
|
|
|
|
int ICCStore::numOfWProfiles () {
|
|
|
|
return sizeof(wpnames)/sizeof(wpnames[0]);
|
|
}
|
|
|
|
TMatrix ICCStore::workingSpaceMatrix (Glib::ustring name) {
|
|
|
|
std::map<Glib::ustring, TMatrix>::iterator r = wMatrices.find (name);
|
|
if (r!=wMatrices.end())
|
|
return r->second;
|
|
else
|
|
return wMatrices["sRGB"];
|
|
}
|
|
|
|
TMatrix ICCStore::workingSpaceInverseMatrix (Glib::ustring name) {
|
|
|
|
std::map<Glib::ustring, TMatrix>::iterator r = iwMatrices.find (name);
|
|
if (r!=iwMatrices.end())
|
|
return r->second;
|
|
else
|
|
return iwMatrices["sRGB"];
|
|
}
|
|
|
|
cmsHPROFILE ICCStore::workingSpace (Glib::ustring name) {
|
|
|
|
std::map<Glib::ustring, cmsHPROFILE>::iterator r = wProfiles.find (name);
|
|
if (r!=wProfiles.end())
|
|
return r->second;
|
|
else
|
|
return wProfiles["sRGB"];
|
|
}
|
|
|
|
cmsHPROFILE ICCStore::workingSpaceGamma (Glib::ustring name) {
|
|
|
|
std::map<Glib::ustring, cmsHPROFILE>::iterator r = wProfilesGamma.find (name);
|
|
if (r!=wProfilesGamma.end())
|
|
return r->second;
|
|
else
|
|
return wProfilesGamma["sRGB"];
|
|
}
|
|
|
|
cmsHPROFILE ICCStore::getProfile (Glib::ustring name) {
|
|
|
|
MyMutex::MyLock lock(mutex_);
|
|
|
|
std::map<Glib::ustring, cmsHPROFILE>::iterator r = fileProfiles.find (name);
|
|
if (r!=fileProfiles.end())
|
|
return r->second;
|
|
else {
|
|
if (!name.compare (0, 5, "file:") && safe_file_test (name.substr(5), Glib::FILE_TEST_EXISTS) && !safe_file_test (name.substr(5), Glib::FILE_TEST_IS_DIR)) {
|
|
ProfileContent pc (name.substr(5));
|
|
if (pc.data) {
|
|
cmsHPROFILE profile = pc.toProfile ();
|
|
if (profile) {
|
|
fileProfiles[name] = profile;
|
|
fileProfileContents[name] = pc;
|
|
return profile;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
cmsHPROFILE ICCStore::getStdProfile (Glib::ustring name) {
|
|
|
|
MyMutex::MyLock lock(mutex_);
|
|
|
|
|
|
std::map<Glib::ustring, cmsHPROFILE>::iterator r = fileStdProfiles.find (name.uppercase());
|
|
if (r==fileStdProfiles.end()) return NULL;
|
|
|
|
return r->second;
|
|
}
|
|
|
|
ProfileContent ICCStore::getContent (Glib::ustring name) {
|
|
|
|
MyMutex::MyLock lock(mutex_);
|
|
|
|
return fileProfileContents[name];
|
|
}
|
|
|
|
// Reads all profiles from the given profiles dir
|
|
void ICCStore::init (Glib::ustring usrICCDir, Glib::ustring rtICCDir) {
|
|
|
|
MyMutex::MyLock lock(mutex_);
|
|
|
|
//
|
|
fileProfiles.clear();
|
|
fileProfileContents.clear();
|
|
// RawTherapee's profiles take precedence if a user's profile of the same name exists
|
|
loadICCs(Glib::build_filename(rtICCDir, "output"), false, fileProfiles, fileProfileContents);
|
|
loadICCs(usrICCDir, false, fileProfiles, fileProfileContents);
|
|
|
|
// Input profiles
|
|
// Load these to different areas, since the short name (e.g. "NIKON D700" may overlap between system/user and RT dir)
|
|
fileStdProfiles.clear();
|
|
fileStdProfileContents.clear();
|
|
loadICCs(Glib::build_filename(rtICCDir, "input"), true, fileStdProfiles, fileStdProfileContents);
|
|
}
|
|
|
|
void ICCStore::loadICCs(Glib::ustring rootDirName, bool nameUpper, std::map<Glib::ustring, cmsHPROFILE>& resultProfiles, std::map<Glib::ustring, ProfileContent> &resultProfileContents) {
|
|
if (rootDirName!="") {
|
|
std::deque<Glib::ustring> qDirs;
|
|
|
|
qDirs.push_front(rootDirName);
|
|
|
|
while (!qDirs.empty()) {
|
|
// process directory
|
|
Glib::ustring dirname = qDirs.back();
|
|
qDirs.pop_back();
|
|
|
|
Glib::Dir* dir = NULL;
|
|
try {
|
|
if (!safe_file_test (dirname, Glib::FILE_TEST_IS_DIR)) return;
|
|
dir = new Glib::Dir (dirname);
|
|
}
|
|
catch (Glib::Exception& fe) {
|
|
return;
|
|
}
|
|
dirname = dirname + "/";
|
|
for (Glib::DirIterator i = dir->begin(); i!=dir->end(); ++i) {
|
|
Glib::ustring fname = dirname + *i;
|
|
Glib::ustring sname = *i;
|
|
// ignore directories
|
|
if (!safe_file_test (fname, Glib::FILE_TEST_IS_DIR)) {
|
|
size_t lastdot = sname.find_last_of ('.');
|
|
if (lastdot!=Glib::ustring::npos && lastdot<=sname.size()-4 && (!sname.casefold().compare (lastdot, 4, ".icm") || !sname.casefold().compare (lastdot, 4, ".icc"))) {
|
|
Glib::ustring name = nameUpper ? sname.substr(0,lastdot).uppercase() : sname.substr(0,lastdot);
|
|
ProfileContent pc (fname);
|
|
if (pc.data) {
|
|
cmsHPROFILE profile = pc.toProfile ();
|
|
if (profile) {
|
|
resultProfiles[name] = profile;
|
|
resultProfileContents[name] = pc;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Removed recursive scanning, see issue #1730.
|
|
// To revert to the recursive method, just uncomment the next line.
|
|
|
|
//else qDirs.push_front(fname); // for later scanning
|
|
}
|
|
delete dir;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Determine the first monitor default profile of operating system, if selected
|
|
void ICCStore::findDefaultMonitorProfile() {
|
|
defaultMonitorProfile="";
|
|
|
|
#ifdef WIN32
|
|
// Get current main monitor. Could be fine tuned to get the current windows monitor (multi monitor setup),
|
|
// but problem is that we live in RTEngine with no GUI window to query around
|
|
HDC hDC=GetDC(NULL);
|
|
|
|
if (hDC!=NULL) {
|
|
if (SetICMMode(hDC, ICM_ON)) {
|
|
char profileName[MAX_PATH+1]; DWORD profileLength=MAX_PATH;
|
|
if (GetICMProfileA(hDC,&profileLength,profileName)) defaultMonitorProfile=Glib::ustring(profileName);
|
|
// might fail if e.g. the monitor has no profile
|
|
}
|
|
|
|
ReleaseDC(NULL,hDC);
|
|
}
|
|
#else
|
|
// TODO: Add other OS specific code here
|
|
#endif
|
|
|
|
if (options.rtSettings.verbose) printf("Default monitor profile is: %s\n", defaultMonitorProfile.c_str());
|
|
}
|
|
|
|
ProfileContent::ProfileContent (Glib::ustring fileName) {
|
|
|
|
data = NULL;
|
|
FILE* f = safe_g_fopen (fileName, "rb");
|
|
if (!f)
|
|
return;
|
|
fseek (f, 0, SEEK_END);
|
|
length = ftell (f);
|
|
fseek (f, 0, SEEK_SET);
|
|
data = new char[length+1];
|
|
fread (data, length, 1, f);
|
|
data[length] = 0;
|
|
fclose (f);
|
|
}
|
|
|
|
ProfileContent::ProfileContent (const ProfileContent& other) {
|
|
|
|
length = other.length;
|
|
if (other.data) {
|
|
data = new char[length+1];
|
|
memcpy (data, other.data, length+1);
|
|
}
|
|
else
|
|
data = NULL;
|
|
}
|
|
|
|
ProfileContent::ProfileContent (cmsHPROFILE hProfile) {
|
|
|
|
data = NULL;
|
|
length = 0;
|
|
if (hProfile != NULL) {
|
|
cmsUInt32Number bytesNeeded = 0;
|
|
cmsSaveProfileToMem(hProfile, 0, &bytesNeeded);
|
|
if (bytesNeeded > 0)
|
|
{
|
|
data = new char[bytesNeeded+1];
|
|
cmsSaveProfileToMem(hProfile, data, &bytesNeeded);
|
|
length = (int)bytesNeeded;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
ProfileContent& ProfileContent::operator= (const ProfileContent& other) {
|
|
|
|
length = other.length;
|
|
if (data)
|
|
delete [] data;
|
|
if (other.data) {
|
|
data = new char[length+1];
|
|
memcpy (data, other.data, length+1);
|
|
}
|
|
else
|
|
data = NULL;
|
|
return *this;
|
|
}
|
|
|
|
ProfileContent::~ProfileContent () {
|
|
|
|
if (data)
|
|
delete [] data;
|
|
}
|
|
|
|
cmsHPROFILE ProfileContent::toProfile () {
|
|
|
|
if (data)
|
|
return cmsOpenProfileFromMem (data, length);
|
|
else
|
|
return NULL;
|
|
}
|
|
|
|
cmsHPROFILE ICCStore::createFromMatrix (const double matrix[3][3], bool gamma, Glib::ustring name) {
|
|
|
|
static const unsigned phead[] =
|
|
{ 1024, 0, 0x2100000, 0x6d6e7472, 0x52474220, 0x58595a20, 0, 0, 0,
|
|
0x61637370, 0, 0, 0, 0, 0, 0, 0, 0xf6d6, 0x10000, 0xd32d };
|
|
unsigned pbody[] =
|
|
{ 10, 0x63707274, 0, 36, /* cprt */
|
|
0x64657363, 0, 40, /* desc */
|
|
0x77747074, 0, 20, /* wtpt */
|
|
0x626b7074, 0, 20, /* bkpt */
|
|
0x72545243, 0, 14, /* rTRC */
|
|
0x67545243, 0, 14, /* gTRC */
|
|
0x62545243, 0, 14, /* bTRC */
|
|
0x7258595a, 0, 20, /* rXYZ */
|
|
0x6758595a, 0, 20, /* gXYZ */
|
|
0x6258595a, 0, 20 }; /* bXYZ */
|
|
static const unsigned pwhite[] = { 0xf351, 0x10000, 0x116cc };//D65
|
|
//static const unsigned pwhite[] = { 0xf6d6, 0x10000, 0xd340 };//D50
|
|
|
|
// 0x63757276 : curveType, 0 : reserved, 1 : entries (1=gamma, 0=identity), 0x1000000=1.0
|
|
unsigned pcurve[] = { 0x63757276, 0, 0, 0x1000000 };
|
|
// unsigned pcurve[] = { 0x63757276, 0, 1, 0x1000000 };
|
|
|
|
if (gamma) {
|
|
pcurve[2] = 1;
|
|
// pcurve[3] = 0x1f00000;// pcurve for gamma BT709 : g=2.22 s=4.5
|
|
// normalize gamma in RT, default (Emil's choice = sRGB)
|
|
pcurve[3] = 0x2390000;//pcurve for gamma sRGB : g:2.4 s=12.92
|
|
|
|
} else {
|
|
// lcms2 up to 2.4 has a bug with linear gamma causing precision loss (banding)
|
|
// of floating point data when a normal icc encoding of linear gamma is used
|
|
// (i e 0 table entries), but by encoding a gamma curve which is 1.0 the
|
|
// floating point path is taken within lcms2 so no precision loss occurs and
|
|
// gamma is still 1.0.
|
|
pcurve[2] = 1;
|
|
pcurve[3] = 0x1000000; //pcurve for gamma 1
|
|
}
|
|
|
|
// constructing profile header
|
|
unsigned* oprof = new unsigned [phead[0]/sizeof(unsigned)];
|
|
memset (oprof, 0, phead[0]);
|
|
memcpy (oprof, phead, sizeof(phead));
|
|
|
|
oprof[0] = 132 + 12*pbody[0];
|
|
|
|
// constructing tag directory (pointers inside the file), and types
|
|
// 0x74657874 : text
|
|
// 0x64657363 : description tag
|
|
for (unsigned int i=0; i < pbody[0]; i++) {
|
|
oprof[oprof[0]/4] = i ? (i > 1 ? 0x58595a20 : 0x64657363) : 0x74657874;
|
|
pbody[i*3+2] = oprof[0];
|
|
oprof[0] += (pbody[i*3+3] + 3) & -4;
|
|
}
|
|
memcpy (oprof+32, pbody, sizeof(pbody));
|
|
|
|
// wtpt
|
|
memcpy ((char *)oprof+pbody[8]+8, pwhite, sizeof(pwhite));
|
|
|
|
// r/g/b TRC
|
|
for (int i=4; i < 7; i++)
|
|
memcpy ((char *)oprof+pbody[i*3+2], pcurve, sizeof(pcurve));
|
|
|
|
// r/g/b XYZ
|
|
// pseudoinverse ((double (*)[3]) out_rgb[output_color-1], inverse, 3);
|
|
for (int i=0; i < 3; i++)
|
|
for (int j=0; j < 3; j++) {
|
|
oprof[pbody[j*3+23]/4+i+2] = matrix[i][j] * 0x10000 + 0.5;
|
|
// for (num = k=0; k < 3; k++)
|
|
// num += xyzd50_srgb[i][k] * inverse[j][k];
|
|
}
|
|
|
|
// convert to network byte order
|
|
for (unsigned int i=0; i < phead[0]/4; i++)
|
|
oprof[i] = htonl(oprof[i]);
|
|
|
|
// cprt
|
|
strcpy ((char *)oprof+pbody[2]+8, "--rawtherapee profile--");
|
|
|
|
// desc
|
|
oprof[pbody[5]/4+2] = name.size() + 1;
|
|
strcpy ((char *)oprof+pbody[5]+12, name.c_str());
|
|
|
|
|
|
cmsHPROFILE p = cmsOpenProfileFromMem (oprof, ntohl(oprof[0]));
|
|
delete [] oprof;
|
|
return p;
|
|
}
|
|
}
|