/*
 *  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 "rawcacorrection.h"
#include "guiutils.h"
#include <sstream>
#include "rtimage.h"

using namespace rtengine;
using namespace rtengine::procparams;

RAWCACorr::RAWCACorr () : FoldableToolPanel(this, "rawcacorrection", M("TP_CHROMATABERR_LABEL"))
{
    Gtk::Image* icaredL =   Gtk::manage (new RTImage ("ajd-ca-red1.png"));
    Gtk::Image* icaredR =   Gtk::manage (new RTImage ("ajd-ca-red2.png"));
    Gtk::Image* icablueL =  Gtk::manage (new RTImage ("ajd-ca-blue1.png"));
    Gtk::Image* icablueR =  Gtk::manage (new RTImage ("ajd-ca-blue2.png"));

    caAutocorrect = Gtk::manage(new Gtk::CheckButton((M("TP_RAWCACORR_AUTO"))));

    caStrength = Gtk::manage(new Adjuster (M("TP_RAWCACORR_CASTR"), 2.0, 8.0, 0.5, 6.0));
    caStrength->setAdjusterListener (this);
    if (caStrength->delay < options.adjusterMaxDelay) {
        caStrength->delay = options.adjusterMaxDelay;
    }

    caStrength->show();
    caRed = Gtk::manage(new Adjuster (M("TP_RAWCACORR_CARED"), -4.0, 4.0, 0.1, 0, icaredL, icaredR));
    caRed->setAdjusterListener (this);

    if (caRed->delay < options.adjusterMaxDelay) {
        caRed->delay = options.adjusterMaxDelay;
    }

    caRed->show();
    caBlue = Gtk::manage(new Adjuster (M("TP_RAWCACORR_CABLUE"), -4.0, 4.0, 0.1, 0, icablueL, icablueR));
    caBlue->setAdjusterListener (this);

    if (caBlue->delay < options.adjusterMaxDelay) {
        caBlue->delay = options.adjusterMaxDelay;
    }

    caBlue->show();

    pack_start( *caAutocorrect, Gtk::PACK_SHRINK, 4);
    pack_start( *caStrength, Gtk::PACK_SHRINK, 4);
    pack_start( *caRed, Gtk::PACK_SHRINK, 4);
    pack_start( *caBlue, Gtk::PACK_SHRINK, 4);

    caacsconn = caAutocorrect->signal_toggled().connect ( sigc::mem_fun(*this, &RAWCACorr::caCorrectionChanged), true);
}

void RAWCACorr::read(const rtengine::procparams::ProcParams* pp, const ParamsEdited* pedited)
{
    disableListener ();
    caacsconn.block (true);

    if(pedited ) {
        caAutocorrect->set_inconsistent(!pedited->raw.caCorrection);
        caStrength->setEditedState( pedited->raw.caAutoStrength ? Edited : UnEdited );
        caRed->setEditedState( pedited->raw.caRed ? Edited : UnEdited );
        caBlue->setEditedState( pedited->raw.caBlue ? Edited : UnEdited );
    }

    lastCA  = pp->raw.ca_autocorrect;

    caStrength->set_sensitive(pp->raw.ca_autocorrect);
    // disable Red and Blue sliders when caAutocorrect is enabled
    caRed->set_sensitive(!pp->raw.ca_autocorrect);
    caBlue->set_sensitive(!pp->raw.ca_autocorrect);

    caAutocorrect->set_active(pp->raw.ca_autocorrect);
    caStrength->setValue (pp->raw.caautostrength);
    caRed->setValue (pp->raw.cared);
    caBlue->setValue (pp->raw.cablue);

    caacsconn.block (false);
    enableListener ();
}

void RAWCACorr::write( rtengine::procparams::ProcParams* pp, ParamsEdited* pedited)
{
    pp->raw.ca_autocorrect = caAutocorrect->get_active();
    pp->raw.caautostrength = caStrength->getValue();
    pp->raw.cared = caRed->getValue();
    pp->raw.cablue = caBlue->getValue();

    if (pedited) {
        pedited->raw.caCorrection = !caAutocorrect->get_inconsistent();
        pedited->raw.caAutoStrength = caStrength->getEditedState ();
        pedited->raw.caRed = caRed->getEditedState ();
        pedited->raw.caBlue = caBlue->getEditedState ();
    }

}

void RAWCACorr::adjusterChanged (Adjuster* a, double newval)
{
    if (listener) {

        Glib::ustring value = a->getTextValue();

        if (a == caRed) {
            listener->panelChanged (EvPreProcessCARed,  value );
        } else if (a == caBlue) {
            listener->panelChanged (EvPreProcessCABlue,  value );
        } else if (a == caStrength) {
            listener->panelChanged (EvPreProcessCAStrength,  value );
        }
    }
}

void RAWCACorr::setBatchMode(bool batchMode)
{
    ToolPanel::setBatchMode (batchMode);
    caStrength->showEditedCB ();
    caRed->showEditedCB ();
    caBlue->showEditedCB ();
}

void RAWCACorr::setDefaults(const rtengine::procparams::ProcParams* defParams, const ParamsEdited* pedited)
{
    caStrength->setDefault( defParams->raw.caautostrength);
    caRed->setDefault( defParams->raw.cared);
    caBlue->setDefault( defParams->raw.cablue);

    if (pedited) {
        caStrength->setDefaultEditedState( pedited->raw.caAutoStrength ? Edited : UnEdited);
        caRed->setDefaultEditedState( pedited->raw.caRed ? Edited : UnEdited);
        caBlue->setDefaultEditedState( pedited->raw.caBlue ? Edited : UnEdited);
    } else {
        caStrength->setDefaultEditedState( Irrelevant );
        caRed->setDefaultEditedState( Irrelevant );
        caBlue->setDefaultEditedState( Irrelevant );
    }
}

void RAWCACorr::caCorrectionChanged()
{
    if (batchMode) {
        if (caAutocorrect->get_inconsistent()) {
            caAutocorrect->set_inconsistent (false);
            caacsconn.block (true);
            caAutocorrect->set_active (false);
            caacsconn.block (false);
        } else if (lastCA) {
            caAutocorrect->set_inconsistent (true);
        }

        lastCA = caAutocorrect->get_active ();

    }

    /*else {
        // For non batch mode, we disable the red and blue slider if caAutocorrect is true
        if (caAutocorrect->get_active ()) {
            caRed->set_sensitive(false);
            caBlue->set_sensitive(false);
        }
        else {
            caRed->set_sensitive(true);
            caBlue->set_sensitive(true);
        }
    }*/

    caStrength->set_sensitive(caAutocorrect->get_active ());
    // disable Red and Blue sliders when caAutocorrect is enabled
    caRed->set_sensitive(!caAutocorrect->get_active ());
    caBlue->set_sensitive(!caAutocorrect->get_active ());

    if (caAutocorrect->get_active ()) {
        // set caRed and caBlue to 0 as RawImageSource::CA_correct_RT uses this as
        // a condition for auto-CA correction. Alternative would be to change
        // RawImageSource::CA_correct_RT and pass it ca_autocorrect value
        caRed->setValue(0);
        caBlue->setValue(0);
    }

    if (listener) {
        listener->panelChanged (EvPreProcessAutoCA, caAutocorrect->get_active() ? M("GENERAL_ENABLED") : M("GENERAL_DISABLED"));
    }
}

void RAWCACorr::setAdjusterBehavior (bool caadd)
{

    caStrength->setAddMode(caadd);
    caRed->setAddMode(caadd);
    caBlue->setAddMode(caadd);
}

void RAWCACorr::trimValues (rtengine::procparams::ProcParams* pp)
{

    caStrength->trimValue(pp->raw.caautostrength);
    caRed->trimValue(pp->raw.cared);
    caBlue->trimValue(pp->raw.cablue);
}