/*
    Scan Tailor - Interactive post-processing tool for scanned pages.
    Copyright (C)  Joseph Artsimovich <joseph.artsimovich@gmail.com>

    This program 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.

    This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef OUTPUT_BLACK_WHITE_OPTIONS_H_
#define OUTPUT_BLACK_WHITE_OPTIONS_H_

class QString;
class QDomDocument;
class QDomElement;

namespace output
{

enum ThresholdFilter { T_OTSU, T_SAUVOLA, T_WOLF, T_WINDOW, T_BRADLEY, T_GRAD, T_EDGEPLUS, T_BLURDIV, T_EDGEDIV };

class BlackWhiteOptions
{
public:
    BlackWhiteOptions();

    BlackWhiteOptions(QDomElement const& el);

    QDomElement toXml(QDomDocument& doc, QString const& name) const;

    int thresholdAdjustment() const
    {
        return m_thresholdAdjustment;
    }

    void setThresholdAdjustment(int val)
    {
        m_thresholdAdjustment = val;
    }

    int thresholdForegroundAdjustment() const
    {
        return m_thresholdForegroundAdjustment;
    }

    void setThresholdForegroundAdjustment(int val)
    {
        m_thresholdForegroundAdjustment = val;
    }

    ThresholdFilter thresholdMethod() const
    {
        return m_thresholdMethod;
    }

    int thresholdWindowSize() const
    {
        return m_thresholdWindowSize;
    }

    double thresholdCoef() const
    {
        return m_thresholdCoef;
    }

    void setThresholdMethod(ThresholdFilter val)
    {
        m_thresholdMethod = val;
    }

    void setThresholdWindowSize(int val)
    {
        m_thresholdWindowSize = val;
    }

    void setThresholdCoef(float val)
    {
        m_thresholdCoef = val;
    }

    bool operator==(BlackWhiteOptions const& other) const;

    bool operator!=(BlackWhiteOptions const& other) const;
private:
    int m_thresholdAdjustment;
    int m_thresholdForegroundAdjustment;
    ThresholdFilter m_thresholdMethod;
    int m_thresholdWindowSize;
    double m_thresholdCoef;

    static ThresholdFilter parseThresholdMethod(QString const& str);

    static QString formatThresholdMethod(ThresholdFilter type);
};

} // namespace output

#endif
