geo/tcNegativeProduct.h

   1 /***************************************************************************
   2  *   This file is part of Tecorrec.                                        *
   3  *   Copyright 2008 James Hogan <james@albanarts.com>                      *
   4  *                                                                         *
   5  *   Tecorrec is free software: you can redistribute it and/or modify      *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation, either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   Tecorrec is distributed in the hope that it will be useful,           *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with Tecorrec.  If not, write to the Free Software Foundation,  *
  17  *   Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.   *
  18  ***************************************************************************/
  19
  20 #ifndef _tcNegativeProduct_h_
  21 #define _tcNegativeProduct_h_
  22
  23 /**
  24  * @file tcNegativeProduct.h
  25  * @brief Highlights the parts of an image which are dark in all channels.
  26  */
  27
  28 #include "tcChannel.h"
  29
  30 #include <QList>
  31
  32 class QSlider;
  33 class QLabel;
  34 class QComboBox;
  35
  36 /// Highlights the parts of an image which are dark in all channels.
  37 class tcNegativeProduct : public tcChannel
  38 {
  39     Q_OBJECT
  40
  41   public:
  42
  43     /*
  44      * Constructors + destructor
  45      */
  46
  47     /// Primary constructor.
  48     tcNegativeProduct(const QList<tcChannel*>& inputChannels);
  49
  50     /// Destructor.
  51     virtual ~tcNegativeProduct();
  52
  53     /*
  54      * Main image interface
  55      */
  56
  57     // Reimplemented
  58     virtual tcChannelConfigWidget* configWidget();
  59
  60     /*
  61      * Parameters
  62      */
  63
  64     /// Load parameters.
  65     void loadParams(const QString& filename);
  66
  67   protected:
  68
  69     /*
  70      * Interface for derived class to implement
  71      */
  72
  73     // Reimplemented
  74     virtual void roundPortion(double* x1, double* y1, double* x2, double* y2);
  75
  76     // Reimplemented
  77     virtual tcAbstractPixelData* loadPortion(double x1, double y1, double x2, double y2, bool changed);
  78
  79   signals:
  80
  81     /*
  82      * Signals
  83      */
  84
  85     /// Emitted when the threshold text changes.
  86     void thresholdChanged(const QString& threshold);
  87
  88   private slots:
  89
  90     /*
  91      * Private slots
  92      */
  93
  94     /// Set whether the theshold is enabled.
  95     void setThresholdEnabled(bool thresholdEnabled);
  96
  97     /// Move the theshold value slider.
  98     void moveThreshold(int threshold);
  99
 100     /// Set the theshold value.
 101     void setThreshold(int threshold);
 102
 103     /// Indicates that the power slider of a channel has moved.
 104     void movePower(int channel);
 105
 106     /// Indicates that the power of a channel has changed.
 107     void changePower(int channel);
 108
 109     /// Optimise the threshold for the parameters using a classification channel.
 110     void optimiseThreshold();
 111
 112     /// Save threshold errors.
 113     void saveThresholdErrors();
 114
 115     /// Optimise the parameters using a classification channel.
 116     void optimiseParameters();
 117
 118     /// Save parameters to a file.
 119     void saveParams();
 120
 121     /// Load parameters from a file.
 122     void loadParams();
 123
 124   private:
 125
 126     /*
 127      * Private functions
 128      */
 129
 130     /// Calculate an error using some channel as a classifier.
 131     float calculateError(tcChannel* classifier);
 132
 133   private:
 134
 135     /*
 136      * Variables
 137      */
 138
 139     /// Input channels.
 140     QList<tcChannel*> m_inputChannels;
 141
 142     /// Whether threshold is enabled.
 143     bool m_thresholdEnabled;
 144
 145     /// Threshold to classify shadows.
 146     float m_threshold;
 147
 148     /// Powers to raise the negative channels to.
 149     float* m_powers;
 150
 151     /// Filename in which to save threshold-error values.
 152     QString m_saveThresholdError;
 153
 154     /// Filename in which to save parameter-error values.
 155     QString m_saveParametersError;
 156
 157     /// Filename in which to save parameters.
 158     QString m_saveParameters;
 159
 160     /// Configuration widget.
 161     tcChannelConfigWidget* m_configWidget;
 162
 163     /// Threshold slider.
 164     QSlider* m_sliderThreshold;
 165
 166     /// Power sliders in config widget.
 167     QList<QSlider*> m_powerSliders;
 168
 169     /// Power sliders labelsin config widget.
 170     QList<QLabel*> m_powerLabels;
 171
 172     /// Classifier channel.
 173     QComboBox* m_comboClassifier;
 174
 175     /// Slider for log learning rate.
 176     QSlider* m_sliderLearningRate;
 177
 178     /// Label to show error.
 179     QLabel* m_labelError;
 180
 181     /// Last calculated optimal threshold error.
 182     int m_optimalThresholdError;
 183 };
 184
 185 #endif