geo/tcElevationOptimization.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 _tcElevationOptimization_h_
  21 #define _tcElevationOptimization_h_
  22
  23 /**
  24  * @file tcElevationOptimization.h
  25  * @brief Optimized elevation.
  26  */
  27
  28 #include "tcChannelDem.h"
  29 #include "tcElevationData.h"
  30
  31 class QSpinBox;
  32
  33 /// Optimized elevation.
  34 class tcElevationOptimization : public tcChannelDem
  35 {
  36     Q_OBJECT
  37
  38   public:
  39
  40     /*
  41      * Constructors + destructor
  42      */
  43
  44     /// Primary constructor.
  45     tcElevationOptimization(tcChannel* shadowChannel, tcChannel* shadingChannel, tcSrtmModel* dem, tcGeoImageData* imagery);
  46
  47     /// Destructor.
  48     virtual ~tcElevationOptimization();
  49
  50     /*
  51      * Main image interface
  52      */
  53
  54     // Reimplemented
  55     virtual tcChannelConfigWidget* configWidget();
  56
  57   private slots:
  58
  59     /*
  60      * Slots
  61      */
  62
  63     /// Reset DEM.
  64     void resetDem();
  65
  66     /// Load elevation data from DEM.
  67     void loadFromDem();
  68
  69     /// Optimize a number of times.
  70     void optimize();
  71
  72   protected:
  73
  74     /*
  75      * Interface for derived class to implement
  76      */
  77
  78     // Reimplemented
  79     virtual void roundPortion(double* x1, double* y1, double* x2, double* y2);
  80
  81     // Reimplemented
  82     virtual tcAbstractPixelData* loadPortion(double x1, double y1, double x2, double y2);
  83
  84   private:
  85
  86     /*
  87      * Private functions
  88      */
  89
  90     /** Calculate the cost for a set of pixels.
  91      * @param x1 Starting horizontal pixel (left).
  92      * @param y1 Starting vertical pixel (top).
  93      * @param x2 Ending horizontal pixel (right).
  94      * @param y2 Ending horizontal pixel (bottom).
  95      * @return Cost of the chosen pixels.
  96      */
  97     float cost(int x1, int y1, int x2, int y2) const;
  98
  99     /** Calculate the relative cost of changing a pixel's elevation.
 100      * @param x Horizontal coordinate of pixel to change.
 101      * @param y Vertical coordinate of pixel to change.
 102      * @param dh Relative change in elevation.
 103      * @return Relative cost of altering the elevation of the specified pixel.
 104      */
 105     float costChange(int x, int y, float dh);
 106
 107     /// Optimize a whole range of elevations.
 108     void optimizeElevations(int x1, int y1, int x2, int y2, float dh);
 109
 110     /// Get pointer to elevation at a pixel.
 111     float* pixelElevation(int x, int y) const;
 112
 113     /// Adjust elevation at a pixel.
 114     float adjustElevation(int x, int y, float dh, bool relative = false);
 115
 116     /*
 117      * Types
 118      */
 119
 120     /// Shadow transition types.
 121     enum TransitionType {
 122       TransitionEntrance = 0,
 123       TransitionExit = 1
 124     };
 125
 126     /// Per-pixel data to cache.
 127     struct PerPixelCache
 128     {
 129       /// Coordinates of shadow entrance and exit.
 130       maths::Vector<2,int> shadowTransition[2];
 131       /// Resolution.
 132       maths::Vector<2,float> resolution;
 133       /// Light direction.
 134       maths::Vector<3,float> light;
 135       /// Length of light vector trace along ground.
 136       float sinLightElevation;
 137       /// How lit pixel is.
 138       float lit;
 139       /// Original elevation.
 140       float originalElevation;
 141       /// Distance to shadow entrance and exit.
 142       float shadowTransitionDistance[2];
 143       /// Reliability of original elevation.
 144       unsigned char reliability;
 145     };
 146
 147     /*
 148      * Variables
 149      */
 150
 151     /// Shadow channel.
 152     tcChannel* m_shadowChannel;
 153
 154     /// Shading channel.
 155     tcChannel* m_shadingChannel;
 156
 157     /// Current elevation data.
 158     tcElevationData*   m_elevationData;
 159
 160     /// per-pixel data.
 161     tcTypedPixelData<PerPixelCache>* m_data;
 162
 163     /// Temporary shadow data.
 164     Reference<tcPixelData<float> > m_shadowData;
 165
 166     /// Temporary shading data.
 167     Reference<tcPixelData<float> > m_shadingData;
 168
 169     /// Weights.
 170     struct
 171     {
 172       float variance;
 173       float roughness;
 174       float steepness;
 175       float litFacing;
 176       float shading;
 177       float transitionElev;
 178       float transitionTangential;
 179       float transitionCurvingDown;
 180     } m_weights;
 181
 182     /*
 183      * Configuration gui
 184      */
 185
 186     /// Configuration widget.
 187     tcChannelConfigWidget* m_configWidget;
 188
 189     /// Number of iterations to perform.
 190     QSpinBox* m_spinIterations;
 191
 192 };
 193
 194 #endif