- Curve editor buttons are set to expand by default, but they are set to shrink as soon as an accompagnying widget is set to expand - White Balance's method button now has a centered ellipse ("...") - White Balance's buttons are now aligned on their right - A "withScrollbar" class is added to MyExpander if the ToolPanel's vertical scrollbar is visible. This can let you add padding space for the scrollbar (see #MyExpander.withScrollbar in RT default theme) - A "maximized" and "fullscreen" class is added to the RTWindow whenever it change state ; BEWARE: if you maximize the window then make it fullscreen, Gtk says that the window is in a "maximized & fullscreen" state, which mean that both class can be added at the same time to the window. One Gtk oddity (at least on Windows) is that you can make your window fullscreen and still drag it around by its header bar... That's not very practical to click on the unfullscreen button if in Single Editor mode with vertical Tab. I also managed to see the window in a Inconified + Maximized state. This part of Gtk doesn't seem very robust, on Windows at least.
887 lines
29 KiB
C++
887 lines
29 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/>.
|
|
*/
|
|
#include "whitebalance.h"
|
|
|
|
#include <iomanip>
|
|
|
|
#include "rtimage.h"
|
|
#include "options.h"
|
|
|
|
#define MINTEMP 1500 //1200
|
|
#define MAXTEMP 60000 //12000
|
|
#define CENTERTEMP 4750
|
|
#define MINGREEN 0.02
|
|
#define MAXGREEN 10.0
|
|
#define MINEQUAL 0.8
|
|
#define MAXEQUAL 1.5
|
|
|
|
using namespace rtengine;
|
|
using namespace rtengine::procparams;
|
|
|
|
Glib::RefPtr<Gdk::Pixbuf> WhiteBalance::wbPixbufs[rtengine::procparams::WBT_CUSTOM + 1];
|
|
/*
|
|
Glib::RefPtr<Gdk::Pixbuf> WhiteBalance::wbCameraPB, WhiteBalance::wbAutoPB, WhiteBalance::wbSunPB, WhiteBalance::wbTungstenPB,
|
|
WhiteBalance::wbCloudyPB, WhiteBalance::wbShadePB, WhiteBalance::wbFluorescentPB, WhiteBalance::wbLampPB,
|
|
WhiteBalance::wbFlashPB, WhiteBalance::wbLedPB, WhiteBalance::wbCustomPB;
|
|
*/
|
|
|
|
void WhiteBalance::init ()
|
|
{
|
|
wbPixbufs[WBT_CAMERA] = RTImage::createFromFile ("wb-camera.png");
|
|
wbPixbufs[WBT_AUTO] = RTImage::createFromFile ("wb-auto.png");
|
|
wbPixbufs[WBT_DAYLIGHT] = RTImage::createFromFile ("wb-sun.png");
|
|
wbPixbufs[WBT_CLOUDY] = RTImage::createFromFile ("wb-cloudy.png");
|
|
wbPixbufs[WBT_SHADE] = RTImage::createFromFile ("wb-shade.png");
|
|
wbPixbufs[WBT_WATER] = RTImage::createFromFile ("wb-water.png");
|
|
// wbPixbufs[WBT_WATER2] = RTImage::createFromFile ("wb-water.png");
|
|
wbPixbufs[WBT_TUNGSTEN] = RTImage::createFromFile ("wb-tungsten.png");
|
|
wbPixbufs[WBT_FLUORESCENT] = RTImage::createFromFile ("wb-fluorescent.png");
|
|
wbPixbufs[WBT_LAMP] = RTImage::createFromFile ("wb-lamp.png");
|
|
wbPixbufs[WBT_FLASH] = RTImage::createFromFile ("wb-flash.png");
|
|
wbPixbufs[WBT_LED] = RTImage::createFromFile ("wb-led.png");
|
|
wbPixbufs[WBT_CUSTOM] = RTImage::createFromFile ("wb-custom.png");
|
|
}
|
|
|
|
void WhiteBalance::cleanup ()
|
|
{
|
|
for (unsigned int i = 0; i < WBT_CUSTOM + 1; i++) {
|
|
wbPixbufs[i].reset();
|
|
}
|
|
}
|
|
|
|
static double wbSlider2Temp(double sval)
|
|
{
|
|
|
|
// slider range: 0 - 10000
|
|
double temp;
|
|
|
|
if (sval <= 5000) {
|
|
// linear below center-temp
|
|
temp = MINTEMP + (sval / 5000.0) * (CENTERTEMP - MINTEMP);
|
|
} else {
|
|
const double slope = (double)(CENTERTEMP - MINTEMP) / (MAXTEMP - CENTERTEMP);
|
|
double x = (sval - 5000) / 5000; // x 0..1
|
|
double y = x * slope + (1.0 - slope) * pow(x, 4.0);
|
|
//double y = pow(x, 4.0);
|
|
temp = CENTERTEMP + y * (MAXTEMP - CENTERTEMP);
|
|
}
|
|
|
|
if (temp < MINTEMP) {
|
|
temp = MINTEMP;
|
|
}
|
|
|
|
if (temp > MAXTEMP) {
|
|
temp = MAXTEMP;
|
|
}
|
|
|
|
return temp;
|
|
}
|
|
|
|
static double wbTemp2Slider(double temp)
|
|
{
|
|
|
|
double sval;
|
|
|
|
if (temp <= CENTERTEMP) {
|
|
sval = ((temp - MINTEMP) / (CENTERTEMP - MINTEMP)) * 5000.0;
|
|
} else {
|
|
const double slope = (double)(CENTERTEMP - MINTEMP) / (MAXTEMP - CENTERTEMP);
|
|
const double y = (temp - CENTERTEMP) / (MAXTEMP - CENTERTEMP);
|
|
double x = pow(y, 0.25); // rough guess of x, will be a little lower
|
|
double k = 0.1;
|
|
bool add = true;
|
|
|
|
// the y=f(x) function is a mess to invert, therefore we have this trial-refinement loop instead.
|
|
// from tests, worst case is about 20 iterations, ie no problem
|
|
for (;;) {
|
|
double y1 = x * slope + (1.0 - slope) * pow(x, 4.0);
|
|
|
|
if (5000 * fabs(y1 - y) < 0.1) {
|
|
break;
|
|
}
|
|
|
|
if (y1 < y) {
|
|
if (!add) {
|
|
k /= 2;
|
|
}
|
|
|
|
x += k;
|
|
add = true;
|
|
} else {
|
|
if (add) {
|
|
k /= 2;
|
|
}
|
|
|
|
x -= k;
|
|
add = false;
|
|
}
|
|
}
|
|
|
|
sval = 5000.0 + x * 5000.0;
|
|
}
|
|
|
|
if (sval < 0) {
|
|
sval = 0;
|
|
}
|
|
|
|
if (sval > 10000) {
|
|
sval = 10000;
|
|
}
|
|
|
|
return sval;
|
|
}
|
|
|
|
WhiteBalance::WhiteBalance () : FoldableToolPanel(this, "whitebalance", M("TP_WBALANCE_LABEL")), wbp(nullptr), wblistener(nullptr)
|
|
{
|
|
|
|
Gtk::HBox* hbox = Gtk::manage (new Gtk::HBox ());
|
|
hbox->set_spacing(4);
|
|
hbox->show ();
|
|
Gtk::Label* lab = Gtk::manage (new Gtk::Label (M("TP_WBALANCE_METHOD")));
|
|
lab->show ();
|
|
|
|
// Create the Tree model
|
|
refTreeModel = Gtk::TreeStore::create(methodColumns);
|
|
// Create the Combobox
|
|
method = Gtk::manage (new MyComboBox ());
|
|
// Assign the model to the Combobox
|
|
method->set_model(refTreeModel);
|
|
|
|
enum WBTypes oldType = WBParams::wbEntries[0]->type;
|
|
enum WBTypes currType;
|
|
Gtk::TreeModel::Row row, childrow;
|
|
|
|
for (unsigned int i = 0; i < WBParams::wbEntries.size(); i++) {
|
|
if (oldType != (currType = WBParams::wbEntries[i]->type)) {
|
|
// New entry type
|
|
if (currType == WBT_FLUORESCENT) {
|
|
// Creating the Fluorescent subcategory header
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = M("TP_WBALANCE_FLUO_HEADER");
|
|
row[methodColumns.colId] = i + 100;
|
|
}
|
|
|
|
if (currType == WBT_WATER) {
|
|
// Creating the under water subcategory header
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = M("TP_WBALANCE_WATER_HEADER");
|
|
row[methodColumns.colId] = i + 100;
|
|
}
|
|
|
|
if (currType == WBT_LAMP) {
|
|
// Creating the Lamp subcategory header
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = M("TP_WBALANCE_LAMP_HEADER");
|
|
row[methodColumns.colId] = i + 100;
|
|
}
|
|
|
|
if (currType == WBT_LED) {
|
|
// Creating the LED subcategory header
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = M("TP_WBALANCE_LED_HEADER");
|
|
row[methodColumns.colId] = i + 100;
|
|
}
|
|
|
|
if (currType == WBT_FLASH) {
|
|
// Creating the Flash subcategory header
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = M("TP_WBALANCE_FLASH_HEADER");
|
|
row[methodColumns.colId] = i + 100;
|
|
}
|
|
}
|
|
|
|
if (currType == WBT_FLUORESCENT
|
|
|| currType == WBT_LAMP
|
|
|| currType == WBT_WATER
|
|
|| currType == WBT_FLASH
|
|
|| currType == WBT_LED
|
|
) {
|
|
childrow = *(refTreeModel->append(row.children()));
|
|
childrow[methodColumns.colIcon] = wbPixbufs[currType];
|
|
childrow[methodColumns.colLabel] = WBParams::wbEntries[i]->GUILabel;
|
|
childrow[methodColumns.colId] = i;
|
|
} else {
|
|
row = *(refTreeModel->append());
|
|
row[methodColumns.colIcon] = wbPixbufs[currType];
|
|
row[methodColumns.colLabel] = WBParams::wbEntries[i]->GUILabel;
|
|
row[methodColumns.colId] = i;
|
|
}
|
|
|
|
oldType = currType;
|
|
|
|
custom_green = 1.0;
|
|
custom_equal = 1.0;
|
|
}
|
|
|
|
//Add the model columns to the Combo (which is a kind of view),
|
|
//rendering them in the default way:
|
|
method->pack_start(methodColumns.colIcon, false);
|
|
method->pack_start(methodColumns.colLabel, true);
|
|
|
|
std::vector<Gtk::CellRenderer*> cells = method->get_cells();
|
|
Gtk::CellRendererText* cellRenderer = dynamic_cast<Gtk::CellRendererText*>(cells.at(1));
|
|
cellRenderer->property_ellipsize() = Pango::ELLIPSIZE_MIDDLE;
|
|
|
|
method->set_active (0); // Camera
|
|
method->show ();
|
|
hbox->pack_start (*lab, Gtk::PACK_SHRINK, 0);
|
|
hbox->pack_start (*method);
|
|
pack_start (*hbox, Gtk::PACK_SHRINK, 0);
|
|
opt = 0;
|
|
|
|
Gtk::HBox* spotbox = Gtk::manage (new Gtk::HBox ());
|
|
spotbox->set_spacing(4);
|
|
spotbox->show ();
|
|
|
|
spotbutton = Gtk::manage (new Gtk::Button ());
|
|
spotbutton->set_tooltip_text(M("TP_WBALANCE_SPOTWB"));
|
|
Gtk::Image* spotimg = Gtk::manage (new RTImage ("gtk-color-picker-small.png"));
|
|
spotimg->show ();
|
|
spotbutton->set_image (*spotimg);
|
|
spotbutton->show ();
|
|
|
|
spotbox->pack_start (*spotbutton);
|
|
|
|
Gtk::Label* slab = Gtk::manage (new Gtk::Label (M("TP_WBALANCE_SIZE")));
|
|
slab->show ();
|
|
|
|
spotsize = Gtk::manage (new MyComboBoxText ());
|
|
spotsize->show ();
|
|
spotsize->append ("2");
|
|
|
|
if (options.whiteBalanceSpotSize == 2) {
|
|
spotsize->set_active(0);
|
|
}
|
|
|
|
spotsize->append ("4");
|
|
|
|
if (options.whiteBalanceSpotSize == 4) {
|
|
spotsize->set_active(1);
|
|
}
|
|
|
|
spotsize->append ("8");
|
|
|
|
if (options.whiteBalanceSpotSize == 8) {
|
|
spotsize->set_active(2);
|
|
}
|
|
|
|
spotsize->append ("16");
|
|
|
|
if (options.whiteBalanceSpotSize == 16) {
|
|
spotsize->set_active(3);
|
|
}
|
|
|
|
spotsize->append ("32");
|
|
|
|
if (options.whiteBalanceSpotSize == 32) {
|
|
spotsize->set_active(4);
|
|
}
|
|
|
|
spotbox->pack_end (*spotsize, Gtk::PACK_EXPAND_WIDGET, 0);
|
|
spotbox->pack_end (*slab, Gtk::PACK_SHRINK, 0);
|
|
|
|
pack_start (*spotbox, Gtk::PACK_SHRINK, 0);
|
|
|
|
Gtk::Image* itempL = Gtk::manage (new RTImage ("ajd-wb-temp1.png"));
|
|
Gtk::Image* itempR = Gtk::manage (new RTImage ("ajd-wb-temp2.png"));
|
|
Gtk::Image* igreenL = Gtk::manage (new RTImage ("ajd-wb-green1.png"));
|
|
Gtk::Image* igreenR = Gtk::manage (new RTImage ("ajd-wb-green2.png"));
|
|
Gtk::Image* iblueredL = Gtk::manage (new RTImage ("ajd-wb-bluered1.png"));
|
|
Gtk::Image* iblueredR = Gtk::manage (new RTImage ("ajd-wb-bluered2.png"));
|
|
|
|
temp = Gtk::manage (new Adjuster (M("TP_WBALANCE_TEMPERATURE"), MINTEMP, MAXTEMP, 5, CENTERTEMP, itempL, itempR, &wbSlider2Temp, &wbTemp2Slider));
|
|
green = Gtk::manage (new Adjuster (M("TP_WBALANCE_GREEN"), MINGREEN, MAXGREEN, 0.001, 1.0, igreenL, igreenR));
|
|
equal = Gtk::manage (new Adjuster (M("TP_WBALANCE_EQBLUERED"), MINEQUAL, MAXEQUAL, 0.001, 1.0, iblueredL, iblueredR));
|
|
cache_customTemp (0);
|
|
cache_customGreen (0);
|
|
cache_customEqual (0);
|
|
equal->set_tooltip_markup (M("TP_WBALANCE_EQBLUERED_TOOLTIP"));
|
|
temp->show ();
|
|
green->show ();
|
|
equal->show ();
|
|
|
|
/* Gtk::HBox* boxgreen = Gtk::manage (new Gtk::HBox ());
|
|
boxgreen->show ();
|
|
|
|
boxgreen->pack_start(*igreenL);
|
|
boxgreen->pack_start(*green);
|
|
boxgreen->pack_start(*igreenR);*/
|
|
|
|
pack_start (*temp);
|
|
//pack_start (*boxgreen);
|
|
pack_start (*green);
|
|
pack_start (*equal);
|
|
|
|
temp->setAdjusterListener (this);
|
|
green->setAdjusterListener (this);
|
|
equal->setAdjusterListener (this);
|
|
|
|
spotbutton->signal_pressed().connect( sigc::mem_fun(*this, &WhiteBalance::spotPressed) );
|
|
methconn = method->signal_changed().connect( sigc::mem_fun(*this, &WhiteBalance::optChanged) );
|
|
spotsize->signal_changed().connect( sigc::mem_fun(*this, &WhiteBalance::spotSizeChanged) );
|
|
}
|
|
|
|
void WhiteBalance::adjusterChanged (Adjuster* a, double newval)
|
|
{
|
|
|
|
int tVal = (int)temp->getValue();
|
|
double gVal = green->getValue();
|
|
double eVal = equal->getValue();
|
|
Gtk::TreeModel::Row row = getActiveMethod();
|
|
|
|
if (row == refTreeModel->children().end()) {
|
|
return;
|
|
}
|
|
|
|
Glib::ustring colLabel = row[methodColumns.colLabel];
|
|
WBEntry* ppMethod = findWBEntry (row[methodColumns.colLabel], WBLT_GUI);
|
|
WBEntry* wbCustom = findWBEntry ("Custom", WBLT_PP);
|
|
|
|
if (!ppMethod || (ppMethod->ppLabel != wbCustom->ppLabel && !(a == equal && ppMethod->type == WBT_AUTO)) ) {
|
|
methconn.block(true);
|
|
opt = setActiveMethod(wbCustom->GUILabel);
|
|
cache_customWB (tVal, gVal);
|
|
cache_customEqual(eVal);
|
|
methconn.block(false);
|
|
}
|
|
|
|
//cache custom WB setting to allow its recall
|
|
if (a == temp) {
|
|
cache_customTemp (tVal);
|
|
} else if (a == green) {
|
|
cache_customGreen (gVal);
|
|
} else if (a == equal) {
|
|
cache_customEqual (eVal);
|
|
|
|
// Recomputing AutoWB if it's the current method
|
|
if (wbp && ppMethod->type == WBT_AUTO) {
|
|
double ctemp = -1.0;
|
|
double cgreen = -1.0;
|
|
wbp->getAutoWB (ctemp, cgreen, eVal);
|
|
|
|
if (ctemp != -1.0) {
|
|
// Set the automatics temperature value only if in SET mode
|
|
if (temp->getEditedState() && !temp->getAddMode() ) {
|
|
temp->setValue (ctemp);
|
|
}
|
|
|
|
// Set the automatics green value only if in SET mode
|
|
if (green->getEditedState() && !green->getAddMode()) {
|
|
green->setValue (cgreen);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (listener) {
|
|
if (a == temp) {
|
|
listener->panelChanged (EvWBTemp, Glib::ustring::format ((int)a->getValue()));
|
|
} else if (a == green) {
|
|
listener->panelChanged (EvWBGreen, Glib::ustring::format (std::setw(4), std::fixed, std::setprecision(3), a->getValue()));
|
|
} else if (a == equal) {
|
|
listener->panelChanged (EvWBequal, Glib::ustring::format (std::setw(4), std::fixed, std::setprecision(3), a->getValue()));
|
|
}
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::optChanged ()
|
|
{
|
|
|
|
Gtk::TreeModel::Row row = getActiveMethod();
|
|
|
|
if (row == refTreeModel->children().end()) {
|
|
return;
|
|
}
|
|
|
|
if (row[methodColumns.colId] >= 100) {
|
|
// "Header" solutions are trapped ; the combo is then set to the previous value
|
|
bool prevState = methconn.block(true);
|
|
method->set_active(opt);
|
|
methconn.block(prevState);
|
|
return;
|
|
}
|
|
|
|
if (opt != row[methodColumns.colId]) {
|
|
|
|
opt = row[methodColumns.colId];
|
|
|
|
if (row[methodColumns.colLabel] == M("GENERAL_UNCHANGED")) {
|
|
temp->setEditedState (UnEdited);
|
|
green->setEditedState (UnEdited);
|
|
equal->setEditedState (UnEdited);
|
|
} else {
|
|
int methodId = findWBEntryId (row[methodColumns.colLabel], WBLT_GUI);
|
|
WBEntry* currMethod = WBParams::wbEntries[methodId];
|
|
|
|
switch (currMethod->type) {
|
|
case WBT_CAMERA:
|
|
if (wbp) {
|
|
double ctemp, cgreen;
|
|
wbp->getCamWB (ctemp, cgreen);
|
|
temp->setValue (temp->getAddMode() ? 0.0 : (int)ctemp);
|
|
green->setValue (green->getAddMode() ? 0.0 : cgreen);
|
|
equal->setValue (equal->getAddMode() ? 0.0 : 1.0);
|
|
|
|
if (batchMode) {
|
|
temp->setEditedState (UnEdited);
|
|
green->setEditedState (UnEdited);
|
|
equal->setEditedState (UnEdited);
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
case WBT_AUTO:
|
|
if (wbp) {
|
|
if (batchMode) {
|
|
temp->setEditedState (UnEdited);
|
|
green->setEditedState (UnEdited);
|
|
// equal remain as is
|
|
}
|
|
|
|
if (!batchMode || equal->getEditedState()) {
|
|
double ctemp, cgreen;
|
|
wbp->getAutoWB (ctemp, cgreen, equal->getValue());
|
|
|
|
if (ctemp != -1.0) {
|
|
temp->setValue (temp->getAddMode() ? 0.0 : (int)ctemp);
|
|
green->setValue (green->getAddMode() ? 0.0 : cgreen);
|
|
}
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
case WBT_CUSTOM:
|
|
if (custom_temp > 0) {
|
|
temp->setValue (temp->getAddMode() ? 0.0 : custom_temp);
|
|
green->setValue (green->getAddMode() ? 0.0 : custom_green);
|
|
equal->setValue (equal->getAddMode() ? 0.0 : custom_equal);
|
|
} else {
|
|
cache_customTemp (temp->getValue());
|
|
cache_customGreen (green->getValue());
|
|
cache_customEqual (equal->getValue());
|
|
}
|
|
|
|
if (batchMode) {
|
|
temp->setEditedState (Edited);
|
|
green->setEditedState (Edited);
|
|
equal->setEditedState (Edited);
|
|
}
|
|
|
|
break;
|
|
|
|
/* All other solution are the default cases
|
|
case WBT_DAYLIGHT:
|
|
case WBT_CLOUDY:
|
|
case WBT_SHADE:
|
|
case WBT_TUNGSTEN:
|
|
case WBT_FLUORESCENT:
|
|
case WBT_LAMP:
|
|
case WBT_FLASH:
|
|
case WBT_LED:*/
|
|
default:
|
|
// recall custom WB settings if it exists, set to 1.0 otherwise
|
|
temp->setValue ( temp->getAddMode() ? 0.0 : (double)(currMethod->temperature));
|
|
green->setValue (green->getAddMode() ? 0.0 : (double)(currMethod->green));
|
|
equal->setValue (equal->getAddMode() ? 0.0 : (double)(currMethod->equal));
|
|
|
|
if (batchMode) {
|
|
temp->setEditedState (Edited);
|
|
green->setEditedState (Edited);
|
|
equal->setEditedState (Edited);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (listener) {
|
|
listener->panelChanged (EvWBMethod, row[methodColumns.colLabel]);
|
|
}
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::spotPressed ()
|
|
{
|
|
|
|
if (wblistener) {
|
|
wblistener->spotWBRequested (getSize());
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::spotSizeChanged ()
|
|
{
|
|
options.whiteBalanceSpotSize = getSize();
|
|
|
|
if (wblistener) {
|
|
wblistener->spotWBRequested (getSize());
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::read (const ProcParams* pp, const ParamsEdited* pedited)
|
|
{
|
|
disableListener ();
|
|
|
|
methconn.block (true);
|
|
equal->setValue (pp->wb.equal);
|
|
|
|
if (pedited) {
|
|
// By default, temperature and green are said "UnEdited", but it may change later
|
|
temp->setEditedState (UnEdited);
|
|
green->setEditedState (UnEdited);
|
|
equal->setEditedState (pedited->wb.equal ? Edited : UnEdited);
|
|
}
|
|
|
|
if (pedited && !pedited->wb.method) {
|
|
opt = setActiveMethod(M("GENERAL_UNCHANGED"));
|
|
} else {
|
|
WBEntry* wbValues = findWBEntry(pp->wb.method, WBLT_PP);
|
|
|
|
if (!wbValues) {
|
|
wbValues = findWBEntry("Camera", WBLT_PP);
|
|
}
|
|
|
|
opt = setActiveMethod(wbValues->GUILabel);
|
|
|
|
// temperature is reset to the associated temperature, or 0.0 if addMode is set.
|
|
switch (wbValues->type) {
|
|
case WBT_CUSTOM:
|
|
temp->setValue (temp->getAddMode() ? 0.0 : pp->wb.temperature);
|
|
green->setValue (green->getAddMode() ? 0.0 : pp->wb.green);
|
|
equal->setValue (equal->getAddMode() ? 0.0 : pp->wb.equal);
|
|
cache_customTemp (pp->wb.temperature);
|
|
cache_customGreen (pp->wb.green);
|
|
cache_customEqual (pp->wb.equal);
|
|
|
|
if (pedited) {
|
|
// The user may have changed the temperature and green value
|
|
temp->setEditedState (pedited->wb.temperature ? Edited : UnEdited);
|
|
green->setEditedState (pedited->wb.green ? Edited : UnEdited);
|
|
}
|
|
|
|
break;
|
|
|
|
case WBT_CAMERA:
|
|
if (wbp) {
|
|
double ctemp = -1.0;
|
|
double cgreen = -1.0;
|
|
wbp->getCamWB (ctemp, cgreen);
|
|
|
|
if (ctemp != -1.0) {
|
|
// Set the camera's temperature value, or 0.0 if in ADD mode
|
|
temp->setValue (temp->getAddMode() ? 0.0 : ctemp);
|
|
// Set the camera's green value, or 0.0 if in ADD mode
|
|
green->setValue (green->getAddMode() ? 0.0 : cgreen);
|
|
equal->setValue (equal->getAddMode() ? 0.0 : 1.);
|
|
} else {
|
|
temp->setValue (temp->getAddMode() ? 0.0 : pp->wb.temperature);
|
|
green->setValue (green->getAddMode() ? 0.0 : pp->wb.green);
|
|
equal->setValue (equal->getAddMode() ? 0.0 : pp->wb.equal);
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
case WBT_AUTO:
|
|
// the equalizer's value is restored for the AutoWB
|
|
equal->setValue (equal->getAddMode() ? 0.0 : pp->wb.equal);
|
|
|
|
// set default values first if in ADD mode, otherwise keep the current ones
|
|
if (temp->getAddMode() ) {
|
|
temp->setValue (0.0);
|
|
}
|
|
|
|
if (green->getAddMode()) {
|
|
green->setValue (0.0);
|
|
}
|
|
|
|
// then check for the correct ones, if possible
|
|
if (wbp) {
|
|
double ctemp = -1.0;
|
|
double cgreen = -1.0;
|
|
wbp->getAutoWB (ctemp, cgreen, pp->wb.equal);
|
|
|
|
if (ctemp != -1.0) {
|
|
// Set the automatics temperature if in SET mode
|
|
if (!pedited || (pedited->wb.temperature && !temp->getAddMode()) ) {
|
|
temp->setValue (ctemp);
|
|
|
|
if (pedited) {
|
|
temp->setEditedState (Edited);
|
|
}
|
|
}
|
|
|
|
// Set the automatics green value if in SET mode
|
|
if (!pedited || (pedited->wb.green && !green->getAddMode())) {
|
|
green->setValue (cgreen);
|
|
|
|
if (pedited) {
|
|
green->setEditedState (Edited);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
/*
|
|
All those types are the "default" case:
|
|
case WBT_DAYLIGHT:
|
|
case WBT_CLOUDY:
|
|
case WBT_SHADE:
|
|
case WBT_TUNGSTEN:
|
|
case WBT_FLUORESCENT:
|
|
case WBT_LAMP:
|
|
case WBT_FLASH:
|
|
case WBT_LED:
|
|
*/
|
|
default:
|
|
// Set the associated temperature, or 0.0 if in ADD mode
|
|
temp->setValue(temp->getAddMode() ? 0.0 : (double)wbValues->temperature);
|
|
// Set the stored temperature, or 0.0 if in ADD mode
|
|
green->setValue(green->getAddMode() ? 0.0 : pp->wb.green);
|
|
equal->setValue(equal->getAddMode() ? 0.0 : pp->wb.equal);
|
|
|
|
// The user may have changed the green value even for predefined WB values
|
|
if (pedited) {
|
|
green->setEditedState (pedited->wb.green ? Edited : UnEdited);
|
|
equal->setEditedState (pedited->wb.equal ? Edited : UnEdited);
|
|
}
|
|
|
|
//cache_customGreen (pp->wb.green);
|
|
break;
|
|
}
|
|
}
|
|
|
|
methconn.block (false);
|
|
enableListener ();
|
|
}
|
|
|
|
void WhiteBalance::write (ProcParams* pp, ParamsEdited* pedited)
|
|
{
|
|
|
|
Gtk::TreeModel::Row row = getActiveMethod();
|
|
|
|
if (pedited) {
|
|
pedited->wb.temperature = temp->getEditedState ();
|
|
pedited->wb.green = green->getEditedState ();
|
|
pedited->wb.equal = equal->getEditedState ();
|
|
pedited->wb.method = row[methodColumns.colLabel] != M("GENERAL_UNCHANGED");
|
|
}
|
|
|
|
WBEntry* ppMethod = findWBEntry (row[methodColumns.colLabel], WBLT_GUI);
|
|
|
|
if (ppMethod) {
|
|
pp->wb.method = ppMethod->ppLabel;
|
|
}
|
|
|
|
pp->wb.temperature = temp->getIntValue ();
|
|
pp->wb.green = green->getValue ();
|
|
pp->wb.equal = equal->getValue ();
|
|
}
|
|
|
|
void WhiteBalance::setDefaults (const ProcParams* defParams, const ParamsEdited* pedited)
|
|
{
|
|
equal->setDefault (defParams->wb.equal);
|
|
|
|
if (wbp && defParams->wb.method == "Camera") {
|
|
double ctemp;
|
|
double cgreen;
|
|
wbp->getCamWB (ctemp, cgreen);
|
|
|
|
// FIXME: Seems to be always -1.0, called too early? Broken!
|
|
if (ctemp != -1.0) {
|
|
temp->setDefault (temp->getAddMode() ? 0 : (int)ctemp);
|
|
green->setDefault (green->getAddMode() ? 0 : cgreen);
|
|
}
|
|
} else if (wbp && defParams->wb.method == "Auto") {
|
|
// this setDefaults method is called too early ; the wbp has been set,
|
|
// but wbp is not ready to provide!
|
|
double ctemp;
|
|
double cgreen;
|
|
wbp->getAutoWB (ctemp, cgreen, defParams->wb.equal);
|
|
|
|
if (ctemp != -1.0) {
|
|
temp->setDefault (temp->getAddMode() ? 0 : (int)ctemp);
|
|
green->setDefault (green->getAddMode() ? 0 : cgreen);
|
|
} else {
|
|
// 6504 & 1.0 = same values as in ProcParams::setDefaults
|
|
temp->setDefault (temp->getAddMode() ? 0 : 6504);
|
|
green->setDefault (green->getAddMode() ? 0 : 1.0);
|
|
}
|
|
} else {
|
|
temp->setDefault (defParams->wb.temperature);
|
|
green->setDefault (defParams->wb.green);
|
|
}
|
|
|
|
if (pedited) {
|
|
temp->setDefaultEditedState (pedited->wb.temperature ? Edited : UnEdited);
|
|
green->setDefaultEditedState (pedited->wb.green ? Edited : UnEdited);
|
|
equal->setDefaultEditedState (pedited->wb.equal ? Edited : UnEdited);
|
|
} else {
|
|
temp->setDefaultEditedState (Irrelevant);
|
|
green->setDefaultEditedState (Irrelevant);
|
|
equal->setDefaultEditedState (Irrelevant);
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::setBatchMode (bool batchMode)
|
|
{
|
|
|
|
ToolPanel::setBatchMode (batchMode);
|
|
temp->showEditedCB ();
|
|
green->showEditedCB ();
|
|
equal->showEditedCB ();
|
|
Gtk::TreeModel::Row row = *(refTreeModel->append());
|
|
row[methodColumns.colId] = WBParams::wbEntries.size();
|
|
row[methodColumns.colLabel] = M("GENERAL_UNCHANGED");
|
|
|
|
}
|
|
|
|
int WhiteBalance::getSize ()
|
|
{
|
|
|
|
return atoi(spotsize->get_active_text().c_str());
|
|
}
|
|
|
|
void WhiteBalance::setWB (int vtemp, double vgreen)
|
|
{
|
|
|
|
methconn.block(true);
|
|
WBEntry *wbValues = findWBEntry("Custom", WBLT_PP);
|
|
temp->setValue (vtemp);
|
|
green->setValue (vgreen);
|
|
opt = setActiveMethod(wbValues->GUILabel);
|
|
cache_customWB (vtemp, vgreen); // sequence in which this call is made is important; must be before "method->set_active (2);"
|
|
cache_customEqual(equal->getValue());
|
|
temp->setEditedState (Edited);
|
|
green->setEditedState (Edited);
|
|
methconn.block(false);
|
|
|
|
if (listener) {
|
|
listener->panelChanged (EvWBTemp, Glib::ustring::compose("%1, %2", (int)temp->getValue(), Glib::ustring::format (std::setw(4), std::fixed, std::setprecision(3), green->getValue())));
|
|
}
|
|
}
|
|
|
|
void WhiteBalance::setAdjusterBehavior (bool tempadd, bool greenadd, bool equaladd)
|
|
{
|
|
|
|
temp->setAddMode(tempadd);
|
|
green->setAddMode(greenadd);
|
|
equal->setAddMode(equaladd);
|
|
}
|
|
|
|
void WhiteBalance::trimValues (rtengine::procparams::ProcParams* pp)
|
|
{
|
|
|
|
temp->trimValue(pp->wb.temperature);
|
|
green->trimValue(pp->wb.green);
|
|
equal->trimValue(pp->wb.equal);
|
|
}
|
|
|
|
inline void WhiteBalance::cache_customTemp(int temp)
|
|
{
|
|
custom_temp = temp;
|
|
}
|
|
|
|
void WhiteBalance::cache_customGreen(double green)
|
|
{
|
|
custom_green = green;
|
|
}
|
|
void WhiteBalance::cache_customEqual(double equal)
|
|
{
|
|
custom_equal = equal;
|
|
}
|
|
|
|
void WhiteBalance::cache_customWB(int temp, double green)
|
|
{
|
|
cache_customTemp (temp);
|
|
cache_customGreen (green);
|
|
}
|
|
|
|
int WhiteBalance::findWBEntryId (Glib::ustring label, enum WB_LabelType lblType)
|
|
{
|
|
for (unsigned int i = 0; i < WBParams::wbEntries.size(); i++) {
|
|
if (label == (lblType == WBLT_GUI ? WBParams::wbEntries[i]->GUILabel : WBParams::wbEntries[i]->ppLabel)) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
WBEntry* WhiteBalance::findWBEntry (Glib::ustring label, enum WB_LabelType lblType)
|
|
{
|
|
for (unsigned int i = 0; i < WBParams::wbEntries.size(); i++) {
|
|
if (label == (lblType == WBLT_GUI ? WBParams::wbEntries[i]->GUILabel : WBParams::wbEntries[i]->ppLabel)) {
|
|
return WBParams::wbEntries[i];
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
int WhiteBalance::_setActiveMethod(Glib::ustring &label, Gtk::TreeModel::Children &children)
|
|
{
|
|
int found = -1;
|
|
|
|
for(Gtk::TreeModel::Children::iterator iter = children.begin(); iter != children.end() && found == -1; ++iter) {
|
|
Gtk::TreeModel::Row row = *iter;
|
|
|
|
if (row[methodColumns.colLabel] == label) {
|
|
method->set_active(iter);
|
|
found = method->get_active_row_number();
|
|
}
|
|
|
|
if (found != -1) {
|
|
return found;
|
|
}
|
|
|
|
Gtk::TreeModel::Children childs = row.children();
|
|
|
|
if (childs.size()) {
|
|
found = _setActiveMethod(label, childs);
|
|
|
|
if (found != -1) {
|
|
return found;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Entry not found
|
|
return -1;
|
|
}
|
|
|
|
int WhiteBalance::setActiveMethod(Glib::ustring label)
|
|
{
|
|
Gtk::TreeModel::Children children = refTreeModel->children();
|
|
return _setActiveMethod(label, children);
|
|
}
|
|
|
|
inline Gtk::TreeRow WhiteBalance::getActiveMethod ()
|
|
{
|
|
return *(method->get_active());
|
|
}
|